\r
LocalTokenNumber = GetPcdDatabase()->Init.LocalTokenNumberTable[TokenNumber];\r
\r
+ //\r
+ // We don't support SET for HII and VPD type PCD entry in PEI phase.\r
+ // So we will assert if any register callback for such PCD entry.\r
+ //\r
ASSERT ((LocalTokenNumber & PCD_TYPE_HII) == 0);\r
ASSERT ((LocalTokenNumber & PCD_TYPE_VPD) == 0);\r
\r
}\r
}\r
\r
- switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
+ switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
case PCD_TYPE_VPD:\r
- Value += sizeof(VPD_HEAD) * i;\r
+ Value = (UINT8 *) &(((VPD_HEAD *) Value)[i]);\r
return ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_VPD);\r
\r
case PCD_TYPE_HII:\r
- Value += sizeof(VARIABLE_HEAD) * i;\r
+ Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[i]);\r
return ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_HII);\r
\r
+ case PCD_TYPE_STRING:\r
+ Value = (UINT8 *) &(((STRING_HEAD *) Value)[i]);\r
+ return ((Value - (UINT8 *) PeiPcdDb) | PCD_TYPE_STRING);\r
+\r
case PCD_TYPE_DATA:\r
Value += Size * i;\r
return (Value - (UINT8 *) PeiPcdDb);\r
- \r
+\r
default:\r
ASSERT (FALSE);\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
\r
EFI_STATUS\r
SetWorker (\r
- UINTN TokenNumber,\r
- VOID *Data,\r
- UINTN Size,\r
- BOOLEAN PtrType\r
+ IN UINTN TokenNumber,\r
+ IN OUT VOID *Data,\r
+ IN OUT UINTN *Size,\r
+ IN BOOLEAN PtrType\r
)\r
{\r
UINT32 LocalTokenNumber;\r
UINT16 StringTableIdx;\r
UINTN Offset;\r
VOID *InternalData;\r
+ UINTN MaxSize;\r
\r
//\r
// TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.\r
\r
LocalTokenNumber = PeiPcdDb->Init.LocalTokenNumberTable[TokenNumber];\r
\r
- if (PtrType) {\r
- ASSERT (PeiPcdDb->Init.SizeTable[TokenNumber] >= Size);\r
- } else {\r
- ASSERT (PeiPcdDb->Init.SizeTable[TokenNumber] == Size);\r
+ if (!PtrType) {\r
+ ASSERT (PeiPcdGetSize(TokenNumber + 1) == *Size);\r
}\r
\r
//\r
// type PCD entry in ExSetWorker.\r
//\r
if (TokenNumber < PEI_NEX_TOKEN_NUMBER) {\r
- InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, Size);\r
+ InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, *Size);\r
}\r
\r
if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
- LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, Size);\r
+ if (PtrType) {\r
+ MaxSize = GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
+ } else {\r
+ MaxSize = *Size;\r
+ }\r
+ LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize);\r
}\r
\r
Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
InternalData = (VOID *) ((UINT8 *) PeiPcdDb + Offset);\r
\r
- switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
+ switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
case PCD_TYPE_VPD:\r
case PCD_TYPE_HII:\r
{\r
}\r
\r
case PCD_TYPE_STRING:\r
- StringTableIdx = *((UINT16 *)InternalData);\r
- CopyMem (&PeiPcdDb->Init.StringTable[StringTableIdx], Data, Size);\r
- return EFI_SUCCESS;\r
+ if (SetPtrTypeSize (TokenNumber, Size, PeiPcdDb)) {\r
+ StringTableIdx = *((UINT16 *)InternalData);\r
+ CopyMem (&PeiPcdDb->Init.StringTable[StringTableIdx], Data, *Size);\r
+ return EFI_SUCCESS;\r
+ } else {\r
+ return EFI_INVALID_PARAMETER;\r
+ }\r
\r
case PCD_TYPE_DATA:\r
{\r
- \r
if (PtrType) {\r
- CopyMem (InternalData, Data, Size);\r
- return EFI_SUCCESS;\r
+ if (SetPtrTypeSize (TokenNumber, Size, PeiPcdDb)) {\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
\r
\r
\r
+EFI_STATUS\r
+ExSetValueWorker (\r
+ IN UINTN ExTokenNumber,\r
+ IN CONST EFI_GUID *Guid,\r
+ IN VOID *Data,\r
+ IN UINTN Size\r
+ )\r
+{\r
+ return ExSetWorker (ExTokenNumber, Guid, Data, &Size, FALSE);\r
+}\r
+\r
+\r
\r
EFI_STATUS\r
ExSetWorker (\r
- IN UINTN ExTokenNumber,\r
- IN CONST EFI_GUID *Guid,\r
- VOID *Data,\r
- UINTN Size,\r
- BOOLEAN PtrType\r
+ IN UINTN ExTokenNumber,\r
+ IN CONST EFI_GUID *Guid,\r
+ IN VOID *Data,\r
+ IN OUT UINTN *Size,\r
+ IN BOOLEAN PtrType\r
)\r
{\r
UINTN TokenNumber;\r
\r
TokenNumber = GetExPcdTokenNumber (Guid, ExTokenNumber);\r
\r
- InvokeCallbackOnSet (ExTokenNumber, Guid, TokenNumber, Data, Size);\r
+ InvokeCallbackOnSet (ExTokenNumber, Guid, TokenNumber, Data, *Size);\r
\r
- SetWorker (TokenNumber, Data, Size, PtrType);\r
+ return SetWorker (TokenNumber, Data, Size, PtrType);\r
\r
- return EFI_SUCCESS;\r
- \r
}\r
\r
\r
UINT16 StringTableIdx;\r
PEI_PCD_DATABASE *PeiPcdDb;\r
UINT32 LocalTokenNumber;\r
- UINTN Size;\r
+ UINTN MaxSize;\r
\r
//\r
// TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.\r
\r
ASSERT (TokenNumber < PEI_LOCAL_TOKEN_NUMBER);\r
\r
- Size = PeiPcdGetSize(TokenNumber + 1);\r
- \r
- ASSERT (GetSize == Size || GetSize == 0);\r
+ ASSERT ((GetSize == PeiPcdGetSize(TokenNumber + 1)) || (GetSize == 0));\r
\r
PeiPcdDb = GetPcdDatabase ();\r
\r
LocalTokenNumber = PeiPcdDb->Init.LocalTokenNumberTable[TokenNumber];\r
\r
if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {\r
- LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, Size);\r
+ if (GetSize == 0) {\r
+ MaxSize = GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
+ } else {\r
+ MaxSize = GetSize;\r
+ }\r
+ LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize);\r
}\r
\r
Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
StringTable = PeiPcdDb->Init.StringTable;\r
\r
- switch (LocalTokenNumber & ~PCD_DATABASE_OFFSET_MASK) {\r
+ switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
case PCD_TYPE_VPD:\r
{\r
VPD_HEAD *VpdHead;\r
return (PEI_PCD_DATABASE *) GET_GUID_HOB_DATA (GuidHob);\r
}\r
\r
+\r
+SKU_ID *\r
+GetSkuIdArray (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ IN PEI_PCD_DATABASE *Database\r
+ )\r
+{\r
+ SKU_HEAD *SkuHead;\r
+ UINTN LocalTokenNumber;\r
+\r
+ LocalTokenNumber = Database->Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];\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 *) ((UINT8 *)Database + SkuHead->SkuIdTableOffset);\r
+ \r
+}\r
+\r
+\r
+\r
+UINTN\r
+GetSizeTableIndex (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ IN PEI_PCD_DATABASE *Database\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 = Database->Init.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, Database);\r
+ SizeTableIdx += (UINTN)*SkuIdTable + 1;\r
+ }\r
+ }\r
+ }\r
+\r
+ }\r
+\r
+ return SizeTableIdx;\r
+}\r
+\r
+\r
+\r
+\r
+UINTN\r
+GetPtrTypeSize (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ OUT UINTN *MaxSize,\r
+ IN PEI_PCD_DATABASE *Database\r
+ )\r
+{\r
+ INTN SizeTableIdx;\r
+ UINTN LocalTokenNumber;\r
+ SKU_ID *SkuIdTable;\r
+ SIZE_INFO *SizeTable;\r
+ UINTN i;\r
+\r
+ SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, Database);\r
+\r
+ LocalTokenNumber = Database->Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+\r
+ ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);\r
+ \r
+ SizeTable = Database->Init.SizeTable;\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, Database);\r
+ for (i = 0; i < SkuIdTable[0]; i++) {\r
+ if (SkuIdTable[1 + i] == Database->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
+ IN PEI_PCD_DATABASE *Database\r
+ )\r
+{\r
+ INTN SizeTableIdx;\r
+ UINTN LocalTokenNumber;\r
+ SKU_ID *SkuIdTable;\r
+ SIZE_INFO *SizeTable;\r
+ UINTN i;\r
+ UINTN MaxSize;\r
+ \r
+ SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, Database);\r
+\r
+ LocalTokenNumber = Database->Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+\r
+ ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);\r
+ \r
+ SizeTable = Database->Init.SizeTable;\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, Database);\r
+ for (i = 0; i < SkuIdTable[0]; i++) {\r
+ if (SkuIdTable[1 + i] == Database->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