1. Use MemoryAllocationLib to replace boot services memory services functions in...

[mirror_edk2.git] / EdkModulePkg / Universal / Variable / RuntimeDxe / reclaim.c

diff --git a/EdkModulePkg/Universal/Variable/RuntimeDxe/reclaim.c b/EdkModulePkg/Universal/Variable/RuntimeDxe/reclaim.c
index 00ebcf84fdb5689115c29cfca5612a5ed3343f33..4c7249046c0df8a26e8395400e32bd930c866b9a 100644 (file)
--- a/EdkModulePkg/Universal/Variable/RuntimeDxe/reclaim.c
+++ b/EdkModulePkg/Universal/Variable/RuntimeDxe/reclaim.c
@@ -1,20 +1,20 @@
 /*++\r
 \r
-Copyright (c) 2006, Intel Corporation                                                         \r
-All rights reserved. This program and the accompanying materials                          \r
-are licensed and made available under the terms and conditions of the BSD License         \r
-which accompanies this distribution.  The full text of the license may be found at        \r
-http://opensource.org/licenses/bsd-license.php                                            \r
-                                                                                          \r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
+Copyright (c) 2006 - 2007, Intel Corporation\r
+All rights reserved. This program and the accompanying materials\r
+are licensed and made available under the terms and conditions of the BSD License\r
+which accompanies this distribution.  The full text of the license may be found at\r
+http://opensource.org/licenses/bsd-license.php\r
+\r
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 \r
 Module Name:\r
 \r
   reclaim.c\r
-  \r
+\r
 Abstract:\r
-  \r
+\r
   Handles non-volatile variable store garbage collection, using FTW\r
   (Fault Tolerant Write) protocol.\r
 \r
@@ -82,7 +82,7 @@ GetFvbHandleByAddress (
     }\r
   }\r
 \r
-  gBS->FreePool (HandleBuffer);\r
+  FreePool (HandleBuffer);\r
   return Status;\r
 }\r
 \r
@@ -216,8 +216,8 @@ Returns:
   // Prepare for the variable data\r
   //\r
   FtwBufferSize = ((VARIABLE_STORE_HEADER *) ((UINTN) VariableBase))->Size;\r
-  Status        = gBS->AllocatePool (EfiRuntimeServicesData, FtwBufferSize, (VOID **) &FtwBuffer);\r
-  if (EFI_ERROR (Status)) {\r
+  FtwBuffer     = AllocateRuntimePool (FtwBufferSize);\r
+  if (FtwBuffer == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
@@ -236,6 +236,6 @@ Returns:
                               FtwBuffer\r
                               );\r
 \r
-  gBS->FreePool (FtwBuffer);\r
+  FreePool (FtwBuffer);\r
   return Status;\r
 }\r