MdeModulePkg/PiSmmCore: AllocatePool should use MemoryType.

[mirror_edk2.git] / MdeModulePkg / Core / PiSmmCore / PiSmmCore.h

diff --git a/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h b/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h
index e2fee549f8d82742c196fbf15aec7781d296e86f..8df1e500f0229604253ea61ea1946b3788447387 100644 (file)
--- a/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h
+++ b/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h
@@ -1109,8 +1109,9 @@ extern LIST_ENTRY  mSmmMemoryMap;
 #define MAX_POOL_INDEX  (MAX_POOL_SHIFT - MIN_POOL_SHIFT + 1)\r
 \r
 typedef struct {\r
-  UINTN        Size;\r
-  BOOLEAN      Available;\r
+  UINTN           Size;\r
+  BOOLEAN         Available;\r
+  EFI_MEMORY_TYPE Type;\r
 } POOL_HEADER;\r
 \r
 typedef struct {\r
@@ -1118,6 +1119,12 @@ typedef struct {
   LIST_ENTRY   Link;\r
 } FREE_POOL_HEADER;\r
 \r
-extern LIST_ENTRY  mSmmPoolLists[MAX_POOL_INDEX];\r
+typedef enum {\r
+  SmmPoolTypeCode,\r
+  SmmPoolTypeData,\r
+  SmmPoolTypeMax,\r
+} SMM_POOL_TYPE;\r
+\r
+extern LIST_ENTRY  mSmmPoolLists[SmmPoolTypeMax][MAX_POOL_INDEX];\r
 \r
 #endif\r