src/udev/udev-builtin.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <getopt.h>
   4 #include <stdio.h>
   5 #include <string.h>
   6
   7 #include "device-private.h"
   8 #include "device-util.h"
   9 #include "string-util.h"
  10 #include "strv.h"
  11 #include "udev-builtin.h"
  12
  13 static bool initialized;
  14
  15 static const struct udev_builtin *builtins[_UDEV_BUILTIN_MAX] = {
  16 #if HAVE_BLKID
  17         [UDEV_BUILTIN_BLKID] = &udev_builtin_blkid,
  18 #endif
  19         [UDEV_BUILTIN_BTRFS] = &udev_builtin_btrfs,
  20         [UDEV_BUILTIN_HWDB] = &udev_builtin_hwdb,
  21         [UDEV_BUILTIN_INPUT_ID] = &udev_builtin_input_id,
  22         [UDEV_BUILTIN_KEYBOARD] = &udev_builtin_keyboard,
  23 #if HAVE_KMOD
  24         [UDEV_BUILTIN_KMOD] = &udev_builtin_kmod,
  25 #endif
  26         [UDEV_BUILTIN_NET_ID] = &udev_builtin_net_id,
  27         [UDEV_BUILTIN_NET_LINK] = &udev_builtin_net_setup_link,
  28         [UDEV_BUILTIN_PATH_ID] = &udev_builtin_path_id,
  29         [UDEV_BUILTIN_USB_ID] = &udev_builtin_usb_id,
  30 #if HAVE_ACL
  31         [UDEV_BUILTIN_UACCESS] = &udev_builtin_uaccess,
  32 #endif
  33 };
  34
  35 void udev_builtin_init(void) {
  36         unsigned i;
  37
  38         if (initialized)
  39                 return;
  40
  41         for (i = 0; i < _UDEV_BUILTIN_MAX; i++)
  42                 if (builtins[i] && builtins[i]->init)
  43                         builtins[i]->init();
  44
  45         initialized = true;
  46 }
  47
  48 void udev_builtin_exit(void) {
  49         unsigned i;
  50
  51         if (!initialized)
  52                 return;
  53
  54         for (i = 0; i < _UDEV_BUILTIN_MAX; i++)
  55                 if (builtins[i] && builtins[i]->exit)
  56                         builtins[i]->exit();
  57
  58         initialized = false;
  59 }
  60
  61 bool udev_builtin_validate(void) {
  62         unsigned i;
  63
  64         for (i = 0; i < _UDEV_BUILTIN_MAX; i++)
  65                 if (builtins[i] && builtins[i]->validate && builtins[i]->validate())
  66                         return true;
  67         return false;
  68 }
  69
  70 void udev_builtin_list(void) {
  71         unsigned i;
  72
  73         for (i = 0; i < _UDEV_BUILTIN_MAX; i++)
  74                 if (builtins[i])
  75                         fprintf(stderr, "  %-14s  %s\n", builtins[i]->name, builtins[i]->help);
  76 }
  77
  78 const char *udev_builtin_name(enum udev_builtin_cmd cmd) {
  79         assert(cmd >= 0 && cmd < _UDEV_BUILTIN_MAX);
  80
  81         if (!builtins[cmd])
  82                 return NULL;
  83
  84         return builtins[cmd]->name;
  85 }
  86
  87 bool udev_builtin_run_once(enum udev_builtin_cmd cmd) {
  88         assert(cmd >= 0 && cmd < _UDEV_BUILTIN_MAX);
  89
  90         if (!builtins[cmd])
  91                 return false;
  92
  93         return builtins[cmd]->run_once;
  94 }
  95
  96 enum udev_builtin_cmd udev_builtin_lookup(const char *command) {
  97         enum udev_builtin_cmd i;
  98         size_t n;
  99
 100         assert(command);
 101
 102         command += strspn(command, WHITESPACE);
 103         n = strcspn(command, WHITESPACE);
 104         for (i = 0; i < _UDEV_BUILTIN_MAX; i++)
 105                 if (builtins[i] && strneq(builtins[i]->name, command, n))
 106                         return i;
 107
 108         return _UDEV_BUILTIN_INVALID;
 109 }
 110
 111 int udev_builtin_run(sd_device *dev, enum udev_builtin_cmd cmd, const char *command, bool test) {
 112         _cleanup_strv_free_ char **argv = NULL;
 113
 114         assert(dev);
 115         assert(cmd >= 0 && cmd < _UDEV_BUILTIN_MAX);
 116         assert(command);
 117
 118         if (!builtins[cmd])
 119                 return -EOPNOTSUPP;
 120
 121         argv = strv_split_full(command, NULL, SPLIT_QUOTES | SPLIT_RELAX);
 122         if (!argv)
 123                 return -ENOMEM;
 124
 125         /* we need '0' here to reset the internal state */
 126         optind = 0;
 127         return builtins[cmd]->cmd(dev, strv_length(argv), argv, test);
 128 }
 129
 130 int udev_builtin_add_property(sd_device *dev, bool test, const char *key, const char *val) {
 131         int r;
 132
 133         assert(dev);
 134         assert(key);
 135
 136         r = device_add_property(dev, key, val);
 137         if (r < 0)
 138                 return log_device_debug_errno(dev, r, "Failed to add property '%s%s%s'",
 139                                               key, val ? "=" : "", strempty(val));
 140
 141         if (test)
 142                 printf("%s=%s\n", key, val);
 143
 144         return 0;
 145 }