From 79cbcdcf3ececdb45f02c528d00e53e9600a4f48 Mon Sep 17 00:00:00 2001 From: Toomas Soome Date: Thu, 20 Feb 2020 08:56:56 +0000 Subject: [PATCH] efipart_inithandles() is missing NULL pointer check Add missing check of malloc() result. --- stand/efi/libefi/efipart.c | 2 ++ 1 file changed, 2 insertions(+) diff --git a/stand/efi/libefi/efipart.c b/stand/efi/libefi/efipart.c index 0710a33fdf9..424df7d2e42 100644 --- a/stand/efi/libefi/efipart.c +++ b/stand/efi/libefi/efipart.c @@ -363,6 +363,8 @@ efipart_inithandles(void) status = BS->LocateHandle(ByProtocol, &blkio_guid, 0, &sz, hin); if (status == EFI_BUFFER_TOO_SMALL) { hin = malloc(sz); + if (hin == NULL) + return (ENOMEM); status = BS->LocateHandle(ByProtocol, &blkio_guid, 0, &sz, hin); if (EFI_ERROR(status))