]> git.proxmox.com Git - mirror_edk2.git/blobdiff - EdkModulePkg/Universal/PCD/Dxe/Service.c
1) Add in support to traverse taken space
[mirror_edk2.git] / EdkModulePkg / Universal / PCD / Dxe / Service.c
index 52097ade7e877d0baef83fd08a6c4362db1becee..97518510d6c63db29b53c4b5f753a2219c140c66 100644 (file)
@@ -22,21 +22,20 @@ Module Name: Service.c
 // Compression Algorithm will take care of the size optimization.\r
 //\r
 \r
-/*\r
-DXE_PCD_DATABASE_INIT gDXEPcdDbInit = {\r
-  DXE_PCD_DB_INIT_VALUE\r
-};\r
-*/\r
+PCD_DATABASE * mPcdDatabase;\r
 \r
-PCD_DATABASE * gPcdDatabase;\r
+LIST_ENTRY *mCallbackFnTable;\r
 \r
 VOID *\r
-GetWorkerByLocalTokenNumber (\r
-  UINT32      LocalTokenNumber,\r
-  BOOLEAN     IsPeiDb,\r
-  UINTN       Size\r
-  ) \r
+GetWorker (\r
+  UINTN             TokenNumber,\r
+  UINTN             GetSize\r
+  )\r
 {\r
+  UINT32              *LocalTokenNumberTable;\r
+  UINT16              *SizeTable;\r
+  BOOLEAN             IsPeiDb;\r
+  UINTN               Size;\r
   UINT32              Offset;\r
   EFI_GUID            *GuidTable;\r
   UINT16              *StringTable;\r
@@ -49,25 +48,52 @@ GetWorkerByLocalTokenNumber (
   VPD_HEAD            *VpdHead;\r
   UINT8               *PcdDb;\r
   UINT16              StringTableIdx;      \r
+  UINT32              LocalTokenNumber;\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
+  ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
+\r
+  Size = DxePcdGetSize (TokenNumber + 1);\r
+  ASSERT (GetSize == Size || GetSize == 0);\r
+\r
+  \r
+  IsPeiDb = (TokenNumber < PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+\r
+  LocalTokenNumberTable  = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
+                                     mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+\r
+  SizeTable              = IsPeiDb ? mPcdDatabase->PeiDb.Init.SizeTable: \r
+                                     mPcdDatabase->DxeDb.Init.SizeTable;\r
 \r
+  TokenNumber            = IsPeiDb ? TokenNumber :\r
+                                     TokenNumber - PEI_LOCAL_TOKEN_NUMBER;\r
+\r
+  LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
+  \r
   if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
     LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, Size, IsPeiDb);\r
   }\r
 \r
-  PcdDb = IsPeiDb ? ((UINT8 *) &gPcdDatabase->PeiDb) : ((UINT8 *) &gPcdDatabase->DxeDb);\r
-  StringTable = IsPeiDb ? gPcdDatabase->PeiDb.Init.StringTable :\r
-                          gPcdDatabase->DxeDb.Init.StringTable;\r
+  PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);\r
+  StringTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.StringTable :\r
+                          mPcdDatabase->DxeDb.Init.StringTable;\r
   \r
   Offset     = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
   \r
   switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
     case PCD_TYPE_VPD:\r
       VpdHead = (VPD_HEAD *) ((UINT8 *) PcdDb + Offset);\r
-      return (VOID *) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
+      return (VOID *) (UINTN) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
       \r
     case PCD_TYPE_HII:\r
-      GuidTable   = IsPeiDb ? gPcdDatabase->PeiDb.Init.GuidTable :\r
-                              gPcdDatabase->DxeDb.Init.GuidTable;\r
+      GuidTable   = IsPeiDb ? mPcdDatabase->PeiDb.Init.GuidTable :\r
+                              mPcdDatabase->DxeDb.Init.GuidTable;\r
                               \r
       VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);\r
       \r
@@ -75,10 +101,15 @@ GetWorkerByLocalTokenNumber (
       Name = &(StringTable[VariableHead->StringIndex]);\r
 \r
       Status = GetHiiVariable (Guid, Name, &Data, &DataSize);\r
-      ASSERT_EFI_ERROR (Status);\r
-      ASSERT (DataSize >= (UINTN) (VariableHead->Offset + Size));\r
-\r
-      return (UINT8 *) Data + VariableHead->Offset;\r
+      if (Status == EFI_SUCCESS) {\r
+        ASSERT (DataSize >= (UINTN) (VariableHead->Offset + Size));\r
+        return (UINT8 *) Data + VariableHead->Offset;\r
+      } else {\r
+        //\r
+        // Return the default value specified by Platform Integrator \r
+        //\r
+        return (VOID *) ((UINT8 *) PcdDb + VariableHead->DefaultValueOffset);\r
+     }\r
 \r
     case PCD_TYPE_STRING:\r
       StringTableIdx = (UINT16) *((UINT8 *) PcdDb + Offset);\r
@@ -97,82 +128,221 @@ GetWorkerByLocalTokenNumber (
   ASSERT (FALSE);\r
       \r
   return NULL;\r
-}\r
   \r
-VOID *\r
-GetWorker (\r
-  UINTN  TokenNumber\r
-  )\r
+}\r
+\r
+\r
+\r
+EFI_STATUS\r
+DxeRegisterCallBackWorker (\r
+  IN  UINTN                   TokenNumber,\r
+  IN  CONST GUID              *Guid, OPTIONAL\r
+  IN  PCD_PROTOCOL_CALLBACK   CallBackFunction\r
+)\r
 {\r
-  UINT32        *LocalTokenNumberTable;\r
-  UINT16        *SizeTable;\r
-  BOOLEAN       IsPeiDb;\r
+  CALLBACK_FN_ENTRY       *FnTableEntry;\r
+  LIST_ENTRY              *ListHead;\r
+  LIST_ENTRY              *ListNode;\r
 \r
-  ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
-  \r
-  IsPeiDb = (TokenNumber <= PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+  if (Guid != NULL) {\r
+    TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) TokenNumber);\r
+  }\r
 \r
-  LocalTokenNumberTable  = IsPeiDb ? gPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
-                                     gPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\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
-  SizeTable              = IsPeiDb ? gPcdDatabase->PeiDb.Init.SizeTable: \r
-                                     gPcdDatabase->DxeDb.Init.SizeTable;\r
+  ListHead = &mCallbackFnTable[TokenNumber];\r
+  ListNode = GetFirstNode (ListHead);\r
 \r
-  TokenNumber            = IsPeiDb ? TokenNumber :\r
-                                     TokenNumber - PEI_LOCAL_TOKEN_NUMBER;\r
-  return GetWorkerByLocalTokenNumber (LocalTokenNumberTable[TokenNumber], IsPeiDb, SizeTable[TokenNumber]);\r
+  while (ListNode != ListHead) {\r
+    FnTableEntry = CR_FNENTRY_FROM_LISTNODE(ListNode, CALLBACK_FN_ENTRY, Node);\r
+\r
+    if (FnTableEntry->CallbackFn == CallBackFunction) {\r
+      //\r
+      // We only allow a Callback function to be register once\r
+      // for a TokenNumber. So just return EFI_SUCCESS\r
+      //\r
+      return EFI_SUCCESS;\r
+    }\r
+    ListNode = GetNextNode (ListHead, ListNode);\r
+  }\r
+\r
+  FnTableEntry = AllocatePool (sizeof(CALLBACK_FN_ENTRY));\r
+  ASSERT (FnTableEntry != NULL);\r
+\r
+  FnTableEntry->CallbackFn = CallBackFunction;\r
+  InsertTailList (ListHead, &FnTableEntry->Node);\r
+  \r
+  return EFI_SUCCESS;\r
 }\r
 \r
 \r
 \r
+\r
 EFI_STATUS\r
-DxeRegisterCallBackWorker (\r
+DxeUnRegisterCallBackWorker (\r
   IN  UINTN                   TokenNumber,\r
   IN  CONST GUID              *Guid, OPTIONAL\r
-  IN  PCD_PROTOCOL_CALLBACK   CallBackFunction,\r
-  IN  BOOLEAN                 Register\r
+  IN  PCD_PROTOCOL_CALLBACK   CallBackFunction\r
 )\r
 {\r
-  \r
-  return EFI_SUCCESS;\r
+  CALLBACK_FN_ENTRY       *FnTableEntry;\r
+  LIST_ENTRY              *ListHead;\r
+  LIST_ENTRY              *ListNode;\r
+\r
+  if (Guid != NULL) {\r
+    TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) TokenNumber);\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
+  ListHead = &mCallbackFnTable[TokenNumber];\r
+  ListNode = GetFirstNode (ListHead);\r
+\r
+  while (ListNode != ListHead) {\r
+    FnTableEntry = CR_FNENTRY_FROM_LISTNODE(ListNode, CALLBACK_FN_ENTRY, Node);\r
+\r
+    if (FnTableEntry->CallbackFn == CallBackFunction) {\r
+      //\r
+      // We only allow a Callback function to be register once\r
+      // for a TokenNumber. So we can safely remove the Node from\r
+      // the Link List and return EFI_SUCCESS.\r
+      //\r
+      RemoveEntryList (ListNode);\r
+      FreePool (FnTableEntry);\r
+      \r
+      return EFI_SUCCESS;\r
+    }\r
+    ListNode = GetNextNode (ListHead, ListNode);\r
+  }\r
+\r
+  return EFI_INVALID_PARAMETER;\r
 }\r
 \r
 \r
-EFI_STATUS\r
-DxeGetNextTokenWorker (\r
-  IN OUT UINTN            *TokenNumber,\r
-  IN CONST GUID           *Guid     OPTIONAL\r
+\r
+UINTN           \r
+ExGetNextTokeNumber (\r
+  IN CONST EFI_GUID         *Guid,\r
+  IN UINTN                  TokenNumber,\r
+  IN EFI_GUID               *GuidTable,\r
+  IN UINTN                  SizeOfGuidTable,\r
+  IN DYNAMICEX_MAPPING      *ExMapTable,\r
+  IN UINTN                  SizeOfExMapTable\r
   )\r
 {\r
-  return EFI_SUCCESS;\r
+  EFI_GUID         *MatchGuid;\r
+  UINTN            Idx;\r
+  UINTN            GuidTableIdx;\r
+  BOOLEAN          Found;\r
+\r
+  MatchGuid = ScanGuid (GuidTable, SizeOfGuidTable, Guid);\r
+  if (MatchGuid == NULL) {\r
+    return PCD_INVALID_TOKEN_NUMBER;\r
+  }\r
+\r
+  Found = FALSE;\r
+  GuidTableIdx = MatchGuid - GuidTable;\r
+  for (Idx = 0; Idx < SizeOfExMapTable; Idx++) {\r
+    if (ExMapTable[Idx].ExGuidIndex == GuidTableIdx) {\r
+      Found = TRUE;\r
+      break;\r
+    }\r
+  }\r
+\r
+  if (Found) {\r
+    if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
+      return ExMapTable[Idx].ExTokenNumber;\r
+    }\r
+\r
+    for ( ; Idx < SizeOfExMapTable; Idx++) {\r
+      if (ExMapTable[Idx].ExTokenNumber == TokenNumber) {\r
+        Idx++;\r
+        if (Idx == SizeOfExMapTable) {\r
+          //\r
+          // Exceed the length of ExMap Table\r
+          //\r
+          return PCD_INVALID_TOKEN_NUMBER;\r
+        } else if (ExMapTable[Idx].ExGuidIndex == GuidTableIdx) {\r
+          //\r
+          // Found the next match\r
+          //\r
+          return ExMapTable[Idx].ExTokenNumber;\r
+        } else {\r
+          //\r
+          // Guid has been changed. It is the next Token Space Guid.\r
+          // We should flag no more TokenNumber.\r
+          //\r
+          return PCD_INVALID_TOKEN_NUMBER;\r
+        }\r
+      }\r
+    }\r
+  }\r
+  \r
+  return PCD_INVALID_TOKEN_NUMBER;\r
 }\r
+  \r
 \r
 \r
 \r
 VOID\r
 BuildPcdDxeDataBase (\r
   VOID\r
-)\r
+  )\r
 {\r
   PEI_PCD_DATABASE    *PeiDatabase;\r
   EFI_HOB_GUID_TYPE   *GuidHob;\r
+  UINTN               Idx;\r
 \r
-  gPcdDatabase = AllocateZeroPool (sizeof(PCD_DATABASE));\r
-  ASSERT (gPcdDatabase != NULL);\r
+  mPcdDatabase = AllocateZeroPool (sizeof(PCD_DATABASE));\r
+  ASSERT (mPcdDatabase != NULL);\r
 \r
   GuidHob = GetFirstGuidHob (&gPcdDataBaseHobGuid);\r
-  ASSERT (GuidHob != NULL);\r
 \r
-  PeiDatabase = (PEI_PCD_DATABASE *) GET_GUID_HOB_DATA (GuidHob);\r
+  if (GuidHob != NULL) {\r
+\r
+    //\r
+    // We will copy over the PEI phase's PCD Database.\r
+    // \r
+    // If no PEIMs use dynamic Pcd Entry, the Pcd Service PEIM\r
+    // should not be included at all. So the GuidHob could\r
+    // be NULL. If it is NULL, we just copy over the DXE Default\r
+    // Value to PCD Database.\r
+    //\r
+    \r
+    PeiDatabase = (PEI_PCD_DATABASE *) GET_GUID_HOB_DATA (GuidHob);\r
+    //\r
+    // Copy PCD Entries refereneced in PEI phase to PCD DATABASE\r
+    //\r
+    CopyMem (&mPcdDatabase->PeiDb, PeiDatabase, sizeof (PEI_PCD_DATABASE));\r
+  }\r
+\r
   //\r
-  // Copy PCD Entries refereneced in PEI phase to PCD DATABASE\r
+  // Copy PCD Entries with default value to PCD DATABASE\r
   //\r
-  CopyMem (&gPcdDatabase->PeiDb, PeiDatabase, sizeof (PEI_PCD_DATABASE));\r
+  CopyMem (&mPcdDatabase->DxeDb.Init, &gDXEPcdDbInit, sizeof(DXE_PCD_DATABASE_INIT));\r
+\r
 \r
   //\r
-  // Copy PCD Entries with default value to PCD DATABASE\r
+  // Initialized the Callback Function Table\r
   //\r
-  CopyMem (&gPcdDatabase->DxeDb.Init, &gDXEPcdDbInit, sizeof(DXE_PCD_DATABASE_INIT));\r
+\r
+  if (PCD_TOTAL_TOKEN_NUMBER != 0) {\r
+    mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER * sizeof (LIST_ENTRY));\r
+  }\r
+  \r
+  for (Idx = 0; Idx < PCD_TOTAL_TOKEN_NUMBER; Idx++) {\r
+    InitializeListHead (&mCallbackFnTable[Idx]);\r
+  }\r
     \r
   return;\r
 }\r
@@ -191,26 +361,36 @@ GetHiiVariable (
   EFI_STATUS Status;\r
   VOID       *Buffer;\r
 \r
+  Size = 0;\r
+  Buffer = NULL;\r
+  \r
   Status = EfiGetVariable (\r
     (UINT16 *)VariableName,\r
     VariableGuid,\r
     NULL,\r
     &Size,\r
-    NULL\r
+    Buffer\r
     );\r
-  ASSERT (Status == EFI_BUFFER_TOO_SMALL);\r
+  \r
+  if (Status == EFI_BUFFER_TOO_SMALL) {\r
 \r
-  Buffer = AllocatePool (Size);\r
+    Buffer = AllocatePool (Size);\r
 \r
-  ASSERT (Buffer != NULL);\r
+    ASSERT (Buffer != NULL);\r
 \r
-  Status = EfiGetVariable (\r
-    VariableName,\r
-    VariableGuid,\r
-    NULL,\r
-    &Size,\r
-    Buffer\r
-    );\r
+    Status = EfiGetVariable (\r
+      VariableName,\r
+      VariableGuid,\r
+      NULL,\r
+      &Size,\r
+      Buffer\r
+      );\r
+\r
+    ASSERT (Status == EFI_SUCCESS);\r
+  }\r
+\r
+  *VariableData = Buffer;\r
+  *VariableSize = Size;\r
 \r
   return Status;\r
 \r
@@ -233,18 +413,18 @@ GetSkuEnabledTokenNumber (
 \r
   ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0);\r
 \r
-  PcdDb = IsPeiDb ? (UINT8 *) &gPcdDatabase->PeiDb : (UINT8 *) &gPcdDatabase->DxeDb;\r
+  PcdDb = IsPeiDb ? (UINT8 *) &mPcdDatabase->PeiDb : (UINT8 *) &mPcdDatabase->DxeDb;\r
 \r
   SkuHead     = (SKU_HEAD *) (PcdDb + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));\r
   Value       = (UINT8 *) (PcdDb + SkuHead->SkuDataStartOffset); \r
 \r
-  PhaseSkuIdTable = IsPeiDb ? gPcdDatabase->PeiDb.Init.SkuIdTable :\r
-                              gPcdDatabase->DxeDb.Init.SkuIdTable;\r
+  PhaseSkuIdTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.SkuIdTable :\r
+                              mPcdDatabase->DxeDb.Init.SkuIdTable;\r
                               \r
   SkuIdTable  = &PhaseSkuIdTable[SkuHead->SkuIdTableOffset];\r
         \r
   for (i = 0; i < SkuIdTable[0]; i++) {\r
-    if (gPcdDatabase->PeiDb.Init.SystemSkuId == SkuIdTable[i + 1]) {\r
+    if (mPcdDatabase->PeiDb.Init.SystemSkuId == SkuIdTable[i + 1]) {\r
       break;\r
     }\r
   }\r
@@ -253,15 +433,15 @@ GetSkuEnabledTokenNumber (
   switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
     case PCD_TYPE_VPD:\r
       Value = (UINT8 *) &(((VPD_HEAD *) Value)[i]);\r
-      return ((Value - PcdDb) | PCD_TYPE_VPD);\r
+      return (UINT32) ((Value - PcdDb) | PCD_TYPE_VPD);\r
 \r
     case PCD_TYPE_HII:\r
       Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[i]);\r
-      return ((Value - PcdDb) | PCD_TYPE_HII);\r
+      return (UINT32) ((Value - PcdDb) | PCD_TYPE_HII);\r
       \r
     case PCD_TYPE_DATA:\r
       Value += Size * i;\r
-      return (Value - PcdDb);\r
+      return (UINT32) (Value - PcdDb);\r
       \r
     default:\r
       ASSERT (FALSE);\r
@@ -286,130 +466,98 @@ InvokeCallbackOnSet (
   UINTN             Size\r
   )\r
 {\r
-  return;\r
-}\r
-\r
-\r
-\r
-\r
-EFI_STATUS\r
-SetWorker (\r
-  UINTN         TokenNumber,\r
-  VOID          *Data,\r
-  UINTN         Size,\r
-  BOOLEAN       PtrType\r
-  )\r
-{\r
-  UINT32              *LocalTokenNumberTable;\r
-  BOOLEAN             IsPeiDb;\r
-\r
-\r
-  ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
+  CALLBACK_FN_ENTRY       *FnTableEntry;\r
+  LIST_ENTRY              *ListHead;\r
+  LIST_ENTRY              *ListNode;\r
 \r
-  if (PtrType) {\r
-    ASSERT (Size <= DxePcdGetSize (TokenNumber));\r
-  } else {\r
-    ASSERT (Size == DxePcdGetSize (TokenNumber));\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
-  IsPeiDb = (TokenNumber <= PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+  ListHead = &mCallbackFnTable[TokenNumber];\r
+  ListNode = GetFirstNode (ListHead);\r
 \r
-  LocalTokenNumberTable  = IsPeiDb ? gPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
-                                     gPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+  while (ListNode != ListHead) {\r
+    FnTableEntry = CR_FNENTRY_FROM_LISTNODE(ListNode, CALLBACK_FN_ENTRY, Node);\r
 \r
-  TokenNumber = IsPeiDb ? TokenNumber\r
-                        : TokenNumber - PEI_LOCAL_TOKEN_NUMBER;\r
+    FnTableEntry->CallbackFn(Guid, \r
+                    (Guid == NULL) ? TokenNumber : ExTokenNumber,\r
+                    Data,\r
+                    Size);\r
+    \r
+    ListNode = GetNextNode (ListHead, ListNode);\r
+  }\r
   \r
-  InvokeCallbackOnSet (0, NULL, TokenNumber, Data, Size);\r
-\r
-  return SetWorkerByLocalTokenNumber (LocalTokenNumberTable[TokenNumber], Data, Size, PtrType, IsPeiDb);\r
-\r
-}\r
-\r
-\r
-\r
-\r
-\r
-VOID *\r
-ExGetWorker (\r
-  IN CONST EFI_GUID         *Guid,\r
-  IN UINTN                  ExTokenNumber,\r
-  IN UINTN                  GetSize\r
-  ) \r
-{\r
-  EX_PCD_ENTRY_ATTRIBUTE Attr;\r
-\r
-  GetExPcdTokenAttributes (Guid, ExTokenNumber, &Attr);\r
-\r
-  ASSERT ((GetSize == Attr.Size) || (GetSize == 0));\r
-\r
-  return GetWorkerByLocalTokenNumber (Attr.LocalTokenNumberAlias,\r
-                                                Attr.IsPeiDb,\r
-                                                Attr.Size\r
-                                                );\r
+  return;\r
 }\r
 \r
 \r
 \r
 \r
-\r
 EFI_STATUS\r
-ExSetWorker (\r
-  IN UINT32               ExTokenNumber,\r
-  IN CONST EFI_GUID       *Guid,\r
+SetWorker (\r
+  UINTN                   TokenNumber,\r
   VOID                    *Data,\r
-  UINTN                   SetSize,\r
+  UINTN                   Size,\r
   BOOLEAN                 PtrType\r
   )\r
 {\r
-  EX_PCD_ENTRY_ATTRIBUTE Attr;\r
-\r
-  GetExPcdTokenAttributes (Guid, ExTokenNumber, &Attr);\r
-\r
-  ASSERT (!PtrType && (SetSize == Attr.Size));\r
-\r
-  ASSERT (PtrType && (SetSize <= Attr.Size));\r
-\r
-  InvokeCallbackOnSet (ExTokenNumber, Guid, Attr.TokenNumber, Data, Attr.Size);\r
-\r
-  SetWorkerByLocalTokenNumber (Attr.LocalTokenNumberAlias, Data, Attr.Size, PtrType, Attr.IsPeiDb);\r
-\r
-  return EFI_SUCCESS;\r
-  \r
-}\r
-\r
-\r
-\r
-\r
-EFI_STATUS\r
-SetWorkerByLocalTokenNumber (\r
-  UINT32        LocalTokenNumber,\r
-  VOID          *Data,\r
-  UINTN         Size,\r
-  BOOLEAN       PtrType,\r
-  BOOLEAN       IsPeiDb\r
-  )\r
-{\r
+  UINT32              *LocalTokenNumberTable;\r
+  BOOLEAN             IsPeiDb;\r
+  UINT32              LocalTokenNumber;\r
   EFI_GUID            *GuidTable;\r
   UINT16              *StringTable;\r
   EFI_GUID            *Guid;\r
   UINT16              *Name;\r
+  UINTN               VariableOffset;\r
   VOID                *InternalData;\r
   VARIABLE_HEAD       *VariableHead;\r
   UINTN               Offset;\r
   UINT8               *PcdDb;\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
+  ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
+\r
+  if (PtrType) {\r
+    ASSERT (Size <= DxePcdGetSize (TokenNumber + 1));\r
+  } else {\r
+    ASSERT (Size == DxePcdGetSize (TokenNumber + 1));\r
+  }\r
+  \r
+  IsPeiDb = (TokenNumber < PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+\r
+  LocalTokenNumberTable  = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
+                                     mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
 \r
+  if ((TokenNumber < PEI_NEX_TOKEN_NUMBER) ||\r
+      (TokenNumber >= PEI_LOCAL_TOKEN_NUMBER || TokenNumber < (PEI_LOCAL_TOKEN_NUMBER + DXE_NEX_TOKEN_NUMBER))) {\r
+    InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, Size);\r
+  }\r
+\r
+  TokenNumber = IsPeiDb ? TokenNumber\r
+                        : TokenNumber - PEI_LOCAL_TOKEN_NUMBER;\r
+\r
+  LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
+  \r
   if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
     LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, Size, IsPeiDb);\r
   }\r
 \r
   Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
 \r
-  PcdDb = IsPeiDb ? ((UINT8 *) &gPcdDatabase->PeiDb) : ((UINT8 *) &gPcdDatabase->DxeDb);\r
+  PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);\r
 \r
-  StringTable = IsPeiDb ? gPcdDatabase->PeiDb.Init.StringTable :\r
-                          gPcdDatabase->DxeDb.Init.StringTable;\r
+  StringTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.StringTable :\r
+                          mPcdDatabase->DxeDb.Init.StringTable;\r
   \r
   InternalData = PcdDb + Offset;\r
 \r
@@ -420,21 +568,22 @@ SetWorkerByLocalTokenNumber (
     \r
     case PCD_TYPE_STRING:\r
       CopyMem (&StringTable[*((UINT16 *)InternalData)], Data, Size);\r
-      break;\r
+      return EFI_SUCCESS;\r
 \r
     case PCD_TYPE_HII:\r
       //\r
       // Bug Bug: Please implement this\r
       //\r
-      GuidTable   = IsPeiDb ? gPcdDatabase->PeiDb.Init.GuidTable :\r
-                              gPcdDatabase->DxeDb.Init.GuidTable;\r
+      GuidTable   = IsPeiDb ? mPcdDatabase->PeiDb.Init.GuidTable :\r
+                              mPcdDatabase->DxeDb.Init.GuidTable;\r
                               \r
       VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);\r
       \r
       Guid = &(GuidTable[VariableHead->GuidTableIndex]);\r
       Name = &(StringTable[VariableHead->StringIndex]);\r
+      VariableOffset = VariableHead->Offset;\r
 \r
-      return EFI_SUCCESS;\r
+      return SetHiiVariable (Guid, Name, Data, Size, VariableOffset);\r
 \r
     case PCD_TYPE_DATA:\r
       if (PtrType) {\r
@@ -475,6 +624,46 @@ SetWorkerByLocalTokenNumber (
 \r
 \r
 \r
+\r
+\r
+VOID *\r
+ExGetWorker (\r
+  IN CONST EFI_GUID         *Guid,\r
+  IN UINTN                  ExTokenNumber,\r
+  IN UINTN                  GetSize\r
+  ) \r
+{\r
+  return GetWorker(GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber), GetSize);\r
+}\r
+\r
+\r
+\r
+\r
+\r
+EFI_STATUS\r
+ExSetWorker (\r
+  IN UINTN                ExTokenNumber,\r
+  IN CONST EFI_GUID       *Guid,\r
+  VOID                    *Data,\r
+  UINTN                   SetSize,\r
+  BOOLEAN                 PtrType\r
+  )\r
+{\r
+  UINTN                   TokenNumber;\r
+  \r
+  TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber);\r
+\r
+  InvokeCallbackOnSet ((UINT32) ExTokenNumber, Guid, TokenNumber, Data, SetSize);\r
+\r
+  SetWorker (TokenNumber, Data, SetSize, PtrType);\r
+\r
+  return EFI_SUCCESS;\r
+  \r
+}\r
+\r
+\r
+\r
+\r
 EFI_STATUS\r
 SetHiiVariable (\r
   IN  EFI_GUID     *VariableGuid,\r
@@ -499,30 +688,41 @@ SetHiiVariable (
     NULL\r
     );\r
 \r
-  ASSERT (Status == EFI_BUFFER_TOO_SMALL);\r
+  if (Status == EFI_BUFFER_TOO_SMALL) {\r
 \r
-  Buffer = AllocatePool (Size);\r
+    Buffer = AllocatePool (Size);\r
 \r
-  ASSERT (Buffer != NULL);\r
+    ASSERT (Buffer != NULL);\r
 \r
-  Status = EfiGetVariable (\r
-    VariableName,\r
-    VariableGuid,\r
-    &Attribute,\r
-    &Size,\r
-    Buffer\r
-    );\r
+    Status = EfiGetVariable (\r
+      VariableName,\r
+      VariableGuid,\r
+      &Attribute,\r
+      &Size,\r
+      Buffer\r
+      );\r
+    \r
+    ASSERT_EFI_ERROR (Status);\r
 \r
+    CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
+\r
+  } else {\r
 \r
-  CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
+    Attribute = EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS;\r
+    Size = DataSize + Offset;\r
+    Buffer = AllocateZeroPool (Size);\r
+    ASSERT (Buffer != NULL);\r
+    CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
+    \r
+  }\r
 \r
   return EfiSetVariable (\r
-    VariableName,\r
-    VariableGuid,\r
-    Attribute,\r
-    Size,\r
-    Buffer\r
-    );\r
+            VariableName,\r
+            VariableGuid,\r
+            Attribute,\r
+            Size,\r
+            Buffer\r
+            );\r
 \r
 }\r
 \r
@@ -530,56 +730,51 @@ SetHiiVariable (
 \r
 \r
 \r
-VOID\r
-GetExPcdTokenAttributes (\r
+UINTN           \r
+GetExPcdTokenNumber (\r
   IN CONST EFI_GUID             *Guid,\r
-  IN UINT32                     ExTokenNumber,\r
-  OUT EX_PCD_ENTRY_ATTRIBUTE    *ExAttr\r
+  IN UINT32                     ExTokenNumber\r
   )\r
 {\r
   UINT32              i;\r
   DYNAMICEX_MAPPING   *ExMap;\r
   EFI_GUID            *GuidTable;\r
-  UINT16              *SizeTable;\r
+  EFI_GUID            *MatchGuid;\r
+  UINTN               MatchGuidIdx;\r
+\r
+  ExMap       = mPcdDatabase->PeiDb.Init.ExMapTable;\r
+  GuidTable   = mPcdDatabase->PeiDb.Init.GuidTable;\r
+  \r
+  MatchGuid   = ScanGuid (GuidTable, sizeof(mPcdDatabase->PeiDb.Init.GuidTable), Guid);\r
+  ASSERT (MatchGuid != NULL);\r
 \r
-  ExMap       = gPcdDatabase->PeiDb.Init.ExMapTable;\r
-  GuidTable   = gPcdDatabase->PeiDb.Init.GuidTable;\r
-  SizeTable   = gPcdDatabase->PeiDb.Init.SizeTable;\r
+  MatchGuidIdx = MatchGuid - GuidTable;\r
   \r
   for (i = 0; i < PEI_EXMAPPING_TABLE_SIZE; i++) {\r
     if ((ExTokenNumber == ExMap[i].ExTokenNumber) &&\r
-        CompareGuid (Guid, (CONST EFI_GUID *) &GuidTable[ExMap[i].ExGuidIndex])\r
-      ) {\r
-\r
-        ExAttr->IsPeiDb               = TRUE;\r
-        ExAttr->Size                  = SizeTable[i + PEI_NEX_TOKEN_NUMBER];\r
-        ExAttr->TokenNumber           = i + PEI_NEX_TOKEN_NUMBER;\r
-        ExAttr->LocalTokenNumberAlias = ExMap[i].LocalTokenNumber;\r
-        return;\r
+        (MatchGuidIdx == ExMap[i].ExGuidIndex)) {\r
+        return ExMap[i].LocalTokenNumber;\r
 \r
     }\r
   }\r
   \r
-  ExMap       = gPcdDatabase->DxeDb.Init.ExMapTable;\r
-  GuidTable   = gPcdDatabase->DxeDb.Init.GuidTable;\r
-  SizeTable   = gPcdDatabase->DxeDb.Init.SizeTable;\r
+  ExMap       = mPcdDatabase->DxeDb.Init.ExMapTable;\r
+  GuidTable   = mPcdDatabase->DxeDb.Init.GuidTable;\r
+\r
+  MatchGuid   = ScanGuid (GuidTable, sizeof(mPcdDatabase->DxeDb.Init.GuidTable), Guid);\r
+  ASSERT (MatchGuid != NULL);\r
+\r
+  MatchGuidIdx = MatchGuid - GuidTable;\r
   \r
   for (i = 0; i < DXE_EXMAPPING_TABLE_SIZE; i++) {\r
     if ((ExTokenNumber == ExMap[i].ExTokenNumber) &&\r
-         CompareGuid (Guid, (CONST EFI_GUID *) &GuidTable[ExMap[i].ExGuidIndex])\r
-      ) {\r
-\r
-        ExAttr->IsPeiDb               = FALSE;\r
-        ExAttr->Size                  = SizeTable[i + DXE_NEX_TOKEN_NUMBER];\r
-        ExAttr->TokenNumber           = i + DXE_NEX_TOKEN_NUMBER;\r
-        ExAttr->LocalTokenNumberAlias = ExMap[i].LocalTokenNumber;\r
-        return;\r
-\r
+         (MatchGuidIdx == ExMap[i].ExGuidIndex)) {\r
+        return ExMap[i].LocalTokenNumber + PEI_LOCAL_TOKEN_NUMBER;\r
     }\r
   }\r
 \r
   ASSERT (FALSE);\r
 \r
-  return;\r
+  return 0;\r
 }\r
 \r