]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/PCD/Pei/Service.c
MdePkg and MdeModulePkg Pcd: Implement PCD Driver for External PCD Database and SKU...
[mirror_edk2.git] / MdeModulePkg / Universal / PCD / Pei / Service.c
index 16a5cc848928c48f91a51fbcbf39e6b5f205416d..47e87d3bcddba3265c67458da167a7a3a2ced783 100644 (file)
@@ -1,8 +1,9 @@
 /** @file\r
-Private functions used by PCD PEIM.\r
+  The driver internal functions are implmented here.\r
+  They build Pei PCD database, and provide access service to PCD database.\r
 \r
-Copyright (c) 2006, Intel Corporation\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
@@ -10,9 +11,6 @@ http://opensource.org/licenses/bsd-license.php
 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
-\r
-Module Name: Service.c\r
-\r
 **/\r
 \r
 #include "Service.h"\r
@@ -21,7 +19,7 @@ Module Name: Service.c
   The function registers the CallBackOnSet fucntion\r
   according to TokenNumber and EFI_GUID space.\r
 \r
-  @param  TokenNumber       The token number.\r
+  @param  ExTokenNumber       The token number.\r
   @param  Guid              The GUID space.\r
   @param  CallBackFunction  The Callback function to be registered.\r
   @param  Register          To register or unregister the callback function.\r
@@ -30,7 +28,8 @@ Module Name: Service.c
   @retval EFI_NOT_FOUND If the PCD Entry is not found according to Token Number and GUID space.\r
   @retval EFI_OUT_OF_RESOURCES If the callback function can't be registered because there is not free\r
                                 slot left in the CallbackFnTable.\r
---*/\r
+  @retval EFI_INVALID_PARAMETER If the callback function want to be de-registered can not be found.\r
+**/\r
 EFI_STATUS\r
 PeiRegisterCallBackWorker (\r
   IN  UINTN                       ExTokenNumber,\r
@@ -44,22 +43,30 @@ PeiRegisterCallBackWorker (
   PCD_PPI_CALLBACK        Compare;\r
   PCD_PPI_CALLBACK        Assign;\r
   UINT32                  LocalTokenNumber;\r
+  UINT32                  LocalTokenCount;\r
+  UINTN                   PeiNexTokenNumber;\r
   UINTN                   TokenNumber;\r
   UINTN                   Idx;\r
+  PEI_PCD_DATABASE        *PeiPcdDb;\r
+\r
+  PeiPcdDb          = GetPcdDatabase();\r
+  LocalTokenCount   = PeiPcdDb->LocalTokenCount;\r
+  PeiNexTokenNumber = PeiPcdDb->LocalTokenCount - PeiPcdDb->ExTokenCount;\r
 \r
   if (Guid == NULL) {\r
     TokenNumber = ExTokenNumber;\r
-\r
     //\r
     // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.\r
     // We have to decrement TokenNumber by 1 to make it usable\r
     // as the array index.\r
     //\r
     TokenNumber--;\r
-    ASSERT (TokenNumber + 1 < PEI_NEX_TOKEN_NUMBER + 1);\r
+    ASSERT (TokenNumber + 1 < (PeiNexTokenNumber + 1));\r
   } else {\r
     TokenNumber = GetExPcdTokenNumber (Guid, ExTokenNumber);\r
-\r
+    if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
+      return EFI_NOT_FOUND;\r
+    }\r
     //\r
     // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.\r
     // We have to decrement TokenNumber by 1 to make it usable\r
@@ -69,11 +76,11 @@ PeiRegisterCallBackWorker (
     // EBC compiler is very choosy. It may report warning about comparison\r
     // between UINTN and 0 . So we add 1 in each size of the \r
     // comparison.\r
-    ASSERT (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+    ASSERT ((TokenNumber + 1) < (LocalTokenCount + 1));\r
   }\r
 \r
 \r
-  LocalTokenNumber = GetPcdDatabase()->Init.LocalTokenNumberTable[TokenNumber];\r
+  LocalTokenNumber = *((UINT32 *)((UINT8 *)PeiPcdDb + PeiPcdDb->LocalTokenNumberTableOffset) + TokenNumber);\r
 \r
   //\r
   // We don't support SET for HII and VPD type PCD entry in PEI phase.\r
@@ -82,67 +89,101 @@ PeiRegisterCallBackWorker (
   ASSERT ((LocalTokenNumber & PCD_TYPE_HII) == 0);\r
   ASSERT ((LocalTokenNumber & PCD_TYPE_VPD) == 0);\r
 \r
-  GuidHob = GetFirstGuidHob (&gPcdPeiCallbackFnTableHobGuid);\r
+  GuidHob = GetFirstGuidHob (&gEfiCallerIdGuid);\r
   ASSERT (GuidHob != NULL);\r
   \r
   CallbackTable = GET_GUID_HOB_DATA (GuidHob);\r
-  CallbackTable = CallbackTable + (TokenNumber * FixedPcdGet32(PcdMaxPeiPcdCallBackNumberPerPcdEntry));\r
+  CallbackTable = CallbackTable + (TokenNumber * PcdGet32 (PcdMaxPeiPcdCallBackNumberPerPcdEntry));\r
 \r
   Compare = Register? NULL: CallBackFunction;\r
   Assign  = Register? CallBackFunction: NULL;\r
 \r
 \r
-  for (Idx = 0; Idx < FixedPcdGet32(PcdMaxPeiPcdCallBackNumberPerPcdEntry); Idx++) {\r
+  for (Idx = 0; Idx < PcdGet32 (PcdMaxPeiPcdCallBackNumberPerPcdEntry); Idx++) {\r
     if (CallbackTable[Idx] == Compare) {\r
       CallbackTable[Idx] = Assign;\r
       return EFI_SUCCESS;\r
     }\r
   }\r
 \r
-  return Register? EFI_OUT_OF_RESOURCES : EFI_NOT_FOUND;\r
+  return Register? EFI_OUT_OF_RESOURCES : EFI_INVALID_PARAMETER;\r
 \r
 }\r
 \r
 \r
+/**\r
+  Find the Pcd database. \r
+\r
+  @param  FileHandle  Handle of the file the external PCD database binary located.\r
+\r
+  @retval The base address of external PCD database binary.\r
+  @retval NULL         Return NULL if not find.\r
+**/\r
+VOID *\r
+LocateExPcdBinary (\r
+  IN EFI_PEI_FILE_HANDLE    FileHandle\r
+  )\r
+{\r
+  EFI_STATUS            Status;\r
+  VOID                  *PcdDb;\r
+\r
+  PcdDb       = NULL;\r
+\r
+  ASSERT (FileHandle != NULL);\r
+\r
+  Status = PeiServicesFfsFindSectionData (EFI_SECTION_RAW, FileHandle, &PcdDb);\r
+  ASSERT_EFI_ERROR (Status);\r
+\r
+  //\r
+  // Check the first bytes (Header Signature Guid) and build version.\r
+  //\r
+  if (!CompareGuid (PcdDb, &gPcdDataBaseSignatureGuid) ||\r
+      (((PEI_PCD_DATABASE *) PcdDb)->BuildVersion != PCD_SERVICE_PEIM_VERSION)) {\r
+    ASSERT (FALSE);\r
+  }\r
+  return PcdDb;\r
+}\r
 \r
 \r
 /**\r
   The function builds the PCD database.\r
 \r
-  @param VOID\r
+  @param  FileHandle  Handle of the file the external PCD database binary located.\r
 \r
-  @retval VOID\r
---*/\r
+**/\r
 VOID\r
 BuildPcdDatabase (\r
-  VOID\r
+  IN EFI_PEI_FILE_HANDLE    FileHandle\r
   )\r
 {\r
-  PEI_PCD_DATABASE  *Database;\r
-  VOID              *CallbackFnTable;\r
-  UINTN             SizeOfCallbackFnTable;\r
-  \r
-  Database = BuildGuidHob (&gPcdDataBaseHobGuid, sizeof (PEI_PCD_DATABASE));\r
+  PEI_PCD_DATABASE       *Database;\r
+  PEI_PCD_DATABASE       *PeiPcdDbBinary;\r
+  VOID                   *CallbackFnTable;\r
+  UINTN                  SizeOfCallbackFnTable;\r
+\r
+  //\r
+  // Locate the external PCD database binary for one section of current FFS\r
+  //\r
+  PeiPcdDbBinary = LocateExPcdBinary (FileHandle);\r
+\r
+  ASSERT(PeiPcdDbBinary != NULL);\r
 \r
-  ZeroMem (Database, sizeof (PEI_PCD_DATABASE));\r
+  Database = BuildGuidHob (&gPcdDataBaseHobGuid, PeiPcdDbBinary->Length + PeiPcdDbBinary->UninitDataBaseSize);\r
+\r
+  ZeroMem (Database, PeiPcdDbBinary->Length  + PeiPcdDbBinary->UninitDataBaseSize);\r
 \r
   //\r
-  // gPEIPcdDbInit is smaller than PEI_PCD_DATABASE\r
+  // PeiPcdDbBinary is smaller than Database\r
   //\r
-  \r
-  CopyMem (&Database->Init, &gPEIPcdDbInit, sizeof (gPEIPcdDbInit));\r
+  CopyMem (Database, PeiPcdDbBinary, PeiPcdDbBinary->Length);\r
 \r
-  SizeOfCallbackFnTable = PEI_LOCAL_TOKEN_NUMBER * sizeof (PCD_PPI_CALLBACK) * FixedPcdGet32(PcdMaxPeiPcdCallBackNumberPerPcdEntry);\r
+  SizeOfCallbackFnTable = Database->LocalTokenCount * sizeof (PCD_PPI_CALLBACK) * PcdGet32 (PcdMaxPeiPcdCallBackNumberPerPcdEntry);\r
 \r
-  CallbackFnTable = BuildGuidHob (&gPcdPeiCallbackFnTableHobGuid, SizeOfCallbackFnTable);\r
+  CallbackFnTable = BuildGuidHob (&gEfiCallerIdGuid, SizeOfCallbackFnTable);\r
   \r
   ZeroMem (CallbackFnTable, SizeOfCallbackFnTable);\r
-  \r
-  return;\r
 }\r
 \r
-\r
-\r
 /**\r
   The function is provided by PCD PEIM and PCD DXE driver to\r
   do the work of reading a HII variable from variable service.\r
@@ -154,8 +195,7 @@ BuildPcdDatabase (
 \r
   @retval EFI_SUCCESS         Operation successful.\r
   @retval EFI_NOT_FOUND         Variablel not found.\r
---*/\r
-STATIC\r
+**/\r
 EFI_STATUS\r
 GetHiiVariable (\r
   IN  CONST EFI_GUID      *VariableGuid,\r
@@ -181,9 +221,8 @@ GetHiiVariable (
                           &Size,\r
                           NULL\r
                           );\r
-  if (Status == EFI_BUFFER_TOO_SMALL) {\r
-\r
 \r
+  if (Status == EFI_BUFFER_TOO_SMALL) {\r
     Status = PeiServicesAllocatePool (Size, &Buffer);\r
     ASSERT_EFI_ERROR (Status);\r
 \r
@@ -201,13 +240,20 @@ GetHiiVariable (
     *VariableData = Buffer;\r
 \r
     return EFI_SUCCESS;\r
-  } else {\r
-    return EFI_NOT_FOUND;\r
   }\r
 \r
+  return EFI_NOT_FOUND;\r
 }\r
 \r
-STATIC\r
+/**\r
+  Find the local token number according to system SKU ID.\r
+\r
+  @param LocalTokenNumber PCD token number\r
+  @param Size             The size of PCD entry.\r
+\r
+  @return Token number according to system SKU ID.\r
+\r
+**/\r
 UINT32\r
 GetSkuEnabledTokenNumber (\r
   UINT32 LocalTokenNumber,\r
@@ -217,8 +263,9 @@ GetSkuEnabledTokenNumber (
   PEI_PCD_DATABASE      *PeiPcdDb;\r
   SKU_HEAD              *SkuHead;\r
   SKU_ID                *SkuIdTable;\r
-  INTN                  i;\r
+  INTN                  Index;\r
   UINT8                 *Value;\r
+  BOOLEAN               FoundSku;\r
 \r
   PeiPcdDb = GetPcdDatabase ();\r
 \r
@@ -227,29 +274,50 @@ GetSkuEnabledTokenNumber (
   SkuHead     = (SKU_HEAD *) ((UINT8 *)PeiPcdDb + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));\r
   Value       = (UINT8 *) ((UINT8 *)PeiPcdDb + (SkuHead->SkuDataStartOffset));\r
   SkuIdTable  = (SKU_ID *) ((UINT8 *)PeiPcdDb + (SkuHead->SkuIdTableOffset));\r
-        \r
-  for (i = 0; i < SkuIdTable[0]; i++) {\r
-    if (PeiPcdDb->Init.SystemSkuId == SkuIdTable[i + 1]) {\r
+\r
+  //\r
+  // Find the current system's SKU ID entry in SKU ID table.\r
+  //\r
+  FoundSku = FALSE;\r
+  for (Index = 0; Index < SkuIdTable[0]; Index++) {\r
+    if (PeiPcdDb->SystemSkuId == SkuIdTable[Index + 1]) {\r
+      FoundSku = TRUE;\r
       break;\r
     }\r
   }\r
 \r
+  //\r
+  // Find the default SKU ID entry in SKU ID table.\r
+  //\r
+  if(!FoundSku) {\r
+    for (Index = 0; Index < SkuIdTable[0]; Index++) {\r
+      if (0 == SkuIdTable[Index + 1]) {\r
+        break;\r
+      }\r
+    }\r
+  }\r
+  ASSERT (Index < SkuIdTable[0]);\r
+\r
   switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
     case PCD_TYPE_VPD:\r
-      Value = (UINT8 *) &(((VPD_HEAD *) Value)[i]);\r
+      Value = (UINT8 *) &(((VPD_HEAD *) Value)[Index]);\r
       return (UINT32) ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_VPD);\r
 \r
     case PCD_TYPE_HII:\r
-      Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[i]);\r
+      Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[Index]);\r
       return (UINT32) ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_HII);\r
-      \r
+\r
+    case PCD_TYPE_HII|PCD_TYPE_STRING:\r
+      Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[Index]);\r
+      return (UINT32) ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_HII | PCD_TYPE_STRING);\r
+\r
     case PCD_TYPE_STRING:\r
-      Value = (UINT8 *) &(((STRING_HEAD *) Value)[i]);\r
+      Value = (UINT8 *) &(((STRING_HEAD *) Value)[Index]);\r
       return (UINT32) ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_STRING);\r
 \r
     case PCD_TYPE_DATA:\r
-      Value += Size * i;\r
-      return (UINT32) (Value - (UINT8 *) PeiPcdDb);\r
+      Value += Size * Index;\r
+      return (UINT32) ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_DATA);\r
 \r
     default:\r
       ASSERT (FALSE);\r
@@ -258,12 +326,21 @@ GetSkuEnabledTokenNumber (
   ASSERT (FALSE);\r
 \r
   return 0;\r
-  \r
 }\r
 \r
+/**\r
+  Invoke the callback function when dynamic PCD entry was set, if this PCD entry \r
+  has registered callback function.\r
 \r
+  @param ExTokenNumber   DynamicEx PCD's token number, if this PCD entry is dyanmicEx\r
+                         type PCD.\r
+  @param Guid            DynamicEx PCD's guid, if this PCD entry is dynamicEx type\r
+                         PCD.\r
+  @param TokenNumber     PCD token number generated by build tools.\r
+  @param Data            Value want to be set for this PCD entry\r
+  @param Size            The size of value\r
 \r
-STATIC\r
+**/\r
 VOID\r
 InvokeCallbackOnSet (\r
   UINTN             ExTokenNumber,\r
@@ -276,6 +353,8 @@ InvokeCallbackOnSet (
   EFI_HOB_GUID_TYPE   *GuidHob;\r
   PCD_PPI_CALLBACK    *CallbackTable;\r
   UINTN               Idx;\r
+  PEI_PCD_DATABASE    *PeiPcdDb;\r
+  UINT32              LocalTokenCount;\r
 \r
   //\r
   // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.\r
@@ -283,35 +362,45 @@ InvokeCallbackOnSet (
   // as the array index.\r
   //\r
   TokenNumber--;\r
-  \r
+\r
+  PeiPcdDb        = GetPcdDatabase ();\r
+  LocalTokenCount = PeiPcdDb->LocalTokenCount;\r
+\r
   if (Guid == NULL) {\r
     // EBC compiler is very choosy. It may report warning about comparison\r
     // between UINTN and 0 . So we add 1 in each size of the \r
     // comparison.\r
-    ASSERT (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+    ASSERT (TokenNumber + 1 < (LocalTokenCount + 1));\r
   }\r
 \r
-  GuidHob = GetFirstGuidHob (&gPcdPeiCallbackFnTableHobGuid);\r
+  GuidHob = GetFirstGuidHob (&gEfiCallerIdGuid);\r
   ASSERT (GuidHob != NULL);\r
   \r
   CallbackTable = GET_GUID_HOB_DATA (GuidHob);\r
 \r
-  CallbackTable += (TokenNumber * FixedPcdGet32(PcdMaxPeiPcdCallBackNumberPerPcdEntry));\r
+  CallbackTable += (TokenNumber * PcdGet32 (PcdMaxPeiPcdCallBackNumberPerPcdEntry));\r
 \r
-  for (Idx = 0; Idx < FixedPcdGet32(PcdMaxPeiPcdCallBackNumberPerPcdEntry); Idx++) {\r
+  for (Idx = 0; Idx < PcdGet32 (PcdMaxPeiPcdCallBackNumberPerPcdEntry); Idx++) {\r
     if (CallbackTable[Idx] != NULL) {\r
       CallbackTable[Idx] (Guid,\r
-                          (Guid == NULL)? TokenNumber: ExTokenNumber,\r
+                          (Guid == NULL) ? (TokenNumber + 1) : ExTokenNumber,\r
                           Data,\r
                           Size\r
                           );\r
     }\r
   }\r
-  \r
 }\r
 \r
+/**\r
+  Wrapper function for setting non-pointer type value for a PCD entry.\r
+\r
+  @param TokenNumber     Pcd token number autogenerated by build tools.\r
+  @param Data            Value want to be set for PCD entry\r
+  @param Size            Size of value.\r
 \r
+  @return status of SetWorker.\r
 \r
+**/\r
 EFI_STATUS\r
 SetValueWorker (\r
   IN          UINTN              TokenNumber,\r
@@ -322,24 +411,40 @@ SetValueWorker (
   return SetWorker (TokenNumber, Data, &Size, FALSE);\r
 }\r
 \r
-\r
-\r
+/**\r
+  Set value for an PCD entry\r
+\r
+  @param TokenNumber     Pcd token number autogenerated by build tools.\r
+  @param Data            Value want to be set for PCD entry\r
+  @param Size            Size of value.\r
+  @param PtrType         If TRUE, the type of PCD entry's value is Pointer.\r
+                         If False, the type of PCD entry's value is not Pointer.\r
+\r
+  @retval EFI_INVALID_PARAMETER  If this PCD type is VPD, VPD PCD can not be set.\r
+  @retval EFI_INVALID_PARAMETER  If Size can not be set to size table.\r
+  @retval EFI_INVALID_PARAMETER  If Size of non-Ptr type PCD does not match the size information in PCD database.\r
+  @retval EFI_NOT_FOUND          If value type of PCD entry is intergrate, but not in\r
+                                 range of UINT8, UINT16, UINT32, UINT64\r
+  @retval EFI_NOT_FOUND          Can not find the PCD type according to token number.                                \r
+**/\r
 EFI_STATUS\r
 SetWorker (\r
   IN          UINTN               TokenNumber,\r
-  IN OUT      VOID                *Data,\r
+  IN          VOID                *Data,\r
   IN OUT      UINTN               *Size,\r
   IN          BOOLEAN             PtrType\r
   )\r
 {\r
   UINT32              LocalTokenNumber;\r
+  UINTN               PeiNexTokenNumber;\r
   PEI_PCD_DATABASE    *PeiPcdDb;\r
-  UINT16              StringTableIdx;\r
+  STRING_HEAD         StringTableIdx;\r
   UINTN               Offset;\r
   VOID                *InternalData;\r
   UINTN               MaxSize;\r
+  UINT32              LocalTokenCount;\r
 \r
-  if (!FeaturePcdGet(PcdPeiPcdDatabaseSetEnabled)) {\r
+  if (!FeaturePcdGet(PcdPeiFullPcdDatabaseEnable)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
   \r
@@ -349,18 +454,29 @@ SetWorker (
   // as the array index.\r
   //\r
   TokenNumber--;\r
+  PeiPcdDb        = GetPcdDatabase ();\r
+  LocalTokenCount = PeiPcdDb->LocalTokenCount;\r
 \r
   // EBC compiler is very choosy. It may report warning about comparison\r
   // between UINTN and 0 . So we add 1 in each size of the \r
   // comparison.\r
-  ASSERT (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
-    \r
-  PeiPcdDb = GetPcdDatabase ();\r
+  ASSERT (TokenNumber + 1 < (LocalTokenCount + 1));\r
 \r
-  LocalTokenNumber = PeiPcdDb->Init.LocalTokenNumberTable[TokenNumber];\r
+  LocalTokenNumber = *((UINT32 *)((UINT8 *)PeiPcdDb + PeiPcdDb->LocalTokenNumberTableOffset) + TokenNumber);\r
 \r
-  if (!PtrType) {\r
-    ASSERT (PeiPcdGetSize(TokenNumber + 1) == *Size);\r
+  if (PtrType) {\r
+    //\r
+    // Get MaxSize first, then check new size with max buffer size.\r
+    //\r
+    GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
+    if (*Size > MaxSize) {\r
+      *Size = MaxSize;\r
+      return EFI_INVALID_PARAMETER;\r
+    }\r
+  } else {\r
+    if (*Size != PeiPcdGetSize (TokenNumber + 1)) {\r
+      return EFI_INVALID_PARAMETER;\r
+    }\r
   }\r
 \r
   //\r
@@ -368,13 +484,14 @@ SetWorker (
   // For Dynamic EX PCD entry, we have invoked the callback function for Dynamic EX\r
   // type PCD entry in ExSetWorker.\r
   //\r
-  if (TokenNumber + 1 < PEI_NEX_TOKEN_NUMBER + 1) {\r
+  PeiNexTokenNumber = PeiPcdDb->LocalTokenCount - PeiPcdDb->ExTokenCount;\r
+  if (TokenNumber + 1 < PeiNexTokenNumber + 1) {\r
     InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, *Size);\r
   }\r
 \r
   if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
     if (PtrType) {\r
-      MaxSize = GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
+      GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
     } else {\r
       MaxSize = *Size;\r
     }\r
@@ -387,6 +504,7 @@ SetWorker (
   switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
     case PCD_TYPE_VPD:\r
     case PCD_TYPE_HII:\r
+    case PCD_TYPE_HII|PCD_TYPE_STRING:\r
     {\r
       ASSERT (FALSE);\r
       return EFI_INVALID_PARAMETER;\r
@@ -394,8 +512,8 @@ SetWorker (
 \r
     case PCD_TYPE_STRING:\r
       if (SetPtrTypeSize (TokenNumber, Size, PeiPcdDb)) {\r
-        StringTableIdx = *((UINT16 *)InternalData);\r
-        CopyMem (&PeiPcdDb->Init.StringTable[StringTableIdx], Data, *Size);\r
+        StringTableIdx = *((STRING_HEAD *)InternalData);\r
+        CopyMem ((UINT8 *)PeiPcdDb + PeiPcdDb->StringTableOffset + StringTableIdx, Data, *Size);\r
         return EFI_SUCCESS;\r
       } else {\r
         return EFI_INVALID_PARAMETER;\r
@@ -442,8 +560,17 @@ SetWorker (
 \r
 }\r
 \r
+/**\r
+  Wrapper function for set PCD value for non-Pointer type dynamic-ex PCD.\r
+\r
+  @param ExTokenNumber   Token number for dynamic-ex PCD.\r
+  @param Guid            Token space guid for dynamic-ex PCD.\r
+  @param Data            Value want to be set.\r
+  @param SetSize         The size of value.\r
 \r
+  @return status of ExSetWorker().\r
 \r
+**/\r
 EFI_STATUS\r
 ExSetValueWorker (\r
   IN          UINTN                ExTokenNumber,\r
@@ -455,8 +582,24 @@ ExSetValueWorker (
   return ExSetWorker (ExTokenNumber, Guid, Data, &Size, FALSE);\r
 }\r
 \r
+/**\r
+  Set value for a dynamic-ex PCD entry.\r
+  \r
+  This routine find the local token number according to dynamic-ex PCD's token \r
+  space guid and token number firstly, and invoke callback function if this PCD\r
+  entry registered callback function. Finally, invoken general SetWorker to set\r
+  PCD value.\r
+  \r
+  @param ExTokenNumber   Dynamic-ex PCD token number.\r
+  @param Guid            Token space guid for dynamic-ex PCD.\r
+  @param Data            PCD value want to be set\r
+  @param SetSize         Size of value.\r
+  @param PtrType         If TRUE, this PCD entry is pointer type.\r
+                         If FALSE, this PCD entry is not pointer type.\r
 \r
+  @return status of SetWorker().\r
 \r
+**/\r
 EFI_STATUS\r
 ExSetWorker (\r
   IN            UINTN                ExTokenNumber,\r
@@ -468,43 +611,59 @@ ExSetWorker (
 {\r
   UINTN                     TokenNumber;\r
 \r
-  if (!FeaturePcdGet(PcdPeiPcdDatabaseSetEnabled)) {\r
+  if (!FeaturePcdGet(PcdPeiFullPcdDatabaseEnable)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
   TokenNumber = GetExPcdTokenNumber (Guid, ExTokenNumber);\r
-\r
+  if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+  \r
   InvokeCallbackOnSet (ExTokenNumber, Guid, TokenNumber, Data, *Size);\r
 \r
   return SetWorker (TokenNumber, Data, Size, PtrType);\r
 \r
 }\r
 \r
+/**\r
+  Wrapper function for get PCD value for dynamic-ex PCD.\r
 \r
+  @param Guid            Token space guid for dynamic-ex PCD.\r
+  @param ExTokenNumber   Token number for dyanmic-ex PCD.\r
+  @param GetSize         The size of dynamic-ex PCD value.\r
 \r
+  @return PCD entry in PCD database.\r
 \r
+**/\r
 VOID *\r
 ExGetWorker (\r
   IN CONST  EFI_GUID  *Guid,\r
   IN UINTN            ExTokenNumber,\r
   IN UINTN            GetSize\r
   )\r
-{\r
-  if (!FeaturePcdGet (PcdPeiPcdDatabaseExEnabled)) {\r
-    ASSERT (FALSE);\r
-    return 0;\r
-  }\r
-  \r
+{ \r
   return GetWorker (GetExPcdTokenNumber (Guid, ExTokenNumber), GetSize);\r
 }\r
 \r
+/**\r
+  Get the PCD entry pointer in PCD database.\r
+  \r
+  This routine will visit PCD database to find the PCD entry according to given\r
+  token number. The given token number is autogened by build tools and it will be \r
+  translated to local token number. Local token number contains PCD's type and \r
+  offset of PCD entry in PCD database.\r
 \r
+  @param TokenNumber     Token's number, it is autogened by build tools\r
+  @param GetSize         The size of token's value\r
 \r
+  @return PCD entry pointer in PCD database\r
 \r
+**/\r
 VOID *\r
 GetWorker (\r
-  UINTN               TokenNumber,\r
-  UINTN               GetSize\r
+  IN UINTN               TokenNumber,\r
+  IN UINTN               GetSize\r
   )\r
 {\r
   UINT32              Offset;\r
@@ -514,11 +673,12 @@ GetWorker (
   EFI_STATUS          Status;\r
   UINTN               DataSize;\r
   VOID                *Data;\r
-  UINT16              *StringTable;\r
-  UINT16              StringTableIdx;\r
+  UINT              *StringTable;\r
+  STRING_HEAD         StringTableIdx;\r
   PEI_PCD_DATABASE    *PeiPcdDb;\r
   UINT32              LocalTokenNumber;\r
   UINTN               MaxSize;\r
+  UINT32              LocalTokenCount;\r
 \r
   //\r
   // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.\r
@@ -527,20 +687,21 @@ GetWorker (
   //\r
   TokenNumber--;\r
 \r
+  PeiPcdDb        = GetPcdDatabase ();\r
+  LocalTokenCount = PeiPcdDb->LocalTokenCount;\r
+\r
   // EBC compiler is very choosy. It may report warning about comparison\r
   // between UINTN and 0 . So we add 1 in each size of the \r
   // comparison.\r
-  ASSERT (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+  ASSERT (TokenNumber + 1 < (LocalTokenCount + 1));\r
 \r
   ASSERT ((GetSize == PeiPcdGetSize(TokenNumber + 1)) || (GetSize == 0));\r
 \r
-  PeiPcdDb        = GetPcdDatabase ();\r
-\r
-  LocalTokenNumber = PeiPcdDb->Init.LocalTokenNumberTable[TokenNumber];\r
+  LocalTokenNumber = *((UINT32 *)((UINT8 *)PeiPcdDb + PeiPcdDb->LocalTokenNumberTableOffset) + TokenNumber);\r
 \r
   if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
     if (GetSize == 0) {\r
-      MaxSize = GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
+      GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
     } else {\r
       MaxSize = GetSize;\r
     }\r
@@ -548,22 +709,23 @@ GetWorker (
   }\r
 \r
   Offset      = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
-  StringTable = PeiPcdDb->Init.StringTable;\r
-  \r
+  StringTable = (UINT8 *)PeiPcdDb + PeiPcdDb->StringTableOffset;\r
+\r
   switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
     case PCD_TYPE_VPD:\r
     {\r
       VPD_HEAD *VpdHead;\r
       VpdHead = (VPD_HEAD *) ((UINT8 *)PeiPcdDb + Offset);\r
-      return (VOID *) (UINTN) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
+      return (VOID *) (UINTN) (PcdGet32 (PcdVpdBaseAddress) + VpdHead->Offset);\r
     }\r
       \r
+    case PCD_TYPE_HII|PCD_TYPE_STRING:\r
     case PCD_TYPE_HII:\r
     {\r
       VariableHead = (VARIABLE_HEAD *) ((UINT8 *)PeiPcdDb + Offset);\r
       \r
-      Guid = &(PeiPcdDb->Init.GuidTable[VariableHead->GuidTableIndex]);\r
-      Name = &StringTable[VariableHead->StringIndex];\r
+      Guid = (EFI_GUID *) ((UINT8 *)PeiPcdDb + PeiPcdDb->GuidTableOffset) + VariableHead->GuidTableIndex;\r
+      Name = (UINT16*)&StringTable[VariableHead->StringIndex];\r
 \r
       Status = GetHiiVariable (Guid, Name, &Data, &DataSize);\r
 \r
@@ -573,7 +735,11 @@ GetWorker (
         //\r
         // Return the default value specified by Platform Integrator \r
         //\r
-        return (VOID *) ((UINT8 *) PeiPcdDb + VariableHead->DefaultValueOffset);\r
+        if ((LocalTokenNumber & PCD_TYPE_ALL_SET) == (PCD_TYPE_HII|PCD_TYPE_STRING)) {\r
+          return (VOID*)&StringTable[*(STRING_HEAD*)((UINT8*)PeiPcdDb + VariableHead->DefaultValueOffset)];\r
+        } else {\r
+          return (VOID *) ((UINT8 *) PeiPcdDb + VariableHead->DefaultValueOffset);\r
+        }\r
       }\r
     }\r
 \r
@@ -581,7 +747,7 @@ GetWorker (
       return (VOID *) ((UINT8 *)PeiPcdDb + Offset);\r
 \r
     case PCD_TYPE_STRING:\r
-      StringTableIdx = (UINT16) *((UINT8 *) PeiPcdDb + Offset);\r
+      StringTableIdx = * (STRING_HEAD*) ((UINT8 *) PeiPcdDb + Offset);\r
       return (VOID *) (&StringTable[StringTableIdx]);\r
 \r
     default:\r
@@ -596,15 +762,26 @@ GetWorker (
   \r
 }\r
 \r
+/**\r
+  Get Token Number according to dynamic-ex PCD's {token space guid:token number}\r
+\r
+  A dynamic-ex type PCD, developer must provide pair of token space guid: token number\r
+  in DEC file. PCD database maintain a mapping table that translate pair of {token\r
+  space guid: token number} to Token Number.\r
+  \r
+  @param Guid            Token space guid for dynamic-ex PCD entry.\r
+  @param ExTokenNumber   Dynamic-ex PCD token number.\r
 \r
+  @return Token Number for dynamic-ex PCD.\r
 \r
+**/\r
 UINTN           \r
 GetExPcdTokenNumber (\r
   IN CONST EFI_GUID             *Guid,\r
   IN UINTN                      ExTokenNumber\r
   )\r
 {\r
-  UINT32              i;\r
+  UINT32              Index;\r
   DYNAMICEX_MAPPING   *ExMap;\r
   EFI_GUID            *GuidTable;\r
   EFI_GUID            *MatchGuid;\r
@@ -612,11 +789,11 @@ GetExPcdTokenNumber (
   PEI_PCD_DATABASE    *PeiPcdDb;\r
 \r
   PeiPcdDb    = GetPcdDatabase();\r
-  \r
-  ExMap       = PeiPcdDb->Init.ExMapTable;\r
-  GuidTable   = PeiPcdDb->Init.GuidTable;\r
 \r
-  MatchGuid = ScanGuid (GuidTable, sizeof(PeiPcdDb->Init.GuidTable), Guid);\r
+  ExMap       = (DYNAMICEX_MAPPING *)((UINT8 *)PeiPcdDb + PeiPcdDb->ExMapTableOffset);\r
+  GuidTable   = (EFI_GUID *)((UINT8 *)PeiPcdDb + PeiPcdDb->GuidTableOffset);\r
+\r
+  MatchGuid = ScanGuid (GuidTable, PeiPcdDb->GuidTableCount * sizeof(EFI_GUID), Guid);\r
   //\r
   // We need to ASSERT here. If GUID can't be found in GuidTable, this is a\r
   // error in the BUILD system.\r
@@ -625,20 +802,22 @@ GetExPcdTokenNumber (
   \r
   MatchGuidIdx = MatchGuid - GuidTable;\r
   \r
-  for (i = 0; i < PEI_EXMAPPING_TABLE_SIZE; i++) {\r
-    if ((ExTokenNumber == ExMap[i].ExTokenNumber) && \r
-        (MatchGuidIdx == ExMap[i].ExGuidIndex)) {\r
-      return ExMap[i].LocalTokenNumber;\r
+  for (Index = 0; Index < PeiPcdDb->ExTokenCount; Index++) {\r
+    if ((ExTokenNumber == ExMap[Index].ExTokenNumber) && \r
+        (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {\r
+      return ExMap[Index].TokenNumber;\r
     }\r
   }\r
   \r
-  ASSERT (FALSE);\r
-  \r
-  return 0;\r
+  return PCD_INVALID_TOKEN_NUMBER;\r
 }\r
 \r
+/**\r
+  Get PCD database from GUID HOB in PEI phase.\r
 \r
+  @return Pointer to PCD database.\r
 \r
+**/\r
 PEI_PCD_DATABASE *\r
 GetPcdDatabase (\r
   VOID\r
@@ -652,8 +831,15 @@ GetPcdDatabase (
   return (PEI_PCD_DATABASE *) GET_GUID_HOB_DATA (GuidHob);\r
 }\r
 \r
+/**\r
+  Get SKU ID table from PCD database.\r
+\r
+  @param LocalTokenNumberTableIdx Index of local token number in token number table.\r
+  @param Database                 PCD database.\r
 \r
+  @return Pointer to SKU ID array table\r
 \r
+**/\r
 SKU_ID *\r
 GetSkuIdArray (\r
   IN    UINTN             LocalTokenNumberTableIdx,\r
@@ -663,7 +849,7 @@ GetSkuIdArray (
   SKU_HEAD *SkuHead;\r
   UINTN     LocalTokenNumber;\r
 \r
-  LocalTokenNumber = Database->Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+  LocalTokenNumber = *((UINT32 *)((UINT8 *)Database + Database->LocalTokenNumberTableOffset) + LocalTokenNumberTableIdx);\r
 \r
   ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) != 0);\r
 \r
@@ -673,36 +859,44 @@ GetSkuIdArray (
   \r
 }\r
 \r
+/**\r
+  Get index of PCD entry in size table.\r
+\r
+  @param LocalTokenNumberTableIdx Index of this PCD in local token number table.\r
+  @param Database                 Pointer to PCD database in PEI phase.\r
 \r
+  @return index of PCD entry in size table.\r
 \r
+**/\r
 UINTN\r
 GetSizeTableIndex (\r
   IN    UINTN             LocalTokenNumberTableIdx,\r
   IN    PEI_PCD_DATABASE  *Database\r
   )\r
 {\r
-  UINTN       i;\r
-  UINTN        SizeTableIdx;\r
+  UINTN       Index;\r
+  UINTN       SizeTableIdx;\r
   UINTN       LocalTokenNumber;\r
   SKU_ID      *SkuIdTable;\r
   \r
   SizeTableIdx = 0;\r
 \r
-  for (i=0; i<LocalTokenNumberTableIdx; i++) {\r
-    LocalTokenNumber = Database->Init.LocalTokenNumberTable[i];\r
+  for (Index = 0; Index < LocalTokenNumberTableIdx; Index++) {\r
+    LocalTokenNumber = *((UINT32 *)((UINT8 *)Database + Database->LocalTokenNumberTableOffset) + Index);\r
 \r
     if ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER) {\r
       //\r
       // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type \r
       // PCD entry.\r
       //\r
-      if (LocalTokenNumber & PCD_TYPE_VPD) {\r
+      if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {\r
           //\r
-          // We have only one entry for VPD enabled PCD entry:\r
+          // We have only two entry for VPD enabled PCD entry:\r
           // 1) MAX Size.\r
-          // We consider current size is equal to MAX size.\r
+          // 2) Current Size\r
+          // Current size is equal to MAX size.\r
           //\r
-          SizeTableIdx++;\r
+          SizeTableIdx += 2;\r
       } else {\r
         if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
           //\r
@@ -717,7 +911,7 @@ GetSizeTableIndex (
           // 1) MAX SIZE\r
           // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
           //\r
-          SkuIdTable = GetSkuIdArray (i, Database);\r
+          SkuIdTable = GetSkuIdArray (Index, Database);\r
           SizeTableIdx += (UINTN)*SkuIdTable + 1;\r
         }\r
       }\r