chiark / gitweb /
release 105
[elogind.git] / udevtrigger.c
index bc8453a82a9c2566a78aaa1ac32a8e7c4e864135..c809d2d47bda67cd696e8bcb53d6620155840402 100644 (file)
@@ -1,6 +1,4 @@
 /*
- * udevtrigger.c
- *
  * Copyright (C) 2004-2006 Kay Sievers <kay@vrfy.org>
  * Copyright (C) 2006 Hannes Reinecke <hare@suse.de>
  *
@@ -15,7 +13,7 @@
  * 
  *     You should have received a copy of the GNU General Public License along
  *     with this program; if not, write to the Free Software Foundation, Inc.,
- *     675 Mass Ave, Cambridge, MA 02139, USA.
+ *     51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  *
  */
 
 #include <string.h>
 #include <stdio.h>
 #include <unistd.h>
+#include <getopt.h>
 #include <errno.h>
 #include <dirent.h>
 #include <fcntl.h>
 #include <syslog.h>
+#include <fnmatch.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 
 #include "udev.h"
+#include "udevd.h"
 
-static const char *udev_log_str;
 static int verbose;
 static int dry_run;
+LIST_HEAD(device_list);
+LIST_HEAD(filter_subsystem_match_list);
+LIST_HEAD(filter_subsystem_nomatch_list);
+LIST_HEAD(filter_attr_match_list);
+LIST_HEAD(filter_attr_nomatch_list);
 
 #ifdef USE_LOG
 void log_message(int priority, const char *format, ...)
@@ -51,60 +56,62 @@ void log_message(int priority, const char *format, ...)
 }
 #endif
 
-/* list of devices that we should run last due to any one of a number of reasons */
-static char *last_list[] = {
-       "/class/block/md",
-       "/class/block/dm-",
-       "/block/md",
-       "/block/dm-",
-       NULL
-};
-
-/* list of devices that we should run first due to any one of a number of reasons */
-static char *first_list[] = {
-       "/class/mem",
-       "/class/tty",
-       NULL
-};
-
-LIST_HEAD(device_first_list);
-LIST_HEAD(device_default_list);
-LIST_HEAD(device_last_list);
-
-static int device_list_insert(const char *path)
+/* devices that should run last cause of their dependencies */
+static int delay_device(const char *devpath)
 {
-       struct list_head *device_list = &device_default_list;
+       static const char *delay_device_list[] = {
+               "*/md*",
+               "*/dm-*",
+               NULL
+       };
        int i;
 
-       for (i = 0; first_list[i] != NULL; i++) {
-               if (strncmp(path, first_list[i], strlen(first_list[i])) == 0) {
-                       device_list = &device_first_list;
-                       break;
-               }
-       }
-       for (i = 0; last_list[i] != NULL; i++) {
-               if (strncmp(path, last_list[i], strlen(last_list[i])) == 0) {
-                       device_list = &device_last_list;
-                       break;
-               }
-       }
+       for (i = 0; delay_device_list[i] != NULL; i++)
+               if (fnmatch(delay_device_list[i], devpath, 0) == 0)
+                       return 1;
+       return 0;
+}
+
+static int device_list_insert(const char *path)
+{
+       char filename[PATH_SIZE];
+       char devpath[PATH_SIZE];
+       struct stat statbuf;
 
        dbg("add '%s'" , path);
-       /* double entries will be ignored */
-       name_list_add(device_list, path, 0);
+
+       /* we only have a device, if we have an uevent file */
+       strlcpy(filename, path, sizeof(filename));
+       strlcat(filename, "/uevent", sizeof(filename));
+       if (stat(filename, &statbuf) < 0)
+               return -1;
+       if (!(statbuf.st_mode & S_IWUSR))
+               return -1;
+
+       strlcpy(devpath, &path[strlen(sysfs_path)], sizeof(devpath));
+
+       /* resolve possible link to real target */
+       if (lstat(path, &statbuf) < 0)
+               return -1;
+       if (S_ISLNK(statbuf.st_mode))
+               if (sysfs_resolve_link(devpath, sizeof(devpath)) != 0)
+                       return -1;
+
+       name_list_add(&device_list, devpath, 1);
        return 0;
 }
 
-static void trigger_uevent(const char *path)
+static void trigger_uevent(const char *devpath)
 {
        char filename[PATH_SIZE];
        int fd;
 
-       strlcpy(filename, path, sizeof(filename));
+       strlcpy(filename, sysfs_path, sizeof(filename));
+       strlcat(filename, devpath, sizeof(filename));
        strlcat(filename, "/uevent", sizeof(filename));
 
        if (verbose)
-               printf("%s\n", path);
+               printf("%s\n", devpath);
 
        if (dry_run)
                return;
@@ -121,58 +128,124 @@ static void trigger_uevent(const char *path)
        close(fd);
 }
 
-static void exec_lists(void)
+static void exec_list(void)
 {
        struct name_entry *loop_device;
        struct name_entry *tmp_device;
 
-       /* handle the devices on the "first" list first */
-       list_for_each_entry_safe(loop_device, tmp_device, &device_first_list, node) {
+       list_for_each_entry_safe(loop_device, tmp_device, &device_list, node) {
+               if (delay_device(loop_device->name))
+                       continue;
+
                trigger_uevent(loop_device->name);
                list_del(&loop_device->node);
                free(loop_device);
        }
 
-       /* handle the devices on the "default" list next */
-       list_for_each_entry_safe(loop_device, tmp_device, &device_default_list, node) {
+       /* trigger remaining delayed devices */
+       list_for_each_entry_safe(loop_device, tmp_device, &device_list, node) {
                trigger_uevent(loop_device->name);
                list_del(&loop_device->node);
                free(loop_device);
        }
+}
 
-       /* handle devices on the "last" list, if any */
-       list_for_each_entry_safe(loop_device, tmp_device, &device_last_list, node) {
-               trigger_uevent(loop_device->name);
-               list_del(&loop_device->node);
-               free(loop_device);
+static int subsystem_filtered(const char *subsystem)
+{
+       struct name_entry *loop_name;
+
+       /* skip devices matching the listed subsystems */
+       list_for_each_entry(loop_name, &filter_subsystem_nomatch_list, node)
+               if (fnmatch(loop_name->name, subsystem, 0) == 0)
+                       return 1;
+
+       /* skip devices not matching the listed subsystems */
+       if (!list_empty(&filter_subsystem_match_list)) {
+               list_for_each_entry(loop_name, &filter_subsystem_match_list, node)
+                       if (fnmatch(loop_name->name, subsystem, 0) == 0)
+                               return 0;
+               return 1;
        }
+
+       return 0;
 }
 
-static int is_device(const char *path)
+static int attr_match(const char *path, const char *attr_value)
 {
-       char filename[PATH_SIZE];
-       struct stat statbuf;
+       char attr[NAME_SIZE];
+       char file[PATH_SIZE];
+       char *match_value;
 
-       /* look for the uevent file of the kobject */
-       strlcpy(filename, path, sizeof(filename));
-       strlcat(filename, "/uevent", sizeof(filename));
-       if (stat(filename, &statbuf) < 0)
-               return 0;
+       strlcpy(attr, attr_value, sizeof(attr));
 
-       if (!(statbuf.st_mode & S_IWUSR))
-               return 0;
+       /* separate attr and match value */
+       match_value = strchr(attr, '=');
+       if (match_value != NULL) {
+               match_value[0] = '\0';
+               match_value = &match_value[1];
+       }
+
+       strlcpy(file, path, sizeof(file));
+       strlcat(file, "/", sizeof(file));
+       strlcat(file, attr, sizeof(file));
+
+       if (match_value != NULL) {
+               /* match file content */
+               char value[NAME_SIZE];
+               int fd;
+               ssize_t size;
+
+               fd = open(file, O_RDONLY);
+               if (fd < 0)
+                       return 0;
+               size = read(fd, value, sizeof(value));
+               close(fd);
+               if (size < 0)
+                       return 0;
+               value[size] = '\0';
+               remove_trailing_chars(value, '\n');
+
+               /* match if attribute value matches */
+               if (fnmatch(match_value, value, 0) == 0)
+                       return 1;
+       } else {
+               /* match if attribute exists */
+               struct stat statbuf;
+
+               if (stat(file, &statbuf) == 0)
+                       return 1;
+       }
+       return 0;
+}
 
-       return 1;
+static int attr_filtered(const char *path)
+{
+       struct name_entry *loop_name;
+
+       /* skip devices matching the listed sysfs attributes */
+       list_for_each_entry(loop_name, &filter_attr_nomatch_list, node)
+               if (attr_match(path, loop_name->name))
+                       return 1;
+
+       /* skip devices not matching the listed sysfs attributes */
+       if (!list_empty(&filter_attr_match_list)) {
+               list_for_each_entry(loop_name, &filter_attr_match_list, node)
+                       if (attr_match(path, loop_name->name))
+                               return 0;
+               return 1;
+       }
+       return 0;
 }
 
-static void udev_scan_bus(void)
+static void scan_subsystem(const char *subsys)
 {
        char base[PATH_SIZE];
        DIR *dir;
        struct dirent *dent;
 
        strlcpy(base, sysfs_path, sizeof(base));
-       strlcat(base, "/bus", sizeof(base));
+       strlcat(base, "/", sizeof(base));
+       strlcat(base, subsys, sizeof(base));
 
        dir = opendir(base);
        if (dir != NULL) {
@@ -184,6 +257,9 @@ static void udev_scan_bus(void)
                        if (dent->d_name[0] == '.')
                                continue;
 
+                       if (subsystem_filtered(dent->d_name))
+                               continue;
+
                        strlcpy(dirname, base, sizeof(dirname));
                        strlcat(dirname, "/", sizeof(dirname));
                        strlcat(dirname, dent->d_name, sizeof(dirname));
@@ -201,9 +277,9 @@ static void udev_scan_bus(void)
                                        strlcpy(dirname2, dirname, sizeof(dirname2));
                                        strlcat(dirname2, "/", sizeof(dirname2));
                                        strlcat(dirname2, dent2->d_name, sizeof(dirname2));
-
-                                       if (is_device(dirname2))
-                                               device_list_insert(dirname2);
+                                       if (attr_filtered(dirname2))
+                                               continue;
+                                       device_list_insert(dirname2);
                                }
                                closedir(dir2);
                        }
@@ -212,17 +288,13 @@ static void udev_scan_bus(void)
        }
 }
 
-static void udev_scan_block(void)
+static void scan_block(void)
 {
        char base[PATH_SIZE];
        DIR *dir;
        struct dirent *dent;
-       struct stat statbuf;
 
-       /* skip if "block" is already a "class" */
-       strlcpy(base, sysfs_path, sizeof(base));
-       strlcat(base, "/class/block", sizeof(base));
-       if (stat(base, &statbuf) == 0)
+       if (subsystem_filtered("block"))
                return;
 
        strlcpy(base, sysfs_path, sizeof(base));
@@ -241,9 +313,9 @@ static void udev_scan_block(void)
                        strlcpy(dirname, base, sizeof(dirname));
                        strlcat(dirname, "/", sizeof(dirname));
                        strlcat(dirname, dent->d_name, sizeof(dirname));
-                       if (is_device(dirname))
-                               device_list_insert(dirname);
-                       else
+                       if (attr_filtered(dirname))
+                               continue;
+                       if (device_list_insert(dirname) != 0)
                                continue;
 
                        /* look for partitions */
@@ -261,8 +333,9 @@ static void udev_scan_block(void)
                                        strlcpy(dirname2, dirname, sizeof(dirname2));
                                        strlcat(dirname2, "/", sizeof(dirname2));
                                        strlcat(dirname2, dent2->d_name, sizeof(dirname2));
-                                       if (is_device(dirname2))
-                                               device_list_insert(dirname2);
+                                       if (attr_filtered(dirname2))
+                                               continue;
+                                       device_list_insert(dirname2);
                                }
                                closedir(dir2);
                        }
@@ -271,7 +344,7 @@ static void udev_scan_block(void)
        }
 }
 
-static void udev_scan_class(void)
+static void scan_class(void)
 {
        char base[PATH_SIZE];
        DIR *dir;
@@ -290,6 +363,9 @@ static void udev_scan_class(void)
                        if (dent->d_name[0] == '.')
                                continue;
 
+                       if (subsystem_filtered(dent->d_name))
+                               continue;
+
                        strlcpy(dirname, base, sizeof(dirname));
                        strlcat(dirname, "/", sizeof(dirname));
                        strlcat(dirname, dent->d_name, sizeof(dirname));
@@ -307,8 +383,9 @@ static void udev_scan_class(void)
                                        strlcpy(dirname2, dirname, sizeof(dirname2));
                                        strlcat(dirname2, "/", sizeof(dirname2));
                                        strlcat(dirname2, dent2->d_name, sizeof(dirname2));
-                                       if (is_device(dirname2))
-                                               device_list_insert(dirname2);
+                                       if (attr_filtered(dirname2))
+                                               continue;
+                                       device_list_insert(dirname2);
                                }
                                closedir(dir2);
                        }
@@ -317,38 +394,140 @@ static void udev_scan_class(void)
        }
 }
 
+static void scan_failed(void)
+{
+       char base[PATH_SIZE];
+       DIR *dir;
+       struct dirent *dent;
+
+       strlcpy(base, udev_root, sizeof(base));
+       strlcat(base, "/" EVENT_FAILED_DIR, sizeof(base));
+
+       dir = opendir(base);
+       if (dir != NULL) {
+               for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
+                       char device[PATH_SIZE];
+                       size_t start, end, i;
+
+                       if (dent->d_name[0] == '.')
+                               continue;
+
+                       strlcpy(device, sysfs_path, sizeof(device));
+                       start = strlcat(device, "/", sizeof(device));
+                       end = strlcat(device, dent->d_name, sizeof(device));
+                       if (end > sizeof(device))
+                               end = sizeof(device);
+
+                       /* replace PATH_TO_NAME_CHAR with '/' */
+                       for (i = start; i < end; i++)
+                               if (device[i] == PATH_TO_NAME_CHAR)
+                                       device[i] = '/';
+
+                       device_list_insert(device);
+               }
+               closedir(dir);
+       }
+}
+
 int main(int argc, char *argv[], char *envp[])
 {
-       int i;
+       int failed = 0;
+       int option;
+       static const struct option options[] = {
+               { "verbose", 0, NULL, 'v' },
+               { "dry-run", 0, NULL, 'n' },
+               { "retry-failed", 0, NULL, 'F' },
+               { "help", 0, NULL, 'h' },
+               { "subsystem-match", 1, NULL, 's' },
+               { "subsystem-nomatch", 1, NULL, 'S' },
+               { "attr-match", 1, NULL, 'a' },
+               { "attr-nomatch", 1, NULL, 'A' },
+               {}
+       };
 
        logging_init("udevtrigger");
        udev_config_init();
        dbg("version %s", UDEV_VERSION);
+       sysfs_init();
 
-       udev_log_str = getenv("UDEV_LOG");
-
-       for (i = 1 ; i < argc; i++) {
-               char *arg = argv[i];
+       while (1) {
+               option = getopt_long(argc, argv, "vnFhs:S:a:A:", options, NULL);
+               if (option == -1)
+                       break;
 
-               if (strcmp(arg, "--verbose") == 0 || strcmp(arg, "-v") == 0)
+               switch (option) {
+               case 'v':
                        verbose = 1;
-               else if (strcmp(arg, "--dry-run") == 0 || strcmp(arg, "-n") == 0)
+                       break;
+               case 'n':
                        dry_run = 1;
-               else {
-                       fprintf(stderr, "Usage: udevtrigger [--verbose] [--dry-run]\n");
+                       break;
+               case 'F':
+                       failed = 1;
+                       break;
+               case 's':
+                       name_list_add(&filter_subsystem_match_list, optarg, 0);
+                       break;
+               case 'S':
+                       name_list_add(&filter_subsystem_nomatch_list, optarg, 0);
+                       break;
+               case 'a':
+                       name_list_add(&filter_attr_match_list, optarg, 0);
+                       break;
+               case 'A':
+                       name_list_add(&filter_attr_nomatch_list, optarg, 0);
+                       break;
+               case 'h':
+                       printf("Usage: udevtrigger OPTIONS\n"
+                              "  --verbose                       print the list of devices while running\n"
+                              "  --dry-run                       do not actually trigger the events\n"
+                              "  --retry-failed                  trigger only the events which have been\n"
+                              "                                  marked as failed during a previous run\n"
+                              "  --subsystem-match=<subsystem>   trigger devices from a matching subystem\n"
+                              "  --subsystem-nomatch=<subsystem> exclude devices from a matching subystem\n"
+                              "  --attr-match=<file[=<value>]>   trigger devices with a matching sysfs\n"
+                              "                                  attribute\n"
+                              "  --attr-nomatch=<file[=<value>]> exclude devices with a matching sysfs\n"
+                              "                                  attribute\n"
+                              "  --help                          print this text\n"
+                              "\n");
+                       goto exit;
+               default:
                        goto exit;
                }
        }
 
-       sysfs_init();
+       if (failed)
+               scan_failed();
+       else {
+               char base[PATH_SIZE];
+               struct stat statbuf;
+
+               /* if we have /sys/subsystem, forget all the old stuff */
+               strlcpy(base, sysfs_path, sizeof(base));
+               strlcat(base, "/subsystem", sizeof(base));
+               if (stat(base, &statbuf) == 0)
+                       scan_subsystem("subsystem");
+               else {
+                       scan_subsystem("bus");
+                       scan_class();
+
+                       /* scan "block" if it isn't a "class" */
+                       strlcpy(base, sysfs_path, sizeof(base));
+                       strlcat(base, "/class/block", sizeof(base));
+                       if (stat(base, &statbuf) != 0)
+                               scan_block();
+               }
+       }
+       exec_list();
 
-       udev_scan_bus();
-       udev_scan_class();
-       udev_scan_block();
-       exec_lists();
+exit:
+       name_list_cleanup(&filter_subsystem_match_list);
+       name_list_cleanup(&filter_subsystem_nomatch_list);
+       name_list_cleanup(&filter_attr_match_list);
+       name_list_cleanup(&filter_attr_nomatch_list);
 
        sysfs_cleanup();
-exit:
        logging_close();
        return 0;
 }