X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fdevice-handler.c;h=d95a37cb0c7c41ae420001d4c227cfa012c33b2f;hp=a8df295df1b6e4763516b68de240c6fa907240a7;hb=120855d64d783aa3b3b0af23923da0169319c338;hpb=5444648fe1ff9b79f3db5ee6feadd51341f59d71 diff --git a/discover/device-handler.c b/discover/device-handler.c index a8df295..d95a37c 100644 --- a/discover/device-handler.c +++ b/discover/device-handler.c @@ -1,38 +1,580 @@ - #include #include +#include #include #include #include +#include +#include #include #include +#include #include #include #include #include #include +#include #include +#include + +#include +#include +#include +#include #include "device-handler.h" #include "discover-server.h" +#include "user-event.h" +#include "platform.h" #include "event.h" #include "parser.h" #include "resource.h" -#include "udev.h" #include "paths.h" +#include "sysinfo.h" #include "boot.h" +#include "udev.h" +#include "network.h" +#include "ipmi.h" + +enum default_priority { + DEFAULT_PRIORITY_REMOTE = 1, + DEFAULT_PRIORITY_LOCAL_FIRST = 2, + DEFAULT_PRIORITY_LOCAL_LAST = 0xfe, + DEFAULT_PRIORITY_DISABLED = 0xff, +}; struct device_handler { struct discover_server *server; int dry_run; + struct pb_udev *udev; + struct network *network; + struct user_event *user_event; + struct discover_device **devices; unsigned int n_devices; + struct waitset *waitset; + struct waiter *timeout_waiter; + bool autoboot_enabled; + unsigned int sec_to_boot; + + struct discover_boot_option *default_boot_option; + int default_boot_option_priority; + struct list unresolved_boot_options; + + struct boot_task *pending_boot; + bool pending_boot_is_default; +}; + +static int mount_device(struct discover_device *dev); +static int umount_device(struct discover_device *dev); + +static int device_handler_init_sources(struct device_handler *handler); +static void device_handler_reinit_sources(struct device_handler *handler); + +static void device_handler_update_lang(const char *lang); + +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_get_device_count - Get the count of current handler devices. + */ + +int device_handler_get_device_count(const struct device_handler *handler) +{ + return handler->n_devices; +} + +/** + * device_handler_get_device - Get a handler device by index. + */ + +const struct discover_device *device_handler_get_device( + const struct device_handler *handler, unsigned int index) +{ + if (index >= handler->n_devices) { + assert(0 && "bad index"); + return NULL; + } + + return handler->devices[index]; +} + +struct discover_boot_option *discover_boot_option_create( + struct discover_context *ctx, + struct discover_device *device) +{ + struct discover_boot_option *opt; + + opt = talloc_zero(ctx, struct discover_boot_option); + opt->option = talloc_zero(opt, struct boot_option); + opt->device = device; + + return opt; +} + +static int device_match_uuid(struct discover_device *dev, const char *uuid) +{ + return dev->uuid && !strcmp(dev->uuid, uuid); +} + +static int device_match_label(struct discover_device *dev, const char *label) +{ + return dev->label && !strcmp(dev->label, label); +} + +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 *), + const char *str) +{ + 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 NULL; +} + +struct discover_device *device_lookup_by_name(struct device_handler *handler, + const char *name) +{ + if (!strncmp(name, "/dev/", strlen("/dev/"))) + name += strlen("/dev/"); + + return device_lookup_by_id(handler, name); +} + +struct discover_device *device_lookup_by_uuid( + struct device_handler *device_handler, + const char *uuid) +{ + return device_lookup(device_handler, device_match_uuid, uuid); +} + +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); +} + +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); +} + +static int destroy_device(void *arg) +{ + struct discover_device *dev = arg; + + umount_device(dev); + + return 0; +} + +struct discover_device *discover_device_create(struct device_handler *handler, + const char *id) +{ + struct discover_device *dev; + + dev = device_lookup_by_id(handler, id); + if (dev) + return dev; + + 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); + + talloc_set_destructor(dev, destroy_device); + + return dev; +} + +struct discover_device_param { + char *name; + char *value; + struct list_item list; }; +void discover_device_set_param(struct discover_device *device, + const char *name, const char *value) +{ + struct discover_device_param *param; + bool found = false; + + list_for_each_entry(&device->params, param, list) { + if (!strcmp(param->name, name)) { + found = true; + break; + } + } + + 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); + } + + param->value = talloc_strdup(param, value); +} + +const char *discover_device_get_param(struct discover_device *device, + const char *name) +{ + struct discover_device_param *param; + + list_for_each_entry(&device->params, param, list) { + if (!strcmp(param->name, name)) + return param->value; + } + return NULL; +} + +struct device_handler *device_handler_init(struct discover_server *server, + struct waitset *waitset, int dry_run) +{ + struct device_handler *handler; + int rc; + + handler = talloc_zero(NULL, struct device_handler); + handler->server = server; + handler->waitset = waitset; + handler->dry_run = dry_run; + 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(); + + if (config_get()->safe_mode) + return handler; + + rc = device_handler_init_sources(handler); + if (rc) { + talloc_free(handler); + return NULL; + } + + return handler; +} + +void device_handler_reinit(struct device_handler *handler) +{ + struct discover_boot_option *opt, *tmp; + unsigned int i; + + device_handler_cancel_default(handler); + + /* free unresolved boot options */ + list_for_each_entry_safe(&handler->unresolved_boot_options, + opt, tmp, list) + talloc_free(opt); + list_init(&handler->unresolved_boot_options); + + /* drop all devices */ + for (i = 0; i < handler->n_devices; i++) + discover_server_notify_device_remove(handler->server, + handler->devices[i]->device); + + talloc_free(handler->devices); + handler->devices = NULL; + handler->n_devices = 0; + + device_handler_reinit_sources(handler); +} + +void device_handler_remove(struct device_handler *handler, + struct discover_device *device) +{ + struct discover_boot_option *opt, *tmp; + unsigned int i; + + for (i = 0; i < handler->n_devices; i++) + if (handler->devices[i] == device) + break; + + if (i == handler->n_devices) { + talloc_free(device); + return; + } + + /* Free any unresolved options, as they're currently allocated + * against the handler */ + list_for_each_entry_safe(&handler->unresolved_boot_options, + opt, tmp, list) { + if (opt->device != device) + continue; + list_remove(&opt->list); + talloc_free(opt); + } + + /* if this is a network device, we have to unregister it from the + * network code */ + if (device->device->type == DEVICE_TYPE_NETWORK) + network_unregister_device(handler->network, device); + + 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); + + if (device->notified) + 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 boot_status status; + + status.type = BOOT_STATUS_INFO; + status.progress = -1; + status.detail = NULL; + status.message = talloc_asprintf(handler, + _("Booting in %d sec: %s"), sec, opt->option->name); + + 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; + + if (handler->pending_boot) + 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; + } + + handler->timeout_waiter = NULL; + + pb_log("Timeout expired, booting default option %s\n", opt->option->id); + + handler->pending_boot = boot(handler, handler->default_boot_option, + NULL, handler->dry_run, boot_status, handler); + handler->pending_boot_is_default = true; + return 0; +} + +struct { + enum ipmi_bootdev ipmi_type; + enum device_type device_type; +} device_type_map[] = { + { IPMI_BOOTDEV_NETWORK, DEVICE_TYPE_NETWORK }, + { IPMI_BOOTDEV_DISK, DEVICE_TYPE_DISK }, + { IPMI_BOOTDEV_CDROM, DEVICE_TYPE_OPTICAL }, +}; + +static bool ipmi_device_type_matches(enum ipmi_bootdev ipmi_type, + enum device_type device_type) +{ + unsigned int i; + + for (i = 0; i < ARRAY_SIZE(device_type_map); i++) { + if (device_type_map[i].device_type == device_type) + return device_type_map[i].ipmi_type == ipmi_type; + } + + return false; +} + +static int autoboot_option_priority(const struct config *config, + struct discover_boot_option *opt) +{ + enum device_type type = opt->device->device->type; + const char *uuid = opt->device->uuid; + struct autoboot_option *auto_opt; + unsigned int i; + + for (i = 0; i < config->n_autoboot_opts; i++) { + auto_opt = &config->autoboot_opts[i]; + if (auto_opt->boot_type == BOOT_DEVICE_UUID) + if (!strcmp(auto_opt->uuid, uuid)) + return DEFAULT_PRIORITY_LOCAL_FIRST + i; + + if (auto_opt->boot_type == BOOT_DEVICE_TYPE) + if (auto_opt->type == type || + auto_opt->type == DEVICE_TYPE_ANY) + return DEFAULT_PRIORITY_LOCAL_FIRST + i; + } + + return -1; +} + +/* + * We have different priorities to resolve conflicts between boot options that + * report to be the default for their device. This function assigns a priority + * for these options. + */ +static enum default_priority default_option_priority( + struct discover_boot_option *opt) +{ + const struct config *config; + + config = config_get(); + + /* We give highest priority to IPMI-configured boot options. If + * we have an IPMI bootdev configuration set, then we don't allow + * any other defaults */ + if (config->ipmi_bootdev) { + bool ipmi_match = ipmi_device_type_matches(config->ipmi_bootdev, + opt->device->device->type); + if (ipmi_match) + return DEFAULT_PRIORITY_REMOTE; + + pb_debug("handler: disabled default priority due to " + "non-matching IPMI type %x\n", + config->ipmi_bootdev); + return DEFAULT_PRIORITY_DISABLED; + } + + /* Next, try to match the option against the user-defined autoboot + * options, either by device UUID or type. */ + if (config->n_autoboot_opts) { + int boot_match = autoboot_option_priority(config, opt); + if (boot_match > 0) + return boot_match; + } + + /* If the option didn't match any entry in the array, it is disabled */ + pb_debug("handler: disabled default priority due to " + "non-matching UUID or type\n"); + return DEFAULT_PRIORITY_DISABLED; +} + +static void set_default(struct device_handler *handler, + struct discover_boot_option *opt) +{ + enum default_priority cur_prio, new_prio; + + if (!handler->autoboot_enabled) + return; + + pb_debug("handler: new default option: %s\n", opt->option->id); + + new_prio = default_option_priority(opt); + + /* Anything outside our range prevents a default boot */ + if (new_prio >= DEFAULT_PRIORITY_DISABLED) + return; + + pb_debug("handler: calculated priority %d\n", new_prio); + + /* 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) { + + cur_prio = handler->default_boot_option_priority; + + if (new_prio < cur_prio) { + pb_log("handler: new prio %d beats " + "old prio %d for %s\n", + new_prio, cur_prio, + handler->default_boot_option + ->option->id); + handler->default_boot_option = opt; + handler->default_boot_option_priority = new_prio; + /* extend the timeout a little, so the user sees some + * indication of the change */ + handler->sec_to_boot += 2; + } + + return; + } + + handler->sec_to_boot = config_get()->autoboot_timeout_sec; + handler->default_boot_option = opt; + handler->default_boot_option_priority = new_prio; + + pb_log("handler: 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; @@ -45,15 +587,21 @@ static bool __attribute__((used)) boot_option_is_resolved( { 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, struct parser *parser, +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_debug("Attempting to resolve resource %s->%s with parser %s\n", + opt->option->id, name, parser->name); parser->resolve_resource(handler, res); return res->resolved; @@ -62,26 +610,49 @@ static bool resource_resolve(struct resource *res, struct parser *parser, static bool boot_option_resolve(struct discover_boot_option *opt, struct device_handler *handler) { - return resource_resolve(opt->boot_image, opt->source, handler) && - resource_resolve(opt->initrd, opt->source, handler) && - resource_resolve(opt->icon, opt->source, 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 discover_boot_option *opt) +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 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) @@ -91,528 +662,711 @@ static void process_boot_option_queue(struct device_handler *handler) list_for_each_entry_safe(&handler->unresolved_boot_options, opt, tmp, list) { + pb_debug("queue: attempting resolution for %s\n", + opt->option->id); + if (!boot_option_resolve(opt, handler)) continue; + pb_debug("\tresolved!\n"); + list_remove(&opt->list); - list_add(&opt->device->boot_options, &opt->list); + list_add_tail(&opt->device->boot_options, &opt->list); talloc_steal(opt->device, opt); - boot_option_finalise(opt); - discover_server_notify_boot_option_add(handler->server, - opt->option); + boot_option_finalise(handler, opt); + 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_zero(handler, struct discover_context); + ctx->device = device; + ctx->network = handler->network; + 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 */ - 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) { list_remove(&opt->list); if (boot_option_resolve(opt, handler)) { - list_add(&dev->boot_options, &opt->list); + 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(opt); - discover_server_notify_boot_option_add(handler->server, - opt->option); + boot_option_finalise(handler, opt); + notify_boot_option(handler, opt); } else { - list_add(&handler->unresolved_boot_options, &opt->list); - talloc_steal(handler, opt); + 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); + } } } } -void discover_context_add_boot_option(struct discover_context *ctx, - struct discover_boot_option *boot_option) +void device_handler_add_device(struct device_handler *handler, + struct discover_device *device) { - boot_option->source = ctx->parser; - list_add(&ctx->boot_options, &boot_option->list); - talloc_steal(ctx, boot_option); + handler->n_devices++; + handler->devices = talloc_realloc(handler, handler->devices, + struct discover_device *, handler->n_devices); + handler->devices[handler->n_devices - 1] = device; + + if (device->device->type == DEVICE_TYPE_NETWORK) + network_register_device(handler->network, device); } -/** - * device_handler_remove - Remove a device from the handler device array. +/* Start discovery on a hotplugged device. The device will be in our devices + * array, but has only just been initialised by the hotplug source. */ - -static void device_handler_remove(struct device_handler *handler, - struct discover_device *device) +int device_handler_discover(struct device_handler *handler, + struct discover_device *dev) { - unsigned int i; + struct discover_context *ctx; + struct boot_status *status; + int rc; - for (i = 0; i < handler->n_devices; i++) - if (handler->devices[i] == device) - break; + status = talloc_zero(handler, struct boot_status); + status->type = BOOT_STATUS_INFO; + status->message = talloc_asprintf(status, "Processing %s device %s", + device_type_display_name(dev->device->type), + dev->device->id); + boot_status(handler, status); - if (i == handler->n_devices) { - assert(0 && "unknown device"); - return; - } + process_boot_option_queue(handler); - 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); + /* create our context */ + ctx = device_handler_discover_context_create(handler, dev); - discover_server_notify_device_remove(handler->server, device->device); + rc = mount_device(dev); + if (rc) + goto out; - talloc_free(device); -} + /* add this device to our system info */ + system_info_register_blockdev(dev->device->id, dev->uuid, + dev->mount_path); -/** - * device_handler_get_device_count - Get the count of current handler devices. - */ + /* run the parsers. This will populate the ctx's boot_option list. */ + iterate_parsers(ctx); -int device_handler_get_device_count(const struct device_handler *handler) -{ - return handler->n_devices; -} + /* add discovered stuff to the handler */ + device_handler_discover_context_commit(handler, ctx); -/** - * device_handler_get_device - Get a handler device by index. - */ +out: + status->message = talloc_asprintf(status,"Processing %s complete\n", + dev->device->id); + boot_status(handler, status); -const struct discover_device *device_handler_get_device( - const struct device_handler *handler, unsigned int index) -{ - if (index >= handler->n_devices) { - assert(0 && "bad index"); - return NULL; - } + talloc_free(status); + talloc_free(ctx); - return handler->devices[index]; + return 0; } -static int mount_device(struct discover_device *dev) +/* Incoming dhcp event */ +int device_handler_dhcp(struct device_handler *handler, + struct discover_device *dev, struct event *event) { - const char *argv[6]; - - if (!dev->mount_path) - 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)); + struct discover_context *ctx; + struct boot_status *status; - argv[0] = pb_system_apps.mount; - argv[1] = dev->device_path; - argv[2] = dev->mount_path; - argv[3] = "-o"; - argv[4] = "ro"; - argv[5] = NULL; + status = talloc_zero(handler, struct boot_status); + status->type = BOOT_STATUS_INFO; + status->message = talloc_asprintf(status, "Processing dhcp event on %s", + dev->device->id); + boot_status(handler, status); - if (pb_run_cmd(argv, 1, 0)) { + /* create our context */ + ctx = device_handler_discover_context_create(handler, dev); + ctx->event = event; - /* Retry mount without ro option. */ + iterate_parsers(ctx); - argv[0] = pb_system_apps.mount; - argv[1] = dev->device_path; - argv[2] = dev->mount_path; - argv[3] = NULL; + device_handler_discover_context_commit(handler, ctx); - if (pb_run_cmd(argv, 1, 0)) - goto out_rmdir; - } + status->message = talloc_asprintf(status,"Processing %s complete\n", + dev->device->id); + boot_status(handler, status); - return 0; + talloc_free(status); + talloc_free(ctx); -out_rmdir: - pb_rmdir_recursive(mount_base(), dev->mount_path); - return -1; + return 0; } -static int umount_device(struct discover_device *dev) +/* incoming conf event */ +int device_handler_conf(struct device_handler *handler, + struct discover_device *dev, struct pb_url *url) { - int status; - pid_t pid; + struct discover_context *ctx; + struct boot_status *status; - if (!dev->mount_path) - return 0; + status = talloc_zero(handler, struct boot_status); + status->type = BOOT_STATUS_INFO; + status->message = talloc_asprintf(status, "Processing user config"); + boot_status(handler, status); - pid = fork(); - if (pid == -1) { - pb_log("%s: fork failed: %s\n", __func__, strerror(errno)); - return -1; - } + /* create our context */ + ctx = device_handler_discover_context_create(handler, dev); + ctx->conf_url = url; - if (pid == 0) { - execl(pb_system_apps.umount, pb_system_apps.umount, - dev->mount_path, NULL); - exit(EXIT_FAILURE); - } + iterate_parsers(ctx); - if (waitpid(pid, &status, 0) == -1) { - pb_log("%s: waitpid failed: %s\n", __func__, - strerror(errno)); - return -1; - } + device_handler_discover_context_commit(handler, ctx); - if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) - return -1; + status->message = talloc_asprintf(status, + "Processing user config complete"); + boot_status(handler, status); - pb_rmdir_recursive(mount_base(), dev->mount_path); + talloc_free(status); + talloc_free(ctx); - return 0; + return 0; } -static struct discover_device *find_device(struct device_handler *handler, - const char *id) +static struct discover_boot_option *find_boot_option_by_id( + 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; + struct discover_device *dev = handler->devices[i]; + struct discover_boot_option *opt; + + list_for_each_entry(&dev->boot_options, opt, list) + if (!strcmp(opt->option->id, id)) + return opt; } return NULL; } -static int destroy_device(void *arg) +void device_handler_boot(struct device_handler *handler, + struct boot_command *cmd) { - struct discover_device *dev = arg; + struct discover_boot_option *opt = NULL; - umount_device(dev); + if (cmd->option_id && strlen(cmd->option_id)) + opt = find_boot_option_by_id(handler, cmd->option_id); - return 0; + if (handler->pending_boot) + boot_cancel(handler->pending_boot); + + platform_pre_boot(); + + handler->pending_boot = boot(handler, opt, cmd, handler->dry_run, + boot_status, handler); + handler->pending_boot_is_default = false; } -static struct discover_device *discover_device_create( - struct device_handler *handler, - struct discover_context *ctx, - struct event *event) +void device_handler_cancel_default(struct device_handler *handler) { - struct discover_device *dev; - const char *devname; + struct boot_status status; - dev = find_device(handler, event->device); - if (dev) - return dev; + if (handler->timeout_waiter) + waiter_remove(handler->timeout_waiter); - dev = talloc_zero(ctx, struct discover_device); - dev->device = talloc_zero(dev, struct device); - list_init(&dev->boot_options); + handler->timeout_waiter = NULL; + handler->autoboot_enabled = false; - devname = event_get_param(ctx->event, "DEVNAME"); - if (devname) - dev->device_path = talloc_strdup(dev, devname); + /* we only send status if we had a default boot option queued */ + if (!handler->default_boot_option) + return; - dev->device->id = talloc_strdup(dev, event->device); + pb_log("Cancelling default boot option\n"); - talloc_set_destructor(dev, destroy_device); + if (handler->pending_boot && handler->pending_boot_is_default) { + boot_cancel(handler->pending_boot); + handler->pending_boot = NULL; + handler->pending_boot_is_default = false; + } - return dev; + 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); } -struct discover_boot_option *discover_boot_option_create( - struct discover_context *ctx, - struct discover_device *device) +void device_handler_update_config(struct device_handler *handler, + struct config *config) { - struct discover_boot_option *opt; + int rc; - opt = talloc_zero(ctx, struct discover_boot_option); - opt->option = talloc_zero(opt, struct boot_option); - opt->device = device; + rc = config_set(config); + if (rc) + return; - return opt; + discover_server_notify_config(handler->server, config); + device_handler_update_lang(config->lang); + device_handler_reinit(handler); } - -static int handle_add_udev_event(struct device_handler *handler, - struct event *event) +static char *device_from_addr(void *ctx, struct pb_url *url) { - struct discover_context *ctx; - struct discover_device *dev; - const char *param; + char *ipaddr, *buf, *tok, *dev = NULL; + const char *delim = " "; + struct sockaddr_in *ip; + struct sockaddr_in si; + struct addrinfo *res; + struct process *p; int rc; - /* create our context */ - ctx = talloc(handler, struct discover_context); - ctx->event = event; - list_init(&ctx->boot_options); + /* Note: IPv4 only */ + rc = inet_pton(AF_INET, url->host, &(si.sin_addr)); + if (rc > 0) { + ipaddr = url->host; + } else { + /* need to turn hostname into a valid IP */ + rc = getaddrinfo(url->host, NULL, NULL, &res); + if (rc) { + pb_debug("%s: Invalid URL\n",__func__); + return NULL; + } + ipaddr = talloc_array(ctx,char,INET_ADDRSTRLEN); + ip = (struct sockaddr_in *) res->ai_addr; + inet_ntop(AF_INET, &(ip->sin_addr), ipaddr, INET_ADDRSTRLEN); + freeaddrinfo(res); + } - /* create our top-level device */ - dev = discover_device_create(handler, ctx, event); + const char *argv[] = { + pb_system_apps.ip, + "route", "show", "to", "match", + ipaddr, + NULL + }; - ctx->device = dev; + p = process_create(ctx); - /* try to parse UUID and labels */ - param = event_get_param(ctx->event, "ID_FS_UUID"); - if (param) - dev->uuid = talloc_strdup(dev, param); + p->path = pb_system_apps.ip; + p->argv = argv; + p->keep_stdout = true; - param = event_get_param(ctx->event, "ID_FS_LABEL"); - if (param) - dev->label = talloc_strdup(dev, param); + rc = process_run_sync(p); - rc = mount_device(dev); if (rc) { - talloc_free(ctx); - return 0; + /* ip has complained for some reason; most likely + * there is no route to the host - bail out */ + pb_debug("%s: No route to %s\n",__func__,url->host); + return NULL; } - /* run the parsers. This will populate the ctx's boot_option list. */ + buf = p->stdout_buf; + /* If a route is found, ip-route output will be of the form + * "... dev DEVNAME ... " */ + tok = strtok(buf, delim); + while (tok) { + if (!strcmp(tok, "dev")) { + tok = strtok(NULL, delim); + dev = talloc_strdup(ctx, tok); + break; + } + tok = strtok(NULL, delim); + } + + process_release(p); + if (dev) + pb_debug("%s: Found interface '%s'\n", __func__,dev); + return dev; +} + + +void device_handler_process_url(struct device_handler *handler, + const char *url) +{ + struct discover_context *ctx; + struct discover_device *dev; + struct boot_status *status; + struct pb_url *pb_url; + struct event *event; + struct param *param; + + status = talloc(handler, struct boot_status); + + status->type = BOOT_STATUS_ERROR; + status->progress = 0; + status->detail = talloc_asprintf(status, + _("Received config URL %s"), url); + + if (!handler->network) { + status->message = talloc_asprintf(handler, + _("No network configured")); + goto msg; + } + + event = talloc(handler, struct event); + event->type = EVENT_TYPE_USER; + event->action = EVENT_ACTION_CONF; + + event->params = talloc_array(event, struct param, 1); + param = &event->params[0]; + param->name = talloc_strdup(event, "pxeconffile"); + param->value = talloc_strdup(event, url); + event->n_params = 1; + + pb_url = pb_url_parse(event, event->params->value); + if (!pb_url || !pb_url->host) { + status->message = talloc_asprintf(handler, + _("Invalid config URL!")); + goto msg; + } + + event->device = device_from_addr(event, pb_url); + if (!event->device) { + status->message = talloc_asprintf(status, + _("Unable to route to host %s"), + pb_url->host); + goto msg; + } + + dev = discover_device_create(handler, event->device); + ctx = device_handler_discover_context_create(handler, dev); + ctx->event = event; + iterate_parsers(ctx); - /* add discovered stuff to the handler */ - context_commit(handler, ctx); + device_handler_discover_context_commit(handler, ctx); talloc_free(ctx); - return 0; + status->type = BOOT_STATUS_INFO; + status->message = talloc_asprintf(status, _("Config file %s parsed"), + pb_url->file); +msg: + boot_status(handler, status); + talloc_free(status); +} + +#ifndef PETITBOOT_TEST + +static void device_handler_update_lang(const char *lang) +{ + const char *cur_lang; + + if (!lang) + return; + + cur_lang = setlocale(LC_ALL, NULL); + if (cur_lang && !strcmp(cur_lang, lang)) + return; + + setlocale(LC_ALL, lang); } -static int handle_remove_udev_event(struct device_handler *handler, - struct event *event) +static int device_handler_init_sources(struct device_handler *handler) { - struct discover_device *dev; + /* init our device sources: udev, network and user events */ + handler->udev = udev_init(handler, handler->waitset); + if (!handler->udev) + return -1; - dev = find_device(handler, event->device); - if (!dev) - return 0; + handler->network = network_init(handler, handler->waitset, + handler->dry_run); + if (!handler->network) + return -1; - /* remove device from handler device array */ - device_handler_remove(handler, dev); + handler->user_event = user_event_init(handler, handler->waitset); + if (!handler->user_event) + return -1; return 0; } -static int handle_add_user_event(struct device_handler *handler, - struct event *event) +static void device_handler_reinit_sources(struct device_handler *handler) { - struct discover_context *ctx; - struct discover_device *dev; + /* if we haven't initialised sources previously (becuase we started in + * safe mode), then init once here. */ + if (!(handler->udev || handler->network || handler->user_event)) { + device_handler_init_sources(handler); + return; + } + + udev_reinit(handler->udev); + + network_shutdown(handler->network); + handler->network = network_init(handler, handler->waitset, + handler->dry_run); +} + +static const char *fs_parameters(unsigned int rw_flags, const char *fstype) +{ + if ((rw_flags | MS_RDONLY) != MS_RDONLY) + return ""; + + /* Avoid writing back to the disk on journaled filesystems */ + if (!strncmp(fstype, "ext4", strlen("ext4"))) + return "norecovery"; + if (!strncmp(fstype, "xfs", strlen("xfs"))) + return "norecovery"; + + return ""; +} + +static bool check_existing_mount(struct discover_device *dev) +{ + struct stat devstat, mntstat; + struct mntent *mnt; + FILE *fp; int rc; - assert(event->device); + rc = stat(dev->device_path, &devstat); + if (rc) { + pb_debug("%s: stat failed: %s\n", __func__, strerror(errno)); + return false; + } - ctx = talloc(handler, struct discover_context); - ctx->event = event; - list_init(&ctx->boot_options); + if (!S_ISBLK(devstat.st_mode)) { + pb_debug("%s: %s isn't a block device?\n", __func__, + dev->device_path); + return false; + } - dev = discover_device_create(handler, ctx, event); - ctx->device = dev; + fp = fopen("/proc/self/mounts", "r"); - rc = parse_user_event(ctx, event); + 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; - if (!rc) - context_commit(handler, ctx); + 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 rc; + return mnt != NULL; } -static int handle_remove_user_event(struct device_handler *handler, - struct event *event) +static int mount_device(struct discover_device *dev) { - struct discover_device *dev = find_device(handler, event->device); + const char *fstype; + int rc; + + if (!dev->device_path) + return -1; - if (!dev) + if (dev->mounted) return 0; - /* remove device from handler device array */ - device_handler_remove(handler, dev); + if (check_existing_mount(dev)) + return 0; - return 0; -} + fstype = discover_device_get_param(dev, "ID_FS_TYPE"); + if (!fstype) + return 0; -typedef int (*event_handler)(struct device_handler *, struct event *); + /* ext3 treats the norecovery option as an error, so mount the device + * as an ext4 filesystem instead */ + if (!strncmp(fstype, "ext3", strlen("ext3"))) { + pb_debug("Mounting ext3 filesystem as ext4\n"); + fstype = talloc_asprintf(dev, "ext4"); + } + + dev->mount_path = join_paths(dev, mount_base(), + dev->device_path); -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, + 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; } -}; -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); + pb_log("mounting device %s read-only\n", dev->device_path); + errno = 0; + rc = mount(dev->device_path, dev->mount_path, fstype, + MS_RDONLY | MS_SILENT, + fs_parameters(MS_RDONLY, fstype)); + if (!rc) { + dev->mounted = true; + dev->mounted_rw = false; + dev->unmount = true; return 0; } - return handlers[event->type][event->action](handler, event); + pb_log("couldn't mount device %s: mount failed: %s\n", + dev->device_path, strerror(errno)); + + pb_rmdir_recursive(mount_base(), dev->mount_path); +err_free: + talloc_free(dev->mount_path); + dev->mount_path = NULL; + return -1; } -struct device_handler *device_handler_init(struct discover_server *server, - int dry_run) +static int umount_device(struct discover_device *dev) { - struct device_handler *handler; + int rc; - handler = talloc(NULL, struct device_handler); - handler->devices = NULL; - handler->n_devices = 0; - handler->server = server; - handler->dry_run = dry_run; - list_init(&handler->unresolved_boot_options); + if (!dev->mounted || !dev->unmount) + return 0; - /* set up our mount point base */ - pb_mkdir_recursive(mount_base()); + pb_log("unmounting device %s\n", dev->device_path); + rc = umount(dev->mount_path); + if (rc) + return -1; - parser_init(); + dev->mounted = false; - return handler; -} + pb_rmdir_recursive(mount_base(), dev->mount_path); -void device_handler_destroy(struct device_handler *handler) -{ - talloc_free(handler); -} + talloc_free(dev->mount_path); + dev->mount_path = NULL; -static int device_match_path(struct discover_device *dev, const char *path) -{ - return !strcmp(dev->device_path, path); + return 0; } -static int device_match_uuid(struct discover_device *dev, const char *uuid) +int device_request_write(struct discover_device *dev, bool *release) { - return dev->uuid && !strcmp(dev->uuid, uuid); -} + const char *fstype; + int rc; -static int device_match_label(struct discover_device *dev, const char *label) -{ - return dev->label && !strcmp(dev->label, label); -} + *release = false; -static int device_match_id(struct discover_device *dev, const char *id) -{ - return !strcmp(dev->device->id, id); -} + if (!dev->mounted) + return -1; -static struct discover_device *device_lookup( - struct device_handler *device_handler, - int (match_fn)(struct discover_device *, const char *), - const char *str) -{ - struct discover_device *dev; - unsigned int i; + if (dev->mounted_rw) + return 0; - if (!str) - return NULL; + fstype = discover_device_get_param(dev, "ID_FS_TYPE"); - for (i = 0; i < device_handler->n_devices; i++) { - dev = device_handler->devices[i]; + pb_log("remounting device %s read-write\n", dev->device_path); - if (match_fn(dev, str)) - return dev; + rc = umount(dev->mount_path); + if (rc) { + pb_log("Failed to unmount %s\n", dev->mount_path); + return -1; } + rc = mount(dev->device_path, dev->mount_path, fstype, + MS_SILENT, + fs_parameters(MS_REMOUNT, fstype)); + if (rc) + goto mount_ro; + + dev->mounted_rw = true; + *release = true; + return 0; - return NULL; +mount_ro: + pb_log("Unable to remount device %s read-write\n", dev->device_path); + rc = mount(dev->device_path, dev->mount_path, fstype, + MS_RDONLY | MS_SILENT, + fs_parameters(MS_RDONLY, fstype)); + if (rc) + pb_log("Unable to recover mount for %s\n", dev->device_path); + return -1; } -struct discover_device *device_lookup_by_name(struct device_handler *handler, - const char *name) +void device_release_write(struct discover_device *dev, bool release) { - struct discover_device *dev; - char *path; - - if (strncmp(name, "/dev/", strlen("/dev/"))) - path = talloc_asprintf(NULL, "/dev/%s", name); - else - path = talloc_strdup(NULL, name); + const char *fstype; - dev = device_lookup_by_path(handler, path); + if (!release) + return; - talloc_free(path); + fstype = discover_device_get_param(dev, "ID_FS_TYPE"); - return dev; + pb_log("remounting device %s read-only\n", dev->device_path); + mount(dev->device_path, dev->mount_path, "", + MS_REMOUNT | MS_RDONLY | MS_SILENT, + fs_parameters(MS_RDONLY, fstype)); + dev->mounted_rw = false; } -struct discover_device *device_lookup_by_path( - struct device_handler *device_handler, - const char *path) +#else + +static void device_handler_update_lang(const char *lang __attribute__((unused))) { - return device_lookup(device_handler, device_match_path, path); } -struct discover_device *device_lookup_by_uuid( - struct device_handler *device_handler, - const char *uuid) +static int device_handler_init_sources( + struct device_handler *handler __attribute__((unused))) { - return device_lookup(device_handler, device_match_uuid, uuid); + return 0; } -struct discover_device *device_lookup_by_label( - struct device_handler *device_handler, - const char *label) +static void device_handler_reinit_sources( + struct device_handler *handler __attribute__((unused))) { - return device_lookup(device_handler, device_match_label, label); } -struct discover_device *device_lookup_by_id( - struct device_handler *device_handler, - const char *id) +static int umount_device(struct discover_device *dev __attribute__((unused))) { - return device_lookup(device_handler, device_match_id, id); + return 0; } -static struct discover_boot_option *find_boot_option_by_id( - struct device_handler *handler, const char *id) +static int __attribute__((unused)) mount_device( + struct discover_device *dev __attribute__((unused))) { - unsigned int i; - - for (i = 0; i < handler->n_devices; i++) { - struct discover_device *dev = handler->devices[i]; - struct discover_boot_option *opt; - - list_for_each_entry(&dev->boot_options, opt, list) - if (!strcmp(opt->option->id, id)) - return opt; - } + return 0; +} - return NULL; +int device_request_write(struct discover_device *dev __attribute__((unused)), + bool *release) +{ + *release = true; + return 0; } -void device_handler_boot(struct device_handler *handler, - struct boot_command *cmd) +void device_release_write(struct discover_device *dev __attribute__((unused)), + bool release __attribute__((unused))) { - struct discover_boot_option *opt; +} - opt = find_boot_option_by_id(handler, cmd->option_id); +#endif - boot(handler, opt, cmd, handler->dry_run); -}