X-Git-Url: https://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fdevice-handler.c;h=142f6a490028508056415ab99e78f020f035b309;hp=7a84302b71314104b74309c3167abe5314d017b2;hb=769fd76dc9bf3440999297ab943af49160df9c45;hpb=e52b37c27c267c882d82cd1e34412817b5a4dbce diff --git a/discover/device-handler.c b/discover/device-handler.c index 7a84302..142f6a4 100644 --- a/discover/device-handler.c +++ b/discover/device-handler.c @@ -1,9 +1,11 @@ #include #include +#include #include #include #include +#include #include #include @@ -21,7 +23,6 @@ #include "event.h" #include "parser.h" #include "resource.h" -#include "udev.h" #include "paths.h" #include "boot.h" @@ -41,6 +42,9 @@ struct device_handler { struct list unresolved_boot_options; }; +static int mount_device(struct discover_device *dev); +static int umount_device(struct discover_device *dev); + void discover_context_add_boot_option(struct discover_context *ctx, struct discover_boot_option *boot_option) { @@ -101,6 +105,12 @@ static int device_match_id(struct discover_device *dev, const char *id) return !strcmp(dev->device->id, id); } +static int device_match_serial(struct discover_device *dev, const char *serial) +{ + const char *val = discover_device_get_param(dev, "ID_SERIAL"); + return val && !strcmp(val, serial); +} + static struct discover_device *device_lookup( struct device_handler *device_handler, int (match_fn)(struct discover_device *, const char *), @@ -152,88 +162,94 @@ struct discover_device *device_lookup_by_id( return device_lookup(device_handler, device_match_id, id); } +struct discover_device *device_lookup_by_serial( + struct device_handler *device_handler, + const char *serial) +{ + return device_lookup(device_handler, device_match_serial, serial); +} + void device_handler_destroy(struct device_handler *handler) { talloc_free(handler); } -#ifdef PETITBOOT_TEST - -/* we have a simplified interface for petitboot testing, but still want - * to keep struct device_handler opaque. */ -struct device_handler *device_handler_init( - struct discover_server *server __attribute__((unused)), - struct waitset *waitset __attribute__((unused)), - int dry_run __attribute__((unused))) +static int destroy_device(void *arg) { - struct device_handler *handler; + struct discover_device *dev = arg; - handler = talloc_zero(NULL, struct device_handler); - list_init(&handler->unresolved_boot_options); + umount_device(dev); - return handler; + return 0; } -void device_handler_add_device(struct device_handler *handler, - struct discover_device *dev) +struct discover_device *discover_device_create(struct device_handler *handler, + const char *id) { - handler->n_devices++; - handler->devices = talloc_realloc(handler, handler->devices, - struct discover_device *, handler->n_devices); - handler->devices[handler->n_devices - 1] = dev; -} - -#else + struct discover_device *dev; -static int mount_device(struct discover_device *dev) -{ - int rc; + dev = device_lookup_by_id(handler, id); + if (dev) + return dev; - if (!dev->device_path) - return -1; + dev = talloc_zero(handler, struct discover_device); + dev->device = talloc_zero(dev, struct device); + dev->device->id = talloc_strdup(dev->device, id); + list_init(&dev->params); + list_init(&dev->boot_options); - if (!dev->mount_path) - dev->mount_path = join_paths(dev, mount_base(), - dev->device_path); + talloc_set_destructor(dev, destroy_device); - if (pb_mkdir_recursive(dev->mount_path)) - pb_log("couldn't create mount directory %s: %s\n", - dev->mount_path, strerror(errno)); + return dev; +} - rc = process_run_simple(dev, pb_system_apps.mount, - dev->device_path, dev->mount_path, - "-o", "ro", NULL); +struct discover_device_param { + char *name; + char *value; + struct list_item list; +}; - if (!rc) - return 0; +void discover_device_set_param(struct discover_device *device, + const char *name, const char *value) +{ + struct discover_device_param *param; + bool found = false; - /* Retry mount without ro option. */ - rc = process_run_simple(dev, pb_system_apps.mount, - dev->device_path, dev->mount_path, NULL); + list_for_each_entry(&device->params, param, list) { + if (!strcmp(param->name, name)) { + found = true; + break; + } + } - if (!rc) - return 0; + if (!found) { + if (!value) + return; + param = talloc(device, struct discover_device_param); + param->name = talloc_strdup(param, name); + list_add(&device->params, ¶m->list); + } else { + if (!value) { + list_remove(¶m->list); + talloc_free(param); + return; + } + talloc_free(param->value); + } - pb_rmdir_recursive(mount_base(), dev->mount_path); - return -1; + param->value = talloc_strdup(param, value); } -static int umount_device(struct discover_device *dev) +const char *discover_device_get_param(struct discover_device *device, + const char *name) { - int status; - - if (!dev->mount_path) - return 0; + struct discover_device_param *param; - status = process_run_simple(dev, pb_system_apps.umount, - dev->mount_path, NULL); - - if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) - return -1; - - pb_rmdir_recursive(mount_base(), dev->mount_path); - - return 0; + list_for_each_entry(&device->params, param, list) { + if (!strcmp(param->name, name)) + return param->name; + } + return NULL; } struct device_handler *device_handler_init(struct discover_server *server, @@ -241,14 +257,12 @@ struct device_handler *device_handler_init(struct discover_server *server, { struct device_handler *handler; - handler = talloc(NULL, struct device_handler); - handler->devices = NULL; - handler->n_devices = 0; + handler = talloc_zero(NULL, struct device_handler); handler->server = server; handler->waitset = waitset; handler->dry_run = dry_run; - handler->default_boot_option = NULL; handler->autoboot_enabled = config_get()->autoboot_enabled; + list_init(&handler->unresolved_boot_options); /* set up our mount point base */ @@ -259,85 +273,8 @@ struct device_handler *device_handler_init(struct discover_server *server, return handler; } -static int destroy_device(void *arg) -{ - struct discover_device *dev = arg; - - umount_device(dev); - - return 0; -} - -static struct discover_device *find_device(struct device_handler *handler, - const char *id) -{ - struct discover_device *dev; - unsigned int i; - - for (i = 0; i < handler->n_devices; i++) { - dev = handler->devices[i]; - if (!strcmp(dev->device->id, id)) - return dev; - } - - return NULL; -} - -static enum device_type event_device_type(struct device *device, - struct event *event) -{ - const char *param; - - param = event_get_param(event, "type"); - if (!param) { - pb_log("%s: empty type\n", device->id); - return DEVICE_TYPE_UNKNOWN; - } - - if (!strcmp(param, "disk") || !strcmp(param, "partition")) - return DEVICE_TYPE_DISK; - - if (!strcmp(param, "net")) - return DEVICE_TYPE_NETWORK; - - pb_log("%s: unknown type '%s'\n", device->id, param); - return DEVICE_TYPE_UNKNOWN; -} - -static struct discover_device *discover_device_create( - struct device_handler *handler, - struct discover_context *ctx, - struct event *event) -{ - struct discover_device *dev; - const char *devnode; - - dev = find_device(handler, event->device); - if (dev) - return dev; - - dev = talloc_zero(ctx, struct discover_device); - dev->device = talloc_zero(dev, struct device); - list_init(&dev->boot_options); - - devnode = event_get_param(ctx->event, "node"); - if (devnode) - dev->device_path = talloc_strdup(dev, devnode); - - dev->device->id = talloc_strdup(dev, event->device); - dev->device->type = event_device_type(dev->device, event); - - talloc_set_destructor(dev, destroy_device); - - return dev; -} - -/** - * device_handler_remove - Remove a device from the handler device array. - */ - -static void device_handler_remove(struct device_handler *handler, - struct discover_device *device) +void device_handler_remove(struct device_handler *handler, + struct discover_device *device) { unsigned int i; @@ -346,7 +283,7 @@ static void device_handler_remove(struct device_handler *handler, break; if (i == handler->n_devices) { - assert(0 && "unknown device"); + talloc_free(device); return; } @@ -356,7 +293,9 @@ static void device_handler_remove(struct device_handler *handler, handler->devices = talloc_realloc(handler, handler->devices, struct discover_device *, handler->n_devices); - discover_server_notify_device_remove(handler->server, device->device); + if (device->notified) + discover_server_notify_device_remove(handler->server, + device->device); talloc_free(device); } @@ -403,6 +342,8 @@ static int default_timeout(void *arg) return 0; } + handler->timeout_waiter = NULL; + pb_log("Timeout expired, booting default option %s\n", opt->option->id); boot(handler, handler->default_boot_option, NULL, @@ -410,17 +351,56 @@ static int default_timeout(void *arg) return 0; } +static bool priority_match(struct boot_priority *prio, + struct discover_boot_option *opt) +{ + return prio->type == opt->device->device->type; +} + +static int default_option_priority(struct discover_boot_option *opt) +{ + const struct config *config; + struct boot_priority *prio; + int i; + + config = config_get(); + + for (i = 0; i < config->n_boot_priorities; i++) { + prio = &config->boot_priorities[i]; + if (priority_match(prio, opt)) + break; + } + + return i; +} + static void set_default(struct device_handler *handler, struct discover_boot_option *opt) { - if (handler->default_boot_option) + if (!handler->autoboot_enabled) return; - if (!handler->autoboot_enabled) + /* Resolve any conflicts: if we have a new default option, it only + * replaces the current if it has a higher priority. */ + if (handler->default_boot_option) { + int new_prio, cur_prio; + + new_prio = default_option_priority(opt); + cur_prio = default_option_priority( + handler->default_boot_option); + + if (new_prio < cur_prio) { + handler->default_boot_option = opt; + /* extend the timeout a little, so the user sees some + * indication of the change */ + handler->sec_to_boot += 2; + } + return; + } - handler->default_boot_option = opt; handler->sec_to_boot = config_get()->autoboot_timeout_sec; + handler->default_boot_option = opt; pb_log("Boot option %s set as default, timeout %u sec.\n", opt->option->id, handler->sec_to_boot); @@ -496,6 +476,18 @@ static void boot_option_finalise(struct device_handler *handler, set_default(handler, opt); } +static void notify_boot_option(struct device_handler *handler, + struct discover_boot_option *opt) +{ + struct discover_device *dev = opt->device; + + if (!dev->notified) + discover_server_notify_device_add(handler->server, + opt->device->device); + dev->notified = true; + discover_server_notify_boot_option_add(handler->server, opt->option); +} + static void process_boot_option_queue(struct device_handler *handler) { struct discover_boot_option *opt, *tmp; @@ -515,46 +507,37 @@ static void process_boot_option_queue(struct device_handler *handler) list_add_tail(&opt->device->boot_options, &opt->list); talloc_steal(opt->device, opt); boot_option_finalise(handler, opt); - discover_server_notify_boot_option_add(handler->server, - opt->option); + notify_boot_option(handler, opt); } } +struct discover_context *device_handler_discover_context_create( + struct device_handler *handler, + struct discover_device *device) +{ + struct discover_context *ctx; + + ctx = talloc(handler, struct discover_context); + ctx->device = device; + ctx->conf_url = NULL; + ctx->test_data = NULL; + list_init(&ctx->boot_options); + + return ctx; +} + /** * context_commit - Commit a temporary discovery context to the handler, * and notify the clients about any new options / devices */ -static void context_commit(struct device_handler *handler, +void device_handler_discover_context_commit(struct device_handler *handler, struct discover_context *ctx) { struct discover_device *dev = ctx->device; struct discover_boot_option *opt, *tmp; - unsigned int i, existing_device = 0; - - /* do we already have this device? */ - for (i = 0; i < handler->n_devices; i++) { - if (ctx->device == handler->devices[i]) { - existing_device = 1; - break; - } - } - - /* if not already present, add the device to the handler's array */ - if (!existing_device) { - handler->n_devices++; - handler->devices = talloc_realloc(handler, handler->devices, - struct discover_device *, handler->n_devices); - handler->devices[handler->n_devices - 1] = dev; - talloc_steal(handler, dev); - - discover_server_notify_device_add(handler->server, dev->device); - - /* this new device might be able to resolve existing boot - * options */ - pb_log("New device %s, processing queue\n", dev->device->id); - process_boot_option_queue(handler); - } + if (!device_lookup_by_id(handler, dev->device->id)) + device_handler_add_device(handler, dev); /* move boot options from the context to the device */ list_for_each_entry_safe(&ctx->boot_options, opt, tmp, list) { @@ -567,8 +550,7 @@ static void context_commit(struct device_handler *handler, list_add_tail(&dev->boot_options, &opt->list); talloc_steal(dev, opt); boot_option_finalise(handler, opt); - discover_server_notify_boot_option_add(handler->server, - opt->option); + notify_boot_option(handler, opt); } else { if (!opt->source->resolve_resource) { pb_log("parser %s gave us an unresolved " @@ -588,236 +570,305 @@ static void context_commit(struct device_handler *handler, } } -static int handle_add_udev_event(struct device_handler *handler, - struct event *event) +void device_handler_add_device(struct device_handler *handler, + struct discover_device *device) { - struct discover_context *ctx; - struct discover_device *dev; - const char *param; - int rc; - - /* create our context */ - ctx = talloc(handler, struct discover_context); - ctx->event = event; - list_init(&ctx->boot_options); + handler->n_devices++; + handler->devices = talloc_realloc(handler, handler->devices, + struct discover_device *, handler->n_devices); + handler->devices[handler->n_devices - 1] = device; - /* create our top-level device */ - dev = discover_device_create(handler, ctx, event); +} - ctx->device = dev; +/* Start discovery on a hotplugged device. The device will be in our devices + * array, but has only just been initialised by the hotplug source. + */ +int device_handler_discover(struct device_handler *handler, + struct discover_device *dev, enum conf_method method) +{ + struct discover_context *ctx; + int rc; - /* try to parse UUID and labels */ - param = event_get_param(ctx->event, "ID_FS_UUID"); - if (param) - dev->uuid = talloc_strdup(dev, param); + process_boot_option_queue(handler); - param = event_get_param(ctx->event, "ID_FS_LABEL"); - if (param) - dev->label = talloc_strdup(dev, param); + /* create our context */ + ctx = device_handler_discover_context_create(handler, dev); rc = mount_device(dev); - if (rc) { - talloc_free(ctx); - return 0; - } + if (rc) + goto out; /* run the parsers. This will populate the ctx's boot_option list. */ - iterate_parsers(ctx, CONF_METHOD_LOCAL_FILE); + iterate_parsers(ctx, method); /* add discovered stuff to the handler */ - context_commit(handler, ctx); + device_handler_discover_context_commit(handler, ctx); +out: talloc_free(ctx); return 0; } -static int handle_remove_udev_event(struct device_handler *handler, - struct event *event) +/* incoming conf event */ +int device_handler_conf(struct device_handler *handler, + struct discover_device *dev, struct pb_url *url, + enum conf_method method) { - struct discover_device *dev; + struct discover_context *ctx; - dev = find_device(handler, event->device); - if (!dev) - return 0; + /* create our context */ + ctx = device_handler_discover_context_create(handler, dev); + ctx->conf_url = url; - /* remove device from handler device array */ - device_handler_remove(handler, dev); + iterate_parsers(ctx, method); + + device_handler_discover_context_commit(handler, ctx); + + talloc_free(ctx); return 0; } -static int handle_add_user_event(struct device_handler *handler, - struct event *event) +static struct discover_boot_option *find_boot_option_by_id( + struct device_handler *handler, const char *id) { - struct discover_context *ctx; - struct discover_device *dev; - int rc; + unsigned int i; - assert(event->device); + for (i = 0; i < handler->n_devices; i++) { + struct discover_device *dev = handler->devices[i]; + struct discover_boot_option *opt; - ctx = talloc(handler, struct discover_context); - ctx->event = event; - list_init(&ctx->boot_options); + list_for_each_entry(&dev->boot_options, opt, list) + if (!strcmp(opt->option->id, id)) + return opt; + } - dev = discover_device_create(handler, ctx, event); - ctx->device = dev; + return NULL; +} - rc = parse_user_event(ctx, event); +void device_handler_boot(struct device_handler *handler, + struct boot_command *cmd) +{ + struct discover_boot_option *opt = NULL; - if (!rc) - context_commit(handler, ctx); + if (cmd->option_id && strlen(cmd->option_id)) + opt = find_boot_option_by_id(handler, cmd->option_id); - return rc; + boot(handler, opt, cmd, handler->dry_run, boot_status, handler); } -static int handle_remove_user_event(struct device_handler *handler, - struct event *event) +void device_handler_cancel_default(struct device_handler *handler) { - struct discover_device *dev = find_device(handler, event->device); + struct boot_status status; - if (!dev) - return 0; + if (handler->timeout_waiter) + waiter_remove(handler->timeout_waiter); - /* remove device from handler device array */ - device_handler_remove(handler, dev); + handler->timeout_waiter = NULL; + handler->autoboot_enabled = false; - return 0; + /* we only send status if we had a default boot option queued */ + if (!handler->default_boot_option) + return; + + pb_log("Cancelling default boot option\n"); + + handler->default_boot_option = NULL; + + status.type = BOOT_STATUS_INFO; + status.progress = -1; + status.detail = NULL; + status.message = "Default boot cancelled"; + + discover_server_notify_boot_status(handler->server, &status); } -static enum conf_method parse_conf_method(const char *str) +#ifndef PETITBOOT_TEST +static bool check_existing_mount(struct discover_device *dev) { + struct stat devstat, mntstat; + struct mntent *mnt; + FILE *fp; + int rc; - if (!strcasecmp(str, "dhcp")) { - return CONF_METHOD_DHCP; + rc = stat(dev->device_path, &devstat); + if (rc) { + pb_debug("%s: stat failed: %s\n", __func__, strerror(errno)); + return false; } - return CONF_METHOD_UNKNOWN; + + if (!S_ISBLK(devstat.st_mode)) { + pb_debug("%s: %s isn't a block device?\n", __func__, + dev->device_path); + return false; + } + + fp = fopen("/proc/self/mounts", "r"); + + for (;;) { + mnt = getmntent(fp); + if (!mnt) + break; + + if (!mnt->mnt_fsname || mnt->mnt_fsname[0] != '/') + continue; + + rc = stat(mnt->mnt_fsname, &mntstat); + if (rc) + continue; + + if (!S_ISBLK(mntstat.st_mode)) + continue; + + if (mntstat.st_rdev == devstat.st_rdev) { + dev->mount_path = talloc_strdup(dev, mnt->mnt_dir); + dev->mounted_rw = !!hasmntopt(mnt, "rw"); + dev->mounted = true; + dev->unmount = false; + + pb_debug("%s: %s is already mounted (r%c) at %s\n", + __func__, dev->device_path, + dev->mounted_rw ? 'w' : 'o', + mnt->mnt_dir); + break; + } + } + + fclose(fp); + + return mnt != NULL; } -static int handle_conf_user_event(struct device_handler *handler, - struct event *event) +static int mount_device(struct discover_device *dev) { - struct discover_context *ctx; - struct discover_device *dev; - enum conf_method method; - const char *val; + int rc; - ctx = talloc(handler, struct discover_context); - ctx->event = event; - list_init(&ctx->boot_options); + if (!dev->device_path) + return -1; - val = event_get_param(event, "url"); - if (!val) { - talloc_free(ctx); + if (dev->mounted) return 0; - } - ctx->conf_url = pb_url_parse(ctx, val); - if (!ctx->conf_url) { - talloc_free(ctx); + if (check_existing_mount(dev)) return 0; + + dev->mount_path = join_paths(dev, mount_base(), + dev->device_path); + + if (pb_mkdir_recursive(dev->mount_path)) { + pb_log("couldn't create mount directory %s: %s\n", + dev->mount_path, strerror(errno)); + goto err_free; } - val = event_get_param(event, "method"); - if (!val) { - talloc_free(ctx); + rc = process_run_simple(dev, pb_system_apps.mount, + dev->device_path, dev->mount_path, + "-o", "ro", NULL); + if (!rc) { + dev->mounted = true; + dev->mounted_rw = false; + dev->unmount = true; return 0; } - method = parse_conf_method(val); - if (method == CONF_METHOD_UNKNOWN) { - talloc_free(ctx); + /* Retry mount without ro option. */ + rc = process_run_simple(dev, pb_system_apps.mount, + dev->device_path, dev->mount_path, NULL); + + if (!rc) { + dev->mounted = true; + dev->mounted_rw = true; + dev->unmount = true; return 0; } - dev = discover_device_create(handler, ctx, event); - ctx->device = dev; + pb_rmdir_recursive(mount_base(), dev->mount_path); +err_free: + talloc_free(dev->mount_path); + dev->mount_path = NULL; + return -1; +} - iterate_parsers(ctx, method); +static int umount_device(struct discover_device *dev) +{ + int status; - context_commit(handler, ctx); + if (!dev->mounted || !dev->unmount) + return 0; - return 0; -} + status = process_run_simple(dev, pb_system_apps.umount, + dev->mount_path, NULL); -typedef int (*event_handler)(struct device_handler *, struct event *); + if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) + return -1; -static event_handler handlers[EVENT_TYPE_MAX][EVENT_ACTION_MAX] = { - [EVENT_TYPE_UDEV] = { - [EVENT_ACTION_ADD] = handle_add_udev_event, - [EVENT_ACTION_REMOVE] = handle_remove_udev_event, - }, - [EVENT_TYPE_USER] = { - [EVENT_ACTION_ADD] = handle_add_user_event, - [EVENT_ACTION_REMOVE] = handle_remove_user_event, - [EVENT_ACTION_CONF] = handle_conf_user_event, - } -}; + dev->mounted = false; -int device_handler_event(struct device_handler *handler, - struct event *event) -{ - if (event->type >= EVENT_TYPE_MAX || - event->action >= EVENT_ACTION_MAX || - !handlers[event->type][event->action]) { - pb_log("%s unknown type/action: %d/%d\n", __func__, - event->type, event->action); - return 0; - } + pb_rmdir_recursive(mount_base(), dev->mount_path); + + talloc_free(dev->mount_path); + dev->mount_path = NULL; - return handlers[event->type][event->action](handler, event); + return 0; } -static struct discover_boot_option *find_boot_option_by_id( - struct device_handler *handler, const char *id) +int device_request_write(struct discover_device *dev, bool *release) { - unsigned int i; - - for (i = 0; i < handler->n_devices; i++) { - struct discover_device *dev = handler->devices[i]; - struct discover_boot_option *opt; + int rc; - list_for_each_entry(&dev->boot_options, opt, list) - if (!strcmp(opt->option->id, id)) - return opt; - } + *release = false; - return NULL; -} + if (!dev->mounted) + return -1; -void device_handler_boot(struct device_handler *handler, - struct boot_command *cmd) -{ - struct discover_boot_option *opt; + if (dev->mounted_rw) + return 0; - opt = find_boot_option_by_id(handler, cmd->option_id); + rc = process_run_simple(dev, pb_system_apps.mount, dev->mount_path, + "-o", "remount,rw", NULL); + if (rc) + return -1; - boot(handler, opt, cmd, handler->dry_run, boot_status, handler); + dev->mounted_rw = true; + *release = true; + return 0; } -void device_handler_cancel_default(struct device_handler *handler) +void device_release_write(struct discover_device *dev, bool release) { - struct boot_status status; - - if (handler->timeout_waiter) - waiter_remove(handler->timeout_waiter); + if (!release) + return; - handler->timeout_waiter = NULL; - handler->autoboot_enabled = false; + process_run_simple(dev, pb_system_apps.mount, dev->mount_path, + "-o", "remount,ro", NULL); + dev->mounted_rw = false; +} - /* we only send status if we had a default boot option queued */ - if (!handler->default_boot_option) - return; +#else - pb_log("Cancelling default boot option\n"); +static int umount_device(struct discover_device *dev __attribute__((unused))) +{ + return 0; +} - handler->default_boot_option = NULL; +static int __attribute__((unused)) mount_device( + struct discover_device *dev __attribute__((unused))) +{ + return 0; +} - status.type = BOOT_STATUS_INFO; - status.progress = -1; - status.detail = NULL; - status.message = "Default boot cancelled"; +int device_request_write(struct discover_device *dev __attribute__((unused)), + bool *release) +{ + *release = true; + return 0; +} - discover_server_notify_boot_status(handler->server, &status); +void device_release_write(struct discover_device *dev __attribute__((unused)), + bool release __attribute__((unused))) +{ } + #endif +