ArmPlatformPkg/MemoryInitPeim: take MAX_ALLOC_ADDRESS into account

[mirror_edk2.git] / ArmPlatformPkg / MemoryInitPei / MemoryInitPeim.c

diff --git a/ArmPlatformPkg/MemoryInitPei/MemoryInitPeim.c b/ArmPlatformPkg/MemoryInitPei/MemoryInitPeim.c
index 389a2e6f1abdc19553a074abbc81c3954c292068..010f93add76b234fc17a078db2f6a3344e996cdd 100644 (file)
--- a/ArmPlatformPkg/MemoryInitPei/MemoryInitPeim.c
+++ b/ArmPlatformPkg/MemoryInitPei/MemoryInitPeim.c
@@ -105,12 +105,12 @@ InitializeMemory (
 \r
   // Ensure PcdSystemMemorySize has been set\r
   ASSERT (PcdGet64 (PcdSystemMemorySize) != 0);\r
-  ASSERT (PcdGet64 (PcdSystemMemoryBase) < (UINT64)MAX_ADDRESS);\r
+  ASSERT (PcdGet64 (PcdSystemMemoryBase) < (UINT64)MAX_ALLOC_ADDRESS);\r
 \r
   SystemMemoryBase = (UINTN)PcdGet64 (PcdSystemMemoryBase);\r
   SystemMemoryTop = SystemMemoryBase + PcdGet64 (PcdSystemMemorySize);\r
-  if (SystemMemoryTop - 1 > MAX_ADDRESS) {\r
-    SystemMemoryTop = (UINT64)MAX_ADDRESS + 1;\r
+  if (SystemMemoryTop - 1 > MAX_ALLOC_ADDRESS) {\r
+    SystemMemoryTop = (UINT64)MAX_ALLOC_ADDRESS + 1;\r
   }\r
   FdBase = (UINTN)PcdGet64 (PcdFdBaseAddress);\r
   FdTop = FdBase + (UINTN)PcdGet32 (PcdFdSize);\r