X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdeModulePkg%2FCore%2FPiSmmCore%2FPiSmmCore.c;h=a7467aca201229dfb9511540ec7f02641e285945;hp=b833763f9a00ea3dc650f02f2210652afd1ccbd0;hb=e63da9f033274843163908ccefa95c892d7944e5;hpb=d76c2da8d50d29a1f38e01ce896ddfce47890c09

diff --git a/MdeModulePkg/Core/PiSmmCore/PiSmmCore.c b/MdeModulePkg/Core/PiSmmCore/PiSmmCore.c
index b833763f9a..a7467aca20 100644
--- a/MdeModulePkg/Core/PiSmmCore/PiSmmCore.c
+++ b/MdeModulePkg/Core/PiSmmCore/PiSmmCore.c
@@ -506,6 +506,11 @@ SmmEntryPoint (
   //
   PlatformHookBeforeSmmDispatch ();
 
+  //
+  // Call memory management hook function
+  //
+  SmmEntryPointMemoryManagementHook ();
+
   //
   // If a legacy boot has occured, then make sure gSmmCorePrivate is not accessed
   //
@@ -699,7 +704,7 @@ SmmMain (
   //
   gSmmCorePrivate->Smst          = &gSmmCoreSmst;
   gSmmCorePrivate->SmmEntryPoint = SmmEntryPoint;
-  
+
   //
   // No need to initialize memory service.
   // It is done in constructor of PiSmmCoreMemoryAllocationLib(),