[mirror_edk2.git] / BaseTools / Source / C / VfrCompile / VfrUtilityLib.cpp

/** @file\r
  \r
  Vfr common library functions.\r
\r
Copyright (c) 2004 - 2016, Intel Corporation. All rights reserved.<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
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
\r
#include "stdio.h"\r
#include "stdlib.h"\r
#include "CommonLib.h"\r
#include "VfrUtilityLib.h"\r
#include "VfrFormPkg.h"\r
\r
VOID\r
CVfrBinaryOutput::WriteLine (\r
  IN FILE         *pFile,\r
  IN UINT32       LineBytes,\r
  IN CONST CHAR8  *LineHeader,\r
  IN CHAR8        *BlkBuf,\r
  IN UINT32       BlkSize\r
  )\r
{\r
  UINT32    Index;\r
\r
  if ((pFile == NULL) || (LineHeader == NULL) || (BlkBuf == NULL)) {\r
    return;\r
  }\r
\r
  for (Index = 0; Index < BlkSize; Index++) {\r
    if ((Index % LineBytes) == 0) {\r
      fprintf (pFile, "\n%s", LineHeader);\r
    }\r
    fprintf (pFile, "0x%02X,  ", (UINT8)BlkBuf[Index]);\r
  }\r
}\r
\r
VOID\r
CVfrBinaryOutput::WriteEnd (\r
  IN FILE         *pFile,\r
  IN UINT32       LineBytes,\r
  IN CONST CHAR8  *LineHeader,\r
  IN CHAR8        *BlkBuf,\r
  IN UINT32       BlkSize\r
  )\r
{\r
  UINT32    Index;\r
\r
  if ((BlkSize == 0) || (pFile == NULL) || (LineHeader == NULL) || (BlkBuf == NULL)) {\r
    return;\r
  }\r
\r
  for (Index = 0; Index < BlkSize - 1; Index++) {\r
    if ((Index % LineBytes) == 0) {\r
      fprintf (pFile, "\n%s", LineHeader);\r
    }\r
    fprintf (pFile, "0x%02X,  ", (UINT8)BlkBuf[Index]);\r
  }\r
\r
  if ((Index % LineBytes) == 0) {\r
    fprintf (pFile, "\n%s", LineHeader);\r
  }\r
  fprintf (pFile, "0x%02X\n", (UINT8)BlkBuf[Index]);\r
}\r
\r
SConfigInfo::SConfigInfo (\r
  IN UINT8              Type, \r
  IN UINT16             Offset, \r
  IN UINT32             Width, \r
  IN EFI_IFR_TYPE_VALUE Value\r
  )\r
{\r
  mNext   = NULL;\r
  mOffset = Offset;\r
  mWidth  = (UINT16)Width;\r
  mValue  = new UINT8[mWidth];\r
  if (mValue == NULL) {\r
    return;\r
  }\r
\r
  switch (Type) {\r
  case EFI_IFR_TYPE_NUM_SIZE_8 :\r
    memcpy (mValue, &Value.u8, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_NUM_SIZE_16 :\r
    memcpy (mValue, &Value.u16, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_NUM_SIZE_32 :\r
    memcpy (mValue, &Value.u32, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_NUM_SIZE_64 :\r
    memcpy (mValue, &Value.u64, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_BOOLEAN :\r
    memcpy (mValue, &Value.b, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_TIME :\r
    memcpy (mValue, &Value.time, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_DATE :\r
    memcpy (mValue, &Value.date, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_STRING :\r
    memcpy (mValue, &Value.string, mWidth);\r
    break;\r
  case EFI_IFR_TYPE_BUFFER :\r
    memcpy (mValue, &Value.u8, mWidth);\r
    break;\r
\r
  case EFI_IFR_TYPE_OTHER :\r
    return;\r
  }\r
}\r
\r
SConfigInfo::~SConfigInfo (\r
  VOID\r
  )\r
{\r
  BUFFER_SAFE_FREE (mValue);\r
}\r
\r
SConfigItem::SConfigItem (\r
  IN CHAR8               *Name,\r
  IN EFI_GUID            *Guid,\r
  IN CHAR8               *Id\r
  )\r
{\r
  mName          = NULL;\r
  mGuid          = NULL;\r
  mId            = NULL;\r
  mInfoStrList = NULL;\r
  mNext        = NULL;\r
\r
  if (Name != NULL) {\r
    if ((mName = new CHAR8[strlen (Name) + 1]) != NULL) {\r
      strcpy (mName, Name);\r
    }\r
  }\r
\r
  if (Guid != NULL) {\r
    if ((mGuid = (EFI_GUID *) new CHAR8[sizeof (EFI_GUID)]) != NULL) {\r
      memcpy (mGuid, Guid, sizeof (EFI_GUID));\r
    }\r
  }\r
\r
  if (Id != NULL) {\r
    if ((mId = new CHAR8[strlen (Id) + 1]) != NULL) {\r
      strcpy (mId, Id);\r
    }\r
  }\r
}\r
\r
SConfigItem::SConfigItem (\r
  IN CHAR8               *Name,\r
  IN EFI_GUID            *Guid,\r
  IN CHAR8               *Id,\r
  IN UINT8               Type,\r
  IN UINT16              Offset,\r
  IN UINT16              Width,\r
  IN EFI_IFR_TYPE_VALUE  Value\r
  )\r
{\r
  mName        = NULL;\r
  mGuid        = NULL;\r
  mId          = NULL;\r
  mInfoStrList = NULL;\r
  mNext        = NULL;\r
\r
  if (Name != NULL) {\r
    if ((mName = new CHAR8[strlen (Name) + 1]) != NULL) {\r
      strcpy (mName, Name);\r
    }\r
  }\r
\r
  if (Guid != NULL) {\r
    if ((mGuid = (EFI_GUID *) new CHAR8[sizeof (EFI_GUID)]) != NULL) {\r
      memcpy (mGuid, Guid, sizeof (EFI_GUID));\r
    }\r
  }\r
\r
  if (Id != NULL) {\r
    if ((mId = new CHAR8[strlen (Id) + 1]) != NULL) {\r
      strcpy (mId, Id);\r
    }\r
  }\r
\r
  mInfoStrList = new SConfigInfo(Type, Offset, Width, Value);\r
}\r
\r
SConfigItem::~SConfigItem (\r
  VOID\r
  )\r
{\r
  SConfigInfo  *Info;\r
\r
  BUFFER_SAFE_FREE (mName);\r
  BUFFER_SAFE_FREE (mGuid);\r
  BUFFER_SAFE_FREE (mId);\r
  while (mInfoStrList != NULL) {\r
    Info = mInfoStrList;\r
    mInfoStrList = mInfoStrList->mNext;\r
\r
    BUFFER_SAFE_FREE (Info);\r
  }\r
}\r
\r
UINT8\r
CVfrBufferConfig::Register (\r
  IN CHAR8               *Name,\r
  IN EFI_GUID            *Guid,\r
  IN CHAR8               *Id\r
  )\r
{\r
  SConfigItem *pNew;\r
\r
  if (Select (Name, Guid) == 0) {\r
    return 1;\r
  }\r
\r
  if ((pNew = new SConfigItem (Name, Guid, Id)) == NULL) {\r
    return 2;\r
  }\r
\r
  if (mItemListHead == NULL) {\r
    mItemListHead = pNew;\r
    mItemListTail = pNew;\r
  } else {\r
    mItemListTail->mNext = pNew;\r
    mItemListTail = pNew;\r
  }\r
  mItemListPos    = pNew;\r
\r
  return 0;\r
}\r
\r
VOID\r
CVfrBufferConfig::Open (\r
  VOID\r
  )\r
{\r
  mItemListPos = mItemListHead;\r
}\r
\r
BOOLEAN\r
CVfrBufferConfig::Eof(\r
  VOID\r
  )\r
{\r
  return (mItemListPos == NULL) ? TRUE : FALSE;\r
}\r
\r
UINT8\r
CVfrBufferConfig::Select (\r
  IN CHAR8    *Name,\r
  IN EFI_GUID *Guid,\r
  IN CHAR8    *Id\r
  )\r
{\r
  SConfigItem *p;\r
\r
  if (Name == NULL || Guid == NULL) {\r
    mItemListPos = mItemListHead;\r
    return 0;\r
  } else {\r
    for (p = mItemListHead; p != NULL; p = p->mNext) {\r
      if ((strcmp (p->mName, Name) != 0) || (memcmp (p->mGuid, Guid, sizeof (EFI_GUID)) != 0)) {\r
        continue;\r
      }\r
\r
      if (Id != NULL) {\r
        if (p->mId == NULL || strcmp (p->mId, Id) != 0) {\r
          continue;\r
        }\r
      } else if (p->mId != NULL) {\r
        continue;\r
      }\r
\r
      mItemListPos = p;\r
      return 0;\r
    }\r
  }\r
\r
  return 1;\r
}\r
\r
UINT8\r
CVfrBufferConfig::Write (\r
  IN CONST CHAR8         Mode,\r
  IN CHAR8               *Name,\r
  IN EFI_GUID            *Guid,\r
  IN CHAR8               *Id,\r
  IN UINT8               Type,\r
  IN UINT16              Offset,\r
  IN UINT32              Width,\r
  IN EFI_IFR_TYPE_VALUE  Value\r
  )\r
{\r
  UINT8         Ret;\r
  SConfigItem   *pItem;\r
  SConfigInfo   *pInfo;\r
\r
  if ((Ret = Select (Name, Guid)) != 0) {\r
    return Ret;\r
  }\r
\r
  switch (Mode) {\r
  case 'a' : // add\r
    if (Select (Name, Guid, Id) != 0) {\r
      if ((pItem = new SConfigItem (Name, Guid, Id, Type, Offset, (UINT16) Width, Value)) == NULL) {\r
        return 2;\r
      }\r
      if (mItemListHead == NULL) {\r
        mItemListHead = pItem;\r
        mItemListTail = pItem;\r
      } else {\r
        mItemListTail->mNext = pItem;\r
        mItemListTail = pItem;\r
      }\r
      mItemListPos = pItem;\r
    } else {\r
      // tranverse the list to find out if there's already the value for the same offset\r
      for (pInfo = mItemListPos->mInfoStrList; pInfo != NULL; pInfo = pInfo->mNext) {\r
        if (pInfo->mOffset == Offset) {\r
          return 0;\r
        }\r
      }\r
      if((pInfo = new SConfigInfo (Type, Offset, Width, Value)) == NULL) {\r
        return 2;\r
      }\r
      pInfo->mNext = mItemListPos->mInfoStrList;\r
      mItemListPos->mInfoStrList = pInfo;\r
    }\r
    break;\r
\r
  case 'd' : // delete\r
    if (mItemListHead == mItemListPos) {\r
      mItemListHead = mItemListPos->mNext;\r
      delete mItemListPos;\r
      break;\r
    }\r
\r
    for (pItem = mItemListHead; pItem->mNext != mItemListPos; pItem = pItem->mNext)\r
      ;\r
\r
    pItem->mNext = mItemListPos->mNext;\r
    if (mItemListTail == mItemListPos) {\r
      mItemListTail = pItem;\r
    }\r
    delete mItemListPos;\r
    mItemListPos = pItem->mNext;\r
    break;\r
\r
  case 'i' : // set info\r
    if (mItemListPos->mId != NULL) {\r
      delete mItemListPos->mId;\r
    }\r
    mItemListPos->mId = NULL;\r
    if (Id != NULL) {\r
      if ((mItemListPos->mId = new CHAR8[strlen (Id) + 1]) == NULL) {\r
        return 2;\r
      }\r
      strcpy (mItemListPos->mId, Id);\r
    }\r
    break;\r
\r
  default :\r
    return 1;\r
  }\r
\r
  return 0;\r
}\r
\r
\r
VOID\r
CVfrBufferConfig::Close (\r
  VOID\r
  )\r
{\r
  mItemListPos = NULL;\r
}\r
\r
#define BYTES_PRE_LINE 0x10\r
\r
VOID\r
CVfrBufferConfig::OutputCFile (\r
  IN FILE  *pFile,\r
  IN CHAR8 *BaseName\r
  )\r
{\r
  CVfrBinaryOutput Output;\r
  SConfigItem      *Item;\r
  SConfigInfo      *Info;\r
  UINT32           TotalLen;\r
\r
  if (pFile == NULL) {\r
    return;\r
  }\r
\r
  for (Item = mItemListHead; Item != NULL; Item = Item->mNext) {\r
    if (Item->mId != NULL || Item->mInfoStrList == NULL) {\r
      continue;\r
    }\r
    fprintf (pFile, "\nunsigned char %s%sBlockName[] = {", BaseName, Item->mName);\r
\r
    TotalLen = sizeof (UINT32);\r
    for (Info = Item->mInfoStrList; Info != NULL; Info = Info->mNext) {\r
      TotalLen += sizeof (UINT16) * 2;\r
    }\r
    Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)&TotalLen, sizeof (UINT32));\r
\r
    for (Info = Item->mInfoStrList; Info != NULL; Info = Info->mNext) {\r
      fprintf (pFile, "\n");\r
      Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)&Info->mOffset, sizeof (UINT16));\r
      Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)&Info->mWidth, sizeof (UINT16));\r
    }\r
    fprintf (pFile, "\n};\n");\r
  }\r
\r
  for (Item = mItemListHead; Item != NULL; Item = Item->mNext) {\r
    if (Item->mId != NULL && Item->mInfoStrList != NULL) {\r
      fprintf (pFile, "\nunsigned char %s%sDefault%s[] = {", BaseName, Item->mName, Item->mId);\r
\r
      TotalLen = sizeof (UINT32);\r
      for (Info = Item->mInfoStrList; Info != NULL; Info = Info->mNext) {\r
        TotalLen += Info->mWidth + sizeof (UINT16) * 2;\r
      }\r
      Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)&TotalLen, sizeof (UINT32));\r
\r
      for (Info = Item->mInfoStrList; Info != NULL; Info = Info->mNext) {\r
        fprintf (pFile, "\n");\r
        Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)&Info->mOffset, sizeof (UINT16));\r
        Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)&Info->mWidth, sizeof (UINT16));\r
        if (Info->mNext == NULL) {\r
          Output.WriteEnd (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)Info->mValue, Info->mWidth);\r
        } else {\r
          Output.WriteLine (pFile, BYTES_PRE_LINE, "  ", (CHAR8 *)Info->mValue, Info->mWidth);\r
        }\r
      }\r
      fprintf (pFile, "\n};\n");\r
    }\r
  }\r
}\r
\r
CVfrBufferConfig::CVfrBufferConfig (\r
  VOID\r
  )\r
{\r
  mItemListHead = NULL;\r
  mItemListTail = NULL;\r
  mItemListPos  = NULL;\r
}\r
\r
CVfrBufferConfig::~CVfrBufferConfig (\r
  VOID\r
  )\r
{\r
  SConfigItem *p;\r
\r
  while (mItemListHead != NULL) {\r
    p = mItemListHead;\r
    mItemListHead = mItemListHead->mNext;\r
    delete p;\r
  }\r
\r
  mItemListHead = NULL;\r
  mItemListTail = NULL;\r
  mItemListPos  = NULL;\r
}\r
\r
CVfrBufferConfig gCVfrBufferConfig;\r
\r
static struct {\r
  CONST CHAR8  *mTypeName;\r
  UINT8  mType;\r
  UINT32 mSize;\r
  UINT32 mAlign;\r
} gInternalTypesTable [] = {\r
  {"UINT64",        EFI_IFR_TYPE_NUM_SIZE_64, sizeof (UINT64),       sizeof (UINT64)},\r
  {"UINT32",        EFI_IFR_TYPE_NUM_SIZE_32, sizeof (UINT32),       sizeof (UINT32)},\r
  {"UINT16",        EFI_IFR_TYPE_NUM_SIZE_16, sizeof (UINT16),       sizeof (UINT16)},\r
  {"UINT8",         EFI_IFR_TYPE_NUM_SIZE_8,  sizeof (UINT8),        sizeof (UINT8)},\r
  {"BOOLEAN",       EFI_IFR_TYPE_BOOLEAN,     sizeof (BOOLEAN),      sizeof (BOOLEAN)},\r
  {"EFI_HII_DATE",  EFI_IFR_TYPE_DATE,        sizeof (EFI_HII_DATE), sizeof (UINT16)},\r
  {"EFI_STRING_ID", EFI_IFR_TYPE_STRING,      sizeof (EFI_STRING_ID),sizeof (EFI_STRING_ID)},\r
  {"EFI_HII_TIME",  EFI_IFR_TYPE_TIME,        sizeof (EFI_HII_TIME), sizeof (UINT8)},\r
  {"EFI_HII_REF",   EFI_IFR_TYPE_REF,         sizeof (EFI_HII_REF),  sizeof (EFI_GUID)},\r
  {NULL,            EFI_IFR_TYPE_OTHER,       0,                     0}\r
};\r
\r
STATIC\r
BOOLEAN\r
_IS_INTERNAL_TYPE (\r
  IN CHAR8 *TypeName\r
  )\r
{\r
  UINT32  Index;\r
\r
  if (TypeName == NULL) {\r
    return FALSE;\r
  }\r
\r
  for (Index = 0; gInternalTypesTable[Index].mTypeName != NULL; Index++) {\r
    if (strcmp (TypeName, gInternalTypesTable[Index].mTypeName) == 0) {\r
      return TRUE;\r
    }\r
  }\r
\r
  return FALSE;\r
}\r
\r
STATIC\r
CHAR8 *\r
TrimHex (\r
  IN  CHAR8   *Str,\r
  OUT bool    *IsHex\r
  )\r
{\r
  *IsHex = FALSE;\r
\r
  while (*Str && *Str == ' ') {\r
    Str++;\r
  }\r
  while (*Str && *Str == '0') {\r
    Str++;\r
  }\r
  if (*Str && (*Str == 'x' || *Str == 'X')) {\r
    Str++;\r
    *IsHex = TRUE;\r
  }\r
\r
  return Str;\r
}\r
\r
UINT32\r
_STR2U32 (\r
  IN CHAR8 *Str\r
  )\r
{\r
  bool    IsHex;\r
  UINT32  Value;\r
  CHAR8    c;\r
\r
  Str = TrimHex (Str, &IsHex);\r
  for (Value = 0; (c = *Str) != '\0'; Str++) {\r
    //\r
    // BUG: does not handle overflow here\r
    //\r
	(IsHex == TRUE) ? (Value <<= 4) : (Value *= 10);\r
\r
    if ((IsHex == TRUE) && (c >= 'a') && (c <= 'f')) {\r
      Value += (c - 'a' + 10);\r
    }\r
    if ((IsHex == TRUE) && (c >= 'A') && (c <= 'F')) {\r
      Value += (c - 'A' + 10);\r
    }\r
    if (c >= '0' && c <= '9') {\r
      Value += (c - '0');\r
    }\r
  }\r
\r
  return Value;\r
}\r
\r
VOID\r
CVfrVarDataTypeDB::RegisterNewType (\r
  IN SVfrDataType  *New\r
  )\r
{\r
  New->mNext               = mDataTypeList;\r
  mDataTypeList            = New;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::ExtractStructTypeName (\r
  IN  CHAR8 *&VarStr,\r
  OUT CHAR8 *TName\r
  )\r
{\r
  if (TName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  while((*VarStr != '\0') && (*VarStr != '.')) {\r
    *TName = *VarStr;\r
    VarStr++;\r
    TName++;\r
  }\r
  *TName = '\0';\r
  if (*VarStr == '.') {\r
    VarStr++;\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::ExtractFieldNameAndArrary (\r
  IN  CHAR8   *&VarStr,\r
  IN  CHAR8   *FName,\r
  OUT UINT32 &ArrayIdx\r
  )\r
{\r
  UINT32 Idx;\r
  CHAR8   ArrayStr[MAX_NAME_LEN + 1];\r
\r
  ArrayIdx = INVALID_ARRAY_INDEX;\r
\r
  if (FName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  while((*VarStr != '\0') &&\r
        (*VarStr != '.') &&\r
        (*VarStr != '[') &&\r
        (*VarStr != ']')) {\r
    *FName = *VarStr;\r
    VarStr++;\r
    FName++;\r
  }\r
  *FName = '\0';\r
\r
  switch (*VarStr) {\r
  case '.' :\r
    VarStr++;\r
  case '\0':\r
    return VFR_RETURN_SUCCESS;\r
  case '[' :\r
    VarStr++;\r
    for (Idx = 0; (Idx < MAX_NAME_LEN) && (*VarStr != '\0') && (*VarStr != ']'); VarStr++, Idx++) {\r
      ArrayStr[Idx] = *VarStr;\r
    }\r
    ArrayStr[Idx] = '\0';\r
\r
    if ((*VarStr != ']') && (ArrayStr[0] == '\0')) {\r
      return VFR_RETURN_DATA_STRING_ERROR;\r
    }\r
    ArrayIdx = _STR2U32 (ArrayStr);\r
    if (*VarStr == ']') {\r
      VarStr++;\r
    }\r
    if (*VarStr == '.') {\r
      VarStr++;\r
    }\r
    return VFR_RETURN_SUCCESS;\r
  case ']':\r
    return VFR_RETURN_DATA_STRING_ERROR;\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetTypeField (\r
  IN  CONST CHAR8   *FName,\r
  IN  SVfrDataType  *Type,\r
  OUT SVfrDataField *&Field\r
  )\r
{\r
  SVfrDataField  *pField = NULL;\r
\r
  if ((FName == NULL) || (Type == NULL)) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pField = Type->mMembers; pField != NULL; pField = pField->mNext) {\r
    //\r
    // For type EFI_IFR_TYPE_TIME, because field name is not correctly wrote,\r
    // add code to adjust it.\r
    //\r
    if (Type->mType == EFI_IFR_TYPE_TIME) {\r
      if (strcmp (FName, "Hour") == 0) {\r
        FName = "Hours";\r
      } else if (strcmp (FName, "Minute") == 0) {\r
        FName = "Minuts";\r
      } else if (strcmp (FName, "Second") == 0) {\r
        FName = "Seconds";\r
      }\r
    }\r
\r
    if (strcmp (pField->mFieldName, FName) == 0) {\r
      Field = pField;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetFieldOffset (\r
  IN  SVfrDataField *Field,\r
  IN  UINT32        ArrayIdx,\r
  OUT UINT32        &Offset\r
  )\r
{\r
  if (Field == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
  \r
  //\r
  // Framework Vfr file Array Index is from 1.\r
  // But Uefi Vfr file Array Index is from 0.\r
  //\r
  if (VfrCompatibleMode && ArrayIdx != INVALID_ARRAY_INDEX) {\r
    if (ArrayIdx == 0) {\r
      return VFR_RETURN_ERROR_ARRARY_NUM;\r
    }\r
    ArrayIdx = ArrayIdx - 1;\r
  }\r
\r
  if ((ArrayIdx != INVALID_ARRAY_INDEX) && ((Field->mArrayNum == 0) || (Field->mArrayNum <= ArrayIdx))) {\r
    return VFR_RETURN_ERROR_ARRARY_NUM;\r
  }\r
  \r
  //\r
  // Be compatible with the current usage\r
  // If ArraryIdx is not specified, the first one is used.\r
  //\r
  // if ArrayNum is larger than zero, ArraryIdx must be specified.\r
  //\r
  // if ((ArrayIdx == INVALID_ARRAY_INDEX) && (Field->mArrayNum > 0)) {\r
  //   return VFR_RETURN_ERROR_ARRARY_NUM;\r
  // }\r
  //\r
\r
  Offset = Field->mOffset + Field->mFieldType->mTotalSize * ((ArrayIdx == INVALID_ARRAY_INDEX) ? 0 : ArrayIdx);\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
UINT8\r
CVfrVarDataTypeDB::GetFieldWidth (\r
  IN SVfrDataField *Field\r
  )\r
{\r
  if (Field == NULL) {\r
    return 0;\r
  }\r
\r
  return Field->mFieldType->mType;\r
}\r
\r
UINT32\r
CVfrVarDataTypeDB::GetFieldSize (\r
  IN SVfrDataField *Field,\r
  IN UINT32       ArrayIdx\r
  )\r
{\r
  if (Field == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if ((ArrayIdx == INVALID_ARRAY_INDEX) && (Field->mArrayNum != 0)) {\r
    return Field->mFieldType->mTotalSize * Field->mArrayNum;\r
  } else {\r
    return Field->mFieldType->mTotalSize;\r
  }\r
}\r
\r
VOID\r
CVfrVarDataTypeDB::InternalTypesListInit (\r
  VOID\r
  )\r
{\r
  SVfrDataType *New   = NULL;\r
  UINT32       Index;\r
\r
  for (Index = 0; gInternalTypesTable[Index].mTypeName != NULL; Index++) {\r
    New                 = new SVfrDataType;\r
    if (New != NULL) {\r
      strcpy (New->mTypeName, gInternalTypesTable[Index].mTypeName);\r
      New->mType        = gInternalTypesTable[Index].mType;\r
      New->mAlign       = gInternalTypesTable[Index].mAlign;\r
      New->mTotalSize   = gInternalTypesTable[Index].mSize;\r
      if (strcmp (gInternalTypesTable[Index].mTypeName, "EFI_HII_DATE") == 0) {\r
        SVfrDataField *pYearField  = new SVfrDataField;\r
        SVfrDataField *pMonthField = new SVfrDataField;\r
        SVfrDataField *pDayField   = new SVfrDataField;\r
\r
        strcpy (pYearField->mFieldName, "Year");\r
        GetDataType ((CHAR8 *)"UINT16", &pYearField->mFieldType);\r
        pYearField->mOffset      = 0;\r
        pYearField->mNext        = pMonthField;\r
        pYearField->mArrayNum    = 0;\r
\r
        strcpy (pMonthField->mFieldName, "Month");\r
        GetDataType ((CHAR8 *)"UINT8", &pMonthField->mFieldType);\r
        pMonthField->mOffset     = 2;\r
        pMonthField->mNext       = pDayField;\r
        pMonthField->mArrayNum   = 0;\r
\r
        strcpy (pDayField->mFieldName, "Day");\r
        GetDataType ((CHAR8 *)"UINT8", &pDayField->mFieldType);\r
        pDayField->mOffset       = 3;\r
        pDayField->mNext         = NULL;\r
        pDayField->mArrayNum     = 0;\r
\r
        New->mMembers            = pYearField;\r
      } else if (strcmp (gInternalTypesTable[Index].mTypeName, "EFI_HII_TIME") == 0) {\r
        SVfrDataField *pHoursField   = new SVfrDataField;\r
        SVfrDataField *pMinutesField = new SVfrDataField;\r
        SVfrDataField *pSecondsField = new SVfrDataField;\r
\r
        strcpy (pHoursField->mFieldName, "Hours");\r
        GetDataType ((CHAR8 *)"UINT8", &pHoursField->mFieldType);\r
        pHoursField->mOffset     = 0;\r
        pHoursField->mNext       = pMinutesField;\r
        pHoursField->mArrayNum   = 0;\r
\r
        strcpy (pMinutesField->mFieldName, "Minutes");\r
        GetDataType ((CHAR8 *)"UINT8", &pMinutesField->mFieldType);\r
        pMinutesField->mOffset   = 1;\r
        pMinutesField->mNext     = pSecondsField;\r
        pMinutesField->mArrayNum = 0;\r
\r
        strcpy (pSecondsField->mFieldName, "Seconds");\r
        GetDataType ((CHAR8 *)"UINT8", &pSecondsField->mFieldType);\r
        pSecondsField->mOffset   = 2;\r
        pSecondsField->mNext     = NULL;\r
        pSecondsField->mArrayNum = 0;\r
\r
        New->mMembers            = pHoursField;\r
      } else if (strcmp (gInternalTypesTable[Index].mTypeName, "EFI_HII_REF") == 0) {\r
        SVfrDataField *pQuestionIdField   = new SVfrDataField;\r
        SVfrDataField *pFormIdField       = new SVfrDataField;\r
        SVfrDataField *pFormSetGuidField  = new SVfrDataField;\r
        SVfrDataField *pDevicePathField   = new SVfrDataField;\r
\r
        strcpy (pQuestionIdField->mFieldName, "QuestionId");\r
        GetDataType ((CHAR8 *)"UINT16", &pQuestionIdField->mFieldType);\r
        pQuestionIdField->mOffset     = 0;\r
        pQuestionIdField->mNext       = pFormIdField;\r
        pQuestionIdField->mArrayNum   = 0;\r
\r
        strcpy (pFormIdField->mFieldName, "FormId");\r
        GetDataType ((CHAR8 *)"UINT16", &pFormIdField->mFieldType);\r
        pFormIdField->mOffset   = 2;\r
        pFormIdField->mNext     = pFormSetGuidField;\r
        pFormIdField->mArrayNum = 0;\r
\r
        strcpy (pFormSetGuidField->mFieldName, "FormSetGuid");\r
        GetDataType ((CHAR8 *)"EFI_GUID", &pFormSetGuidField->mFieldType);\r
        pFormSetGuidField->mOffset   = 4;\r
        pFormSetGuidField->mNext     = pDevicePathField;\r
        pFormSetGuidField->mArrayNum = 0;\r
\r
        strcpy (pDevicePathField->mFieldName, "DevicePath");\r
        GetDataType ((CHAR8 *)"EFI_STRING_ID", &pDevicePathField->mFieldType);\r
        pDevicePathField->mOffset   = 20;\r
        pDevicePathField->mNext     = NULL;\r
        pDevicePathField->mArrayNum = 0;\r
\r
        New->mMembers            = pQuestionIdField;\r
      } else {\r
        New->mMembers            = NULL;\r
      }\r
      New->mNext                 = NULL;\r
      RegisterNewType (New);\r
      New                        = NULL;\r
    }\r
  }\r
}\r
\r
CVfrVarDataTypeDB::CVfrVarDataTypeDB (\r
  VOID\r
  )\r
{\r
  mDataTypeList  = NULL;\r
  mNewDataType   = NULL;\r
  mCurrDataField = NULL;\r
  mPackAlign     = DEFAULT_PACK_ALIGN;\r
  mPackStack     = NULL;\r
  mFirstNewDataTypeName = NULL;\r
\r
  InternalTypesListInit ();\r
}\r
\r
CVfrVarDataTypeDB::~CVfrVarDataTypeDB (\r
  VOID\r
  )\r
{\r
  SVfrDataType      *pType;\r
  SVfrDataField     *pField;\r
  SVfrPackStackNode *pPack;\r
\r
  if (mNewDataType != NULL) {\r
    delete mNewDataType;\r
  }\r
\r
  while (mDataTypeList != NULL) {\r
    pType = mDataTypeList;\r
    mDataTypeList = mDataTypeList->mNext;\r
    while(pType->mMembers != NULL) {\r
      pField = pType->mMembers;\r
      pType->mMembers = pType->mMembers->mNext;\r
      delete pField;\r
    }\r
	delete pType;\r
  }\r
\r
  while (mPackStack != NULL) {\r
    pPack = mPackStack;\r
    mPackStack = mPackStack->mNext;\r
    delete pPack;\r
  }\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::Pack (\r
  IN UINT32         LineNum,\r
  IN UINT8          Action,\r
  IN CHAR8           *Identifier,\r
  IN UINT32         Number\r
  )\r
{\r
  UINT32            PackAlign;\r
  CHAR8             Msg[MAX_STRING_LEN] = {0, };\r
\r
  if (Action & VFR_PACK_SHOW) {\r
    sprintf (Msg, "value of pragma pack(show) == %d", mPackAlign);\r
    gCVfrErrorHandle.PrintMsg (LineNum, NULL, "Warning", Msg);\r
  }\r
\r
  if (Action & VFR_PACK_PUSH) {\r
    SVfrPackStackNode *pNew = NULL;\r
\r
    if ((pNew = new SVfrPackStackNode (Identifier, mPackAlign)) == NULL) {\r
      return VFR_RETURN_FATAL_ERROR;\r
    }\r
    pNew->mNext = mPackStack;\r
    mPackStack  = pNew;\r
  }\r
\r
  if (Action & VFR_PACK_POP) {\r
    SVfrPackStackNode *pNode = NULL;\r
\r
    if (mPackStack == NULL) {\r
      gCVfrErrorHandle.PrintMsg (LineNum, NULL, "Error", "#pragma pack(pop...) : more pops than pushes");\r
    }\r
\r
    for (pNode = mPackStack; pNode != NULL; pNode = pNode->mNext) {\r
      if (pNode->Match (Identifier) == TRUE) {\r
        mPackAlign = pNode->mNumber;\r
        mPackStack = pNode->mNext;\r
      }\r
    }\r
  }\r
\r
  if (Action & VFR_PACK_ASSIGN) {\r
    PackAlign = (Number > 1) ? Number + Number % 2 : Number;\r
    if ((PackAlign == 0) || (PackAlign > 16)) {\r
      gCVfrErrorHandle.PrintMsg (LineNum, NULL, "Error", "expected pragma parameter to be '1', '2', '4', '8', or '16'");\r
    } else {\r
      mPackAlign = PackAlign;\r
    }\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
VOID\r
CVfrVarDataTypeDB::DeclareDataTypeBegin (\r
  VOID\r
  )\r
{\r
  SVfrDataType *pNewType = NULL;\r
\r
  pNewType               = new SVfrDataType;\r
  pNewType->mTypeName[0] = '\0';\r
  pNewType->mType        = EFI_IFR_TYPE_OTHER;\r
  pNewType->mAlign       = DEFAULT_ALIGN;\r
  pNewType->mTotalSize   = 0;\r
  pNewType->mMembers     = NULL;\r
  pNewType->mNext        = NULL;\r
\r
  mNewDataType           = pNewType;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::SetNewTypeName (\r
  IN CHAR8   *TypeName\r
  )\r
{\r
  SVfrDataType *pType;\r
\r
  if (mNewDataType == NULL) {\r
    return VFR_RETURN_ERROR_SKIPED;\r
  }\r
  if (TypeName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
  if (strlen(TypeName) >= MAX_NAME_LEN) {\r
    return VFR_RETURN_INVALID_PARAMETER;\r
  }\r
\r
  for (pType = mDataTypeList; pType != NULL; pType = pType->mNext) {\r
    if (strcmp(pType->mTypeName, TypeName) == 0) {\r
      return VFR_RETURN_REDEFINED;\r
    }\r
  }\r
\r
  strcpy(mNewDataType->mTypeName, TypeName);\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::DataTypeAddField (\r
  IN CHAR8   *FieldName,\r
  IN CHAR8   *TypeName,\r
  IN UINT32 ArrayNum\r
  )\r
{\r
  SVfrDataField       *pNewField  = NULL;\r
  SVfrDataType        *pFieldType = NULL;\r
  SVfrDataField       *pTmp;\r
  UINT32              Align;\r
\r
  CHECK_ERROR_RETURN (GetDataType (TypeName, &pFieldType), VFR_RETURN_SUCCESS);\r
\r
  if (strlen (FieldName) >= MAX_NAME_LEN) {\r
   return VFR_RETURN_INVALID_PARAMETER;\r
  }\r
\r
  for (pTmp = mNewDataType->mMembers; pTmp != NULL; pTmp = pTmp->mNext) {\r
    if (strcmp (pTmp->mFieldName, FieldName) == 0) {\r
      return VFR_RETURN_REDEFINED;\r
    }\r
  }\r
\r
  Align = MIN (mPackAlign, pFieldType->mAlign);\r
\r
  if ((pNewField = new SVfrDataField) == NULL) {\r
    return VFR_RETURN_OUT_FOR_RESOURCES;\r
  }\r
  strcpy (pNewField->mFieldName, FieldName);\r
  pNewField->mFieldType    = pFieldType;\r
  pNewField->mArrayNum     = ArrayNum;\r
  if ((mNewDataType->mTotalSize % Align) == 0) {\r
    pNewField->mOffset     = mNewDataType->mTotalSize;\r
  } else {\r
    pNewField->mOffset     = mNewDataType->mTotalSize + ALIGN_STUFF(mNewDataType->mTotalSize, Align);\r
  }\r
  if (mNewDataType->mMembers == NULL) {\r
    mNewDataType->mMembers = pNewField;\r
    pNewField->mNext       = NULL;\r
  } else {\r
    for (pTmp = mNewDataType->mMembers; pTmp->mNext != NULL; pTmp = pTmp->mNext)\r
      ;\r
    pTmp->mNext            = pNewField;\r
    pNewField->mNext       = NULL;\r
  }\r
\r
  mNewDataType->mAlign     = MIN (mPackAlign, MAX (pFieldType->mAlign, mNewDataType->mAlign));\r
  mNewDataType->mTotalSize = pNewField->mOffset + (pNewField->mFieldType->mTotalSize) * ((ArrayNum == 0) ? 1 : ArrayNum);\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
VOID\r
CVfrVarDataTypeDB::DeclareDataTypeEnd (\r
  VOID\r
  )\r
{\r
  if (mNewDataType->mTypeName[0] == '\0') {\r
    return;\r
  }\r
\r
  if ((mNewDataType->mTotalSize % mNewDataType->mAlign) !=0) {\r
    mNewDataType->mTotalSize += ALIGN_STUFF (mNewDataType->mTotalSize, mNewDataType->mAlign);\r
  }\r
\r
  RegisterNewType (mNewDataType);\r
  if (mFirstNewDataTypeName == NULL) {\r
    mFirstNewDataTypeName = mNewDataType->mTypeName;\r
  }\r
\r
  mNewDataType             = NULL;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetDataType (\r
  IN  CHAR8         *TypeName,\r
  OUT SVfrDataType **DataType\r
  )\r
{\r
  SVfrDataType *pDataType = NULL;\r
\r
  if (TypeName == NULL) {\r
    return VFR_RETURN_ERROR_SKIPED;\r
  }\r
\r
  if (DataType == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  *DataType = NULL;\r
\r
  for (pDataType = mDataTypeList; pDataType != NULL; pDataType = pDataType->mNext) {\r
    if (strcmp (TypeName, pDataType->mTypeName) == 0) {\r
      *DataType = pDataType;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetDataTypeSize (\r
  IN  UINT8   DataType,\r
  OUT UINT32 *Size\r
  )\r
{\r
  SVfrDataType *pDataType = NULL;\r
\r
  if (Size == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  *Size    = 0;\r
  DataType = DataType & 0x0F;\r
\r
  //\r
  // For user defined data type, the size can't be got by this function.\r
  //\r
  if (DataType == EFI_IFR_TYPE_OTHER) {\r
    return VFR_RETURN_SUCCESS;\r
  }\r
\r
  for (pDataType = mDataTypeList; pDataType != NULL; pDataType = pDataType->mNext) {\r
    if (DataType == pDataType->mType) {\r
      *Size = pDataType->mTotalSize;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetDataTypeSize (\r
  IN  CHAR8   *TypeName,\r
  OUT UINT32 *Size\r
  )\r
{\r
  SVfrDataType *pDataType = NULL;\r
\r
  if (Size == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  *Size = 0;\r
\r
  for (pDataType = mDataTypeList; pDataType != NULL; pDataType = pDataType->mNext) {\r
    if (strcmp (TypeName, pDataType->mTypeName) == 0) {\r
      *Size = pDataType->mTotalSize;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetDataFieldInfo (\r
  IN  CHAR8     *VarStr,\r
  OUT UINT16   &Offset,\r
  OUT UINT8    &Type,\r
  OUT UINT32   &Size\r
  )\r
{\r
  CHAR8               TName[MAX_NAME_LEN], FName[MAX_NAME_LEN];\r
  UINT32              ArrayIdx, Tmp;\r
  SVfrDataType        *pType  = NULL;\r
  SVfrDataField       *pField = NULL;\r
\r
  Offset = 0;\r
  Type   = EFI_IFR_TYPE_OTHER;\r
  Size   = 0;\r
\r
  CHECK_ERROR_RETURN (ExtractStructTypeName (VarStr, TName), VFR_RETURN_SUCCESS);\r
  CHECK_ERROR_RETURN (GetDataType (TName, &pType), VFR_RETURN_SUCCESS);\r
\r
  //\r
  // if it is not struct data type\r
  //\r
  Type  = pType->mType;\r
  Size  = pType->mTotalSize;\r
\r
  while (*VarStr != '\0') {\r
  	CHECK_ERROR_RETURN(ExtractFieldNameAndArrary(VarStr, FName, ArrayIdx), VFR_RETURN_SUCCESS);\r
    CHECK_ERROR_RETURN(GetTypeField (FName, pType, pField), VFR_RETURN_SUCCESS);\r
    pType  = pField->mFieldType;\r
    CHECK_ERROR_RETURN(GetFieldOffset (pField, ArrayIdx, Tmp), VFR_RETURN_SUCCESS);\r
    Offset = (UINT16) (Offset + Tmp);\r
    Type   = GetFieldWidth (pField);\r
    Size   = GetFieldSize (pField, ArrayIdx);\r
  }\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrVarDataTypeDB::GetUserDefinedTypeNameList  (\r
  OUT CHAR8      ***NameList,\r
  OUT UINT32    *ListSize\r
  )\r
{\r
  UINT32       Index;\r
  SVfrDataType *pType;\r
\r
  if ((NameList == NULL) || (ListSize == NULL)) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  *NameList = NULL;\r
  *ListSize = 0;\r
\r
  for (pType = mDataTypeList; pType != NULL; pType = pType->mNext) {\r
    if (_IS_INTERNAL_TYPE(pType->mTypeName) == FALSE) {\r
      (*ListSize)++;\r
    }\r
  }\r
\r
  if (*ListSize == 0) {\r
    return VFR_RETURN_SUCCESS;\r
  }\r
\r
  if ((*NameList = new CHAR8*[*ListSize]) == NULL) {\r
    *ListSize = 0;\r
    return VFR_RETURN_OUT_FOR_RESOURCES;\r
  }\r
\r
  for (Index = 0, pType = mDataTypeList; pType != NULL; pType = pType->mNext, Index++) {\r
    if (_IS_INTERNAL_TYPE(pType->mTypeName) == FALSE) {\r
      (*NameList)[Index] = pType->mTypeName;\r
    }\r
  }\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
BOOLEAN\r
CVfrVarDataTypeDB::IsTypeNameDefined (\r
  IN CHAR8 *TypeName\r
  )\r
{\r
  SVfrDataType *pType;\r
\r
  if (TypeName == NULL) {\r
    return FALSE;\r
  }\r
\r
  for (pType = mDataTypeList; pType != NULL; pType = pType->mNext) {\r
    if (strcmp (pType->mTypeName, TypeName) == 0) {\r
      return TRUE;\r
    }\r
  }\r
\r
  return FALSE;\r
}\r
\r
VOID\r
CVfrVarDataTypeDB::Dump (\r
  IN FILE         *File\r
  )\r
{\r
  SVfrDataType  *pTNode;\r
  SVfrDataField *pFNode;\r
\r
  fprintf (File, "\n\n***************************************************************\n");\r
  fprintf (File, "\t\tmPackAlign = %x\n", mPackAlign);\r
  for (pTNode = mDataTypeList; pTNode != NULL; pTNode = pTNode->mNext) {\r
    fprintf (File, "\t\tstruct %s : mAlign [%d] mTotalSize [0x%x]\n\n", pTNode->mTypeName, pTNode->mAlign, pTNode->mTotalSize);\r
    fprintf (File, "\t\tstruct %s {\n", pTNode->mTypeName);\r
    for (pFNode = pTNode->mMembers; pFNode != NULL; pFNode = pFNode->mNext) {\r
      if (pFNode->mArrayNum > 0) {\r
        fprintf (File, "\t\t\t+%08d[%08x] %s[%d] <%s>\n", pFNode->mOffset, pFNode->mOffset, \r
                  pFNode->mFieldName, pFNode->mArrayNum, pFNode->mFieldType->mTypeName);\r
      } else {\r
        fprintf (File, "\t\t\t+%08d[%08x] %s <%s>\n", pFNode->mOffset, pFNode->mOffset, \r
                  pFNode->mFieldName, pFNode->mFieldType->mTypeName);\r
      }\r
    }\r
    fprintf (File, "\t\t};\n");\r
  fprintf (File, "---------------------------------------------------------------\n");\r
  }\r
  fprintf (File, "***************************************************************\n");\r
}\r
\r
#ifdef CVFR_VARDATATYPEDB_DEBUG\r
VOID\r
CVfrVarDataTypeDB::ParserDB (\r
  VOID\r
  )\r
{\r
  SVfrDataType  *pTNode;\r
  SVfrDataField *pFNode;\r
\r
  printf ("***************************************************************\n");\r
  printf ("\t\tmPackAlign = %x\n", mPackAlign);\r
  for (pTNode = mDataTypeList; pTNode != NULL; pTNode = pTNode->mNext) {\r
    printf ("\t\tstruct %s : mAlign [%x] mTotalSize [%x]\n\n", pTNode->mTypeName, pTNode->mAlign, pTNode->mTotalSize);\r
    printf ("\t\tstruct %s {\n", pTNode->mTypeName);\r
    for (pFNode = pTNode->mMembers; pFNode != NULL; pFNode = pFNode->mNext) {\r
      printf ("\t\t\t%s\t%s\n", pFNode->mFieldType->mTypeName, pFNode->mFieldName);\r
    }\r
    printf ("\t\t};\n");\r
	printf ("---------------------------------------------------------------\n");\r
  }\r
  printf ("***************************************************************\n");\r
}\r
#endif\r
\r
SVfrVarStorageNode::SVfrVarStorageNode (\r
  IN EFI_GUID              *Guid,\r
  IN CHAR8                 *StoreName,\r
  IN EFI_VARSTORE_ID       VarStoreId,\r
  IN EFI_STRING_ID         VarName,\r
  IN UINT32                VarSize,\r
  IN BOOLEAN               Flag\r
  )\r
{\r
  if (Guid != NULL) {\r
    mGuid = *Guid;\r
  } else {\r
    memset (&mGuid, 0, sizeof (EFI_GUID));\r
  }\r
  if (StoreName != NULL) {\r
    mVarStoreName = new CHAR8[strlen(StoreName) + 1];\r
    strcpy (mVarStoreName, StoreName);\r
  } else {\r
    mVarStoreName = NULL;\r
  }\r
  mNext                            = NULL;\r
  mVarStoreId                      = VarStoreId;\r
  mVarStoreType                    = EFI_VFR_VARSTORE_EFI;\r
  mStorageInfo.mEfiVar.mEfiVarName = VarName;\r
  mStorageInfo.mEfiVar.mEfiVarSize = VarSize;\r
  mAssignedFlag                    = Flag;\r
}\r
\r
SVfrVarStorageNode::SVfrVarStorageNode (\r
  IN EFI_GUID              *Guid,\r
  IN CHAR8                 *StoreName,\r
  IN EFI_VARSTORE_ID       VarStoreId,\r
  IN SVfrDataType          *DataType,\r
  IN BOOLEAN               Flag\r
  )\r
{\r
  if (Guid != NULL) {\r
    mGuid = *Guid;\r
  } else {\r
    memset (&mGuid, 0, sizeof (EFI_GUID));\r
  }\r
  if (StoreName != NULL) {\r
    mVarStoreName = new CHAR8[strlen(StoreName) + 1];\r
    strcpy (mVarStoreName, StoreName);\r
  } else {\r
    mVarStoreName = NULL;\r
  }\r
  mNext                    = NULL;\r
  mVarStoreId              = VarStoreId;\r
  mVarStoreType            = EFI_VFR_VARSTORE_BUFFER;\r
  mStorageInfo.mDataType   = DataType;\r
  mAssignedFlag            = Flag;\r
}\r
\r
SVfrVarStorageNode::SVfrVarStorageNode (\r
  IN CHAR8                 *StoreName,\r
  IN EFI_VARSTORE_ID       VarStoreId\r
  )\r
{\r
  if (StoreName != NULL) {\r
    mVarStoreName = new CHAR8[strlen(StoreName) + 1];\r
    strcpy (mVarStoreName, StoreName);\r
  } else {\r
    mVarStoreName = NULL;\r
  }\r
  mNext                              = NULL;\r
  mVarStoreId                        = VarStoreId;\r
  mVarStoreType                      = EFI_VFR_VARSTORE_NAME;\r
  mStorageInfo.mNameSpace.mNameTable = new EFI_VARSTORE_ID[DEFAULT_NAME_TABLE_ITEMS];\r
  mStorageInfo.mNameSpace.mTableSize = 0;\r
}\r
\r
SVfrVarStorageNode::~SVfrVarStorageNode (\r
  VOID\r
  )\r
{\r
  if (mVarStoreName != NULL) {\r
    delete mVarStoreName;\r
  }\r
\r
  if (mVarStoreType == EFI_VFR_VARSTORE_NAME) {\r
    delete mStorageInfo.mNameSpace.mNameTable;\r
  }\r
}\r
\r
CVfrDataStorage::CVfrDataStorage (\r
  VOID\r
  )\r
{\r
  UINT32 Index;\r
\r
  for (Index = 0; Index < EFI_FREE_VARSTORE_ID_BITMAP_SIZE; Index++) {\r
    mFreeVarStoreIdBitMap[Index] = 0;\r
  }\r
\r
  // Question ID 0 is reserved.\r
  mFreeVarStoreIdBitMap[0] = 0x80000000;\r
\r
  mBufferVarStoreList      = NULL;\r
  mEfiVarStoreList         = NULL;\r
  mNameVarStoreList        = NULL;\r
  mCurrVarStorageNode      = NULL;\r
  mNewVarStorageNode       = NULL;\r
  mBufferFieldInfoListHead = NULL;\r
  mBufferFieldInfoListTail = NULL;\r
}\r
\r
CVfrDataStorage::~CVfrDataStorage (\r
  VOID\r
  )\r
{\r
  SVfrVarStorageNode *pNode;\r
\r
  while (mBufferVarStoreList != NULL) {\r
    pNode = mBufferVarStoreList;\r
    mBufferVarStoreList = mBufferVarStoreList->mNext;\r
    delete pNode;\r
  }\r
  while (mEfiVarStoreList != NULL) {\r
    pNode = mEfiVarStoreList;\r
    mEfiVarStoreList = mEfiVarStoreList->mNext;\r
    delete pNode;\r
  }\r
  while (mNameVarStoreList != NULL) {\r
    pNode = mNameVarStoreList;\r
    mNameVarStoreList = mNameVarStoreList->mNext;\r
    delete pNode;\r
  }\r
  if (mNewVarStorageNode != NULL) {\r
    delete mNewVarStorageNode;\r
  }\r
}\r
\r
EFI_VARSTORE_ID\r
CVfrDataStorage::GetFreeVarStoreId (\r
  EFI_VFR_VARSTORE_TYPE VarType\r
  )\r
{\r
  UINT32  Index, Mask, Offset;\r
  \r
  //\r
  // Assign the different ID range for the different type VarStore to support Framework Vfr\r
  //\r
  Index = 0;\r
  if ((!VfrCompatibleMode) || (VarType == EFI_VFR_VARSTORE_BUFFER)) {\r
    Index = 0;\r
  } else if (VarType == EFI_VFR_VARSTORE_EFI) {\r
    Index = 1;\r
  } else if (VarType == EFI_VFR_VARSTORE_NAME) {\r
    Index = 2;\r
  }\r
\r
  for (; Index < EFI_FREE_VARSTORE_ID_BITMAP_SIZE; Index++) {\r
    if (mFreeVarStoreIdBitMap[Index] != 0xFFFFFFFF) {\r
      break;\r
    }\r
  }\r
\r
  for (Offset = 0, Mask = 0x80000000; Mask != 0; Mask >>= 1, Offset++) {\r
    if ((mFreeVarStoreIdBitMap[Index] & Mask) == 0) {\r
      mFreeVarStoreIdBitMap[Index] |= Mask;\r
      return (EFI_VARSTORE_ID)((Index << EFI_BITS_SHIFT_PER_UINT32) + Offset);\r
    }\r
  }\r
\r
  return EFI_VARSTORE_ID_INVALID;\r
}\r
\r
BOOLEAN\r
CVfrDataStorage::ChekVarStoreIdFree (\r
  IN EFI_VARSTORE_ID VarStoreId\r
  )\r
{\r
  UINT32 Index  = (VarStoreId / EFI_BITS_PER_UINT32);\r
  UINT32 Offset = (VarStoreId % EFI_BITS_PER_UINT32);\r
\r
  return (mFreeVarStoreIdBitMap[Index] & (0x80000000 >> Offset)) == 0;\r
}\r
\r
VOID\r
CVfrDataStorage::MarkVarStoreIdUsed (\r
  IN EFI_VARSTORE_ID VarStoreId\r
  )\r
{\r
  UINT32 Index  = (VarStoreId / EFI_BITS_PER_UINT32);\r
  UINT32 Offset = (VarStoreId % EFI_BITS_PER_UINT32);\r
\r
  mFreeVarStoreIdBitMap[Index] |= (0x80000000 >> Offset);\r
}\r
\r
VOID\r
CVfrDataStorage::MarkVarStoreIdUnused (\r
  IN EFI_VARSTORE_ID VarStoreId\r
  )\r
{\r
  UINT32 Index  = (VarStoreId / EFI_BITS_PER_UINT32);\r
  UINT32 Offset = (VarStoreId % EFI_BITS_PER_UINT32);\r
\r
  mFreeVarStoreIdBitMap[Index] &= ~(0x80000000 >> Offset);\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::DeclareNameVarStoreBegin (\r
  IN CHAR8           *StoreName,\r
  IN EFI_VARSTORE_ID VarStoreId\r
  )\r
{\r
  SVfrVarStorageNode *pNode = NULL;\r
  EFI_VARSTORE_ID    TmpVarStoreId;\r
\r
  if (StoreName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if (GetVarStoreId (StoreName, &TmpVarStoreId) == VFR_RETURN_SUCCESS) {\r
    return VFR_RETURN_REDEFINED;\r
  }\r
  \r
  if (VarStoreId == EFI_VARSTORE_ID_INVALID) {\r
    VarStoreId = GetFreeVarStoreId (EFI_VFR_VARSTORE_NAME);\r
  } else {\r
    if (ChekVarStoreIdFree (VarStoreId) == FALSE) {\r
      return VFR_RETURN_VARSTOREID_REDEFINED;\r
    }\r
    MarkVarStoreIdUsed (VarStoreId);\r
  }\r
\r
  if ((pNode = new SVfrVarStorageNode (StoreName, VarStoreId)) == NULL) {\r
    return VFR_RETURN_UNDEFINED;\r
  }\r
\r
  mNewVarStorageNode = pNode;\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::NameTableAddItem (\r
  IN EFI_STRING_ID  Item\r
  )\r
{\r
  EFI_VARSTORE_ID *NewTable, *OldTable;\r
  UINT32          TableSize;\r
\r
  OldTable  = mNewVarStorageNode->mStorageInfo.mNameSpace.mNameTable;\r
  TableSize = mNewVarStorageNode->mStorageInfo.mNameSpace.mTableSize;\r
\r
  if ((TableSize != 0) && ((TableSize % DEFAULT_NAME_TABLE_ITEMS) == 0)) {\r
    if ((NewTable = new EFI_VARSTORE_ID[TableSize + DEFAULT_NAME_TABLE_ITEMS]) == NULL) {\r
      return VFR_RETURN_OUT_FOR_RESOURCES;\r
    }\r
    memcpy (NewTable, OldTable, TableSize);\r
    mNewVarStorageNode->mStorageInfo.mNameSpace.mNameTable = NewTable;\r
  }\r
\r
  mNewVarStorageNode->mStorageInfo.mNameSpace.mNameTable[TableSize++] = Item;\r
  mNewVarStorageNode->mStorageInfo.mNameSpace.mTableSize = TableSize;\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::DeclareNameVarStoreEnd (\r
  IN EFI_GUID *Guid\r
  )\r
{\r
  mNewVarStorageNode->mGuid = *Guid;\r
  mNewVarStorageNode->mNext = mNameVarStoreList;\r
  mNameVarStoreList         = mNewVarStorageNode;\r
\r
  mNewVarStorageNode        = NULL;\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE \r
CVfrDataStorage::DeclareEfiVarStore (\r
  IN CHAR8          *StoreName, \r
  IN EFI_GUID       *Guid, \r
  IN EFI_STRING_ID  NameStrId,\r
  IN UINT32         VarSize,\r
  IN BOOLEAN        Flag\r
  )\r
{\r
  SVfrVarStorageNode *pNode;\r
  EFI_VARSTORE_ID    VarStoreId;\r
\r
  if ((StoreName == NULL) || (Guid == NULL)) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if (VarSize > sizeof (UINT64)) {\r
    return VFR_RETURN_EFIVARSTORE_SIZE_ERROR;\r
  }\r
\r
  if (GetVarStoreId (StoreName, &VarStoreId, Guid) == VFR_RETURN_SUCCESS) {\r
    return VFR_RETURN_REDEFINED;\r
  }\r
\r
  VarStoreId = GetFreeVarStoreId (EFI_VFR_VARSTORE_EFI);\r
  if ((pNode = new SVfrVarStorageNode (Guid, StoreName, VarStoreId, NameStrId, VarSize, Flag)) == NULL) {\r
    return VFR_RETURN_OUT_FOR_RESOURCES;\r
  }\r
\r
  pNode->mNext       = mEfiVarStoreList;\r
  mEfiVarStoreList   = pNode;\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE \r
CVfrDataStorage::DeclareBufferVarStore (\r
  IN CHAR8             *StoreName, \r
  IN EFI_GUID          *Guid, \r
  IN CVfrVarDataTypeDB *DataTypeDB,\r
  IN CHAR8             *TypeName,\r
  IN EFI_VARSTORE_ID   VarStoreId,\r
  IN BOOLEAN           Flag\r
  )\r
{\r
  SVfrVarStorageNode   *pNew = NULL;\r
  SVfrDataType         *pDataType = NULL;\r
  EFI_VARSTORE_ID      TempVarStoreId;\r
\r
  if ((StoreName == NULL) || (Guid == NULL) || (DataTypeDB == NULL)) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if (GetVarStoreId (StoreName, &TempVarStoreId, Guid) == VFR_RETURN_SUCCESS) {\r
    return VFR_RETURN_REDEFINED;\r
  }\r
\r
  CHECK_ERROR_RETURN(DataTypeDB->GetDataType (TypeName, &pDataType), VFR_RETURN_SUCCESS);\r
\r
  if (VarStoreId == EFI_VARSTORE_ID_INVALID) {\r
    VarStoreId = GetFreeVarStoreId (EFI_VFR_VARSTORE_BUFFER);\r
  } else {\r
    if (ChekVarStoreIdFree (VarStoreId) == FALSE) {\r
      return VFR_RETURN_VARSTOREID_REDEFINED;\r
    }\r
    MarkVarStoreIdUsed (VarStoreId);\r
  }\r
\r
  if ((pNew = new SVfrVarStorageNode (Guid, StoreName, VarStoreId, pDataType, Flag)) == NULL) {\r
    return VFR_RETURN_OUT_FOR_RESOURCES;\r
  }\r
\r
  pNew->mNext         = mBufferVarStoreList;\r
  mBufferVarStoreList = pNew;\r
\r
  if (gCVfrBufferConfig.Register(StoreName, Guid) != 0) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE \r
CVfrDataStorage::GetVarStoreByDataType (\r
  IN  CHAR8              *DataTypeName,\r
  OUT SVfrVarStorageNode **VarNode,\r
  IN  EFI_GUID           *VarGuid\r
  )\r
{\r
  SVfrVarStorageNode    *pNode;\r
  SVfrVarStorageNode    *MatchNode;\r
  \r
  //\r
  // Framework VFR uses Data type name as varstore name, so don't need check again.\r
  //\r
  if (VfrCompatibleMode) {\r
    return VFR_RETURN_UNDEFINED;\r
  }\r
\r
  MatchNode = NULL;\r
  for (pNode = mBufferVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mStorageInfo.mDataType->mTypeName, DataTypeName) != 0) {\r
      continue;\r
    }\r
\r
    if ((VarGuid != NULL)) {\r
      if (memcmp (VarGuid, &pNode->mGuid, sizeof (EFI_GUID)) == 0) {\r
        *VarNode = pNode;\r
        return VFR_RETURN_SUCCESS;\r
      }\r
    } else {\r
      if (MatchNode == NULL) {\r
        MatchNode = pNode;\r
      } else {\r
        //\r
        // More than one varstores referred the same data structures.\r
        //\r
        return VFR_RETURN_VARSTORE_DATATYPE_REDEFINED_ERROR;\r
      }\r
    }\r
  }\r
  \r
  if (MatchNode == NULL) {\r
    return VFR_RETURN_UNDEFINED;\r
  }\r
\r
  *VarNode = MatchNode;\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VARSTORE_ID \r
CVfrDataStorage::CheckGuidField (\r
  IN  SVfrVarStorageNode   *pNode,\r
  IN  EFI_GUID             *StoreGuid,\r
  IN  BOOLEAN              *HasFoundOne,\r
  OUT EFI_VFR_RETURN_CODE  *ReturnCode\r
  )\r
{\r
  if (StoreGuid != NULL) {\r
    //\r
    // If has guid info, compare the guid filed.\r
    //\r
    if (memcmp (StoreGuid, &pNode->mGuid, sizeof (EFI_GUID)) == 0) {\r
      //\r
      // Both name and guid are same, this this varstore.\r
      //\r
      mCurrVarStorageNode = pNode;\r
      *ReturnCode = VFR_RETURN_SUCCESS;\r
      return TRUE;\r
    }\r
  } else {\r
    //\r
    // Not has Guid field, check whether this name is the only one.\r
    //\r
    if (*HasFoundOne) {\r
      //\r
      // The name has conflict, return name redefined.\r
      //\r
      *ReturnCode = VFR_RETURN_VARSTORE_NAME_REDEFINED_ERROR;\r
      return TRUE;\r
    }\r
\r
    *HasFoundOne = TRUE;\r
    mCurrVarStorageNode = pNode;\r
  }\r
\r
  return FALSE;\r
}\r
\r
/**\r
  Base on the input store name and guid to find the varstore id. \r
\r
  If both name and guid are inputed, base on the name and guid to\r
  found the varstore. If only name inputed, base on the name to\r
  found the varstore and go on to check whether more than one varstore\r
  has the same name. If only has found one varstore, return this\r
  varstore; if more than one varstore has same name, return varstore\r
  name redefined error. If no varstore found by varstore name, call\r
  function GetVarStoreByDataType and use inputed varstore name as \r
  data type name to search.\r
**/\r
EFI_VFR_RETURN_CODE \r
CVfrDataStorage::GetVarStoreId (\r
  IN  CHAR8           *StoreName,\r
  OUT EFI_VARSTORE_ID *VarStoreId,\r
  IN  EFI_GUID        *StoreGuid\r
  )\r
{\r
  EFI_VFR_RETURN_CODE   ReturnCode;\r
  SVfrVarStorageNode    *pNode;\r
  BOOLEAN               HasFoundOne = FALSE;\r
\r
  mCurrVarStorageNode = NULL;\r
\r
  for (pNode = mBufferVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mVarStoreName, StoreName) == 0) {\r
      if (CheckGuidField(pNode, StoreGuid, &HasFoundOne, &ReturnCode)) {\r
        *VarStoreId = mCurrVarStorageNode->mVarStoreId;\r
        return ReturnCode;\r
      }\r
    }\r
  }\r
\r
  for (pNode = mEfiVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mVarStoreName, StoreName) == 0) {\r
      if (CheckGuidField(pNode, StoreGuid, &HasFoundOne, &ReturnCode)) {\r
        *VarStoreId = mCurrVarStorageNode->mVarStoreId;\r
        return ReturnCode;\r
      }\r
    }\r
  }\r
\r
  for (pNode = mNameVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mVarStoreName, StoreName) == 0) {\r
      if (CheckGuidField(pNode, StoreGuid, &HasFoundOne, &ReturnCode)) {\r
        *VarStoreId = mCurrVarStorageNode->mVarStoreId;\r
        return ReturnCode;\r
      }\r
    }\r
  }\r
\r
  if (HasFoundOne) {\r
    *VarStoreId = mCurrVarStorageNode->mVarStoreId;\r
    return VFR_RETURN_SUCCESS;\r
  }\r
\r
  *VarStoreId         = EFI_VARSTORE_ID_INVALID;\r
\r
  //\r
  // Assume that Data strucutre name is used as StoreName, and check again. \r
  //\r
  ReturnCode = GetVarStoreByDataType (StoreName, &pNode, StoreGuid);\r
  if (pNode != NULL) {\r
    mCurrVarStorageNode = pNode;\r
    *VarStoreId = pNode->mVarStoreId;\r
  }\r
  \r
  return ReturnCode;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::GetBufferVarStoreDataTypeName (\r
  IN  EFI_VARSTORE_ID        VarStoreId,\r
  OUT CHAR8                  **DataTypeName\r
  )\r
{\r
  SVfrVarStorageNode    *pNode;\r
\r
  if (VarStoreId == EFI_VARSTORE_ID_INVALID) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pNode = mBufferVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      *DataTypeName = pNode->mStorageInfo.mDataType->mTypeName;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_VARSTORE_TYPE\r
CVfrDataStorage::GetVarStoreType (\r
  IN  EFI_VARSTORE_ID        VarStoreId\r
  )\r
{\r
  SVfrVarStorageNode    *pNode;\r
  EFI_VFR_VARSTORE_TYPE VarStoreType;\r
\r
  VarStoreType = EFI_VFR_VARSTORE_INVALID;\r
\r
  if (VarStoreId == EFI_VARSTORE_ID_INVALID) {\r
    return VarStoreType;\r
  }\r
\r
  for (pNode = mBufferVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      VarStoreType = pNode->mVarStoreType;\r
      return VarStoreType;\r
    }\r
  }\r
\r
  for (pNode = mEfiVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      VarStoreType = pNode->mVarStoreType;\r
      return VarStoreType;\r
    }\r
  }\r
\r
  for (pNode = mNameVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      VarStoreType = pNode->mVarStoreType;\r
      return VarStoreType;\r
    }\r
  }\r
\r
  return VarStoreType;\r
}\r
\r
EFI_GUID *\r
CVfrDataStorage::GetVarStoreGuid (\r
  IN  EFI_VARSTORE_ID        VarStoreId\r
  )\r
{\r
  SVfrVarStorageNode    *pNode;\r
  EFI_GUID              *VarGuid;\r
\r
  VarGuid = NULL;\r
\r
  if (VarStoreId == EFI_VARSTORE_ID_INVALID) {\r
    return VarGuid;\r
  }\r
\r
  for (pNode = mBufferVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      VarGuid = &pNode->mGuid;\r
      return VarGuid;\r
    }\r
  }\r
\r
  for (pNode = mEfiVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      VarGuid = &pNode->mGuid;\r
      return VarGuid;\r
    }\r
  }\r
\r
  for (pNode = mNameVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      VarGuid = &pNode->mGuid;\r
      return VarGuid;\r
    }\r
  }\r
\r
  return VarGuid;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::GetVarStoreName (\r
  IN  EFI_VARSTORE_ID VarStoreId, \r
  OUT CHAR8           **VarStoreName\r
  )\r
{\r
  SVfrVarStorageNode    *pNode;\r
\r
  if (VarStoreName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pNode = mBufferVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      *VarStoreName = pNode->mVarStoreName;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  for (pNode = mEfiVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      *VarStoreName = pNode->mVarStoreName;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  for (pNode = mNameVarStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mVarStoreId == VarStoreId) {\r
      *VarStoreName = pNode->mVarStoreName;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  *VarStoreName = NULL;\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::GetEfiVarStoreInfo (\r
  IN OUT EFI_VARSTORE_INFO  *Info\r
  )\r
{\r
  if (Info == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if (mCurrVarStorageNode == NULL) {\r
    return VFR_RETURN_GET_EFIVARSTORE_ERROR;\r
  }\r
\r
  Info->mInfo.mVarName = mCurrVarStorageNode->mStorageInfo.mEfiVar.mEfiVarName;\r
  Info->mVarTotalSize  = mCurrVarStorageNode->mStorageInfo.mEfiVar.mEfiVarSize;\r
  switch (Info->mVarTotalSize) {\r
  case 1:\r
    Info->mVarType = EFI_IFR_TYPE_NUM_SIZE_8;\r
    break;\r
  case 2:\r
    Info->mVarType = EFI_IFR_TYPE_NUM_SIZE_16;\r
    break;\r
  case 4:\r
    Info->mVarType = EFI_IFR_TYPE_NUM_SIZE_32;\r
    break;\r
  case 8:\r
    Info->mVarType = EFI_IFR_TYPE_NUM_SIZE_64;\r
    break;\r
  default :\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::AddBufferVarStoreFieldInfo (\r
  IN EFI_VARSTORE_INFO  *Info\r
  )\r
{\r
  BufferVarStoreFieldInfoNode *pNew;\r
\r
  if ((pNew = new BufferVarStoreFieldInfoNode(Info)) == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if (mBufferFieldInfoListHead == NULL) {\r
    mBufferFieldInfoListHead = pNew;\r
    mBufferFieldInfoListTail= pNew;\r
  } else {\r
    mBufferFieldInfoListTail->mNext = pNew;\r
    mBufferFieldInfoListTail = pNew;\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::GetBufferVarStoreFieldInfo (\r
  IN OUT EFI_VARSTORE_INFO  *Info\r
  )\r
{\r
  BufferVarStoreFieldInfoNode *pNode;\r
\r
  pNode = mBufferFieldInfoListHead;\r
  while (pNode != NULL) {\r
    if (Info->mVarStoreId == pNode->mVarStoreInfo.mVarStoreId &&\r
      Info->mInfo.mVarOffset == pNode->mVarStoreInfo.mInfo.mVarOffset) {\r
      Info->mVarTotalSize = pNode->mVarStoreInfo.mVarTotalSize;\r
      Info->mVarType      = pNode->mVarStoreInfo.mVarType;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
    pNode = pNode->mNext;\r
  }\r
  return VFR_RETURN_FATAL_ERROR;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDataStorage::GetNameVarStoreInfo (\r
  OUT EFI_VARSTORE_INFO  *Info,\r
  IN  UINT32             Index\r
  )\r
{\r
  if (Info == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  if (mCurrVarStorageNode == NULL) {\r
    return VFR_RETURN_GET_NVVARSTORE_ERROR;\r
  }\r
  \r
  //\r
  // Framework Vfr file Index is from 1, but Uefi Vfr file Index is from 0.\r
  //\r
  if (VfrCompatibleMode) {\r
    if (Index == 0) {\r
      return VFR_RETURN_ERROR_ARRARY_NUM;\r
    }\r
    Index --;\r
  }\r
\r
  Info->mInfo.mVarName = mCurrVarStorageNode->mStorageInfo.mNameSpace.mNameTable[Index];\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
SVfrDefaultStoreNode::SVfrDefaultStoreNode (\r
  IN EFI_IFR_DEFAULTSTORE *ObjBinAddr,\r
  IN CHAR8                *RefName, \r
  IN EFI_STRING_ID        DefaultStoreNameId, \r
  IN UINT16               DefaultId\r
  )\r
{\r
  mObjBinAddr = ObjBinAddr;\r
\r
  if (RefName != NULL) {\r
    mRefName          = new CHAR8[strlen (RefName) + 1];\r
    strcpy (mRefName, RefName);\r
  } else {\r
    mRefName          = NULL;\r
  }\r
\r
  mNext               = NULL;\r
  mDefaultId          = DefaultId;\r
  mDefaultStoreNameId = DefaultStoreNameId;\r
}\r
\r
SVfrDefaultStoreNode::~SVfrDefaultStoreNode (\r
  VOID\r
  )\r
{\r
  if (mRefName != NULL) {\r
    delete mRefName;\r
  }\r
}\r
\r
CVfrDefaultStore::CVfrDefaultStore (\r
  VOID\r
  )\r
{\r
  mDefaultStoreList = NULL;\r
}\r
\r
CVfrDefaultStore::~CVfrDefaultStore (\r
  VOID\r
  )\r
{\r
  SVfrDefaultStoreNode *pTmp = NULL;\r
\r
  while (mDefaultStoreList != NULL) {\r
    pTmp = mDefaultStoreList;\r
    mDefaultStoreList = mDefaultStoreList->mNext;\r
    delete pTmp;\r
  }\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDefaultStore::RegisterDefaultStore (\r
  IN CHAR8                *ObjBinAddr,\r
  IN CHAR8                *RefName,\r
  IN EFI_STRING_ID        DefaultStoreNameId,\r
  IN UINT16               DefaultId\r
  )\r
{\r
  SVfrDefaultStoreNode *pNode = NULL;\r
\r
  if (RefName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pNode = mDefaultStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mRefName, RefName) == 0) {\r
      return VFR_RETURN_REDEFINED;\r
    }\r
  }\r
\r
  if ((pNode = new SVfrDefaultStoreNode ((EFI_IFR_DEFAULTSTORE *)ObjBinAddr, RefName, DefaultStoreNameId, DefaultId)) == NULL) {\r
    return VFR_RETURN_OUT_FOR_RESOURCES;\r
  }\r
\r
  pNode->mNext               = mDefaultStoreList;\r
  mDefaultStoreList          = pNode;\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
/*\r
 * assign new reference name or new default store name id only if \r
 * the original is invalid\r
 */\r
EFI_VFR_RETURN_CODE\r
CVfrDefaultStore::ReRegisterDefaultStoreById (\r
  IN UINT16          DefaultId,\r
  IN CHAR8           *RefName,\r
  IN EFI_STRING_ID   DefaultStoreNameId\r
  )\r
{\r
  SVfrDefaultStoreNode *pNode = NULL;\r
\r
  for (pNode = mDefaultStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mDefaultId == DefaultId) {\r
      break;\r
    }\r
  }\r
\r
  if (pNode == NULL) {\r
    return VFR_RETURN_UNDEFINED;\r
  } else {\r
    if (pNode->mDefaultStoreNameId == EFI_STRING_ID_INVALID) {\r
      pNode->mDefaultStoreNameId  = DefaultStoreNameId;\r
      if (pNode->mObjBinAddr != NULL) {\r
        pNode->mObjBinAddr->DefaultName = DefaultStoreNameId;\r
      }\r
    } else {\r
      return VFR_RETURN_REDEFINED;\r
    }\r
\r
    if (RefName != NULL) {\r
      delete pNode->mRefName;\r
      pNode->mRefName = new CHAR8[strlen (RefName) + 1];\r
      if (pNode->mRefName != NULL) {\r
        strcpy (pNode->mRefName, RefName);\r
      }\r
    }\r
  }\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
BOOLEAN\r
CVfrDefaultStore::DefaultIdRegistered (\r
  IN UINT16          DefaultId\r
  )\r
{\r
  SVfrDefaultStoreNode *pNode = NULL;\r
\r
  for (pNode = mDefaultStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mDefaultId == DefaultId) {\r
      return TRUE;\r
    }\r
  }\r
\r
  return FALSE;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDefaultStore::GetDefaultId (\r
  IN  CHAR8           *RefName,\r
  OUT UINT16          *DefaultId\r
  )\r
{\r
  SVfrDefaultStoreNode *pTmp = NULL;\r
\r
  if (DefaultId == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pTmp = mDefaultStoreList; pTmp != NULL; pTmp = pTmp->mNext) {\r
    if (strcmp (pTmp->mRefName, RefName) == 0) {\r
      *DefaultId = pTmp->mDefaultId;\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrDefaultStore::BufferVarStoreAltConfigAdd (\r
  IN EFI_VARSTORE_ID    DefaultId,\r
  IN EFI_VARSTORE_INFO  &Info,\r
  IN CHAR8              *VarStoreName,\r
  IN EFI_GUID           *VarStoreGuid,\r
  IN UINT8              Type,\r
  IN EFI_IFR_TYPE_VALUE Value\r
  )\r
{\r
  SVfrDefaultStoreNode  *pNode = NULL;\r
  CHAR8                 NewAltCfg[2 * 2 * sizeof (UINT16) + 1] = {0,};\r
  INTN                  Returnvalue = 0;\r
\r
  if (VarStoreName == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pNode = mDefaultStoreList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mDefaultId == DefaultId) {\r
      break;\r
    }\r
  }\r
\r
  if (pNode == NULL) {\r
    return VFR_RETURN_UNDEFINED;\r
  }\r
\r
  gCVfrBufferConfig.Open ();\r
\r
  sprintf (NewAltCfg, "%04x", pNode->mDefaultId);\r
  if ((Returnvalue = gCVfrBufferConfig.Select(VarStoreName, VarStoreGuid)) == 0) {\r
    if ((Returnvalue = gCVfrBufferConfig.Write ('a', VarStoreName, VarStoreGuid, NewAltCfg, Type, Info.mInfo.mVarOffset, Info.mVarTotalSize, Value)) != 0) {\r
      goto WriteError;\r
    }\r
  }\r
  \r
  gCVfrBufferConfig.Close ();\r
\r
  return VFR_RETURN_SUCCESS;\r
\r
WriteError:\r
  gCVfrBufferConfig.Close ();\r
  return (EFI_VFR_RETURN_CODE)Returnvalue;\r
}\r
\r
SVfrRuleNode::SVfrRuleNode (\r
  IN CHAR8        *RuleName,\r
  IN UINT8       RuleId\r
  )\r
{\r
  if (RuleName != NULL) {\r
    mRuleName = new CHAR8[strlen (RuleName) + 1];\r
    strcpy (mRuleName, RuleName);\r
  } else {\r
    mRuleName = NULL;\r
  }\r
\r
  mNext       = NULL;\r
  mRuleId     = RuleId;\r
}\r
\r
SVfrRuleNode::~SVfrRuleNode (\r
  VOID\r
  )\r
{\r
  if (mRuleName != NULL) {\r
    delete mRuleName;\r
  }\r
}\r
\r
CVfrRulesDB::CVfrRulesDB ()\r
{\r
  mRuleList   = NULL;\r
  mFreeRuleId = EFI_VARSTORE_ID_START;\r
}\r
\r
CVfrRulesDB::~CVfrRulesDB ()\r
{\r
  SVfrRuleNode *pNode;\r
\r
  while(mRuleList != NULL) {\r
    pNode = mRuleList;\r
    mRuleList = mRuleList->mNext;\r
    delete pNode;\r
  }\r
}\r
\r
VOID\r
CVfrRulesDB::RegisterRule (\r
  IN CHAR8  *RuleName\r
  )\r
{\r
  SVfrRuleNode *pNew;\r
\r
  if (RuleName == NULL) {\r
    return ;\r
  }\r
\r
  if ((pNew = new SVfrRuleNode (RuleName, mFreeRuleId)) == NULL) {\r
    return ;\r
  }\r
\r
  mFreeRuleId++;\r
\r
  pNew->mNext = mRuleList;\r
  mRuleList   = pNew;\r
}\r
\r
UINT8\r
CVfrRulesDB::GetRuleId (\r
  IN CHAR8  *RuleName\r
  )\r
{\r
  SVfrRuleNode *pNode;\r
\r
  if (RuleName == NULL) {\r
    return EFI_RULE_ID_INVALID;\r
  }\r
\r
  for (pNode = mRuleList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mRuleName, RuleName) == 0) {\r
      return pNode->mRuleId;\r
    }\r
  }\r
\r
  return EFI_RULE_ID_INVALID;\r
}\r
\r
CVfrRulesDB gCVfrRulesDB;\r
\r
EFI_VARSTORE_INFO::EFI_VARSTORE_INFO (\r
  VOID\r
  )\r
{\r
  mVarStoreId      = EFI_VARSTORE_ID_INVALID;\r
  mInfo.mVarName   = EFI_STRING_ID_INVALID;\r
  mInfo.mVarOffset = EFI_VAROFFSET_INVALID;\r
  mVarType         = EFI_IFR_TYPE_OTHER;\r
  mVarTotalSize    = 0;\r
}\r
\r
EFI_VARSTORE_INFO::EFI_VARSTORE_INFO (\r
  IN EFI_VARSTORE_INFO &Info\r
  )\r
{\r
  mVarStoreId      = Info.mVarStoreId;\r
  mInfo.mVarName   = Info.mInfo.mVarName;\r
  mInfo.mVarOffset = Info.mInfo.mVarOffset;\r
  mVarType         = Info.mVarType;\r
  mVarTotalSize    = Info.mVarTotalSize;\r
}\r
\r
BOOLEAN\r
EFI_VARSTORE_INFO::operator == (\r
  IN EFI_VARSTORE_INFO  *Info\r
  )\r
{\r
  if ((mVarStoreId == Info->mVarStoreId) &&\r
  	  (mInfo.mVarName == Info->mInfo.mVarName) &&\r
      (mInfo.mVarOffset == Info->mInfo.mVarOffset) &&\r
      (mVarType == Info->mVarType) &&\r
      (mVarTotalSize == Info->mVarTotalSize)) {\r
    return TRUE;\r
  }\r
\r
  return FALSE;\r
}\r
\r
BufferVarStoreFieldInfoNode::BufferVarStoreFieldInfoNode(\r
  IN EFI_VARSTORE_INFO  *Info\r
  )\r
{\r
  mVarStoreInfo.mVarType               = Info->mVarType;\r
  mVarStoreInfo.mVarTotalSize          = Info->mVarTotalSize;\r
  mVarStoreInfo.mInfo.mVarOffset       = Info->mInfo.mVarOffset;\r
  mVarStoreInfo.mVarStoreId            = Info->mVarStoreId;\r
  mNext = NULL;\r
}\r
\r
BufferVarStoreFieldInfoNode::~BufferVarStoreFieldInfoNode ()\r
{\r
  mVarStoreInfo.mVarType               = EFI_IFR_TYPE_OTHER;\r
  mVarStoreInfo.mVarTotalSize          = 0;\r
  mVarStoreInfo.mInfo.mVarOffset       = EFI_VAROFFSET_INVALID;\r
  mVarStoreInfo.mVarStoreId            = EFI_VARSTORE_ID_INVALID;\r
  mNext = NULL;\r
}\r
\r
static EFI_VARSTORE_INFO gEfiInvalidVarStoreInfo;\r
\r
EFI_QUESTION_ID\r
CVfrQuestionDB::GetFreeQuestionId (\r
  VOID\r
  )\r
{\r
  UINT32  Index, Mask, Offset;\r
\r
  for (Index = 0; Index < EFI_FREE_QUESTION_ID_BITMAP_SIZE; Index++) {\r
    if (mFreeQIdBitMap[Index] != 0xFFFFFFFF) {\r
      break;\r
    }\r
  }\r
\r
  for (Offset = 0, Mask = 0x80000000; Mask != 0; Mask >>= 1, Offset++) {\r
    if ((mFreeQIdBitMap[Index] & Mask) == 0) {\r
      mFreeQIdBitMap[Index] |= Mask;\r
      return (EFI_QUESTION_ID)((Index << EFI_BITS_SHIFT_PER_UINT32) + Offset);\r
    }\r
  }\r
\r
  return EFI_QUESTION_ID_INVALID;\r
}\r
\r
BOOLEAN\r
CVfrQuestionDB::ChekQuestionIdFree (\r
  IN EFI_QUESTION_ID QId\r
  )\r
{\r
  UINT32 Index  = (QId / EFI_BITS_PER_UINT32);\r
  UINT32 Offset = (QId % EFI_BITS_PER_UINT32);\r
\r
  return (mFreeQIdBitMap[Index] & (0x80000000 >> Offset)) == 0;\r
}\r
\r
VOID\r
CVfrQuestionDB::MarkQuestionIdUsed (\r
  IN EFI_QUESTION_ID QId\r
  )\r
{\r
  UINT32 Index  = (QId / EFI_BITS_PER_UINT32);\r
  UINT32 Offset = (QId % EFI_BITS_PER_UINT32);\r
\r
  mFreeQIdBitMap[Index] |= (0x80000000 >> Offset);\r
}\r
\r
VOID\r
CVfrQuestionDB::MarkQuestionIdUnused (\r
  IN EFI_QUESTION_ID QId\r
  )\r
{\r
  UINT32 Index  = (QId / EFI_BITS_PER_UINT32);\r
  UINT32 Offset = (QId % EFI_BITS_PER_UINT32);\r
\r
  mFreeQIdBitMap[Index] &= ~(0x80000000 >> Offset);\r
}\r
\r
SVfrQuestionNode::SVfrQuestionNode (\r
  IN CHAR8  *Name,\r
  IN CHAR8  *VarIdStr,\r
  IN UINT32 BitMask\r
  )\r
{\r
  mName       = NULL;\r
  mVarIdStr   = NULL;\r
  mQuestionId = EFI_QUESTION_ID_INVALID;\r
  mBitMask    = BitMask;\r
  mNext       = NULL;\r
  mQtype      = QUESTION_NORMAL;\r
\r
  if (Name == NULL) {\r
    mName = new CHAR8[strlen ("$DEFAULT") + 1];\r
    strcpy (mName, "$DEFAULT");\r
  } else {\r
    mName = new CHAR8[strlen (Name) + 1];\r
    strcpy (mName, Name);\r
  }\r
\r
  if (VarIdStr != NULL) {\r
    mVarIdStr = new CHAR8[strlen (VarIdStr) + 1];\r
    strcpy (mVarIdStr, VarIdStr);\r
  } else {\r
    mVarIdStr = new CHAR8[strlen ("$") + 1];\r
    strcpy (mVarIdStr, "$");\r
  }\r
}\r
\r
SVfrQuestionNode::~SVfrQuestionNode (\r
  VOID\r
  )\r
{\r
  if (mName != NULL) {\r
    delete mName;\r
  }\r
\r
  if (mVarIdStr != NULL) {\r
    delete mVarIdStr;\r
  }\r
}\r
\r
CVfrQuestionDB::CVfrQuestionDB ()\r
{\r
  UINT32 Index;\r
\r
  for (Index = 0; Index < EFI_FREE_QUESTION_ID_BITMAP_SIZE; Index++) {\r
    mFreeQIdBitMap[Index] = 0;\r
  }\r
\r
  // Question ID 0 is reserved.\r
  mFreeQIdBitMap[0] = 0x80000000;\r
  mQuestionList     = NULL;\r
}\r
\r
CVfrQuestionDB::~CVfrQuestionDB ()\r
{\r
  SVfrQuestionNode     *pNode;\r
\r
  while (mQuestionList != NULL) {\r
    pNode = mQuestionList;\r
    mQuestionList = mQuestionList->mNext;\r
    delete pNode;\r
  }\r
}\r
\r
//\r
// Reset to init state\r
//\r
VOID\r
CVfrQuestionDB::ResetInit(\r
  IN VOID\r
  )\r
{\r
  UINT32               Index;\r
  SVfrQuestionNode     *pNode;\r
\r
  while (mQuestionList != NULL) {\r
    pNode = mQuestionList;\r
    mQuestionList = mQuestionList->mNext;\r
    delete pNode;\r
  }\r
\r
  for (Index = 0; Index < EFI_FREE_QUESTION_ID_BITMAP_SIZE; Index++) {\r
    mFreeQIdBitMap[Index] = 0;\r
  }\r
\r
  // Question ID 0 is reserved.\r
  mFreeQIdBitMap[0] = 0x80000000;\r
  mQuestionList     = NULL;   \r
}\r
\r
VOID\r
CVfrQuestionDB::PrintAllQuestion (\r
  VOID\r
  )\r
{\r
  SVfrQuestionNode *pNode = NULL;\r
\r
  for (pNode = mQuestionList; pNode != NULL; pNode = pNode->mNext) {\r
    printf ("Question VarId is %s and QuesitonId is 0x%x\n", pNode->mVarIdStr, pNode->mQuestionId);\r
  }\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrQuestionDB::RegisterQuestion (\r
  IN     CHAR8             *Name,\r
  IN     CHAR8             *VarIdStr,\r
  IN OUT EFI_QUESTION_ID   &QuestionId\r
  )\r
{\r
  SVfrQuestionNode *pNode = NULL;\r
\r
  if ((Name != NULL) && (FindQuestion(Name) == VFR_RETURN_SUCCESS)) {\r
    return VFR_RETURN_REDEFINED;\r
  }\r
\r
  if ((pNode = new SVfrQuestionNode (Name, VarIdStr)) == NULL) {\r
    return VFR_RETURN_OUT_FOR_RESOURCES;\r
  }\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    QuestionId = GetFreeQuestionId ();\r
  } else {\r
    //\r
    // For Framework Vfr, don't check question ID conflict.\r
    //\r
    if (!VfrCompatibleMode && ChekQuestionIdFree (QuestionId) == FALSE) {\r
      delete pNode;\r
      return VFR_RETURN_QUESTIONID_REDEFINED;\r
    }\r
    MarkQuestionIdUsed (QuestionId);\r
  }\r
  pNode->mQuestionId = QuestionId;\r
\r
  pNode->mNext       = mQuestionList;\r
  mQuestionList      = pNode;\r
\r
  gCFormPkg.DoPendingAssign (VarIdStr, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
VOID\r
CVfrQuestionDB::RegisterOldDateQuestion (\r
  IN     CHAR8            *YearVarId,\r
  IN     CHAR8            *MonthVarId,\r
  IN     CHAR8            *DayVarId,\r
  IN OUT EFI_QUESTION_ID &QuestionId\r
  )\r
{\r
  SVfrQuestionNode *pNode[3] = {NULL, };\r
  UINT32           Index;\r
\r
  if ((YearVarId == NULL) || (MonthVarId == NULL) || (DayVarId == NULL)) {\r
    return;\r
  }\r
\r
  if ((pNode[0] = new SVfrQuestionNode (NULL, YearVarId, DATE_YEAR_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[1] = new SVfrQuestionNode (NULL, MonthVarId, DATE_MONTH_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[2] = new SVfrQuestionNode (NULL, DayVarId, DATE_DAY_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    QuestionId = GetFreeQuestionId ();\r
  } else {\r
    if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
      goto Err;\r
    }\r
    MarkQuestionIdUsed (QuestionId);\r
  }\r
\r
  pNode[0]->mQuestionId = QuestionId;\r
  pNode[1]->mQuestionId = QuestionId;\r
  pNode[2]->mQuestionId = QuestionId;\r
  pNode[0]->mQtype      = QUESTION_DATE;\r
  pNode[1]->mQtype      = QUESTION_DATE;\r
  pNode[2]->mQtype      = QUESTION_DATE;\r
  pNode[0]->mNext       = pNode[1];\r
  pNode[1]->mNext       = pNode[2];\r
  pNode[2]->mNext       = mQuestionList;\r
  mQuestionList         = pNode[0];\r
\r
  gCFormPkg.DoPendingAssign (YearVarId, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (MonthVarId, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (DayVarId, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
\r
  return;\r
\r
Err:\r
  for (Index = 0; Index < 3; Index++) {\r
    if (pNode[Index] != NULL) {\r
      delete pNode[Index];\r
    }\r
  }\r
  QuestionId = EFI_QUESTION_ID_INVALID;\r
}\r
\r
VOID\r
CVfrQuestionDB::RegisterNewDateQuestion (\r
  IN     CHAR8            *Name,\r
  IN     CHAR8            *BaseVarId,\r
  IN OUT EFI_QUESTION_ID &QuestionId\r
  )\r
{\r
  SVfrQuestionNode     *pNode[3] = {NULL, };\r
  UINT32               Len;\r
  CHAR8                *VarIdStr[3] = {NULL, };\r
  CHAR8                 Index;\r
\r
  if (BaseVarId == NULL && Name == NULL) {\r
    if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
      QuestionId = GetFreeQuestionId ();\r
    } else {\r
      if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
        goto Err;\r
      }\r
      MarkQuestionIdUsed (QuestionId);\r
    }\r
    return;\r
  }\r
\r
  if (BaseVarId != NULL) {\r
    Len = strlen (BaseVarId);\r
\r
    VarIdStr[0] = new CHAR8[Len + strlen (".Year") + 1];\r
    if (VarIdStr[0] != NULL) {\r
      strcpy (VarIdStr[0], BaseVarId);\r
      strcat (VarIdStr[0], ".Year");\r
    }\r
    VarIdStr[1] = new CHAR8[Len + strlen (".Month") + 1];\r
    if (VarIdStr[1] != NULL) {\r
      strcpy (VarIdStr[1], BaseVarId);\r
      strcat (VarIdStr[1], ".Month");\r
    }\r
    VarIdStr[2] = new CHAR8[Len + strlen (".Day") + 1];\r
    if (VarIdStr[2] != NULL) {\r
      strcpy (VarIdStr[2], BaseVarId);\r
      strcat (VarIdStr[2], ".Day");\r
    }\r
  } else {\r
    Len = strlen (Name);\r
\r
    VarIdStr[0] = new CHAR8[Len + strlen (".Year") + 1];\r
    if (VarIdStr[0] != NULL) {\r
      strcpy (VarIdStr[0], Name);\r
      strcat (VarIdStr[0], ".Year");\r
    }\r
    VarIdStr[1] = new CHAR8[Len + strlen (".Month") + 1];\r
    if (VarIdStr[1] != NULL) {\r
      strcpy (VarIdStr[1], Name);\r
      strcat (VarIdStr[1], ".Month");\r
    }\r
    VarIdStr[2] = new CHAR8[Len + strlen (".Day") + 1];\r
    if (VarIdStr[2] != NULL) {\r
      strcpy (VarIdStr[2], Name);\r
      strcat (VarIdStr[2], ".Day");\r
    }\r
  }\r
\r
  if ((pNode[0] = new SVfrQuestionNode (Name, VarIdStr[0], DATE_YEAR_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[1] = new SVfrQuestionNode (Name, VarIdStr[1], DATE_MONTH_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[2] = new SVfrQuestionNode (Name, VarIdStr[2], DATE_DAY_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    QuestionId = GetFreeQuestionId ();\r
  } else {\r
    if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
      goto Err;\r
    }\r
    MarkQuestionIdUsed (QuestionId);\r
  }\r
\r
  pNode[0]->mQuestionId = QuestionId;\r
  pNode[1]->mQuestionId = QuestionId;\r
  pNode[2]->mQuestionId = QuestionId;\r
  pNode[0]->mQtype      = QUESTION_DATE;\r
  pNode[1]->mQtype      = QUESTION_DATE;\r
  pNode[2]->mQtype      = QUESTION_DATE;\r
  pNode[0]->mNext       = pNode[1];\r
  pNode[1]->mNext       = pNode[2];\r
  pNode[2]->mNext       = mQuestionList;\r
  mQuestionList         = pNode[0];\r
\r
  for (Index = 0; Index < 3; Index++) {\r
    if (VarIdStr[Index] != NULL) {\r
      delete VarIdStr[Index];\r
    }\r
  }\r
\r
  gCFormPkg.DoPendingAssign (VarIdStr[0], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[1], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[2], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
\r
  return;\r
\r
Err:\r
  for (Index = 0; Index < 3; Index++) {\r
    if (pNode[Index] != NULL) {\r
      delete pNode[Index];\r
    }\r
\r
    if (VarIdStr[Index] != NULL) {\r
      delete VarIdStr[Index];\r
    }\r
  }\r
}\r
\r
VOID\r
CVfrQuestionDB::RegisterOldTimeQuestion (\r
  IN     CHAR8            *HourVarId,\r
  IN     CHAR8            *MinuteVarId,\r
  IN     CHAR8            *SecondVarId,\r
  IN OUT EFI_QUESTION_ID &QuestionId\r
  )\r
{\r
  SVfrQuestionNode *pNode[3] = {NULL, };\r
  UINT32           Index;\r
\r
  if ((HourVarId == NULL) || (MinuteVarId == NULL) || (SecondVarId == NULL)) {\r
    return;\r
  }\r
\r
  if ((pNode[0] = new SVfrQuestionNode (NULL, HourVarId, TIME_HOUR_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[1] = new SVfrQuestionNode (NULL, MinuteVarId, TIME_MINUTE_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[2] = new SVfrQuestionNode (NULL, SecondVarId, TIME_SECOND_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    QuestionId = GetFreeQuestionId ();\r
  } else {\r
    if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
      goto Err;\r
    }\r
    MarkQuestionIdUsed (QuestionId);\r
  }\r
\r
  pNode[0]->mQuestionId = QuestionId;\r
  pNode[1]->mQuestionId = QuestionId;\r
  pNode[2]->mQuestionId = QuestionId;\r
  pNode[0]->mQtype      = QUESTION_TIME;\r
  pNode[1]->mQtype      = QUESTION_TIME;\r
  pNode[2]->mQtype      = QUESTION_TIME;\r
  pNode[0]->mNext       = pNode[1];\r
  pNode[1]->mNext       = pNode[2];\r
  pNode[2]->mNext       = mQuestionList;\r
  mQuestionList         = pNode[0];\r
\r
  gCFormPkg.DoPendingAssign (HourVarId, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (MinuteVarId, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (SecondVarId, (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
\r
  return;\r
\r
Err:\r
  for (Index = 0; Index < 3; Index++) {\r
    if (pNode[Index] != NULL) {\r
      delete pNode[Index];\r
    }\r
  }\r
  QuestionId = EFI_QUESTION_ID_INVALID;\r
}\r
\r
VOID\r
CVfrQuestionDB::RegisterNewTimeQuestion (\r
  IN     CHAR8           *Name,\r
  IN     CHAR8           *BaseVarId,\r
  IN OUT EFI_QUESTION_ID &QuestionId\r
  )\r
{\r
  SVfrQuestionNode     *pNode[3] = {NULL, };\r
  UINT32               Len;\r
  CHAR8                *VarIdStr[3] = {NULL, };\r
  CHAR8                 Index;\r
\r
  if (BaseVarId == NULL && Name == NULL) {\r
    if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
      QuestionId = GetFreeQuestionId ();\r
    } else {\r
      if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
        goto Err;\r
      }\r
      MarkQuestionIdUsed (QuestionId);\r
    }\r
    return;\r
  }\r
\r
  if (BaseVarId != NULL) {\r
    Len = strlen (BaseVarId);\r
\r
    VarIdStr[0] = new CHAR8[Len + strlen (".Hour") + 1];\r
    if (VarIdStr[0] != NULL) {\r
      strcpy (VarIdStr[0], BaseVarId);\r
      strcat (VarIdStr[0], ".Hour");\r
    }\r
    VarIdStr[1] = new CHAR8[Len + strlen (".Minute") + 1];\r
    if (VarIdStr[1] != NULL) {\r
      strcpy (VarIdStr[1], BaseVarId);\r
      strcat (VarIdStr[1], ".Minute");\r
    }\r
    VarIdStr[2] = new CHAR8[Len + strlen (".Second") + 1];\r
    if (VarIdStr[2] != NULL) {\r
      strcpy (VarIdStr[2], BaseVarId);\r
      strcat (VarIdStr[2], ".Second");\r
    }\r
  } else {\r
    Len = strlen (Name);\r
\r
    VarIdStr[0] = new CHAR8[Len + strlen (".Hour") + 1];\r
    if (VarIdStr[0] != NULL) {\r
      strcpy (VarIdStr[0], Name);\r
      strcat (VarIdStr[0], ".Hour");\r
    }\r
    VarIdStr[1] = new CHAR8[Len + strlen (".Minute") + 1];\r
    if (VarIdStr[1] != NULL) {\r
      strcpy (VarIdStr[1], Name);\r
      strcat (VarIdStr[1], ".Minute");\r
    }\r
    VarIdStr[2] = new CHAR8[Len + strlen (".Second") + 1];\r
    if (VarIdStr[2] != NULL) {\r
      strcpy (VarIdStr[2], Name);\r
      strcat (VarIdStr[2], ".Second");\r
    }\r
  }\r
\r
  if ((pNode[0] = new SVfrQuestionNode (Name, VarIdStr[0], TIME_HOUR_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[1] = new SVfrQuestionNode (Name, VarIdStr[1], TIME_MINUTE_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[2] = new SVfrQuestionNode (Name, VarIdStr[2], TIME_SECOND_BITMASK)) == NULL) {\r
    goto Err;\r
  }\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    QuestionId = GetFreeQuestionId ();\r
  } else {\r
    if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
      goto Err;\r
    }\r
    MarkQuestionIdUsed (QuestionId);\r
  }\r
\r
  pNode[0]->mQuestionId = QuestionId;\r
  pNode[1]->mQuestionId = QuestionId;\r
  pNode[2]->mQuestionId = QuestionId;\r
  pNode[0]->mQtype      = QUESTION_TIME;\r
  pNode[1]->mQtype      = QUESTION_TIME;\r
  pNode[2]->mQtype      = QUESTION_TIME;\r
  pNode[0]->mNext       = pNode[1];\r
  pNode[1]->mNext       = pNode[2];\r
  pNode[2]->mNext       = mQuestionList;\r
  mQuestionList         = pNode[0];\r
\r
  for (Index = 0; Index < 3; Index++) {\r
    if (VarIdStr[Index] != NULL) {\r
      delete VarIdStr[Index];\r
    }\r
  }\r
\r
  gCFormPkg.DoPendingAssign (VarIdStr[0], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[1], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[2], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
\r
  return;\r
\r
Err:\r
  for (Index = 0; Index < 3; Index++) {\r
    if (pNode[Index] != NULL) {\r
      delete pNode[Index];\r
    }\r
\r
    if (VarIdStr[Index] != NULL) {\r
      delete VarIdStr[Index];\r
    }\r
  }\r
}\r
\r
VOID \r
CVfrQuestionDB::RegisterRefQuestion (\r
  IN     CHAR8           *Name,\r
  IN     CHAR8           *BaseVarId,\r
  IN OUT EFI_QUESTION_ID &QuestionId\r
  )\r
{\r
  SVfrQuestionNode     *pNode[4] = {NULL, };\r
  UINT32               Len;\r
  CHAR8                *VarIdStr[4] = {NULL, };\r
  CHAR8                 Index;\r
\r
  if (BaseVarId == NULL && Name == NULL) {\r
    return;\r
  }\r
\r
  if (BaseVarId != NULL) {\r
    Len = strlen (BaseVarId);\r
\r
    VarIdStr[0] = new CHAR8[Len + strlen (".QuestionId") + 1];\r
    if (VarIdStr[0] != NULL) {\r
      strcpy (VarIdStr[0], BaseVarId);\r
      strcat (VarIdStr[0], ".QuestionId");\r
    }\r
    VarIdStr[1] = new CHAR8[Len + strlen (".FormId") + 1];\r
    if (VarIdStr[1] != NULL) {\r
      strcpy (VarIdStr[1], BaseVarId);\r
      strcat (VarIdStr[1], ".FormId");\r
    }\r
    VarIdStr[2] = new CHAR8[Len + strlen (".FormSetGuid") + 1];\r
    if (VarIdStr[2] != NULL) {\r
      strcpy (VarIdStr[2], BaseVarId);\r
      strcat (VarIdStr[2], ".FormSetGuid");\r
    }\r
    VarIdStr[3] = new CHAR8[Len + strlen (".DevicePath") + 1];\r
    if (VarIdStr[3] != NULL) {\r
      strcpy (VarIdStr[3], BaseVarId);\r
      strcat (VarIdStr[3], ".DevicePath");\r
    }\r
  } else {\r
    Len = strlen (Name);\r
\r
    VarIdStr[0] = new CHAR8[Len + strlen (".QuestionId") + 1];\r
    if (VarIdStr[0] != NULL) {\r
      strcpy (VarIdStr[0], Name);\r
      strcat (VarIdStr[0], ".QuestionId");\r
    }\r
    VarIdStr[1] = new CHAR8[Len + strlen (".FormId") + 1];\r
    if (VarIdStr[1] != NULL) {\r
      strcpy (VarIdStr[1], Name);\r
      strcat (VarIdStr[1], ".FormId");\r
    }\r
    VarIdStr[2] = new CHAR8[Len + strlen (".FormSetGuid") + 1];\r
    if (VarIdStr[2] != NULL) {\r
      strcpy (VarIdStr[2], Name);\r
      strcat (VarIdStr[2], ".FormSetGuid");\r
    }\r
    VarIdStr[3] = new CHAR8[Len + strlen (".DevicePath") + 1];\r
    if (VarIdStr[3] != NULL) {\r
      strcpy (VarIdStr[3], Name);\r
      strcat (VarIdStr[3], ".DevicePath");\r
    }\r
  }\r
\r
  if ((pNode[0] = new SVfrQuestionNode (Name, VarIdStr[0])) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[1] = new SVfrQuestionNode (Name, VarIdStr[1])) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[2] = new SVfrQuestionNode (Name, VarIdStr[2])) == NULL) {\r
    goto Err;\r
  }\r
  if ((pNode[3] = new SVfrQuestionNode (Name, VarIdStr[3])) == NULL) {\r
    goto Err;\r
  }\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    QuestionId = GetFreeQuestionId ();\r
  } else {\r
    if (ChekQuestionIdFree (QuestionId) == FALSE) {\r
      goto Err;\r
    }\r
    MarkQuestionIdUsed (QuestionId);\r
  }\r
\r
  pNode[0]->mQuestionId = QuestionId;\r
  pNode[1]->mQuestionId = QuestionId;\r
  pNode[2]->mQuestionId = QuestionId;\r
  pNode[3]->mQuestionId = QuestionId;  \r
  pNode[0]->mQtype      = QUESTION_REF;\r
  pNode[1]->mQtype      = QUESTION_REF;\r
  pNode[2]->mQtype      = QUESTION_REF;\r
  pNode[3]->mQtype      = QUESTION_REF;  \r
  pNode[0]->mNext       = pNode[1];\r
  pNode[1]->mNext       = pNode[2];\r
  pNode[2]->mNext       = pNode[3];\r
  pNode[3]->mNext       = mQuestionList;  \r
  mQuestionList         = pNode[0];\r
\r
  gCFormPkg.DoPendingAssign (VarIdStr[0], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[1], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[2], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
  gCFormPkg.DoPendingAssign (VarIdStr[3], (VOID *)&QuestionId, sizeof(EFI_QUESTION_ID));\r
\r
  return;\r
\r
  Err:\r
  for (Index = 0; Index < 4; Index++) {\r
    if (pNode[Index] != NULL) {\r
      delete pNode[Index];\r
    }\r
\r
    if (VarIdStr[Index] != NULL) {\r
      delete VarIdStr[Index];\r
    }\r
  }\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrQuestionDB::UpdateQuestionId (\r
  IN EFI_QUESTION_ID   QId,\r
  IN EFI_QUESTION_ID   NewQId\r
  )\r
{\r
  SVfrQuestionNode *pNode = NULL;\r
  \r
  if (QId == NewQId) {\r
    // don't update\r
    return VFR_RETURN_SUCCESS;\r
  }\r
  \r
  //\r
  // For Framework Vfr, don't check question ID conflict.\r
  //  \r
  if (!VfrCompatibleMode && ChekQuestionIdFree (NewQId) == FALSE) {\r
    return VFR_RETURN_REDEFINED;\r
  }\r
\r
  for (pNode = mQuestionList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mQuestionId == QId) {\r
      break;\r
    }\r
  }\r
\r
  if (pNode == NULL) {\r
    return VFR_RETURN_UNDEFINED;\r
  }\r
\r
  MarkQuestionIdUnused (QId);\r
  pNode->mQuestionId = NewQId;\r
  MarkQuestionIdUsed (NewQId);\r
\r
  gCFormPkg.DoPendingAssign (pNode->mVarIdStr, (VOID *)&NewQId, sizeof(EFI_QUESTION_ID));\r
\r
  return VFR_RETURN_SUCCESS;\r
}\r
\r
VOID\r
CVfrQuestionDB::GetQuestionId (\r
  IN  CHAR8             *Name,\r
  IN  CHAR8             *VarIdStr,\r
  OUT EFI_QUESTION_ID   &QuestionId,\r
  OUT UINT32            &BitMask,\r
  OUT EFI_QUESION_TYPE  *QType\r
  )\r
{\r
  SVfrQuestionNode *pNode;\r
\r
  QuestionId = EFI_QUESTION_ID_INVALID;\r
  BitMask    = 0x00000000;\r
  if (QType != NULL) {\r
    *QType = QUESTION_NORMAL;\r
  }\r
\r
  if ((Name == NULL) && (VarIdStr == NULL)) {\r
    return ;\r
  }\r
\r
  for (pNode = mQuestionList; pNode != NULL; pNode = pNode->mNext) {\r
    if (Name != NULL) {\r
      if (strcmp (pNode->mName, Name) != 0) {\r
        continue;\r
      }\r
    }\r
\r
    if (VarIdStr != NULL) {\r
      if (strcmp (pNode->mVarIdStr, VarIdStr) != 0) {\r
        continue;\r
      }\r
  	}\r
\r
    QuestionId = pNode->mQuestionId;\r
    BitMask    = pNode->mBitMask;\r
    if (QType != NULL) {\r
      *QType     = pNode->mQtype;\r
    }\r
    break;\r
  }\r
\r
  return ;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrQuestionDB::FindQuestion (\r
  IN EFI_QUESTION_ID QuestionId\r
  )\r
{\r
  SVfrQuestionNode *pNode;\r
\r
  if (QuestionId == EFI_QUESTION_ID_INVALID) {\r
    return VFR_RETURN_INVALID_PARAMETER;\r
  }\r
\r
  for (pNode = mQuestionList; pNode != NULL; pNode = pNode->mNext) {\r
    if (pNode->mQuestionId == QuestionId) {\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
EFI_VFR_RETURN_CODE\r
CVfrQuestionDB::FindQuestion (\r
  IN CHAR8 *Name\r
  )\r
{\r
  SVfrQuestionNode *pNode;\r
\r
  if (Name == NULL) {\r
    return VFR_RETURN_FATAL_ERROR;\r
  }\r
\r
  for (pNode = mQuestionList; pNode != NULL; pNode = pNode->mNext) {\r
    if (strcmp (pNode->mName, Name) == 0) {\r
      return VFR_RETURN_SUCCESS;\r
    }\r
  }\r
\r
  return VFR_RETURN_UNDEFINED;\r
}\r
\r
CVfrStringDB::CVfrStringDB ()\r
{\r
  mStringFileName = NULL;\r
}\r
\r
CVfrStringDB::~CVfrStringDB ()\r
{\r
  if (mStringFileName != NULL) {\r
    delete mStringFileName;\r
  }\r
  mStringFileName = NULL;\r
}\r
\r
\r
VOID \r
CVfrStringDB::SetStringFileName(IN CHAR8 *StringFileName)\r
{\r
  UINT32 FileLen = 0;\r
\r
  if (StringFileName == NULL) {\r
    return;\r
  }\r
\r
  FileLen = strlen (StringFileName) + 1;\r
  mStringFileName = new CHAR8[FileLen];\r
  if (mStringFileName == NULL) {\r
    return;\r
  }\r
\r
  strcpy (mStringFileName, StringFileName);\r
  mStringFileName[FileLen - 1] = '\0';\r
}\r
\r
\r
/**\r
  Returns TRUE or FALSE whether SupportedLanguages contains the best matching language \r
  from a set of supported languages.\r
\r
  @param[in]  SupportedLanguages  A pointer to a Null-terminated ASCII string that\r
                                  contains a set of language codes.\r
  @param[in]  Language            A variable that contains pointers to Null-terminated\r
                                  ASCII strings that contain one language codes.\r
\r
  @retval FALSE   The best matching language could not be found in SupportedLanguages.\r
  @retval TRUE    The best matching language could be found in SupportedLanguages.\r
\r
**/\r
BOOLEAN\r
CVfrStringDB::GetBestLanguage (\r
  IN CONST CHAR8  *SupportedLanguages,\r
  IN CHAR8        *Language\r
  )\r
{\r
  UINTN        CompareLength;\r
  UINTN        LanguageLength;\r
  CONST CHAR8  *Supported;\r
\r
  if (SupportedLanguages == NULL || Language == NULL){\r
    return FALSE;\r
  }\r
\r
  //\r
  // Determine the length of the first RFC 4646 language code in Language\r
  //\r
  for (LanguageLength = 0; Language[LanguageLength] != 0 && Language[LanguageLength] != ';'; LanguageLength++);\r
\r
  //\r
  // Trim back the length of Language used until it is empty\r
  //\r
  while (LanguageLength > 0) {\r
    //\r
    // Loop through all language codes in SupportedLanguages\r
    //\r
    for (Supported = SupportedLanguages; *Supported != '\0'; Supported += CompareLength) {\r
      //\r
      // Skip ';' characters in Supported\r
      //\r
      for (; *Supported != '\0' && *Supported == ';'; Supported++);\r
      //\r
      // Determine the length of the next language code in Supported\r
      //\r
      for (CompareLength = 0; Supported[CompareLength] != 0 && Supported[CompareLength] != ';'; CompareLength++);\r
      //\r
      // If Language is longer than the Supported, then skip to the next language\r
      //\r
      if (LanguageLength > CompareLength) {\r
        continue;\r
      }\r
\r
      //\r
      // See if the first LanguageLength characters in Supported match Language\r
      //\r
      if (strncmp (Supported, Language, LanguageLength) == 0) {\r
        return TRUE;\r
      }\r
    }\r
\r
    //\r
    // Trim Language from the right to the next '-' character \r
    //\r
    for (LanguageLength--; LanguageLength > 0 && Language[LanguageLength] != '-'; LanguageLength--);\r
  }\r
\r
  //\r
  // No matches were found \r
  //\r
  return FALSE;\r
}\r
\r
\r
CHAR8 *\r
CVfrStringDB::GetVarStoreNameFormStringId (\r
  IN EFI_STRING_ID StringId\r
  )\r
{\r
  FILE        *pInFile    = NULL;\r
  UINT32      NameOffset;\r
  UINT32      Length;\r
  UINT8       *StringPtr;\r
  CHAR8       *StringName;\r
  CHAR16      *UnicodeString;\r
  CHAR8       *VarStoreName = NULL;\r
  CHAR8       *DestTmp;\r
  UINT8       *Current;\r
  EFI_STATUS  Status;\r
  CHAR8       LineBuf[EFI_IFR_MAX_LENGTH];\r
  UINT8       BlockType;\r
  EFI_HII_STRING_PACKAGE_HDR *PkgHeader;\r
  \r
  if (mStringFileName == '\0' ) {\r
    return NULL;\r
  }\r
\r
  if ((pInFile = fopen (LongFilePath (mStringFileName), "rb")) == NULL) {\r
    return NULL;\r
  }\r
\r
  //\r
  // Get file length.\r
  //\r
  fseek (pInFile, 0, SEEK_END);\r
  Length = ftell (pInFile);\r
  fseek (pInFile, 0, SEEK_SET);\r
\r
  //\r
  // Get file data.\r
  //\r
  StringPtr = new UINT8[Length];\r
  if (StringPtr == NULL) {\r
    fclose (pInFile);\r
    return NULL;\r
  }\r
  fread ((char *)StringPtr, sizeof (UINT8), Length, pInFile);\r
  fclose (pInFile);\r
\r
  PkgHeader = (EFI_HII_STRING_PACKAGE_HDR *) StringPtr;\r
  //\r
  // Check the String package.\r
  //\r
  if (PkgHeader->Header.Type != EFI_HII_PACKAGE_STRINGS) {\r
    delete StringPtr;\r
    return NULL;\r
  }\r
\r
  //\r
  // Search the language, get best language base on RFC 4647 matching algorithm.\r
  //\r
  Current = StringPtr;\r
  while (!GetBestLanguage ("en", PkgHeader->Language)) {\r
    Current += PkgHeader->Header.Length;\r
    PkgHeader = (EFI_HII_STRING_PACKAGE_HDR *) Current;\r
    //\r
    // If can't find string package base on language, just return the first string package.\r
    //\r
    if (Current - StringPtr >= Length) {\r
      Current = StringPtr;\r
      PkgHeader = (EFI_HII_STRING_PACKAGE_HDR *) StringPtr;\r
      break;\r
    }\r
  }\r
\r
  Current += PkgHeader->HdrSize;\r
  //\r
  // Find the string block according the stringId.\r
  //\r
  Status = FindStringBlock(Current, StringId, &NameOffset, &BlockType);\r
  if (Status != EFI_SUCCESS) {\r
    delete StringPtr;\r
    return NULL;\r
  }\r
\r
  //\r
  // Get varstore name according the string type.\r
  //\r
  switch (BlockType) {\r
  case EFI_HII_SIBT_STRING_SCSU:\r
  case EFI_HII_SIBT_STRING_SCSU_FONT:\r
  case EFI_HII_SIBT_STRINGS_SCSU:\r
  case EFI_HII_SIBT_STRINGS_SCSU_FONT:\r
    StringName = (CHAR8*)(Current + NameOffset);\r
    VarStoreName = new CHAR8[strlen(StringName) + 1];\r
    strcpy (VarStoreName, StringName);\r
    break;\r
  case EFI_HII_SIBT_STRING_UCS2:\r
  case EFI_HII_SIBT_STRING_UCS2_FONT:\r
  case EFI_HII_SIBT_STRINGS_UCS2:\r
  case EFI_HII_SIBT_STRINGS_UCS2_FONT:\r
    UnicodeString = (CHAR16*)(Current + NameOffset);\r
    Length = GetUnicodeStringTextSize ((UINT8*)UnicodeString) ;\r
    DestTmp = new CHAR8[Length / 2 + 1];\r
    VarStoreName = DestTmp;\r
    while (*UnicodeString != '\0') {\r
      *(DestTmp++) = (CHAR8) *(UnicodeString++);\r
    }\r
    *DestTmp = '\0';\r
    break;\r
  default:\r
    break;\r
  }\r
\r
  delete StringPtr;\r
\r
  return VarStoreName;\r
}\r
\r
EFI_STATUS\r
CVfrStringDB::FindStringBlock (\r
  IN  UINT8                           *StringData,\r
  IN  EFI_STRING_ID                   StringId,\r
  OUT UINT32                          *StringTextOffset,\r
  OUT UINT8                           *BlockType\r
  )\r
{\r
  UINT8                                *BlockHdr;\r
  EFI_STRING_ID                        CurrentStringId;\r
  UINT32                               BlockSize;\r
  UINT32                               Index;\r
  UINT8                                *StringTextPtr;\r
  UINT32                               Offset;\r
  UINT16                               StringCount;\r
  UINT16                               SkipCount;\r
  UINT8                                Length8;\r
  EFI_HII_SIBT_EXT2_BLOCK              Ext2;\r
  UINT32                               Length32;\r
  UINT32                               StringSize;\r
\r
  CurrentStringId = 1;\r
\r
  //\r
  // Parse the string blocks to get the string text and font.\r
  //\r
  BlockHdr  = StringData;\r
  BlockSize = 0;\r
  Offset    = 0;\r
  while (*BlockHdr != EFI_HII_SIBT_END) {\r
    switch (*BlockHdr) {\r
    case EFI_HII_SIBT_STRING_SCSU:\r
      Offset = sizeof (EFI_HII_STRING_BLOCK);\r
      StringTextPtr = BlockHdr + Offset;\r
      BlockSize += Offset + strlen ((CHAR8 *) StringTextPtr) + 1;\r
      CurrentStringId++;\r
      break;\r
\r
    case EFI_HII_SIBT_STRING_SCSU_FONT:\r
      Offset = sizeof (EFI_HII_SIBT_STRING_SCSU_FONT_BLOCK) - sizeof (UINT8);\r
      StringTextPtr = BlockHdr + Offset;\r
      BlockSize += Offset + strlen ((CHAR8 *) StringTextPtr) + 1;\r
      CurrentStringId++;\r
      break;\r
\r
    case EFI_HII_SIBT_STRINGS_SCSU:\r
      memcpy (&StringCount, BlockHdr + sizeof (EFI_HII_STRING_BLOCK), sizeof (UINT16));\r
      StringTextPtr = BlockHdr + sizeof (EFI_HII_SIBT_STRINGS_SCSU_BLOCK) - sizeof (UINT8);\r
      BlockSize += StringTextPtr - BlockHdr;\r
\r
      for (Index = 0; Index < StringCount; Index++) {\r
        BlockSize += strlen ((CHAR8 *) StringTextPtr) + 1;\r
        if (CurrentStringId == StringId) {\r
          *BlockType        = *BlockHdr;\r
          *StringTextOffset = StringTextPtr - StringData;\r
          return EFI_SUCCESS;\r
        }\r
        StringTextPtr = StringTextPtr + strlen ((CHAR8 *) StringTextPtr) + 1;\r
        CurrentStringId++;\r
      }\r
      break;\r
\r
    case EFI_HII_SIBT_STRINGS_SCSU_FONT:\r
      memcpy (\r
        &StringCount,\r
        BlockHdr + sizeof (EFI_HII_STRING_BLOCK) + sizeof (UINT8),\r
        sizeof (UINT16)\r
        );\r
      StringTextPtr = BlockHdr + sizeof (EFI_HII_SIBT_STRINGS_SCSU_FONT_BLOCK) - sizeof (UINT8);\r
      BlockSize += StringTextPtr - BlockHdr;\r
\r
      for (Index = 0; Index < StringCount; Index++) {\r
        BlockSize += strlen ((CHAR8 *) StringTextPtr) + 1;\r
        if (CurrentStringId == StringId) {\r
          *BlockType        = *BlockHdr;\r
          *StringTextOffset = StringTextPtr - StringData;\r
          return EFI_SUCCESS;\r
        }\r
        StringTextPtr = StringTextPtr + strlen ((CHAR8 *) StringTextPtr) + 1;\r
        CurrentStringId++;\r
      }\r
      break;\r
\r
    case EFI_HII_SIBT_STRING_UCS2:\r
      Offset        = sizeof (EFI_HII_STRING_BLOCK);\r
      StringTextPtr = BlockHdr + Offset;\r
      //\r
      // Use StringSize to store the size of the specified string, including the NULL\r
      // terminator.\r
      //\r
      StringSize = GetUnicodeStringTextSize (StringTextPtr);\r
      BlockSize += Offset + StringSize;\r
      CurrentStringId++;\r
      break;\r
\r
    case EFI_HII_SIBT_STRING_UCS2_FONT:\r
      Offset = sizeof (EFI_HII_SIBT_STRING_UCS2_FONT_BLOCK)  - sizeof (CHAR16);\r
      StringTextPtr = BlockHdr + Offset;\r
      //\r
      // Use StrSize to store the size of the specified string, including the NULL\r
      // terminator.\r
      //\r
      StringSize = GetUnicodeStringTextSize (StringTextPtr);\r
      BlockSize += Offset + StringSize;\r
      CurrentStringId++;\r
      break;\r
\r
    case EFI_HII_SIBT_STRINGS_UCS2:\r
      Offset = sizeof (EFI_HII_SIBT_STRINGS_UCS2_BLOCK) - sizeof (CHAR16);\r
      StringTextPtr = BlockHdr + Offset;\r
      BlockSize += Offset;\r
      memcpy (&StringCount, BlockHdr + sizeof (EFI_HII_STRING_BLOCK), sizeof (UINT16));\r
      for (Index = 0; Index < StringCount; Index++) {\r
        StringSize = GetUnicodeStringTextSize (StringTextPtr);\r
        BlockSize += StringSize;\r
        if (CurrentStringId == StringId) {\r
          *BlockType        = *BlockHdr;\r
          *StringTextOffset = StringTextPtr - StringData;\r
          return EFI_SUCCESS;\r
        }\r
        StringTextPtr = StringTextPtr + StringSize;\r
        CurrentStringId++;\r
      }\r
      break;\r
\r
    case EFI_HII_SIBT_STRINGS_UCS2_FONT:\r
      Offset = sizeof (EFI_HII_SIBT_STRINGS_UCS2_FONT_BLOCK) - sizeof (CHAR16);\r
      StringTextPtr = BlockHdr + Offset;\r
      BlockSize += Offset;\r
      memcpy (\r
        &StringCount,\r
        BlockHdr + sizeof (EFI_HII_STRING_BLOCK) + sizeof (UINT8),\r
        sizeof (UINT16)\r
        );\r
      for (Index = 0; Index < StringCount; Index++) {\r
        StringSize = GetUnicodeStringTextSize (StringTextPtr);\r
        BlockSize += StringSize;\r
        if (CurrentStringId == StringId) {\r
          *BlockType        = *BlockHdr;\r
          *StringTextOffset = StringTextPtr - StringData;\r
          return EFI_SUCCESS;\r
        }\r
        StringTextPtr = StringTextPtr + StringSize;\r
        CurrentStringId++;\r
      }\r
      break;\r
\r
    case EFI_HII_SIBT_DUPLICATE:\r
      if (CurrentStringId == StringId) {\r
        //\r
        // Incoming StringId is an id of a duplicate string block.\r
        // Update the StringId to be the previous string block.\r
        // Go back to the header of string block to search.\r
        //\r
        memcpy (\r
          &StringId,\r
          BlockHdr + sizeof (EFI_HII_STRING_BLOCK),\r
          sizeof (EFI_STRING_ID)\r
          );\r
        CurrentStringId = 1;\r
        BlockSize       = 0;\r
      } else {\r
        BlockSize       += sizeof (EFI_HII_SIBT_DUPLICATE_BLOCK);\r
        CurrentStringId++;\r
      }\r
      break;\r
\r
    case EFI_HII_SIBT_SKIP1:\r
      SkipCount = (UINT16) (*(BlockHdr + sizeof (EFI_HII_STRING_BLOCK)));\r
      CurrentStringId = (UINT16) (CurrentStringId + SkipCount);\r
      BlockSize       +=  sizeof (EFI_HII_SIBT_SKIP1_BLOCK);\r
      break;\r
\r
    case EFI_HII_SIBT_SKIP2:\r
      memcpy (&SkipCount, BlockHdr + sizeof (EFI_HII_STRING_BLOCK), sizeof (UINT16));\r
      CurrentStringId = (UINT16) (CurrentStringId + SkipCount);\r
      BlockSize       +=  sizeof (EFI_HII_SIBT_SKIP2_BLOCK);\r
      break;\r
\r
    case EFI_HII_SIBT_EXT1:\r
      memcpy (\r
        &Length8,\r
        BlockHdr + sizeof (EFI_HII_STRING_BLOCK) + sizeof (UINT8),\r
        sizeof (UINT8)\r
        );\r
      BlockSize += Length8;\r
      break;\r
\r
    case EFI_HII_SIBT_EXT2:\r
      memcpy (&Ext2, BlockHdr, sizeof (EFI_HII_SIBT_EXT2_BLOCK));\r
      BlockSize += Ext2.Length;\r
      break;\r
\r
    case EFI_HII_SIBT_EXT4:\r
      memcpy (\r
        &Length32,\r
        BlockHdr + sizeof (EFI_HII_STRING_BLOCK) + sizeof (UINT8),\r
        sizeof (UINT32)\r
        );\r
\r
      BlockSize += Length32;\r
      break;\r
\r
    default:\r
      break;\r
    }\r
\r
    if (StringId > 0 && StringId != (EFI_STRING_ID)(-1)) {\r
      *StringTextOffset = BlockHdr - StringData + Offset;\r
      *BlockType        = *BlockHdr;\r
\r
      if (StringId == CurrentStringId - 1) {\r
        //\r
        // if only one skip item, return EFI_NOT_FOUND.\r
        //\r
        if(*BlockType == EFI_HII_SIBT_SKIP2 || *BlockType == EFI_HII_SIBT_SKIP1) {\r
          return EFI_NOT_FOUND;\r
        } else {\r
          return EFI_SUCCESS;\r
        }\r
      }\r
\r
      if (StringId < CurrentStringId - 1) {\r
        return EFI_NOT_FOUND;\r
      }\r
    }\r
    BlockHdr  = StringData + BlockSize;\r
  }\r
\r
  return EFI_NOT_FOUND;\r
}\r
\r
UINT32\r
CVfrStringDB::GetUnicodeStringTextSize (\r
  IN  UINT8            *StringSrc\r
  )\r
{\r
  UINT32 StringSize;\r
  CHAR16 *StringPtr;\r
\r
  StringSize = sizeof (CHAR16);\r
  StringPtr  = (UINT16*)StringSrc;\r
  while (*StringPtr++ != L'\0') {\r
    StringSize += sizeof (CHAR16);\r
  }\r
\r
  return StringSize;\r
}\r
\r
BOOLEAN  VfrCompatibleMode = FALSE;\r
\r
CVfrVarDataTypeDB gCVfrVarDataTypeDB;\r
CVfrDefaultStore  gCVfrDefaultStore;\r
CVfrDataStorage  gCVfrDataStorage;\r
\r
\r