]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/HiiDatabaseDxe/ConfigRouting.c
BaseTools:To generate string default type correctly in VfrCompiler
[mirror_edk2.git] / MdeModulePkg / Universal / HiiDatabaseDxe / ConfigRouting.c
index d6705d02473edecb5555b851e0e5cdb96e285c42..56868831a19e9203e72f1d4945d01fd459d9223e 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
 Implementation of interfaces function for EFI_HII_CONFIG_ROUTING_PROTOCOL.\r
 \r
-Copyright (c) 2007 - 2013, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2007 - 2015, 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
@@ -252,8 +252,7 @@ GenerateSubStr (
   Str    = AllocateZeroPool (Length * sizeof (CHAR16));\r
   ASSERT (Str != NULL);\r
 \r
-  StrCpy (Str, String);\r
-  Length = (BufferLen * 2 + 1) * sizeof (CHAR16);\r
+  StrCpyS (Str, Length, String);\r
 \r
   StringHeader = Str + StrLen (String);\r
   TemString    = (CHAR16 *) StringHeader;\r
@@ -297,7 +296,7 @@ GenerateSubStr (
   //\r
   // Convert the uppercase to lowercase since <HexAf> is defined in lowercase format.\r
   //\r
-  StrCat (Str, L"&");  \r
+  StrCatS (Str, Length, L"&");  \r
   HiiToLower (Str);\r
 \r
   *SubStr = Str;\r
@@ -355,6 +354,9 @@ OutputConfigBody (
   }\r
 \r
   Length = TmpPtr - String;\r
+  if (Length == 0) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
   Result = AllocateCopyPool (Length * sizeof (CHAR16), String);\r
   if (Result == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
@@ -389,6 +391,7 @@ AppendToMultiString (
 {\r
   UINTN AppendStringSize;\r
   UINTN MultiStringSize;\r
+  UINTN MaxLen;\r
 \r
   if (MultiString == NULL || *MultiString == NULL || AppendString == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -396,6 +399,7 @@ AppendToMultiString (
 \r
   AppendStringSize = StrSize (AppendString);\r
   MultiStringSize  = StrSize (*MultiString);\r
+  MaxLen = MAX_STRING_LENGTH / sizeof (CHAR16);\r
 \r
   //\r
   // Enlarge the buffer each time when length exceeds MAX_STRING_LENGTH.\r
@@ -407,12 +411,13 @@ AppendToMultiString (
                                   MultiStringSize + AppendStringSize,\r
                                   (VOID *) (*MultiString)\r
                                   );\r
+    MaxLen = (MultiStringSize + AppendStringSize) / sizeof (CHAR16);\r
     ASSERT (*MultiString != NULL);\r
   }\r
   //\r
   // Append the incoming string\r
   //\r
-  StrCat (*MultiString, AppendString);\r
+  StrCatS (*MultiString, MaxLen, AppendString);\r
 \r
   return EFI_SUCCESS;\r
 }\r
@@ -533,6 +538,8 @@ MergeDefaultString (
   EFI_STRING   AltConfigHdr;\r
   UINTN        HeaderLength;\r
   UINTN        SizeAltCfgResp;\r
+  UINTN        MaxLen;\r
+  UINTN        TotalSize;\r
   \r
   if (*AltCfgResp == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -569,13 +576,14 @@ MergeDefaultString (
   // Construct AltConfigHdr string  "&<ConfigHdr>&ALTCFG=XXXX\0"\r
   //                                  |1| StrLen (ConfigHdr) | 8 | 4 | 1 |\r
   //\r
-  AltConfigHdr = AllocateZeroPool ((1 + HeaderLength + 8 + 4 + 1) * sizeof (CHAR16));\r
+  MaxLen = 1 + HeaderLength + 8 + 4 + 1;\r
+  AltConfigHdr = AllocateZeroPool (MaxLen * sizeof (CHAR16));\r
   if (AltConfigHdr == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
-  StrCpy (AltConfigHdr, L"&");\r
-  StrnCat (AltConfigHdr, *AltCfgResp, HeaderLength);\r
-  StrCat (AltConfigHdr, L"&ALTCFG=");\r
+  StrCpyS (AltConfigHdr, MaxLen, L"&");\r
+  StrnCatS (AltConfigHdr, MaxLen, *AltCfgResp, HeaderLength);\r
+  StrCatS (AltConfigHdr, MaxLen, L"&ALTCFG=");\r
   HeaderLength = StrLen (AltConfigHdr);\r
   \r
   StringPtrDefault = StrStr (DefaultAltCfgResp, AltConfigHdr);\r
@@ -583,7 +591,7 @@ MergeDefaultString (
     //\r
     // Get AltCfg Name\r
     //\r
-    StrnCat (AltConfigHdr, StringPtrDefault + HeaderLength, 4);\r
+    StrnCatS (AltConfigHdr, MaxLen, StringPtrDefault + HeaderLength, 4);\r
     StringPtr = StrStr (*AltCfgResp, AltConfigHdr); \r
     \r
     //\r
@@ -592,34 +600,35 @@ MergeDefaultString (
     if (StringPtr == NULL) {\r
       StringPtrEnd   = StrStr (StringPtrDefault + 1, L"&GUID");\r
       SizeAltCfgResp = StrSize (*AltCfgResp);\r
+      TotalSize = SizeAltCfgResp + StrSize (StringPtrDefault);\r
       if (StringPtrEnd == NULL) {\r
         //\r
         // No more default string is found.\r
         //\r
         *AltCfgResp    = (EFI_STRING) ReallocatePool (\r
                                      SizeAltCfgResp,\r
-                                     SizeAltCfgResp + StrSize (StringPtrDefault),\r
+                                     TotalSize,\r
                                      (VOID *) (*AltCfgResp)\r
                                      );\r
         if (*AltCfgResp == NULL) {\r
           FreePool (AltConfigHdr);\r
           return EFI_OUT_OF_RESOURCES;\r
         }\r
-        StrCat (*AltCfgResp, StringPtrDefault);\r
+        StrCatS (*AltCfgResp, TotalSize / sizeof (CHAR16), StringPtrDefault);\r
         break;\r
       } else {\r
         TempChar = *StringPtrEnd;\r
         *StringPtrEnd = L'\0';\r
         *AltCfgResp = (EFI_STRING) ReallocatePool (\r
                                      SizeAltCfgResp,\r
-                                     SizeAltCfgResp + StrSize (StringPtrDefault),\r
+                                     TotalSize,\r
                                      (VOID *) (*AltCfgResp)\r
                                      );\r
         if (*AltCfgResp == NULL) {\r
           FreePool (AltConfigHdr);\r
           return EFI_OUT_OF_RESOURCES;\r
         }\r
-        StrCat (*AltCfgResp, StringPtrDefault);\r
+        StrCatS (*AltCfgResp, TotalSize / sizeof (CHAR16), StringPtrDefault);\r
         *StringPtrEnd = TempChar;\r
       }\r
     }\r
@@ -1109,6 +1118,7 @@ GetVarStoreType (
 {\r
   EFI_STATUS               Status;\r
   UINTN                    IfrOffset;\r
+  UINTN                    PackageOffset;\r
   EFI_IFR_OP_HEADER        *IfrOpHdr;\r
   CHAR16                   *VarStoreName;\r
   EFI_STRING               GuidStr;\r
@@ -1118,6 +1128,7 @@ GetVarStoreType (
   UINT8                    *HiiFormPackage;\r
   UINTN                    PackageSize;\r
   EFI_IFR_VARSTORE_EFI     *IfrEfiVarStore;\r
+  EFI_HII_PACKAGE_HEADER   *PackageHeader;\r
   \r
   HiiFormPackage = NULL;\r
   LengthString     = 0;\r
@@ -1132,10 +1143,26 @@ GetVarStoreType (
     return Status;\r
   }\r
 \r
-  IfrOffset   = sizeof (EFI_HII_PACKAGE_HEADER);\r
+  IfrOffset     = sizeof (EFI_HII_PACKAGE_HEADER);\r
+  PackageOffset = IfrOffset;\r
+  PackageHeader = (EFI_HII_PACKAGE_HEADER *) HiiFormPackage;\r
+\r
   while (IfrOffset < PackageSize) {\r
-    IfrOpHdr  = (EFI_IFR_OP_HEADER *) (HiiFormPackage + IfrOffset);    \r
+    //\r
+    // More than one form packages exist.\r
+    //\r
+    if (PackageOffset >= PackageHeader->Length) {\r
+        //\r
+        // Process the new form package.\r
+        //\r
+        PackageOffset = sizeof (EFI_HII_PACKAGE_HEADER);\r
+        IfrOffset    += PackageOffset;\r
+        PackageHeader = (EFI_HII_PACKAGE_HEADER *) (HiiFormPackage + IfrOffset);\r
+    }\r
+\r
+    IfrOpHdr  = (EFI_IFR_OP_HEADER *) (HiiFormPackage + IfrOffset);\r
     IfrOffset += IfrOpHdr->Length;\r
+    PackageOffset += IfrOpHdr->Length;\r
 \r
     if (IfrOpHdr->OpCode == EFI_IFR_VARSTORE_EFI_OP ) {\r
       IfrEfiVarStore = (EFI_IFR_VARSTORE_EFI *) IfrOpHdr;\r
@@ -1167,8 +1194,8 @@ GetVarStoreType (
         Status = EFI_OUT_OF_RESOURCES;\r
         goto Done;\r
       }\r
-      StrCpy (TempStr, GuidStr);\r
-      StrCat (TempStr, NameStr);\r
+      StrCpyS (TempStr, LengthString, GuidStr);\r
+      StrCatS (TempStr, LengthString, NameStr);\r
       if (ConfigHdr == NULL || StrnCmp (ConfigHdr, TempStr, StrLen (TempStr)) == 0) {\r
         *EfiVarStore = (EFI_IFR_VARSTORE_EFI *) AllocateZeroPool (IfrOpHdr->Length);\r
         if (*EfiVarStore == NULL) {\r
@@ -1190,6 +1217,13 @@ GetVarStoreType (
       FreePool (GuidStr);\r
       FreePool (NameStr);\r
       FreePool (TempStr);\r
+\r
+      //\r
+      // Already found the varstore, break;\r
+      //\r
+      if (*IsEfiVarstore) {\r
+        break;\r
+      }\r
     }\r
   }\r
 Done:\r
@@ -1256,6 +1290,13 @@ IsThisVarstore (
   GuidStr      = NULL;\r
   TempStr      = NULL;\r
 \r
+  //\r
+  // If ConfigHdr has name field and varstore not has name, return FALSE.\r
+  //\r
+  if (Name == NULL && ConfigHdr != NULL && StrStr (ConfigHdr, L"NAME=&") == NULL) {\r
+    return FALSE;\r
+  }\r
+\r
   GenerateSubStr (L"GUID=", sizeof (EFI_GUID), (VOID *)VarstoreGuid, 1, &GuidStr);\r
   if (Name != NULL) {\r
     GenerateSubStr (L"NAME=", StrLen (Name) * sizeof (CHAR16), (VOID *) Name, 2, &NameStr);\r
@@ -1269,8 +1310,8 @@ IsThisVarstore (
     goto Done;\r
   }\r
 \r
-  StrCpy (TempStr, GuidStr);\r
-  StrCat (TempStr, NameStr);\r
+  StrCpyS (TempStr, LengthString, GuidStr);\r
+  StrCatS (TempStr, LengthString, NameStr);\r
 \r
   if (ConfigHdr == NULL || StrnCmp (ConfigHdr, TempStr, StrLen (TempStr)) == 0) {\r
     RetVal = TRUE;\r
@@ -1292,6 +1333,128 @@ Done:
   return RetVal;\r
 }\r
 \r
+/**\r
+  This function parses Form Package to get the efi varstore info according to the request ConfigHdr.\r
+\r
+  @param  DataBaseRecord        The DataBaseRecord instance contains the found Hii handle and package.\r
+  @param  ConfigHdr             Request string ConfigHdr. If it is NULL,\r
+                                the first found varstore will be as ConfigHdr.\r
+  @retval  TRUE                 This hii package is the reqeust one.\r
+  @retval  FALSE                This hii package is not the reqeust one.\r
+**/                                \r
+BOOLEAN\r
+IsThisPackageList (\r
+  IN     HII_DATABASE_RECORD        *DataBaseRecord,\r
+  IN     EFI_STRING                 ConfigHdr\r
+  )\r
+{\r
+  EFI_STATUS               Status;\r
+  UINTN                    IfrOffset;\r
+  UINTN                    PackageOffset;\r
+  EFI_IFR_OP_HEADER        *IfrOpHdr;\r
+  CHAR16                   *VarStoreName;\r
+  UINT8                    *HiiFormPackage;\r
+  UINTN                    PackageSize;\r
+  EFI_IFR_VARSTORE_EFI     *IfrEfiVarStore;\r
+  EFI_HII_PACKAGE_HEADER   *PackageHeader;\r
+  EFI_IFR_VARSTORE         *IfrVarStore;\r
+  EFI_IFR_VARSTORE_NAME_VALUE *IfrNameValueVarStore;\r
+  BOOLEAN                  FindVarstore;\r
+\r
+  HiiFormPackage   = NULL;\r
+  VarStoreName     = NULL;\r
+  Status           = EFI_SUCCESS;\r
+  FindVarstore     = FALSE;\r
+\r
+  Status = GetFormPackageData(DataBaseRecord, &HiiFormPackage, &PackageSize);\r
+  if (EFI_ERROR (Status)) {\r
+    return FALSE;\r
+  }\r
+\r
+  IfrOffset     = sizeof (EFI_HII_PACKAGE_HEADER);\r
+  PackageOffset = IfrOffset;\r
+  PackageHeader = (EFI_HII_PACKAGE_HEADER *) HiiFormPackage;\r
+\r
+  while (IfrOffset < PackageSize) {\r
+    //\r
+    // More than one form packages exist.\r
+    //\r
+    if (PackageOffset >= PackageHeader->Length) {\r
+        //\r
+        // Process the new form package.\r
+        //\r
+        PackageOffset = sizeof (EFI_HII_PACKAGE_HEADER);\r
+        IfrOffset    += PackageOffset;\r
+        PackageHeader = (EFI_HII_PACKAGE_HEADER *) (HiiFormPackage + IfrOffset);\r
+    }\r
+\r
+    IfrOpHdr  = (EFI_IFR_OP_HEADER *) (HiiFormPackage + IfrOffset);\r
+    IfrOffset += IfrOpHdr->Length;\r
+    PackageOffset += IfrOpHdr->Length;\r
+\r
+    switch (IfrOpHdr->OpCode) {\r
+    \r
+    case EFI_IFR_VARSTORE_OP:\r
+      IfrVarStore = (EFI_IFR_VARSTORE *) IfrOpHdr;\r
+\r
+      VarStoreName = AllocateZeroPool (AsciiStrSize ((CHAR8 *)IfrVarStore->Name) * sizeof (CHAR16));\r
+      if (VarStoreName == NULL) {\r
+        goto Done;\r
+      }\r
+      AsciiStrToUnicodeStr ((CHAR8 *)IfrVarStore->Name, VarStoreName);\r
+\r
+      if (IsThisVarstore((VOID *)&IfrVarStore->Guid, VarStoreName, ConfigHdr)) {\r
+        FindVarstore = TRUE;\r
+        goto Done;\r
+      }\r
+      break;\r
+\r
+    case EFI_IFR_VARSTORE_EFI_OP:\r
+      IfrEfiVarStore = (EFI_IFR_VARSTORE_EFI *) IfrOpHdr;\r
+      VarStoreName = AllocateZeroPool (AsciiStrSize ((CHAR8 *)IfrEfiVarStore->Name) * sizeof (CHAR16));\r
+      if (VarStoreName == NULL) {\r
+        goto Done;\r
+      }\r
+      AsciiStrToUnicodeStr ((CHAR8 *)IfrEfiVarStore->Name, VarStoreName);\r
+\r
+      if (IsThisVarstore (&IfrEfiVarStore->Guid, VarStoreName, ConfigHdr)) {\r
+        FindVarstore = TRUE;\r
+        goto Done;\r
+      }\r
+      break;\r
+\r
+    case EFI_IFR_VARSTORE_NAME_VALUE_OP:\r
+      IfrNameValueVarStore = (EFI_IFR_VARSTORE_NAME_VALUE *) IfrOpHdr;\r
+\r
+      if (IsThisVarstore (&IfrNameValueVarStore->Guid, NULL, ConfigHdr)) {\r
+        FindVarstore = TRUE;\r
+        goto Done;\r
+      }\r
+      break;\r
+      \r
+    case EFI_IFR_FORM_OP:\r
+    case EFI_IFR_FORM_MAP_OP:\r
+      //\r
+      // No matched varstore is found and directly return.\r
+      //\r
+      goto Done;\r
+\r
+    default:\r
+      break;\r
+    }\r
+  }\r
+Done:\r
+  if (HiiFormPackage != NULL) {\r
+    FreePool (HiiFormPackage);\r
+  }\r
+\r
+  if (VarStoreName != NULL) {\r
+    FreePool (VarStoreName);\r
+  }\r
+\r
+  return FindVarstore;\r
+}\r
+\r
 /**\r
   Check whether the this op code is required.\r
 \r
@@ -1415,6 +1578,7 @@ ParseIfrData (
 {\r
   EFI_STATUS               Status;\r
   UINTN                    IfrOffset;\r
+  UINTN                    PackageOffset;\r
   EFI_IFR_VARSTORE         *IfrVarStore;\r
   EFI_IFR_VARSTORE_EFI     *IfrEfiVarStore;\r
   EFI_IFR_OP_HEADER        *IfrOpHdr;\r
@@ -1438,25 +1602,50 @@ ParseIfrData (
   LIST_ENTRY               *LinkData;\r
   LIST_ENTRY               *LinkDefault;\r
   EFI_IFR_VARSTORE_NAME_VALUE *IfrNameValueVarStore;\r
+  EFI_HII_PACKAGE_HEADER   *PackageHeader;\r
+  EFI_VARSTORE_ID          VarStoreId;\r
 \r
   Status           = EFI_SUCCESS;\r
   BlockData        = NULL;\r
   DefaultDataPtr   = NULL;\r
   FirstOneOfOption = FALSE;\r
+  VarStoreId       = 0;\r
   ZeroMem (&DefaultData, sizeof (IFR_DEFAULT_DATA));\r
 \r
   //\r
   // Go through the form package to parse OpCode one by one.\r
   //\r
-  IfrOffset   = sizeof (EFI_HII_PACKAGE_HEADER);\r
+  PackageOffset = sizeof (EFI_HII_PACKAGE_HEADER);\r
+  PackageHeader = (EFI_HII_PACKAGE_HEADER *) Package;\r
+  IfrOffset     = PackageOffset;\r
   while (IfrOffset < PackageLength) {\r
+\r
+    //\r
+    // More than one form package found.\r
+    //\r
+    if (PackageOffset >= PackageHeader->Length) {\r
+        //\r
+        // Already found varstore for this request, break;\r
+        //\r
+        if (VarStoreId != 0) {\r
+          VarStoreId = 0;\r
+        }\r
+\r
+        //\r
+        // Get next package header info.\r
+        //\r
+        IfrOffset    += sizeof (EFI_HII_PACKAGE_HEADER);\r
+        PackageOffset = sizeof (EFI_HII_PACKAGE_HEADER);\r
+        PackageHeader = (EFI_HII_PACKAGE_HEADER *) (Package + IfrOffset);\r
+    }\r
+\r
     IfrOpHdr  = (EFI_IFR_OP_HEADER *) (Package + IfrOffset);\r
     switch (IfrOpHdr->OpCode) {\r
     case EFI_IFR_VARSTORE_OP:\r
       //\r
       // VarStore is found. Don't need to search any more.\r
       //\r
-      if (VarStorageData->VarStoreId != 0) {\r
+      if (VarStoreId != 0) {\r
         break;\r
       }\r
 \r
@@ -1474,10 +1663,10 @@ ParseIfrData (
         // Find the matched VarStore\r
         //\r
         CopyGuid (&VarStorageData->Guid, (EFI_GUID *) (VOID *) &IfrVarStore->Guid);\r
-        VarStorageData->VarStoreId = IfrVarStore->VarStoreId;\r
         VarStorageData->Size       = IfrVarStore->Size;\r
         VarStorageData->Name       = VarStoreName;\r
         VarStorageData->Type       = EFI_HII_VARSTORE_BUFFER;\r
+        VarStoreId                 = IfrVarStore->VarStoreId;\r
       }\r
       break;\r
 \r
@@ -1485,7 +1674,7 @@ ParseIfrData (
       //\r
       // VarStore is found. Don't need to search any more.\r
       //\r
-      if (VarStorageData->VarStoreId != 0) {\r
+      if (VarStoreId != 0) {\r
         break;\r
       }\r
 \r
@@ -1512,10 +1701,10 @@ ParseIfrData (
         // Find the matched VarStore\r
         //\r
         CopyGuid (&VarStorageData->Guid, (EFI_GUID *) (VOID *) &IfrEfiVarStore->Guid);\r
-        VarStorageData->VarStoreId = IfrEfiVarStore->VarStoreId;\r
         VarStorageData->Size       = IfrEfiVarStore->Size;\r
         VarStorageData->Name       = VarStoreName;\r
         VarStorageData->Type       = EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER;\r
+        VarStoreId                 = IfrEfiVarStore->VarStoreId;\r
       }\r
       break;\r
 \r
@@ -1523,7 +1712,7 @@ ParseIfrData (
       //\r
       // VarStore is found. Don't need to search any more.\r
       //\r
-      if (VarStorageData->VarStoreId != 0) {\r
+      if (VarStoreId != 0) {\r
         break;\r
       }\r
 \r
@@ -1534,8 +1723,8 @@ ParseIfrData (
         // Find the matched VarStore\r
         //\r
         CopyGuid (&VarStorageData->Guid, (EFI_GUID *) (VOID *) &IfrNameValueVarStore->Guid);\r
-        VarStorageData->VarStoreId = IfrNameValueVarStore->VarStoreId;\r
         VarStorageData->Type       = EFI_HII_VARSTORE_NAME_VALUE;\r
+        VarStoreId                 = IfrNameValueVarStore->VarStoreId;\r
       }\r
       break;\r
 \r
@@ -1558,7 +1747,7 @@ ParseIfrData (
       //\r
       // No matched varstore is found and directly return.\r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if ( VarStoreId == 0) {\r
         Status = EFI_SUCCESS;\r
         goto Done;\r
       }\r
@@ -1568,7 +1757,7 @@ ParseIfrData (
       //\r
       // Ref question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if ( VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1576,7 +1765,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrRef = (EFI_IFR_REF4 *) IfrOpHdr;\r
-      if (IfrRef->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrRef->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
       VarWidth  = (UINT16) (sizeof (EFI_HII_REF));\r
@@ -1596,7 +1785,7 @@ ParseIfrData (
       //\r
       // Numeric and OneOf question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1604,7 +1793,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrOneOf = (EFI_IFR_ONE_OF *) IfrOpHdr;\r
-      if (IfrOneOf->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrOneOf->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
       VarWidth  = (UINT16) (1 << (IfrOneOf->Flags & EFI_IFR_NUMERIC_SIZE));\r
@@ -1673,7 +1862,7 @@ ParseIfrData (
       //\r
       // OrderedList question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1681,7 +1870,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrOrderedList = (EFI_IFR_ORDERED_LIST *) IfrOpHdr;\r
-      if (IfrOrderedList->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrOrderedList->Question.VarStoreId != VarStoreId) {\r
         BlockData = NULL;\r
         break;\r
       }\r
@@ -1705,7 +1894,7 @@ ParseIfrData (
       //\r
       // CheckBox question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1713,7 +1902,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrCheckBox = (EFI_IFR_CHECKBOX *) IfrOpHdr;\r
-      if (IfrCheckBox->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrCheckBox->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
       VarWidth  = (UINT16) sizeof (BOOLEAN);\r
@@ -1792,7 +1981,7 @@ ParseIfrData (
       //\r
       // Date question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1800,7 +1989,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrDate = (EFI_IFR_DATE *) IfrOpHdr;\r
-      if (IfrDate->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrDate->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
 \r
@@ -1821,7 +2010,7 @@ ParseIfrData (
       //\r
       // Time question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1829,7 +2018,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrTime = (EFI_IFR_TIME *) IfrOpHdr;\r
-      if (IfrTime->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrTime->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
 \r
@@ -1850,7 +2039,7 @@ ParseIfrData (
       //\r
       // String question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1858,7 +2047,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrString = (EFI_IFR_STRING *) IfrOpHdr;\r
-      if (IfrString->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrString->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
 \r
@@ -1884,7 +2073,7 @@ ParseIfrData (
       //\r
       // Password question is not in IFR Form. This IFR form is not valid. \r
       //\r
-      if (VarStorageData->VarStoreId == 0) {\r
+      if (VarStoreId == 0) {\r
         Status = EFI_INVALID_PARAMETER;\r
         goto Done;\r
       }\r
@@ -1892,7 +2081,7 @@ ParseIfrData (
       // Check whether this question is for the requested varstore.\r
       //\r
       IfrPassword = (EFI_IFR_PASSWORD *) IfrOpHdr;\r
-      if (IfrPassword->Question.VarStoreId != VarStorageData->VarStoreId) {\r
+      if (IfrPassword->Question.VarStoreId != VarStoreId) {\r
         break;\r
       }\r
 \r
@@ -2077,19 +2266,32 @@ ParseIfrData (
       //\r
       // End Opcode is for Var question.\r
       //\r
-      if (BlockData != NULL && BlockData->Scope > 0) {\r
-        BlockData->Scope--;\r
+      if (BlockData != NULL) {\r
+        if (BlockData->Scope > 0) {\r
+          BlockData->Scope--;\r
+        }\r
+        if (BlockData->Scope == 0) {\r
+          BlockData = NULL;\r
+        }\r
       }\r
+\r
       break;\r
 \r
     default:\r
-      if (BlockData != NULL && BlockData->Scope > 0) {\r
-        BlockData->Scope = (UINT8) (BlockData->Scope + IfrOpHdr->Scope);\r
+      if (BlockData != NULL) {\r
+        if (BlockData->Scope > 0) {\r
+          BlockData->Scope = (UINT8) (BlockData->Scope + IfrOpHdr->Scope);\r
+        }\r
+\r
+        if (BlockData->Scope == 0) {\r
+          BlockData = NULL;\r
+        }\r
       }\r
       break;\r
     }\r
 \r
-    IfrOffset += IfrOpHdr->Length;\r
+    IfrOffset     += IfrOpHdr->Length;\r
+    PackageOffset += IfrOpHdr->Length;\r
   }\r
 \r
 Done:\r
@@ -2133,6 +2335,8 @@ GetBlockElement (
   IFR_BLOCK_DATA       *NextBlockData;\r
   UINTN                Length;\r
 \r
+  TmpBuffer = NULL;\r
+\r
   //\r
   // Init RequestBlockArray\r
   //\r
@@ -2468,7 +2672,7 @@ GenerateConfigRequest (
   //\r
   // Start with <ConfigHdr>\r
   //\r
-  StrCpy (StringPtr, ConfigHdr);\r
+  StrCpyS (StringPtr, Length, ConfigHdr);\r
   StringPtr += StrLen (StringPtr);\r
 \r
   //\r
@@ -2567,12 +2771,12 @@ GenerateHdr (
     Status = EFI_OUT_OF_RESOURCES;\r
     goto Done;\r
   }\r
-  StrCpy (*ConfigHdr, GuidStr);\r
-  StrCat (*ConfigHdr, NameStr);\r
+  StrCpyS (*ConfigHdr, Length, GuidStr);\r
+  StrCatS (*ConfigHdr, Length, NameStr);\r
   if (VarStorageData->Name == NULL) {\r
-    StrCat (*ConfigHdr, L"&");\r
+    StrCatS (*ConfigHdr, Length, L"&");\r
   }\r
-  StrCat (*ConfigHdr, PathStr);\r
+  StrCatS (*ConfigHdr, Length, PathStr);\r
 \r
   //\r
   // Remove the last character L'&'\r
@@ -2736,7 +2940,7 @@ GenerateAltConfigResp (
   //\r
   // Start with <ConfigHdr>\r
   //\r
-  StrCpy (StringPtr, ConfigHdr);\r
+  StrCpyS (StringPtr, Length, ConfigHdr);\r
   StringPtr += StrLen (StringPtr);\r
 \r
   for (Link = DefaultIdArray->Entry.ForwardLink; Link != &DefaultIdArray->Entry; Link = Link->ForwardLink) {\r
@@ -2887,7 +3091,7 @@ GetFullStringFromHiiFormPackages (
 \r
   Status = GetFormPackageData (DataBaseRecord, &HiiFormPackage, &PackageSize);\r
   if (EFI_ERROR (Status)) {\r
-    return Status;\r
+    goto Done;\r
   }\r
 \r
   //\r
@@ -2990,7 +3194,7 @@ GetFullStringFromHiiFormPackages (
   //\r
   // No requested varstore in IFR data and directly return\r
   //\r
-  if (VarStorageData->VarStoreId == 0) {\r
+  if (VarStorageData->Type == 0 && VarStorageData->Name == NULL) {\r
     Status = EFI_SUCCESS;\r
     goto Done;\r
   }\r
@@ -3148,10 +3352,11 @@ GetConfigRespFromEfiVarStore (
   UINT8      *VarStore;\r
   UINTN      BufferSize;\r
 \r
-  Status       = EFI_SUCCESS;\r
-  BufferSize   = 0;\r
-  VarStore     = NULL;\r
-  VarStoreName = NULL;\r
+  Status          = EFI_SUCCESS;\r
+  BufferSize      = 0;\r
+  VarStore        = NULL;\r
+  VarStoreName    = NULL;\r
+  *AccessProgress = Request;\r
   \r
   VarStoreName = AllocateZeroPool (AsciiStrSize ((CHAR8 *)EfiVarStoreInfo->Name) * sizeof (CHAR16));\r
   if (VarStoreName == NULL) {\r
@@ -3448,8 +3653,10 @@ ConfigRequestValidate (
                                  Progress parameter is set to NULL.\r
   @retval EFI_INVALID_PARAMETER  Illegal syntax. Progress set to most recent &\r
                                  before the error or the beginning of the string.\r
-  @retval EFI_INVALID_PARAMETER  Unknown name. Progress points to the & before the\r
-                                 name in question.\r
+  @retval EFI_INVALID_PARAMETER  The ExtractConfig function of the underlying HII\r
+                                 Configuration Access Protocol returned \r
+                                 EFI_INVALID_PARAMETER. Progress set to most recent\r
+                                 & before the error or the beginning of the string.\r
 \r
 **/\r
 EFI_STATUS\r
@@ -3483,6 +3690,7 @@ HiiConfigRoutingExtractConfig (
   BOOLEAN                             IsEfiVarStore;\r
   EFI_IFR_VARSTORE_EFI                *EfiVarStoreInfo;\r
   EFI_STRING                          ErrorPtr;\r
+  UINTN                               DevicePathSize;\r
 \r
   if (This == NULL || Progress == NULL || Results == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -3568,11 +3776,8 @@ HiiConfigRoutingExtractConfig (
       Database = CR (Link, HII_DATABASE_RECORD, DatabaseEntry, HII_DATABASE_RECORD_SIGNATURE);\r
       if ((DevicePathPkg = Database->PackageList->DevicePathPkg) != NULL) {\r
         CurrentDevicePath = DevicePathPkg + sizeof (EFI_HII_PACKAGE_HEADER);\r
-        if (CompareMem (\r
-              DevicePath,\r
-              CurrentDevicePath,\r
-              GetDevicePathSize ((EFI_DEVICE_PATH_PROTOCOL *) CurrentDevicePath)\r
-              ) == 0) {\r
+        DevicePathSize    = GetDevicePathSize ((EFI_DEVICE_PATH_PROTOCOL *) CurrentDevicePath);\r
+        if ((CompareMem (DevicePath,CurrentDevicePath,DevicePathSize) == 0) && IsThisPackageList(Database, ConfigRequest)) {\r
           DriverHandle = Database->DriverHandle;\r
           HiiHandle    = Database->Handle;\r
           break;\r
@@ -4006,6 +4211,7 @@ HiiConfigRoutingRouteConfig (
   EFI_STRING                          AccessProgress;\r
   EFI_IFR_VARSTORE_EFI                *EfiVarStoreInfo;\r
   BOOLEAN                             IsEfiVarstore;\r
+  UINTN                               DevicePathSize;\r
 \r
   if (This == NULL || Progress == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -4077,11 +4283,8 @@ HiiConfigRoutingRouteConfig (
 \r
       if ((DevicePathPkg = Database->PackageList->DevicePathPkg) != NULL) {\r
         CurrentDevicePath = DevicePathPkg + sizeof (EFI_HII_PACKAGE_HEADER);\r
-        if (CompareMem (\r
-              DevicePath,\r
-              CurrentDevicePath,\r
-              GetDevicePathSize ((EFI_DEVICE_PATH_PROTOCOL *) CurrentDevicePath)\r
-              ) == 0) {\r
+        DevicePathSize    = GetDevicePathSize ((EFI_DEVICE_PATH_PROTOCOL *) CurrentDevicePath);\r
+        if ((CompareMem (DevicePath,CurrentDevicePath,DevicePathSize) == 0) && IsThisPackageList(Database, ConfigResp)) {\r
           DriverHandle = Database->DriverHandle;\r
           break;\r
         }\r
@@ -4144,6 +4347,7 @@ HiiConfigRoutingRouteConfig (
                                );\r
     }\r
     if (EFI_ERROR (Status)) {\r
+      ASSERT (AccessProgress != NULL);\r
       //\r
       // AccessProgress indicates the parsing progress on <ConfigResp>.\r
       // Map it to the progress on <MultiConfigResp> then return it.\r
@@ -4236,6 +4440,8 @@ HiiBlockToConfig (
   CHAR16                              *TemString;\r
   CHAR16                              TemChar;\r
 \r
+  TmpBuffer = NULL;\r
+\r
   if (This == NULL || Progress == NULL || Config == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
@@ -4285,12 +4491,11 @@ HiiBlockToConfig (
   }\r
   if (*StringPtr == 0) {\r
     *Progress = StringPtr;\r
-    Status = EFI_SUCCESS;\r
 \r
     AppendToMultiString(Config, ConfigRequest);\r
     HiiToLower (*Config);\r
 \r
-    goto Exit;\r
+    return EFI_SUCCESS;\r
   }\r
   //\r
   // Skip '&'\r
@@ -4322,7 +4527,7 @@ HiiBlockToConfig (
     //\r
     Status = GetValueOfNumber (StringPtr, &TmpBuffer, &Length);\r
     if (EFI_ERROR (Status)) {\r
-      *Progress = ConfigRequest;\r
+      *Progress = TmpPtr - 1;\r
       goto Exit;\r
     }\r
     Offset = 0;\r
@@ -4335,7 +4540,7 @@ HiiBlockToConfig (
 \r
     StringPtr += Length;\r
     if (StrnCmp (StringPtr, L"&WIDTH=", StrLen (L"&WIDTH=")) != 0) {\r
-      *Progress = StringPtr - Length - StrLen (L"OFFSET=") - 1;\r
+      *Progress = TmpPtr - 1;\r
       Status = EFI_INVALID_PARAMETER;\r
       goto Exit;\r
     }\r
@@ -4346,7 +4551,7 @@ HiiBlockToConfig (
     //\r
     Status = GetValueOfNumber (StringPtr, &TmpBuffer, &Length);\r
     if (EFI_ERROR (Status)) {\r
-      *Progress = ConfigRequest;\r
+      *Progress =  TmpPtr - 1;\r
       goto Exit;\r
     }\r
     Width = 0;\r
@@ -4359,7 +4564,7 @@ HiiBlockToConfig (
 \r
     StringPtr += Length;\r
     if (*StringPtr != 0 && *StringPtr != L'&') {\r
-      *Progress = StringPtr - Length - StrLen (L"&WIDTH=");\r
+      *Progress =  TmpPtr - 1;\r
       Status = EFI_INVALID_PARAMETER;\r
       goto Exit;\r
     }\r
@@ -4413,8 +4618,8 @@ HiiBlockToConfig (
       *(ConfigElement + (StringPtr - TmpPtr)) = L'&';\r
     }\r
     *(ConfigElement + (StringPtr - TmpPtr) + 1) = 0;\r
-    StrCat (ConfigElement, L"VALUE=");\r
-    StrCat (ConfigElement, ValueStr);\r
+    StrCatS (ConfigElement, Length, L"VALUE=");\r
+    StrCatS (ConfigElement, Length, ValueStr);\r
 \r
     AppendToMultiString (Config, ConfigElement);\r
 \r
@@ -4523,6 +4728,7 @@ HiiConfigToBlock (
 {\r
   HII_DATABASE_PRIVATE_DATA           *Private;\r
   EFI_STRING                          StringPtr;\r
+  EFI_STRING                          TmpPtr;\r
   UINTN                               Length;\r
   EFI_STATUS                          Status;\r
   UINT8                               *TmpBuffer;\r
@@ -4532,6 +4738,8 @@ HiiConfigToBlock (
   UINTN                               BufferSize;\r
   UINTN                               MaxBlockSize;\r
 \r
+  TmpBuffer = NULL;\r
+\r
   if (This == NULL || BlockSize == NULL || Progress == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
@@ -4581,13 +4789,14 @@ HiiConfigToBlock (
   // <BlockConfig> ::= 'OFFSET='<Number>&'WIDTH='<Number>&'VALUE='<Number>\r
   //\r
   while (*StringPtr != 0 && StrnCmp (StringPtr, L"&OFFSET=", StrLen (L"&OFFSET=")) == 0) {\r
+    TmpPtr     = StringPtr;\r
     StringPtr += StrLen (L"&OFFSET=");\r
     //\r
     // Get Offset\r
     //\r
     Status = GetValueOfNumber (StringPtr, &TmpBuffer, &Length);\r
     if (EFI_ERROR (Status)) {\r
-      *Progress = ConfigResp;\r
+      *Progress = TmpPtr;\r
       goto Exit;\r
     }\r
     Offset = 0;\r
@@ -4600,7 +4809,7 @@ HiiConfigToBlock (
 \r
     StringPtr += Length;\r
     if (StrnCmp (StringPtr, L"&WIDTH=", StrLen (L"&WIDTH=")) != 0) {\r
-      *Progress = StringPtr - Length - StrLen (L"&OFFSET=");\r
+      *Progress = TmpPtr;\r
       Status = EFI_INVALID_PARAMETER;\r
       goto Exit;\r
     }\r
@@ -4611,7 +4820,7 @@ HiiConfigToBlock (
     //\r
     Status = GetValueOfNumber (StringPtr, &TmpBuffer, &Length);\r
     if (EFI_ERROR (Status)) {\r
-      *Progress = ConfigResp;\r
+      *Progress = TmpPtr;\r
       goto Exit;\r
     }\r
     Width = 0;\r
@@ -4624,7 +4833,7 @@ HiiConfigToBlock (
 \r
     StringPtr += Length;\r
     if (StrnCmp (StringPtr, L"&VALUE=", StrLen (L"&VALUE=")) != 0) {\r
-      *Progress = StringPtr - Length - StrLen (L"&WIDTH=");\r
+      *Progress = TmpPtr;\r
       Status = EFI_INVALID_PARAMETER;\r
       goto Exit;\r
     }\r
@@ -4635,13 +4844,13 @@ HiiConfigToBlock (
     //\r
     Status = GetValueOfNumber (StringPtr, &Value, &Length);\r
     if (EFI_ERROR (Status)) {\r
-      *Progress = ConfigResp;\r
+      *Progress = TmpPtr;\r
       goto Exit;\r
     }\r
 \r
     StringPtr += Length;\r
     if (*StringPtr != 0 && *StringPtr != L'&') {\r
-      *Progress = StringPtr - Length - StrLen (L"&VALUE=");\r
+      *Progress = TmpPtr;\r
       Status = EFI_INVALID_PARAMETER;\r
       goto Exit;\r
     }\r
@@ -4927,8 +5136,8 @@ Exit:
     if (*AltCfgResp == NULL) {\r
       Status = EFI_OUT_OF_RESOURCES;\r
     } else {\r
-      StrnCpy (*AltCfgResp, HdrStart, HdrEnd - HdrStart);\r
-      StrCat (*AltCfgResp, Result);\r
+      StrnCpyS (*AltCfgResp, Length, HdrStart, HdrEnd - HdrStart);\r
+      StrCatS (*AltCfgResp, Length, Result);\r
       Status = EFI_SUCCESS;\r
     }\r
   }\r