MdeModulePkg/BdsDxe: Avoid overwriting PlatformRecovery####

[mirror_edk2.git] / MdeModulePkg / Universal / LoadFileOnFv2 / LoadFileOnFv2.c

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