efi_selftest: error handling in efi_selftest_tcg2

If memory allocation fails, write an error message.

Signed-off-by: Heinrich Schuchardt <heinrich.schuchardt@canonical.com>

diff --git a/lib/efi_selftest/efi_selftest_tcg2.c b/lib/efi_selftest/efi_selftest_tcg2.c
index a2b4a79e9b24e3322a471c61738cbb47a9b14c5e..67a886efaa8506b68407c127f1c94d1260e35d28 100644 (file)
--- a/lib/efi_selftest/efi_selftest_tcg2.c
+++ b/lib/efi_selftest/efi_selftest_tcg2.c
@@ -631,8 +631,10 @@ static int efi_st_tcg2_setup(const efi_handle_t img_handle,
                                      sizeof(struct efi_tcg2_event) +
                                      sizeof(struct uefi_image_load_event),
                                      (void **)&efi_tcg2_event);
-       if (!efi_tcg2_event)
+       if (ret != EFI_SUCCESS) {
+               efi_st_error("Out of memory\n");
                return EFI_ST_FAILURE;
+       }
 
        efi_tcg2_event->size = sizeof(struct efi_tcg2_event) +
                               sizeof(struct uefi_image_load_event);
@@ -659,8 +661,10 @@ static int efi_st_tcg2_setup(const efi_handle_t img_handle,
                                      (EFI_TCG2_MAX_PCR_INDEX + 1) *
                                      TPM2_SHA256_DIGEST_SIZE,
                                      (void **)&pcrs);
-       if (!pcrs)
+       if (ret != EFI_SUCCESS) {
+               efi_st_error("Out of memory\n");
                return EFI_ST_FAILURE;
+       }
 
        boottime->set_mem(pcrs, (EFI_TCG2_MAX_PCR_INDEX + 1) * TPM2_SHA256_DIGEST_SIZE, 0);