]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/PCD/Dxe/Service.c
MdeModulePkg Pcd: Check the input SkuId in SetSku()
[mirror_edk2.git] / MdeModulePkg / Universal / PCD / Dxe / Service.c
index 513402cad91f1b591b44926258ead587ee541586..9b4701bdd74996bbb253948ae97bae84bff3b21d 100644 (file)
@@ -1,7 +1,8 @@
 /** @file\r
     Help functions used by PCD DXE driver.\r
 \r
-Copyright (c) 2006 - 2013, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2014, Hewlett-Packard Development Company, L.P.<BR>\r
+Copyright (c) 2006 - 2015, 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
@@ -35,6 +36,333 @@ LIST_ENTRY    *mCallbackFnTable;
 EFI_GUID     **TmpTokenSpaceBuffer;\r
 UINTN          TmpTokenSpaceBufferCount; \r
 \r
+/**\r
+  Get Local Token Number by Token Number.\r
+\r
+  @param[in]    IsPeiDb     If TRUE, the pcd entry is initialized in PEI phase,\r
+                            If FALSE, the pcd entry is initialized in DXE phase.\r
+  @param[in]    TokenNumber The PCD token number.\r
+\r
+  @return       Local Token Number.\r
+**/\r
+UINT32\r
+GetLocalTokenNumber (\r
+  IN BOOLEAN            IsPeiDb,\r
+  IN UINTN              TokenNumber\r
+  )\r
+{\r
+  UINTN                 TmpTokenNumber;\r
+  UINT32                *LocalTokenNumberTable;\r
+  UINT32                LocalTokenNumber;\r
+  UINTN                 Size;\r
+  UINTN                 MaxSize;\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
+\r
+  //\r
+  // Backup the TokenNumber passed in as GetPtrTypeSize need the original TokenNumber\r
+  // \r
+  TmpTokenNumber = TokenNumber;\r
+\r
+  LocalTokenNumberTable  = IsPeiDb ? (UINT32 *)((UINT8 *)mPcdDatabase.PeiDb + mPcdDatabase.PeiDb->LocalTokenNumberTableOffset) : \r
+                                     (UINT32 *)((UINT8 *)mPcdDatabase.DxeDb + mPcdDatabase.DxeDb->LocalTokenNumberTableOffset);\r
+  TokenNumber            = IsPeiDb ? TokenNumber : TokenNumber - mPeiLocalTokenCount;\r
+\r
+  LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
+\r
+  Size = (LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) >> PCD_DATUM_TYPE_SHIFT;\r
+\r
+  if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
+    if (Size == 0) {\r
+      GetPtrTypeSize (TmpTokenNumber, &MaxSize);\r
+    } else {\r
+      MaxSize = Size;\r
+    }\r
+    LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);\r
+  }\r
+\r
+  return LocalTokenNumber;\r
+}\r
+\r
+/**\r
+  Get PCD type by Local Token Number.\r
+\r
+  @param[in]    LocalTokenNumber The PCD local token number.\r
+\r
+  @return       PCD type.\r
+**/\r
+EFI_PCD_TYPE\r
+GetPcdType (\r
+  IN UINT32             LocalTokenNumber\r
+  )\r
+{\r
+  switch (LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) {\r
+    case PCD_DATUM_TYPE_POINTER:\r
+      return EFI_PCD_TYPE_PTR;\r
+    case PCD_DATUM_TYPE_UINT8:\r
+      if ((LocalTokenNumber & PCD_DATUM_TYPE_UINT8_BOOLEAN) == PCD_DATUM_TYPE_UINT8_BOOLEAN) {\r
+        return EFI_PCD_TYPE_BOOL;\r
+      } else {\r
+        return EFI_PCD_TYPE_8;\r
+      }\r
+    case PCD_DATUM_TYPE_UINT16:\r
+      return EFI_PCD_TYPE_16;\r
+    case PCD_DATUM_TYPE_UINT32:\r
+      return EFI_PCD_TYPE_32;\r
+    case PCD_DATUM_TYPE_UINT64:\r
+      return EFI_PCD_TYPE_64;\r
+    default:\r
+      ASSERT (FALSE);\r
+      return EFI_PCD_TYPE_8;\r
+  }\r
+}\r
+\r
+/**\r
+  Get PCD name.\r
+\r
+  @param[in]    OnlyTokenSpaceName  If TRUE, only need to get the TokenSpaceCName.\r
+                                    If FALSE, need to get the full PCD name.\r
+  @param[in]    IsPeiDb             If TRUE, the pcd entry is initialized in PEI phase,\r
+                                    If FALSE, the pcd entry is initialized in DXE phase.\r
+  @param[in]    TokenNumber         The PCD token number.\r
+\r
+  @return       The TokenSpaceCName or full PCD name.\r
+**/\r
+CHAR8 *\r
+GetPcdName (\r
+  IN BOOLEAN            OnlyTokenSpaceName,\r
+  IN BOOLEAN            IsPeiDb,\r
+  IN UINTN              TokenNumber\r
+  )\r
+{\r
+  PCD_DATABASE_INIT *Database;\r
+  UINT8             *StringTable;\r
+  PCD_NAME_INDEX    *PcdNameIndex;\r
+  CHAR8             *TokenSpaceName;\r
+  CHAR8             *PcdName;\r
+  CHAR8             *Name;\r
+\r
+  //\r
+  // Return NULL when PCD name table is absent. \r
+  //\r
+  if (IsPeiDb) {\r
+    if (mPcdDatabase.PeiDb->PcdNameTableOffset == 0) {\r
+      return NULL;\r
+    }\r
+  } else {\r
+    if (mPcdDatabase.DxeDb->PcdNameTableOffset == 0) {\r
+      return NULL;\r
+    }\r
+  }\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
+\r
+  Database = IsPeiDb ? mPcdDatabase.PeiDb: mPcdDatabase.DxeDb;\r
+  TokenNumber = IsPeiDb ? TokenNumber : TokenNumber - mPeiLocalTokenCount;\r
+\r
+  StringTable = (UINT8 *) Database + Database->StringTableOffset;\r
+\r
+  //\r
+  // Get the PCD name index.\r
+  //\r
+  PcdNameIndex = (PCD_NAME_INDEX *)((UINT8 *) Database + Database->PcdNameTableOffset) + TokenNumber;\r
+  TokenSpaceName = (CHAR8 *)&StringTable[PcdNameIndex->TokenSpaceCNameIndex];\r
+  PcdName = (CHAR8 *)&StringTable[PcdNameIndex->PcdCNameIndex];\r
+\r
+  if (OnlyTokenSpaceName) {\r
+    //\r
+    // Only need to get the TokenSpaceCName.\r
+    //\r
+    Name = AllocateCopyPool (AsciiStrSize (TokenSpaceName), TokenSpaceName);\r
+  } else {\r
+    //\r
+    // Need to get the full PCD name.\r
+    //\r
+    Name = AllocateZeroPool (AsciiStrSize (TokenSpaceName) + AsciiStrSize (PcdName));\r
+    ASSERT (Name != NULL);\r
+    //\r
+    // Catenate TokenSpaceCName and PcdCName with a '.' to form the full PCD name.\r
+    //\r
+    AsciiStrCat (Name, TokenSpaceName);\r
+    Name[AsciiStrSize (TokenSpaceName) - sizeof (CHAR8)] = '.';\r
+    AsciiStrCat (Name, PcdName);  \r
+  }\r
+\r
+  return Name;\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
+  @param[in]    IsPeiDb     If TRUE, the pcd entry is initialized in PEI phase,\r
+                            If FALSE, the pcd entry is initialized in DXE phase.\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
+  @retval  EFI_SUCCESS      The PCD information was returned successfully\r
+  @retval  EFI_NOT_FOUND    The PCD service could not find the requested token number.\r
+**/\r
+EFI_STATUS\r
+ExGetPcdInfo (\r
+  IN        BOOLEAN             IsPeiDb,\r
+  IN CONST  EFI_GUID            *Guid,\r
+  IN        UINTN               TokenNumber,\r
+  OUT       EFI_PCD_INFO        *PcdInfo\r
+  )\r
+{\r
+  PCD_DATABASE_INIT     *Database;\r
+  UINTN                 GuidTableIdx;\r
+  EFI_GUID              *MatchGuid;\r
+  EFI_GUID              *GuidTable;\r
+  DYNAMICEX_MAPPING     *ExMapTable;\r
+  UINTN                 Index;\r
+  UINT32                LocalTokenNumber;\r
+\r
+  Database = IsPeiDb ? mPcdDatabase.PeiDb: mPcdDatabase.DxeDb;\r
+\r
+  GuidTable = (EFI_GUID *)((UINT8 *)Database + Database->GuidTableOffset);\r
+  MatchGuid = ScanGuid (GuidTable, Database->GuidTableCount * sizeof(EFI_GUID), Guid);\r
+\r
+  if (MatchGuid == NULL) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  GuidTableIdx = MatchGuid - GuidTable;\r
+\r
+  ExMapTable = (DYNAMICEX_MAPPING *)((UINT8 *)Database + Database->ExMapTableOffset);\r
+\r
+  //\r
+  // Find the PCD by GuidTableIdx and ExTokenNumber in ExMapTable.\r
+  //\r
+  for (Index = 0; Index < Database->ExTokenCount; Index++) {\r
+    if (ExMapTable[Index].ExGuidIndex == GuidTableIdx) {\r
+      if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
+        //\r
+        // TokenNumber is 0, follow spec to set PcdType to EFI_PCD_TYPE_8,\r
+        // PcdSize to 0 and PcdName to the null-terminated ASCII string\r
+        // associated with the token's namespace Guid.\r
+        //\r
+        PcdInfo->PcdType = EFI_PCD_TYPE_8;\r
+        PcdInfo->PcdSize = 0;\r
+        //\r
+        // Here use one representative in the token space to get the TokenSpaceCName.\r
+        // \r
+        PcdInfo->PcdName = GetPcdName (TRUE, IsPeiDb, ExMapTable[Index].TokenNumber);\r
+        return EFI_SUCCESS;\r
+      } else if (ExMapTable[Index].ExTokenNumber == TokenNumber) {\r
+        PcdInfo->PcdSize = DxePcdGetSize (ExMapTable[Index].TokenNumber);\r
+        LocalTokenNumber = GetLocalTokenNumber (IsPeiDb, ExMapTable[Index].TokenNumber);\r
+        PcdInfo->PcdType = GetPcdType (LocalTokenNumber);\r
+        PcdInfo->PcdName = GetPcdName (FALSE, IsPeiDb, ExMapTable[Index].TokenNumber);\r
+        return EFI_SUCCESS;\r
+      }\r
+    }\r
+  }\r
+\r
+  return EFI_NOT_FOUND;\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
+  @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
+  @retval  EFI_SUCCESS      The PCD information was returned successfully.\r
+  @retval  EFI_NOT_FOUND    The PCD service could not find the requested token number.\r
+**/\r
+EFI_STATUS\r
+DxeGetPcdInfo (\r
+  IN CONST  EFI_GUID        *Guid,\r
+  IN        UINTN           TokenNumber,\r
+  OUT       EFI_PCD_INFO    *PcdInfo\r
+  )\r
+{\r
+  EFI_STATUS            Status;\r
+  BOOLEAN               PeiExMapTableEmpty;\r
+  BOOLEAN               DxeExMapTableEmpty;\r
+  UINT32                LocalTokenNumber;\r
+  BOOLEAN               IsPeiDb;\r
+\r
+  ASSERT (PcdInfo != NULL);\r
+\r
+  Status = EFI_NOT_FOUND;\r
+  PeiExMapTableEmpty = mPeiExMapTableEmpty;\r
+  DxeExMapTableEmpty = mDxeExMapTableEmpty;\r
+\r
+  if (Guid == NULL) {\r
+    if (((TokenNumber + 1 > mPeiNexTokenCount + 1) && (TokenNumber + 1 <= mPeiLocalTokenCount + 1)) ||\r
+        ((TokenNumber + 1 > (mPeiLocalTokenCount + mDxeNexTokenCount + 1)))) {\r
+      return EFI_NOT_FOUND;\r
+    } else if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
+      //\r
+      // TokenNumber is 0, follow spec to set PcdType to EFI_PCD_TYPE_8,\r
+      // PcdSize to 0 and PcdName to NULL for default Token Space.\r
+      //\r
+      PcdInfo->PcdType = EFI_PCD_TYPE_8;\r
+      PcdInfo->PcdSize = 0;\r
+      PcdInfo->PcdName = NULL;\r
+    } else {\r
+      PcdInfo->PcdSize = DxePcdGetSize (TokenNumber);\r
+      IsPeiDb = FALSE;\r
+      if ((TokenNumber + 1 <= mPeiNexTokenCount + 1)) {\r
+        IsPeiDb = TRUE;\r
+      }\r
+      LocalTokenNumber = GetLocalTokenNumber (IsPeiDb, TokenNumber);\r
+      PcdInfo->PcdType = GetPcdType (LocalTokenNumber);\r
+      PcdInfo->PcdName = GetPcdName (FALSE, IsPeiDb, TokenNumber);\r
+    }\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  if (PeiExMapTableEmpty && DxeExMapTableEmpty) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  if (!PeiExMapTableEmpty) {\r
+    Status = ExGetPcdInfo (\r
+               TRUE,\r
+               Guid,\r
+               TokenNumber,\r
+               PcdInfo\r
+               );\r
+  }\r
+\r
+  if (Status == EFI_SUCCESS) {\r
+    return Status;\r
+  }\r
+\r
+  if (!DxeExMapTableEmpty) {\r
+    Status = ExGetPcdInfo (\r
+               FALSE,\r
+               Guid,\r
+               TokenNumber,\r
+               PcdInfo\r
+               );\r
+  }\r
+\r
+  return Status;\r
+}\r
 \r
 /**\r
   Get the PCD entry pointer in PCD database.\r
@@ -56,7 +384,6 @@ GetWorker (
   IN UINTN             GetSize\r
   )\r
 {\r
-  UINT32              *LocalTokenNumberTable;\r
   EFI_GUID            *GuidTable;\r
   UINT8               *StringTable;\r
   EFI_GUID            *Guid;\r
@@ -67,7 +394,6 @@ GetWorker (
   VPD_HEAD            *VpdHead;\r
   UINT8               *PcdDb;\r
   VOID                *RetPtr;\r
-  UINTN               MaxSize;\r
   UINTN               TmpTokenNumber;\r
   UINTN               DataSize;\r
   EFI_STATUS          Status;\r
@@ -107,22 +433,7 @@ GetWorker (
   // comparison.\r
   IsPeiDb = (BOOLEAN) ((TokenNumber + 1 < mPeiLocalTokenCount + 1) ? TRUE : FALSE);\r
 \r
-  LocalTokenNumberTable  = IsPeiDb ? (UINT32 *)((UINT8 *)mPcdDatabase.PeiDb + mPcdDatabase.PeiDb->LocalTokenNumberTableOffset) : \r
-                                     (UINT32 *)((UINT8 *)mPcdDatabase.DxeDb + mPcdDatabase.DxeDb->LocalTokenNumberTableOffset);\r
-\r
-  TokenNumber            = IsPeiDb ? TokenNumber :\r
-                                     TokenNumber - mPeiLocalTokenCount;\r
-\r
-  LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
-  \r
-  if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
-    if (GetSize == 0) {\r
-      GetPtrTypeSize (TmpTokenNumber, &MaxSize);\r
-    } else {\r
-      MaxSize = GetSize;\r
-    }\r
-    LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);\r
-  }\r
+  LocalTokenNumber = GetLocalTokenNumber (IsPeiDb, TokenNumber + 1);\r
 \r
   PcdDb = IsPeiDb ? ((UINT8 *) mPcdDatabase.PeiDb) : ((UINT8 *) mPcdDatabase.DxeDb);\r
                                     \r
@@ -160,25 +471,33 @@ GetWorker (
         // string array in string table.\r
         //\r
         StringTableIdx = *(STRING_HEAD*)((UINT8 *) PcdDb + VariableHead->DefaultValueOffset);   \r
-        VaraiableDefaultBuffer = (VOID *) (StringTable + StringTableIdx);     \r
+        VaraiableDefaultBuffer = (UINT8 *) (StringTable + StringTableIdx);     \r
       } else {\r
         VaraiableDefaultBuffer = (UINT8 *) PcdDb + VariableHead->DefaultValueOffset;\r
       }\r
       Status = GetHiiVariable (Guid, Name, &Data, &DataSize);\r
       if (Status == EFI_SUCCESS) {\r
-        if (GetSize == 0) {\r
+        if (DataSize >= (VariableHead->Offset + GetSize)) {\r
+          if (GetSize == 0) {\r
+            //\r
+            // It is a pointer type. So get the MaxSize reserved for\r
+            // this PCD entry.\r
+            //\r
+            GetPtrTypeSize (TmpTokenNumber, &GetSize);\r
+            if (GetSize > (DataSize - VariableHead->Offset)) {\r
+              //\r
+              // Use actual valid size.\r
+              //\r
+              GetSize = DataSize - VariableHead->Offset;\r
+            }\r
+          }\r
           //\r
-          // It is a pointer type. So get the MaxSize reserved for\r
-          // this PCD entry.\r
+          // If the operation is successful, we copy the data\r
+          // to the default value buffer in the PCD Database.\r
+          // So that we can free the Data allocated in GetHiiVariable.\r
           //\r
-          GetPtrTypeSize (TmpTokenNumber, &GetSize);\r
+          CopyMem (VaraiableDefaultBuffer, Data + VariableHead->Offset, GetSize);\r
         }\r
-        //\r
-        // If the operation is successful, we copy the data\r
-        // to the default value buffer in the PCD Database.\r
-        // So that we can free the Data allocated in GetHiiVariable.\r
-        //\r
-        CopyMem (VaraiableDefaultBuffer, Data + VariableHead->Offset, GetSize);\r
         FreePool (Data);\r
       }\r
       RetPtr = (VOID *) VaraiableDefaultBuffer;\r
@@ -477,6 +796,18 @@ BuildPcdDxeDataBase (
   UINT32              PcdDxeDbLen;\r
   VOID                *PcdDxeDb;\r
 \r
+  //\r
+  // Assign PCD Entries with default value to PCD DATABASE\r
+  //\r
+  mPcdDatabase.DxeDb = LocateExPcdBinary ();\r
+  ASSERT(mPcdDatabase.DxeDb != NULL);\r
+  PcdDxeDbLen = mPcdDatabase.DxeDb->Length + mPcdDatabase.DxeDb->UninitDataBaseSize;\r
+  PcdDxeDb = AllocateZeroPool (PcdDxeDbLen);\r
+  ASSERT (PcdDxeDb != NULL);\r
+  CopyMem (PcdDxeDb, mPcdDatabase.DxeDb, mPcdDatabase.DxeDb->Length);\r
+  FreePool (mPcdDatabase.DxeDb);\r
+  mPcdDatabase.DxeDb = PcdDxeDb;\r
+\r
   GuidHob = GetFirstGuidHob (&gPcdDataBaseHobGuid);\r
   if (GuidHob != NULL) {\r
 \r
@@ -492,19 +823,15 @@ BuildPcdDxeDataBase (
     // Assign PCD Entries refereneced in PEI phase to PCD DATABASE\r
     //\r
     mPcdDatabase.PeiDb = PeiDatabase;\r
+    //\r
+    // Inherit the SystemSkuId from PEI phase.\r
+    //\r
+    mPcdDatabase.DxeDb->SystemSkuId = mPcdDatabase.PeiDb->SystemSkuId;\r
+  } else {\r
+    mPcdDatabase.PeiDb = AllocateZeroPool (sizeof (PEI_PCD_DATABASE));\r
+    ASSERT(mPcdDatabase.PeiDb != NULL);\r
   }\r
 \r
-  //\r
-  // Assign PCD Entries with default value to PCD DATABASE\r
-  //\r
-  mPcdDatabase.DxeDb = LocateExPcdBinary ();\r
-  ASSERT(mPcdDatabase.DxeDb != NULL);\r
-  PcdDxeDbLen = mPcdDatabase.DxeDb->Length + mPcdDatabase.DxeDb->UninitDataBaseSize;\r
-  PcdDxeDb = AllocateZeroPool (PcdDxeDbLen);\r
-  CopyMem (PcdDxeDb, mPcdDatabase.DxeDb, mPcdDatabase.DxeDb->Length);\r
-  FreePool (mPcdDatabase.DxeDb);\r
-  mPcdDatabase.DxeDb = PcdDxeDb;\r
-\r
   //\r
   // Initialized the external PCD database local variables\r
   //\r
@@ -647,7 +974,7 @@ GetSkuEnabledTokenNumber (
   //\r
   FoundSku = FALSE;\r
   for (Index = 0; Index < SkuIdTable[0]; Index++) {\r
-    if (mPcdDatabase.PeiDb->SystemSkuId == SkuIdTable[Index + 1]) {\r
+    if (mPcdDatabase.DxeDb->SystemSkuId == SkuIdTable[Index + 1]) {\r
       FoundSku = TRUE;\r
       break;\r
     }\r
@@ -791,7 +1118,6 @@ SetWorker (
   IN          BOOLEAN                 PtrType\r
   )\r
 {\r
-  UINT32              *LocalTokenNumberTable;\r
   BOOLEAN             IsPeiDb;\r
   UINT32              LocalTokenNumber;\r
   EFI_GUID            *GuidTable;\r
@@ -799,6 +1125,7 @@ SetWorker (
   EFI_GUID            *Guid;\r
   UINT16              *Name;\r
   UINTN               VariableOffset;\r
+  UINT32              Attributes;\r
   VOID                *InternalData;\r
   VARIABLE_HEAD       *VariableHead;\r
   UINTN               Offset;\r
@@ -860,22 +1187,7 @@ SetWorker (
   //\r
   IsPeiDb = (BOOLEAN) ((TokenNumber + 1 < mPeiLocalTokenCount + 1) ? TRUE : FALSE);\r
 \r
-  LocalTokenNumberTable  = IsPeiDb ? (UINT32 *)((UINT8 *)mPcdDatabase.PeiDb + mPcdDatabase.PeiDb->LocalTokenNumberTableOffset) : \r
-                                     (UINT32 *)((UINT8 *)mPcdDatabase.DxeDb + mPcdDatabase.DxeDb->LocalTokenNumberTableOffset) ;\r
-\r
-  TokenNumber = IsPeiDb ? TokenNumber\r
-                        : TokenNumber - mPeiLocalTokenCount;\r
-\r
-  LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
-  \r
-  if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
-    if (PtrType) {\r
-      GetPtrTypeSize (TmpTokenNumber, &MaxSize);\r
-    } else {\r
-      MaxSize = *Size;\r
-    }\r
-    LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);\r
-  }\r
+  LocalTokenNumber = GetLocalTokenNumber (IsPeiDb, TokenNumber + 1);\r
 \r
   Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
 \r
@@ -925,20 +1237,8 @@ SetWorker (
       Guid = GuidTable + VariableHead->GuidTableIndex;\r
       Name = (UINT16*) (StringTable + VariableHead->StringIndex);\r
       VariableOffset = VariableHead->Offset;\r
-      Status = SetHiiVariable (Guid, Name, Data, *Size, VariableOffset);\r
-\r
-      if (EFI_NOT_FOUND == Status) {\r
-        if ((LocalTokenNumber & PCD_TYPE_ALL_SET) == (PCD_TYPE_HII|PCD_TYPE_STRING))  {\r
-          CopyMem (\r
-            StringTable + *(STRING_HEAD *)(PcdDb + VariableHead->DefaultValueOffset),\r
-            Data,\r
-            *Size\r
-            );\r
-        } else {\r
-          CopyMem (PcdDb + VariableHead->DefaultValueOffset, Data, *Size);\r
-        } \r
-        Status = EFI_SUCCESS;\r
-      }\r
+      Attributes = VariableHead->Attributes;\r
+      Status = SetHiiVariable (Guid, Name, Attributes, Data, *Size, VariableOffset);\r
       break;\r
       \r
     case PCD_TYPE_DATA:\r
@@ -1075,6 +1375,7 @@ ExSetWorker (
   \r
   @param VariableGuid    Guid of variable which stored value of a HII-type PCD.\r
   @param VariableName    Unicode name of variable which stored value of a HII-type PCD.\r
+  @param SetAttributes   Attributes bitmask to set for the variable.\r
   @param Data            Value want to be set.\r
   @param DataSize        Size of value\r
   @param Offset          Value offset of HII-type PCD in variable.\r
@@ -1086,6 +1387,7 @@ EFI_STATUS
 SetHiiVariable (\r
   IN  EFI_GUID     *VariableGuid,\r
   IN  UINT16       *VariableName,\r
+  IN  UINT32       SetAttributes,\r
   IN  CONST VOID   *Data,\r
   IN  UINTN        DataSize,\r
   IN  UINTN        Offset\r
@@ -1115,7 +1417,7 @@ SetHiiVariable (
     //\r
     // Patch new PCD's value to offset in given HII variable.\r
     //\r
-    if  (Size >= (DataSize + Offset)) {\r
+    if (Size >= (DataSize + Offset)) {\r
       SetSize = Size;\r
     } else {\r
       SetSize = DataSize + Offset;\r
@@ -1135,10 +1437,14 @@ SetHiiVariable (
 \r
     CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
 \r
+    if (SetAttributes == 0) {\r
+      SetAttributes = Attribute;\r
+    }\r
+\r
     Status = gRT->SetVariable (\r
               VariableName,\r
               VariableGuid,\r
-              Attribute,\r
+              SetAttributes,\r
               SetSize,\r
               Buffer\r
               );\r
@@ -1156,11 +1462,15 @@ SetHiiVariable (
     ASSERT (Buffer != NULL);\r
     \r
     CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
-    \r
+\r
+    if (SetAttributes == 0) {\r
+      SetAttributes = EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE;\r
+    }\r
+\r
     Status = gRT->SetVariable (\r
               VariableName,\r
               VariableGuid,\r
-              EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,\r
+              SetAttributes,\r
               Size,\r
               Buffer\r
               );\r
@@ -1170,8 +1480,7 @@ SetHiiVariable (
   }\r
   \r
   //\r
-  // If we drop to here, the value is failed to be written in to variable area\r
-  // So, we will save the data in the PCD Database's volatile area.\r
+  // If we drop to here, the value is failed to be written in to variable area.\r
   //\r
   return Status;\r
 }\r
@@ -1211,7 +1520,7 @@ GetExPcdTokenNumber (
 \r
       MatchGuidIdx = MatchGuid - GuidTable;\r
 \r
-      for (Index = 0; Index < mPeiExMapppingTableSize; Index++) {\r
+      for (Index = 0; Index < mPcdDatabase.PeiDb->ExTokenCount; Index++) {\r
         if ((ExTokenNumber == ExMap[Index].ExTokenNumber) &&\r
             (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {\r
             return ExMap[Index].TokenNumber;\r
@@ -1232,7 +1541,7 @@ GetExPcdTokenNumber (
 \r
   MatchGuidIdx = MatchGuid - GuidTable;\r
 \r
-  for (Index = 0; Index < mDxeExMapppingTableSize; Index++) {\r
+  for (Index = 0; Index < mPcdDatabase.DxeDb->ExTokenCount; Index++) {\r
     if ((ExTokenNumber == ExMap[Index].ExTokenNumber) &&\r
          (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {\r
         return ExMap[Index].TokenNumber;\r
@@ -1422,7 +1731,7 @@ GetPtrTypeSize (
       //\r
       SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);\r
       for (Index = 0; Index < SkuIdTable[0]; Index++) {\r
-        if (SkuIdTable[1 + Index] == mPcdDatabase.PeiDb->SystemSkuId) {\r
+        if (SkuIdTable[1 + Index] == mPcdDatabase.DxeDb->SystemSkuId) {\r
           return SizeTable[SizeTableIdx + 1 + Index];\r
         }\r
       }\r
@@ -1512,7 +1821,7 @@ SetPtrTypeSize (
       //\r
       SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);\r
       for (Index = 0; Index < SkuIdTable[0]; Index++) {\r
-        if (SkuIdTable[1 + Index] == mPcdDatabase.PeiDb->SystemSkuId) {\r
+        if (SkuIdTable[1 + Index] == mPcdDatabase.DxeDb->SystemSkuId) {\r
           SizeTable[SizeTableIdx + 1 + Index] = (SIZE_INFO) *CurrentSize;\r
           return TRUE;\r
         }\r
@@ -1522,3 +1831,91 @@ SetPtrTypeSize (
     }\r
   }\r
 }\r
+\r
+/**\r
+  VariableLock DynamicHiiPcd.\r
+\r
+  @param[in] IsPeiDb        If TRUE, the pcd entry is initialized in PEI phase,\r
+                            If FALSE, the pcd entry is initialized in DXE phase.\r
+  @param[in] VariableLock   Pointer to VariableLockProtocol.\r
+\r
+**/\r
+VOID\r
+VariableLockDynamicHiiPcd (\r
+  IN BOOLEAN                        IsPeiDb,\r
+  IN EDKII_VARIABLE_LOCK_PROTOCOL   *VariableLock\r
+  )\r
+{\r
+  EFI_STATUS                Status;\r
+  PCD_DATABASE_INIT         *Database;\r
+  UINT32                    LocalTokenCount; \r
+  UINTN                     TokenNumber;\r
+  UINT32                    LocalTokenNumber;\r
+  UINTN                     Offset;\r
+  EFI_GUID                  *GuidTable;\r
+  UINT8                     *StringTable;\r
+  VARIABLE_HEAD             *VariableHead;\r
+  EFI_GUID                  *Guid;\r
+  UINT16                    *Name;\r
+\r
+  Database = IsPeiDb ? mPcdDatabase.PeiDb: mPcdDatabase.DxeDb;\r
+  LocalTokenCount = IsPeiDb ? mPeiLocalTokenCount: mDxeLocalTokenCount;\r
+\r
+  //\r
+  // Go through PCD database to find out DynamicHii PCDs.\r
+  //\r
+  for (TokenNumber = 0; TokenNumber < LocalTokenCount; TokenNumber++) {\r
+    if (IsPeiDb) {\r
+      LocalTokenNumber = GetLocalTokenNumber (TRUE, TokenNumber);\r
+    } else {\r
+      LocalTokenNumber = GetLocalTokenNumber (FALSE, TokenNumber + mPeiLocalTokenCount);\r
+    }\r
+    if ((LocalTokenNumber & PCD_TYPE_HII) != 0) {\r
+      Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
+      VariableHead = (VARIABLE_HEAD *) ((UINT8 *) Database + Offset);\r
+      //\r
+      // Why not to set property by VarCheckProtocol with Attributes and Property directly here?\r
+      // It is because that set property by VarCheckProtocol will indicate the variable to\r
+      // be a system variable, but the unknown max size of the variable is dangerous to\r
+      // the system variable region.\r
+      //\r
+      if ((VariableHead->Property & VAR_CHECK_VARIABLE_PROPERTY_READ_ONLY) != 0) {\r
+        //\r
+        // DynamicHii PCD with RO property set in *.dsc.\r
+        //\r
+        StringTable = (UINT8 *) ((UINT8 *) Database + Database->StringTableOffset);\r
+        GuidTable = (EFI_GUID *) ((UINT8 *) Database + Database->GuidTableOffset);\r
+        Guid = GuidTable + VariableHead->GuidTableIndex;\r
+        Name = (UINT16*) (StringTable + VariableHead->StringIndex);\r
+        Status = VariableLock->RequestToLock (VariableLock, Name, Guid);\r
+        ASSERT_EFI_ERROR (Status);\r
+      }\r
+    }\r
+  }\r
+}\r
+\r
+/**\r
+  VariableLockProtocol callback\r
+  to lock the variables referenced by DynamicHii PCDs with RO property set in *.dsc.\r
+\r
+  @param[in] Event      Event whose notification function is being invoked.\r
+  @param[in] Context    Pointer to the notification function's context.\r
+\r
+**/\r
+VOID\r
+EFIAPI\r
+VariableLockCallBack (\r
+  IN EFI_EVENT          Event,\r
+  IN VOID               *Context\r
+  )\r
+{\r
+  EFI_STATUS                    Status;\r
+  EDKII_VARIABLE_LOCK_PROTOCOL  *VariableLock;\r
+\r
+  Status = gBS->LocateProtocol (&gEdkiiVariableLockProtocolGuid, NULL, (VOID **) &VariableLock);\r
+  if (!EFI_ERROR (Status)) {\r
+    VariableLockDynamicHiiPcd (TRUE, VariableLock);\r
+    VariableLockDynamicHiiPcd (FALSE, VariableLock);\r
+  }\r
+}\r
+\r