ArmPkg,ArmPlatformPkg: Free memory allocated by Get.*SpaceMap()

[mirror_edk2.git] / ArmPkg / Drivers / CpuDxe / AArch64 / Mmu.c

diff --git a/ArmPkg/Drivers/CpuDxe/AArch64/Mmu.c b/ArmPkg/Drivers/CpuDxe/AArch64/Mmu.c
index 9043afaffad5ba0402bbda87aef3416274de004a..da6e6225bed28751dc1c043261912232e422df4f 100644 (file)
--- a/ArmPkg/Drivers/CpuDxe/AArch64/Mmu.c
+++ b/ArmPkg/Drivers/CpuDxe/AArch64/Mmu.c
@@ -15,6 +15,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 --*/\r
 \r
+#include <Library/MemoryAllocationLib.h>\r
 #include "CpuDxe.h"\r
 \r
 #define TT_ATTR_INDX_INVALID    ((UINT32)~0)\r
@@ -122,6 +123,8 @@ GetNextEntryAttribute (
     }\r
   }\r
 \r
+  FreePool (MemorySpaceMap);\r
+\r
   return BaseAddress + (EntryCount * TT_ADDRESS_AT_LEVEL(TableLevel));\r
 }\r
 \r
@@ -185,5 +188,7 @@ SyncCacheConfig (
       EndAddressGcdRegion - BaseAddressGcdRegion,\r
       PageAttributeToGcdAttribute (PageAttribute));\r
 \r
+  FreePool (MemorySpaceMap);\r
+\r
   return EFI_SUCCESS;\r
 }\r