[mirror_edk2.git] / MdeModulePkg / Universal / PCD / Dxe / Service.c

/** @file\r
Private functions used by PCD DXE driver.\r
\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
http://opensource.org/licenses/bsd-license.php\r
\r
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
\r
\r
PCD_DATABASE * mPcdDatabase;\r
\r
LIST_ENTRY *mCallbackFnTable;\r
\r
VOID *\r
GetWorker (\r
  UINTN             TokenNumber,\r
  UINTN             GetSize\r
  )\r
{\r
  UINT32              *LocalTokenNumberTable;\r
  EFI_GUID            *GuidTable;\r
  UINT16              *StringTable;\r
  EFI_GUID            *Guid;\r
  UINT16              *Name;\r
  VARIABLE_HEAD       *VariableHead;\r
  UINT8               *VaraiableDefaultBuffer;\r
  UINT8               *Data;\r
  VPD_HEAD            *VpdHead;\r
  UINT8               *PcdDb;\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
  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
  //\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 = (BOOLEAN) ((TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? TRUE : FALSE);\r
\r
  LocalTokenNumberTable  = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
                                     mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\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
    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
  StringTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.StringTable :\r
                          mPcdDatabase->DxeDb.Init.StringTable;\r
  \r
  Offset     = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
  \r
  switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
    case PCD_TYPE_VPD:\r
      VpdHead = (VPD_HEAD *) ((UINT8 *) PcdDb + 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
                              mPcdDatabase->DxeDb.Init.GuidTable;\r
                              \r
      VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);\r
      \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
      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
      RetPtr = (VOID *) &StringTable[StringTableIdx];\r
      break;\r
\r
    case PCD_TYPE_DATA:\r
      RetPtr = (VOID *) ((UINT8 *) PcdDb + Offset);\r
      break;\r
\r
    default:\r
      ASSERT (FALSE);\r
      break;\r
      \r
  }\r
\r
  EfiReleaseLock (&mPcdDatabaseLock);\r
  \r
  return RetPtr;\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
  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 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
DxeUnRegisterCallBackWorker (\r
  IN  UINTN                   TokenNumber,\r
  IN  CONST GUID              *Guid, OPTIONAL\r
  IN  PCD_PROTOCOL_CALLBACK   CallBackFunction\r
)\r
{\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
\r
EFI_STATUS\r
ExGetNextTokeNumber (\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
  UINTN            Idx;\r
  UINTN            GuidTableIdx;\r
  BOOLEAN          Found;\r
\r
  MatchGuid = ScanGuid (GuidTable, SizeOfGuidTable, Guid);\r
  if (MatchGuid == NULL) {\r
    return EFI_NOT_FOUND;\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
      *TokenNumber = ExMapTable[Idx].ExTokenNumber;\r
      return EFI_SUCCESS;\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
          *TokenNumber = PCD_INVALID_TOKEN_NUMBER;\r
          return EFI_SUCCESS;\r
        } else if (ExMapTable[Idx].ExGuidIndex == GuidTableIdx) {\r
          //\r
          // Found the next match\r
          //\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
          *TokenNumber = PCD_INVALID_TOKEN_NUMBER;\r
          return EFI_SUCCESS;\r
        }\r
      }\r
    }\r
  }\r
  \r
  return EFI_NOT_FOUND;\r
}\r
  \r
\r
\r
\r
VOID\r
BuildPcdDxeDataBase (\r
  VOID\r
  )\r
{\r
  PEI_PCD_DATABASE    *PeiDatabase;\r
  EFI_HOB_GUID_TYPE   *GuidHob;\r
  UINTN               Idx;\r
\r
  mPcdDatabase = AllocateZeroPool (sizeof(PCD_DATABASE));\r
  ASSERT (mPcdDatabase != NULL);\r
\r
  GuidHob = GetFirstGuidHob (&gPcdDataBaseHobGuid);\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 with default value to PCD DATABASE\r
  //\r
  CopyMem (&mPcdDatabase->DxeDb.Init, &gDXEPcdDbInit, sizeof(DXE_PCD_DATABASE_INIT));\r
\r
\r
  //\r
  // Initialized the Callback Function Table\r
  //\r
\r
  mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER * sizeof (LIST_ENTRY));\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
  for (Idx = 0; Idx + 1 < PCD_TOTAL_TOKEN_NUMBER + 1; Idx++) {\r
    InitializeListHead (&mCallbackFnTable[Idx]);\r
  }\r
    \r
  return;\r
}\r
\r
\r
\r
EFI_STATUS\r
GetHiiVariable (\r
  IN  EFI_GUID      *VariableGuid,\r
  IN  UINT16        *VariableName,\r
  OUT UINT8         **VariableData,\r
  OUT UINTN         *VariableSize\r
  )\r
{\r
  UINTN      Size;\r
  EFI_STATUS Status;\r
  UINT8      *Buffer;\r
\r
  Size = 0;\r
  Buffer = NULL;\r
  \r
  Status = gRT->GetVariable (\r
    (UINT16 *)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
UINT32\r
GetSkuEnabledTokenNumber (\r
  UINT32    LocalTokenNumber,\r
  UINTN     Size,\r
  BOOLEAN   IsPeiDb\r
  ) \r
{\r
  SKU_HEAD              *SkuHead;\r
  SKU_ID                *SkuIdTable;\r
  INTN                  i;\r
  UINT8                 *Value;\r
  SKU_ID                *PhaseSkuIdTable;\r
  UINT8                 *PcdDb;\r
\r
  ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0);\r
\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 ? 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 (mPcdDatabase->PeiDb.Init.SystemSkuId == SkuIdTable[i + 1]) {\r
      break;\r
    }\r
  }\r
  ASSERT (i < SkuIdTable[0]);\r
\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
\r
    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
    default:\r
      ASSERT (FALSE);\r
  }\r
\r
  ASSERT (FALSE);\r
\r
  return 0;\r
  \r
}\r
\r
\r
\r
\r
STATIC\r
VOID\r
InvokeCallbackOnSet (\r
  UINT32            ExTokenNumber,\r
  CONST EFI_GUID    *Guid, OPTIONAL\r
  UINTN             TokenNumber,\r
  VOID              *Data,\r
  UINTN             Size\r
  )\r
{\r
  CALLBACK_FN_ENTRY       *FnTableEntry;\r
  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
  while (ListNode != ListHead) {\r
    FnTableEntry = CR_FNENTRY_FROM_LISTNODE(ListNode, CALLBACK_FN_ENTRY, Node);\r
\r
    FnTableEntry->CallbackFn(Guid, \r
                    (Guid == NULL) ? TokenNumber : ExTokenNumber,\r
                    Data,\r
                    Size);\r
    \r
    ListNode = GetNextNode (ListHead, ListNode);\r
  }\r
  \r
  return;\r
}\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
  IN          UINTN                   TokenNumber,\r
  IN          VOID                    *Data,\r
  IN OUT      UINTN                   *Size,\r
  IN          BOOLEAN                 PtrType\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
  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
  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
  ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);\r
\r
  if (!PtrType) {\r
    ASSERT (*Size == DxePcdGetSize (TokenNumber + 1));\r
  }\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) ((TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? TRUE : FALSE);\r
\r
  LocalTokenNumberTable  = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
                                     mPcdDatabase->DxeDb.Init.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
  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
                        : TokenNumber - PEI_LOCAL_TOKEN_NUMBER;\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
\r
  Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
\r
  PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);\r
\r
  StringTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.StringTable :\r
                          mPcdDatabase->DxeDb.Init.StringTable;\r
  \r
  InternalData = PcdDb + Offset;\r
\r
  switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
    case PCD_TYPE_VPD:\r
      ASSERT (FALSE);\r
      Status = EFI_INVALID_PARAMETER;\r
      break;\r
    \r
    case PCD_TYPE_STRING:\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
      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
      VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);\r
      \r
      Guid = &(GuidTable[VariableHead->GuidTableIndex]);\r
      Name = &(StringTable[VariableHead->StringIndex]);\r
      VariableOffset = VariableHead->Offset;\r
\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
        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
      Status = EFI_SUCCESS;\r
      switch (*Size) {\r
        case sizeof(UINT8):\r
          *((UINT8 *) InternalData) = *((UINT8 *) Data);\r
          break;\r
\r
        case sizeof(UINT16):\r
          *((UINT16 *) InternalData) = *((UINT16 *) Data);\r
          break;\r
\r
        case sizeof(UINT32):\r
          *((UINT32 *) InternalData) = *((UINT32 *) Data);\r
          break;\r
\r
        case sizeof(UINT64):\r
          *((UINT64 *) InternalData) = *((UINT64 *) Data);\r
          break;\r
\r
        default:\r
          ASSERT (FALSE);\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
  EfiReleaseLock (&mPcdDatabaseLock);\r
  \r
  return Status;\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
  return GetWorker(GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber), GetSize);\r
}\r
\r
\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
  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
\r
  return SetWorker (TokenNumber, Data, SetSize, PtrType);\r
\r
}\r
\r
\r
\r
\r
EFI_STATUS\r
SetHiiVariable (\r
  IN  EFI_GUID     *VariableGuid,\r
  IN  UINT16       *VariableName,\r
  IN  CONST VOID   *Data,\r
  IN  UINTN        DataSize,\r
  IN  UINTN        Offset\r
  )\r
{\r
  UINTN       Size;\r
  VOID        *Buffer;\r
  EFI_STATUS  Status;\r
  UINT32      Attribute;\r
\r
  Size = 0;\r
\r
  Status = gRT->GetVariable (\r
    (UINT16 *)VariableName,\r
    VariableGuid,\r
    NULL,\r
    &Size,\r
    NULL\r
    );\r
\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
\r
    Buffer = AllocatePool (Size);\r
\r
    ASSERT (Buffer != NULL);\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
    Status = gRT->SetVariable (\r
              VariableName,\r
              VariableGuid,\r
              Attribute,\r
              Size,\r
              Buffer\r
              );\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
\r
\r
\r
UINTN           \r
GetExPcdTokenNumber (\r
  IN CONST EFI_GUID             *Guid,\r
  IN UINT32                     ExTokenNumber\r
  )\r
{\r
  UINT32              i;\r
  DYNAMICEX_MAPPING   *ExMap;\r
  EFI_GUID            *GuidTable;\r
  EFI_GUID            *MatchGuid;\r
  UINTN               MatchGuidIdx;\r
\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
\r
        }\r
      }\r
    }\r
  }\r
  \r
  ExMap       = mPcdDatabase->DxeDb.Init.ExMapTable;\r
  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
  \r
  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;\r
    }\r
  }\r
\r
  ASSERT (FALSE);\r
\r
  return 0;\r
}\r
\r
\r
STATIC\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
STATIC\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
STATIC\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