From 9945bc4f867a9d0dd2c69b86b182265320f37d4b Mon Sep 17 00:00:00 2001 From: Raymond Mao Date: Mon, 19 Jun 2023 14:22:59 -0700 Subject: 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 Reviewed-by: Ilias Apalodimas --- lib/efi_loader/efi_bootmgr.c | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'lib') diff --git a/lib/efi_loader/efi_bootmgr.c b/lib/efi_loader/efi_bootmgr.c index f6110f2269..993bb1139c 100644 --- a/lib/efi_loader/efi_bootmgr.c +++ b/lib/efi_loader/efi_bootmgr.c @@ -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; } -- cgit v1.2.3