MdeModulePkg/PeiMain: PeiAllocatePool: output NULL if HOB creation fails

[mirror_edk2.git] / MdeModulePkg / Core / Pei / Memory / MemoryServices.c

diff --git a/MdeModulePkg/Core/Pei/Memory/MemoryServices.c b/MdeModulePkg/Core/Pei/Memory/MemoryServices.c
index 42f79ab07633a591f5ea0a2f3d64e134e2640a27..706837890fde0658ca601d347f136b6bcd3b9f72 100644 (file)
--- a/MdeModulePkg/Core/Pei/Memory/MemoryServices.c
+++ b/MdeModulePkg/Core/Pei/Memory/MemoryServices.c
@@ -802,7 +802,12 @@ PeiAllocatePool (
              (VOID **)&Hob\r
              );\r
   ASSERT_EFI_ERROR (Status);\r
-  *Buffer = Hob+1;\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    *Buffer = NULL;\r
+  } else {\r
+    *Buffer = Hob + 1;\r
+  }\r
 \r
   return Status;\r
 }\r