Fix incorrect return code of boot option update

Correct the return code for out-of-memory and no boot option found

Signed-off-by: Raymond Mao <raymond.mao@linaro.org>
Reviewed-by: Ilias Apalodimas <ilias.apalodimas@linaro.org>
This commit is contained in:
Raymond Mao 2023-06-19 14:22:59 -07:00 committed by Heinrich Schuchardt
parent 339b527bd4
commit 9945bc4f86
3 changed files with 8 additions and 4 deletions

View File

@ -352,7 +352,7 @@ static struct bootmenu_data *bootmenu_create(int delay)
* a architecture-specific default image name such as BOOTAA64.EFI.
*/
efi_ret = efi_bootmgr_update_media_device_boot_option();
if (efi_ret != EFI_SUCCESS && efi_ret != EFI_NOT_FOUND)
if (efi_ret != EFI_SUCCESS)
goto cleanup;
ret = prepare_uefi_bootorder_entry(menu, &iter, &i);

View File

@ -2314,7 +2314,7 @@ static int do_eficonfig(struct cmd_tbl *cmdtp, int flag, int argc, char *const a
return CMD_RET_FAILURE;
ret = efi_bootmgr_update_media_device_boot_option();
if (ret != EFI_SUCCESS && ret != EFI_NOT_FOUND)
if (ret != EFI_SUCCESS)
return ret;
while (1) {

View File

@ -660,11 +660,13 @@ efi_status_t efi_bootmgr_update_media_device_boot_option(void)
NULL, &count,
(efi_handle_t **)&volume_handles);
if (ret != EFI_SUCCESS)
return ret;
goto out;
opt = calloc(count, sizeof(struct eficonfig_media_boot_option));
if (!opt)
if (!opt) {
ret = EFI_OUT_OF_RESOURCES;
goto out;
}
/* enumerate all devices supporting EFI_SIMPLE_FILE_SYSTEM_PROTOCOL */
ret = efi_bootmgr_enumerate_boot_option(opt, volume_handles, count);
@ -717,5 +719,7 @@ out:
free(opt);
efi_free_pool(volume_handles);
if (ret == EFI_NOT_FOUND)
return EFI_SUCCESS;
return ret;
}