mok import: handle OOM case

Coverity pointed out that AllocateZeroPool() can error, and then we
immediately dereference the NULL.

This handles the error case.

Signed-off-by: Peter Jones <pjones@redhat.com>

1 files changed, 4 insertions, 0 deletions

diff --git a/mok.c b/mok.c
index d1e51d0f..63ddfcaa 100644
--- a/mok.c
+++ b/mok.c
@@ -871,6 +871,10 @@ EFI_STATUS import_one_mok_state(struct mok_state_variable *v,
 		if (efi_status == EFI_NOT_FOUND &&
 		    v->flags & MOK_VARIABLE_INVERSE) {
 			v->data = AllocateZeroPool(4);
+			if (!v->data) {
+				perror(L"Out of memory\n");
+				return EFI_OUT_OF_RESOURCES;
+			}
 			v->data[0] = 0x01;
 			v->data_size = 1;
 		} else if (efi_status == EFI_NOT_FOUND) {