]> git.proxmox.com Git - mirror_edk2.git/blobdiff - EdkModulePkg/Universal/PCD/Dxe/Service.c
Updated the copyright year.
[mirror_edk2.git] / EdkModulePkg / Universal / PCD / Dxe / Service.c
index 4b86cd7795ca38eab1f7871d58e2955bc4a97916..51e16624efcd51aa35eab70c9c34d12b98a84bfe 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
-Private functions used by PCD DXE driver.s\r
+Private functions used by PCD DXE driver.\r
 \r
-Copyright (c) 2006, Intel Corporation\r
+Copyright (c) 2006 - 2007, Intel Corporation\r
 All rights reserved. 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
@@ -17,11 +17,6 @@ Module Name: Service.c
 #include "Service.h"\r
 \r
 \r
-//\r
-// Build Tool will generate DXE_PCD_DB_INIT_VALUE in Autogen.h\r
-// Compression Algorithm will take care of the size optimization.\r
-//\r
-\r
 PCD_DATABASE * mPcdDatabase;\r
 \r
 LIST_ENTRY *mCallbackFnTable;\r
@@ -34,30 +29,54 @@ GetWorker (
 {\r
   UINT32              *LocalTokenNumberTable;\r
   UINT16              *SizeTable;\r
-  BOOLEAN             IsPeiDb;\r
-  UINTN               Size;\r
-  UINT32              Offset;\r
   EFI_GUID            *GuidTable;\r
   UINT16              *StringTable;\r
   EFI_GUID            *Guid;\r
   UINT16              *Name;\r
   VARIABLE_HEAD       *VariableHead;\r
-  EFI_STATUS          Status;\r
-  UINTN               DataSize;\r
-  VOID                *Data;\r
+  UINT8               *VaraiableDefaultBuffer;\r
+  UINT8               *Data;\r
   VPD_HEAD            *VpdHead;\r
   UINT8               *PcdDb;\r
-  UINT16              StringTableIdx;      \r
+  VOID                *RetPtr;\r
+  UINTN               MaxSize;\r
+  UINTN               TmpTokenNumber;\r
+  UINTN               DataSize;\r
+  EFI_STATUS          Status;\r
   UINT32              LocalTokenNumber;\r
+  UINT32              Offset;\r
+  UINT16              StringTableIdx;      \r
+  BOOLEAN             IsPeiDb;\r
 \r
+  //\r
+  // Aquire lock to prevent reentrance from TPL_CALLBACK level\r
+  //\r
+  EfiAcquireLock (&mPcdDatabaseLock);\r
 \r
-  ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
-\r
-  Size = DxePcdGetSize (TokenNumber);\r
-  ASSERT (GetSize == Size || GetSize == 0);\r
+  RetPtr = NULL;\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
+  TmpTokenNumber = TokenNumber;\r
   \r
-  IsPeiDb = (TokenNumber <= PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+  //\r
+  // PCD_TOTAL_TOKEN_NUMBER is a auto-generated constant.\r
+  // It could be zero. EBC compiler is very choosy. It may\r
+  // report warning. So we add 1 in each size of the \r
+  // comparison.\r
+  //\r
+  ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);\r
+\r
+  ASSERT ((GetSize == DxePcdGetSize (TokenNumber + 1)) || (GetSize == 0));\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
+  IsPeiDb = (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? TRUE : FALSE;\r
 \r
   LocalTokenNumberTable  = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
                                      mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
@@ -71,7 +90,12 @@ GetWorker (
   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
+    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
 \r
   PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);\r
@@ -80,10 +104,11 @@ GetWorker (
   \r
   Offset     = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
   \r
-  switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
+  switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
     case PCD_TYPE_VPD:\r
       VpdHead = (VPD_HEAD *) ((UINT8 *) PcdDb + Offset);\r
-      return (VOID *) (UINTN) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
+      RetPtr = (VOID *) (UINTN) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
+      break;\r
       \r
     case PCD_TYPE_HII:\r
       GuidTable   = IsPeiDb ? mPcdDatabase->PeiDb.Init.GuidTable :\r
@@ -93,19 +118,40 @@ GetWorker (
       \r
       Guid = &(GuidTable[VariableHead->GuidTableIndex]);\r
       Name = &(StringTable[VariableHead->StringIndex]);\r
+      VaraiableDefaultBuffer = (UINT8 *) PcdDb + VariableHead->DefaultValueOffset;\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
+        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
+        }\r
+        CopyMem (VaraiableDefaultBuffer, Data + VariableHead->Offset, GetSize);\r
+        FreePool (Data);\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
+      //\r
+      // If the operation is not successful, \r
+      // Return 1) either the default value specified by Platform Integrator \r
+      //        2) Or the value Set by a PCD set operation.\r
+      //\r
+      RetPtr = (VOID *) VaraiableDefaultBuffer;\r
+      break;\r
 \r
     case PCD_TYPE_STRING:\r
       StringTableIdx = (UINT16) *((UINT8 *) PcdDb + Offset);\r
-      return (VOID *) &StringTable[StringTableIdx];\r
+      RetPtr = (VOID *) &StringTable[StringTableIdx];\r
+      break;\r
 \r
     case PCD_TYPE_DATA:\r
-      return (VOID *) ((UINT8 *) PcdDb + Offset);\r
+      RetPtr = (VOID *) ((UINT8 *) PcdDb + Offset);\r
       break;\r
 \r
     default:\r
@@ -114,9 +160,9 @@ GetWorker (
       \r
   }\r
 \r
-  ASSERT (FALSE);\r
-      \r
-  return NULL;\r
+  EfiReleaseLock (&mPcdDatabaseLock);\r
+  \r
+  return RetPtr;\r
   \r
 }\r
 \r
@@ -137,6 +183,13 @@ DxeRegisterCallBackWorker (
     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
@@ -180,6 +233,13 @@ DxeUnRegisterCallBackWorker (
     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
@@ -205,14 +265,14 @@ DxeUnRegisterCallBackWorker (
 \r
 \r
 \r
-UINTN           \r
+EFI_STATUS\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
+  IN      CONST EFI_GUID         *Guid,\r
+  IN OUT  UINTN                  *TokenNumber,\r
+  IN      EFI_GUID               *GuidTable,\r
+  IN      UINTN                  SizeOfGuidTable,\r
+  IN      DYNAMICEX_MAPPING      *ExMapTable,\r
+  IN      UINTN                  SizeOfExMapTable\r
   )\r
 {\r
   EFI_GUID         *MatchGuid;\r
@@ -222,7 +282,7 @@ ExGetNextTokeNumber (
 \r
   MatchGuid = ScanGuid (GuidTable, SizeOfGuidTable, Guid);\r
   if (MatchGuid == NULL) {\r
-    return PCD_INVALID_TOKEN_NUMBER;\r
+    return EFI_NOT_FOUND;\r
   }\r
 \r
   Found = FALSE;\r
@@ -235,35 +295,39 @@ ExGetNextTokeNumber (
   }\r
 \r
   if (Found) {\r
-    if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
-      return ExMapTable[Idx].ExTokenNumber;\r
+    if (*TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
+      *TokenNumber = ExMapTable[Idx].ExTokenNumber;\r
+      return EFI_SUCCESS;\r
     }\r
-    \r
+\r
     for ( ; Idx < SizeOfExMapTable; Idx++) {\r
-      if (ExMapTable[Idx].ExTokenNumber == TokenNumber) {\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
+          *TokenNumber = PCD_INVALID_TOKEN_NUMBER;\r
+          return EFI_SUCCESS;\r
         } else if (ExMapTable[Idx].ExGuidIndex == GuidTableIdx) {\r
           //\r
           // Found the next match\r
           //\r
-          return ExMapTable[Idx].ExTokenNumber;\r
+          *TokenNumber = ExMapTable[Idx].ExTokenNumber;\r
+          return EFI_SUCCESS;\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
+          *TokenNumber = PCD_INVALID_TOKEN_NUMBER;\r
+          return EFI_SUCCESS;\r
         }\r
       }\r
     }\r
   }\r
   \r
-  return PCD_INVALID_TOKEN_NUMBER;\r
+  return EFI_NOT_FOUND;\r
 }\r
   \r
 \r
@@ -282,7 +346,6 @@ BuildPcdDxeDataBase (
   ASSERT (mPcdDatabase != NULL);\r
 \r
   GuidHob = GetFirstGuidHob (&gPcdDataBaseHobGuid);\r
-\r
   if (GuidHob != NULL) {\r
 \r
     //\r
@@ -311,11 +374,12 @@ BuildPcdDxeDataBase (
   // Initialized the Callback Function Table\r
   //\r
 \r
-  if (PCD_TOTAL_TOKEN_NUMBER != 0) {\r
-    mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER * sizeof (LIST_ENTRY));\r
-  }\r
+  mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER * sizeof (LIST_ENTRY));\r
   \r
-  for (Idx = 0; Idx < PCD_TOTAL_TOKEN_NUMBER; Idx++) {\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
+  for (Idx = 0; Idx + 1 < PCD_TOTAL_TOKEN_NUMBER + 1; Idx++) {\r
     InitializeListHead (&mCallbackFnTable[Idx]);\r
   }\r
     \r
@@ -328,37 +392,44 @@ EFI_STATUS
 GetHiiVariable (\r
   IN  EFI_GUID      *VariableGuid,\r
   IN  UINT16        *VariableName,\r
-  OUT VOID          ** VariableData,\r
+  OUT UINT8         **VariableData,\r
   OUT UINTN         *VariableSize\r
   )\r
 {\r
   UINTN      Size;\r
   EFI_STATUS Status;\r
-  VOID       *Buffer;\r
+  UINT8      *Buffer;\r
 \r
-  Status = EfiGetVariable (\r
+  Size = 0;\r
+  Buffer = NULL;\r
+  \r
+  Status = gRT->GetVariable (\r
     (UINT16 *)VariableName,\r
     VariableGuid,\r
     NULL,\r
     &Size,\r
-    NULL\r
-    );\r
-  ASSERT (Status == EFI_BUFFER_TOO_SMALL);\r
-\r
-  Buffer = AllocatePool (Size);\r
-\r
-  ASSERT (Buffer != NULL);\r
-\r
-  Status = EfiGetVariable (\r
-    VariableName,\r
-    VariableGuid,\r
-    NULL,\r
-    &Size,\r
     Buffer\r
     );\r
+  \r
+  if (Status == EFI_BUFFER_TOO_SMALL) {\r
+    Buffer = (UINT8 *) AllocatePool (Size);\r
+\r
+    ASSERT (Buffer != NULL);\r
+\r
+    Status = gRT->GetVariable (\r
+      VariableName,\r
+      VariableGuid,\r
+      NULL,\r
+      &Size,\r
+      Buffer\r
+      );\r
+\r
+    ASSERT (Status == EFI_SUCCESS);\r
+    *VariableData = Buffer;\r
+    *VariableSize = Size;\r
+  }\r
 \r
   return Status;\r
-\r
 }\r
 \r
 \r
@@ -395,7 +466,7 @@ GetSkuEnabledTokenNumber (
   }\r
   ASSERT (i < SkuIdTable[0]);\r
 \r
-  switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
+  switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
     case PCD_TYPE_VPD:\r
       Value = (UINT8 *) &(((VPD_HEAD *) Value)[i]);\r
       return (UINT32) ((Value - PcdDb) | PCD_TYPE_VPD);\r
@@ -403,11 +474,15 @@ GetSkuEnabledTokenNumber (
     case PCD_TYPE_HII:\r
       Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[i]);\r
       return (UINT32) ((Value - PcdDb) | PCD_TYPE_HII);\r
+\r
+    case PCD_TYPE_STRING:\r
+      Value = (UINT8 *) &(((STRING_HEAD *) Value)[i]);\r
+      return (UINT32) ((Value - PcdDb) | PCD_TYPE_STRING);\r
       \r
     case PCD_TYPE_DATA:\r
       Value += Size * i;\r
       return (UINT32) (Value - PcdDb);\r
-      \r
+\r
     default:\r
       ASSERT (FALSE);\r
   }\r
@@ -435,6 +510,13 @@ InvokeCallbackOnSet (
   LIST_ENTRY              *ListHead;\r
   LIST_ENTRY              *ListNode;\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
@@ -453,14 +535,23 @@ InvokeCallbackOnSet (
 }\r
 \r
 \r
+EFI_STATUS\r
+SetValueWorker (\r
+  IN UINTN                   TokenNumber,\r
+  IN VOID                    *Data,\r
+  IN UINTN                   Size\r
+  )\r
+{\r
+  return SetWorker (TokenNumber, Data, &Size, FALSE);\r
+}\r
 \r
 \r
 EFI_STATUS\r
 SetWorker (\r
-  UINTN                   TokenNumber,\r
-  VOID                    *Data,\r
-  UINTN                   Size,\r
-  BOOLEAN                 PtrType\r
+  IN          UINTN                   TokenNumber,\r
+  IN          VOID                    *Data,\r
+  IN OUT      UINTN                   *Size,\r
+  IN          BOOLEAN                 PtrType\r
   )\r
 {\r
   UINT32              *LocalTokenNumberTable;\r
@@ -470,28 +561,53 @@ SetWorker (
   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
+  EFI_STATUS          Status;\r
+  UINTN               MaxSize;\r
+  UINTN               TmpTokenNumber;\r
+\r
+  //\r
+  // Aquire lock to prevent reentrance from TPL_CALLBACK level\r
+  //\r
+  EfiAcquireLock (&mPcdDatabaseLock);\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
+  TmpTokenNumber = TokenNumber;\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
 \r
-  if (PtrType) {\r
-    ASSERT (Size <= DxePcdGetSize (TokenNumber));\r
-  } else {\r
-    ASSERT (Size == DxePcdGetSize (TokenNumber));\r
+  ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);\r
+\r
+  if (!PtrType) {\r
+    ASSERT (*Size == DxePcdGetSize (TokenNumber + 1));\r
   }\r
   \r
-  IsPeiDb = (TokenNumber <= PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\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
+  IsPeiDb = (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? 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, Data, Size);\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
+  if ((TokenNumber + 1 < PEI_NEX_TOKEN_NUMBER + 1) ||\r
+      (TokenNumber + 1 >= PEI_LOCAL_TOKEN_NUMBER + 1 || TokenNumber + 1 < (PEI_LOCAL_TOKEN_NUMBER + DXE_NEX_TOKEN_NUMBER + 1))) {\r
+    InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, *Size);\r
   }\r
 \r
   TokenNumber = IsPeiDb ? TokenNumber\r
@@ -500,7 +616,12 @@ SetWorker (
   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
+    if (PtrType) {\r
+      GetPtrTypeSize (TmpTokenNumber, &MaxSize);\r
+    } else {\r
+      MaxSize = *Size;\r
+    }\r
+    LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);\r
   }\r
 \r
   Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
@@ -512,19 +633,29 @@ SetWorker (
   \r
   InternalData = PcdDb + Offset;\r
 \r
-  switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
+  switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
     case PCD_TYPE_VPD:\r
       ASSERT (FALSE);\r
-      return EFI_INVALID_PARAMETER;\r
+      Status = EFI_INVALID_PARAMETER;\r
+      break;\r
     \r
     case PCD_TYPE_STRING:\r
-      CopyMem (&StringTable[*((UINT16 *)InternalData)], Data, Size);\r
+      if (SetPtrTypeSize (TmpTokenNumber, Size)) {\r
+        CopyMem (&StringTable[*((UINT16 *)InternalData)], Data, *Size);\r
+        Status = EFI_SUCCESS;\r
+      } else {\r
+        Status = EFI_INVALID_PARAMETER;\r
+      }\r
       break;\r
 \r
     case PCD_TYPE_HII:\r
-      //\r
-      // Bug Bug: Please implement this\r
-      //\r
+      if (PtrType) {\r
+        if (!SetPtrTypeSize (TmpTokenNumber, Size)) {\r
+          Status = EFI_INVALID_PARAMETER;\r
+          break;\r
+        }\r
+      }\r
+      \r
       GuidTable   = IsPeiDb ? mPcdDatabase->PeiDb.Init.GuidTable :\r
                               mPcdDatabase->DxeDb.Init.GuidTable;\r
                               \r
@@ -532,44 +663,61 @@ SetWorker (
       \r
       Guid = &(GuidTable[VariableHead->GuidTableIndex]);\r
       Name = &(StringTable[VariableHead->StringIndex]);\r
+      VariableOffset = VariableHead->Offset;\r
 \r
-      return EFI_SUCCESS;\r
+      Status = SetHiiVariable (Guid, Name, Data, *Size, VariableOffset);\r
 \r
+      if (EFI_NOT_FOUND == Status) {\r
+        CopyMem (PcdDb + VariableHead->DefaultValueOffset, Data, *Size);\r
+        Status = EFI_SUCCESS;\r
+      } \r
+      break;\r
+      \r
     case PCD_TYPE_DATA:\r
       if (PtrType) {\r
-        CopyMem (InternalData, Data, Size);\r
-        return EFI_SUCCESS;\r
+        if (SetPtrTypeSize (TmpTokenNumber, Size)) {\r
+          CopyMem (InternalData, Data, *Size);\r
+          Status = EFI_SUCCESS;\r
+        } else {\r
+          Status = EFI_INVALID_PARAMETER;\r
+        }\r
+        break;\r
       }\r
 \r
-      switch (Size) {\r
+      Status = EFI_SUCCESS;\r
+      switch (*Size) {\r
         case sizeof(UINT8):\r
           *((UINT8 *) InternalData) = *((UINT8 *) Data);\r
-          return EFI_SUCCESS;\r
+          break;\r
 \r
         case sizeof(UINT16):\r
           *((UINT16 *) InternalData) = *((UINT16 *) Data);\r
-          return EFI_SUCCESS;\r
+          break;\r
 \r
         case sizeof(UINT32):\r
           *((UINT32 *) InternalData) = *((UINT32 *) Data);\r
-          return EFI_SUCCESS;\r
+          break;\r
 \r
         case sizeof(UINT64):\r
           *((UINT64 *) InternalData) = *((UINT64 *) Data);\r
-          return EFI_SUCCESS;\r
+          break;\r
 \r
         default:\r
           ASSERT (FALSE);\r
-          return EFI_NOT_FOUND;\r
+          Status = EFI_NOT_FOUND;\r
+          break;\r
       }\r
+      break;\r
 \r
     default:\r
       ASSERT (FALSE);\r
+      Status = EFI_NOT_FOUND;\r
       break;\r
     }\r
-      \r
-  ASSERT (FALSE);\r
-  return EFI_NOT_FOUND;\r
+\r
+  EfiReleaseLock (&mPcdDatabaseLock);\r
+  \r
+  return Status;\r
 }\r
 \r
 \r
@@ -589,26 +737,35 @@ ExGetWorker (
 \r
 \r
 \r
+EFI_STATUS\r
+ExSetValueWorker (\r
+  IN          UINTN                ExTokenNumber,\r
+  IN          CONST EFI_GUID       *Guid,\r
+  IN          VOID                 *Data,\r
+  IN          UINTN                SetSize\r
+  )\r
+{\r
+  return ExSetWorker (ExTokenNumber, Guid, Data, &SetSize, FALSE);\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
+  IN          UINTN                ExTokenNumber,\r
+  IN          CONST EFI_GUID       *Guid,\r
+  IN          VOID                 *Data,\r
+  IN OUT      UINTN                *SetSize,\r
+  IN          BOOLEAN              PtrType\r
   )\r
 {\r
   UINTN                   TokenNumber;\r
   \r
   TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber);\r
 \r
-  InvokeCallbackOnSet ((UINT32) ExTokenNumber, Guid, TokenNumber, Data, SetSize);\r
+  InvokeCallbackOnSet ((UINT32) ExTokenNumber, Guid, TokenNumber, Data, *SetSize);\r
 \r
-  SetWorker (TokenNumber, Data, SetSize, PtrType);\r
+  return SetWorker (TokenNumber, Data, SetSize, PtrType);\r
 \r
-  return EFI_SUCCESS;\r
-  \r
 }\r
 \r
 \r
@@ -630,39 +787,51 @@ SetHiiVariable (
 \r
   Size = 0;\r
 \r
-  Status = EfiGetVariable (\r
+  Status = gRT->GetVariable (\r
     (UINT16 *)VariableName,\r
     VariableGuid,\r
-    &Attribute,\r
+    NULL,\r
     &Size,\r
     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 = gRT->GetVariable (\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
-  CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
+    Status = gRT->SetVariable (\r
+              VariableName,\r
+              VariableGuid,\r
+              Attribute,\r
+              Size,\r
+              Buffer\r
+              );\r
 \r
-  return EfiSetVariable (\r
-    VariableName,\r
-    VariableGuid,\r
-    Attribute,\r
-    Size,\r
-    Buffer\r
-    );\r
+    FreePool (Buffer);\r
+    return Status;\r
 \r
+  } \r
+  \r
+  //\r
+  // If we drop to here, we don't have a Variable entry in\r
+  // the variable service yet. So, we will save the data\r
+  // in the PCD Database's volatile area.\r
+  //\r
+  return Status;\r
 }\r
 \r
 \r
@@ -681,19 +850,23 @@ GetExPcdTokenNumber (
   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
+  if (!PEI_DATABASE_EMPTY) {\r
+    ExMap       = mPcdDatabase->PeiDb.Init.ExMapTable;\r
+    GuidTable   = mPcdDatabase->PeiDb.Init.GuidTable;\r
+    \r
+    MatchGuid   = ScanGuid (GuidTable, sizeof(mPcdDatabase->PeiDb.Init.GuidTable), Guid);\r
+    \r
+    if (MatchGuid != NULL) {\r
 \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
+      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
 \r
+        }\r
+      }\r
     }\r
   }\r
   \r
@@ -701,6 +874,10 @@ GetExPcdTokenNumber (
   GuidTable   = mPcdDatabase->DxeDb.Init.GuidTable;\r
 \r
   MatchGuid   = ScanGuid (GuidTable, sizeof(mPcdDatabase->DxeDb.Init.GuidTable), 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
+  //\r
   ASSERT (MatchGuid != NULL);\r
 \r
   MatchGuidIdx = MatchGuid - GuidTable;\r
@@ -708,7 +885,7 @@ GetExPcdTokenNumber (
   for (i = 0; i < DXE_EXMAPPING_TABLE_SIZE; i++) {\r
     if ((ExTokenNumber == ExMap[i].ExTokenNumber) &&\r
          (MatchGuidIdx == ExMap[i].ExGuidIndex)) {\r
-        return ExMap[i].LocalTokenNumber + PEI_LOCAL_TOKEN_NUMBER;\r
+        return ExMap[i].LocalTokenNumber;\r
     }\r
   }\r
 \r
@@ -717,3 +894,264 @@ GetExPcdTokenNumber (
   return 0;\r
 }\r
 \r
+\r
+\r
+SKU_ID *\r
+GetSkuIdArray (\r
+  IN    UINTN             LocalTokenNumberTableIdx,\r
+  IN    BOOLEAN           IsPeiPcd\r
+  )\r
+{\r
+  SKU_HEAD  *SkuHead;\r
+  UINTN     LocalTokenNumber;\r
+  UINT8     *Database;\r
+\r
+  if (IsPeiPcd) {\r
+    LocalTokenNumber = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+    Database         = (UINT8 *) &mPcdDatabase->PeiDb;\r
+  } else {\r
+    LocalTokenNumber = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx - PEI_LOCAL_TOKEN_NUMBER];\r
+    Database         = (UINT8 *) &mPcdDatabase->DxeDb;\r
+  }\r
+\r
+  ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) != 0);\r
+\r
+  SkuHead = (SKU_HEAD *) ((UINT8 *)Database + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));\r
+\r
+  return (SKU_ID *) (Database + SkuHead->SkuIdTableOffset);\r
+  \r
+}\r
+\r
+\r
+\r
+UINTN\r
+GetSizeTableIndexA (\r
+  IN UINTN        LocalTokenNumberTableIdx,\r
+  IN UINT32       *LocalTokenNumberTable,\r
+  IN BOOLEAN      IsPeiDb\r
+  )\r
+{\r
+  UINTN       i;\r
+  UINTN       SizeTableIdx;\r
+  UINTN       LocalTokenNumber;\r
+  SKU_ID      *SkuIdTable;\r
+  \r
+  SizeTableIdx = 0;\r
+\r
+  for (i=0; i<LocalTokenNumberTableIdx; i++) {\r
+    LocalTokenNumber = LocalTokenNumberTable[i];\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
+          //\r
+          // We have only one entry for VPD enabled PCD entry:\r
+          // 1) MAX Size.\r
+          // We consider current size is equal to MAX size.\r
+          //\r
+          SizeTableIdx++;\r
+      } else {\r
+        if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
+          //\r
+          // We have only two entry for Non-Sku enabled PCD entry:\r
+          // 1) MAX SIZE\r
+          // 2) Current Size\r
+          //\r
+          SizeTableIdx += 2;\r
+        } else {\r
+          //\r
+          // We have these entry for SKU enabled PCD entry\r
+          // 1) MAX SIZE\r
+          // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
+          //\r
+          SkuIdTable = GetSkuIdArray (i, IsPeiDb);\r
+          SizeTableIdx += (UINTN)*SkuIdTable + 1;\r
+        }\r
+      }\r
+    }\r
+\r
+  }\r
+\r
+  return SizeTableIdx;\r
+}\r
+\r
+\r
+\r
+\r
+UINTN\r
+GetSizeTableIndex (\r
+  IN    UINTN             LocalTokenNumberTableIdx,\r
+  IN    BOOLEAN           IsPeiDb\r
+  )\r
+{\r
+  UINT32 *LocalTokenNumberTable;\r
+  \r
+  if (IsPeiDb) {\r
+    LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;\r
+  } else {\r
+    LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+  }\r
+  return GetSizeTableIndexA (LocalTokenNumberTableIdx, \r
+                              LocalTokenNumberTable,\r
+                              IsPeiDb);\r
+}\r
+\r
+\r
+\r
+UINTN\r
+GetPtrTypeSize (\r
+  IN    UINTN             LocalTokenNumberTableIdx,\r
+  OUT   UINTN             *MaxSize\r
+  )\r
+{\r
+  INTN        SizeTableIdx;\r
+  UINTN       LocalTokenNumber;\r
+  SKU_ID      *SkuIdTable;\r
+  SIZE_INFO   *SizeTable;\r
+  UINTN       i;\r
+  BOOLEAN     IsPeiDb;\r
+  UINT32      *LocalTokenNumberTable;\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
+  IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+\r
+\r
+  if (IsPeiDb) {\r
+    LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;\r
+    SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;\r
+  } else {\r
+    LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;\r
+    LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+    SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;\r
+  }\r
+\r
+  LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+\r
+  ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);\r
+  \r
+  SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);\r
+\r
+  *MaxSize = SizeTable[SizeTableIdx];\r
+  //\r
+  // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type \r
+  // PCD entry.\r
+  //\r
+  if (LocalTokenNumber & PCD_TYPE_VPD) {\r
+      //\r
+      // We have only one entry for VPD enabled PCD entry:\r
+      // 1) MAX Size.\r
+      // We consider current size is equal to MAX size.\r
+      //\r
+      return *MaxSize;\r
+  } else {\r
+    if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
+      //\r
+      // We have only two entry for Non-Sku enabled PCD entry:\r
+      // 1) MAX SIZE\r
+      // 2) Current Size\r
+      //\r
+      return SizeTable[SizeTableIdx + 1];\r
+    } else {\r
+      //\r
+      // We have these entry for SKU enabled PCD entry\r
+      // 1) MAX SIZE\r
+      // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
+      //\r
+      SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);\r
+      for (i = 0; i < SkuIdTable[0]; i++) {\r
+        if (SkuIdTable[1 + i] == mPcdDatabase->PeiDb.Init.SystemSkuId) {\r
+          return SizeTable[SizeTableIdx + 1 + i];\r
+        }\r
+      }\r
+      return SizeTable[SizeTableIdx + 1];\r
+    }\r
+  }\r
+}\r
+\r
+\r
+\r
+BOOLEAN\r
+SetPtrTypeSize (\r
+  IN          UINTN             LocalTokenNumberTableIdx,\r
+  IN    OUT   UINTN             *CurrentSize\r
+  )\r
+{\r
+  INTN        SizeTableIdx;\r
+  UINTN       LocalTokenNumber;\r
+  SKU_ID      *SkuIdTable;\r
+  SIZE_INFO   *SizeTable;\r
+  UINTN       i;\r
+  UINTN       MaxSize;\r
+  BOOLEAN     IsPeiDb;\r
+  UINT32      *LocalTokenNumberTable;\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
+  IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+\r
+  if (IsPeiDb) {\r
+    LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;\r
+    SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;\r
+  } else {\r
+    LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;\r
+    LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+    SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;\r
+  }\r
+\r
+  LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+\r
+  ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);\r
+  \r
+  SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);\r
+\r
+  MaxSize = SizeTable[SizeTableIdx];\r
+  //\r
+  // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type \r
+  // PCD entry.\r
+  //\r
+  if (LocalTokenNumber & PCD_TYPE_VPD) {\r
+      //\r
+      // We shouldn't come here as we don't support SET for VPD\r
+      //\r
+      ASSERT (FALSE);\r
+      return FALSE;\r
+  } else {\r
+    if ((*CurrentSize > MaxSize) ||\r
+      (*CurrentSize == MAX_ADDRESS)) {\r
+       *CurrentSize = MaxSize;\r
+       return FALSE;\r
+    } \r
+    \r
+    if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
+      //\r
+      // We have only two entry for Non-Sku enabled PCD entry:\r
+      // 1) MAX SIZE\r
+      // 2) Current Size\r
+      //\r
+      SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;\r
+      return TRUE;\r
+    } else {\r
+      //\r
+      // We have these entry for SKU enabled PCD entry\r
+      // 1) MAX SIZE\r
+      // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
+      //\r
+      SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);\r
+      for (i = 0; i < SkuIdTable[0]; i++) {\r
+        if (SkuIdTable[1 + i] == mPcdDatabase->PeiDb.Init.SystemSkuId) {\r
+          SizeTable[SizeTableIdx + 1 + i] = (SIZE_INFO) *CurrentSize;\r
+          return TRUE;\r
+        }\r
+      }\r
+      SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;\r
+      return TRUE;\r
+    }\r
+  }\r
+}\r
+\r