ArmPkg: Bug fix for UncachedMemoryAllocationLib

[mirror_edk2.git] / ArmPkg / Library / UncachedMemoryAllocationLib / UncachedMemoryAllocationLib.c

diff --git a/ArmPkg/Library/UncachedMemoryAllocationLib/UncachedMemoryAllocationLib.c b/ArmPkg/Library/UncachedMemoryAllocationLib/UncachedMemoryAllocationLib.c
index e70d8777d76d47706565cbd69e92dc15ff200399..b859f63c2b5a0256b01f650a4f29bce4ecf893de 100644 (file)
--- a/ArmPkg/Library/UncachedMemoryAllocationLib/UncachedMemoryAllocationLib.c
+++ b/ArmPkg/Library/UncachedMemoryAllocationLib/UncachedMemoryAllocationLib.c
@@ -125,9 +125,10 @@ AllocatePagesFromList (
   }\r
   // Check if we have found a node that could contain our new allocation\r
   if (NewNode != NULL) {\r
-    NewNode->Allocated = TRUE;\r
-    Node->Allocation   = (VOID*)(UINTN)Node->Base;\r
-    *Allocation        = Node->Allocation;\r
+    NewNode->Allocated  = TRUE;\r
+    NewNode->Allocation = (VOID*)(UINTN)NewNode->Base;\r
+    *Allocation         = NewNode->Allocation;\r
+    mFreedBufferSize    -= NewNode->Pages * EFI_PAGE_SIZE;\r
     return EFI_SUCCESS;\r
   }\r
 \r