X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=udev.c;h=b872f42494e68070e34452e6221dca619fb5e650;hp=bf170d1651d55c69c253d72144f8bb9b07d99bb8;hb=54988802b795328ceba29480611102902e88f572;hpb=2232cac825e4f8b53273d16154309278cf639e3c

diff --git a/udev.c b/udev.c
index bf170d165..b872f4249 100644
--- a/udev.c
+++ b/udev.c
@@ -27,13 +27,39 @@
 #include <fcntl.h>
 #include <unistd.h>
 #include <errno.h>
+#include <ctype.h>
+#include <signal.h>
+#include <stdarg.h>
 
 #include "udev.h"
 #include "udev_version.h"
+#include "udev_dbus.h"
+#include "logging.h"
 #include "namedev.h"
+#include "udevdb.h"
+#include "libsysfs/libsysfs.h"
 
+/* global variables */
+char **main_argv;
+char **main_envp;
 
-static char *get_action(void)
+static void sig_handler(int signum)
+{
+	dbg("caught signal %d", signum);
+	switch (signum) {
+		case SIGINT:
+		case SIGTERM:
+		case SIGKILL:
+			sysbus_disconnect();
+			udevdb_exit();
+			exit(20 + signum);
+			break;
+		default:
+			dbg("unhandled signal");
+	}
+}
+
+static inline char *get_action(void)
 {
 	char *action;
 
@@ -41,266 +67,291 @@ static char *get_action(void)
 	return action;
 }
 
-
-static char *get_device(void)
+static inline char *get_devpath(void)
 {
-	static char device[255];
-	char *temp;
-
-	temp = getenv("DEVPATH");
-	if (temp == NULL)
-		return NULL;
-	strcpy(device, SYSFS_ROOT);
-	strcat(device, temp);
+	char *devpath;
 
-	return device;
+	devpath = getenv("DEVPATH");
+	return devpath;
 }
 
-/* 
- * Right now the major/minor of a device is stored in a file called
- * "dev" in sysfs.
- * The number is stored as:
- * 	MMmm
- * 		MM is the major
- * 		mm is the minor
- * 		The value is in hex.
- * Yes, this will probably change when we go to a bigger major/minor
- * range, and will have to be changed at that time.
- */
-static int get_major_minor (char *dev, int *major, int *minor)
+static inline char *get_seqnum(void)
 {
-	char filename[255];
-	char line[20];
-	char temp[3];
-	int fd;
-	int retval = 0;
-
-	/* add the dev file to the directory and see if it's present */
-	strncpy(filename, dev, sizeof(filename));
-	strncat(filename, DEV_FILE, sizeof(filename));
-	fd = open(filename, O_RDONLY);
-	if (fd < 0) {
-		dbg("Can't open %s", filename);
-		return -ENODEV;
-	}
+	char *seqnum;
 
-	/* get the major/minor */
-	retval = read(fd, line, sizeof(line));
-	if (retval < 0) {
-		dbg("read error on %s", dev);
-		goto exit;
-	}
+	seqnum = getenv("SEQNUM");
+	return seqnum;
+}
 
-	temp[0] = line[0];
-	temp[1] = line[1];
-	temp[2] = 0x00;
-	*major = (int)strtol(&temp[0], NULL, 16);
+static void print_record(char *path, struct udevice *dev)
+{
+	printf("P: %s\n", path);
+	printf("N: %s\n", dev->name);
+	printf("S: %s\n", dev->symlink);
+	printf("O: %s\n", dev->owner);
+	printf("G: %s\n", dev->group);
+	printf("\n");
+}
 
-	temp[0] = line[2];
-	temp[1] = line[3];
-	temp[2] = 0x00;
-	*minor = (int)strtol(&temp[0], NULL, 16);
+enum query_type {
+	NONE,
+	NAME,
+	SYMLINK,
+	OWNER,
+	GROUP
+};
 
-	dbg("found major = %d, minor = %d", *major, *minor);
+static inline int udev_user(int argc, char **argv)
+{
+	static const char short_options[] = "dp:q:rVh";
+	int option;
+	int retval = -EINVAL;
+	struct udevice dev;
+	int root = 0;
+	enum query_type query = NONE;
+	char result[NAME_SIZE] = "";
+	char path[NAME_SIZE] = "";
+
+	/* get command line options */
+	while (1) {
+		option = getopt(argc, argv, short_options);
+		if (option == -1)
+			break;
 
-	retval = 0;
-exit:
-	close(fd);
-	return retval;
-}
+		dbg("option '%c'", option);
+		switch (option) {
+		case 'p':
+			dbg("udev path: %s\n", optarg);
+			strfieldcpy(path, optarg);
+			break;
 
-/*
- * Here would go a call to the naming deamon, to get the name we want to have
- * for this device.  But for now, let's just default to whatever the kernel is
- * calling the device as that will keep the "old-style" naming policy
- */
-static char *get_name(char *dev, int major, int minor)
-{
-	static char name[100];
-	char *temp;
+		case 'q':
+			dbg("udev query: %s\n", optarg);
 
-	temp = strrchr(dev, '/');
-	if (temp == NULL)
-		return NULL;
-	strncpy(name, &temp[1], sizeof(name));
+			if (strcmp(optarg, "name") == 0) {
+				query = NAME;
+				break;
+			}
 
-	dbg("name is %s", name);
+			if (strcmp(optarg, "symlink") == 0) {
+				query = SYMLINK;
+				break;
+			}
 
-	return &name[0];
-}
+			if (strcmp(optarg, "owner") == 0) {
+				query = OWNER;
+				break;
+			}
 
-/*
- * Again, this will live in the naming deamon
- */
-static int get_mode(char *name, char *dev, int major, int minor)
-{
-	/* just default everyone to rw for the world! */
-	return 0666;
-}
+			if (strcmp(optarg, "group") == 0) {
+				query = GROUP;
+				break;
+			}
 
-/*
- * We also want to add some permissions here, and possibly some symlinks
- */
-static int create_node(char *name, char type, int major, int minor, int mode)
-{
-	char *argv[7];
-	char mode_string[100];
-	char type_string[3];
-	char major_string[20];
-	char minor_string[20];
-	char filename[255];
-	int retval = 0;
-
-	strncpy(filename, UDEV_ROOT, sizeof(filename));
-	strncat(filename, name, sizeof(filename));
-
-	snprintf(mode_string, sizeof(mode_string), "--mode=%#o", mode);
-	snprintf(type_string, sizeof(type_string), "%c", type);
-	snprintf(major_string, sizeof(major_string), "%d", major);
-	snprintf(minor_string, sizeof(minor_string), "%d", minor);
-	
-	argv[0] = MKNOD;
-	argv[1] = mode_string;
-	argv[2] = filename;
-	argv[3] = type_string;
-	argv[4] = major_string;
-	argv[5] = minor_string;
-	argv[6] = NULL;
-	dbg ("executing %s %s %s %s %s %s",
-		argv[0], argv[1], argv[2], argv[3], argv[4], argv[5]);
-	switch (fork()) {
-		case 0:
-			/* we are the child, so lets run the program */
-			execv (MKNOD, argv);
-			exit(0);
-			break;
-		case (-1):
-			dbg ("fork failed.");
-			retval = -EFAULT;
+			printf("unknown query type\n");
+			return -EINVAL;
+
+		case 'r':
+			root = 1;
 			break;
+
+		case 'd':
+			retval = udevdb_open_ro();
+			if (retval != 0) {
+				printf("unable to open udev database\n");
+				return -EACCES;
+			}
+			retval = udevdb_dump(print_record);
+			udevdb_exit();
+			return retval;
+
+		case 'V':
+			printf("udev, version %s\n", UDEV_VERSION);
+			return 0;
+
+		case 'h':
+			retval = 0;
+		case '?':
 		default:
-			break;
+			goto help;
+		}
 	}
-	return retval;
-}
 
-/*
- * We also want to clean up any symlinks that were created in create_node()
- */
-static int delete_node(char *name)
-{
-	char filename[255];
+	/* process options */
+	if (query != NONE) {
+		if (path[0] == '\0') {
+			printf("query needs device path specified\n");
+			return -EINVAL;
+		}
+
+		retval = udevdb_open_ro();
+		if (retval != 0) {
+			printf("unable to open udev database\n");
+			return -EACCES;
+		}
+		retval = udevdb_get_dev(path, &dev);
+		if (retval == 0) {
+			switch(query) {
+			case NAME:
+				if (root)
+				strfieldcpy(result, udev_root);
+				strncat(result, dev.name, sizeof(result));
+				break;
+
+			case SYMLINK:
+				strfieldcpy(result, dev.symlink);
+				break;
+
+			case GROUP:
+				strfieldcpy(result, dev.group);
+				break;
+
+			case OWNER:
+				strfieldcpy(result, dev.owner);
+				break;
+
+			default:
+				break;
+			}
+			printf("%s\n", result);
+		} else {
+			printf("device not found in udev database\n");
+		}
+		udevdb_exit();
+		return retval;
+	}
 
-	strncpy(filename, UDEV_ROOT, sizeof(filename));
-	strncat(filename, name, sizeof(filename));
+	if (root) {
+		printf("%s\n", udev_root);
+		return 0;
+	}
 
-	dbg("unlinking %s", filename);
-	return unlink(filename);
+help:
+	printf("Usage: [-pqrdVh]\n"
+	       "  -q TYPE  query database for the specified value:\n"
+	       "             'name'    name of device node\n"
+	       "             'symlink' pointing to node\n"
+	       "             'owner'   of node\n"
+	       "             'group'   of node\n"
+	       "  -p PATH  sysfs device path used for query\n"
+	       "  -r       print udev root\n"
+	       "  -d       dump whole database\n"
+	       "  -V       print udev version\n"
+	       "  -h       print this help text\n"
+	       "\n");
+	return retval;
 }
 
-static int add_device(char *device, char type)
+static char *subsystem_blacklist[] = {
+	"net",
+	"scsi_host",
+	"scsi_device",
+	"usb_host",
+	"pci_bus",
+	"",
+};
+
+static inline int udev_hotplug(int argc, char **argv)
 {
-	char *name;
-	int major;
-	int minor;
-	int mode;
+	char *action;
+	char *devpath;
+	char *subsystem;
 	int retval = -EINVAL;
+	int i;
 
-	retval = get_major_minor(device, &major, &minor);
-	if (retval) {
-		dbg ("get_major_minor failed");
-		goto exit;
-	}
+	subsystem = argv[1];
 
-	name = get_name(device, major, minor);
-	if (name == NULL) {
-		dbg ("get_name failed");
-		retval = -ENODEV;
+	devpath = get_devpath();
+	if (!devpath) {
+		dbg ("no devpath?");
 		goto exit;
 	}
+	dbg("looking at '%s'", devpath);
 
-	mode = get_mode(name, device, major, minor);
-	if (mode < 0) {
-		dbg ("get_mode failed");
-		retval = -EINVAL;
+	/* 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;
 	}
 
-	return create_node(name, type, major, minor, mode);
-
-exit:
-	return retval;
-}
-
-static int remove_device(char *device)
-{
-	char *name;
-	int retval = 0;
+	/* 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++;
+	}
 
-	name = get_name(device, 0, 0);
-	if (name == NULL) {
-		dbg ("get_name failed");
-		retval = -ENODEV;
+	action = get_action();
+	if (!action) {
+		dbg ("no action?");
 		goto exit;
 	}
 
-	return delete_node(name);
+	/* connect to the system message bus */
+	sysbus_connect();
 
-exit:
-	return retval;
-}
-
-int main(int argc, char *argv[])
-{
-	char *subsystem;
-	char *action;
-	char *device;
-	char type;
-	int retval = -EINVAL;
-	
-	if (argc != 2) {
-		dbg ("unknown number of arguments");
-		goto exit;
+	/* initialize udev database */
+	retval = udevdb_init(UDEVDB_DEFAULT);
+	if (retval != 0) {
+		dbg("unable to initialize database");
+		goto exit_sysbus;
 	}
 
+	/* set up a default signal handler for now */
+	signal(SIGINT, sig_handler);
+	signal(SIGTERM, sig_handler);
+	signal(SIGKILL, sig_handler);
+
+	/* initialize the naming deamon */
 	namedev_init();
 
-	/* sleep for a second or two to give the kernel a chance to
-	 * create the dev file
-	 */
-	sleep(2);
+	if (strcmp(action, "add") == 0)
+		retval = udev_add_device(devpath, subsystem);
 
-	/* for now, the block layer is the only place where block devices are */
-	subsystem = argv[1];
-	if (strcmp(subsystem, "block") == 0)
-		type = 'b';
-	else
-		type = 'c';
+	else if (strcmp(action, "remove") == 0)
+		retval = udev_remove_device(devpath, subsystem);
 
-	action = get_action();
-	if (!action) {
-		dbg ("no action?");
-		goto exit;
+	else {
+		dbg("unknown action '%s'", action);
+		retval = -EINVAL;
 	}
+	udevdb_exit();
 
-	device = get_device();
-	if (!device) {
-		dbg ("no device?");
-		goto exit;
-	}
-	dbg("looking at %s", device);
+exit_sysbus:
+	/* disconnect from the system message bus */
+	sysbus_disconnect();
 
-	if (strcmp(action, "add") == 0)
-		return add_device(device, type);
+exit:
+	if (retval > 0)
+		retval = 0;
 
-	if (strcmp(action, "remove") == 0)
-		return remove_device(device);
+	return -retval;
+}
 
-	dbg("Unknown action: %s", action);
-	return -EINVAL;
+int main(int argc, char **argv, char **envp)
+{
+	int retval;
+	main_argv = argv;
+	main_envp = envp;
+
+	dbg("version %s", UDEV_VERSION);
+
+	/* initialize our configuration */
+	udev_init_config();
+
+	if (argc == 2 && argv[1][0] != '-') {
+		dbg("called by hotplug");
+		retval = udev_hotplug(argc, argv);
+	} else {
+		dbg("called by user");
+		retval = udev_user(argc, argv);
+	}
 
-	retval = 0;
-exit:	
 	return retval;
 }
 
+