chiark / gitweb /
libudev: enumerate - split new() and scan()
[elogind.git] / udev / lib / libudev-enumerate.c
index fc853b8d631098adb79091e5bb30e8fa03760d32..266164ae517ebfebe4da97cc34b9af44749b7728 100644 (file)
@@ -54,7 +54,14 @@ void udev_enumerate_unref(struct udev_enumerate *udev_enumerate)
        free(udev_enumerate);
 }
 
-struct udev_list *udev_enumerate_get_list(struct udev_enumerate *udev_enumerate)
+struct udev *udev_enumerate_get_udev(struct udev_enumerate *udev_enumerate)
+{
+       if (udev_enumerate == NULL)
+               return NULL;
+       return udev_enumerate->udev;
+}
+
+struct udev_list_entry *udev_enumerate_get_list_entry(struct udev_enumerate *udev_enumerate)
 {
        if (udev_enumerate == NULL)
                return NULL;
@@ -63,7 +70,7 @@ struct udev_list *udev_enumerate_get_list(struct udev_enumerate *udev_enumerate)
 
 static int devices_scan_subsystem(struct udev *udev,
                                  const char *basedir, const char *subsystem, const char *subdir,
-                                 struct list_node *device_list)
+                                 struct list_node *devices_list)
 {
        char path[UTIL_PATH_SIZE];
        DIR *dir;
@@ -71,8 +78,10 @@ static int devices_scan_subsystem(struct udev *udev,
 
        util_strlcpy(path, udev_get_sys_path(udev), sizeof(path));
        util_strlcat(path, basedir, sizeof(path));
-       util_strlcat(path, "/", sizeof(path));
-       util_strlcat(path, subsystem, sizeof(path));
+       if (subsystem != NULL) {
+               util_strlcat(path, "/", sizeof(path));
+               util_strlcat(path, subsystem, sizeof(path));
+       }
        if (subdir != NULL)
                util_strlcat(path, subdir, sizeof(path));
        dir = opendir(path);
@@ -80,41 +89,60 @@ static int devices_scan_subsystem(struct udev *udev,
                return -1;
        for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
                char syspath[UTIL_PATH_SIZE];
+               char filename[UTIL_PATH_SIZE];
+               struct stat statbuf;
 
                if (dent->d_name[0] == '.')
                        continue;
                util_strlcpy(syspath, path, sizeof(syspath));
                util_strlcat(syspath, "/", sizeof(syspath));
                util_strlcat(syspath, dent->d_name, sizeof(syspath));
+               util_strlcpy(filename, syspath, sizeof(filename));
+               util_strlcat(filename, "/uevent", sizeof(filename));
+               if (stat(filename, &statbuf) != 0)
+                       continue;
                util_resolve_sys_link(udev, syspath, sizeof(syspath));
-               list_insert_entry(udev, device_list, syspath, NULL, 1);
+               list_entry_add(udev, devices_list, syspath, NULL, 1, 1);
        }
        closedir(dir);
        return 0;
 }
 
 static int devices_scan_subsystems(struct udev *udev,
-                                  const char *basedir, const char *subsystem, const char *subdir,
-                                  struct list_node *device_list)
+                                  const char *basedir, const char *subdir,
+                                  struct udev_list_entry *subsystem_include_list,
+                                  struct udev_list_entry *subsystem_exclude_list,
+                                  struct list_node *devices_list)
 {
-       char path[UTIL_PATH_SIZE];
-       DIR *dir;
-       struct dirent *dent;
+       if (subsystem_include_list != NULL) {
+               struct udev_list_entry *list_entry;
 
-       if (subsystem != NULL)
-               return devices_scan_subsystem(udev, basedir, subsystem, subdir, device_list);
+               /* if list of subsystems to scan is given, just use this list */
+               udev_list_entry_foreach(list_entry, subsystem_include_list) {
+                       if (udev_list_entry_get_by_name(subsystem_exclude_list, udev_list_entry_get_name(list_entry)) != NULL)
+                                       continue;
+                       devices_scan_subsystem(udev, basedir, udev_list_entry_get_name(list_entry), subdir, devices_list);
+               }
+       } else {
+               char path[UTIL_PATH_SIZE];
+               DIR *dir;
+               struct dirent *dent;
 
-       util_strlcpy(path, udev_get_sys_path(udev), sizeof(path));
-       util_strlcat(path, basedir, sizeof(path));
-       dir = opendir(path);
-       if (dir == NULL)
-               return -1;
-       for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
-               if (dent->d_name[0] == '.')
-                       continue;
-               devices_scan_subsystem(udev, basedir, dent->d_name, subdir, device_list);
+               /* if no list of subsystems to scan is given, scan all, and possible exclude some subsystems */
+               util_strlcpy(path, udev_get_sys_path(udev), sizeof(path));
+               util_strlcat(path, basedir, sizeof(path));
+               dir = opendir(path);
+               if (dir == NULL)
+                       return -1;
+               for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
+                       if (dent->d_name[0] == '.')
+                               continue;
+                       if (udev_list_entry_get_by_name(subsystem_exclude_list, dent->d_name) != NULL)
+                                       continue;
+                       devices_scan_subsystem(udev, basedir, dent->d_name, subdir, devices_list);
+               }
+               closedir(dir);
        }
-       closedir(dir);
        return 0;
 }
 
@@ -140,21 +168,14 @@ static int devices_delay(struct udev *udev, const char *syspath)
 }
 
 /**
- * udev_enumerate_new_from_subsystems:
+ * udev_enumerate_new:
  * @udev: udev library context
- * @subsystem: the subsystem to enumerate
  *
  * Returns: an enumeration context
  **/
-struct udev_enumerate *udev_enumerate_new_from_subsystems(struct udev *udev, const char *subsystem)
+struct udev_enumerate *udev_enumerate_new(struct udev *udev)
 {
        struct udev_enumerate *udev_enumerate;
-       char base[UTIL_PATH_SIZE];
-       struct stat statbuf;
-       struct udev_list *list;
-
-       if (udev == NULL)
-               return NULL;
 
        udev_enumerate = malloc(sizeof(struct udev_enumerate));
        if (udev_enumerate == NULL)
@@ -163,23 +184,121 @@ struct udev_enumerate *udev_enumerate_new_from_subsystems(struct udev *udev, con
        udev_enumerate->refcount = 1;
        udev_enumerate->udev = udev;
        list_init(&udev_enumerate->devices_list);
+       return udev_enumerate;
+}
+
+/**
+ * udev_enumerate_scan_devices:
+ * @udev_enumerate: udev enumeration context
+ * @subsystem: the list of names of subsystems to look for devices
+ *
+ * Returns: 0 on success.
+ **/
+int udev_enumerate_scan_devices(struct udev_enumerate *udev_enumerate, const char *subsystem, ...)
+{
+       struct udev *udev = udev_enumerate_get_udev(udev_enumerate);
+       va_list vargs;
+       const char *arg;
+       char base[UTIL_PATH_SIZE];
+       struct stat statbuf;
+       struct list_node subsystem_include_list;
+       struct list_node subsystem_exclude_list;
+       struct udev_list_entry *list_entry;
+
+       if (udev_enumerate == NULL)
+               return -EINVAL;
+
+       va_start(vargs, subsystem);
+       list_init(&subsystem_include_list);
+       list_init(&subsystem_exclude_list);
+       for (arg = subsystem; arg != NULL; arg = va_arg(vargs, const char *)) {
+               if (arg[0] != '!')
+                       list_entry_add(udev, &subsystem_include_list, arg, NULL, 1, 0);
+               else
+                       list_entry_add(udev, &subsystem_exclude_list, &arg[1], NULL, 1, 0);
+       }
+       va_end(vargs);
 
-       /* if we have /sys/subsystem/, forget all the old stuff */
        util_strlcpy(base, udev_get_sys_path(udev), sizeof(base));
        util_strlcat(base, "/subsystem", sizeof(base));
        if (stat(base, &statbuf) == 0) {
-               devices_scan_subsystems(udev, "/subsystem", subsystem, "/devices", &udev_enumerate->devices_list);
+               /* we have /subsystem/, forget all the old stuff */
+               info(udev, "searching '/subsystem/*/devices/*' dir\n");
+               devices_scan_subsystems(udev, "/subsystem", "/devices",
+                                       list_get_entry(&subsystem_include_list),
+                                       list_get_entry(&subsystem_exclude_list),
+                                       &udev_enumerate->devices_list);
        } else {
-               devices_scan_subsystems(udev, "/bus", subsystem, "/devices", &udev_enumerate->devices_list);
-               devices_scan_subsystems(udev, "/class", subsystem, NULL, &udev_enumerate->devices_list);
+               info(udev, "searching '/bus/*/devices/*' dir\n");
+               devices_scan_subsystems(udev, "/bus", "/devices",
+                                       list_get_entry(&subsystem_include_list),
+                                       list_get_entry(&subsystem_exclude_list),
+                                       &udev_enumerate->devices_list);
+               info(udev, "searching '/class/*' dir\n");
+               devices_scan_subsystems(udev, "/class", NULL,
+                                       list_get_entry(&subsystem_include_list),
+                                       list_get_entry(&subsystem_exclude_list),
+                                       &udev_enumerate->devices_list);
+               /* if block isn't a class, scan /block/ */
+               util_strlcpy(base, udev_get_sys_path(udev), sizeof(base));
+               util_strlcat(base, "/class/block", sizeof(base));
+               if (stat(base, &statbuf) != 0) {
+                       struct udev_list_entry *include_list = list_get_entry(&subsystem_include_list);
+                       struct udev_list_entry *exclude_list = list_get_entry(&subsystem_exclude_list);
+                       int include_block = (include_list == NULL || udev_list_entry_get_by_name(include_list, "block") != NULL);
+                       int exclude_block = (udev_list_entry_get_by_name(exclude_list, "block") != NULL);
+
+                       if (include_block && !exclude_block) {
+                               info(udev, "searching '/block/*' dir\n");
+                               /* scan disks */
+                               devices_scan_subsystem(udev, "/block", NULL, NULL, &udev_enumerate->devices_list);
+                               /* scan partitions */
+                               info(udev, "searching '/block/*/*' dir\n");
+                               devices_scan_subsystems(udev, "/block", NULL,
+                                                       NULL, NULL,
+                                                       &udev_enumerate->devices_list);
+                       }
+               }
        }
 
+       list_cleanup(udev, &subsystem_include_list);
+       list_cleanup(udev, &subsystem_exclude_list);
+
        /* sort delayed devices to the end of the list */
-       list = list_get_entry(&udev_enumerate->devices_list);
-       while (list != NULL) {
-               if (devices_delay(udev, udev_list_entry_get_name(list)))
-                       list_move_entry_to_end(list, &udev_enumerate->devices_list);
-               list = udev_list_entry_get_next(list);
+       udev_list_entry_foreach(list_entry, list_get_entry(&udev_enumerate->devices_list)) {
+               if (devices_delay(udev, udev_list_entry_get_name(list_entry)))
+                       list_entry_move_to_end(list_entry);
        }
-       return udev_enumerate;
+       return 0;
+}
+
+/**
+ * udev_enumerate_scan_subsystems:
+ * @udev_enumerate: udev enumeration context
+ *
+ * Returns: 0 on success.
+ **/
+int udev_enumerate_scan_subsystems(struct udev_enumerate *udev_enumerate)
+{
+       struct udev *udev = udev_enumerate_get_udev(udev_enumerate);
+       char base[UTIL_PATH_SIZE];
+       struct stat statbuf;
+       const char *subsysdir;
+
+       if (udev_enumerate == NULL)
+               return -EINVAL;
+
+       util_strlcpy(base, udev_get_sys_path(udev), sizeof(base));
+       util_strlcat(base, "/subsystem", sizeof(base));
+       if (stat(base, &statbuf) == 0)
+               subsysdir = "/subsystem";
+       else
+               subsysdir = "/bus";
+       info(udev, "searching '%s/*' dir\n", subsysdir);
+       devices_scan_subsystem(udev, subsysdir, NULL, NULL, &udev_enumerate->devices_list);
+       info(udev, "searching '%s/*/drivers/*' dir\n", subsysdir);
+       devices_scan_subsystems(udev, subsysdir, "/drivers",
+                               NULL, NULL,
+                               &udev_enumerate->devices_list);
+       return 0;
 }