On Fri, Apr 07, 2023 at 12:13:36PM +0200, Heinrich Schuchardt wrote: > On 4/5/23 02:06, Raymond Mao wrote: > > Changes for complying to EFI spec ยง3.5.1.1 > > 'Removable Media Boot Behavior'. > > Boot variables can be automatically generated during a removable > > media is probed. At the same time, unused boot variables will be > > detected and removed. > > Related APIs are renamed and moved from cmd to lib for re-use > > between eficonfig and bootmgr. > > > > Signed-off-by: Raymond Mao <raymond....@linaro.org> > > We should split the patch into moving and renaming functions and > individual patches for each changed functionality. > > > --- > > Changes in v2 > > - Ignore EFI_NOT_FOUND returned from > > efi_bootmgr_update_media_device_boot_option which means no boot > > options scanned. > > > > cmd/bootmenu.c | 2 +- > > cmd/eficonfig.c | 408 +----------------------------- > > include/efi_config.h | 5 - > > include/efi_loader.h | 11 + > > lib/efi_loader/efi_bootmgr.c | 380 ++++++++++++++++++++++++++++ > > lib/efi_loader/efi_disk.c | 7 + > > lib/efi_loader/efi_helper.c | 25 ++ > > lib/efi_loader/efi_variable.c | 6 +- > > lib/efi_loader/efi_variable_tee.c | 3 +- > > 9 files changed, 437 insertions(+), 410 deletions(-) > > > > diff --git a/cmd/bootmenu.c b/cmd/bootmenu.c > > index 6baeedc69f..01daddca7b 100644 > > --- a/cmd/bootmenu.c > > +++ b/cmd/bootmenu.c > > @@ -351,7 +351,7 @@ static struct bootmenu_data *bootmenu_create(int delay) > > * UEFI specification requires booting from removal media using > > * a architecture-specific default image name such as > > BOOTAA64.EFI. > > */ > > - efi_ret = eficonfig_generate_media_device_boot_option(); > > + efi_ret = efi_bootmgr_update_media_device_boot_option(); > > if (efi_ret != EFI_SUCCESS && efi_ret != EFI_NOT_FOUND) > > goto cleanup; > > > > diff --git a/cmd/eficonfig.c b/cmd/eficonfig.c > > index 720f52b48b..82a80306f4 100644 > > --- a/cmd/eficonfig.c > > +++ b/cmd/eficonfig.c > > @@ -1134,43 +1134,6 @@ out: > > return ret; > > } > > > > -/** > > - * eficonfig_get_unused_bootoption() - get unused "Boot####" index > > - * > > - * @buf: pointer to the buffer to store boot option variable name > > - * @buf_size: buffer size > > - * @index: pointer to store the index in the BootOrder variable > > - * Return: status code > > - */ > > -efi_status_t eficonfig_get_unused_bootoption(u16 *buf, efi_uintn_t > > buf_size, > > - unsigned int *index) > > -{ > > - u32 i; > > - efi_status_t ret; > > - efi_uintn_t size; > > - > > - if (buf_size < u16_strsize(u"Boot####")) > > - return EFI_BUFFER_TOO_SMALL; > > - > > - for (i = 0; i <= 0xFFFF; i++) { > > - size = 0; > > - efi_create_indexed_name(buf, buf_size, "Boot", i); > > - ret = efi_get_variable_int(buf, &efi_global_variable_guid, > > - NULL, &size, NULL, NULL); > > - if (ret == EFI_BUFFER_TOO_SMALL) > > - continue; > > - else > > - break; > > - } > > - > > - if (i > 0xFFFF) > > - return EFI_OUT_OF_RESOURCES; > > - > > - *index = i; > > - > > - return EFI_SUCCESS; > > -} > > - > > /** > > * eficonfig_set_boot_option() - set boot option > > * > > @@ -1208,46 +1171,6 @@ static efi_status_t eficonfig_set_boot_option(u16 > > *varname, struct efi_device_pa > > return ret; > > } > > > > -/** > > - * eficonfig_append_bootorder() - append new boot option in BootOrder > > variable > > - * > > - * @index: "Boot####" index to append to BootOrder variable > > - * Return: status code > > - */ > > -efi_status_t eficonfig_append_bootorder(u16 index) > > -{ > > - u16 *bootorder; > > - efi_status_t ret; > > - u16 *new_bootorder = NULL; > > - efi_uintn_t last, size, new_size; > > - > > - /* append new boot option */ > > - bootorder = efi_get_var(u"BootOrder", &efi_global_variable_guid, &size); > > - last = size / sizeof(u16); > > - new_size = size + sizeof(u16); > > - new_bootorder = calloc(1, new_size); > > - if (!new_bootorder) { > > - ret = EFI_OUT_OF_RESOURCES; > > - goto out; > > - } > > - memcpy(new_bootorder, bootorder, size); > > - new_bootorder[last] = index; > > - > > - ret = efi_set_variable_int(u"BootOrder", &efi_global_variable_guid, > > - EFI_VARIABLE_NON_VOLATILE | > > - EFI_VARIABLE_BOOTSERVICE_ACCESS | > > - EFI_VARIABLE_RUNTIME_ACCESS, > > - new_size, new_bootorder, false); > > - if (ret != EFI_SUCCESS) > > - goto out; > > - > > -out: > > - free(bootorder); > > - free(new_bootorder); > > - > > - return ret; > > -} > > - > > /** > > * create_boot_option_entry() - create boot option entry > > * > > @@ -1619,7 +1542,7 @@ static efi_status_t > > eficonfig_process_add_boot_option(void *data) > > if (!bo) > > return EFI_OUT_OF_RESOURCES; > > > > - ret = eficonfig_get_unused_bootoption(varname, sizeof(varname), > > &bo->boot_index); > > + ret = efi_bootmgr_get_unused_bootoption(varname, sizeof(varname), > > &bo->boot_index); > > if (ret != EFI_SUCCESS) > > return ret; > > > > @@ -1627,7 +1550,7 @@ static efi_status_t > > eficonfig_process_add_boot_option(void *data) > > if (ret != EFI_SUCCESS) > > goto out; > > > > - ret = eficonfig_append_bootorder((u16)bo->boot_index); > > + ret = efi_bootmgr_append_bootorder((u16)bo->boot_index); > > if (ret != EFI_SUCCESS) > > goto out; > > > > @@ -1656,31 +1579,6 @@ static efi_status_t > > eficonfig_process_boot_selected(void *data) > > return EFI_SUCCESS; > > } > > > > -/** > > - * search_bootorder() - search the boot option index in BootOrder > > - * > > - * @bootorder: pointer to the BootOrder variable > > - * @num: number of BootOrder entry > > - * @target: target boot option index to search > > - * @index: pointer to store the index of BootOrder variable > > - * Return: true if exists, false otherwise > > - */ > > -static bool search_bootorder(u16 *bootorder, efi_uintn_t num, u32 target, > > u32 *index) > > -{ > > - u32 i; > > - > > - for (i = 0; i < num; i++) { > > - if (target == bootorder[i]) { > > - if (index) > > - *index = i; > > - > > - return true; > > - } > > - } > > - > > - return false; > > -} > > - > > /** > > * eficonfig_add_boot_selection_entry() - add boot option menu entry > > * > > @@ -1805,7 +1703,7 @@ static efi_status_t > > eficonfig_show_boot_selection(unsigned int *selected) > > > > if (efi_varname_is_load_option(var_name16, &index)) { > > /* If the index is included in the BootOrder, skip it */ > > - if (search_bootorder(bootorder, num, index, NULL)) > > + if (efi_search_bootorder(bootorder, num, index, NULL)) > > continue; > > > > ret = eficonfig_add_boot_selection_entry(efi_menu, > > index, selected); > > @@ -2202,7 +2100,7 @@ static efi_status_t > > eficonfig_create_change_boot_order_entry(struct efimenu *efi > > > > if (efi_varname_is_load_option(var_name16, &index)) { > > /* If the index is included in the BootOrder, skip it */ > > - if (search_bootorder(bootorder, num, index, NULL)) > > + if (efi_search_bootorder(bootorder, num, index, NULL)) > > continue; > > > > ret = eficonfig_add_change_boot_order_entry(efi_menu, > > index, false); > > @@ -2304,50 +2202,6 @@ out: > > return ret; > > } > > > > -/** > > - * delete_boot_option() - delete selected boot option > > - * > > - * @boot_index: boot option index to delete > > - * Return: status code > > - */ > > -static efi_status_t delete_boot_option(u16 boot_index) > > -{ > > - u16 *bootorder; > > - u16 varname[9]; > > - efi_status_t ret; > > - unsigned int index; > > - efi_uintn_t num, size; > > - > > - efi_create_indexed_name(varname, sizeof(varname), > > - "Boot", boot_index); > > - ret = efi_set_variable_int(varname, &efi_global_variable_guid, > > - 0, 0, NULL, false); > > - if (ret != EFI_SUCCESS) { > > - log_err("delete boot option(%ls) failed\n", varname); > > - return ret; > > - } > > - > > - /* update BootOrder if necessary */ > > - bootorder = efi_get_var(u"BootOrder", &efi_global_variable_guid, &size); > > - if (!bootorder) > > - return EFI_SUCCESS; > > - > > - num = size / sizeof(u16); > > - if (!search_bootorder(bootorder, num, boot_index, &index)) > > - return EFI_SUCCESS; > > - > > - memmove(&bootorder[index], &bootorder[index + 1], > > - (num - index - 1) * sizeof(u16)); > > - size -= sizeof(u16); > > - ret = efi_set_variable_int(u"BootOrder", &efi_global_variable_guid, > > - EFI_VARIABLE_NON_VOLATILE | > > - EFI_VARIABLE_BOOTSERVICE_ACCESS | > > - EFI_VARIABLE_RUNTIME_ACCESS, > > - size, bootorder, false); > > - > > - return ret; > > -} > > - > > /** > > * eficonfig_process_delete_boot_option() - handler to delete boot option > > * > > @@ -2362,7 +2216,7 @@ static efi_status_t > > eficonfig_process_delete_boot_option(void *data) > > while (1) { > > ret = eficonfig_show_boot_selection(&selected); > > if (ret == EFI_SUCCESS) > > - ret = delete_boot_option(selected); > > + ret = efi_bootmgr_delete_boot_option(selected); > > Please, pass the number in the Boot#### variable and not an index in > BootOrder. > > > > > if (ret != EFI_SUCCESS) > > break; > > @@ -2374,256 +2228,6 @@ static efi_status_t > > eficonfig_process_delete_boot_option(void *data) > > return ret; > > } > > > > -/** > > - * eficonfig_enumerate_boot_option() - enumerate the possible bootable > > media > > - * > > - * @opt: pointer to the media boot option structure > > - * @volume_handles: pointer to the efi handles > > - * @count: number of efi handle > > - * Return: status code > > - */ > > -efi_status_t eficonfig_enumerate_boot_option(struct > > eficonfig_media_boot_option *opt, > > - efi_handle_t *volume_handles, > > efi_status_t count) > > -{ > > - u32 i; > > - struct efi_handler *handler; > > - efi_status_t ret = EFI_SUCCESS; > > - > > - for (i = 0; i < count; i++) { > > - u16 *p; > > - u16 dev_name[BOOTMENU_DEVICE_NAME_MAX]; > > - char *optional_data; > > - struct efi_load_option lo; > > - char buf[BOOTMENU_DEVICE_NAME_MAX]; > > - struct efi_device_path *device_path; > > - > > - ret = efi_search_protocol(volume_handles[i], > > &efi_guid_device_path, &handler); > > - if (ret != EFI_SUCCESS) > > - continue; > > - ret = efi_protocol_open(handler, (void **)&device_path, > > - efi_root, NULL, > > EFI_OPEN_PROTOCOL_GET_PROTOCOL); > > - if (ret != EFI_SUCCESS) > > - continue; > > - > > - ret = efi_disk_get_device_name(volume_handles[i], buf, > > BOOTMENU_DEVICE_NAME_MAX); > > - if (ret != EFI_SUCCESS) > > - continue; > > - > > - p = dev_name; > > - utf8_utf16_strncpy(&p, buf, strlen(buf)); > > - > > - lo.label = dev_name; > > - lo.attributes = LOAD_OPTION_ACTIVE; > > - lo.file_path = device_path; > > - lo.file_path_length = efi_dp_size(device_path) + sizeof(END); > > - /* > > - * Set the dedicated guid to optional_data, it is used to > > identify > > - * the boot option that automatically generated by the bootmenu. > > - * efi_serialize_load_option() expects optional_data is > > null-terminated > > - * utf8 string, so set the "1234567" string to allocate enough > > space > > - * to store guid, instead of realloc the load_option. > > - */ > > - lo.optional_data = "1234567"; > > - opt[i].size = efi_serialize_load_option(&lo, (u8 **)&opt[i].lo); > > - if (!opt[i].size) { > > - ret = EFI_OUT_OF_RESOURCES; > > - goto out; > > - } > > - /* set the guid */ > > - optional_data = (char *)opt[i].lo + (opt[i].size - > > u16_strsize(u"1234567")); > > - memcpy(optional_data, &efi_guid_bootmenu_auto_generated, > > sizeof(efi_guid_t)); > > - } > > - > > -out: > > - return ret; > > -} > > - > > -/** > > - * eficonfig_delete_invalid_boot_option() - delete non-existing boot option > > - * > > - * @opt: pointer to the media boot option structure > > - * @count: number of media boot option structure > > - * Return: status code > > - */ > > -efi_status_t eficonfig_delete_invalid_boot_option(struct > > eficonfig_media_boot_option *opt, > > - efi_status_t count) > > -{ > > - efi_uintn_t size; > > - void *load_option; > > - u32 i, list_size = 0; > > - struct efi_load_option lo; > > - u16 *var_name16 = NULL; > > - u16 varname[] = u"Boot####"; > > - efi_status_t ret = EFI_SUCCESS; > > - u16 *delete_index_list = NULL, *p; > > - efi_uintn_t buf_size; > > - > > - buf_size = 128; > > - var_name16 = malloc(buf_size); > > - if (!var_name16) > > - return EFI_OUT_OF_RESOURCES; > > - > > - var_name16[0] = 0; > > - for (;;) { > > - int index; > > - efi_guid_t guid; > > - efi_uintn_t tmp; > > - > > - ret = efi_next_variable_name(&buf_size, &var_name16, &guid); > > - if (ret == EFI_NOT_FOUND) { > > - /* > > - * EFI_NOT_FOUND indicates we retrieved all EFI > > variables. > > - * This should be treated as success. > > - */ > > - ret = EFI_SUCCESS; > > - break; > > - } > > - if (ret != EFI_SUCCESS) > > - goto out; > > - > > - if (!efi_varname_is_load_option(var_name16, &index)) > > - continue; > > - > > - efi_create_indexed_name(varname, sizeof(varname), "Boot", > > index); > > - load_option = efi_get_var(varname, &efi_global_variable_guid, > > &size); > > - if (!load_option) > > - continue; > > - > > - tmp = size; > > - ret = efi_deserialize_load_option(&lo, load_option, &size); > > - if (ret != EFI_SUCCESS) > > - goto next; > > - > > - if (size >= sizeof(efi_guid_bootmenu_auto_generated) && > > - !guidcmp(lo.optional_data, > > &efi_guid_bootmenu_auto_generated)) { > > - for (i = 0; i < count; i++) { > > - if (opt[i].size == tmp && > > - memcmp(opt[i].lo, load_option, tmp) == 0) { > > - opt[i].exist = true; > > - break; > > - } > > - } > > - > > - /* > > - * The entire list of variables must be retrieved by > > - * efi_get_next_variable_name_int() before deleting the > > invalid > > - * boot option, just save the index here. > > - */ > > - if (i == count) { > > - p = realloc(delete_index_list, sizeof(u32) * > > - (list_size + 1)); > > - if (!p) { > > - ret = EFI_OUT_OF_RESOURCES; > > - goto out; > > - } > > - delete_index_list = p; > > - delete_index_list[list_size++] = index; > > - } > > - } > > -next: > > - free(load_option); > > - } > > - > > - /* delete all invalid boot options */ > > - for (i = 0; i < list_size; i++) { > > - ret = delete_boot_option(delete_index_list[i]); > > - if (ret != EFI_SUCCESS) > > - goto out; > > - } > > - > > -out: > > - free(var_name16); > > - free(delete_index_list); > > - > > - return ret; > > -} > > - > > -/** > > - * eficonfig_generate_media_device_boot_option() - generate the media > > device boot option > > - * > > - * This function enumerates all devices supporting > > EFI_SIMPLE_FILE_SYSTEM_PROTOCOL > > - * and generate the bootmenu entries. > > - * This function also provide the BOOT#### variable maintenance for > > - * the media device entries. > > - * - Automatically create the BOOT#### variable for the newly detected > > device, > > - * this BOOT#### variable is distinguished by the special GUID > > - * stored in the EFI_LOAD_OPTION.optional_data > > - * - If the device is not attached to the system, the associated > > BOOT#### variable > > - * is automatically deleted. > > - * > > - * Return: status code > > - */ > > -efi_status_t eficonfig_generate_media_device_boot_option(void) > > -{ > > - u32 i; > > - efi_status_t ret; > > - efi_uintn_t count; > > - efi_handle_t *volume_handles = NULL; > > - struct eficonfig_media_boot_option *opt = NULL; > > - > > - ret = efi_locate_handle_buffer_int(BY_PROTOCOL, > > &efi_simple_file_system_protocol_guid, > > - NULL, &count, (efi_handle_t > > **)&volume_handles); > > - if (ret != EFI_SUCCESS) > > - return ret; > > - > > - opt = calloc(count, sizeof(struct eficonfig_media_boot_option)); > > - if (!opt) > > - goto out; > > - > > - /* enumerate all devices supporting EFI_SIMPLE_FILE_SYSTEM_PROTOCOL */ > > - ret = eficonfig_enumerate_boot_option(opt, volume_handles, count); > > - if (ret != EFI_SUCCESS) > > - goto out; > > - > > - /* > > - * System hardware configuration may vary depending on the user setup. > > - * The boot option is automatically added by the bootmenu. > > - * If the device is not attached to the system, the boot option needs > > - * to be deleted. > > - */ > > - ret = eficonfig_delete_invalid_boot_option(opt, count); > > - if (ret != EFI_SUCCESS) > > - goto out; > > - > > - /* add non-existent boot option */ > > - for (i = 0; i < count; i++) { > > - u32 boot_index; > > - u16 var_name[9]; > > - > > - if (!opt[i].exist) { > > - ret = eficonfig_get_unused_bootoption(var_name, > > sizeof(var_name), > > - &boot_index); > > - if (ret != EFI_SUCCESS) > > - goto out; > > - > > - ret = efi_set_variable_int(var_name, > > &efi_global_variable_guid, > > - EFI_VARIABLE_NON_VOLATILE | > > - > > EFI_VARIABLE_BOOTSERVICE_ACCESS | > > - EFI_VARIABLE_RUNTIME_ACCESS, > > - opt[i].size, opt[i].lo, > > false); > > - if (ret != EFI_SUCCESS) > > - goto out; > > - > > - ret = eficonfig_append_bootorder(boot_index); > > - if (ret != EFI_SUCCESS) { > > - efi_set_variable_int(var_name, > > &efi_global_variable_guid, > > - 0, 0, NULL, false); > > - goto out; > > - } > > - } > > - } > > - > > -out: > > - if (opt) { > > - for (i = 0; i < count; i++) > > - free(opt[i].lo); > > - } > > - free(opt); > > - efi_free_pool(volume_handles); > > - > > - return ret; > > -} > > - > > /** > > * eficonfig_init() - do required initialization for eficonfig command > > * > > @@ -2709,7 +2313,7 @@ static int do_eficonfig(struct cmd_tbl *cmdtp, int > > flag, int argc, char *const a > > if (ret != EFI_SUCCESS) > > return CMD_RET_FAILURE; > > > > - ret = eficonfig_generate_media_device_boot_option(); > > + ret = efi_bootmgr_update_media_device_boot_option(); > > if (ret != EFI_SUCCESS && ret != EFI_NOT_FOUND) > > return ret; > > > > diff --git a/include/efi_config.h b/include/efi_config.h > > index 01ce9b2b06..d7c1601137 100644 > > --- a/include/efi_config.h > > +++ b/include/efi_config.h > > @@ -105,11 +105,6 @@ efi_status_t eficonfig_process_common(struct efimenu > > *efi_menu, > > void (*item_data_print)(void *), > > char *(*item_choice)(void *)); > > efi_status_t eficonfig_process_select_file(void *data); > > -efi_status_t eficonfig_get_unused_bootoption(u16 *buf, > > - efi_uintn_t buf_size, u32 *index); > > -efi_status_t eficonfig_append_bootorder(u16 index); > > -efi_status_t eficonfig_generate_media_device_boot_option(void); > > - > > efi_status_t eficonfig_append_menu_entry(struct efimenu *efi_menu, > > char *title, eficonfig_entry_func func, > > void *data); > > diff --git a/include/efi_loader.h b/include/efi_loader.h > > index 1542b4b625..31ca1f5d1d 100644 > > --- a/include/efi_loader.h > > +++ b/include/efi_loader.h > > @@ -516,6 +516,17 @@ extern struct list_head efi_register_notify_events; > > int efi_init_early(void); > > /* Initialize efi execution environment */ > > efi_status_t efi_init_obj_list(void); > > +/* Append new boot option in BootOrder variable */ > > +efi_status_t efi_bootmgr_append_bootorder(u16 index); > > +/* Get unused "Boot####" index */ > > +efi_status_t efi_bootmgr_get_unused_bootoption(u16 *buf, > > + efi_uintn_t buf_size, u32 > > *index); > > +/* Generate the media device boot option */ > > +efi_status_t efi_bootmgr_update_media_device_boot_option(void); > > +/* Delete selected boot option */ > > +efi_status_t efi_bootmgr_delete_boot_option(u16 boot_index); > > +/* search the boot option index in BootOrder */ > > +bool efi_search_bootorder(u16 *bootorder, efi_uintn_t num, u32 target, u32 > > *index); > > /* Set up console modes */ > > void efi_setup_console_size(void); > > /* Install device tree */ > > diff --git a/lib/efi_loader/efi_bootmgr.c b/lib/efi_loader/efi_bootmgr.c > > index 4b24b41047..c329428973 100644 > > --- a/lib/efi_loader/efi_bootmgr.c > > +++ b/lib/efi_loader/efi_bootmgr.c > > @@ -347,3 +347,383 @@ efi_status_t efi_bootmgr_load(efi_handle_t *handle, > > void **load_options) > > error: > > return ret; > > } > > + > > +/** > > + * efi_bootmgr_enumerate_boot_option() - enumerate the possible bootable > > media > > + * > > + * @opt: pointer to the media boot option structure > > + * @volume_handles: pointer to the efi handles > > + * @count: number of efi handle > > + * Return: status code > > + */ > > +static efi_status_t efi_bootmgr_enumerate_boot_option(struct > > eficonfig_media_boot_option *opt, > > + efi_handle_t > > *volume_handles, > > + efi_status_t count) > > +{ > > + u32 i; > > + struct efi_handler *handler; > > + efi_status_t ret = EFI_SUCCESS; > > + > > + for (i = 0; i < count; i++) { > > + u16 *p; > > + u16 dev_name[BOOTMENU_DEVICE_NAME_MAX]; > > + char *optional_data; > > + struct efi_load_option lo; > > + char buf[BOOTMENU_DEVICE_NAME_MAX]; > > + struct efi_device_path *device_path; > > + > > + ret = efi_search_protocol(volume_handles[i], > > &efi_guid_device_path, &handler); > > + if (ret != EFI_SUCCESS) > > + continue; > > + ret = efi_protocol_open(handler, (void **)&device_path, > > + efi_root, NULL, > > EFI_OPEN_PROTOCOL_GET_PROTOCOL); > > + if (ret != EFI_SUCCESS) > > + continue; > > + > > + ret = efi_disk_get_device_name(volume_handles[i], buf, > > BOOTMENU_DEVICE_NAME_MAX); > > + if (ret != EFI_SUCCESS) > > + continue; > > + > > + p = dev_name; > > + utf8_utf16_strncpy(&p, buf, strlen(buf)); > > + > > + lo.label = dev_name; > > + lo.attributes = LOAD_OPTION_ACTIVE; > > + lo.file_path = device_path; > > + lo.file_path_length = efi_dp_size(device_path) + sizeof(END); > > + /* > > + * Set the dedicated guid to optional_data, it is used to > > identify > > + * the boot option that automatically generated by the bootmenu. > > + * efi_serialize_load_option() expects optional_data is > > null-terminated > > + * utf8 string, so set the "1234567" string to allocate enough > > space > > + * to store guid, instead of realloc the load_option. > > + */ > > + lo.optional_data = "1234567"; > > + opt[i].size = efi_serialize_load_option(&lo, (u8 **)&opt[i].lo); > > + if (!opt[i].size) { > > + ret = EFI_OUT_OF_RESOURCES; > > + goto out; > > + } > > + /* set the guid */ > > + optional_data = (char *)opt[i].lo + (opt[i].size - > > u16_strsize(u"1234567")); > > + memcpy(optional_data, &efi_guid_bootmenu_auto_generated, > > sizeof(efi_guid_t)); > > + } > > + > > +out: > > + return ret; > > +} > > + > > +/** > > + * efi_bootmgr_delete_invalid_boot_option() - delete non-existing boot > > option > > You can't delete what does not exist. > > > + * > > + * @opt: pointer to the media boot option structure > > + * @count: number of media boot option structure > > Why would you need such a parameter? > > > + * Return: status code > > + */ > > +static efi_status_t efi_bootmgr_delete_invalid_boot_option(struct > > eficonfig_media_boot_option *opt, > > + efi_status_t count) > > Why don't you pass in the boot option num > > > +{ > > + efi_uintn_t size; > > + void *load_option; > > + u32 i, list_size = 0; > > + struct efi_load_option lo; > > + u16 *var_name16 = NULL; > > + u16 varname[] = u"Boot####"; > > Why would you initialize the variable? > > > + efi_status_t ret = EFI_SUCCESS; > > + u16 *delete_index_list = NULL, *p; > > + efi_uintn_t buf_size; > > + > > + buf_size = 128; > > + var_name16 = malloc(buf_size); > > + if (!var_name16) > > + return EFI_OUT_OF_RESOURCES; > > + > > + var_name16[0] = 0; > > + for (;;) { > > + int index; > > + efi_guid_t guid; > > + efi_uintn_t tmp; > > + > > + ret = efi_next_variable_name(&buf_size, &var_name16, &guid); > > Can't we simply put the boot option number into struct > eficonfig_media_boot_option and save 100 lines code? > > > + if (ret == EFI_NOT_FOUND) { > > + log_debug("last efi var\n"); > > + /* > > + * EFI_NOT_FOUND indicates we retrieved all EFI > > variables. > > + * This should be treated as success. > > + */ > > + ret = EFI_SUCCESS; > > + break; > > + } > > + > > + if (ret != EFI_SUCCESS) > > + goto out; > > + > > + if (!efi_varname_is_load_option(var_name16, &index)) > > + continue; > > + > > + efi_create_indexed_name(varname, sizeof(varname), "Boot", > > index); > > + load_option = efi_get_var(varname, &efi_global_variable_guid, > > &size); > > + if (!load_option) > > + continue; > > + > > + tmp = size; > > + ret = efi_deserialize_load_option(&lo, load_option, &size); > > + if (ret != EFI_SUCCESS) > > + goto next; > > + > > + if (size >= sizeof(efi_guid_bootmenu_auto_generated) && > > + !guidcmp(lo.optional_data, > > &efi_guid_bootmenu_auto_generated)) { > > + for (i = 0; i < count; i++) { > > + if (opt[i].size == tmp && > > + memcmp(opt[i].lo, load_option, tmp) == 0) { > > + opt[i].exist = true; > > + break; > > + } > > + } > > + > > + /* > > + * The entire list of variables must be retrieved by > > + * efi_get_next_variable_name_int() before deleting the > > invalid > > + * boot option, just save the index here. > > + */ > > + if (i == count) { > > + log_debug("found index %d can be deleted\n", > > index); > > + p = realloc(delete_index_list, sizeof(u32) * > > + (list_size + 1)); > > + if (!p) { > > + ret = EFI_OUT_OF_RESOURCES; > > + goto out; > > + } > > + delete_index_list = p; > > + delete_index_list[list_size++] = index; > > + } > > + } > > +next: > > + free(load_option); > > + } > > + > > + /* delete all invalid boot options */ > > + for (i = 0; i < list_size; i++) { > > + log_debug("deleting index %d\n", delete_index_list[i]); > > + ret = efi_bootmgr_delete_boot_option(delete_index_list[i]); > > + if (ret != EFI_SUCCESS) > > + goto out; > > + } > > + > > +out: > > + free(var_name16); > > + free(delete_index_list); > > + > > + return ret; > > +} > > + > > +/** > > + * efi_bootmgr_get_unused_bootoption() - get unused "Boot####" index > > + * > > + * @buf: pointer to the buffer to store boot option variable name > > + * @buf_size: buffer size > > + * @index: pointer to store the index in the BootOrder variable > > + * Return: status code > > + */ > > +efi_status_t efi_bootmgr_get_unused_bootoption(u16 *buf, efi_uintn_t > > buf_size, > > + unsigned int *index) > > This function should be static. The exported functions should not take > the inddex in the BootOrder variable as a parameter. > > > +{ > > + u32 i; > > + efi_status_t ret; > > + efi_uintn_t size; > > + > > + if (buf_size < u16_strsize(u"Boot####")) > > + return EFI_BUFFER_TOO_SMALL; > > + > > + for (i = 0; i <= 0xFFFF; i++) { > > + size = 0; > > + efi_create_indexed_name(buf, buf_size, "Boot", i); > > + ret = efi_get_variable_int(buf, &efi_global_variable_guid, > > + NULL, &size, NULL, NULL); > > + if (ret == EFI_BUFFER_TOO_SMALL) > > + continue; > > + else > > + break; > > + } > > + > > + if (i > 0xFFFF) > > + return EFI_OUT_OF_RESOURCES; > > + > > + *index = i; > > + > > + return EFI_SUCCESS; > > +} > > + > > +/** > > + * efi_bootmgr_append_bootorder() - append new boot option in BootOrder > > variable > > + * > > + * @index: "Boot####" index to append to BootOrder variable > > + * Return: status code > > + */ > > +efi_status_t efi_bootmgr_append_bootorder(u16 index) > > +{ > > + u16 *bootorder; > > + efi_status_t ret; > > + u16 *new_bootorder = NULL; > > + efi_uintn_t last, size, new_size; > > + > > + /* append new boot option */ > > + bootorder = efi_get_var(u"BootOrder", &efi_global_variable_guid, &size); > > + last = size / sizeof(u16); > > + new_size = size + sizeof(u16); > > + new_bootorder = calloc(1, new_size); > > + if (!new_bootorder) { > > + ret = EFI_OUT_OF_RESOURCES; > > + goto out; > > + } > > + memcpy(new_bootorder, bootorder, size); > > + new_bootorder[last] = index; > > + > > + ret = efi_set_variable_int(u"BootOrder", &efi_global_variable_guid, > > + EFI_VARIABLE_NON_VOLATILE | > > + EFI_VARIABLE_BOOTSERVICE_ACCESS | > > + EFI_VARIABLE_RUNTIME_ACCESS, > > + new_size, new_bootorder, false); > > + if (ret != EFI_SUCCESS) > > + goto out; > > + > > +out: > > + free(bootorder); > > + free(new_bootorder); > > + > > + return ret; > > +} > > + > > +/** > > + * efi_bootmgr_delete_boot_option() - delete selected boot option > > + * > > + * @boot_index: boot option index to delete > > + * Return: status code > > + */ > > +efi_status_t efi_bootmgr_delete_boot_option(u16 boot_index) > > +{ > > + u16 *bootorder; > > + u16 varname[9]; > > + efi_status_t ret; > > + unsigned int index; > > + efi_uintn_t num, size; > > + > > + efi_create_indexed_name(varname, sizeof(varname), > > + "Boot", boot_index); > > + ret = efi_set_variable_int(varname, &efi_global_variable_guid, > > + 0, 0, NULL, false); > > + if (ret != EFI_SUCCESS) { > > + log_err("delete boot option(%ls) failed\n", varname); > > + return ret; > > + } > > + > > + /* update BootOrder if necessary */ > > + bootorder = efi_get_var(u"BootOrder", &efi_global_variable_guid, &size); > > + if (!bootorder) > > + return EFI_SUCCESS; > > + > > + num = size / sizeof(u16); > > + if (!efi_search_bootorder(bootorder, num, boot_index, &index)) > > + return EFI_SUCCESS; > > + > > + memmove(&bootorder[index], &bootorder[index + 1], > > + (num - index - 1) * sizeof(u16)); > > + size -= sizeof(u16); > > + ret = efi_set_variable_int(u"BootOrder", &efi_global_variable_guid, > > + EFI_VARIABLE_NON_VOLATILE | > > + EFI_VARIABLE_BOOTSERVICE_ACCESS | > > + EFI_VARIABLE_RUNTIME_ACCESS, > > + size, bootorder, false); > > + > > + return ret; > > +} > > + > > +/** > > + * efi_bootmgr_update_media_device_boot_option() - generate the media > > device boot option > > + * > > + * This function enumerates all devices supporting > > EFI_SIMPLE_FILE_SYSTEM_PROTOCOL > > + * and generate the bootmenu entries. > > + * This function also provide the BOOT#### variable maintenance for > > + * the media device entries. > > + * - Automatically create the BOOT#### variable for the newly detected > > device, > > + * this BOOT#### variable is distinguished by the special GUID > > + * stored in the EFI_LOAD_OPTION.optional_data > > + * - If the device is not attached to the system, the associated > > BOOT#### variable > > + * is automatically deleted. > > + * > > + * Return: status code > > + */ > > +efi_status_t efi_bootmgr_update_media_device_boot_option(void) > > +{ > > + u32 i; > > + efi_status_t ret; > > + efi_uintn_t count; > > + efi_handle_t *volume_handles = NULL; > > + struct eficonfig_media_boot_option *opt = NULL; > > + > > + ret = efi_locate_handle_buffer_int(BY_PROTOCOL, > > + > > &efi_simple_file_system_protocol_guid, > > + NULL, &count, > > + (efi_handle_t **)&volume_handles); > > + if (ret != EFI_SUCCESS) > > + return ret; > > + > > + opt = calloc(count, sizeof(struct eficonfig_media_boot_option)); > > + if (!opt) > > + goto out; > > + > > + /* enumerate all devices supporting EFI_SIMPLE_FILE_SYSTEM_PROTOCOL */ > > + ret = efi_bootmgr_enumerate_boot_option(opt, volume_handles, count); > > + if (ret != EFI_SUCCESS) > > + goto out; > > + > > + /* > > + * System hardware configuration may vary depending on the user setup. > > + * The boot option is automatically added by the bootmenu. > > + * If the device is not attached to the system, the boot option needs > > + * to be deleted. > > + */ > > + ret = efi_bootmgr_delete_invalid_boot_option(opt, count); > > + if (ret != EFI_SUCCESS) > > + goto out; > > + > > + /* add non-existent boot option */ > > + for (i = 0; i < count; i++) { > > + u32 boot_index; > > + u16 var_name[9]; > > + > > + if (!opt[i].exist) { > > + ret = efi_bootmgr_get_unused_bootoption(var_name, > > sizeof(var_name), > > + &boot_index); > > + if (ret != EFI_SUCCESS) > > + goto out; > > + > > + ret = efi_set_variable_int(var_name, > > &efi_global_variable_guid, > > + EFI_VARIABLE_NON_VOLATILE | > > + > > EFI_VARIABLE_BOOTSERVICE_ACCESS | > > + EFI_VARIABLE_RUNTIME_ACCESS, > > + opt[i].size, opt[i].lo, > > false); > > + if (ret != EFI_SUCCESS) > > + goto out; > > + > > + ret = efi_bootmgr_append_bootorder(boot_index); > > + if (ret != EFI_SUCCESS) { > > + efi_set_variable_int(var_name, > > &efi_global_variable_guid, > > + 0, 0, NULL, false); > > + goto out; > > + } > > + } > > + } > > + > > +out: > > + if (opt) { > > + for (i = 0; i < count; i++) { > > + if (opt[i].lo) > > + free(opt[i].lo); > > + } > > + free(opt); > > + } > > + efi_free_pool(volume_handles); > > + > > + return ret; > > +} > > diff --git a/lib/efi_loader/efi_disk.c b/lib/efi_loader/efi_disk.c > > index d2256713a8..1309e28134 100644 > > --- a/lib/efi_loader/efi_disk.c > > +++ b/lib/efi_loader/efi_disk.c > > @@ -687,6 +687,13 @@ int efi_disk_probe(void *ctx, struct event *event) > > return -1; > > } > > > > + /* only do the boot option management when UEFI sub-system is > > initialized */ > > + if (efi_obj_list_initialized == EFI_SUCCESS) { > > + ret = efi_bootmgr_update_media_device_boot_option(); > > Why should we rescan all devices if a single device is added?
Then, don't forget to call the 'remove' function in efi_disk_remove() (or _delete_raw() and _delete_part()). > The only place where we need the list of removable media devices is when > executing the boot manager. Not sure what you mean, but we may want to confirm available boot options with efidebug or eficonfig before invoking the boot manager. -Takahiro Akashi > > > + if (ret) > > + return -1; > > + } > > + > > return 0; > > } > > > > diff --git a/lib/efi_loader/efi_helper.c b/lib/efi_loader/efi_helper.c > > index 1f4ab2b419..cdfd16ea77 100644 > > --- a/lib/efi_loader/efi_helper.c > > +++ b/lib/efi_loader/efi_helper.c > > @@ -257,3 +257,28 @@ efi_status_t efi_next_variable_name(efi_uintn_t *size, > > u16 **buf, efi_guid_t *gu > > > > return ret; > > } > > + > > +/** > > + * efi_search_bootorder() - search the boot option index in BootOrder > > + * > > + * @bootorder: pointer to the BootOrder variable > > + * @num: number of BootOrder entry > > + * @target: target boot option index to search > > + * @index: pointer to store the index of BootOrder variable > > + * Return: true if exists, false otherwise > > + */ > > +bool efi_search_bootorder(u16 *bootorder, efi_uintn_t num, u32 target, u32 > > *index) > > This function should be static. The exported function to delete a boot > option should not use the index in the boot order in their interface but > the number encoded in the variable name. > > We can simplify the function: > > Return -1 if the entry is not found and the index otherwise. > > > +{ > > + u32 i; > > + > > + for (i = 0; i < num; i++) { > > + if (target == bootorder[i]) { > > + if (index) > > + *index = i; > > + > > + return true; > > + } > > + } > > + > > + return false; > > +} > > diff --git a/lib/efi_loader/efi_variable.c b/lib/efi_loader/efi_variable.c > > index be95ed44e6..2f251553e1 100644 > > --- a/lib/efi_loader/efi_variable.c > > +++ b/lib/efi_loader/efi_variable.c > > @@ -476,6 +476,10 @@ efi_status_t efi_init_variables(void) > > log_err("Invalid EFI variable seed\n"); > > } > > > > + ret = efi_init_secure_state(); > > + if (ret != EFI_SUCCESS) > > + return ret; > > > > - return efi_init_secure_state(); > > + /* update boot option management after variable service initialized */ > > + return efi_bootmgr_update_media_device_boot_option(); > > The right place to call the function might be efi_init_obj_list(). > > Best regards > > Heinrich > > > } > > diff --git a/lib/efi_loader/efi_variable_tee.c > > b/lib/efi_loader/efi_variable_tee.c > > index dfef18435d..a48d313ef0 100644 > > --- a/lib/efi_loader/efi_variable_tee.c > > +++ b/lib/efi_loader/efi_variable_tee.c > > @@ -748,5 +748,6 @@ efi_status_t efi_init_variables(void) > > if (ret != EFI_SUCCESS) > > return ret; > > > > - return EFI_SUCCESS; > > + /* update boot option management after variable service initialized */ > > + return efi_bootmgr_update_media_device_boot_option(); > > } >