Add in support for MaxSize and CurrentSize for PCD entry.

[mirror_edk2.git] / MdePkg / Library / PeiPcdLib / PeiPcdLib.c

diff --git a/MdePkg/Library/PeiPcdLib/PeiPcdLib.c b/MdePkg/Library/PeiPcdLib/PeiPcdLib.c
index bee53860977c9f522e691064c135a54b9c919592..b8a53185bd8d59b4c94571321e814a5941359bca 100644 (file)
--- a/MdePkg/Library/PeiPcdLib/PeiPcdLib.c
+++ b/MdePkg/Library/PeiPcdLib/PeiPcdLib.c
@@ -586,7 +586,6 @@ LibPcdSetPtr (
 {\r
   EFI_STATUS Status;\r
   PCD_PPI    *PcdPpi;\r
 {\r
   EFI_STATUS Status;\r
   PCD_PPI    *PcdPpi;\r

-  UINTN      Size;\r

 \r
   ASSERT (SizeOfBuffer != NULL);\r
 \r
 \r
   ASSERT (SizeOfBuffer != NULL);\r
 \r


@@ -596,17 +595,12 @@ LibPcdSetPtr (
 \r
   PcdPpi = GetPcdPpiPtr ();\r
   \r
 \r
   PcdPpi = GetPcdPpiPtr ();\r
   \r

-  Size = LibPcdGetSize (TokenNumber);\r
-  \r
-  if (*SizeOfBuffer > Size) {\r
-    *SizeOfBuffer = Size;\r
+  Status = PcdPpi->SetPtr (TokenNumber, SizeOfBuffer, Buffer);\r
+\r
+  if (EFI_ERROR (Status)) {\r

     return NULL;\r
   }\r
 \r
     return NULL;\r
   }\r
 \r

-  Status = PcdPpi->SetPtr (TokenNumber, *SizeOfBuffer, Buffer);\r
-\r
-  ASSERT_EFI_ERROR (Status);\r
-\r

   return Buffer;\r
 }\r
 \r
   return Buffer;\r
 }\r
 \r


@@ -817,7 +811,6 @@ LibPcdSetExPtr (
 {\r
   EFI_STATUS      Status;\r
   PCD_PPI         *PcdPpi;\r
 {\r
   EFI_STATUS      Status;\r
   PCD_PPI         *PcdPpi;\r

-  UINTN           Size;\r

 \r
   if (*SizeOfBuffer > 0) {\r
     ASSERT (Buffer != NULL);\r
 \r
   if (*SizeOfBuffer > 0) {\r
     ASSERT (Buffer != NULL);\r


@@ -825,16 +818,12 @@ LibPcdSetExPtr (
 \r
   PcdPpi = GetPcdPpiPtr ();\r
 \r
 \r
   PcdPpi = GetPcdPpiPtr ();\r
 \r

-  Size = LibPcdGetExSize (Guid, TokenNumber);\r
-  if (*SizeOfBuffer > Size) {\r
-    *SizeOfBuffer = Size;\r
+  Status = PcdPpi->SetPtrEx (Guid, TokenNumber, SizeOfBuffer, Buffer);\r
+\r
+  if (EFI_ERROR (Status)) {\r

     return NULL;\r
   }\r
 \r
     return NULL;\r
   }\r
 \r

-  Status = PcdPpi->SetPtrEx (Guid, TokenNumber, *SizeOfBuffer, Buffer);\r
-\r
-  ASSERT_EFI_ERROR (Status);\r
-\r

   return Buffer;\r
 }\r
 \r
   return Buffer;\r
 }\r
 \r