X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fdevice-handler.c;h=ab27b51b2eaf764d98a37b7df3185dd49d46a20b;hp=6e03ef3fa159136b6015983a0054411cb7f126c8;hb=bd06734362bb727b09b943688d9b69aa0a84590d;hpb=52b9db95764fcdee9195113d7df225634a19c9f4 diff --git a/discover/device-handler.c b/discover/device-handler.c index 6e03ef3..ab27b51 100644 --- a/discover/device-handler.c +++ b/discover/device-handler.c @@ -10,7 +10,7 @@ #include #include #include -#include +#include #include #include "device-handler.h" @@ -19,13 +19,11 @@ #include "parser.h" #include "udev.h" #include "paths.h" - -#define MOUNT_BIN "/bin/mount" - -#define UMOUNT_BIN "/bin/umount" +#include "boot.h" struct device_handler { struct discover_server *server; + int dry_run; struct device **devices; unsigned int n_devices; @@ -33,11 +31,6 @@ struct device_handler { struct list contexts; }; -struct mount_map { - char *device_path; - char *mount_point; -}; - /** * device_handler_add - Add a device to the handler device array. */ @@ -194,7 +187,7 @@ static int mount_device(struct discover_context *ctx) pb_log("couldn't create mount directory %s: %s\n", ctx->mount_path, strerror(errno)); - argv[0] = MOUNT_BIN; + argv[0] = pb_system_apps.mount; argv[1] = ctx->device_path; argv[2] = ctx->mount_path; argv[3] = "-o"; @@ -205,7 +198,7 @@ static int mount_device(struct discover_context *ctx) /* Retry mount without ro option. */ - argv[0] = MOUNT_BIN; + argv[0] = pb_system_apps.mount; argv[1] = ctx->device_path; argv[2] = ctx->mount_path; argv[3] = NULL; @@ -236,7 +229,8 @@ static int umount_device(struct discover_context *ctx) } if (pid == 0) { - execl(UMOUNT_BIN, UMOUNT_BIN, ctx->mount_path, NULL); + execl(pb_system_apps.umount, pb_system_apps.umount, + ctx->mount_path, NULL); exit(EXIT_FAILURE); } @@ -281,6 +275,7 @@ static int handle_add_udev_event(struct device_handler *handler, struct event *event) { struct discover_context *ctx; + struct boot_option *opt; const char *devname; int rc; @@ -317,7 +312,10 @@ static int handle_add_udev_event(struct device_handler *handler, /* add device to handler device array */ device_handler_add(handler, ctx->device); - discover_server_notify_add(handler->server, 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); return 0; } @@ -331,7 +329,7 @@ static int handle_remove_udev_event(struct device_handler *handler, if (!ctx) return 0; - discover_server_notify_remove(handler->server, ctx->device); + discover_server_notify_device_remove(handler->server, ctx->device); /* remove device from handler device array */ device_handler_remove(handler, ctx->device); @@ -344,6 +342,7 @@ 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; assert(event->device); @@ -358,7 +357,10 @@ static int handle_add_user_event(struct device_handler *handler, parse_user_event(device, event); - discover_server_notify_add(handler->server, device); + discover_server_notify_device_add(handler->server, device); + + list_for_each_entry(&device->boot_options, opt, list) + discover_server_notify_boot_option_add(handler->server, opt); /* add device to handler device array */ device_handler_add(handler, device); @@ -378,7 +380,7 @@ static int handle_remove_user_event(struct device_handler *handler, if (!device) return 0; - discover_server_notify_remove(handler->server, device); + discover_server_notify_device_remove(handler->server, device); /* remove device from handler device array */ device_handler_remove(handler, device); @@ -387,49 +389,35 @@ static int handle_remove_user_event(struct device_handler *handler, return 0; } +typedef int (*event_handler)(struct device_handler *, struct event *); + +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, + } +}; + int device_handler_event(struct device_handler *handler, struct event *event) { - int rc = 0; - - switch (event->type) { - case EVENT_TYPE_UDEV: - switch (event->action) { - case EVENT_ACTION_ADD: - rc = handle_add_udev_event(handler, event); - break; - case EVENT_ACTION_REMOVE: - rc = handle_remove_udev_event(handler, event); - break; - default: - pb_log("%s unknown action: %d\n", __func__, - event->action); - break; - } - break; - case EVENT_TYPE_USER: - switch (event->action) { - case EVENT_ACTION_ADD: - rc = handle_add_user_event(handler, event); - break; - case EVENT_ACTION_REMOVE: - rc = handle_remove_user_event(handler, event); - break; - default: - pb_log("%s unknown action: %d\n", __func__, - event->action); - break; - } - break; - default: - pb_log("%s unknown type: %d\n", __func__, event->type); - break; + 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; } - return rc; + return handlers[event->type][event->action](handler, event); } -struct device_handler *device_handler_init(struct discover_server *server) +struct device_handler *device_handler_init(struct discover_server *server, + int dry_run) { struct device_handler *handler; @@ -437,6 +425,7 @@ struct device_handler *device_handler_init(struct discover_server *server) handler->devices = NULL; handler->n_devices = 0; handler->server = server; + handler->dry_run = dry_run; list_init(&handler->contexts); @@ -452,3 +441,30 @@ void device_handler_destroy(struct device_handler *handler) { talloc_free(handler); } + +static struct 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; + + list_for_each_entry(&dev->boot_options, opt, list) + if (!strcmp(opt->id, id)) + return opt; + } + + return NULL; +} + +void device_handler_boot(struct device_handler *handler, + struct boot_command *cmd) +{ + struct boot_option *opt; + + opt = find_boot_option_by_id(handler, cmd->option_id); + + boot(handler, opt, cmd, handler->dry_run); +}