+\r
+\r
+SKU_ID *\r
+GetSkuIdArray (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ IN BOOLEAN IsPeiPcd\r
+ )\r
+{\r
+ SKU_HEAD *SkuHead;\r
+ UINTN LocalTokenNumber;\r
+ UINT8 *Database;\r
+\r
+ if (IsPeiPcd) {\r
+ LocalTokenNumber = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+ Database = (UINT8 *) &mPcdDatabase->PeiDb;\r
+ } else {\r
+ LocalTokenNumber = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx - PEI_LOCAL_TOKEN_NUMBER];\r
+ Database = (UINT8 *) &mPcdDatabase->DxeDb;\r
+ }\r
+\r
+ ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) != 0);\r
+\r
+ SkuHead = (SKU_HEAD *) ((UINT8 *)Database + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));\r
+\r
+ return (SKU_ID *) (Database + SkuHead->SkuIdTableOffset);\r
+ \r
+}\r
+\r
+\r
+\r
+UINTN\r
+GetSizeTableIndexA (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ IN UINT32 *LocalTokenNumberTable,\r
+ IN BOOLEAN IsPeiDb\r
+ )\r
+{\r
+ UINTN i;\r
+ UINTN SizeTableIdx;\r
+ UINTN LocalTokenNumber;\r
+ SKU_ID *SkuIdTable;\r
+ \r
+ SizeTableIdx = 0;\r
+\r
+ for (i=0; i<LocalTokenNumberTableIdx; i++) {\r
+ LocalTokenNumber = LocalTokenNumberTable[i];\r
+\r
+ if ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER) {\r
+ //\r
+ // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type \r
+ // PCD entry.\r
+ //\r
+ if (LocalTokenNumber & PCD_TYPE_VPD) {\r
+ //\r
+ // We have only one entry for VPD enabled PCD entry:\r
+ // 1) MAX Size.\r
+ // We consider current size is equal to MAX size.\r
+ //\r
+ SizeTableIdx++;\r
+ } else {\r
+ if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
+ //\r
+ // We have only two entry for Non-Sku enabled PCD entry:\r
+ // 1) MAX SIZE\r
+ // 2) Current Size\r
+ //\r
+ SizeTableIdx += 2;\r
+ } else {\r
+ //\r
+ // We have these entry for SKU enabled PCD entry\r
+ // 1) MAX SIZE\r
+ // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
+ //\r
+ SkuIdTable = GetSkuIdArray (i, IsPeiDb);\r
+ SizeTableIdx += (UINTN)*SkuIdTable + 1;\r
+ }\r
+ }\r
+ }\r
+\r
+ }\r
+\r
+ return SizeTableIdx;\r
+}\r
+\r
+\r
+\r
+\r
+UINTN\r
+GetSizeTableIndex (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ IN BOOLEAN IsPeiDb\r
+ )\r
+{\r
+ UINT32 *LocalTokenNumberTable;\r
+ \r
+ if (IsPeiDb) {\r
+ LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;\r
+ } else {\r
+ LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+ }\r
+ return GetSizeTableIndexA (LocalTokenNumberTableIdx, \r
+ LocalTokenNumberTable,\r
+ IsPeiDb);\r
+}\r
+\r
+\r
+\r
+UINTN\r
+GetPtrTypeSize (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ OUT UINTN *MaxSize\r
+ )\r
+{\r
+ INTN SizeTableIdx;\r
+ UINTN LocalTokenNumber;\r
+ SKU_ID *SkuIdTable;\r
+ SIZE_INFO *SizeTable;\r
+ UINTN i;\r
+ BOOLEAN IsPeiDb;\r
+ UINT32 *LocalTokenNumberTable;\r
+\r
+ // EBC compiler is very choosy. It may report warning about comparison\r
+ // between UINTN and 0 . So we add 1 in each size of the \r
+ // comparison.\r
+ IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+\r
+\r
+ if (IsPeiDb) {\r
+ LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;\r
+ SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;\r
+ } else {\r
+ LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;\r
+ LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+ SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;\r
+ }\r
+\r
+ LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+\r
+ ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);\r
+ \r
+ SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);\r
+\r
+ *MaxSize = SizeTable[SizeTableIdx];\r
+ //\r
+ // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type \r
+ // PCD entry.\r
+ //\r
+ if (LocalTokenNumber & PCD_TYPE_VPD) {\r
+ //\r
+ // We have only one entry for VPD enabled PCD entry:\r
+ // 1) MAX Size.\r
+ // We consider current size is equal to MAX size.\r
+ //\r
+ return *MaxSize;\r
+ } else {\r
+ if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
+ //\r
+ // We have only two entry for Non-Sku enabled PCD entry:\r
+ // 1) MAX SIZE\r
+ // 2) Current Size\r
+ //\r
+ return SizeTable[SizeTableIdx + 1];\r
+ } else {\r
+ //\r
+ // We have these entry for SKU enabled PCD entry\r
+ // 1) MAX SIZE\r
+ // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
+ //\r
+ SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);\r
+ for (i = 0; i < SkuIdTable[0]; i++) {\r
+ if (SkuIdTable[1 + i] == mPcdDatabase->PeiDb.Init.SystemSkuId) {\r
+ return SizeTable[SizeTableIdx + 1 + i];\r
+ }\r
+ }\r
+ return SizeTable[SizeTableIdx + 1];\r
+ }\r
+ }\r
+}\r
+\r
+\r
+\r
+BOOLEAN\r
+SetPtrTypeSize (\r
+ IN UINTN LocalTokenNumberTableIdx,\r
+ IN OUT UINTN *CurrentSize\r
+ )\r
+{\r
+ INTN SizeTableIdx;\r
+ UINTN LocalTokenNumber;\r
+ SKU_ID *SkuIdTable;\r
+ SIZE_INFO *SizeTable;\r
+ UINTN i;\r
+ UINTN MaxSize;\r
+ BOOLEAN IsPeiDb;\r
+ UINT32 *LocalTokenNumberTable;\r
+\r
+ // EBC compiler is very choosy. It may report warning about comparison\r
+ // between UINTN and 0 . So we add 1 in each size of the \r
+ // comparison.\r
+ IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);\r
+\r
+ if (IsPeiDb) {\r
+ LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;\r
+ SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;\r
+ } else {\r
+ LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;\r
+ LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;\r
+ SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;\r
+ }\r
+\r
+ LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];\r
+\r
+ ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);\r
+ \r
+ SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);\r
+\r
+ MaxSize = SizeTable[SizeTableIdx];\r
+ //\r
+ // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type \r
+ // PCD entry.\r
+ //\r
+ if (LocalTokenNumber & PCD_TYPE_VPD) {\r
+ //\r
+ // We shouldn't come here as we don't support SET for VPD\r
+ //\r
+ ASSERT (FALSE);\r
+ return FALSE;\r
+ } else {\r
+ if ((*CurrentSize > MaxSize) ||\r
+ (*CurrentSize == MAX_ADDRESS)) {\r
+ *CurrentSize = MaxSize;\r
+ return FALSE;\r
+ } \r
+ \r
+ if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {\r
+ //\r
+ // We have only two entry for Non-Sku enabled PCD entry:\r
+ // 1) MAX SIZE\r
+ // 2) Current Size\r
+ //\r
+ SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;\r
+ return TRUE;\r
+ } else {\r
+ //\r
+ // We have these entry for SKU enabled PCD entry\r
+ // 1) MAX SIZE\r
+ // 2) Current Size for each SKU_ID (It is equal to MaxSku).\r
+ //\r
+ SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);\r
+ for (i = 0; i < SkuIdTable[0]; i++) {\r
+ if (SkuIdTable[1 + i] == mPcdDatabase->PeiDb.Init.SystemSkuId) {\r
+ SizeTable[SizeTableIdx + 1 + i] = (SIZE_INFO) *CurrentSize;\r
+ return TRUE;\r
+ }\r
+ }\r
+ SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;\r
+ return TRUE;\r
+ }\r
+ }\r
+}\r
+\r