[PATCH v7 5/9] efimenu: add "Change Boot Order" menu entry
Masahisa Kojima
masahisa.kojima at linaro.org
Mon Jun 13 11:38:49 CEST 2022
This commit adds the menu entry to update UEFI BootOrder variable.
User moves the entry with UP/DOWN key, changes the order
with PLUS/MINUS key, then finalizes the order by ENTER key.
The U-Boot menu framework is well designed for static menu,
this commit implements the own menu display and key handling
for dynamically change the order of menu entry.
Signed-off-by: Masahisa Kojima <masahisa.kojima at linaro.org>
---
Changes in v7:
- use UP/DOWN and PLUS/MINUS key to change to order
no update in v6:
cmd/efimenu.c | 207 ++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 207 insertions(+)
diff --git a/cmd/efimenu.c b/cmd/efimenu.c
index 918e9dffa2..1be483a082 100644
--- a/cmd/efimenu.c
+++ b/cmd/efimenu.c
@@ -70,6 +70,13 @@ struct efimenu_file_entry_data {
u16 *file_name;
};
+struct efimenu_boot_order {
+ u32 num;
+ u16 *description;
+ u32 prev_index;
+ struct list_head list;
+};
+
/**
* efimenu_print_msg() - print message
*
@@ -1359,6 +1366,205 @@ out:
return ret;
}
+static void efimenu_display_change_boot_order(struct list_head *bo_list,
+ struct efimenu *efi_menu)
+{
+ bool reverse;
+ struct list_head *pos, *n;
+ struct efimenu_boot_order *entry;
+
+ printf(ANSI_CLEAR_CONSOLE
+ ANSI_CURSOR_POSITION
+ "\n ** Change Boot Order **\n"
+ ANSI_CURSOR_POSITION
+ "\n"
+ " Press UP/DOWN to move, +/- to change order\n ENTER to complete, ESC/CTRL+C to quit"
+ ANSI_CURSOR_POSITION,
+ 0, 1, efi_menu->count + 5, 1, 4, 1);
+
+ /* draw boot order list */
+ list_for_each_safe(pos, n, bo_list) {
+ entry = list_entry(pos, struct efimenu_boot_order, list);
+ reverse = (entry->num == efi_menu->active);
+ puts(" ");
+ if (reverse)
+ puts(ANSI_COLOR_REVERSE);
+
+ printf("%ls\n", entry->description);
+
+ if (reverse)
+ puts(ANSI_COLOR_RESET);
+ }
+}
+
+static efi_status_t efimenu_choice_change_boot_order(struct list_head *bo_list,
+ struct efimenu *efi_menu)
+{
+ int esc = 0;
+ struct list_head *pos, *n;
+ struct efimenu_boot_order *tmp;
+ enum bootmenu_key key = KEY_NONE;
+ struct efimenu_boot_order *entry;
+
+ while (1) {
+ bootmenu_loop(NULL, &key, &esc);
+
+ switch (key) {
+ case KEY_PLUS:
+ if (efi_menu->active > 0) {
+ list_for_each_safe(pos, n, bo_list) {
+ entry = list_entry(pos, struct efimenu_boot_order, list);
+ if (entry->num == efi_menu->active)
+ break;
+ }
+ tmp = list_entry(pos->prev, struct efimenu_boot_order, list);
+ entry->num--;
+ tmp->num++;
+ list_del(&tmp->list);
+ list_add(&tmp->list, &entry->list);
+ }
+ fallthrough;
+ case KEY_UP:
+ if (efi_menu->active > 0)
+ --efi_menu->active;
+ return EFI_NOT_READY;
+ case KEY_MINUS:
+ if (efi_menu->active < efi_menu->count - 1) {
+ list_for_each_safe(pos, n, bo_list) {
+ entry = list_entry(pos, struct efimenu_boot_order, list);
+ if (entry->num == efi_menu->active)
+ break;
+ }
+ tmp = list_entry(pos->next, struct efimenu_boot_order, list);
+ entry->num++;
+ tmp->num--;
+ list_del(&entry->list);
+ list_add(&entry->list, &tmp->list);
+ }
+ fallthrough;
+ case KEY_DOWN:
+ if (efi_menu->active < efi_menu->count - 1)
+ ++efi_menu->active;
+ return EFI_NOT_READY;
+ case KEY_SELECT:
+ return EFI_SUCCESS;
+ case KEY_QUIT:
+ return EFI_ABORTED;
+ default:
+ break;
+ }
+ }
+}
+
+static efi_status_t efimenu_process_change_boot_order(void *data)
+{
+ u32 i;
+ u16 *bootorder;
+ efi_status_t ret;
+ efi_uintn_t num, size;
+ struct list_head bo_list;
+ struct list_head *pos, *n;
+ struct efimenu_boot_order *entry;
+ struct efimenu efi_menu;
+
+ bootorder = efi_get_var(u"BootOrder", &efi_global_variable_guid, &size);
+ if (!bootorder) {
+ ret = EFI_NOT_FOUND;
+ return ret;
+ }
+
+ INIT_LIST_HEAD(&bo_list);
+
+ num = size / sizeof(u16);
+ for (i = 0; i < num; i++) {
+ void *load_option;
+ struct efi_load_option lo;
+ u16 varname[] = u"Boot####";
+
+ efi_create_indexed_name(varname, sizeof(varname),
+ "Boot", bootorder[i]);
+ load_option = efi_get_var(varname, &efi_global_variable_guid, &size);
+ if (!load_option) {
+ ret = EFI_NOT_FOUND;
+ goto out;
+ }
+
+ ret = efi_deserialize_load_option(&lo, load_option, &size);
+ if (ret != EFI_SUCCESS) {
+ free(load_option);
+ goto out;
+ }
+
+ entry = calloc(1, sizeof(struct efimenu_boot_order));
+ if (!entry) {
+ free(load_option);
+ goto out;
+ }
+ entry->num = i;
+ entry->description = u16_strdup(lo.label);
+ entry->prev_index = i;
+ list_add_tail(&entry->list, &bo_list);
+
+ free(load_option);
+ }
+
+ efi_menu.delay = -1;
+ efi_menu.active = 0;
+ efi_menu.count = num;
+
+ while (1) {
+ efimenu_display_change_boot_order(&bo_list, &efi_menu);
+
+ ret = efimenu_choice_change_boot_order(&bo_list, &efi_menu);
+ if (ret == EFI_SUCCESS) {
+ u32 i = 0;
+ u16 *new_bootorder;
+
+ /* update the BootOrder variable */
+ size = num * sizeof(u16);
+ new_bootorder = calloc(1, size);
+ if (!new_bootorder) {
+ ret = EFI_OUT_OF_RESOURCES;
+ goto out;
+ }
+
+ list_for_each_safe(pos, n, &bo_list) {
+ entry = list_entry(pos, struct efimenu_boot_order, list);
+ new_bootorder[i] = bootorder[entry->prev_index];
+ i++;
+ }
+
+ ret = efi_set_variable_int(u"BootOrder", &efi_global_variable_guid,
+ EFI_VARIABLE_NON_VOLATILE |
+ EFI_VARIABLE_BOOTSERVICE_ACCESS |
+ EFI_VARIABLE_RUNTIME_ACCESS,
+ size, new_bootorder, false);
+
+ free(new_bootorder);
+ goto out;
+ } else if (ret == EFI_NOT_READY) {
+ continue;
+ } else {
+ goto out;
+ }
+ }
+
+out:
+ list_for_each_safe(pos, n, &bo_list) {
+ entry = list_entry(pos, struct efimenu_boot_order, list);
+ list_del(&entry->list);
+ free(entry->description);
+ free(entry);
+ }
+
+ free(bootorder);
+
+ /* to stay the parent menu */
+ ret = (ret == EFI_ABORTED) ? EFI_NOT_READY : ret;
+
+ return ret;
+}
+
static efi_status_t efimenu_init(void)
{
efi_status_t ret;
@@ -1393,6 +1599,7 @@ static efi_status_t efimenu_init(void)
static const struct efimenu_item maintenance_menu_items[] = {
{"Add Boot Option", efimenu_process_add_boot_option},
{"Edit Boot Option", efimenu_process_edit_boot_option},
+ {"Change Boot Order", efimenu_process_change_boot_order},
{"Quit", efimenu_process_quit},
};
--
2.17.1
More information about the U-Boot
mailing list