chiark / gitweb /
allow RUN to send the environment to a local socket
[elogind.git] / udevstart.c
index 867919972ad9273e1a60984f7d32e8b99e3fff1a..5f3ba711f0303deffaa1a5dc160ecdbd9d114597 100644 (file)
@@ -2,10 +2,12 @@
  * udevstart.c
  *
  * Copyright (C) 2004 Greg Kroah-Hartman <greg@kroah.com>
- * 
+ * Copyright (C) 2004 Kay Sievers <kay@vrfy.org>
+ *
  * Quick and dirty way to populate a /dev with udev if your system
- * does not have access to a shell.  Based originally on a patch to udev 
- * from Harald Hoyer <harald@redhat.com>
+ * does not have access to a shell.  Based originally on a patch
+ * from:
+ *     Harald Hoyer <harald@redhat.com>
  *
  *     This program is free software; you can redistribute it and/or modify it
  *     under the terms of the GNU General Public License as published by the
  */
 
 #include <stdlib.h>
+#include <stddef.h>
 #include <string.h>
 #include <stdio.h>
+#include <unistd.h>
 #include <errno.h>
 #include <ctype.h>
-#include <signal.h>
 #include <dirent.h>
+#include <signal.h>
 #include <sys/wait.h>
-
+#include <sys/stat.h>
+#include <sys/types.h>
+
+#include "libsysfs/sysfs/libsysfs.h"
+#include "udev_libc_wrapper.h"
+#include "udev_sysfs.h"
+#include "udev.h"
+#include "udev_version.h"
 #include "logging.h"
+#include "udev_utils.h"
+#include "udev_rules.h"
+#include "list.h"
 
+static const char *udev_run_str;
+static const char *udev_log_str;
+static struct udev_rules rules;
 
-#ifdef LOG
-unsigned char logname[42];
-void log_message(int level, const char *format, ...)
+#ifdef USE_LOG
+void log_message(int priority, const char *format, ...)
 {
        va_list args;
 
+       if (priority > udev_log_priority)
+               return;
+
        va_start(args, format);
-       vsyslog(level, format, args);
+       vsyslog(priority, format, args);
        va_end(args);
 }
 #endif
 
+struct device {
+       struct list_head node;
+       char path[PATH_SIZE];
+       char subsys[NAME_SIZE];
+};
+
+/* sort files in lexical order */
+static int device_list_insert(const char *path, char *subsystem, struct list_head *device_list)
+{
+       struct device *loop_device;
+       struct device *new_device;
+
+       dbg("insert: '%s'\n", path);
+
+       list_for_each_entry(loop_device, device_list, node) {
+               if (strcmp(loop_device->path, path) > 0) {
+                       break;
+               }
+       }
+
+       new_device = malloc(sizeof(struct device));
+       if (new_device == NULL) {
+               dbg("error malloc");
+               return -ENOMEM;
+       }
+
+       strlcpy(new_device->path, path, sizeof(new_device->path));
+       strlcpy(new_device->subsys, subsystem, sizeof(new_device->subsys));
+       list_add_tail(&new_device->node, &loop_device->node);
+       dbg("add '%s' from subsys '%s'", new_device->path, new_device->subsys);
+       return 0;
+}
+
+/* list of devices that we should run last due to any one of a number of reasons */
+static char *last_list[] = {
+       "/block/dm",    /* on here because dm wants to have the block devices around before it */
+       NULL,
+};
 
-#define MAX_PATHLEN    1024
-#define SYSBLOCK       "/sys/block"
-#define SYSCLASS       "/sys/class"
+/* list of devices that we should run first due to any one of a number of reasons */
+static char *first_list[] = {
+       "/class/mem",   /* people tend to like their memory devices around first... */
+       NULL,
+};
 
-static int execute_udev(char *path, char *value, int len)
+static int add_device(const char *path, const char *subsystem)
 {
-       int retval;
-       int res;
-       int status;
-       int fds[2];
-       pid_t pid;
-       int value_set = 0;
-       char buffer[255];
-       char *pos;
-
-       retval = pipe(fds);
-       if (retval != 0) {
-               dbg("pipe failed");
+       struct udevice udev;
+       struct sysfs_class_device *class_dev;
+       const char *devpath;
+
+       devpath = &path[strlen(sysfs_path)];
+
+       /* clear and set environment for next event */
+       clearenv();
+       setenv("ACTION", "add", 1);
+       setenv("DEVPATH", devpath, 1);
+       setenv("SUBSYSTEM", subsystem, 1);
+       setenv("UDEV_START", "1", 1);
+       if (udev_log_str)
+               setenv("UDEV_LOG", udev_log_str, 1);
+       if (udev_run_str)
+               setenv("UDEV_RUN", udev_run_str, 1);
+       dbg("exec: '%s' (%s)\n", devpath, path);
+
+       class_dev = sysfs_open_class_device_path(path);
+       if (class_dev == NULL) {
+               dbg("sysfs_open_class_device_path failed");
                return -1;
        }
-       pid = fork();
-       switch(pid) {
-       case 0:
-               /* child */
-               close(STDOUT_FILENO);
-
-               /* dup write side of pipe to STDOUT */
-               dup(fds[1]);
-
-               dbg("executing /sbin/udev '%s'", path);
-               retval = execl("/sbin/udev", "/sbin/udev", path, NULL);
-
-               info("execution of '%s' failed", path);
-               exit(1);
-       case -1:
-               dbg("fork failed");
+
+       udev_init_device(&udev, devpath, subsystem, "add");
+       udev.devt = get_devt(class_dev);
+       if (!udev.devt && udev.type != DEV_NET) {
+               dbg("sysfs_open_class_device_path failed");
                return -1;
-       default:
-               /* parent reads from fds[0] */
-               close(fds[1]);
-               retval = 0;
-               while (1) {
-                       res = read(fds[0], buffer, sizeof(buffer) - 1);
-                       if (res <= 0)
+       }
+       udev_rules_get_name(&rules, &udev, class_dev);
+       if (udev.ignore_device) {
+               dbg("device event will be ignored");
+               goto exit;
+       }
+       if (udev.name[0] == '\0') {
+               dbg("device node creation supressed");
+               goto run;
+       }
+
+       udev_add_device(&udev, class_dev);
+       if (udev.devname[0] != '\0')
+               setenv("DEVNAME", udev.devname, 1);
+run:
+       if (udev_run && !list_empty(&udev.run_list)) {
+               struct name_entry *name_loop;
+
+               dbg("executing run list");
+               list_for_each_entry(name_loop, &udev.run_list, node) {
+                       if (strncmp(name_loop->name, "socket:", strlen("socket:")) == 0)
+                               pass_env_to_socket(&name_loop->name[strlen("socket:")], devpath, "add");
+                       else
+                               execute_program(name_loop->name, udev.subsystem, NULL, 0, NULL);
+               }
+       }
+exit:
+       sysfs_close_class_device(class_dev);
+       udev_cleanup_device(&udev);
+
+       return 0;
+}
+
+static void exec_list(struct list_head *device_list)
+{
+       struct device *loop_device;
+       struct device *tmp_device;
+       int i;
+
+       /* handle the "first" type devices first */
+       list_for_each_entry_safe(loop_device, tmp_device, device_list, node) {
+               for (i = 0; first_list[i] != NULL; i++) {
+                       if (strncmp(loop_device->path, first_list[i], strlen(first_list[i])) == 0) {
+                               add_device(loop_device->path, loop_device->subsys);
+                               list_del(&loop_device->node);
+                               free(loop_device);
                                break;
-                       buffer[res] = '\0';
-                       if (res > len) {
-                               dbg("result len %d too short", len);
-                               retval = -1;
-                       }
-                       if (value_set) {
-                               dbg("result value already set");
-                               retval = -1;
-                       } else {
-                               value_set = 1;
-                               strncpy(value, buffer, len);
-                               pos = value + strlen(value)-1;
-                               if (pos[0] == '\n')
-                                       pos[0] = '\0';
-                               dbg("result is '%s'", value);
                        }
                }
-               close(fds[0]);
-               res = wait(&status);
-               if (res < 0) {
-                       dbg("wait failed result %d", res);
-                       retval = -1;
-               }
+       }
 
-               if (!WIFEXITED(status) || (WEXITSTATUS(status) != 0)) {
-                       dbg("exec program status 0x%x", status);
-                       retval = -1;
+       /* handle the devices we are allowed to, excluding the "last" type devices */
+       list_for_each_entry_safe(loop_device, tmp_device, device_list, node) {
+               int found = 0;
+               for (i = 0; last_list[i] != NULL; i++) {
+                       if (strncmp(loop_device->path, last_list[i], strlen(last_list[i])) == 0) {
+                               found = 1;
+                               break;
+                       }
                }
+               if (found)
+                       continue;
+
+               add_device(loop_device->path, loop_device->subsys);
+               list_del(&loop_device->node);
+               free(loop_device);
+       }
+
+       /* handle the rest of the devices left over, if any */
+       list_for_each_entry_safe(loop_device, tmp_device, device_list, node) {
+               add_device(loop_device->path, loop_device->subsys);
+               list_del(&loop_device->node);
+               free(loop_device);
        }
-       return retval;
 }
 
-static int udev_scan(void)
+static int has_devt(const char *directory)
 {
-       char           *devpath;
-       DIR            *dir;
-       struct dirent  *dent;
-       int             retval = -EINVAL;
-       char scratch[200];
-
-       devpath = "block";
-       dir = opendir(SYSBLOCK);
-       if (dir) {
-               for (dent = readdir(dir); dent; dent = readdir(dir)) {
-                       char            dirname[MAX_PATHLEN];
-                       DIR            *dir2;
-                       struct dirent  *dent2;
-                       if ((strcmp(dent->d_name, ".") == 0)
-                           || (strcmp(dent->d_name, "..") == 0))
-                               continue;
+       char filename[PATH_SIZE];
+       struct stat statbuf;
 
-                       snprintf(dirname, MAX_PATHLEN, "/block/%s", dent->d_name);
+       snprintf(filename, sizeof(filename), "%s/dev", directory);
+       filename[sizeof(filename)-1] = '\0';
 
-                       setenv("DEVPATH", dirname, 1);
-                       dbg("udev block, 'DEVPATH' = '%s'", dirname);
-                       execute_udev("block", scratch, sizeof(scratch));
+       if (stat(filename, &statbuf) == 0)
+               return 1;
 
-                       snprintf(dirname, MAX_PATHLEN, "%s/%s", SYSBLOCK, dent->d_name);
+       return 0;
+}
+
+static void udev_scan_block(void)
+{
+       char base[PATH_SIZE];
+       DIR *dir;
+       struct dirent *dent;
+       LIST_HEAD(device_list);
+
+       snprintf(base, sizeof(base), "%s/block", sysfs_path);
+       base[sizeof(base)-1] = '\0';
+
+       dir = opendir(base);
+       if (dir != NULL) {
+               for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
+                       char dirname[PATH_SIZE];
+                       DIR *dir2;
+                       struct dirent *dent2;
+
+                       if (dent->d_name[0] == '.')
+                               continue;
 
+                       snprintf(dirname, sizeof(dirname), "%s/%s", base, dent->d_name);
+                       dirname[sizeof(dirname)-1] = '\0';
+                       if (has_devt(dirname))
+                               device_list_insert(dirname, "block", &device_list);
+                       else
+                               continue;
+
+                       /* look for partitions */
                        dir2 = opendir(dirname);
-                       if (dir2) {
-                               for (dent2 = readdir(dir2); dent2; dent2 = readdir(dir2)) {
-                                       char            dirname2[MAX_PATHLEN];
-                                       DIR            *dir3;
-                                       struct dirent  *dent3;
-
-                                       if ((strcmp(dent2->d_name, ".") == 0) ||
-                                           (strcmp(dent2->d_name, "..") == 0))
+                       if (dir2 != NULL) {
+                               for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
+                                       char dirname2[PATH_SIZE];
+
+                                       if (dent2->d_name[0] == '.')
                                                continue;
 
-                                       snprintf(dirname2, MAX_PATHLEN, "%s/%s", dirname, dent2->d_name);
-
-                                       dir3 = opendir(dirname2);
-                                       if (dir3) {
-                                               for (dent3 = readdir(dir3); dent3; dent3 = readdir(dir3)) {
-                                                       char filename[MAX_PATHLEN];
-
-                                                       if (strcmp(dent3->d_name, "dev") == 0) {
-                                                               snprintf(filename, MAX_PATHLEN, "/block/%s/%s", dent->d_name, dent2->d_name);
-                                                               setenv("DEVPATH", filename, 1);
-                                                               dbg("udev block, 'DEVPATH' = '%s'", filename);
-                                                               execute_udev("block", scratch, sizeof(scratch));
-                                                       }
-                                               }
-                                       }
+                                       snprintf(dirname2, sizeof(dirname2), "%s/%s", dirname, dent2->d_name);
+                                       dirname2[sizeof(dirname2)-1] = '\0';
+
+                                       if (has_devt(dirname2))
+                                               device_list_insert(dirname2, "block", &device_list);
                                }
+                               closedir(dir2);
                        }
                }
+               closedir(dir);
        }
+       exec_list(&device_list);
+}
 
-       devpath = "class";
-       dir = opendir(SYSCLASS);
-       if (dir) {
-               for (dent = readdir(dir); dent; dent = readdir(dir)) {
-                       char            dirname[MAX_PATHLEN];
-                       DIR            *dir2;
-                       struct dirent  *dent2;
-                       if ((strcmp(dent->d_name, ".") == 0)
-                           || (strcmp(dent->d_name, "..") == 0))
+static void udev_scan_class(void)
+{
+       char base[PATH_SIZE];
+       DIR *dir;
+       struct dirent *dent;
+       LIST_HEAD(device_list);
+
+       snprintf(base, sizeof(base), "%s/class", sysfs_path);
+       base[sizeof(base)-1] = '\0';
+
+       dir = opendir(base);
+       if (dir != NULL) {
+               for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
+                       char dirname[PATH_SIZE];
+                       DIR *dir2;
+                       struct dirent *dent2;
+
+                       if (dent->d_name[0] == '.')
                                continue;
 
-                       snprintf(dirname, MAX_PATHLEN, "%s/%s", SYSCLASS, dent->d_name);
+                       snprintf(dirname, sizeof(dirname), "%s/%s", base, dent->d_name);
+                       dirname[sizeof(dirname)-1] = '\0';
 
                        dir2 = opendir(dirname);
-                       if (dir2) {
-                               for (dent2 = readdir(dir2); dent2; dent2 = readdir(dir2)) {
-                                       char            dirname2[MAX_PATHLEN];
-                                       DIR            *dir3;
-                                       struct dirent  *dent3;
+                       if (dir2 != NULL) {
+                               for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
+                                       char dirname2[PATH_SIZE];
 
-                                       if ((strcmp(dent2->d_name, ".") == 0) || (strcmp(dent2->d_name, "..") == 0))
+                                       if (dent2->d_name[0] == '.')
                                                continue;
 
-                                       snprintf(dirname2, MAX_PATHLEN, "%s/%s", dirname, dent2->d_name);
-
-                                       dir3 = opendir(dirname2);
-                                       if (dir3) {
-                                               for (dent3 = readdir(dir3); dent3; dent3 = readdir(dir3)) {
-                                                       char
-                                                                       filename[MAX_PATHLEN];
-
-                                                       if (strcmp(dent3->d_name, "dev") == 0) {
-                                                               snprintf
-                                                                   (filename,
-                                                                    MAX_PATHLEN,
-                                                                    "/class/%s/%s", dent->d_name, dent2->d_name);
-                                                               setenv("DEVPATH", filename, 1);
-                                                               dbg("udev '%s', 'DEVPATH' = '%s'", dent->d_name, filename);
-                                                               execute_udev(dent->d_name, scratch, sizeof(scratch));
-                                                       }
-                                               }
-                                       }
+                                       snprintf(dirname2, sizeof(dirname2), "%s/%s", dirname, dent2->d_name);
+                                       dirname2[sizeof(dirname2)-1] = '\0';
+
+                                       /* pass the net class as it is */
+                                       if (strcmp(dent->d_name, "net") == 0)
+                                               device_list_insert(dirname2, "net", &device_list);
+                                       else if (has_devt(dirname2))
+                                               device_list_insert(dirname2, dent->d_name, &device_list);
                                }
+                               closedir(dir2);
                        }
                }
+               closedir(dir);
        }
-
-       if (retval > 0)
-               retval = 0;
-
-       return -retval;
+       exec_list(&device_list);
 }
 
+static void asmlinkage sig_handler(int signum)
+{
+       switch (signum) {
+               case SIGALRM:
+                       exit(1);
+               case SIGINT:
+               case SIGTERM:
+                       exit(20 + signum);
+       }
+}
 
-int main(int argc, char **argv, char **envp)
+int main(int argc, char *argv[], char *envp[])
 {
-       init_logging("udevstart");
+       struct sigaction act;
 
-       setenv("ACTION", "add", 1);
-       setenv("UDEV_NO_SLEEP", "1", 1);
+       logging_init("udevstart");
+       udev_init_config();
+       dbg("version %s", UDEV_VERSION);
+
+       udev_run_str = getenv("UDEV_RUN");
+       udev_log_str = getenv("UDEV_LOG");
+
+       /* disable all logging if not explicitely requested */
+       if (udev_log_str == NULL)
+               udev_log_priority = 0;
+
+       /* set signal handlers */
+       memset(&act, 0x00, sizeof(act));
+       act.sa_handler = (void (*) (int))sig_handler;
+       sigemptyset (&act.sa_mask);
+       act.sa_flags = 0;
+       sigaction(SIGALRM, &act, NULL);
+       sigaction(SIGINT, &act, NULL);
+       sigaction(SIGTERM, &act, NULL);
+
+       /* trigger timeout to prevent hanging processes */
+       alarm(UDEV_ALARM_TIMEOUT);
+
+       udev_rules_init(&rules, 1);
+
+       udev_scan_block();
+       udev_scan_class();
 
-       return udev_scan();
+       udev_rules_close(&rules);
+       logging_close();
+       return 0;
 }