[PATCH] PATCH some cleanups and security fixes

[elogind.git] / udev.c

diff --git a/udev.c b/udev.c
index 1ac1c985076be16e6d1eab73fdd65d7b58cb7d33..e6f2744fcbefe0826255550a0879e9f2f1c440a3 100644 (file)
--- a/udev.c
+++ b/udev.c
@@ -3,8 +3,7 @@
  *
  * Userspace devfs
  *
- * Copyright (C) 2003 Greg Kroah-Hartman <greg@kroah.com>
- *
+ * Copyright (C) 2003,2004 Greg Kroah-Hartman <greg@kroah.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
@@ -21,63 +20,162 @@
  *
  */
 
+#include <stdio.h>
+#include <stddef.h>
 #include <stdlib.h>
 #include <string.h>
+#include <ctype.h>
+#include <errno.h>
+#include <signal.h>
+
+#include "libsysfs/sysfs/libsysfs.h"
 #include "udev.h"
+#include "udev_lib.h"
 #include "udev_version.h"
+#include "logging.h"
+#include "namedev.h"
+#include "udevdb.h"
 
+/* global variables */
+char **main_argv;
+char **main_envp;
 
-static char *get_action(void)
+#ifdef LOG
+unsigned char logname[LOGNAME_SIZE];
+void log_message(int level, const char *format, ...)
 {
-       char *action;
-
-       action = getenv("ACTION");
-       return action;
-}
+       va_list args;
 
+       if (!udev_log)
+               return;
 
-/* yeah this should be dynamically allocated... */
-static char device[255];
+       va_start(args, format);
+       vsyslog(level, format, args);
+       va_end(args);
+}
+#endif
 
-static char *get_device(void)
+asmlinkage static void sig_handler(int signum)
 {
-       char *temp;
-
-       temp = getenv("DEVPATH");
-       if (temp == NULL)
-               return NULL;
-       strcpy(device, SYSFS_ROOT);
-       strcat(device, temp);
-
-       return device;
+       switch (signum) {
+               case SIGINT:
+               case SIGTERM:
+                       udevdb_exit();
+                       exit(20 + signum);
+               default:
+                       dbg("unhandled signal %d", signum);
+       }
 }
 
+static char *subsystem_blacklist[] = {
+       "scsi_host",
+       "scsi_device",
+       "usb_host",
+       "pci_bus",
+       "pcmcia_socket",
+       ""
+};
 
-int main(int argc, char *argv[])
+int main(int argc, char *argv[], char *envp[])
 {
-       char *subsystem;
+       main_argv = argv;
+       main_envp = envp;
+       struct sigaction act;
        char *action;
-       char *dev;
-       
-       if (argc != 2) {
-               dbg ("unknown number of arguments");
-               return 1;
+       char *devpath = "";
+       char *subsystem = "";
+       int i;
+       int retval = -EINVAL;
+       enum {
+               ADD,
+               REMOVE,
+               UDEVSTART,
+       } act_type;
+
+       dbg("version %s", UDEV_VERSION);
+
+       /* initialize our configuration */
+       udev_init_config();
+
+       if (strstr(argv[0], "udevstart")) {
+               act_type = UDEVSTART;
+       } else {
+               action = get_action();
+               if (!action) {
+                       dbg("no action?");
+                       goto exit;
+               }
+               if (strcmp(action, "add") == 0) {
+                       act_type = ADD;
+               } else if (strcmp(action, "remove") == 0) {
+                       act_type = REMOVE;
+               } else {
+                       dbg("unknown action '%s'", action);
+                       goto exit;
+               }
+
+               devpath = get_devpath();
+               if (!devpath) {
+                       dbg("no devpath?");
+                       goto exit;
+               }
+               dbg("looking at '%s'", devpath);
+
+               /* we only care about class devices and block stuff */
+               if (!strstr(devpath, "class") && !strstr(devpath, "block")) {
+                       dbg("not a block or class device");
+                       goto exit;
+               }
+
+               subsystem = get_subsystem(main_argv[1]);
+               if (!subsystem) {
+                       dbg("no subsystem?");
+                       goto exit;
+               }
+
+               /* skip blacklisted subsystems */
+               i = 0;
+               while (subsystem_blacklist[i][0] != '\0') {
+                       if (strcmp(subsystem, subsystem_blacklist[i]) == 0) {
+                               dbg("don't care about '%s' devices", subsystem);
+                               goto exit;
+                       }
+                       i++;
+               }
        }
 
-       subsystem = argv[1];
-
-       action = get_action();
-       if (!action) {
-               dbg ("no action?");
-               return 1;
+       /* set signal handlers */
+       act.sa_handler = sig_handler;
+       sigemptyset (&act.sa_mask);
+       act.sa_flags = SA_RESTART;
+       sigaction(SIGINT, &act, NULL);
+       sigaction(SIGTERM, &act, NULL);
+
+       /* initialize udev database */
+       if (udevdb_init(UDEVDB_DEFAULT) != 0) {
+               dbg("unable to initialize database");
+               goto exit;
        }
 
-       dev = get_device();
-       if (!dev) {
-               dbg ("no device?");
-               return 1;
+       switch(act_type) {
+       case UDEVSTART:
+               dbg("udevstart");
+               namedev_init();
+               udev_sleep = 0;
+               retval = udev_start();
+               break;
+       case ADD:
+               dbg("udev add");
+               namedev_init();
+               retval = udev_add_device(devpath, subsystem, NOFAKE);
+               break;
+       case REMOVE:
+               dbg("udev remove");
+               retval = udev_remove_device(devpath, subsystem);
        }
 
-       return 0;
-}
+       udevdb_exit();
 
+exit:
+       return retval;
+}