X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fdevice-handler.c;h=7a84302b71314104b74309c3167abe5314d017b2;hp=4ba7405f2de73f29ded296340ec310104393a8d6;hb=e52b37c27c267c882d82cd1e34412817b5a4dbce;hpb=12a5c9153ce95ddde7fb14eaba9d779933a3637c diff --git a/discover/device-handler.c b/discover/device-handler.c index 4ba7405..7a84302 100644 --- a/discover/device-handler.c +++ b/discover/device-handler.c @@ -7,315 +7,627 @@ #include #include +#include #include #include #include #include #include +#include +#include #include "device-handler.h" #include "discover-server.h" #include "event.h" #include "parser.h" +#include "resource.h" #include "udev.h" #include "paths.h" #include "boot.h" struct device_handler { - struct discover_server *server; - int dry_run; + struct discover_server *server; + int dry_run; - struct device **devices; - unsigned int n_devices; + struct discover_device **devices; + unsigned int n_devices; - struct list contexts; + struct waitset *waitset; + struct waiter *timeout_waiter; + bool autoboot_enabled; + unsigned int sec_to_boot; + + struct discover_boot_option *default_boot_option; + struct list unresolved_boot_options; }; +void discover_context_add_boot_option(struct discover_context *ctx, + struct discover_boot_option *boot_option) +{ + boot_option->source = ctx->parser; + list_add_tail(&ctx->boot_options, &boot_option->list); + talloc_steal(ctx, boot_option); +} + /** - * device_handler_add - Add a device to the handler device array. + * device_handler_get_device_count - Get the count of current handler devices. */ -static void device_handler_add(struct device_handler *handler, - struct device *device) +int device_handler_get_device_count(const struct device_handler *handler) { - handler->n_devices++; - handler->devices = talloc_realloc(handler, handler->devices, - struct device *, handler->n_devices); - handler->devices[handler->n_devices - 1] = device; + return handler->n_devices; } /** - * device_handler_remove - Remove a device from the handler device array. + * device_handler_get_device - Get a handler device by index. */ -static void device_handler_remove(struct device_handler *handler, - struct device *device) +const struct discover_device *device_handler_get_device( + const struct device_handler *handler, unsigned int index) { - unsigned int i; - - for (i = 0; i < handler->n_devices; i++) - if (handler->devices[i] == device) - break; - - if (i == handler->n_devices) { - assert(0 && "unknown device"); - return; + if (index >= handler->n_devices) { + assert(0 && "bad index"); + return NULL; } - handler->n_devices--; - memmove(&handler->devices[i], &handler->devices[i + 1], - (handler->n_devices - i) * sizeof(handler->devices[0])); - handler->devices = talloc_realloc(handler, handler->devices, - struct device *, handler->n_devices); + return handler->devices[index]; } -/** - * device_handler_find - Find a handler device by id. - */ - -static struct device *device_handler_find(struct device_handler *handler, - const char *id) +struct discover_boot_option *discover_boot_option_create( + struct discover_context *ctx, + struct discover_device *device) { - unsigned int i; + struct discover_boot_option *opt; - assert(id); + opt = talloc_zero(ctx, struct discover_boot_option); + opt->option = talloc_zero(opt, struct boot_option); + opt->device = device; - for (i = 0; i < handler->n_devices; i++) - if (handler->devices[i]->id - && streq(handler->devices[i]->id, id)) - return handler->devices[i]; - - pb_log("%s: unknown device: %s\n", __func__, id); - return NULL; + return opt; } -/** - * device_handler_get_device_count - Get the count of current handler devices. - */ +static int device_match_uuid(struct discover_device *dev, const char *uuid) +{ + return dev->uuid && !strcmp(dev->uuid, uuid); +} -int device_handler_get_device_count(const struct device_handler *handler) +static int device_match_label(struct discover_device *dev, const char *label) { - return handler->n_devices; + return dev->label && !strcmp(dev->label, label); } -/** - * device_handler_get_device - Get a handler device by index. - */ +static int device_match_id(struct discover_device *dev, const char *id) +{ + return !strcmp(dev->device->id, id); +} -const struct device *device_handler_get_device( - const struct device_handler *handler, unsigned int index) +static struct discover_device *device_lookup( + struct device_handler *device_handler, + int (match_fn)(struct discover_device *, const char *), + const char *str) { - if (index >= handler->n_devices) { - assert(0 && "bad index"); + struct discover_device *dev; + unsigned int i; + + if (!str) return NULL; + + for (i = 0; i < device_handler->n_devices; i++) { + dev = device_handler->devices[i]; + + if (match_fn(dev, str)) + return dev; } - return handler->devices[index]; + return NULL; } -static void setup_device_links(struct discover_context *ctx) -{ - struct link { - char *env, *dir; - } *link, links[] = { - { - .env = "ID_FS_UUID", - .dir = "disk/by-uuid" - }, - { - .env = "ID_FS_LABEL", - .dir = "disk/by-label" - }, - { - .env = NULL - } - }; +struct discover_device *device_lookup_by_name(struct device_handler *handler, + const char *name) +{ + if (!strncmp(name, "/dev/", strlen("/dev/"))) + name += strlen("/dev/"); - for (link = links; link->env; link++) { - char *enc, *dir, *path; - const char *value; + return device_lookup_by_id(handler, name); +} - value = event_get_param(ctx->event, link->env); - if (!value || !*value) - continue; +struct discover_device *device_lookup_by_uuid( + struct device_handler *device_handler, + const char *uuid) +{ + return device_lookup(device_handler, device_match_uuid, uuid); +} - enc = encode_label(ctx, value); - dir = join_paths(ctx, mount_base(), link->dir); - path = join_paths(ctx, dir, value); - - if (!pb_mkdir_recursive(dir)) { - unlink(path); - if (symlink(ctx->mount_path, path)) { - pb_log("symlink(%s,%s): %s\n", - ctx->mount_path, path, - strerror(errno)); - talloc_free(path); - } else { - int i = ctx->n_links++; - ctx->links = talloc_realloc(ctx, - ctx->links, char *, - ctx->n_links); - ctx->links[i] = path; - } +struct discover_device *device_lookup_by_label( + struct device_handler *device_handler, + const char *label) +{ + return device_lookup(device_handler, device_match_label, label); +} - } +struct discover_device *device_lookup_by_id( + struct device_handler *device_handler, + const char *id) +{ + return device_lookup(device_handler, device_match_id, id); +} - talloc_free(dir); - talloc_free(enc); - } +void device_handler_destroy(struct device_handler *handler) +{ + talloc_free(handler); } -static void remove_device_links(struct discover_context *ctx) +#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))) { - int i; + struct device_handler *handler; - for (i = 0; i < ctx->n_links; i++) - unlink(ctx->links[i]); + handler = talloc_zero(NULL, struct device_handler); + list_init(&handler->unresolved_boot_options); + + return handler; } -static int mount_device(struct discover_context *ctx) +void device_handler_add_device(struct device_handler *handler, + struct discover_device *dev) { - const char *mountpoint; - const char *argv[6]; + handler->n_devices++; + handler->devices = talloc_realloc(handler, handler->devices, + struct discover_device *, handler->n_devices); + handler->devices[handler->n_devices - 1] = dev; +} - if (!ctx->mount_path) { - mountpoint = mountpoint_for_device(ctx->device_path); - ctx->mount_path = talloc_strdup(ctx, mountpoint); - } +#else - if (pb_mkdir_recursive(ctx->mount_path)) - pb_log("couldn't create mount directory %s: %s\n", - ctx->mount_path, strerror(errno)); +static int mount_device(struct discover_device *dev) +{ + int rc; - argv[0] = pb_system_apps.mount; - argv[1] = ctx->device_path; - argv[2] = ctx->mount_path; - argv[3] = "-o"; - argv[4] = "ro"; - argv[5] = NULL; + if (!dev->device_path) + return -1; - if (pb_run_cmd(argv, 1, 0)) { + if (!dev->mount_path) + dev->mount_path = join_paths(dev, mount_base(), + dev->device_path); - /* Retry mount without ro option. */ + if (pb_mkdir_recursive(dev->mount_path)) + pb_log("couldn't create mount directory %s: %s\n", + dev->mount_path, strerror(errno)); - argv[0] = pb_system_apps.mount; - argv[1] = ctx->device_path; - argv[2] = ctx->mount_path; - argv[3] = NULL; + rc = process_run_simple(dev, pb_system_apps.mount, + dev->device_path, dev->mount_path, + "-o", "ro", NULL); - if (pb_run_cmd(argv, 1, 0)) - goto out_rmdir; - } + if (!rc) + return 0; - setup_device_links(ctx); - return 0; + /* Retry mount without ro option. */ + rc = process_run_simple(dev, pb_system_apps.mount, + dev->device_path, dev->mount_path, NULL); + + if (!rc) + return 0; -out_rmdir: - pb_rmdir_recursive(mount_base(), ctx->mount_path); + pb_rmdir_recursive(mount_base(), dev->mount_path); return -1; } -static int umount_device(struct discover_context *ctx) +static int umount_device(struct discover_device *dev) { int status; - pid_t pid; - remove_device_links(ctx); + if (!dev->mount_path) + return 0; + + status = process_run_simple(dev, pb_system_apps.umount, + dev->mount_path, NULL); - pid = fork(); - if (pid == -1) { - pb_log("%s: fork failed: %s\n", __func__, strerror(errno)); + if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) return -1; - } - if (pid == 0) { - execl(pb_system_apps.umount, pb_system_apps.umount, - ctx->mount_path, NULL); - exit(EXIT_FAILURE); - } + pb_rmdir_recursive(mount_base(), dev->mount_path); - if (waitpid(pid, &status, 0) == -1) { - pb_log("%s: waitpid failed: %s\n", __func__, - strerror(errno)); - return -1; - } + return 0; +} - if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) - return -1; +struct device_handler *device_handler_init(struct discover_server *server, + struct waitset *waitset, int dry_run) +{ + struct device_handler *handler; + + handler = talloc(NULL, struct device_handler); + handler->devices = NULL; + handler->n_devices = 0; + 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 */ + pb_mkdir_recursive(mount_base()); + + parser_init(); + + return handler; +} + +static int destroy_device(void *arg) +{ + struct discover_device *dev = arg; - pb_rmdir_recursive(mount_base(), ctx->mount_path); + umount_device(dev); return 0; } -static struct discover_context *find_context(struct device_handler *handler, +static struct discover_device *find_device(struct device_handler *handler, const char *id) { - struct discover_context *ctx; + struct discover_device *dev; + unsigned int i; - list_for_each_entry(&handler->contexts, ctx, list) { - if (!strcmp(ctx->id, id)) - return ctx; + for (i = 0; i < handler->n_devices; i++) { + dev = handler->devices[i]; + if (!strcmp(dev->device->id, id)) + return dev; } return NULL; } -static int destroy_context(void *arg) +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) +{ + unsigned int i; + + for (i = 0; i < handler->n_devices; i++) + if (handler->devices[i] == device) + break; + + if (i == handler->n_devices) { + assert(0 && "unknown device"); + return; + } + + handler->n_devices--; + memmove(&handler->devices[i], &handler->devices[i + 1], + (handler->n_devices - i) * sizeof(handler->devices[0])); + handler->devices = talloc_realloc(handler, handler->devices, + struct discover_device *, handler->n_devices); + + discover_server_notify_device_remove(handler->server, device->device); + + talloc_free(device); +} + +static void boot_status(void *arg, struct boot_status *status) +{ + struct device_handler *handler = arg; + + discover_server_notify_boot_status(handler->server, status); +} + +static void countdown_status(struct device_handler *handler, + struct discover_boot_option *opt, unsigned int sec) { - struct discover_context *ctx = arg; + struct boot_status status; + + status.type = BOOT_STATUS_INFO; + status.progress = -1; + status.detail = NULL; + status.message = talloc_asprintf(handler, + "Booting %s in %u sec", opt->option->name, sec); + + discover_server_notify_boot_status(handler->server, &status); + + talloc_free(status.message); +} + +static int default_timeout(void *arg) +{ + struct device_handler *handler = arg; + struct discover_boot_option *opt; + + if (!handler->default_boot_option) + return 0; + + opt = handler->default_boot_option; + + if (handler->sec_to_boot) { + countdown_status(handler, opt, handler->sec_to_boot); + handler->sec_to_boot--; + handler->timeout_waiter = waiter_register_timeout( + handler->waitset, 1000, + default_timeout, handler); + return 0; + } - list_remove(&ctx->list); - umount_device(ctx); + pb_log("Timeout expired, booting default option %s\n", opt->option->id); + boot(handler, handler->default_boot_option, NULL, + handler->dry_run, boot_status, handler); return 0; } +static void set_default(struct device_handler *handler, + struct discover_boot_option *opt) +{ + if (handler->default_boot_option) + return; + + if (!handler->autoboot_enabled) + return; + + handler->default_boot_option = opt; + handler->sec_to_boot = config_get()->autoboot_timeout_sec; + + pb_log("Boot option %s set as default, timeout %u sec.\n", + opt->option->id, handler->sec_to_boot); + + default_timeout(handler); +} + +static bool resource_is_resolved(struct resource *res) +{ + return !res || res->resolved; +} + +/* We only use this in an assert, which will disappear if we're compiling + * with NDEBUG, so we need the 'used' attribute for these builds */ +static bool __attribute__((used)) boot_option_is_resolved( + struct discover_boot_option *opt) +{ + return resource_is_resolved(opt->boot_image) && + resource_is_resolved(opt->initrd) && + resource_is_resolved(opt->dtb) && + resource_is_resolved(opt->icon); +} + +static bool resource_resolve(struct resource *res, const char *name, + struct discover_boot_option *opt, + struct device_handler *handler) +{ + struct parser *parser = opt->source; + + if (resource_is_resolved(res)) + return true; + + pb_log("Attempting to resolve resource %s->%s with parser %s\n", + opt->option->id, name, parser->name); + parser->resolve_resource(handler, res); + + return res->resolved; +} + +static bool boot_option_resolve(struct discover_boot_option *opt, + struct device_handler *handler) +{ + return resource_resolve(opt->boot_image, "boot_image", opt, handler) && + resource_resolve(opt->initrd, "initrd", opt, handler) && + resource_resolve(opt->dtb, "dtb", opt, handler) && + resource_resolve(opt->icon, "icon", opt, handler); +} + +static void boot_option_finalise(struct device_handler *handler, + struct discover_boot_option *opt) +{ + assert(boot_option_is_resolved(opt)); + + /* check that the parsers haven't set any of the final data */ + assert(!opt->option->boot_image_file); + assert(!opt->option->initrd_file); + assert(!opt->option->dtb_file); + assert(!opt->option->icon_file); + assert(!opt->option->device_id); + + if (opt->boot_image) + opt->option->boot_image_file = opt->boot_image->url->full; + if (opt->initrd) + opt->option->initrd_file = opt->initrd->url->full; + if (opt->dtb) + opt->option->dtb_file = opt->dtb->url->full; + if (opt->icon) + opt->option->icon_file = opt->icon->url->full; + + opt->option->device_id = opt->device->device->id; + + if (opt->option->is_default) + set_default(handler, opt); +} + +static void process_boot_option_queue(struct device_handler *handler) +{ + struct discover_boot_option *opt, *tmp; + + list_for_each_entry_safe(&handler->unresolved_boot_options, + opt, tmp, list) { + + pb_log("queue: attempting resolution for %s\n", + opt->option->id); + + if (!boot_option_resolve(opt, handler)) + continue; + + pb_log("\tresolved!\n"); + + list_remove(&opt->list); + 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); + } +} + +/** + * 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, + 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); + } + + + /* move boot options from the context to the device */ + list_for_each_entry_safe(&ctx->boot_options, opt, tmp, list) { + list_remove(&opt->list); + + if (boot_option_resolve(opt, handler)) { + pb_log("boot option %s is resolved, " + "sending to clients\n", + opt->option->id); + 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); + } else { + if (!opt->source->resolve_resource) { + pb_log("parser %s gave us an unresolved " + "resource (%s), but no way to " + "resolve it\n", + opt->source->name, opt->option->id); + talloc_free(opt); + } else { + pb_log("boot option %s is unresolved, " + "adding to queue\n", + opt->option->id); + list_add(&handler->unresolved_boot_options, + &opt->list); + talloc_steal(handler, opt); + } + } + } +} + static int handle_add_udev_event(struct device_handler *handler, struct event *event) { struct discover_context *ctx; - struct boot_option *opt; - const char *devname; + struct discover_device *dev; + const char *param; int rc; /* create our context */ ctx = talloc(handler, struct discover_context); ctx->event = event; - ctx->mount_path = NULL; - ctx->links = NULL; - ctx->n_links = 0; + list_init(&ctx->boot_options); + + /* create our top-level device */ + dev = discover_device_create(handler, ctx, event); - ctx->id = talloc_strdup(ctx, event->device); + ctx->device = dev; - devname = event_get_param(ctx->event, "DEVNAME"); - assert(devname); - ctx->device_path = talloc_strdup(ctx, devname); + /* try to parse UUID and labels */ + param = event_get_param(ctx->event, "ID_FS_UUID"); + if (param) + dev->uuid = talloc_strdup(dev, param); - rc = mount_device(ctx); + param = event_get_param(ctx->event, "ID_FS_LABEL"); + if (param) + dev->label = talloc_strdup(dev, param); + + rc = mount_device(dev); if (rc) { talloc_free(ctx); return 0; } - list_add(&handler->contexts, &ctx->list); - talloc_set_destructor(ctx, destroy_context); - - /* set up the top-level device */ - ctx->device = talloc_zero(ctx, struct device); - ctx->device->id = talloc_strdup(ctx->device, ctx->id); - list_init(&ctx->device->boot_options); + /* run the parsers. This will populate the ctx's boot_option list. */ + iterate_parsers(ctx, CONF_METHOD_LOCAL_FILE); - /* run the parsers */ - iterate_parsers(ctx); + /* add discovered stuff to the handler */ + context_commit(handler, ctx); - /* add device to handler device array */ - device_handler_add(handler, ctx->device); - - discover_server_notify_device_add(handler->server, ctx->device); - - list_for_each_entry(&ctx->device->boot_options, opt, list) - discover_server_notify_boot_option_add(handler->server, opt); + talloc_free(ctx); return 0; } @@ -323,18 +635,14 @@ static int handle_add_udev_event(struct device_handler *handler, static int handle_remove_udev_event(struct device_handler *handler, struct event *event) { - struct discover_context *ctx; + struct discover_device *dev; - ctx = find_context(handler, event->device); - if (!ctx) + dev = find_device(handler, event->device); + if (!dev) return 0; - discover_server_notify_device_remove(handler->server, ctx->device); - /* remove device from handler device array */ - device_handler_remove(handler, ctx->device); - - talloc_free(ctx); + device_handler_remove(handler, dev); return 0; } @@ -342,52 +650,93 @@ static int handle_remove_udev_event(struct device_handler *handler, static int handle_add_user_event(struct device_handler *handler, struct event *event) { - struct boot_option *opt; - struct device *device; + struct discover_context *ctx; + struct discover_device *dev; + int rc; assert(event->device); - device = device_handler_find(handler, event->device); + ctx = talloc(handler, struct discover_context); + ctx->event = event; + list_init(&ctx->boot_options); - if (!device) { - device = talloc_zero(handler, struct device); + dev = discover_device_create(handler, ctx, event); + ctx->device = dev; - if (!device) - goto fail; + rc = parse_user_event(ctx, event); - device->id = talloc_strdup(device, event->device); - list_init(&device->boot_options); + if (!rc) + context_commit(handler, ctx); - /* add device to handler device array */ - device_handler_add(handler, device); + return rc; +} - discover_server_notify_device_add(handler->server, device); - } +static int handle_remove_user_event(struct device_handler *handler, + struct event *event) +{ + struct discover_device *dev = find_device(handler, event->device); - opt = parse_user_event(device, event); - discover_server_notify_boot_option_add(handler->server, opt); + if (!dev) + return 0; - return 0; + /* remove device from handler device array */ + device_handler_remove(handler, dev); -fail: - talloc_free(device); return 0; } -static int handle_remove_user_event(struct device_handler *handler, +static enum conf_method parse_conf_method(const char *str) +{ + + if (!strcasecmp(str, "dhcp")) { + return CONF_METHOD_DHCP; + } + return CONF_METHOD_UNKNOWN; +} + +static int handle_conf_user_event(struct device_handler *handler, struct event *event) { - struct device *device = device_handler_find(handler, event->device); + struct discover_context *ctx; + struct discover_device *dev; + enum conf_method method; + const char *val; - if (!device) + ctx = talloc(handler, struct discover_context); + ctx->event = event; + list_init(&ctx->boot_options); + + val = event_get_param(event, "url"); + if (!val) { + talloc_free(ctx); return 0; + } - discover_server_notify_device_remove(handler->server, device); + ctx->conf_url = pb_url_parse(ctx, val); + if (!ctx->conf_url) { + talloc_free(ctx); + return 0; + } - /* remove device from handler device array */ - device_handler_remove(handler, device); + val = event_get_param(event, "method"); + if (!val) { + talloc_free(ctx); + return 0; + } + + method = parse_conf_method(val); + if (method == CONF_METHOD_UNKNOWN) { + talloc_free(ctx); + return 0; + } + + dev = discover_device_create(handler, ctx, event); + ctx->device = dev; + + iterate_parsers(ctx, method); + + context_commit(handler, ctx); - talloc_free(device); return 0; } @@ -401,6 +750,7 @@ static event_handler handlers[EVENT_TYPE_MAX][EVENT_ACTION_MAX] = { [EVENT_TYPE_USER] = { [EVENT_ACTION_ADD] = handle_add_user_event, [EVENT_ACTION_REMOVE] = handle_remove_user_event, + [EVENT_ACTION_CONF] = handle_conf_user_event, } }; @@ -418,43 +768,17 @@ int device_handler_event(struct device_handler *handler, return handlers[event->type][event->action](handler, event); } -struct device_handler *device_handler_init(struct discover_server *server, - int dry_run) -{ - struct device_handler *handler; - - handler = talloc(NULL, struct device_handler); - handler->devices = NULL; - handler->n_devices = 0; - handler->server = server; - handler->dry_run = dry_run; - - list_init(&handler->contexts); - - /* set up our mount point base */ - pb_mkdir_recursive(mount_base()); - - parser_init(); - - return handler; -} - -void device_handler_destroy(struct device_handler *handler) -{ - talloc_free(handler); -} - -static struct boot_option *find_boot_option_by_id( +static struct discover_boot_option *find_boot_option_by_id( struct device_handler *handler, const char *id) { unsigned int i; for (i = 0; i < handler->n_devices; i++) { - struct device *dev = handler->devices[i]; - struct boot_option *opt; + struct discover_device *dev = handler->devices[i]; + struct discover_boot_option *opt; list_for_each_entry(&dev->boot_options, opt, list) - if (!strcmp(opt->id, id)) + if (!strcmp(opt->option->id, id)) return opt; } @@ -464,9 +788,36 @@ static struct boot_option *find_boot_option_by_id( void device_handler_boot(struct device_handler *handler, struct boot_command *cmd) { - struct boot_option *opt; + struct discover_boot_option *opt; opt = find_boot_option_by_id(handler, cmd->option_id); - boot(handler, opt, cmd, handler->dry_run); + boot(handler, opt, cmd, handler->dry_run, boot_status, handler); +} + +void device_handler_cancel_default(struct device_handler *handler) +{ + struct boot_status status; + + if (handler->timeout_waiter) + waiter_remove(handler->timeout_waiter); + + handler->timeout_waiter = NULL; + handler->autoboot_enabled = false; + + /* 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); } +#endif