+/** @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 common header file for this module.\r
+//\r
+#include "CommonHeader.h"\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