ui/ncurses: in lockdown ensure system reboot in ncurses menu exit

[petitboot] / ui / ncurses / nc-menu.c

diff --git a/ui/ncurses/nc-menu.c b/ui/ncurses/nc-menu.c
index b8f9a35e0cddcd7ac16bbd330e73104ee8a0a3e4..a90a02e3f89d43a2be8b1aa284f4b43096a413fd 100644 (file)
--- a/ui/ncurses/nc-menu.c
+++ b/ui/ncurses/nc-menu.c
@@ -117,12 +117,37 @@ static const char *pmenu_item_label(struct pmenu_item *item, const char *name)
        label = talloc_array(item, char, len + 1);
        wcstombs(label, tmp, len + 1);
 
-       pb_log("%s: %s\n", __func__, label);
+       pb_log_fn("%s\n", label);
 
        talloc_free(tmp);
        return label;
 }
 
+/**
+ * pmenu_item_update - Update the label of an existing pmenu_item.
+ *
+ * The item array must be disconnected prior to calling.
+ */
+int pmenu_item_update(struct pmenu_item *item, const char *name)
+{
+       const char *label;
+       ITEM *i;
+
+       if (!item || !item->nci)
+               return -1;
+
+       label = pmenu_item_label(item, name);
+
+       if (!label)
+               return -1;
+
+       i = item->nci;
+       i->name.str = label;
+       i->name.length = strncols(label);
+
+       return 0;
+}
+
 /**
  * pmenu_item_create - Allocate and initialize a new pmenu_item instance.
  *
@@ -230,7 +255,7 @@ struct pmenu_item *pmenu_find_device(struct pmenu *menu, struct device *dev,
 
        for (i = 0; i < menu->item_count; i++) {
                item = item_userptr(menu->items[i]);
-               cod = item->data;
+               cod = cod_from_item(item);
                /* boot entries will have opt defined */
                if (!cod || cod->opt)
                        continue;
@@ -253,6 +278,7 @@ struct pmenu_item *pmenu_find_device(struct pmenu *menu, struct device *dev,
        switch (dev->type) {
        case DEVICE_TYPE_OPTICAL:
        case DEVICE_TYPE_DISK:
+       case DEVICE_TYPE_USB:
                /* Find block info */
                for (i = 0; sys && i < sys->n_blockdevs; i++) {
                        bd = sys->blockdevs[i];
@@ -263,8 +289,7 @@ struct pmenu_item *pmenu_find_device(struct pmenu *menu, struct device *dev,
                }
                if (matched) {
                        snprintf(buf,sizeof(buf),"[%s: %s / %s]",
-                               dev->type == DEVICE_TYPE_DISK ?
-                                       _("Disk") : _("CD/DVD"),
+                               device_type_display_name(dev->type),
                                bd->name, bd->uuid);
                }
                break;
@@ -285,6 +310,12 @@ struct pmenu_item *pmenu_find_device(struct pmenu *menu, struct device *dev,
                                _("Network"), intf->name, hwaddr);
                }
                break;
+       case DEVICE_TYPE_ANY:
+               /* This is an option found from a file:// url, not associated
+                * with any device */
+               snprintf(buf, sizeof(buf), "[Custom Local Options]");
+               matched = true;
+               break;
 
        default:
                /* Assume the device may be able to boot */
@@ -308,10 +339,9 @@ struct pmenu_item *pmenu_find_device(struct pmenu *menu, struct device *dev,
 
        /* We identify dev_hdr items as having a valid c->name,
         * but a NULL c->opt */
-       cod = talloc(dev_hdr, struct cui_opt_data);
+       cod = talloc_zero(dev_hdr, struct cui_opt_data);
        cod->name = talloc_strdup(dev_hdr, opt->device_id);
        cod->dev = dev;
-       cod->opt = NULL;
        dev_hdr->data = cod;
 
        pb_debug("%s: returning %s\n",__func__,cod->name);
@@ -322,11 +352,12 @@ static int pmenu_item_get_index(const struct pmenu_item *item)
 {
        unsigned int i;
 
-       for (i = 0; i < item->pmenu->item_count; i++)
-               if (item->pmenu->items[i] == item->nci)
-                       return i;
+       if (item)
+               for (i = 0; i < item->pmenu->item_count; i++)
+                       if (item->pmenu->items[i] == item->nci)
+                               return i;
 
-       pb_log("%s: not found: %p %s\n", __func__, item,
+       pb_log_fn("not found: %p %s\n", item,
                (item ? item->nci->name.str : "(null)"));
        return -1;
 }
@@ -342,6 +373,34 @@ static void pmenu_move_cursor(struct pmenu *menu, int req)
        wrefresh(menu->scr.main_ncw);
 }
 
+/**
+ * pmenu_main_hot_keys - Hot keys for the main boot menu
+ */
+int pmenu_main_hot_keys(struct pmenu *menu, struct pmenu_item *item, int c)
+{
+       struct nc_scr *scr = &menu->scr;
+       (void)item;
+
+       switch (c) {
+       case 'i':
+               cui_show_sysinfo(cui_from_arg(scr->ui_ctx));
+               break;
+       case 'c':
+               cui_show_config(cui_from_arg(scr->ui_ctx));
+               break;
+       case 'l':
+               cui_show_lang(cui_from_arg(scr->ui_ctx));
+               break;
+       case 'g':
+               cui_show_statuslog(cui_from_arg(scr->ui_ctx));
+               break;
+       default:
+               return 0;
+       }
+
+       return c;
+}
+
 /**
  * pmenu_process_key - Process a user keystroke.
  */
@@ -350,11 +409,15 @@ static void pmenu_process_key(struct nc_scr *scr, int key)
 {
        struct pmenu *menu = pmenu_from_scr(scr);
        struct pmenu_item *item = pmenu_find_selected(menu);
+       unsigned int i;
 
        nc_scr_status_free(&menu->scr);
 
-       if (menu->hot_key)
-               key = menu->hot_key(menu, item, key);
+       if (menu->hot_keys)
+               for (i = 0; i < menu->n_hot_keys; i++) {
+                       if (menu->hot_keys[i](menu, item, key))
+                               return;
+               }
 
        switch (key) {
        case 27: /* ESC */
@@ -402,15 +465,6 @@ static void pmenu_process_key(struct nc_scr *scr, int key)
                if (item->on_execute)
                        item->on_execute(item);
                break;
-       case 'i':
-               cui_show_sysinfo(cui_from_arg(scr->ui_ctx));
-               break;
-       case 'c':
-               cui_show_config(cui_from_arg(scr->ui_ctx));
-               break;
-       case 'l':
-               cui_show_lang(cui_from_arg(scr->ui_ctx));
-               break;
        case KEY_F(1):
        case 'h':
                if (menu->help_text)
@@ -437,7 +491,7 @@ unsigned int pmenu_grow(struct pmenu *menu, unsigned int count)
 
        assert(item_count(menu->ncm) == 0 && "not disconnected");
 
-       pb_log("%s: %u current + %u new = %u\n", __func__, menu->item_count,
+       pb_log_fn("%u current + %u new = %u\n", menu->item_count,
                count, menu->item_count + count);
 
        /* Note that items array has a null terminator. */
@@ -472,7 +526,7 @@ int pmenu_remove(struct pmenu *menu, struct pmenu_item *item)
 
        assert(menu->item_count);
 
-       pb_log("%s: %u\n", __func__, menu->item_count);
+       pb_log_fn("%u\n", menu->item_count);
 
        index = pmenu_item_get_index(item);