]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/HiiDatabaseDxe/ConfigRouting.c
MdeModulePkg: Convert HiiDatabaseDxe ConfigRouting ASSERT to return an error
[mirror_edk2.git] / MdeModulePkg / Universal / HiiDatabaseDxe / ConfigRouting.c
index a83a2ff77779dfebfc39f24776a838ef2019e7a4..55ac08cedb29859a7b55858f8bf18e7b2f0c388e 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
@@ -1125,16 +1136,33 @@ GetVarStoreType (
   GuidStr          = NULL;\r
   NameStr          = NULL;\r
   TempStr          = NULL;\r
+  *IsEfiVarstore   = FALSE;\r
 \r
   Status = GetFormPackageData(DataBaseRecord, &HiiFormPackage, &PackageSize);\r
   if (EFI_ERROR (Status)) {\r
     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
@@ -1166,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
@@ -1189,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
@@ -1255,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
@@ -1268,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
@@ -1291,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
@@ -1414,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
@@ -1433,37 +1598,56 @@ ParseIfrData (
   CHAR16                   *VarStoreName;\r
   UINT16                   VarWidth;\r
   UINT16                   VarDefaultId;\r
-  EFI_STRING               GuidStr;\r
-  EFI_STRING               NameStr;\r
-  EFI_STRING               TempStr;\r
-  UINTN                    LengthString;\r
   BOOLEAN                  FirstOneOfOption;\r
+  BOOLEAN                  FirstOrderedList;\r
   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
-  LengthString     = 0;\r
   Status           = EFI_SUCCESS;\r
-  GuidStr          = NULL;\r
-  NameStr          = NULL;\r
-  TempStr          = NULL;\r
   BlockData        = NULL;\r
   DefaultDataPtr   = NULL;\r
   FirstOneOfOption = FALSE;\r
+  VarStoreId       = 0;\r
+  FirstOrderedList = FALSE;\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
@@ -1481,10 +1665,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
@@ -1492,7 +1676,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
@@ -1519,10 +1703,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
@@ -1530,7 +1714,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
@@ -1541,8 +1725,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
@@ -1565,7 +1749,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
@@ -1575,7 +1759,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
@@ -1583,7 +1767,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
@@ -1603,7 +1787,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
@@ -1611,7 +1795,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
@@ -1674,13 +1858,13 @@ ParseIfrData (
       //\r
       // offset by question header\r
       // width by EFI_IFR_ORDERED_LIST MaxContainers * OneofOption Type\r
-      // no default value and default id, how to define its default value?\r
       //\r
 \r
+      FirstOrderedList = TRUE;\r
       //\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
@@ -1688,7 +1872,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
@@ -1712,7 +1896,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
@@ -1720,7 +1904,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
@@ -1799,7 +1983,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
@@ -1807,7 +1991,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
@@ -1828,7 +2012,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
@@ -1836,7 +2020,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
@@ -1857,7 +2041,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
@@ -1865,7 +2049,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
@@ -1874,11 +2058,6 @@ ParseIfrData (
       if (EFI_ERROR (Status)) {\r
         goto Done;\r
       }\r
-\r
-      //\r
-      // No default value for string.\r
-      //\r
-      BlockData = NULL;\r
       break;\r
 \r
     case EFI_IFR_PASSWORD_OP:\r
@@ -1891,7 +2070,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
@@ -1899,7 +2078,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
@@ -1925,6 +2104,10 @@ ParseIfrData (
 \r
       IfrOneOfOption = (EFI_IFR_ONE_OF_OPTION *) IfrOpHdr;\r
       if (BlockData->OpCode == EFI_IFR_ORDERED_LIST_OP) {\r
+\r
+        if (!FirstOrderedList){\r
+          break;\r
+        }\r
         //\r
         // Get ordered list option data type.\r
         //\r
@@ -1981,10 +2164,9 @@ ParseIfrData (
         // Add Block Data into VarStorageData BlockEntry\r
         //\r
         InsertBlockData (&VarStorageData->BlockEntry, &BlockData);\r
-        //\r
-        // No default data for OrderedList.\r
-        //\r
-        BlockData = NULL;\r
+\r
+        FirstOrderedList = FALSE;\r
+\r
         break;\r
       }\r
 \r
@@ -2045,12 +2227,6 @@ ParseIfrData (
         break;\r
       }\r
 \r
-      if (BlockData->OpCode == EFI_IFR_ORDERED_LIST_OP) {\r
-        //\r
-        // OrderedList Opcode is no default value.\r
-        //\r
-        break;\r
-      }\r
       //\r
       // Get the DefaultId\r
       //\r
@@ -2084,19 +2260,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
@@ -2140,6 +2329,8 @@ GetBlockElement (
   IFR_BLOCK_DATA       *NextBlockData;\r
   UINTN                Length;\r
 \r
+  TmpBuffer = NULL;\r
+\r
   //\r
   // Init RequestBlockArray\r
   //\r
@@ -2475,7 +2666,7 @@ GenerateConfigRequest (
   //\r
   // Start with <ConfigHdr>\r
   //\r
-  StrCpy (StringPtr, ConfigHdr);\r
+  StrCpyS (StringPtr, Length, ConfigHdr);\r
   StringPtr += StrLen (StringPtr);\r
 \r
   //\r
@@ -2574,12 +2765,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
@@ -2653,6 +2844,7 @@ GetStorageWidth (
 /**\r
   Generate ConfigAltResp string base on the varstore info.\r
 \r
+  @param      HiiHandle             Hii Handle for this hii package.\r
   @param      ConfigHdr             The config header for this varstore.\r
   @param      VarStorageData        The varstore info.\r
   @param      DefaultIdArray        The Default id array.\r
@@ -2663,6 +2855,7 @@ GetStorageWidth (
 **/\r
 EFI_STATUS\r
 GenerateAltConfigResp (\r
+  IN  EFI_HII_HANDLE               HiiHandle,\r
   IN  CHAR16                       *ConfigHdr,\r
   IN  IFR_VARSTORAGE_DATA          *VarStorageData,\r
   IN  IFR_DEFAULT_DATA             *DefaultIdArray,\r
@@ -2681,10 +2874,11 @@ GenerateAltConfigResp (
   IFR_DEFAULT_DATA      *DefaultValueData;\r
   UINTN                 Width;\r
   UINT8                 *TmpBuffer;\r
+  CHAR16                *DefaultString;\r
 \r
   BlockData     = NULL;\r
   DataExist     = FALSE;\r
-\r
+  DefaultString = NULL;\r
   //\r
   // Add length for <ConfigHdr> + '\0'\r
   //\r
@@ -2743,7 +2937,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
@@ -2796,10 +2990,19 @@ GenerateAltConfigResp (
         // Convert Value to a hex string in "%x" format\r
         // NOTE: This is in the opposite byte that GUID and PATH use\r
         //\r
-        TmpBuffer = (UINT8 *) &(DefaultValueData->Value);\r
-        for (; Width > 0; Width--) {\r
+        if (BlockData->OpCode == EFI_IFR_STRING_OP){\r
+          DefaultString   = InternalGetString(HiiHandle, DefaultValueData->Value.string);\r
+          TmpBuffer = (UINT8 *) DefaultString;\r
+        } else {\r
+          TmpBuffer = (UINT8 *) &(DefaultValueData->Value);\r
+        }\r
+        for (; Width > 0 && (TmpBuffer != NULL); Width--) {\r
           StringPtr += UnicodeValueToString (StringPtr, PREFIX_ZERO | RADIX_HEX, TmpBuffer[Width - 1], 2);\r
         }\r
+        if (DefaultString != NULL){\r
+          FreePool(DefaultString);\r
+          DefaultString = NULL;\r
+        }\r
       }\r
     }\r
   }\r
@@ -2875,7 +3078,6 @@ GetFullStringFromHiiFormPackages (
   EFI_STRING                   ConfigHdr;\r
   EFI_STRING                   StringPtr;\r
   EFI_STRING                   Progress;\r
-  BOOLEAN                      DataExist;\r
 \r
   if (DataBaseRecord == NULL || DevicePath == NULL || Request == NULL || AltCfgResp == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -2891,12 +3093,11 @@ GetFullStringFromHiiFormPackages (
   ConfigHdr         = NULL;\r
   HiiFormPackage    = NULL;\r
   PackageSize       = 0;\r
-  DataExist         = FALSE;\r
   Progress          = *Request;\r
 \r
   Status = GetFormPackageData (DataBaseRecord, &HiiFormPackage, &PackageSize);\r
   if (EFI_ERROR (Status)) {\r
-    return Status;\r
+    goto Done;\r
   }\r
 \r
   //\r
@@ -2999,7 +3200,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
@@ -3023,7 +3224,7 @@ GetFullStringFromHiiFormPackages (
   // Go through all VarStorageData Entry and get the DefaultId array for each one\r
   // Then construct them all to : ConfigHdr AltConfigHdr ConfigBody AltConfigHdr ConfigBody\r
   //\r
-  Status = GenerateAltConfigResp (ConfigHdr, VarStorageData, DefaultIdArray, &DefaultAltCfgResp);\r
+  Status = GenerateAltConfigResp (DataBaseRecord->Handle,ConfigHdr, VarStorageData, DefaultIdArray, &DefaultAltCfgResp);\r
   if (EFI_ERROR (Status)) {\r
     goto Done;\r
   }\r
@@ -3157,10 +3358,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
@@ -3457,8 +3659,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
@@ -3492,6 +3696,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
@@ -3577,11 +3782,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
@@ -3671,7 +3873,9 @@ HiiConfigRoutingExtractConfig (
                       &gEfiHiiConfigAccessProtocolGuid,\r
                       (VOID **) &ConfigAccess\r
                       );\r
-      ASSERT_EFI_ERROR (Status);\r
+      if (EFI_ERROR (Status)) {\r
+        goto Done;\r
+      }\r
 \r
       Status = ConfigAccess->ExtractConfig (\r
                                ConfigAccess,\r
@@ -3776,9 +3980,9 @@ Done:
                                  instance.\r
   @param  Results                Null-terminated Unicode string in\r
                                  <MultiConfigAltResp> format which has all values\r
-                                 filled in for the names in the Request string.\r
-                                 String to be allocated by the  called function.\r
-                                 De-allocation is up to the caller.\r
+                                 filled in for the entirety of the current HII \r
+                                 database. String to be allocated by the  called \r
+                                 function. De-allocation is up to the caller.\r
 \r
   @retval EFI_SUCCESS            The Results string is filled with the values\r
                                  corresponding to all requested names.\r
@@ -4015,6 +4219,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
@@ -4086,11 +4291,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
@@ -4153,6 +4355,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
@@ -4245,6 +4448,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
@@ -4294,12 +4499,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
@@ -4331,7 +4535,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
@@ -4344,7 +4548,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
@@ -4355,7 +4559,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
@@ -4368,7 +4572,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
@@ -4422,8 +4626,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
@@ -4491,8 +4695,9 @@ Exit:
                                  (see below)  is returned.\r
   @param  BlockSize              The length of the Block in units of UINT8.  On\r
                                  input, this is the size of the Block. On output,\r
-                                 if successful, contains the index of the  last\r
-                                 modified byte in the Block.\r
+                                 if successful, contains the largest index of the\r
+                                 modified byte in the Block, or the required buffer\r
+                                 size if the Block is not large enough.\r
   @param  Progress               On return, points to an element of the ConfigResp\r
                                  string filled in with the offset of the most\r
                                  recent '&' before the first failing name / value\r
@@ -4512,7 +4717,8 @@ Exit:
                                  value pair. Block is left updated and\r
                                  Progress points at the '&' preceding the first\r
                                  non-<BlockName>.\r
-  @retval EFI_DEVICE_ERROR       Block not large enough. Progress undefined.\r
+  @retval EFI_BUFFER_TOO_SMALL   Block not large enough. Progress undefined. \r
+                                 BlockSize is updated with the required buffer size.\r
   @retval EFI_NOT_FOUND          Target for the specified routing data was not found.\r
                                  Progress points to the "G" in "GUID" of the errant\r
                                  routing data.\r
@@ -4530,6 +4736,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
@@ -4539,6 +4746,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
@@ -4588,13 +4797,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
@@ -4607,7 +4817,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
@@ -4618,7 +4828,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
@@ -4631,7 +4841,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
@@ -4642,13 +4852,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
@@ -4689,7 +4899,7 @@ HiiConfigToBlock (
   if (MaxBlockSize > BufferSize) {\r
     *BlockSize = MaxBlockSize;\r
     if (Block != NULL) {\r
-      return EFI_DEVICE_ERROR;\r
+      return EFI_BUFFER_TOO_SMALL;\r
     }\r
   }\r
 \r
@@ -4934,8 +5144,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