X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fudev.c;h=3127ec25edc68dbc1cf484b973f76905c3ae0ebc;hp=f9eb26d078213a9a323d74ff4e7447ca652a75bf;hb=416647ec10a0ca505de35348620391fde80043c0;hpb=e52b37c27c267c882d82cd1e34412817b5a4dbce diff --git a/discover/udev.c b/discover/udev.c index f9eb26d..3127ec2 100644 --- a/discover/udev.c +++ b/discover/udev.c @@ -1,5 +1,7 @@ -#define _GNU_SOURCE +#if defined(HAVE_CONFIG_H) +#include "config.h" +#endif #include #include @@ -13,6 +15,7 @@ #include #include +#include #include #include #include @@ -21,15 +24,7 @@ #include "udev.h" #include "pb-discover.h" #include "device-handler.h" - -#if defined(DEBUG) -#define DBG(fmt, args...) pb_log("DBG: " fmt, ## args) -#define DBGS(fmt, args...) \ - pb_log("DBG:%s:%d: " fmt, __func__, __LINE__, ## args) -#else -#define DBG(fmt, args...) -#define DBGS(fmt, args...) -#endif +#include "cdrom.h" struct pb_udev { struct udev *udev; @@ -50,93 +45,198 @@ static int udev_destructor(void *p) return 0; } -static void udev_setup_event_params(struct udev_device *dev, - struct event *event) +static void udev_setup_device_params(struct udev_device *udev, + struct discover_device *dev) { struct udev_list_entry *list, *entry; - list = udev_device_get_properties_list_entry(dev); + list = udev_device_get_properties_list_entry(udev); if (!list) return; udev_list_entry_foreach(entry, list) - event_set_param(event,udev_list_entry_get_name(entry), + discover_device_set_param(dev, + udev_list_entry_get_name(entry), udev_list_entry_get_value(entry)); } -static int udev_handle_dev_action(struct udev_device *dev, const char *action) +static int udev_handle_dev_add(struct pb_udev *udev, struct udev_device *dev) { - const char *devtype; - const char *devpath; - const char *devnode; - struct pb_udev *udev; - struct event *event; - enum event_action eva = 0; + struct discover_device *ddev; + const char *typestr; + const char *uuid; + const char *path; + const char *name; + const char *node; + const char *prop; + bool cdrom; + + name = udev_device_get_sysname(dev); + if (!name) { + pb_debug("udev_device_get_sysname failed\n"); + return -1; + } - assert(dev); - assert(action); + typestr = udev_device_get_devtype(dev); + if (!typestr) { + pb_debug("udev_device_get_devtype failed\n"); + return -1; + } - devtype = udev_device_get_devtype(dev); /* DEVTYPE */ + if (!(!strcmp(typestr, "disk") || !strcmp(typestr, "partition"))) { + pb_debug("SKIP %s: invalid type %s\n", name, typestr); + return 0; + } - if (!devtype) { - pb_log("udev_device_get_devtype failed\n"); - return -1; + node = udev_device_get_devnode(dev); + path = udev_device_get_devpath(dev); + if (path && (strstr(path, "virtual/block/loop") + || strstr(path, "virtual/block/ram"))) { + pb_debug("SKIP: %s: ignored (path=%s)\n", name, path); + return 0; } - devpath = udev_device_get_devpath(dev); /* DEVPATH */ + cdrom = node && !!udev_device_get_property_value(dev, "ID_CDROM"); + if (cdrom) { + /* CDROMs require a little initialisation, to get + * petitboot-compatible tray behaviour */ + cdrom_init(node); + if (!cdrom_media_present(node)) { + pb_debug("SKIP: %s: no media present\n", name); + return 0; + } + } - if (!devpath) { - pb_log("udev_device_get_devpath failed\n"); + /* We have enough info to create the device and start discovery */ + ddev = device_lookup_by_id(udev->handler, name); + if (ddev) { + pb_debug("device %s is already present?\n", name); return -1; } - devnode = udev_device_get_devnode(dev); /* DEVNAME */ + /* We may see multipath devices; they'll have the same uuid as an + * existing device, so only parse the first. */ + uuid = udev_device_get_property_value(dev, "ID_FS_UUID"); + if (uuid) { + ddev = device_lookup_by_uuid(udev->handler, uuid); + if (ddev) { + pb_log("SKIP: %s UUID [%s] already present (as %s)\n", + name, uuid, ddev->device->id); + return -1; + } + } + + ddev = discover_device_create(udev->handler, name); + + ddev->device_path = talloc_strdup(ddev, node); + + if (uuid) + ddev->uuid = talloc_strdup(ddev, uuid); + prop = udev_device_get_property_value(dev, "ID_FS_LABEL"); + if (prop) + ddev->label = talloc_strdup(ddev, prop); + ddev->device->type = cdrom ? DEVICE_TYPE_OPTICAL : DEVICE_TYPE_DISK; + + udev_setup_device_params(dev, ddev); + + device_handler_discover(udev->handler, ddev); + + return 0; +} + +static int udev_handle_dev_remove(struct pb_udev *udev, struct udev_device *dev) +{ + struct discover_device *ddev; + const char *name; - if (!devnode) { - pb_log("udev_device_get_devnode failed\n"); + name = udev_device_get_sysname(dev); + if (!name) { + pb_debug("udev_device_get_sysname failed\n"); return -1; } - /* Ignore non disk or partition, ram, loop. */ + ddev = device_lookup_by_id(udev->handler, name); + if (!ddev) + return 0; + + device_handler_remove(udev->handler, ddev); + + return 0; +} + +static int udev_handle_dev_change(struct pb_udev *udev, struct udev_device *dev) +{ + struct discover_device *ddev; + const char *name, *node; + + name = udev_device_get_sysname(dev); + node = udev_device_get_devnode(dev); - if (!(strstr(devtype, "disk") || strstr(devtype, "partition")) - || strstr(devpath, "virtual/block/loop") - || strstr(devpath, "virtual/block/ram")) { - pb_log("SKIP: %s - %s\n", devtype, devnode); + /* we're only interested in CDROM change events at present */ + if (!udev_device_get_property_value(dev, "ID_CDROM")) return 0; - } - if (!strcmp(action, "add")) { - pb_log("ADD: %s - %s\n", devtype, devnode); - eva = EVENT_ACTION_ADD; - } else if (!strcmp(action, "remove")) { - pb_log("REMOVE: %s - %s\n", devtype, devnode); - eva = EVENT_ACTION_REMOVE; - } else { - pb_log("SKIP: %s: %s - %s\n", action, devtype, devnode); + /* handle CDROM eject requests */ + if (udev_device_get_property_value(dev, "DISK_EJECT_REQUEST")) { + bool eject = false; + + pb_debug("udev: eject request\n"); + + /* If the device is mounted, cdrom_id's own eject request may + * have failed. So, we'll need to do our own here. + */ + ddev = device_lookup_by_id(udev->handler, name); + if (ddev) { + eject = ddev->mounted; + udev_handle_dev_remove(udev, dev); + } + + if (eject) + cdrom_eject(node); + return 0; } - event = talloc(NULL, struct event); + if (udev_device_get_property_value(dev, "DISK_MEDIA_CHANGE")) { + if (cdrom_media_present(node)) + return udev_handle_dev_add(udev, dev); + else + return udev_handle_dev_remove(udev, dev); + } - event->type = EVENT_TYPE_UDEV; - event->action = eva; - event->device = devnode; + return 0; +} - event->n_params = 0; - event->params = NULL; - event_set_param(event, "path", devpath); - event_set_param(event, "node", devnode); - event_set_param(event, "type", devtype); +static int udev_handle_dev_action(struct udev_device *dev, const char *action) +{ + struct pb_udev *udev = udev_get_userdata(udev_device_get_udev(dev)); + +#ifdef DEBUG + { + struct udev_list_entry *list; + const char *name; + + list = udev_device_get_properties_list_entry(dev); + name = udev_device_get_sysname(dev); + + pb_debug("%s: action %s, device %s\n", __func__, action, name); + pb_debug("%s properties:\n", __func__); + + for (; list; list = udev_list_entry_get_next(list)) + pb_log("\t%-20s: %s\n", udev_list_entry_get_name(list), + udev_list_entry_get_value(list)); + } while (0); +#endif - udev_setup_event_params(dev, event); + if (!strcmp(action, "add")) + return udev_handle_dev_add(udev, dev); - udev = udev_get_userdata(udev_device_get_udev(dev)); - assert(udev); + else if (!strcmp(action, "remove")) + return udev_handle_dev_remove(udev, dev); - device_handler_event(udev->handler, event); + else if (!strcmp(action, "change")) + return udev_handle_dev_change(udev, dev); - talloc_free(event); return 0; } @@ -301,7 +401,7 @@ struct pb_udev *udev_init(struct waitset *waitset, waiter_register_io(waitset, udev_monitor_get_fd(udev->monitor), WAIT_IN, udev_process, udev->monitor); - pb_log("%s: waiting on udev\n", __func__); + pb_debug("%s: waiting on udev\n", __func__); return udev;