X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=MdeModulePkg%2FUniversal%2FLoadFileOnFv2%2FLoadFileOnFv2.c;h=bd86db50f0ef8096e6018a3c43110fc3d4d2a6d0;hb=0dc3fb06b270ee8b370c5b219ce590fcfe66d551;hp=9eea50dd33fdf9debc6ad2277864bc664924683e;hpb=b40ad7b54d1dbdded77f779fa4d1dbe860783c8e;p=mirror_edk2.git

diff --git a/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c b/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
index 9eea50dd33..bd86db50f0 100644
--- a/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
+++ b/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
@@ -157,7 +157,7 @@ LOAD_FILE_ON_FV2_PRIVATE_DATA  mLoadFileOnFv2PrivateDataTemplate = {
 /**
   Check if the FFS has been installed LoadFileProtocol for it.
 
-  @param EFI_GUID     File GUID.
+  @param[in] NameGuid Point to FFS File GUID to be checked.
 
   @retval TRUE        The FFS's FileLoadProtocol is in list.
   @retval FALSE       The FFS's FileLoadProtocol is not in list.
@@ -345,6 +345,9 @@ FvNotificationEvent (
   Index      = 0;
   BufferSize = sizeof (EFI_HANDLE);
   Handle     = AllocateZeroPool (BufferSize);
+  if (Handle == NULL) {
+    return;
+  }
   Status = gBS->LocateHandle (
                     ByProtocol,
                     &gEfiFirmwareVolume2ProtocolGuid,
@@ -355,6 +358,9 @@ FvNotificationEvent (
   if (EFI_BUFFER_TOO_SMALL == Status) {
     FreePool (Handle);
     Handle = AllocateZeroPool (BufferSize);
+    if (Handle == NULL) {
+      return;
+    }
     Status = gBS->LocateHandle (
                     ByProtocol,
                     &gEfiFirmwareVolume2ProtocolGuid,