MdeModule PeiCore: Support pre memory page allocation

[mirror_edk2.git] / MdeModulePkg / Core / Pei / PeiMain / PeiMain.c

diff --git a/MdeModulePkg/Core/Pei/PeiMain/PeiMain.c b/MdeModulePkg/Core/Pei/PeiMain/PeiMain.c
index d2897437d776b3bfb8e24d1699850004cbb7d1f1..3cd61906c3dff8a7166980d668e457e7af51ab86 100644 (file)
--- a/MdeModulePkg/Core/Pei/PeiMain/PeiMain.c
+++ b/MdeModulePkg/Core/Pei/PeiMain/PeiMain.c
@@ -64,7 +64,8 @@ EFI_PEI_SERVICES  gPs = {
   PeiRegisterForShadow,\r
   PeiFfsFindSectionData3,\r
   PeiFfsGetFileInfo2,\r
-  PeiResetSystem2\r
+  PeiResetSystem2,\r
+  PeiFreePages,\r
 };\r
 \r
 /**\r
@@ -231,6 +232,11 @@ PeiCore (
       HandoffInformationTable->EfiFreeMemoryTop    = OldCoreData->FreePhysicalMemoryTop;\r
       HandoffInformationTable->EfiFreeMemoryBottom = HandoffInformationTable->EfiEndOfHobList + sizeof (EFI_HOB_GENERIC_HEADER);\r
 \r
+      //\r
+      // We need convert MemoryBaseAddress in memory allocation HOBs\r
+      //\r
+      ConvertMemoryAllocationHobs (OldCoreData);\r
+\r
       //\r
       // We need convert the PPI descriptor's pointer\r
       //\r