Help functions used by PCD DXE driver.\r
\r
Copyright (c) 2014, Hewlett-Packard Development Company, L.P.<BR>\r
-Copyright (c) 2006 - 2014, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2006 - 2017, Intel Corporation. All rights reserved.<BR>\r
+(C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>\r
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
{\r
PCD_DATABASE_INIT *Database;\r
UINT8 *StringTable;\r
+ UINTN NameSize;\r
PCD_NAME_INDEX *PcdNameIndex;\r
CHAR8 *TokenSpaceName;\r
CHAR8 *PcdName;\r
CHAR8 *Name;\r
\r
+ //\r
+ // Return NULL when PCD name table is absent. \r
+ //\r
+ if (IsPeiDb) {\r
+ if (mPcdDatabase.PeiDb->PcdNameTableOffset == 0) {\r
+ return NULL;\r
+ }\r
+ } else {\r
+ if (mPcdDatabase.DxeDb->PcdNameTableOffset == 0) {\r
+ return NULL;\r
+ }\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
//\r
// Need to get the full PCD name.\r
//\r
- Name = AllocateZeroPool (AsciiStrSize (TokenSpaceName) + AsciiStrSize (PcdName));\r
+ NameSize = AsciiStrSize (TokenSpaceName) + AsciiStrSize (PcdName);\r
+ Name = AllocateZeroPool (NameSize);\r
ASSERT (Name != NULL);\r
//\r
// Catenate TokenSpaceCName and PcdCName with a '.' to form the full PCD name.\r
//\r
- AsciiStrCat (Name, TokenSpaceName);\r
+ AsciiStrCatS (Name, NameSize, TokenSpaceName);\r
Name[AsciiStrSize (TokenSpaceName) - sizeof (CHAR8)] = '.';\r
- AsciiStrCat (Name, PcdName); \r
+ AsciiStrCatS (Name, NameSize, PcdName); \r
}\r
\r
return Name;\r
switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {\r
case PCD_TYPE_VPD:\r
VpdHead = (VPD_HEAD *) ((UINT8 *) PcdDb + Offset);\r
- RetPtr = (VOID *) (UINTN) (PcdGet32 (PcdVpdBaseAddress) + VpdHead->Offset);\r
+ RetPtr = (VOID *) ((UINTN) PcdGet32 (PcdVpdBaseAddress) + VpdHead->Offset);\r
\r
break;\r
\r
}\r
\r
for ( ; Index < ExMapTableCount; Index++) {\r
- if (ExMapTable[Index].ExTokenNumber == *TokenNumber) {\r
+ if ((ExMapTable[Index].ExTokenNumber == *TokenNumber) && (ExMapTable[Index].ExGuidIndex == GuidTableIdx)) {\r
break;\r
}\r
}\r
{\r
SKU_HEAD *SkuHead;\r
SKU_ID *SkuIdTable;\r
- INTN Index;\r
+ UINTN Index;\r
UINT8 *Value;\r
UINT8 *PcdDb;\r
BOOLEAN FoundSku;\r
\r
}\r
\r
+/**\r
+ Get variable size and data from HII-type PCDs.\r
+\r
+ @param[in] VariableGuid Guid of variable which stored value of a HII-type PCD.\r
+ @param[in] VariableName Unicode name of variable which stored value of a HII-type PCD.\r
+ @param[out] VariableSize Pointer to variable size got from HII-type PCDs.\r
+ @param[out] VariableData Pointer to variable data got from HII-type PCDs.\r
+\r
+**/\r
+VOID\r
+GetVariableSizeAndDataFromHiiPcd (\r
+ IN EFI_GUID *VariableGuid,\r
+ IN UINT16 *VariableName,\r
+ OUT UINTN *VariableSize,\r
+ OUT VOID *VariableData OPTIONAL\r
+ )\r
+{\r
+ BOOLEAN IsPeiDb;\r
+ PCD_DATABASE_INIT *Database;\r
+ UINTN TokenNumber;\r
+ UINT32 LocalTokenNumber;\r
+ UINTN Offset;\r
+ EFI_GUID *GuidTable;\r
+ UINT8 *StringTable;\r
+ VARIABLE_HEAD *VariableHead;\r
+ EFI_GUID *Guid;\r
+ UINT16 *Name;\r
+ UINTN PcdDataSize;\r
+ UINTN Size;\r
+ UINT8 *VaraiableDefaultBuffer;\r
+ STRING_HEAD StringTableIdx;\r
+\r
+ *VariableSize = 0;\r
+\r
+ //\r
+ // Go through PCD database to find out DynamicHii PCDs.\r
+ //\r
+ for (TokenNumber = 1; TokenNumber <= mPcdTotalTokenCount; TokenNumber++) {\r
+ IsPeiDb = (BOOLEAN) ((TokenNumber + 1 < mPeiLocalTokenCount + 1) ? TRUE : FALSE);\r
+ Database = IsPeiDb ? mPcdDatabase.PeiDb: mPcdDatabase.DxeDb;\r
+ LocalTokenNumber = GetLocalTokenNumber (IsPeiDb, TokenNumber);\r
+ if ((LocalTokenNumber & PCD_TYPE_HII) != 0) {\r
+ //\r
+ // Get the Variable Guid and Name pointer.\r
+ //\r
+ Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;\r
+ VariableHead = (VARIABLE_HEAD *) ((UINT8 *) Database + Offset);\r
+ StringTable = (UINT8 *) ((UINT8 *) Database + Database->StringTableOffset);\r
+ GuidTable = (EFI_GUID *) ((UINT8 *) Database + Database->GuidTableOffset);\r
+ Guid = GuidTable + VariableHead->GuidTableIndex;\r
+ Name = (UINT16*) (StringTable + VariableHead->StringIndex);\r
+ if (CompareGuid (VariableGuid, Guid) && (StrCmp (VariableName, Name) == 0)) {\r
+ //\r
+ // It is the matched DynamicHii PCD.\r
+ //\r
+ PcdDataSize = DxePcdGetSize (TokenNumber);\r
+ Size = VariableHead->Offset + PcdDataSize;\r
+ if (Size > *VariableSize) {\r
+ *VariableSize = Size;\r
+ }\r
+ if (VariableData != NULL) {\r
+ if ((LocalTokenNumber & PCD_TYPE_ALL_SET) == (PCD_TYPE_HII|PCD_TYPE_STRING)) {\r
+ //\r
+ // If a HII type PCD's datum type is VOID*, the DefaultValueOffset is the index of\r
+ // string array in string table.\r
+ //\r
+ StringTableIdx = *(STRING_HEAD *) ((UINT8 *) Database + VariableHead->DefaultValueOffset);\r
+ VaraiableDefaultBuffer = (UINT8 *) (StringTable + StringTableIdx);\r
+ } else {\r
+ VaraiableDefaultBuffer = (UINT8 *) Database + VariableHead->DefaultValueOffset;\r
+ }\r
+ CopyMem ((UINT8 *) VariableData + VariableHead->Offset, VaraiableDefaultBuffer, PcdDataSize);\r
+ }\r
+ }\r
+ }\r
+ }\r
+}\r
+\r
/**\r
Set value for HII-type PCD.\r
\r
//\r
// If variable does not exist, a new variable need to be created.\r
//\r
- \r
- Size = Offset + DataSize;\r
- \r
+\r
+ //\r
+ // Get size, allocate buffer and get data.\r
+ //\r
+ GetVariableSizeAndDataFromHiiPcd (VariableGuid, VariableName, &Size, NULL);\r
Buffer = AllocateZeroPool (Size);\r
ASSERT (Buffer != NULL);\r
- \r
+ GetVariableSizeAndDataFromHiiPcd (VariableGuid, VariableName, &Size, Buffer);\r
+\r
+ //\r
+ // Update buffer.\r
+ //\r
CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);\r
\r
if (SetAttributes == 0) {\r
//\r
// Go through PCD database to find out DynamicHii PCDs.\r
//\r
- for (TokenNumber = 0; TokenNumber < LocalTokenCount; TokenNumber++) {\r
+ for (TokenNumber = 1; TokenNumber <= LocalTokenCount; TokenNumber++) {\r
if (IsPeiDb) {\r
LocalTokenNumber = GetLocalTokenNumber (TRUE, TokenNumber);\r
} else {\r
projects / mirror_edk2.git / blobdiff