\r
VOID *\r
GetWorker (\r
- PCD_TOKEN_NUMBER TokenNumber,\r
+ UINTN TokenNumber,\r
UINTN GetSize\r
)\r
{\r
UINT32 *LocalTokenNumberTable;\r
UINT16 *SizeTable;\r
BOOLEAN IsPeiDb;\r
- UINTN Size;\r
UINT32 Offset;\r
EFI_GUID *GuidTable;\r
UINT16 *StringTable;\r
EFI_GUID *Guid;\r
UINT16 *Name;\r
VARIABLE_HEAD *VariableHead;\r
+ UINT8 *VaraiableDefaultBuffer;\r
EFI_STATUS Status;\r
UINTN DataSize;\r
- VOID *Data;\r
+ UINT8 *Data;\r
VPD_HEAD *VpdHead;\r
UINT8 *PcdDb;\r
UINT16 StringTableIdx; \r
UINT32 LocalTokenNumber;\r
+ UINTN MaxSize;\r
+ UINTN TmpTokenNumber;\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
ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
\r
- Size = DxePcdGetSize (TokenNumber);\r
- ASSERT (GetSize == Size || GetSize == 0);\r
+ ASSERT ((GetSize == DxePcdGetSize (TokenNumber + 1)) || (GetSize == 0));\r
\r
\r
- IsPeiDb = (TokenNumber <= PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+ IsPeiDb = (TokenNumber < PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
\r
LocalTokenNumberTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
\r
if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
- LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, Size, IsPeiDb);\r
+ 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
\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 *) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
+ return (VOID *) (UINTN) (FixedPcdGet32(PcdVpdBaseAddress) + VpdHead->Offset);\r
\r
case PCD_TYPE_HII:\r
GuidTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.GuidTable :\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
- ASSERT_EFI_ERROR (Status);\r
- ASSERT (DataSize >= (UINTN) (VariableHead->Offset + Size));\r
-\r
- return (UINT8 *) Data + VariableHead->Offset;\r
+ Status = GetHiiVariable (Guid, Name, (VOID*)&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 (TokenNumber, &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
+ return (VOID *) VaraiableDefaultBuffer;\r
\r
case PCD_TYPE_STRING:\r
StringTableIdx = (UINT16) *((UINT8 *) PcdDb + Offset);\r
\r
EFI_STATUS\r
DxeRegisterCallBackWorker (\r
- IN PCD_TOKEN_NUMBER TokenNumber,\r
+ IN UINTN TokenNumber,\r
IN CONST GUID *Guid, OPTIONAL\r
IN PCD_PROTOCOL_CALLBACK CallBackFunction\r
)\r
LIST_ENTRY *ListNode;\r
\r
if (Guid != NULL) {\r
- TokenNumber = GetExPcdTokenNumber (Guid, TokenNumber);\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
\r
EFI_STATUS\r
DxeUnRegisterCallBackWorker (\r
- IN PCD_TOKEN_NUMBER TokenNumber,\r
+ IN UINTN TokenNumber,\r
IN CONST GUID *Guid, OPTIONAL\r
IN PCD_PROTOCOL_CALLBACK CallBackFunction\r
)\r
LIST_ENTRY *ListNode;\r
\r
if (Guid != NULL) {\r
- TokenNumber = GetExPcdTokenNumber (Guid, TokenNumber);\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
\r
\r
\r
-PCD_TOKEN_NUMBER\r
+UINTN \r
ExGetNextTokeNumber (\r
IN CONST EFI_GUID *Guid,\r
- IN PCD_TOKEN_NUMBER TokenNumber,\r
+ IN UINTN TokenNumber,\r
IN EFI_GUID *GuidTable,\r
IN UINTN SizeOfGuidTable,\r
IN DYNAMICEX_MAPPING *ExMapTable,\r
if (TokenNumber == PCD_INVALID_TOKEN_NUMBER) {\r
return ExMapTable[Idx].ExTokenNumber;\r
}\r
- \r
+\r
for ( ; Idx < SizeOfExMapTable; Idx++) {\r
if (ExMapTable[Idx].ExTokenNumber == TokenNumber) {\r
Idx++;\r
VOID\r
BuildPcdDxeDataBase (\r
VOID\r
-)\r
+ )\r
{\r
PEI_PCD_DATABASE *PeiDatabase;\r
EFI_HOB_GUID_TYPE *GuidHob;\r
//\r
\r
if (PCD_TOTAL_TOKEN_NUMBER != 0) {\r
- mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER);\r
+ mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER * sizeof (LIST_ENTRY));\r
}\r
\r
for (Idx = 0; Idx < PCD_TOTAL_TOKEN_NUMBER; Idx++) {\r
GetHiiVariable (\r
IN EFI_GUID *VariableGuid,\r
IN UINT16 *VariableName,\r
- OUT VOID ** VariableData,\r
+ OUT VOID **VariableData,\r
OUT UINTN *VariableSize\r
)\r
{\r
EFI_STATUS Status;\r
VOID *Buffer;\r
\r
+ Size = 0;\r
+ Buffer = NULL;\r
+ \r
Status = EfiGetVariable (\r
(UINT16 *)VariableName,\r
VariableGuid,\r
NULL,\r
&Size,\r
- NULL\r
- );\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 = AllocatePool (Size);\r
+\r
+ ASSERT (Buffer != NULL);\r
+\r
+ Status = EfiGetVariable (\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
}\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 ((Value - PcdDb) | PCD_TYPE_VPD);\r
+ return (UINT32) ((Value - PcdDb) | PCD_TYPE_VPD);\r
\r
case PCD_TYPE_HII:\r
Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[i]);\r
- return ((Value - PcdDb) | PCD_TYPE_HII);\r
+ return (UINT32) ((Value - PcdDb) | PCD_TYPE_HII);\r
+\r
+ case PCD_TYPE_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 (Value - PcdDb);\r
- \r
+ return (UINT32) (Value - PcdDb);\r
+\r
default:\r
ASSERT (FALSE);\r
}\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
}\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
- PCD_TOKEN_NUMBER 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
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
+ // 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
ASSERT (TokenNumber < PCD_TOTAL_TOKEN_NUMBER);\r
\r
- if (PtrType) {\r
- ASSERT (Size <= DxePcdGetSize (TokenNumber));\r
- } else {\r
- ASSERT (Size == DxePcdGetSize (TokenNumber));\r
+ if (!PtrType) {\r
+ ASSERT (*Size == DxePcdGetSize (TokenNumber + 1));\r
}\r
\r
- IsPeiDb = (TokenNumber <= PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
+ IsPeiDb = (TokenNumber < PEI_LOCAL_TOKEN_NUMBER) ? TRUE : FALSE;\r
\r
LocalTokenNumberTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable : \r
mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
\r
if ((TokenNumber < PEI_NEX_TOKEN_NUMBER) ||\r
(TokenNumber >= PEI_LOCAL_TOKEN_NUMBER || TokenNumber < (PEI_LOCAL_TOKEN_NUMBER + DXE_NEX_TOKEN_NUMBER))) {\r
- InvokeCallbackOnSet (0, NULL, TokenNumber, Data, Size);\r
+ InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, *Size);\r
}\r
\r
TokenNumber = IsPeiDb ? TokenNumber\r
LocalTokenNumber = LocalTokenNumberTable[TokenNumber];\r
\r
if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
- LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, Size, IsPeiDb);\r
+ 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
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
\r
case PCD_TYPE_STRING:\r
- CopyMem (&StringTable[*((UINT16 *)InternalData)], Data, Size);\r
- break;\r
+ if (SetPtrTypeSize (TokenNumber, Size)) {\r
+ CopyMem (&StringTable[*((UINT16 *)InternalData)], Data, *Size);\r
+ return EFI_SUCCESS;\r
+ } else {\r
+ return EFI_INVALID_PARAMETER;\r
+ }\r
\r
case PCD_TYPE_HII:\r
- //\r
- // Bug Bug: Please implement this\r
- //\r
+ if (PtrType) {\r
+ if (!SetPtrTypeSize (TokenNumber, Size)) {\r
+ return EFI_INVALID_PARAMETER;\r
+ }\r
+ }\r
+ \r
GuidTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.GuidTable :\r
mPcdDatabase->DxeDb.Init.GuidTable;\r
\r
\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
+ return EFI_SUCCESS;\r
+ } else {\r
+ return Status;\r
+ }\r
+ \r
+ //\r
+ // Bug Bug: Please implement this\r
+ //\r
\r
case PCD_TYPE_DATA:\r
if (PtrType) {\r
- CopyMem (InternalData, Data, Size);\r
- return EFI_SUCCESS;\r
+ if (SetPtrTypeSize (TokenNumber, Size)) {\r
+ CopyMem (InternalData, Data, *Size);\r
+ return EFI_SUCCESS;\r
+ } else {\r
+ return EFI_INVALID_PARAMETER;\r
+ }\r
}\r
\r
- switch (Size) {\r
+ switch (*Size) {\r
case sizeof(UINT8):\r
*((UINT8 *) InternalData) = *((UINT8 *) Data);\r
return EFI_SUCCESS;\r
IN UINTN GetSize\r
) \r
{\r
- return GetWorker(GetExPcdTokenNumber (Guid, ExTokenNumber), GetSize);\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 PCD_TOKEN_NUMBER 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
- PCD_TOKEN_NUMBER TokenNumber;\r
+ UINTN TokenNumber;\r
\r
- TokenNumber = GetExPcdTokenNumber (Guid, ExTokenNumber);\r
+ TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber);\r
\r
- InvokeCallbackOnSet (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
NULL\r
);\r
\r
- ASSERT (Status == EFI_BUFFER_TOO_SMALL);\r
+ if (Status == EFI_BUFFER_TOO_SMALL) {\r
\r
- Buffer = AllocatePool (Size);\r
+ Buffer = AllocatePool (Size);\r
\r
- ASSERT (Buffer != NULL);\r
+ ASSERT (Buffer != NULL);\r
\r
- Status = EfiGetVariable (\r
- VariableName,\r
- VariableGuid,\r
- &Attribute,\r
- &Size,\r
- Buffer\r
- );\r
+ Status = EfiGetVariable (\r
+ VariableName,\r
+ VariableGuid,\r
+ &Attribute,\r
+ &Size,\r
+ Buffer\r
+ );\r
+ \r
+ ASSERT_EFI_ERROR (Status);\r
\r
+ CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
\r
- CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
+ Status = EfiSetVariable (\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
\r
\r
\r
-PCD_TOKEN_NUMBER\r
+UINTN \r
GetExPcdTokenNumber (\r
IN CONST EFI_GUID *Guid,\r
- IN PCD_TOKEN_NUMBER ExTokenNumber\r
+ IN UINT32 ExTokenNumber\r
)\r
{\r
UINT32 i;\r
EFI_GUID *MatchGuid;\r
UINTN MatchGuidIdx;\r
\r
- ExMap = mPcdDatabase->PeiDb.Init.ExMapTable;\r
- GuidTable = mPcdDatabase->PeiDb.Init.GuidTable;\r
- \r
- MatchGuid = ScanGuid (GuidTable, sizeof(mPcdDatabase->PeiDb.Init.GuidTable), Guid);\r
- ASSERT (MatchGuid != NULL);\r
+ 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
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
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
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
+ IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx < PEI_LOCAL_TOKEN_NUMBER);\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
+ IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx < PEI_LOCAL_TOKEN_NUMBER);\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