chiark / gitweb /
[PATCH] udev - safer string handling - part four
[elogind.git] / namedev.c
index 59ec08c715e58bdc92a52f669ecc86b8845e6e1e..21f52d1f3cfa31811110997d762be77e12744a1d 100644 (file)
--- a/namedev.c
+++ b/namedev.c
 #include <sys/wait.h>
 #include <sys/stat.h>
 
+#include "libsysfs/sysfs/libsysfs.h"
 #include "list.h"
 #include "udev.h"
 #include "udev_version.h"
 #include "logging.h"
 #include "namedev.h"
-#include "libsysfs/libsysfs.h"
 #include "klibc_fixups.h"
 
+static struct sysfs_attribute *find_sysfs_attribute(struct sysfs_class_device *class_dev, struct sysfs_device *sysfs_device, char *attr);
+
 LIST_HEAD(config_device_list);
 LIST_HEAD(perm_device_list);
 
@@ -112,7 +114,7 @@ int add_perm_dev(struct perm_device *new_dev)
 
        /* update the values if we already have the device */
        list_for_each_entry(dev, &perm_device_list, node) {
-               if (strcmp_pattern(new_dev->name, dev->name))
+               if (strcmp(new_dev->name, dev->name))
                        continue;
                copy_var(dev, new_dev, mode);
                copy_string(dev, new_dev, owner);
@@ -155,7 +157,7 @@ static mode_t get_default_mode(void)
 static char *get_default_owner(void)
 {
        if (strlen(default_owner_str) == 0)
-               strncpy(default_owner_str, "root", OWNER_SIZE);
+               strfieldcpy(default_owner_str, "root");
 
        return default_owner_str;
 }
@@ -163,12 +165,53 @@ static char *get_default_owner(void)
 static char *get_default_group(void)
 {
        if (strlen(default_group_str) == 0)
-               strncpy(default_group_str, "root", GROUP_SIZE);
+               strfieldcpy(default_group_str, "root");
 
        return default_group_str;
 }
 
-static void apply_format(struct udevice *udev, unsigned char *string)
+/* extract possible {attr} and move str behind it */
+static char *get_format_attribute(char **str)
+{
+       char *pos;
+       char *attr = NULL;
+
+       if (*str[0] == '{') {
+               pos = strchr(*str, '}');
+               if (pos == NULL) {
+                       dbg("missing closing brace for format");
+                       return NULL;
+               }
+               pos[0] = '\0';
+               attr = *str+1;
+               *str = pos+1;
+               dbg("attribute='%s', str='%s'", attr, *str);
+       }
+       return attr;
+}
+
+/* extract possible format length and move str behind it*/
+static int get_format_len(char **str)
+{
+       int num;
+       char *tail;
+
+       if (isdigit(*str[0])) {
+               num = (int) strtoul(*str, &tail, 10);
+               if (tail != NULL) {
+                       *str = tail;
+                       dbg("format length=%i", num);
+                       return num;
+               } else {
+                       dbg("format parsing error '%s'", *str);
+               }
+       }
+       return -1;
+}
+
+static void apply_format(struct udevice *udev, char *string, size_t maxsize,
+                        struct sysfs_class_device *class_dev,
+                        struct sysfs_device *sysfs_device)
 {
        char temp[NAME_SIZE];
        char temp1[NAME_SIZE];
@@ -176,96 +219,119 @@ static void apply_format(struct udevice *udev, unsigned char *string)
        char *pos;
        char *pos2;
        char *pos3;
+       char *attr;
        int num;
+       int i;
+       char c;
+       struct sysfs_attribute *tmpattr;
 
        pos = string;
-       while (1) {
-               num = 0;
-               pos = strchr(pos, '%');
 
-               if (pos) {
+       while (1) {
+               pos = strchr(string, '%');
+               if (pos != NULL) {
                        pos[0] = '\0';
                        tail = pos+1;
-                       if (isdigit(tail[0])) {
-                               num = (int) strtoul(&pos[1], &tail, 10);
-                               if (tail == NULL) {
-                                       dbg("format parsing error '%s'", pos+1);
-                                       break;
-                               }
-                       }
+                       num = get_format_len(&tail);
+                       c = tail[0];
                        strfieldcpy(temp, tail+1);
+                       tail = temp;
+               } else {
+                       break;
+               }
+               dbg("format=%c, string='%s', tail='%s'",c , string, tail);
 
-                       switch (tail[0]) {
-                       case 'b':
-                               if (strlen(udev->bus_id) == 0)
-                                       break;
-                               strcat(pos, udev->bus_id);
-                               dbg("substitute bus_id '%s'", udev->bus_id);
+               attr = get_format_attribute(&tail);
+
+               switch (c) {
+               case 'b':
+                       if (strlen(udev->bus_id) == 0)
                                break;
-                       case 'k':
-                               if (strlen(udev->kernel_name) == 0)
-                                       break;
-                               strcat(pos, udev->kernel_name);
-                               dbg("substitute kernel name '%s'", udev->kernel_name);
+                       strnfieldcat(string, udev->bus_id, maxsize);
+                       dbg("substitute bus_id '%s'", udev->bus_id);
+                       break;
+               case 'k':
+                       if (strlen(udev->kernel_name) == 0)
                                break;
-                       case 'n':
-                               if (strlen(udev->kernel_number) == 0)
-                                       break;
-                               strcat(pos, udev->kernel_number);
-                               dbg("substitute kernel number '%s'", udev->kernel_number);
+                       strnfieldcat(string, udev->kernel_name, maxsize);
+                       dbg("substitute kernel name '%s'", udev->kernel_name);
+                       break;
+               case 'n':
+                       if (strlen(udev->kernel_number) == 0)
                                break;
-                       case 'm':
-                               sprintf(pos, "%u", udev->minor);
-                               dbg("substitute minor number '%u'", udev->minor);
+                       strnfieldcat(string, udev->kernel_number, maxsize);
+                       dbg("substitute kernel number '%s'", udev->kernel_number);
                                break;
+               case 'm':
+                       sprintf(pos, "%u", udev->minor);
+                       dbg("substitute minor number '%u'", udev->minor);
+                       break;
                        case 'M':
-                               sprintf(pos, "%u", udev->major);
-                               dbg("substitute major number '%u'", udev->major);
+                       sprintf(pos, "%u", udev->major);
+                       dbg("substitute major number '%u'", udev->major);
+                       break;
+               case 'c':
+                       if (strlen(udev->program_result) == 0)
                                break;
-                       case 'c':
-                               if (strlen(udev->program_result) == 0)
-                                       break;
-                               if (num) {
-                                       /* get part of return string */
-                                       strncpy(temp1, udev->program_result, sizeof(temp1));
-                                       pos2 = temp1;
-                                       while (num) {
-                                               num--;
-                                               pos3 = strsep(&pos2, " ");
-                                               if (pos3 == NULL) {
-                                                       dbg("requested part of result string not found");
-                                                       break;
-                                               }
-                                       }
-                                       if (pos3) {
-                                               strcat(pos, pos3);
-                                               dbg("substitute part of result string '%s'", pos3);
+                       /* get part part of the result string */
+                       i = num; /* num syntax is deprecated and will be removed  */
+                       if (attr != NULL)
+                               i = atoi(attr);
+                       if (i > 0) {
+                               strfieldcpy(temp1, udev->program_result);
+                               pos2 = temp1;
+                               while (i) {
+                                       i--;
+                                       pos3 = strsep(&pos2, " ");
+                                       if (pos3 == NULL) {
+                                               dbg("requested part of result string not found");
+                                               break;
                                        }
-                               } else {
-                                       strcat(pos, udev->program_result);
-                                       dbg("substitute result string '%s'", udev->program_result);
                                }
-                               break;
-                       case '%':
-                               strcat(pos, "%");
-                               pos++;
-                               break;
-                       default:
-                               dbg("unknown substitution type '%%%c'", pos[1]);
-                               break;
+                               if (pos3) {
+                                       strnfieldcat(string, pos3, maxsize);
+                                       dbg("substitute part of result string '%s'", pos3);
+                               }
+                       } else {
+                               strnfieldcat(string, udev->program_result, maxsize);
+                               dbg("substitute result string '%s'", udev->program_result);
+                       }
+                       break;
+               case 's':
+                       if (attr != NULL) {
+                               tmpattr = find_sysfs_attribute(class_dev, sysfs_device, attr);
+                               if (tmpattr == NULL) {
+                                       dbg("sysfa attribute '%s' not found", attr);
+                                       break;
+                               }
+                               strnfieldcat(string, tmpattr->value, maxsize);
+                               dbg("substitute sysfs value '%s'", tmpattr->value);
+                       } else {
+                               dbg("missing attribute");
                        }
-                       strcat(string, temp);
-               } else
                        break;
+               case '%':
+                       strnfieldcat(string, "%", maxsize);
+                       break;
+               default:
+                       dbg("unknown substitution type '%%%c'", c);
+                       break;
+               }
+               strnfieldcat(string, tail, maxsize);
        }
 }
 
+/* 
+ * Note, we can have multiple files for different busses in here due
+ * to the mess that USB has for its device tree...
+ */
 static struct bus_file {
        char *bus;
        char *file;
 } bus_files[] = {
        { .bus = "scsi",        .file = "vendor" },
        { .bus = "usb",         .file = "idVendor" },
+       { .bus = "usb",         .file = "iInterface" },
        { .bus = "usb-serial",  .file = "detach_state" },
        { .bus = "ide",         .file = "detach_state" },
        { .bus = "pci",         .file = "vendor" },
@@ -285,30 +351,35 @@ static void wait_for_device_to_initialize(struct sysfs_device *sysfs_device)
         */
        struct bus_file *b = &bus_files[0];
        struct sysfs_attribute *tmpattr;
-       int loop;
+       int found = 0;
+       int loop = SECONDS_TO_WAIT_FOR_FILE;
 
        while (1) {
-               if (b->bus == NULL)
-                       break;
+               if (b->bus == NULL) {
+                       if (!found)
+                               break;
+                       /* sleep to give the kernel a chance to create the file */
+                       sleep(1);
+                       --loop;
+                       if (loop == 0)
+                               break;
+                       b = &bus_files[0];
+               }
                if (strcmp(sysfs_device->bus, b->bus) == 0) {
-                       tmpattr = NULL;
-                       loop = SECONDS_TO_WAIT_FOR_FILE;
-                       while (loop--) {
-                               dbg("looking for file '%s' on bus '%s'", b->file, b->bus);
-                               tmpattr = sysfs_get_device_attr(sysfs_device, b->file);
-                               if (tmpattr) {
-                                       /* found it! */
-                                       goto exit;
-                               }
-                               /* sleep to give the kernel a chance to create the file */
-                               sleep(1);
+                       found = 1;
+                       dbg("looking for file '%s' on bus '%s'", b->file, b->bus);
+                       tmpattr = sysfs_get_device_attr(sysfs_device, b->file);
+                       if (tmpattr) {
+                               /* found it! */
+                               goto exit;
                        }
-                       dbg("timed out waiting for '%s' file, continuing on anyway...", b->file);
-                       goto exit;
+                       dbg("can't find '%s' file", b->file);
                }
-               b++;
+               ++b;
        }
-       dbg("did not find bus type '%s' on list of bus_id_files, contact greg@kroah.com", sysfs_device->bus);
+       if (!found)
+               dbg("did not find bus type '%s' on list of bus_id_files, "
+                   "contact greg@kroah.com", sysfs_device->bus);
 exit:
        return; /* here to prevent compiler warning... */
 }
@@ -334,7 +405,7 @@ static int execute_program(char *path, char *value, int len)
        int fds[2];
        pid_t pid;
        int value_set = 0;
-       char buffer[256];
+       char buffer[255];
        char *pos;
        char *args[PROGRAM_MAXARG];
        int i;
@@ -393,7 +464,7 @@ static int execute_program(char *path, char *value, int len)
                                strncpy(value, buffer, len);
                                pos = value + strlen(value)-1;
                                if (pos[0] == '\n')
-                               pos[0] = '\0';
+                                       pos[0] = '\0';
                                dbg("result is '%s'", value);
                        }
                }
@@ -412,32 +483,46 @@ static int execute_program(char *path, char *value, int len)
        return retval;
 }
 
-static int compare_sysfs_attribute(struct sysfs_class_device *class_dev, struct sysfs_device *sysfs_device, struct sysfs_pair *pair)
+static struct sysfs_attribute *find_sysfs_attribute(struct sysfs_class_device *class_dev, struct sysfs_device *sysfs_device, char *attr)
 {
        struct sysfs_attribute *tmpattr = NULL;
        char *c;
 
-       if ((pair == NULL) || (pair->file[0] == '\0') || (pair->value == '\0'))
-               return -ENODEV;
-
-       dbg("look for device attribute '%s'", pair->file);
+       dbg("look for device attribute '%s'", attr);
        /* try to find the attribute in the class device directory */
-       tmpattr = sysfs_get_classdev_attr(class_dev, pair->file);
+       tmpattr = sysfs_get_classdev_attr(class_dev, attr);
        if (tmpattr)
-               goto label_found;
+               goto attr_found;
 
        /* look in the class device directory if present */
        if (sysfs_device) {
-               tmpattr = sysfs_get_device_attr(sysfs_device, pair->file);
+               tmpattr = sysfs_get_device_attr(sysfs_device, attr);
                if (tmpattr)
-                       goto label_found;
+                       goto attr_found;
        }
-       return -ENODEV;
 
-label_found:
-       c = tmpattr->value + strlen(tmpattr->value)-1;
-       if (*c == '\n')
-               *c = 0x00;
+       return NULL;
+
+attr_found:
+       c = strchr(tmpattr->value, '\n');
+       if (c != NULL)
+               c[0] = '\0';
+
+       dbg("found attribute '%s'", tmpattr->path);
+       return tmpattr;
+}
+
+static int compare_sysfs_attribute(struct sysfs_class_device *class_dev, struct sysfs_device *sysfs_device, struct sysfs_pair *pair)
+{
+       struct sysfs_attribute *tmpattr;
+
+       if ((pair == NULL) || (pair->file[0] == '\0') || (pair->value == '\0'))
+               return -ENODEV;
+
+       tmpattr = find_sysfs_attribute(class_dev, sysfs_device, pair->file);
+       if (tmpattr == NULL)
+               return -ENODEV;
+
        dbg("compare attribute '%s' value '%s' with '%s'",
                  pair->file, tmpattr->value, pair->value);
        if (strcmp_pattern(pair->value, tmpattr->value) != 0)
@@ -477,8 +562,9 @@ static int match_id(struct config_device *dev, struct sysfs_class_device *class_
 
        strfieldcpy(path, sysfs_device->path);
        temp = strrchr(path, '/');
+       temp++;
        dbg("search '%s' in '%s', path='%s'", dev->id, temp, path);
-       if (strstr(temp, dev->id) == NULL)
+       if (strcmp_pattern(dev->id, temp) != 0)
                return -ENODEV;
        else
                return 0;
@@ -539,7 +625,8 @@ static struct sysfs_device *get_sysfs_device(struct sysfs_class_device *class_de
        tspec.tv_nsec = 10000000;  /* sleep 10 millisec */
        loop = 10;
        while (loop--) {
-               nanosleep(&tspec, NULL);
+               if (udev_sleep)
+                       nanosleep(&tspec, NULL);
                if (class_dev_parent)
                        sysfs_device = sysfs_get_classdev_device(class_dev_parent);
                else
@@ -566,7 +653,8 @@ device_found:
                loop = 10;
                tspec.tv_nsec = 10000000;
                while (loop--) {
-                       nanosleep(&tspec, NULL);
+                       if (udev_sleep)
+                               nanosleep(&tspec, NULL);
                        sysfs_get_device_bus(sysfs_device);
                        
                        if (sysfs_device->bus[0] != '\0')
@@ -647,7 +735,8 @@ static int match_rule(struct config_device *dev, struct sysfs_class_device *clas
                /* execute external program */
                if (dev->program[0] != '\0') {
                        dbg("check " FIELD_PROGRAM);
-                       apply_format(udev, dev->program);
+                       apply_format(udev, dev->program, sizeof(dev->program),
+                                    class_dev, sysfs_device);
                        if (execute_program(dev->program, udev->program_result, NAME_SIZE) != 0) {
                                dbg(FIELD_PROGRAM " returned nozero");
                                goto try_parent;
@@ -734,14 +823,16 @@ int namedev_name_device(struct sysfs_class_device *class_dev, struct udevice *ud
        }
 
        /* no rule was found so we use the kernel name */
-       strfieldcpy(udev->name, class_dev->name);
+       strfieldcpy(udev->name, udev->kernel_name);
        goto done;
 
 found:
        /* substitute placeholder */
-       apply_format(udev, udev->name);
-       apply_format(udev, udev->symlink);
-
+       apply_format(udev, udev->name, sizeof(udev->name),
+                    class_dev, sysfs_device);
+       apply_format(udev, udev->symlink, sizeof(udev->symlink),
+                    class_dev, sysfs_device);
+       udev->partitions = dev->partitions;
 done:
        perm = find_perm(udev->name);
        if (perm) {
@@ -751,8 +842,8 @@ done:
        } else {
                /* no matching perms found :( */
                udev->mode = get_default_mode();
-               strncpy(udev->owner, get_default_owner(), OWNER_SIZE);
-               strncpy(udev->group, get_default_group(), GROUP_SIZE);
+               strfieldcpy(udev->owner, get_default_owner());
+               strfieldcpy(udev->group, get_default_group());
        }
        dbg("name, '%s' is going to have owner='%s', group='%s', mode = %#o",
            udev->name, udev->owner, udev->group, udev->mode);