[PATCH v3 1/3] Move bootorder and bootoption apis to lib
Raymond Mao
raymond.mao at linaro.org
Tue May 16 16:32:43 CEST 2023
Hi Ilias,
I think we can address Heinrich's comments in a follow up patch set.
The concerns are more related to refactoring on the logic of existing
functions and less to the functionality we want to enable with these
patches.
Thanks.
Regards,
Raymond
On Tue, 16 May 2023 at 01:58, Ilias Apalodimas <ilias.apalodimas at linaro.org>
wrote:
> Hi Raymond,
>
> Heinrichm you had some concerns on the previous version of this patchset
> wrt to the current code. Do you think the rework should take place here?
> Or are you ok in cleaning it up in a follow up series?
>
> Thanks
> /Ilias
> On Tue, May 02, 2023 at 12:12:17PM -0700, Raymond Mao wrote:
> > Rename and move bootorder and bootoption apis from cmd to lib
> > for re-use between eficonfig and bootmgr
> >
> > Signed-off-by: Raymond Mao <raymond.mao at linaro.org>
> > ---
> > Changes in v2
> > - Ignore EFI_NOT_FOUND returned from
> > efi_bootmgr_update_media_device_boot_option which means no boot
> > options scanned.
> > Changes in v3
> > - Split the patch into moving and renaming functions and
> > individual patches for each changed functionality
> >
> > 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_helper.c | 25 +++
> > 6 files changed, 423 insertions(+), 408 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);
> >
> > 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
> > + *
> > + * @opt: pointer to the media boot option structure
> > + * @count: number of media boot option structure
> > + * Return: status code
> > + */
> > +static efi_status_t efi_bootmgr_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) {
> > + 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)
> > +{
> > + 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_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)
> > +{
> > + u32 i;
> > +
> > + for (i = 0; i < num; i++) {
> > + if (target == bootorder[i]) {
> > + if (index)
> > + *index = i;
> > +
> > + return true;
> > + }
> > + }
> > +
> > + return false;
> > +}
> > --
> > 2.25.1
> >
>
More information about the U-Boot
mailing list