X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fdevice-handler.c;h=4e1b3ca77fc0f447bfd0febfdf66db22c7121ed6;hp=38584440801963e193a7aee2ad9c7a62cb89f9d3;hb=b9b5fa357e5921eb752e62f7146339007d9f767f;hpb=6704a228c6389607f775bbaf2ee7de786d7b9996 diff --git a/discover/device-handler.c b/discover/device-handler.c index 3858444..4e1b3ca 100644 --- a/discover/device-handler.c +++ b/discover/device-handler.c @@ -18,6 +18,7 @@ #include #include #include +#include #include #include @@ -84,6 +85,10 @@ struct device_handler { struct list progress; unsigned int n_progress; + + struct plugin_option **plugins; + unsigned int n_plugins; + bool plugin_installing; }; static int mount_device(struct discover_device *dev); @@ -126,6 +131,28 @@ const struct discover_device *device_handler_get_device( return handler->devices[index]; } +/** + * device_handler_get_plugin_count - Get the count of current handler plugins. + */ +int device_handler_get_plugin_count(const struct device_handler *handler) +{ + return handler->n_plugins; +} + +/** + * discover_handler_get_plugin - Get a handler plugin by index. + */ +const struct plugin_option *device_handler_get_plugin( + const struct device_handler *handler, unsigned int index) +{ + if (index >= handler->n_plugins) { + assert(0 && "bad index"); + return NULL; + } + + return handler->plugins[index]; +} + struct network *device_handler_get_network( const struct device_handler *handler) { @@ -312,6 +339,22 @@ const char *discover_device_get_param(struct discover_device *device, return NULL; } +static void set_env_variables(const struct config *config) +{ + if (config->http_proxy) + setenv("http_proxy", config->http_proxy, 1); + else + unsetenv("http_proxy"); + + if (config->https_proxy) + setenv("https_proxy", config->https_proxy, 1); + else + unsetenv("https_proxy"); + + /* Reduce noise in the log from LVM listing open file descriptors */ + setenv("LVM_SUPPRESS_FD_WARNINGS", "1", 1); +} + struct device_handler *device_handler_init(struct discover_server *server, struct waitset *waitset, int dry_run) { @@ -336,6 +379,8 @@ struct device_handler *device_handler_init(struct discover_server *server, if (config_get()->safe_mode) return handler; + set_env_variables(config_get()); + rc = device_handler_init_sources(handler); if (rc) { talloc_free(handler); @@ -349,6 +394,7 @@ void device_handler_reinit(struct device_handler *handler) { struct discover_boot_option *opt, *tmp; struct ramdisk_device *ramdisk; + struct config *config; unsigned int i; device_handler_cancel_default(handler); @@ -385,6 +431,25 @@ void device_handler_reinit(struct device_handler *handler) handler->ramdisks = NULL; handler->n_ramdisks = 0; + /* drop any known plugins */ + for (i = 0; i < handler->n_plugins; i++) + talloc_free(handler->plugins[i]); + talloc_free(handler->plugins); + handler->plugins = NULL; + handler->n_plugins = 0; + + discover_server_notify_plugins_remove(handler->server); + + set_env_variables(config_get()); + + /* If the safe mode warning was active disable it now */ + if (config_get()->safe_mode) { + config = config_copy(handler, config_get()); + config->safe_mode = false; + config_set(config); + discover_server_notify_config(handler->server, config); + } + device_handler_reinit_sources(handler); } @@ -603,6 +668,21 @@ void device_handler_status_download(struct device_handler *handler, } } +static void device_handler_plugin_scan_device(struct device_handler *handler, + struct discover_device *dev) +{ + int rc; + + pb_debug("Scanning %s for plugin files\n", dev->device->id); + + rc = process_run_simple(handler, pb_system_apps.pb_plugin, + "scan", dev->mount_path, + NULL); + if (rc) + pb_log("Error from pb-plugin scan %s\n", + dev->mount_path); +} + void device_handler_status_download_remove(struct device_handler *handler, struct process_info *procinfo) { @@ -628,7 +708,8 @@ static void countdown_status(struct device_handler *handler, status.type = STATUS_INFO; status.message = talloc_asprintf(handler, - _("Booting in %d sec: %s"), sec, opt->option->name); + _("Booting in %d sec: [%s] %s"), sec, + opt->device->device->id, opt->option->name); status.backlog = false; device_handler_status(handler, &status); @@ -1068,6 +1149,9 @@ int device_handler_discover(struct device_handler *handler, device_handler_discover_context_commit(handler, ctx); process_boot_option_queue(handler); + + /* Check this device for pb-plugins */ + device_handler_plugin_scan_device(handler, dev); out: talloc_unlink(handler, ctx); @@ -1353,6 +1437,61 @@ void device_handler_process_url(struct device_handler *handler, talloc_unlink(handler, ctx); } +static void plugin_install_cb(struct process *process) +{ + struct device_handler *handler = process->data; + + if (!handler) { + pb_log("%s: Missing data!\n", __func__); + return; + } + + handler->plugin_installing = false; + if (process->exit_status) { + device_handler_status_err(handler, "Plugin failed to install!"); + pb_log("Failed to install plugin:\n%s\n", process->stdout_buf); + } +} + +void device_handler_install_plugin(struct device_handler *handler, + const char *plugin_file) +{ + struct process *p; + int result; + + if (handler->plugin_installing) { + pb_log("Plugin install cancelled - install already running"); + return; + } + + p = process_create(handler); + if (!p) { + pb_log("install_plugin: Failed to create process\n"); + return; + } + + const char *argv[] = { + pb_system_apps.pb_plugin, + "install", + "auto", + plugin_file, + NULL + }; + + p->path = pb_system_apps.pb_plugin; + p->argv = argv; + p->exit_cb = plugin_install_cb; + p->data = handler; + p->keep_stdout = true; + + result = process_run_async(p); + + if (result) + device_handler_status_err(handler, "Could not install plugin"); + else + handler->plugin_installing = true; +} + #ifndef PETITBOOT_TEST /** @@ -1407,6 +1546,36 @@ void device_handler_discover_context_commit(struct device_handler *handler, } } +void device_handler_add_plugin_option(struct device_handler *handler, + struct plugin_option *opt) +{ + struct plugin_option *tmp; + unsigned int i; + + for (i = 0; i < handler->n_plugins; i++) { + tmp = handler->plugins[i]; + /* If both id and version match, ignore */ + if (strncmp(opt->id, tmp->id, strlen(opt->id)) == 0 && + strncmp(opt->version, tmp->version, + strlen(opt->version) == 0)) { + pb_log("discover: Plugin '%s' already exists, ignoring\n", + opt->id); + return; + } + } + + handler->plugins = talloc_realloc(handler, handler->plugins, + struct plugin_option *, handler->n_plugins + 1); + if (!handler->plugins) { + pb_log("Failed to allocate memory for new plugin\n"); + handler->n_plugins = 0; + return; + } + + handler->plugins[handler->n_plugins++] = opt; + discover_server_notify_plugin_option_add(handler->server, opt); +} + static void device_handler_update_lang(const char *lang) { const char *cur_lang; @@ -1424,8 +1593,8 @@ static void device_handler_update_lang(const char *lang) static int device_handler_init_sources(struct device_handler *handler) { /* init our device sources: udev, network and user events */ - handler->udev = udev_init(handler, handler->waitset); - if (!handler->udev) + handler->user_event = user_event_init(handler, handler->waitset); + if (!handler->user_event) return -1; handler->network = network_init(handler, handler->waitset, @@ -1433,8 +1602,8 @@ static int device_handler_init_sources(struct device_handler *handler) if (!handler->network) return -1; - handler->user_event = user_event_init(handler, handler->waitset); - if (!handler->user_event) + handler->udev = udev_init(handler, handler->waitset); + if (!handler->udev) return -1; return 0; @@ -1451,11 +1620,11 @@ static void device_handler_reinit_sources(struct device_handler *handler) system_info_reinit(); - udev_reinit(handler->udev); - network_shutdown(handler->network); handler->network = network_init(handler, handler->waitset, handler->dry_run); + + udev_reinit(handler->udev); } static inline const char *get_device_path(struct discover_device *dev)