From: Jeremy Kerr Date: Mon, 11 Mar 2013 05:43:48 +0000 (+0800) Subject: protocol: Separate device add from boot-option add messages X-Git-Tag: v1.0.0~713 X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=commitdiff_plain;h=bd06734362bb727b09b943688d9b69aa0a84590d protocol: Separate device add from boot-option add messages We want to cater for situations where boot options may be discovered some time after we get notificiation about devices. For instance, discovering boot options from DHCP configuration parameters. In this case, we'll need to notify UIs of boot options appear some time after the device (and/or other boot options on the same device) has appeared. This change adds a new protocol message type, PB_PROTOCOL_ACTION_BOOT_OPTION_ADD. We also rename PB_PROTOCOL_ACTION_ADD to make it clear that it is just for devices. The discover server is updated to send boot option add events at device discover time, but we are now able to decouple this later. We also update the clients to handle the boot option add events separately. Signed-off-by: Jeremy Kerr --- diff --git a/discover/device-handler.c b/discover/device-handler.c index 12bd5ae..ab27b51 100644 --- a/discover/device-handler.c +++ b/discover/device-handler.c @@ -275,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; @@ -311,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; } @@ -325,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); @@ -338,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); @@ -352,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); @@ -372,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); diff --git a/discover/discover-server.c b/discover/discover-server.c index 9ec3382..9f6e7da 100644 --- a/discover/discover-server.c +++ b/discover/discover-server.c @@ -91,7 +91,7 @@ static int client_write_message( return rc; } -static int write_add_message(struct discover_server *server, +static int write_device_add_message(struct discover_server *server, struct client *client, const struct device *dev) { struct pb_protocol_message *message; @@ -100,7 +100,7 @@ static int write_add_message(struct discover_server *server, len = pb_protocol_device_len(dev); message = pb_protocol_create_message(client, - PB_PROTOCOL_ACTION_ADD, len); + PB_PROTOCOL_ACTION_DEVICE_ADD, len); if (!message) return -1; @@ -109,7 +109,25 @@ static int write_add_message(struct discover_server *server, return client_write_message(server, client, message); } -static int write_remove_message(struct discover_server *server, +static int write_boot_option_add_message(struct discover_server *server, + struct client *client, const struct boot_option *opt) +{ + struct pb_protocol_message *message; + int len; + + len = pb_protocol_boot_option_len(opt); + + message = pb_protocol_create_message(client, + PB_PROTOCOL_ACTION_BOOT_OPTION_ADD, len); + if (!message) + return -1; + + pb_protocol_serialise_boot_option(opt, message->payload, len); + + return client_write_message(server, client, message); +} + +static int write_device_remove_message(struct discover_server *server, struct client *client, char *dev_id) { struct pb_protocol_message *message; @@ -118,7 +136,7 @@ static int write_remove_message(struct discover_server *server, len = strlen(dev_id) + sizeof(uint32_t); message = pb_protocol_create_message(client, - PB_PROTOCOL_ACTION_REMOVE, len); + PB_PROTOCOL_ACTION_DEVICE_REMOVE, len); if (!message) return -1; @@ -183,9 +201,14 @@ static int discover_server_process_connection(void *arg) n_devices = device_handler_get_device_count(server->device_handler); for (i = 0; i < n_devices; i++) { const struct device *device; + struct boot_option *opt; device = device_handler_get_device(server->device_handler, i); - write_add_message(server, client, device); + write_device_add_message(server, client, device); + + list_for_each_entry(&device->boot_options, opt, list) + discover_server_notify_boot_option_add(server, opt); + } waiter_register(server->waitset, client->fd, WAIT_IN, @@ -194,23 +217,32 @@ static int discover_server_process_connection(void *arg) return 0; } -void discover_server_notify_add(struct discover_server *server, +void discover_server_notify_device_add(struct discover_server *server, struct device *device) { struct client *client; list_for_each_entry(&server->clients, client, list) - write_add_message(server, client, device); + write_device_add_message(server, client, device); + +} +void discover_server_notify_boot_option_add(struct discover_server *server, + struct boot_option *boot_option) +{ + struct client *client; + + list_for_each_entry(&server->clients, client, list) + write_boot_option_add_message(server, client, boot_option); } -void discover_server_notify_remove(struct discover_server *server, +void discover_server_notify_device_remove(struct discover_server *server, struct device *device) { struct client *client; list_for_each_entry(&server->clients, client, list) - write_remove_message(server, client, device->id); + write_device_remove_message(server, client, device->id); } diff --git a/discover/discover-server.h b/discover/discover-server.h index 6650bba..e47cf5f 100644 --- a/discover/discover-server.h +++ b/discover/discover-server.h @@ -5,6 +5,7 @@ struct discover_server; struct device_handler; +struct boot_option; struct device; struct discover_server *discover_server_init(struct waitset *waitset); @@ -14,8 +15,10 @@ void discover_server_destroy(struct discover_server *server); void discover_server_set_device_source(struct discover_server *server, struct device_handler *handler); -void discover_server_notify_add(struct discover_server *server, +void discover_server_notify_device_add(struct discover_server *server, struct device *device); -void discover_server_notify_remove(struct discover_server *server, +void discover_server_notify_boot_option_add(struct discover_server *server, + struct boot_option *option); +void discover_server_notify_device_remove(struct discover_server *server, struct device *device); #endif /* _DISCOVER_SERVER_H */ diff --git a/discover/event-parser.c b/discover/event-parser.c index 0b96a09..1eec5c9 100644 --- a/discover/event-parser.c +++ b/discover/event-parser.c @@ -31,6 +31,7 @@ int parse_user_event(struct device *device, struct event *event) } opt->id = talloc_asprintf(opt, "%s#%s", device->id, p); + opt->device_id = talloc_strdup(opt, device->id); opt->name = talloc_strdup(opt, p); p = event_get_param(event, "image"); diff --git a/lib/pb-protocol/pb-protocol.c b/lib/pb-protocol/pb-protocol.c index c6d8f63..ded35eb 100644 --- a/lib/pb-protocol/pb-protocol.c +++ b/lib/pb-protocol/pb-protocol.c @@ -161,26 +161,23 @@ static int optional_strlen(const char *str) int pb_protocol_device_len(const struct device *dev) { - struct boot_option *opt; - int len; - - len = 4 + optional_strlen(dev->id) + + return 4 + optional_strlen(dev->id) + 4 + optional_strlen(dev->name) + 4 + optional_strlen(dev->description) + - 4 + optional_strlen(dev->icon_file) + - 4; + 4 + optional_strlen(dev->icon_file); +} - list_for_each_entry(&dev->boot_options, opt, list) { - len += 4 + optional_strlen(opt->id) + - 4 + optional_strlen(opt->name) + - 4 + optional_strlen(opt->description) + - 4 + optional_strlen(opt->icon_file) + - 4 + optional_strlen(opt->boot_image_file) + - 4 + optional_strlen(opt->initrd_file) + - 4 + optional_strlen(opt->boot_args); - } +int pb_protocol_boot_option_len(const struct boot_option *opt) +{ - return len; + return 4 + optional_strlen(opt->device_id) + + 4 + optional_strlen(opt->id) + + 4 + optional_strlen(opt->name) + + 4 + optional_strlen(opt->description) + + 4 + optional_strlen(opt->icon_file) + + 4 + optional_strlen(opt->boot_image_file) + + 4 + optional_strlen(opt->initrd_file) + + 4 + optional_strlen(opt->boot_args); } int pb_protocol_boot_len(const struct boot_command *boot) @@ -191,39 +188,35 @@ int pb_protocol_boot_len(const struct boot_command *boot) 4 + optional_strlen(boot->boot_args); } -int pb_protocol_serialise_device(const struct device *dev, char *buf, int buf_len) +int pb_protocol_serialise_device(const struct device *dev, + char *buf, int buf_len) { - struct boot_option *opt; - uint32_t n; - char *pos; - - pos = buf; + char *pos = buf; - /* construct payload into buffer */ pos += pb_protocol_serialise_string(pos, dev->id); pos += pb_protocol_serialise_string(pos, dev->name); pos += pb_protocol_serialise_string(pos, dev->description); pos += pb_protocol_serialise_string(pos, dev->icon_file); - /* write option count */ - n = 0; + assert(pos <= buf + buf_len); + (void)buf_len; - list_for_each_entry(&dev->boot_options, opt, list) - n++; + return 0; +} - *(uint32_t *)pos = __cpu_to_be32(n); - pos += sizeof(uint32_t); +int pb_protocol_serialise_boot_option(const struct boot_option *opt, + char *buf, int buf_len) +{ + char *pos = buf; - /* write each option */ - list_for_each_entry(&dev->boot_options, opt, list) { - pos += pb_protocol_serialise_string(pos, opt->id); - pos += pb_protocol_serialise_string(pos, opt->name); - pos += pb_protocol_serialise_string(pos, opt->description); - pos += pb_protocol_serialise_string(pos, opt->icon_file); - pos += pb_protocol_serialise_string(pos, opt->boot_image_file); - pos += pb_protocol_serialise_string(pos, opt->initrd_file); - pos += pb_protocol_serialise_string(pos, opt->boot_args); - } + pos += pb_protocol_serialise_string(pos, opt->device_id); + pos += pb_protocol_serialise_string(pos, opt->id); + pos += pb_protocol_serialise_string(pos, opt->name); + pos += pb_protocol_serialise_string(pos, opt->description); + pos += pb_protocol_serialise_string(pos, opt->icon_file); + pos += pb_protocol_serialise_string(pos, opt->boot_image_file); + pos += pb_protocol_serialise_string(pos, opt->initrd_file); + pos += pb_protocol_serialise_string(pos, opt->boot_args); assert(pos <= buf + buf_len); (void)buf_len; @@ -341,89 +334,95 @@ struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd) int pb_protocol_deserialise_device(struct device *dev, const struct pb_protocol_message *message) { - const char *pos; - int i, n_options; unsigned int len; + const char *pos; + int rc = -1; len = message->payload_len; pos = message->payload; if (read_string(dev, &pos, &len, &dev->id)) - goto out_err; + goto out; if (read_string(dev, &pos, &len, &dev->name)) - goto out_err; + goto out; if (read_string(dev, &pos, &len, &dev->description)) - goto out_err; + goto out; if (read_string(dev, &pos, &len, &dev->icon_file)) - goto out_err; + goto out; - n_options = __be32_to_cpu(*(uint32_t *)pos); - pos += sizeof(uint32_t); + rc = 0; - dev->n_options = n_options; - - list_init(&dev->boot_options); - - for (i = 0; i < n_options; i++) { - struct boot_option *opt; - - opt = talloc(dev, struct boot_option); - - if (read_string(opt, &pos, &len, &opt->id)) - goto out_err; - if (read_string(opt, &pos, &len, &opt->name)) - goto out_err; - if (read_string(opt, &pos, &len, - &opt->description)) - goto out_err; - if (read_string(opt, &pos, &len, - &opt->icon_file)) - goto out_err; - if (read_string(opt, &pos, &len, - &opt->boot_image_file)) - goto out_err; - if (read_string(opt, &pos, &len, - &opt->initrd_file)) - goto out_err; - if (read_string(opt, &pos, &len, - &opt->boot_args)) - goto out_err; - - list_add(&dev->boot_options, &opt->list); - } +out: + return rc; +} - return 0; +int pb_protocol_deserialise_boot_option(struct boot_option *opt, + const struct pb_protocol_message *message) +{ + unsigned int len; + const char *pos; + int rc = -1; -out_err: - return -1; + len = message->payload_len; + pos = message->payload; + + if (read_string(opt, &pos, &len, &opt->device_id)) + goto out; + + if (read_string(opt, &pos, &len, &opt->id)) + goto out; + + if (read_string(opt, &pos, &len, &opt->name)) + goto out; + + if (read_string(opt, &pos, &len, &opt->description)) + goto out; + + if (read_string(opt, &pos, &len, &opt->icon_file)) + goto out; + + if (read_string(opt, &pos, &len, &opt->boot_image_file)) + goto out; + + if (read_string(opt, &pos, &len, &opt->initrd_file)) + goto out; + + if (read_string(opt, &pos, &len, &opt->boot_args)) + goto out; + + rc = 0; + +out: + return rc; } int pb_protocol_deserialise_boot_command(struct boot_command *cmd, const struct pb_protocol_message *message) { - const char *pos; unsigned int len; + const char *pos; + int rc = -1; len = message->payload_len; pos = message->payload; if (read_string(cmd, &pos, &len, &cmd->option_id)) - goto out_err; + goto out; if (read_string(cmd, &pos, &len, &cmd->boot_image_file)) - goto out_err; + goto out; if (read_string(cmd, &pos, &len, &cmd->initrd_file)) - goto out_err; + goto out; if (read_string(cmd, &pos, &len, &cmd->boot_args)) - goto out_err; + goto out; - return 0; + rc = 0; -out_err: - return -1; +out: + return rc; } diff --git a/lib/pb-protocol/pb-protocol.h b/lib/pb-protocol/pb-protocol.h index de2ae7c..35e96f0 100644 --- a/lib/pb-protocol/pb-protocol.h +++ b/lib/pb-protocol/pb-protocol.h @@ -12,9 +12,11 @@ #define PB_PROTOCOL_MAX_PAYLOAD_SIZE (64 * 1024) enum pb_protocol_action { - PB_PROTOCOL_ACTION_ADD = 0x1, - PB_PROTOCOL_ACTION_REMOVE = 0x2, - PB_PROTOCOL_ACTION_BOOT = 0x3, + PB_PROTOCOL_ACTION_DEVICE_ADD = 0x1, + PB_PROTOCOL_ACTION_BOOT_OPTION_ADD = 0x2, + PB_PROTOCOL_ACTION_DEVICE_REMOVE = 0x3, +/* PB_PROTOCOL_ACTION_BOOT_OPTION_REMOVE = 0x4, */ + PB_PROTOCOL_ACTION_BOOT = 0x5, }; struct pb_protocol_message { @@ -26,6 +28,7 @@ struct pb_protocol_message { void pb_protocol_dump_device(const struct device *dev, const char *text, FILE *stream); int pb_protocol_device_len(const struct device *dev); +int pb_protocol_boot_option_len(const struct boot_option *opt); int pb_protocol_boot_len(const struct boot_command *boot); int pb_protocol_device_cmp(const struct device *a, const struct device *b); @@ -36,7 +39,10 @@ int pb_protocol_serialise_string(char *pos, const char *str); char *pb_protocol_deserialise_string(void *ctx, const struct pb_protocol_message *message); -int pb_protocol_serialise_device(const struct device *dev, char *buf, int buf_len); +int pb_protocol_serialise_device(const struct device *dev, + char *buf, int buf_len); +int pb_protocol_serialise_boot_option(const struct boot_option *opt, + char *buf, int buf_len); int pb_protocol_serialise_boot_command(const struct boot_command *boot, char *buf, int buf_len); @@ -50,6 +56,9 @@ struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd); int pb_protocol_deserialise_device(struct device *dev, const struct pb_protocol_message *message); +int pb_protocol_deserialise_boot_option(struct boot_option *opt, + const struct pb_protocol_message *message); + int pb_protocol_deserialise_boot_command(struct boot_command *cmd, const struct pb_protocol_message *message); diff --git a/lib/types/types.h b/lib/types/types.h index 1843557..fb53826 100644 --- a/lib/types/types.h +++ b/lib/types/types.h @@ -16,6 +16,7 @@ struct device { }; struct boot_option { + char *device_id; char *id; char *name; char *description; diff --git a/ui/common/discover-client.c b/ui/common/discover-client.c index c0cfea0..42aa9de 100644 --- a/ui/common/discover-client.c +++ b/ui/common/discover-client.c @@ -1,4 +1,5 @@ +#include #include #include #include @@ -37,6 +38,20 @@ void discover_client_destroy(struct discover_client *client) talloc_free(client); } +static struct device *find_device(struct discover_client *client, + const char *id) +{ + int i; + + for (i = 0; i < client->n_devices; i++) { + struct device *dev = client->devices[i]; + if (!strcmp(dev->id, id)) + return dev; + } + + return NULL; +} + static void device_add(struct discover_client *client, struct device *device) { client->n_devices++; @@ -46,7 +61,25 @@ static void device_add(struct discover_client *client, struct device *device) client->devices[client->n_devices - 1] = device; talloc_steal(client, device); - client->ops.device_add(device, client->ops.cb_arg); + if (client->ops.device_add) + client->ops.device_add(device, client->ops.cb_arg); +} + +static void boot_option_add(struct discover_client *client, + struct boot_option *opt) +{ + struct device *dev; + + dev = find_device(client, opt->device_id); + + /* we require that devices are already present before any boot options + * are added */ + assert(dev); + + talloc_steal(dev, opt); + + if (client->ops.boot_option_add) + client->ops.boot_option_add(dev, opt, client->ops.cb_arg); } static void device_remove(struct discover_client *client, const char *id) @@ -81,6 +114,7 @@ static int discover_client_process(void *arg) { struct discover_client *client = arg; struct pb_protocol_message *message; + struct boot_option *opt; struct device *dev; char *dev_id; int rc; @@ -91,8 +125,9 @@ static int discover_client_process(void *arg) return -1; switch (message->action) { - case PB_PROTOCOL_ACTION_ADD: - dev = talloc(client, struct device); + case PB_PROTOCOL_ACTION_DEVICE_ADD: + dev = talloc_zero(client, struct device); + list_init(&dev->boot_options); rc = pb_protocol_deserialise_device(dev, message); if (rc) { @@ -102,7 +137,18 @@ static int discover_client_process(void *arg) device_add(client, dev); break; - case PB_PROTOCOL_ACTION_REMOVE: + case PB_PROTOCOL_ACTION_BOOT_OPTION_ADD: + opt = talloc_zero(client, struct boot_option); + + rc = pb_protocol_deserialise_boot_option(opt, message); + if (rc) { + pb_log("%s: no boot_option?\n", __func__); + return 0; + } + + boot_option_add(client, opt); + break; + case PB_PROTOCOL_ACTION_DEVICE_REMOVE: dev_id = pb_protocol_deserialise_string(client, message); if (!dev_id) { pb_log("%s: no device id?\n", __func__); diff --git a/ui/common/discover-client.h b/ui/common/discover-client.h index 6d5d1c4..5317b44 100644 --- a/ui/common/discover-client.h +++ b/ui/common/discover-client.h @@ -21,6 +21,8 @@ struct pb_boot_data { struct discover_client_ops { int (*device_add)(struct device *device, void *arg); + int (*boot_option_add)(struct device *dev, struct boot_option *option, + void *arg); void (*device_remove)(struct device *device, void *arg); void *cb_arg; }; diff --git a/ui/ncurses/nc-cui.c b/ui/ncurses/nc-cui.c index 2d4dea5..c26819f 100644 --- a/ui/ncurses/nc-cui.c +++ b/ui/ncurses/nc-cui.c @@ -142,7 +142,7 @@ static int cui_boot(struct pmenu_item *item) def_prog_mode(); - result = discover_client_boot(cui->client, cod->dev, cod->opt, cod->bd); + result = discover_client_boot(cui->client, NULL, cod->opt, cod->bd); reset_prog_mode(); redrawwin(cui->current->main_ncw); @@ -354,16 +354,18 @@ void cui_on_open(struct pmenu *menu) * menu_items into the main menu. Redraws the main menu if it is active. */ -static int cui_device_add(struct device *dev, void *arg) +static int cui_boot_option_add(struct device *dev, struct boot_option *opt, + void *arg) { struct cui *cui = cui_from_arg(arg); - int result; - struct boot_option *opt; - unsigned int o_count; /* device opts */ + struct cui_opt_data *cod; unsigned int insert_pt; + struct pmenu_item *i; ITEM *selected; + char *name; + int result; - pb_log("%s: %p %s\n", __func__, dev, dev->id); + pb_log("%s: %p %s\n", __func__, opt, opt->id); selected = current_item(cui->main->ncm); @@ -377,57 +379,42 @@ static int cui_device_add(struct device *dev, void *arg) if (result) pb_log("%s: set_menu_items failed: %d\n", __func__, result); - o_count = 0; - list_for_each_entry(&dev->boot_options, opt, list) - o_count++; - /* Insert new items at insert_pt. */ + insert_pt = pmenu_grow(cui->main, 1); - insert_pt = pmenu_grow(cui->main, o_count); - - list_for_each_entry(&dev->boot_options, opt, list) { - struct pmenu_item *i; - struct cui_opt_data *cod; - char *name; - - /* Save the item in opt->ui_info for cui_device_remove() */ + /* Save the item in opt->ui_info for cui_device_remove() */ - opt->ui_info = i = pmenu_item_alloc(cui->main); + opt->ui_info = i = pmenu_item_alloc(cui->main); - i->on_edit = cui_boot_editor_run; - i->on_execute = cui_boot; - i->data = cod = talloc(i, struct cui_opt_data); - - cod->dev = dev; - cod->opt = opt; - cod->opt_hash = pb_opt_hash(dev, opt); - cod->name = opt->name; - cod->bd = talloc(i, struct pb_boot_data); - - cod->bd->image = talloc_strdup(cod->bd, opt->boot_image_file); - cod->bd->initrd = talloc_strdup(cod->bd, opt->initrd_file); - cod->bd->args = talloc_strdup(cod->bd, opt->boot_args); - - name = cui_make_item_name(i, cod); - pmenu_item_setup(cui->main, i, insert_pt, name); - - insert_pt++; - - pb_log("%s: adding opt '%s'\n", __func__, cod->name); - pb_log(" image '%s'\n", cod->bd->image); - pb_log(" initrd '%s'\n", cod->bd->initrd); - pb_log(" args '%s'\n", cod->bd->args); - - /* If this is the default_item select it and start timer. */ - - if (cod->opt_hash == cui->default_item) { - selected = i->nci; - ui_timer_kick(&cui->timer); - } + i->on_edit = cui_boot_editor_run; + i->on_execute = cui_boot; + i->data = cod = talloc(i, struct cui_opt_data); + + cod->opt = opt; + cod->dev = dev; + cod->opt_hash = pb_opt_hash(dev, opt); + cod->name = opt->name; + cod->bd = talloc(i, struct pb_boot_data); + + cod->bd->image = talloc_strdup(cod->bd, opt->boot_image_file); + cod->bd->initrd = talloc_strdup(cod->bd, opt->initrd_file); + cod->bd->args = talloc_strdup(cod->bd, opt->boot_args); + + name = cui_make_item_name(i, cod); + pmenu_item_setup(cui->main, i, insert_pt, name); + + pb_log("%s: adding opt '%s'\n", __func__, cod->name); + pb_log(" image '%s'\n", cod->bd->image); + pb_log(" initrd '%s'\n", cod->bd->initrd); + pb_log(" args '%s'\n", cod->bd->args); + + /* If this is the default_item select it and start timer. */ + if (cod->opt_hash == cui->default_item) { + selected = i->nci; + ui_timer_kick(&cui->timer); } /* Re-attach the items array. */ - result = set_menu_items(cui->main->ncm, cui->main->items); if (result) @@ -506,7 +493,8 @@ static void cui_device_remove(struct device *dev, void *arg) } static struct discover_client_ops cui_client_ops = { - .device_add = cui_device_add, + .device_add = NULL, + .boot_option_add = cui_boot_option_add, .device_remove = cui_device_remove, }; diff --git a/ui/twin/pbt-client.c b/ui/twin/pbt-client.c index 875b408..09d8e99 100644 --- a/ui/twin/pbt-client.c +++ b/ui/twin/pbt-client.c @@ -63,7 +63,7 @@ static int pbt_client_boot(struct pbt_item *item) pbt_item_name(item)); result = discover_client_boot(item->pbt_client->discover_client, - opt_data->dev, opt_data->opt, opt_data->bd); + NULL, opt_data->opt, opt_data->bd); if (result) { pb_log("%s: failed: %s\n", __func__, opt_data->bd->image); @@ -83,101 +83,89 @@ static int pbt_client_on_edit(struct pbt_item *item) static int pbt_device_add(struct device *dev, struct pbt_client *client) { struct pbt_frame *const frame = &client->frame; - struct pbt_item *device_item; - struct boot_option *opt; + struct pbt_item *item; struct pbt_quad q; const char *icon_file; - struct pbt_item *selected_item = NULL; pb_log("%s: %p %s: n_options %d\n", __func__, dev, dev->id, dev->n_options); pb_protocol_dump_device(dev, "", pb_log_get_stream()); - /* device_item */ - // FIXME: check for existing dev->id icon_file = dev->icon_file ? dev->icon_file : pbt_icon_chooser(dev->id); - device_item = pbt_item_create_reduced(frame->top_menu, dev->id, + item = pbt_item_create_reduced(frame->top_menu, dev->id, frame->top_menu->n_items, icon_file); - if (!device_item) - goto fail_device_item_create; + if (!item) + return -1; - device_item->pb_device = dev; + item->pb_device = dev; + dev->ui_info = item; frame->top_menu->n_items++; /* sub_menu */ - q.x = frame->top_menu->window->pixmap->width; q.y = 0; q.width = frame->top_menu->scr->tscreen->width - q.x; q.height = frame->top_menu->scr->tscreen->height; - device_item->sub_menu = pbt_menu_create(device_item, dev->id, + item->sub_menu = pbt_menu_create(item, dev->id, frame->top_menu->scr, frame->top_menu, &q, &frame->top_menu->layout); - if (!device_item->sub_menu) - goto fail_sub_menu_create; + if (!item->sub_menu) + return -1; + + pbt_menu_show(frame->top_menu, 1); - list_for_each_entry(&dev->boot_options, opt, list) { - struct pbt_item *i; - struct pb_opt_data *opt_data; + return 0; +} + +static int pbt_boot_option_add(struct device *dev, struct boot_option *opt, + struct pbt_client *client) +{ + struct pbt_item *opt_item, *device_item; + struct pb_opt_data *opt_data; - i = pbt_item_create(device_item->sub_menu, + device_item = dev->ui_info; + + opt_item = pbt_item_create(device_item->sub_menu, opt->id, device_item->sub_menu->n_items++, opt->icon_file, opt->name, opt->description); - if (!i) { - assert(0); - break; - } - - i->pb_opt = opt; - i->pbt_client = client; - i->on_execute = pbt_client_boot; - i->on_edit = pbt_client_on_edit; - - i->data = opt_data = talloc(i, struct pb_opt_data); - opt_data->name = opt->name; - opt_data->bd = talloc(i, struct pb_boot_data); - opt_data->bd->image = talloc_strdup(opt_data->bd, - opt->boot_image_file); - opt_data->bd->initrd = talloc_strdup(opt_data->bd, - opt->initrd_file); - opt_data->bd->args = talloc_strdup(opt_data->bd, - opt->boot_args); - opt_data->dev = dev; - opt_data->opt = opt; - opt_data->opt_hash = pb_opt_hash(dev, opt); - - /* Select the first item as default. */ - - if (!selected_item) - selected_item = i; - - /* If this is the default_item select it and start timer. */ - - if (opt_data->opt_hash - == device_item->sub_menu->default_item_hash) { - selected_item = i; - ui_timer_kick(&client->signal_data.timer); - } + opt_item->pb_opt = opt; + opt_item->pbt_client = client; + opt_item->on_execute = pbt_client_boot; + opt_item->on_edit = pbt_client_on_edit; + + opt_item->data = opt_data = talloc(opt_item, struct pb_opt_data); + opt_data->name = opt->name; + opt_data->bd = talloc(opt_item, struct pb_boot_data); + opt_data->bd->image = talloc_strdup(opt_data->bd, + opt->boot_image_file); + opt_data->bd->initrd = talloc_strdup(opt_data->bd, + opt->initrd_file); + opt_data->bd->args = talloc_strdup(opt_data->bd, + opt->boot_args); + opt_data->opt = opt; + opt_data->opt_hash = pb_opt_hash(dev, opt); + + /* If this is the default_item select it and start timer. */ + if (opt_data->opt_hash == device_item->sub_menu->default_item_hash) { + device_item->selected_item = opt_item; + pbt_menu_set_selected(device_item->sub_menu, opt_item); + ui_timer_kick(&client->signal_data.timer); } - pbt_menu_set_selected(device_item->sub_menu, selected_item); + /* Select the first item as default. */ + if (!device_item->selected_item) + pbt_menu_set_selected(device_item->sub_menu, opt_item); - pbt_menu_show(frame->top_menu, 1); twin_screen_update(client->frame.scr->tscreen); return 0; - -fail_sub_menu_create: -fail_device_item_create: - assert(0); - return -1; } static void pbt_device_remove(struct device *dev, struct pbt_client *client) @@ -248,6 +236,7 @@ static void pbt_device_remove(struct device *dev, struct pbt_client *client) static struct discover_client_ops pbt_client_ops = { .device_add = (void *)pbt_device_add, + .boot_option_add = (void *)pbt_boot_option_add, .device_remove = (void *)pbt_device_remove, }; diff --git a/ui/twin/pbt-menu.h b/ui/twin/pbt-menu.h index cb696ea..698b64d 100644 --- a/ui/twin/pbt-menu.h +++ b/ui/twin/pbt-menu.h @@ -41,6 +41,7 @@ struct pbt_item twin_pixmap_t *pixmap_active; struct pbt_menu *sub_menu; + struct pbt_item *selected_item; int (*on_execute)(struct pbt_item *item); int (*on_edit)(struct pbt_item *item);