MdeModulePkg/Core: allow HeapGuard even before CpuArchProtocol installed

[mirror_edk2.git] / MdeModulePkg / Core / Dxe / Mem / HeapGuard.h

diff --git a/MdeModulePkg/Core/Dxe/Mem/HeapGuard.h b/MdeModulePkg/Core/Dxe/Mem/HeapGuard.h
index 7208ab143719a2b2923b9db97585d52de000f891..8c34692439d37d37b6dd778ce589b0a5800635af 100644 (file)
--- a/MdeModulePkg/Core/Dxe/Mem/HeapGuard.h
+++ b/MdeModulePkg/Core/Dxe/Mem/HeapGuard.h
@@ -399,6 +399,14 @@ IsHeapGuardEnabled (
   VOID\r
   );\r
 \r
   VOID\r
   );\r
 \r

+/**\r
+  Notify function used to set all Guard pages after CPU Arch Protocol installed.\r
+**/\r
+VOID\r
+HeapGuardCpuArchProtocolNotify (\r
+  VOID\r
+  );\r
+\r

 extern BOOLEAN mOnGuarding;\r
 \r
 #endif\r
 extern BOOLEAN mOnGuarding;\r
 \r
 #endif\r