]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdePkg/Library/DxePcdLib/DxePcdLib.c
MdePkg PcdLib: Follow the comment of LibPcdGetNextToken to ASSERT when an invalid...
[mirror_edk2.git] / MdePkg / Library / DxePcdLib / DxePcdLib.c
index e6feeaccdc65ab2b3b6f71a8de9a8e7c2aa1a25f..f14a74652978bb38f6c1af867ab4294c7faa6e77 100644 (file)
@@ -1,8 +1,8 @@
 /** @file\r
 Implementation of PcdLib class library for DXE phase.\r
 \r
-Copyright (c) 2006 - 2009, Intel Corporation<BR>\r
-All rights reserved. This program and the accompanying materials                          \r
+Copyright (c) 2006 - 2013, Intel Corporation. All rights reserved.<BR>\r
+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
@@ -18,51 +18,108 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 #include <Protocol/Pcd.h>\r
 #include <Protocol/PiPcd.h>\r
+#include <Protocol/PcdInfo.h>\r
+#include <Protocol/PiPcdInfo.h>\r
 \r
 #include <Library/PcdLib.h>\r
 #include <Library/DebugLib.h>\r
 #include <Library/UefiBootServicesTableLib.h>\r
 #include <Library/BaseMemoryLib.h>\r
 \r
-PCD_PROTOCOL     *mPcd = NULL;\r
-EFI_PCD_PROTOCOL *mPiPcd = NULL;\r
+PCD_PROTOCOL                *mPcd       = NULL;\r
+EFI_PCD_PROTOCOL            *mPiPcd     = NULL;\r
+GET_PCD_INFO_PROTOCOL       *mPcdInfo   = NULL;\r
+EFI_GET_PCD_INFO_PROTOCOL   *mPiPcdInfo = NULL;\r
 \r
 /**\r
-  The constructor function caches the PCD_PROTOCOL pointer.\r
+  Retrieves the PI PCD protocol from the handle database.\r
 \r
-  @param[in] ImageHandle The firmware allocated handle for the EFI image.  \r
-  @param[in] SystemTable A pointer to the EFI System Table.\r
-  \r
-  @retval EFI_SUCCESS The constructor always return EFI_SUCCESS.\r
+  @retval EFI_PCD_PROTOCOL * The pointer to the EFI_PCD_PROTOCOL.\r
+**/\r
+EFI_PCD_PROTOCOL *\r
+EFIAPI\r
+GetPiPcdProtocol (\r
+  VOID\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
 \r
+  if (mPiPcd == NULL) {\r
+    //\r
+    // PI Pcd protocol defined in PI 1.2 vol3 should be installed before the module \r
+    // access DynamicEx type PCD.\r
+    //\r
+    Status = gBS->LocateProtocol (&gEfiPcdProtocolGuid, NULL, (VOID **) &mPiPcd);\r
+    ASSERT_EFI_ERROR (Status);\r
+    ASSERT (mPiPcd != NULL);\r
+  }\r
+  return mPiPcd;\r
+}\r
+\r
+/**\r
+  Retrieves the PCD protocol from the handle database.\r
+\r
+  @retval PCD_PROTOCOL * The pointer to the PCD_PROTOCOL.\r
 **/\r
-EFI_STATUS\r
+PCD_PROTOCOL *\r
 EFIAPI\r
-PcdLibConstructor (\r
-  IN EFI_HANDLE        ImageHandle,\r
-  IN EFI_SYSTEM_TABLE  *SystemTable\r
+GetPcdProtocol (\r
+  VOID\r
   )\r
 {\r
   EFI_STATUS  Status;\r
 \r
-  //\r
-  // PCD protocol need to be installed before the module access Dynamic type PCD.\r
-  // But dynamic type PCD is not required in PI 1.2 specification.\r
-  // \r
-  gBS->LocateProtocol (&gPcdProtocolGuid, NULL, (VOID **)&mPcd);\r
-  \r
-  //\r
-  // PI Pcd protocol defined in PI 1.2 vol3 should be installed before the module \r
-  // access DynamicEx type PCD.\r
-  //\r
-  Status = gBS->LocateProtocol (&gEfiPcdProtocolGuid, NULL, (VOID **) mPiPcd);\r
-  \r
-  ASSERT_EFI_ERROR (Status);\r
-  ASSERT (mPiPcd!= NULL);\r
+  if (mPcd == NULL) {\r
+    //\r
+    // PCD protocol need to be installed before the module access Dynamic type PCD.\r
+    // But dynamic type PCD is not required in PI 1.2 specification.\r
+    // \r
+    Status = gBS->LocateProtocol (&gPcdProtocolGuid, NULL, (VOID **)&mPcd);\r
+    ASSERT_EFI_ERROR (Status);\r
+    ASSERT (mPcd != NULL);\r
+  }\r
+  return mPcd;\r
+}\r
+\r
+/**\r
+  Retrieves the PI PCD info protocol from the handle database.\r
 \r
-  return Status;\r
+  @retval EFI_GET_PCD_INFO_PROTOCOL * The pointer to the EFI_GET_PCD_INFO_PROTOCOL defined in PI 1.2.1 Vol 3.\r
+**/\r
+EFI_GET_PCD_INFO_PROTOCOL *\r
+GetPiPcdInfoProtocolPointer (\r
+  VOID\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+\r
+  if (mPiPcdInfo == NULL) {\r
+    Status = gBS->LocateProtocol (&gEfiGetPcdInfoProtocolGuid, NULL, (VOID **)&mPiPcdInfo);\r
+    ASSERT_EFI_ERROR (Status);\r
+    ASSERT (mPiPcdInfo != NULL);\r
+  }\r
+  return mPiPcdInfo;\r
 }\r
 \r
+/**\r
+  Retrieves the PCD info protocol from the handle database.\r
+\r
+  @retval GET_PCD_INFO_PROTOCOL * The pointer to the GET_PCD_INFO_PROTOCOL.\r
+**/\r
+GET_PCD_INFO_PROTOCOL *\r
+GetPcdInfoProtocolPointer (\r
+  VOID\r
+  ) \r
+{\r
+  EFI_STATUS  Status;\r
+\r
+  if (mPcdInfo == NULL) {\r
+    Status = gBS->LocateProtocol (&gGetPcdInfoProtocolGuid, NULL, (VOID **)&mPcdInfo);\r
+    ASSERT_EFI_ERROR (Status);\r
+    ASSERT (mPcdInfo != NULL);\r
+  }\r
+  return mPcdInfo;\r
+}\r
 \r
 /**\r
   This function provides a means by which SKU support can be established in the PCD infrastructure.\r
@@ -82,10 +139,9 @@ LibPcdSetSku (
   IN UINTN   SkuId\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
   ASSERT (SkuId < PCD_MAX_SKU_ID);\r
 \r
-  mPcd->SetSku (SkuId);\r
+  GetPcdProtocol()->SetSku (SkuId);\r
 \r
   return SkuId;\r
 }\r
@@ -108,8 +164,7 @@ LibPcdGet8 (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->Get8 (TokenNumber);\r
+  return GetPcdProtocol()->Get8 (TokenNumber);\r
 }\r
 \r
 \r
@@ -130,8 +185,7 @@ LibPcdGet16 (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->Get16 (TokenNumber);\r
+  return GetPcdProtocol()->Get16 (TokenNumber);\r
 }\r
 \r
 \r
@@ -152,8 +206,7 @@ LibPcdGet32 (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->Get32 (TokenNumber);\r
+  return GetPcdProtocol()->Get32 (TokenNumber);\r
 }\r
 \r
 \r
@@ -174,8 +227,7 @@ LibPcdGet64 (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->Get64 (TokenNumber);\r
+  return GetPcdProtocol()->Get64 (TokenNumber);\r
 }\r
 \r
 \r
@@ -196,8 +248,7 @@ LibPcdGetPtr (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->GetPtr (TokenNumber);\r
+  return GetPcdProtocol()->GetPtr (TokenNumber);\r
 }\r
 \r
 \r
@@ -218,8 +269,7 @@ LibPcdGetBool (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->GetBool (TokenNumber);\r
+  return GetPcdProtocol()->GetBool (TokenNumber);\r
 }\r
 \r
 \r
@@ -238,8 +288,7 @@ LibPcdGetSize (
   IN UINTN             TokenNumber\r
   )\r
 {\r
-  ASSERT (mPcd != NULL);\r
-  return mPcd->GetSize (TokenNumber);\r
+  return GetPcdProtocol()->GetSize (TokenNumber);\r
 }\r
 \r
 \r
@@ -251,7 +300,7 @@ LibPcdGetSize (
   \r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid         Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid         The pointer to a 128-bit unique value that designates \r
                            which namespace to retrieve a value from.\r
   @param[in]  TokenNumber  The PCD token number to retrieve a current value for.\r
 \r
@@ -267,7 +316,7 @@ LibPcdGetEx8 (
 {\r
   ASSERT (Guid != NULL);\r
   \r
-  return mPiPcd->Get8Ex (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->Get8 (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -278,7 +327,7 @@ LibPcdGetEx8 (
   \r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid         Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid         The pointer to a 128-bit unique value that designates \r
                            which namespace to retrieve a value from.\r
   @param[in]  TokenNumber  The PCD token number to retrieve a current value for.\r
 \r
@@ -294,7 +343,7 @@ LibPcdGetEx16 (
 {\r
   ASSERT (Guid != NULL);\r
 \r
-  return mPiPcd->Get16Ex (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->Get16 (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -302,7 +351,7 @@ LibPcdGetEx16 (
   Returns the 32-bit value for the token specified by TokenNumber and Guid.\r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid         Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid         The pointer to a 128-bit unique value that designates \r
                            which namespace to retrieve a value from.\r
   @param[in]  TokenNumber  The PCD token number to retrieve a current value for.\r
 \r
@@ -318,7 +367,7 @@ LibPcdGetEx32 (
 {\r
   ASSERT (Guid != NULL);\r
 \r
-  return mPiPcd->Get32Ex (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->Get32 (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -330,7 +379,7 @@ LibPcdGetEx32 (
   \r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that designates \r
                             which namespace to retrieve a value from.\r
   @param[in]  TokenNumber   The PCD token number to retrieve a current value for.\r
 \r
@@ -346,7 +395,7 @@ LibPcdGetEx64 (
 {\r
   ASSERT (Guid != NULL);\r
   \r
-  return mPiPcd->Get64Ex (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->Get64 (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -358,7 +407,7 @@ LibPcdGetEx64 (
   \r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that designates \r
                             which namespace to retrieve a value from.\r
   @param[in]  TokenNumber   The PCD token number to retrieve a current value for.\r
 \r
@@ -374,7 +423,7 @@ LibPcdGetExPtr (
 {\r
   ASSERT (Guid != NULL);\r
 \r
-  return mPiPcd->GetPtrEx (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->GetPtr (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -386,7 +435,7 @@ LibPcdGetExPtr (
   \r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that designates \r
                             which namespace to retrieve a value from.\r
   @param[in]  TokenNumber   The PCD token number to retrieve a current value for.\r
 \r
@@ -402,7 +451,7 @@ LibPcdGetExBool (
 {\r
   ASSERT (Guid != NULL);\r
 \r
-  return mPiPcd->GetBoolEx (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->GetBool (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -414,7 +463,7 @@ LibPcdGetExBool (
   \r
   If Guid is NULL, then ASSERT(). \r
 \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that designates \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that designates \r
                             which namespace to retrieve a value from.\r
   @param[in]  TokenNumber   The PCD token number to retrieve a current value for.\r
 \r
@@ -430,7 +479,7 @@ LibPcdGetExSize (
 {\r
   ASSERT (Guid != NULL);\r
 \r
-  return mPiPcd->GetSizeEx (Guid, TokenNumber);\r
+  return GetPiPcdProtocol()->GetSize (Guid, TokenNumber);\r
 }\r
 \r
 \r
@@ -444,7 +493,7 @@ LibPcdGetExSize (
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 8-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT8\r
@@ -456,9 +505,7 @@ LibPcdSet8 (
 {\r
   EFI_STATUS Status;\r
 \r
-  ASSERT (mPcd != NULL);\r
-  Status = mPcd->Set8 (TokenNumber, Value);\r
-\r
+  Status = GetPcdProtocol()->Set8 (TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
   \r
   return Value;\r
@@ -475,7 +522,7 @@ LibPcdSet8 (
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 16-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT16\r
@@ -487,9 +534,7 @@ LibPcdSet16 (
 {\r
   EFI_STATUS Status;\r
 \r
-  ASSERT (mPcd != NULL);\r
-  Status = mPcd->Set16 (TokenNumber, Value);\r
-\r
+  Status = GetPcdProtocol()->Set16 (TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
   \r
   return Value;\r
@@ -506,7 +551,7 @@ LibPcdSet16 (
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 32-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT32\r
@@ -518,9 +563,7 @@ LibPcdSet32 (
 {\r
   EFI_STATUS Status;\r
   \r
-  ASSERT (mPcd != NULL);\r
-  Status = mPcd->Set32 (TokenNumber, Value);\r
-\r
+  Status = GetPcdProtocol()->Set32 (TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -537,7 +580,7 @@ LibPcdSet32 (
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 64-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT64\r
@@ -549,9 +592,7 @@ LibPcdSet64 (
 {\r
   EFI_STATUS Status;\r
 \r
-  ASSERT (mPcd != NULL);\r
-  Status = mPcd->Set64 (TokenNumber, Value);\r
-\r
+  Status = GetPcdProtocol()->Set64 (TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -591,20 +632,18 @@ LibPcdSetPtr (
 {\r
   EFI_STATUS Status;\r
 \r
-  ASSERT (mPcd != NULL);\r
   ASSERT (SizeOfBuffer != NULL);\r
 \r
   if (*SizeOfBuffer > 0) {\r
     ASSERT (Buffer != NULL);\r
   }\r
 \r
-  Status = mPcd->SetPtr (TokenNumber, SizeOfBuffer, (VOID *) Buffer);\r
-\r
+  Status = GetPcdProtocol()->SetPtr (TokenNumber, SizeOfBuffer, (VOID *) Buffer);\r
   if (EFI_ERROR (Status)) {\r
     return NULL;\r
   }\r
 \r
-  return (VOID *) Buffer;\r
+  return (VOID *)Buffer;\r
 }\r
 \r
 \r
@@ -618,7 +657,7 @@ LibPcdSetPtr (
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The boolean value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 BOOLEAN\r
@@ -630,9 +669,7 @@ LibPcdSetBool (
 {\r
   EFI_STATUS Status;\r
 \r
-  ASSERT (mPcd != NULL);\r
-  Status = mPcd->SetBool (TokenNumber, Value);\r
-\r
+  Status = GetPcdProtocol()->SetBool (TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -648,12 +685,12 @@ LibPcdSetBool (
   \r
   If Guid is NULL, then ASSERT().\r
   \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that \r
                             designates which namespace to set a value from.\r
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 8-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT8\r
@@ -668,8 +705,7 @@ LibPcdSetEx8 (
 \r
   ASSERT (Guid != NULL);\r
 \r
-  Status = mPiPcd->Set8Ex (Guid, TokenNumber, Value);\r
-\r
+  Status = GetPiPcdProtocol()->Set8 (Guid, TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -685,12 +721,12 @@ LibPcdSetEx8 (
   \r
   If Guid is NULL, then ASSERT().\r
   \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that \r
                             designates which namespace to set a value from.\r
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 16-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT16\r
@@ -705,8 +741,7 @@ LibPcdSetEx16 (
 \r
   ASSERT (Guid != NULL);\r
 \r
-  Status = mPiPcd->Set16Ex (Guid, TokenNumber, Value);\r
-\r
+  Status = GetPiPcdProtocol()->Set16 (Guid, TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -722,12 +757,12 @@ LibPcdSetEx16 (
   \r
   If Guid is NULL, then ASSERT().\r
   \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that \r
                             designates which namespace to set a value from.\r
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 32-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT32\r
@@ -742,8 +777,7 @@ LibPcdSetEx32 (
 \r
   ASSERT (Guid != NULL);\r
 \r
-  Status = mPiPcd->Set32Ex (Guid, TokenNumber, Value);\r
-\r
+  Status = GetPiPcdProtocol()->Set32 (Guid, TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -758,12 +792,12 @@ LibPcdSetEx32 (
   Guid to the value specified by Value. Value is returned.\r
   If Guid is NULL, then ASSERT().\r
   \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that \r
                             designates which namespace to set a value from.\r
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The 64-bit value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 UINT64\r
@@ -778,8 +812,7 @@ LibPcdSetEx64 (
 \r
   ASSERT (Guid != NULL);\r
 \r
-  Status = mPiPcd->Set64Ex (Guid, TokenNumber, Value);\r
-\r
+  Status = GetPiPcdProtocol()->Set64 (Guid, TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -800,13 +833,13 @@ LibPcdSetEx64 (
   If SizeOfBuffer is NULL, then ASSERT().\r
   If SizeOfBuffer > 0 and Buffer is NULL, then ASSERT().\r
   \r
-  @param[in]  Guid              Pointer to a 128-bit unique value that \r
+  @param[in]  Guid              The pointer to a 128-bit unique value that \r
                                 designates which namespace to set a value from.\r
   @param[in]  TokenNumber       The PCD token number to set a current value for.\r
   @param[in, out] SizeOfBuffer  The size, in bytes, of Buffer.\r
   @param[in]  Buffer            A pointer to the buffer to set.\r
 \r
-  @return Return the pinter to the buffer been set.\r
+  @return Return the pointer to the buffer been set.\r
 \r
 **/\r
 VOID *\r
@@ -828,8 +861,7 @@ LibPcdSetExPtr (
     ASSERT (Buffer != NULL);\r
   }\r
 \r
-  Status = mPiPcd->SetPtrEx (Guid, TokenNumber, SizeOfBuffer, Buffer);\r
-\r
+  Status = GetPiPcdProtocol()->SetPtr (Guid, TokenNumber, SizeOfBuffer, Buffer);\r
   if (EFI_ERROR (Status)) {\r
     return NULL;\r
   }\r
@@ -847,12 +879,12 @@ LibPcdSetExPtr (
   \r
   If Guid is NULL, then ASSERT().\r
   \r
-  @param[in]  Guid          Pointer to a 128-bit unique value that \r
+  @param[in]  Guid          The pointer to a 128-bit unique value that \r
                             designates which namespace to set a value from.\r
   @param[in]  TokenNumber   The PCD token number to set a current value for.\r
   @param[in]  Value         The Boolean value to set.\r
 \r
-  @return Return the value been set.\r
+  @return Return the value that was set.\r
 \r
 **/\r
 BOOLEAN\r
@@ -867,8 +899,7 @@ LibPcdSetExBool (
 \r
   ASSERT (Guid != NULL);\r
 \r
-  Status = mPiPcd->SetBoolEx (Guid, TokenNumber, Value);\r
-\r
+  Status = GetPiPcdProtocol()->SetBool (Guid, TokenNumber, Value);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return Value;\r
@@ -884,7 +915,7 @@ LibPcdSetExBool (
   If Guid is NULL, then the default token space is used.\r
   If NotificationFunction is NULL, then ASSERT().\r
 \r
-  @param[in]  Guid                  Pointer to a 128-bit unique value that designates which \r
+  @param[in]  Guid                  The pointer to a 128-bit unique value that designates which \r
                                     namespace to set a value from.  If NULL, then the default \r
                                     token space is used.\r
   @param[in]  TokenNumber           The PCD token number to monitor.\r
@@ -904,8 +935,7 @@ LibPcdCallbackOnSet (
 \r
   ASSERT (NotificationFunction != NULL);\r
 \r
-  Status = mPiPcd->CallbackOnSet (Guid, TokenNumber, (EFI_PCD_PROTOCOL_CALLBACK) NotificationFunction);\r
-\r
+  Status = GetPiPcdProtocol()->CallbackOnSet (Guid, TokenNumber, (EFI_PCD_PROTOCOL_CALLBACK) NotificationFunction);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return;\r
@@ -938,8 +968,7 @@ LibPcdCancelCallback (
 \r
   ASSERT (NotificationFunction != NULL);\r
     \r
-  Status = mPiPcd->CancelCallback (Guid, TokenNumber, (EFI_PCD_PROTOCOL_CALLBACK) NotificationFunction);\r
-\r
+  Status = GetPiPcdProtocol()->CancelCallback (Guid, TokenNumber, (EFI_PCD_PROTOCOL_CALLBACK) NotificationFunction);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
   return;\r
@@ -958,7 +987,7 @@ LibPcdCancelCallback (
   \r
   If TokenNumber is not 0 and is not in the token space specified by Guid, then ASSERT().\r
 \r
-  @param[in]  Guid        Pointer to a 128-bit unique value that designates which namespace \r
+  @param[in]  Guid        The pointer to a 128-bit unique value that designates which namespace \r
                           to set a value from.  If NULL, then the default token space is used.\r
   @param[in]  TokenNumber The previous PCD token number.  If 0, then retrieves the first PCD \r
                           token number.\r
@@ -973,11 +1002,10 @@ LibPcdGetNextToken (
   IN UINTN                    TokenNumber\r
   )\r
 {\r
-  EFI_STATUS Status;\r
-\r
-  Status = mPiPcd->GetNextToken (Guid, &TokenNumber);\r
+  EFI_STATUS    Status;\r
 \r
-  ASSERT_EFI_ERROR (Status);\r
+  Status = GetPiPcdProtocol()->GetNextToken (Guid, &TokenNumber);\r
+  ASSERT (!EFI_ERROR (Status) || TokenNumber == 0);\r
 \r
   return TokenNumber;\r
 }\r
@@ -992,7 +1020,7 @@ LibPcdGetNextToken (
   If TokenSpaceGuid is NULL, then a pointer to the first PCD token spaces returned.\r
   If TokenSpaceGuid is the last PCD token space GUID in the list, then NULL is returned.\r
   \r
-  @param  TokenSpaceGuid  Pointer to the a PCD token space GUID\r
+  @param  TokenSpaceGuid  The pointer to the a PCD token space GUID.\r
 \r
   @return The next valid token namespace.\r
 \r
@@ -1003,13 +1031,9 @@ LibPcdGetNextTokenSpace (
   IN CONST GUID  *TokenSpaceGuid\r
   )\r
 {\r
-  EFI_STATUS Status;\r
-\r
-  Status = mPiPcd->GetNextTokenSpace (&TokenSpaceGuid);\r
-\r
-  ASSERT_EFI_ERROR (Status);\r
+  GetPiPcdProtocol()->GetNextTokenSpace (&TokenSpaceGuid);\r
 \r
-  return (GUID *) TokenSpaceGuid;\r
+  return (GUID *)TokenSpaceGuid;\r
 }\r
 \r
 \r
@@ -1063,5 +1087,78 @@ LibPatchPcdSetPtr (
   return (VOID *) Buffer;\r
 }\r
 \r
+/**\r
+  Retrieve additional information associated with a PCD token.\r
+\r
+  This includes information such as the type of value the TokenNumber is associated with as well as possible\r
+  human readable name that is associated with the token.\r
+\r
+  If TokenNumber is not in the default token space specified, then ASSERT().\r
+\r
+  @param[in]    TokenNumber The PCD token number.\r
+  @param[out]   PcdInfo     The returned information associated with the requested TokenNumber.\r
+                            The caller is responsible for freeing the buffer that is allocated by callee for PcdInfo->PcdName.\r
+**/\r
+VOID\r
+EFIAPI\r
+LibPcdGetInfo (\r
+  IN        UINTN           TokenNumber,\r
+  OUT       PCD_INFO        *PcdInfo\r
+  )\r
+{\r
+  EFI_STATUS Status;\r
+\r
+  Status = GetPcdInfoProtocolPointer()->GetInfo (TokenNumber, (EFI_PCD_INFO *) PcdInfo);\r
+  ASSERT_EFI_ERROR (Status);\r
+}\r
+\r
+/**\r
+  Retrieve additional information associated with a PCD token.\r
+\r
+  This includes information such as the type of value the TokenNumber is associated with as well as possible\r
+  human readable name that is associated with the token.\r
+\r
+  If TokenNumber is not in the token space specified by Guid, then ASSERT().\r
 \r
+  @param[in]    Guid        The 128-bit unique value that designates the namespace from which to extract the value.\r
+  @param[in]    TokenNumber The PCD token number.\r
+  @param[out]   PcdInfo     The returned information associated with the requested TokenNumber.\r
+                            The caller is responsible for freeing the buffer that is allocated by callee for PcdInfo->PcdName.\r
+**/\r
+VOID\r
+EFIAPI\r
+LibPcdGetInfoEx (\r
+  IN CONST  GUID            *Guid,\r
+  IN        UINTN           TokenNumber,\r
+  OUT       PCD_INFO        *PcdInfo\r
+  )\r
+{\r
+  EFI_STATUS Status;\r
+\r
+  Status = GetPiPcdInfoProtocolPointer()->GetInfo (Guid, TokenNumber, (EFI_PCD_INFO *) PcdInfo);\r
+  ASSERT_EFI_ERROR (Status);\r
+}\r
+\r
+/**\r
+  Retrieve the currently set SKU Id.\r
+\r
+  If the sku id got >= PCD_MAX_SKU_ID, then ASSERT().\r
+\r
+  @return   The currently set SKU Id. If the platform has not set at a SKU Id, then the\r
+            default SKU Id value of 0 is returned. If the platform has set a SKU Id, then the currently set SKU\r
+            Id is returned.\r
+**/\r
+UINTN\r
+EFIAPI\r
+LibPcdGetSku (\r
+  VOID\r
+  )\r
+{\r
+  UINTN SkuId;\r
+\r
+  SkuId = GetPiPcdInfoProtocolPointer()->GetSku ();\r
+  ASSERT (SkuId < PCD_MAX_SKU_ID);\r
+\r
+  return SkuId;\r
+}\r
 \r