]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/SetupBrowserDxe/Setup.c
Add check to make code more safely.
[mirror_edk2.git] / MdeModulePkg / Universal / SetupBrowserDxe / Setup.c
index ed904161b619dd538b8ce296c83764791da2f225..ff3b7e5b4d9621c473552845a8c2118c439145ef 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
 Entry and initialization module for the browser.\r
 \r
-Copyright (c) 2007 - 2013, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2007 - 2014, 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
@@ -26,167 +26,420 @@ SETUP_DRIVER_PRIVATE_DATA  mPrivateData = {
     RegisterHotKey,\r
     RegiserExitHandler,\r
     SaveReminder\r
+  },\r
+  {\r
+    BROWSER_EXTENSION2_VERSION_1_1,\r
+    SetScope,\r
+    RegisterHotKey,\r
+    RegiserExitHandler,\r
+    IsBrowserDataModified,\r
+    ExecuteAction,\r
+    {NULL,NULL},\r
+    {NULL,NULL},\r
+    IsResetRequired\r
   }\r
 };\r
 \r
 EFI_HII_DATABASE_PROTOCOL         *mHiiDatabase;\r
-EFI_HII_STRING_PROTOCOL           *mHiiString;\r
 EFI_HII_CONFIG_ROUTING_PROTOCOL   *mHiiConfigRouting;\r
 EFI_DEVICE_PATH_FROM_TEXT_PROTOCOL *mPathFromText;\r
+EDKII_FORM_DISPLAY_ENGINE_PROTOCOL *mFormDisplay;\r
 \r
 UINTN           gBrowserContextCount = 0;\r
 LIST_ENTRY      gBrowserContextList = INITIALIZE_LIST_HEAD_VARIABLE (gBrowserContextList);\r
 LIST_ENTRY      gBrowserFormSetList = INITIALIZE_LIST_HEAD_VARIABLE (gBrowserFormSetList);\r
 LIST_ENTRY      gBrowserHotKeyList  = INITIALIZE_LIST_HEAD_VARIABLE (gBrowserHotKeyList);\r
-LIST_ENTRY      gBrowserStorageList  = INITIALIZE_LIST_HEAD_VARIABLE (gBrowserStorageList);\r
+LIST_ENTRY      gBrowserStorageList = INITIALIZE_LIST_HEAD_VARIABLE (gBrowserStorageList);\r
+LIST_ENTRY      gBrowserSaveFailFormSetList = INITIALIZE_LIST_HEAD_VARIABLE (gBrowserSaveFailFormSetList);\r
 \r
-BANNER_DATA           *gBannerData;\r
-EFI_HII_HANDLE        gFrontPageHandle;\r
-UINTN                 gClassOfVfr;\r
-UINTN                 gFunctionKeySetting;\r
+BOOLEAN               mSystemSubmit = FALSE;\r
 BOOLEAN               gResetRequired;\r
-EFI_HII_HANDLE        gHiiHandle;\r
-UINT16                gDirection;\r
-EFI_SCREEN_DESCRIPTOR gScreenDimensions;\r
+BOOLEAN               gExitRequired;\r
 BROWSER_SETTING_SCOPE gBrowserSettingScope = FormSetLevel;\r
 BOOLEAN               mBrowserScopeFirstSet = TRUE;\r
 EXIT_HANDLER          ExitHandlerFunction = NULL;\r
-UINTN                 gFooterHeight;\r
+FORM_BROWSER_FORMSET  *mSystemLevelFormSet;\r
 \r
 //\r
 // Browser Global Strings\r
 //\r
-CHAR16            *gSaveFailed;\r
-CHAR16            *gDiscardFailed;\r
-CHAR16            *gDefaultFailed;\r
-CHAR16            *gEnterString;\r
-CHAR16            *gEnterCommitString;\r
-CHAR16            *gEnterEscapeString;\r
-CHAR16            *gEscapeString;\r
-CHAR16            *gMoveHighlight;\r
-CHAR16            *gMakeSelection;\r
-CHAR16            *gDecNumericInput;\r
-CHAR16            *gHexNumericInput;\r
-CHAR16            *gToggleCheckBox;\r
-CHAR16            *gPromptForData;\r
-CHAR16            *gPromptForPassword;\r
-CHAR16            *gPromptForNewPassword;\r
-CHAR16            *gConfirmPassword;\r
-CHAR16            *gConfirmError;\r
-CHAR16            *gPassowordInvalid;\r
-CHAR16            *gPressEnter;\r
 CHAR16            *gEmptyString;\r
-CHAR16            *gAreYouSure;\r
-CHAR16            *gYesResponse;\r
-CHAR16            *gNoResponse;\r
-CHAR16            *gMiniString;\r
-CHAR16            *gPlusString;\r
-CHAR16            *gMinusString;\r
-CHAR16            *gAdjustNumber;\r
-CHAR16            *gSaveChanges;\r
-CHAR16            *gOptionMismatch;\r
-CHAR16            *gFormSuppress;\r
-CHAR16            *gProtocolNotFound;\r
-\r
 CHAR16            *mUnknownString = L"!";\r
 \r
-CHAR16            gPromptBlockWidth;\r
-CHAR16            gOptionBlockWidth;\r
-CHAR16            gHelpBlockWidth;\r
-\r
 EFI_GUID  gZeroGuid = {0, 0, 0, {0, 0, 0, 0, 0, 0, 0, 0}};\r
-EFI_GUID  gSetupBrowserGuid = {\r
-  0xab368524, 0xb60c, 0x495b, {0xa0, 0x9, 0x12, 0xe8, 0x5b, 0x1a, 0xea, 0x32}\r
-};\r
 \r
-FORM_BROWSER_FORMSET  *gOldFormSet = NULL;\r
+extern EFI_GUID        mCurrentFormSetGuid;\r
+extern EFI_HII_HANDLE  mCurrentHiiHandle;\r
+extern UINT16          mCurrentFormId;\r
+extern FORM_DISPLAY_ENGINE_FORM gDisplayFormData;\r
+\r
+/**\r
+  Create a menu with specified formset GUID and form ID, and add it as a child\r
+  of the given parent menu.\r
+\r
+  @param  HiiHandle              Hii handle related to this formset.\r
+  @param  FormSetGuid            The Formset Guid of menu to be added.\r
+  @param  FormId                 The Form ID of menu to be added.\r
+  @param  QuestionId             The question id of this menu to be added.\r
+\r
+  @return A pointer to the newly added menu or NULL if memory is insufficient.\r
+\r
+**/\r
+FORM_ENTRY_INFO *\r
+UiAddMenuList (\r
+  IN EFI_HII_HANDLE       HiiHandle,\r
+  IN EFI_GUID             *FormSetGuid,\r
+  IN UINT16               FormId,\r
+  IN UINT16               QuestionId\r
+  )\r
+{\r
+  FORM_ENTRY_INFO  *MenuList;\r
+\r
+  MenuList = AllocateZeroPool (sizeof (FORM_ENTRY_INFO));\r
+  if (MenuList == NULL) {\r
+    return NULL;\r
+  }\r
+\r
+  MenuList->Signature = FORM_ENTRY_INFO_SIGNATURE;\r
+\r
+  MenuList->HiiHandle  = HiiHandle;\r
+  CopyMem (&MenuList->FormSetGuid, FormSetGuid, sizeof (EFI_GUID));\r
+  MenuList->FormId     = FormId;\r
+  MenuList->QuestionId = QuestionId;\r
 \r
-FUNCTIION_KEY_SETTING gFunctionKeySettingTable[] = {\r
   //\r
-  // Boot Manager\r
+  // If parent is not specified, it is the root Form of a Formset\r
   //\r
-  {\r
-    {\r
-      0x847bc3fe,\r
-      0xb974,\r
-      0x446d,\r
-      {\r
-        0x94,\r
-        0x49,\r
-        0x5a,\r
-        0xd5,\r
-        0x41,\r
-        0x2e,\r
-        0x99,\r
-        0x3b\r
+  InsertTailList (&mPrivateData.FormBrowserEx2.FormViewHistoryHead, &MenuList->Link);\r
+\r
+  return MenuList;\r
+}\r
+\r
+/**\r
+  Return the form id for the input hiihandle and formset.\r
+\r
+  @param  HiiHandle              HiiHandle for FormSet.\r
+  @param  FormSetGuid            The Formset GUID of the menu to search.\r
+\r
+  @return First form's id for this form set.\r
+\r
+**/\r
+EFI_FORM_ID\r
+GetFirstFormId (\r
+  IN EFI_HII_HANDLE       HiiHandle,\r
+  IN EFI_GUID             *FormSetGuid\r
+  )\r
+{\r
+  LIST_ENTRY         *Link;\r
+  FORM_BROWSER_FORM  *Form;\r
+\r
+  Link = GetFirstNode (&gCurrentSelection->FormSet->FormListHead);\r
+  Form = FORM_BROWSER_FORM_FROM_LINK (Link);\r
+\r
+  return Form->FormId;\r
+}\r
+\r
+/**\r
+  Search Menu with given FormSetGuid and FormId in all cached menu list.\r
+\r
+  @param  HiiHandle              HiiHandle for FormSet.\r
+  @param  FormSetGuid            The Formset GUID of the menu to search.\r
+  @param  FormId                 The Form ID of menu to search.\r
+\r
+  @return A pointer to menu found or NULL if not found.\r
+\r
+**/\r
+FORM_ENTRY_INFO *\r
+UiFindMenuList (\r
+  IN EFI_HII_HANDLE       HiiHandle, \r
+  IN EFI_GUID             *FormSetGuid,\r
+  IN UINT16               FormId\r
+  )\r
+{\r
+  LIST_ENTRY         *Link;\r
+  FORM_ENTRY_INFO    *MenuList;\r
+  FORM_ENTRY_INFO    *RetMenu;\r
+  EFI_FORM_ID        FirstFormId;\r
+\r
+  RetMenu = NULL;\r
+\r
+  Link = GetFirstNode (&mPrivateData.FormBrowserEx2.FormViewHistoryHead);\r
+  while (!IsNull (&mPrivateData.FormBrowserEx2.FormViewHistoryHead, Link)) {\r
+    MenuList = FORM_ENTRY_INFO_FROM_LINK (Link);\r
+    Link = GetNextNode (&mPrivateData.FormBrowserEx2.FormViewHistoryHead, Link);\r
+    \r
+    //\r
+    // If already find the menu, free the menus behind it.\r
+    //\r
+    if (RetMenu != NULL) {\r
+      RemoveEntryList (&MenuList->Link);\r
+      FreePool (MenuList);\r
+      continue;\r
+    }\r
+\r
+    //\r
+    // Find the same FromSet.\r
+    //\r
+    if (MenuList->HiiHandle == HiiHandle) {\r
+      if (CompareGuid (&MenuList->FormSetGuid, &gZeroGuid)) {\r
+        //\r
+        // FormSetGuid is not specified.\r
+        //\r
+        RetMenu = MenuList;\r
+      } else if (CompareGuid (&MenuList->FormSetGuid, FormSetGuid)) {\r
+        if (MenuList->FormId == FormId) {\r
+          RetMenu = MenuList;\r
+        } else if (FormId == 0 || MenuList->FormId == 0 ) {\r
+          FirstFormId = GetFirstFormId (HiiHandle, FormSetGuid);\r
+          if ((FormId == 0 && FirstFormId == MenuList->FormId) || (MenuList->FormId ==0 && FirstFormId == FormId)) {\r
+            RetMenu = MenuList;\r
+          }\r
+        }\r
       }\r
-    },\r
-    NONE_FUNCTION_KEY_SETTING\r
-  },\r
+    }\r
+  }\r
+\r
+  return RetMenu;\r
+}\r
+\r
+/**\r
+  Find parent menu for current menu.\r
+\r
+  @param  CurrentMenu    Current Menu\r
+  @param  SettingLevel   Whether find parent menu in Form Level or Formset level.\r
+                         In form level, just find the parent menu; \r
+                         In formset level, find the parent menu which has different\r
+                         formset guid value.\r
+\r
+  @retval   The parent menu for current menu.\r
+**/\r
+FORM_ENTRY_INFO *\r
+UiFindParentMenu (\r
+  IN FORM_ENTRY_INFO          *CurrentMenu,\r
+  IN BROWSER_SETTING_SCOPE    SettingLevel\r
+  )\r
+{\r
+  FORM_ENTRY_INFO    *ParentMenu;\r
+  LIST_ENTRY         *Link;\r
+\r
+  ASSERT (SettingLevel == FormLevel || SettingLevel == FormSetLevel);\r
+\r
+  if (CurrentMenu == NULL) {\r
+    return NULL;\r
+  }\r
+  \r
+  ParentMenu = NULL;\r
+  Link       = &CurrentMenu->Link;\r
+\r
+  while (Link->BackLink != &mPrivateData.FormBrowserEx2.FormViewHistoryHead) {\r
+    ParentMenu = FORM_ENTRY_INFO_FROM_LINK (Link->BackLink);\r
+\r
+    if (SettingLevel == FormLevel) {\r
+      //\r
+      // For FormLevel, just find the parent menu, return.\r
+      //\r
+      break;\r
+    }\r
+\r
+    if (!CompareGuid (&CurrentMenu->FormSetGuid, &ParentMenu->FormSetGuid)) {\r
+      //\r
+      // For SystemLevel, must find the menu which has different formset.\r
+      //\r
+      break;\r
+    }\r
+\r
+    Link = Link->BackLink;\r
+  }\r
+\r
   //\r
-  // Device Manager\r
+  // Not find the parent menu, just return NULL.\r
   //\r
-  {\r
-    {\r
-      0x3ebfa8e6,\r
-      0x511d,\r
-      0x4b5b,\r
-      {\r
-        0xa9,\r
-        0x5f,\r
-        0xfb,\r
-        0x38,\r
-        0x26,\r
-        0xf,\r
-        0x1c,\r
-        0x27\r
-      }\r
-    },\r
-    NONE_FUNCTION_KEY_SETTING\r
-  },\r
+  if (Link->BackLink == &mPrivateData.FormBrowserEx2.FormViewHistoryHead) {\r
+    return NULL;\r
+  }\r
+\r
+  return ParentMenu;\r
+}\r
+\r
+/**\r
+  Free Menu list linked list.\r
+\r
+  @param  MenuListHead    One Menu list point in the menu list.\r
+\r
+**/\r
+VOID\r
+UiFreeMenuList (\r
+  LIST_ENTRY   *MenuListHead\r
+  )\r
+{\r
+  FORM_ENTRY_INFO    *MenuList;\r
+\r
+  while (!IsListEmpty (MenuListHead)) {\r
+    MenuList = FORM_ENTRY_INFO_FROM_LINK (MenuListHead->ForwardLink);\r
+    RemoveEntryList (&MenuList->Link);\r
+\r
+    FreePool (MenuList);\r
+  }\r
+}\r
+\r
+/**\r
+  Copy current Menu list to the new menu list.\r
+  \r
+  @param  NewMenuListHead        New create Menu list.\r
+  @param  CurrentMenuListHead    Current Menu list.\r
+\r
+**/\r
+VOID\r
+UiCopyMenuList (\r
+  OUT LIST_ENTRY   *NewMenuListHead,\r
+  IN  LIST_ENTRY   *CurrentMenuListHead\r
+  )\r
+{\r
+  LIST_ENTRY         *Link;\r
+  FORM_ENTRY_INFO    *MenuList;\r
+  FORM_ENTRY_INFO    *NewMenuEntry;\r
+\r
   //\r
-  // BMM FormSet.\r
+  // If new menu list not empty, free it first.\r
   //\r
-  {\r
-    {\r
-      0x642237c7,\r
-      0x35d4,\r
-      0x472d,\r
-      {\r
-        0x83,\r
-        0x65,\r
-        0x12,\r
-        0xe0,\r
-        0xcc,\r
-        0xf2,\r
-        0x7a,\r
-        0x22\r
-      }\r
-    },\r
-    NONE_FUNCTION_KEY_SETTING\r
-  },\r
+  UiFreeMenuList (NewMenuListHead);\r
+\r
+  Link = GetFirstNode (CurrentMenuListHead);\r
+  while (!IsNull (CurrentMenuListHead, Link)) {\r
+    MenuList = FORM_ENTRY_INFO_FROM_LINK (Link);\r
+    Link = GetNextNode (CurrentMenuListHead, Link);\r
+\r
+    NewMenuEntry = AllocateZeroPool (sizeof (FORM_ENTRY_INFO));\r
+    ASSERT (NewMenuEntry != NULL);\r
+    NewMenuEntry->Signature  = FORM_ENTRY_INFO_SIGNATURE;\r
+    NewMenuEntry->HiiHandle  = MenuList->HiiHandle;\r
+    CopyMem (&NewMenuEntry->FormSetGuid, &MenuList->FormSetGuid, sizeof (EFI_GUID));\r
+    NewMenuEntry->FormId     = MenuList->FormId;\r
+    NewMenuEntry->QuestionId = MenuList->QuestionId;\r
+\r
+    InsertTailList (NewMenuListHead, &NewMenuEntry->Link);\r
+  }\r
+}\r
+\r
+/**\r
+  Load all hii formset to the browser.\r
+\r
+**/\r
+VOID\r
+LoadAllHiiFormset (\r
+  VOID\r
+  )\r
+{\r
+  FORM_BROWSER_FORMSET    *LocalFormSet;\r
+  EFI_HII_HANDLE          *HiiHandles;\r
+  UINTN                   Index;\r
+  EFI_GUID                ZeroGuid;\r
+  EFI_STATUS              Status;\r
+  FORM_BROWSER_FORMSET    *OldFormset;\r
+\r
+  OldFormset = mSystemLevelFormSet;\r
+\r
   //\r
-  // BMM File Explorer FormSet.\r
+  // Get all the Hii handles\r
   //\r
-  {\r
-    {\r
-      0x1f2d63e1,\r
-      0xfebd,\r
-      0x4dc7,\r
-      {\r
-        0x9c,\r
-        0xc5,\r
-        0xba,\r
-        0x2b,\r
-        0x1c,\r
-        0xef,\r
-        0x9c,\r
-        0x5b\r
-      }\r
-    },\r
-    NONE_FUNCTION_KEY_SETTING\r
-  },\r
-};\r
+  HiiHandles = HiiGetHiiHandles (NULL);\r
+  ASSERT (HiiHandles != NULL);\r
+\r
+  //\r
+  // Search for formset of each class type\r
+  //\r
+  for (Index = 0; HiiHandles[Index] != NULL; Index++) {\r
+    //\r
+    // Check HiiHandles[Index] does exist in global maintain list.\r
+    //\r
+    if (GetFormSetFromHiiHandle (HiiHandles[Index]) != NULL) {\r
+      continue;\r
+    }\r
+\r
+    //\r
+    // Initilize FormSet Setting\r
+    //\r
+    LocalFormSet = AllocateZeroPool (sizeof (FORM_BROWSER_FORMSET));\r
+    ASSERT (LocalFormSet != NULL);\r
+    mSystemLevelFormSet = LocalFormSet;\r
+\r
+    ZeroMem (&ZeroGuid, sizeof (ZeroGuid));\r
+    Status = InitializeFormSet (HiiHandles[Index], &ZeroGuid, LocalFormSet);\r
+    if (EFI_ERROR (Status) || IsListEmpty (&LocalFormSet->FormListHead)) {\r
+      DestroyFormSet (LocalFormSet);\r
+      continue;\r
+    }\r
+    InitializeCurrentSetting (LocalFormSet);\r
+\r
+    //\r
+    // Initilize Questions' Value\r
+    //\r
+    Status = LoadFormSetConfig (NULL, LocalFormSet);\r
+    if (EFI_ERROR (Status)) {\r
+      DestroyFormSet (LocalFormSet);\r
+      continue;\r
+    }\r
+  }\r
+\r
+  //\r
+  // Free resources, and restore gOldFormSet and gClassOfVfr\r
+  //\r
+  FreePool (HiiHandles);\r
+\r
+  mSystemLevelFormSet = OldFormset;\r
+}\r
+\r
+/**\r
+  Pop up the error info.\r
+\r
+  @param      BrowserStatus    The input browser status.\r
+  @param      HiiHandle        The Hiihandle for this opcode.\r
+  @param      OpCode           The opcode use to get the erro info and timeout value.\r
+  @param      ErrorString      Error string used by BROWSER_NO_SUBMIT_IF.\r
+\r
+**/\r
+UINT32\r
+PopupErrorMessage (\r
+  IN UINT32                BrowserStatus,\r
+  IN EFI_HII_HANDLE        HiiHandle,\r
+  IN EFI_IFR_OP_HEADER     *OpCode, OPTIONAL\r
+  IN CHAR16                *ErrorString\r
+  )\r
+{\r
+  FORM_DISPLAY_ENGINE_STATEMENT *Statement;\r
+  USER_INPUT                    UserInputData;\r
+\r
+  Statement = NULL;\r
+\r
+  if (OpCode != NULL) {\r
+    Statement = AllocateZeroPool (sizeof(FORM_DISPLAY_ENGINE_STATEMENT));\r
+    ASSERT (Statement != NULL);\r
+    Statement->OpCode = OpCode;\r
+    gDisplayFormData.HighLightedStatement = Statement;\r
+  }\r
+\r
+  //\r
+  // Used to compatible with old display engine.\r
+  // New display engine not use this field.\r
+  //\r
+  gDisplayFormData.ErrorString   = ErrorString;\r
+  gDisplayFormData.BrowserStatus = BrowserStatus;\r
+\r
+  if (HiiHandle != NULL) {\r
+    gDisplayFormData.HiiHandle     = HiiHandle;\r
+  }\r
+\r
+  mFormDisplay->FormDisplay (&gDisplayFormData, &UserInputData);\r
+\r
+  gDisplayFormData.BrowserStatus = BROWSER_SUCCESS;\r
+  gDisplayFormData.ErrorString   = NULL;\r
+\r
+  if (OpCode != NULL) {\r
+    FreePool (Statement);\r
+  }\r
+\r
+  return UserInputData.Action;\r
+}\r
 \r
 /**\r
   This is the routine which an external caller uses to direct the browser\r
@@ -229,26 +482,14 @@ SendForm (
   UI_MENU_SELECTION             *Selection;\r
   UINTN                         Index;\r
   FORM_BROWSER_FORMSET          *FormSet;\r
-  LIST_ENTRY                    *Link;\r
+  FORM_ENTRY_INFO               *MenuList;\r
 \r
   //\r
-  // Calculate total number of Register HotKeys. \r
+  // If EDKII_FORM_DISPLAY_ENGINE_PROTOCOL not found, return EFI_UNSUPPORTED.\r
   //\r
-  Index = 0;\r
-  Link  = GetFirstNode (&gBrowserHotKeyList);\r
-  while (!IsNull (&gBrowserHotKeyList, Link)) {\r
-    Link = GetNextNode (&gBrowserHotKeyList, Link);\r
-    Index ++;\r
+  if (mFormDisplay == NULL) {\r
+    return EFI_UNSUPPORTED;\r
   }\r
-  //\r
-  // Show three HotKeys help information on one ROW.\r
-  //\r
-  gFooterHeight = FOOTER_HEIGHT + (Index / 3);\r
-\r
-  //\r
-  // Clean the history menu list.\r
-  //\r
-  InitializeListHead (&gMenuList);\r
 \r
   //\r
   // Save globals used by SendForm()\r
@@ -256,65 +497,10 @@ SendForm (
   SaveBrowserContext ();\r
 \r
   gResetRequired = FALSE;\r
-  Status = EFI_SUCCESS;\r
-  ZeroMem (&gScreenDimensions, sizeof (EFI_SCREEN_DESCRIPTOR));\r
-\r
-  //\r
-  // Seed the dimensions in the global\r
-  //\r
-  gST->ConOut->QueryMode (\r
-                 gST->ConOut,\r
-                 gST->ConOut->Mode->Mode,\r
-                 &gScreenDimensions.RightColumn,\r
-                 &gScreenDimensions.BottomRow\r
-                 );\r
-\r
-  if (ScreenDimensions != NULL) {\r
-    //\r
-    // Check local dimension vs. global dimension.\r
-    //\r
-    if ((gScreenDimensions.RightColumn < ScreenDimensions->RightColumn) ||\r
-        (gScreenDimensions.BottomRow < ScreenDimensions->BottomRow)\r
-        ) {\r
-      Status = EFI_INVALID_PARAMETER;\r
-      goto Done;\r
-    } else {\r
-      //\r
-      // Local dimension validation.\r
-      //\r
-      if ((ScreenDimensions->RightColumn > ScreenDimensions->LeftColumn) &&\r
-          (ScreenDimensions->BottomRow > ScreenDimensions->TopRow) &&\r
-          ((ScreenDimensions->RightColumn - ScreenDimensions->LeftColumn) > 2) &&\r
-          (\r
-            (ScreenDimensions->BottomRow - ScreenDimensions->TopRow) > STATUS_BAR_HEIGHT +\r
-            SCROLL_ARROW_HEIGHT *\r
-            2 +\r
-            FRONT_PAGE_HEADER_HEIGHT +\r
-            gFooterHeight +\r
-            1\r
-          )\r
-        ) {\r
-        CopyMem (&gScreenDimensions, (VOID *) ScreenDimensions, sizeof (EFI_SCREEN_DESCRIPTOR));\r
-      } else {\r
-        Status = EFI_INVALID_PARAMETER;\r
-        goto Done;\r
-      }\r
-    }\r
-  }\r
-\r
-  gOptionBlockWidth = (CHAR16) ((gScreenDimensions.RightColumn - gScreenDimensions.LeftColumn) / 3);\r
-  gPromptBlockWidth = (CHAR16) (gOptionBlockWidth + LEFT_SKIPPED_COLUMNS);\r
-  gHelpBlockWidth   = (CHAR16) (gOptionBlockWidth - LEFT_SKIPPED_COLUMNS);\r
-\r
-  //\r
-  // Initialize the strings for the browser, upon exit of the browser, the strings will be freed\r
-  //\r
-  InitializeBrowserStrings ();\r
-\r
-  //\r
-  // Ensure we are in Text mode\r
-  //\r
-  gST->ConOut->SetAttribute (gST->ConOut, EFI_TEXT_ATTR (EFI_LIGHTGRAY, EFI_BLACK));\r
+  gExitRequired  = FALSE;\r
+  Status         = EFI_SUCCESS;\r
+  gEmptyString   = L"";\r
+  gDisplayFormData.ScreenDimensions = (EFI_SCREEN_DESCRIPTOR *) ScreenDimensions;\r
 \r
   for (Index = 0; Index < HandleCount; Index++) {\r
     Selection = AllocateZeroPool (sizeof (UI_MENU_SELECTION));\r
@@ -332,20 +518,24 @@ SendForm (
       FormSet = AllocateZeroPool (sizeof (FORM_BROWSER_FORMSET));\r
       ASSERT (FormSet != NULL);\r
 \r
+      //\r
+      // Validate the HiiHandle\r
+      // if validate failed, find the first validate parent HiiHandle.\r
+      //\r
+      if (!ValidateHiiHandle(Selection->Handle)) {\r
+        FindNextMenu (Selection, FormSetLevel);\r
+      }\r
+\r
       //\r
       // Initialize internal data structures of FormSet\r
       //\r
-      Status = InitializeFormSet (Selection->Handle, &Selection->FormSetGuid, FormSet, TRUE);\r
+      Status = InitializeFormSet (Selection->Handle, &Selection->FormSetGuid, FormSet);\r
       if (EFI_ERROR (Status) || IsListEmpty (&FormSet->FormListHead)) {\r
         DestroyFormSet (FormSet);\r
         break;\r
       }\r
       Selection->FormSet = FormSet;\r
-\r
-      //\r
-      // Try to find pre FormSet in the maintain backup list.\r
-      //\r
-      gOldFormSet = GetFormSetFromHiiHandle (Selection->Handle);\r
+      mSystemLevelFormSet = FormSet;\r
 \r
       //\r
       // Display this formset\r
@@ -355,24 +545,21 @@ SendForm (
       Status = SetupBrowser (Selection);\r
 \r
       gCurrentSelection = NULL;\r
+      mSystemLevelFormSet = NULL;\r
 \r
-      if (EFI_ERROR (Status)) {\r
-        break;\r
-      }\r
-\r
-    } while (Selection->Action == UI_ACTION_REFRESH_FORMSET);\r
-\r
-    if (gOldFormSet != NULL) {\r
       //\r
       // If no data is changed, don't need to save current FormSet into the maintain list.\r
       //\r
-      if (!IsNvUpdateRequired (gOldFormSet)) {\r
-        CleanBrowserStorage(gOldFormSet);\r
-        RemoveEntryList (&gOldFormSet->Link);\r
-        DestroyFormSet (gOldFormSet);\r
+      if (!IsNvUpdateRequiredForFormSet (FormSet)) {\r
+        CleanBrowserStorage(FormSet);\r
+        RemoveEntryList (&FormSet->Link);\r
+        DestroyFormSet (FormSet);\r
       }\r
-      gOldFormSet = NULL;\r
-    }\r
+\r
+      if (EFI_ERROR (Status)) {\r
+        break;\r
+      }\r
+    } while (Selection->Action == UI_ACTION_REFRESH_FORMSET);\r
 \r
     FreePool (Selection);\r
   }\r
@@ -384,13 +571,17 @@ SendForm (
     }\r
   }\r
 \r
-  FreeBrowserStrings ();\r
-  UiFreeMenuList(&gMenuList);\r
+  mFormDisplay->ExitDisplay();\r
 \r
-  gST->ConOut->SetAttribute (gST->ConOut, EFI_TEXT_ATTR (EFI_LIGHTGRAY, EFI_BLACK));\r
-  gST->ConOut->ClearScreen (gST->ConOut);\r
+  //\r
+  // Clear the menu history data.\r
+  //\r
+  while (!IsListEmpty (&mPrivateData.FormBrowserEx2.FormViewHistoryHead)) {\r
+    MenuList = FORM_ENTRY_INFO_FROM_LINK (mPrivateData.FormBrowserEx2.FormViewHistoryHead.ForwardLink);\r
+    RemoveEntryList (&MenuList->Link);\r
+    FreePool (MenuList);\r
+  }\r
 \r
-Done:\r
   //\r
   // Restore globals used by SendForm()\r
   //\r
@@ -399,15 +590,10 @@ Done:
   return Status;\r
 }\r
 \r
-\r
 /**\r
-  This function is called by a callback handler to retrieve uncommitted state\r
-  data from the browser.\r
+  Get or set data to the storage.\r
 \r
-  @param  This                   A pointer to the EFI_FORM_BROWSER2_PROTOCOL\r
-                                 instance.\r
-  @param  ResultsDataSize        A pointer to the size of the buffer associated\r
-                                 with ResultsData.\r
+  @param  ResultsDataSize        The size of the buffer associatedwith ResultsData.\r
   @param  ResultsData            A string returned from an IFR browser or\r
                                  equivalent. The results string will have no\r
                                  routing information in them.\r
@@ -416,9 +602,97 @@ Done:
                                  browser state information or set (if RetrieveData\r
                                  = FALSE) data in the uncommitted browser state\r
                                  information.\r
-  @param  VariableGuid           An optional field to indicate the target variable\r
-                                 GUID name to use.\r
-  @param  VariableName           An optional field to indicate the target\r
+  @param  Storage                The pointer to the storage.\r
+\r
+  @retval EFI_SUCCESS            The results have been distributed or are awaiting\r
+                                 distribution.\r
+\r
+**/\r
+EFI_STATUS \r
+ProcessStorage (\r
+  IN OUT UINTN                         *ResultsDataSize,\r
+  IN OUT EFI_STRING                    *ResultsData,\r
+  IN BOOLEAN                           RetrieveData,\r
+  IN BROWSER_STORAGE                   *Storage\r
+  )\r
+{\r
+  CHAR16                *ConfigResp;\r
+  EFI_STATUS            Status;\r
+  CHAR16                *StrPtr;\r
+  UINTN                 BufferSize;\r
+  UINTN                 TmpSize;\r
+\r
+  if (RetrieveData) {\r
+    //\r
+    // Generate <ConfigResp>\r
+    //\r
+    Status = StorageToConfigResp (Storage, &ConfigResp, Storage->ConfigRequest, TRUE);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+\r
+    //\r
+    // Skip <ConfigHdr> and '&' to point to <ConfigBody> when first copy the configbody.\r
+    // Also need to consider add "\0" at first time.\r
+    //\r
+    StrPtr     = ConfigResp + StrLen (Storage->ConfigHdr) + 1;\r
+    BufferSize = StrSize (StrPtr);\r
+\r
+\r
+    //\r
+    // Copy the data if the input buffer is bigger enough.\r
+    //\r
+    if (*ResultsDataSize >= BufferSize) {\r
+      StrCpy (*ResultsData, StrPtr);\r
+    }\r
+\r
+    *ResultsDataSize = BufferSize;\r
+    FreePool (ConfigResp);\r
+  } else {\r
+    //\r
+    // Prepare <ConfigResp>\r
+    //\r
+    TmpSize = StrLen (*ResultsData);\r
+    BufferSize = (TmpSize + StrLen (Storage->ConfigHdr) + 2) * sizeof (CHAR16);\r
+    ConfigResp = AllocateZeroPool (BufferSize);\r
+    ASSERT (ConfigResp != NULL);\r
+\r
+    StrCpy (ConfigResp, Storage->ConfigHdr);\r
+    StrCat (ConfigResp, L"&");\r
+    StrCat (ConfigResp, *ResultsData);\r
+\r
+    //\r
+    // Update Browser uncommited data\r
+    //\r
+    Status = ConfigRespToStorage (Storage, ConfigResp);\r
+    FreePool (ConfigResp);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+  }\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  This routine called this service in the browser to retrieve or set certain uncommitted \r
+  state information that resides in the open formsets. \r
+\r
+  @param  This                   A pointer to the EFI_FORM_BROWSER2_PROTOCOL\r
+                                 instance.\r
+  @param  ResultsDataSize        A pointer to the size of the buffer associated\r
+                                 with ResultsData.\r
+  @param  ResultsData            A string returned from an IFR browser or\r
+                                 equivalent. The results string will have no\r
+                                 routing information in them.\r
+  @param  RetrieveData           A BOOLEAN field which allows an agent to retrieve\r
+                                 (if RetrieveData = TRUE) data from the uncommitted\r
+                                 browser state information or set (if RetrieveData\r
+                                 = FALSE) data in the uncommitted browser state\r
+                                 information.\r
+  @param  VariableGuid           An optional field to indicate the target variable\r
+                                 GUID name to use.\r
+  @param  VariableName           An optional field to indicate the target\r
                                  human-readable variable name.\r
 \r
   @retval EFI_SUCCESS            The results have been distributed or are awaiting\r
@@ -440,60 +714,75 @@ BrowserCallback (
 {\r
   EFI_STATUS            Status;\r
   LIST_ENTRY            *Link;\r
-  FORMSET_STORAGE       *Storage;\r
-  FORM_BROWSER_FORMSET  *FormSet;\r
+  BROWSER_STORAGE       *Storage;\r
+  FORMSET_STORAGE       *FormsetStorage;\r
+  UINTN                 TotalSize;\r
   BOOLEAN               Found;\r
-  CHAR16                *ConfigResp;\r
-  CHAR16                *StrPtr;\r
-  UINTN                 BufferSize;\r
-  UINTN                 TmpSize;\r
 \r
   if (ResultsDataSize == NULL || ResultsData == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  if (gCurrentSelection == NULL) {\r
-    return EFI_NOT_READY;\r
-  }\r
-\r
-  Storage = NULL;\r
-  ConfigResp = NULL;\r
-  FormSet = gCurrentSelection->FormSet;\r
-\r
-  //\r
-  // Find target storage\r
-  //\r
-  Link = GetFirstNode (&FormSet->StorageListHead);\r
-  if (IsNull (&FormSet->StorageListHead, Link)) {\r
-    return EFI_UNSUPPORTED;\r
-  }\r
+  TotalSize = *ResultsDataSize;\r
+  Storage   = NULL;\r
+  Found     = FALSE;\r
+  Status    = EFI_SUCCESS;\r
 \r
   if (VariableGuid != NULL) {\r
     //\r
-    // Try to find target storage\r
+    // Try to find target storage in the current formset.\r
     //\r
-    Found = FALSE;\r
-    while (!IsNull (&FormSet->StorageListHead, Link)) {\r
-      Storage = FORMSET_STORAGE_FROM_LINK (Link);\r
-      Link = GetNextNode (&FormSet->StorageListHead, Link);\r
+    Link = GetFirstNode (&gBrowserStorageList);\r
+    while (!IsNull (&gBrowserStorageList, Link)) {\r
+      Storage = BROWSER_STORAGE_FROM_LINK (Link);\r
+      Link = GetNextNode (&gBrowserStorageList, Link);\r
+      //\r
+      // Check the current storage.\r
+      //\r
+      if (!CompareGuid (&Storage->Guid, (EFI_GUID *) VariableGuid)) {\r
+        continue;\r
+      }\r
 \r
-      if (CompareGuid (&Storage->BrowserStorage->Guid, (EFI_GUID *) VariableGuid)) {\r
-        if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_BUFFER ||\r
-            Storage->BrowserStorage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
-          //\r
-          // Buffer storage require both GUID and Name\r
-          //\r
-          if (VariableName == NULL) {\r
-            return EFI_NOT_FOUND;\r
-          }\r
+      if (Storage->Type == EFI_HII_VARSTORE_BUFFER ||\r
+          Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
+        //\r
+        // Buffer storage require both GUID and Name\r
+        //\r
+        if (VariableName == NULL) {\r
+          return EFI_NOT_FOUND;\r
+        }\r
 \r
-          if (StrCmp (Storage->BrowserStorage->Name, (CHAR16 *) VariableName) != 0) {\r
-            continue;\r
-          }\r
+        if (StrCmp (Storage->Name, (CHAR16 *) VariableName) != 0) {\r
+          continue;\r
         }\r
-        Found = TRUE;\r
-        break;\r
       }\r
+\r
+      if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE ||\r
+          Storage->Type == EFI_HII_VARSTORE_BUFFER) {\r
+        if (mSystemLevelFormSet == NULL || mSystemLevelFormSet->HiiHandle == NULL) {\r
+          return EFI_NOT_FOUND;\r
+        }\r
+\r
+        if (Storage->HiiHandle != mSystemLevelFormSet->HiiHandle) {\r
+          continue;\r
+        }\r
+      }\r
+\r
+      Status = ProcessStorage (&TotalSize, &ResultsData, RetrieveData, Storage);\r
+      if (EFI_ERROR (Status)) {\r
+        return Status;\r
+      }\r
+\r
+      if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
+        ConfigRequestAdjust (Storage, ResultsData, TRUE);\r
+      }\r
+\r
+      //\r
+      // Different formsets may have same varstore, so here just set the flag\r
+      // not exit the circle.\r
+      // \r
+      Found = TRUE;\r
+      break;\r
     }\r
 \r
     if (!Found) {\r
@@ -503,65 +792,61 @@ BrowserCallback (
     //\r
     // GUID/Name is not specified, take the first storage in FormSet\r
     //\r
-    Storage = FORMSET_STORAGE_FROM_LINK (Link);\r
-  }\r
-\r
-  if (RetrieveData) {\r
-    //\r
-    // Skip if there is no RequestElement\r
-    //\r
-    if (Storage->ElementCount == 0) {\r
-      return EFI_SUCCESS;\r
+    if (mSystemLevelFormSet == NULL) {\r
+      return EFI_NOT_READY;\r
     }\r
 \r
     //\r
     // Generate <ConfigResp>\r
     //\r
-    Status = StorageToConfigResp (Storage->BrowserStorage, &ConfigResp, Storage->ConfigRequest);\r
+    Link = GetFirstNode (&mSystemLevelFormSet->StorageListHead);\r
+    if (IsNull (&mSystemLevelFormSet->StorageListHead, Link)) {\r
+      return EFI_UNSUPPORTED;\r
+    }\r
+\r
+    FormsetStorage = FORMSET_STORAGE_FROM_LINK (Link);\r
+    \r
+    Status = ProcessStorage (&TotalSize, &ResultsData, RetrieveData, FormsetStorage->BrowserStorage);\r
     if (EFI_ERROR (Status)) {\r
       return Status;\r
     }\r
+  }\r
 \r
-    //\r
-    // Skip <ConfigHdr> and '&' to point to <ConfigBody>\r
-    //\r
-    StrPtr = ConfigResp + StrLen (Storage->BrowserStorage->ConfigHdr) + 1;\r
+  if (RetrieveData) {\r
+    Status = TotalSize <= *ResultsDataSize ? EFI_SUCCESS : EFI_BUFFER_TOO_SMALL;\r
+    *ResultsDataSize = TotalSize;\r
+  }\r
+   \r
+  return Status;\r
 \r
-    BufferSize = StrSize (StrPtr);\r
-    if (*ResultsDataSize < BufferSize) {\r
-      *ResultsDataSize = BufferSize;\r
+}\r
 \r
-      FreePool (ConfigResp);\r
-      return EFI_BUFFER_TOO_SMALL;\r
-    }\r
 \r
-    *ResultsDataSize = BufferSize;\r
-    CopyMem (ResultsData, StrPtr, BufferSize);\r
+/**\r
+  Callback function for SimpleTextInEx protocol install events\r
 \r
-    FreePool (ConfigResp);\r
-  } else {\r
-    //\r
-    // Prepare <ConfigResp>\r
-    //\r
-    TmpSize = StrLen (ResultsData);\r
-    BufferSize = (TmpSize + StrLen (Storage->BrowserStorage->ConfigHdr) + 2) * sizeof (CHAR16);\r
-    ConfigResp = AllocateZeroPool (BufferSize);\r
-    ASSERT (ConfigResp != NULL);\r
+  @param Event           the event that is signaled.\r
+  @param Context         not used here.\r
 \r
-    StrCpy (ConfigResp, Storage->BrowserStorage->ConfigHdr);\r
-    StrCat (ConfigResp, L"&");\r
-    StrCat (ConfigResp, ResultsData);\r
+**/\r
+VOID\r
+EFIAPI\r
+FormDisplayCallback (\r
+  IN EFI_EVENT    Event,\r
+  IN VOID         *Context\r
+  )\r
+{\r
+  EFI_STATUS                  Status;\r
 \r
-    //\r
-    // Update Browser uncommited data\r
-    //\r
-    Status = ConfigRespToStorage (Storage->BrowserStorage, ConfigResp);\r
-    if (EFI_ERROR (Status)) {\r
-      return Status;\r
-    }\r
+  if (mFormDisplay != NULL) {\r
+    return;\r
   }\r
 \r
-  return EFI_SUCCESS;\r
+  Status = gBS->LocateProtocol (\r
+                  &gEdkiiFormDisplayEngineProtocolGuid,\r
+                  NULL,\r
+                  (VOID **) &mFormDisplay\r
+                  );\r
 }\r
 \r
 /**\r
@@ -582,8 +867,7 @@ InitializeSetup (
   )\r
 {\r
   EFI_STATUS                  Status;\r
-  EFI_INPUT_KEY               DefaultHotKey;\r
-  EFI_STRING                  HelpString;\r
+  VOID                        *Registration;\r
 \r
   //\r
   // Locate required Hii relative protocols\r
@@ -595,13 +879,6 @@ InitializeSetup (
                   );\r
   ASSERT_EFI_ERROR (Status);\r
 \r
-  Status = gBS->LocateProtocol (\r
-                  &gEfiHiiStringProtocolGuid,\r
-                  NULL,\r
-                  (VOID **) &mHiiString\r
-                  );\r
-  ASSERT_EFI_ERROR (Status);\r
-\r
   Status = gBS->LocateProtocol (\r
                   &gEfiHiiConfigRoutingProtocolGuid,\r
                   NULL,\r
@@ -615,30 +892,6 @@ InitializeSetup (
                   (VOID **) &mPathFromText\r
                   );\r
 \r
-  //\r
-  // Publish our HII data\r
-  //\r
-  gHiiHandle = HiiAddPackages (\r
-                 &gSetupBrowserGuid,\r
-                 ImageHandle,\r
-                 SetupBrowserStrings,\r
-                 NULL\r
-                 );\r
-  ASSERT (gHiiHandle != NULL);\r
-\r
-  //\r
-  // Initialize Driver private data\r
-  //\r
-  gBannerData = AllocateZeroPool (sizeof (BANNER_DATA));\r
-  ASSERT (gBannerData != NULL);\r
-  \r
-  //\r
-  // Initialize generic help strings.\r
-  //\r
-  gSaveFailed    = GetToken (STRING_TOKEN (SAVE_FAILED), gHiiHandle);\r
-  gDiscardFailed = GetToken (STRING_TOKEN (DISCARD_FAILED), gHiiHandle);\r
-  gDefaultFailed = GetToken (STRING_TOKEN (DEFAULT_FAILED), gHiiHandle);\r
-\r
   //\r
   // Install FormBrowser2 protocol\r
   //\r
@@ -652,25 +905,19 @@ InitializeSetup (
   ASSERT_EFI_ERROR (Status);\r
 \r
   //\r
-  // Install default HotKey F10 for Save\r
-  //\r
-  DefaultHotKey.UnicodeChar = CHAR_NULL;\r
-  HelpString             = GetToken (STRING_TOKEN (FUNCTION_TEN_STRING), gHiiHandle);\r
-  DefaultHotKey.ScanCode = SCAN_F10;\r
-  RegisterHotKey (&DefaultHotKey, BROWSER_ACTION_SUBMIT, 0, HelpString);\r
-  FreePool (HelpString);\r
-  //\r
-  // Install default HotKey F9 for Reset To Defaults\r
-  //\r
-  DefaultHotKey.ScanCode    = SCAN_F9;\r
-  HelpString                = GetToken (STRING_TOKEN (FUNCTION_NINE_STRING), gHiiHandle);\r
-  RegisterHotKey (&DefaultHotKey, BROWSER_ACTION_DEFAULT, EFI_HII_DEFAULT_CLASS_STANDARD, HelpString);\r
-  FreePool (HelpString);\r
-\r
-  //\r
-  // Install FormBrowserEx protocol\r
+  // Install FormBrowserEx2 protocol\r
   //\r
+  InitializeListHead (&mPrivateData.FormBrowserEx2.FormViewHistoryHead);\r
+  InitializeListHead (&mPrivateData.FormBrowserEx2.OverrideQestListHead);\r
   mPrivateData.Handle = NULL;\r
+  Status = gBS->InstallProtocolInterface (\r
+                  &mPrivateData.Handle,\r
+                  &gEdkiiFormBrowserEx2ProtocolGuid,\r
+                  EFI_NATIVE_INTERFACE,\r
+                  &mPrivateData.FormBrowserEx2\r
+                  );\r
+  ASSERT_EFI_ERROR (Status);\r
+  \r
   Status = gBS->InstallProtocolInterface (\r
                   &mPrivateData.Handle,\r
                   &gEfiFormBrowserExProtocolGuid,\r
@@ -679,7 +926,25 @@ InitializeSetup (
                   );\r
   ASSERT_EFI_ERROR (Status);\r
 \r
-  return Status;\r
+  InitializeDisplayFormData ();\r
+\r
+  Status = gBS->LocateProtocol (\r
+                  &gEdkiiFormDisplayEngineProtocolGuid,\r
+                  NULL,\r
+                  (VOID **) &mFormDisplay\r
+                  );\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    EfiCreateProtocolNotifyEvent (\r
+      &gEdkiiFormDisplayEngineProtocolGuid,\r
+      TPL_CALLBACK,\r
+      FormDisplayCallback,\r
+      NULL,\r
+      &Registration\r
+      );\r
+  }\r
+  \r
+  return EFI_SUCCESS;\r
 }\r
 \r
 \r
@@ -815,58 +1080,6 @@ NewStringCat (
   *Dest = NewString;\r
 }\r
 \r
-\r
-/**\r
-  Synchronize or restore Storage's Edit copy and Shadow copy.\r
-\r
-  @param  Storage          The Storage to be synchronized.\r
-  @param  SyncOrRestore    Sync the buffer to editbuffer or Restore  the \r
-                           editbuffer to buffer\r
-                           if TRUE, copy the editbuffer to the buffer.\r
-                           if FALSE, copy the buffer to the editbuffer.\r
-\r
-**/\r
-VOID\r
-SynchronizeStorage (\r
-  IN BROWSER_STORAGE         *Storage,\r
-  IN BOOLEAN                 SyncOrRestore\r
-  )\r
-{\r
-  LIST_ENTRY              *Link;\r
-  NAME_VALUE_NODE         *Node;\r
-\r
-  switch (Storage->Type) {\r
-  case EFI_HII_VARSTORE_BUFFER:\r
-  case EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER:\r
-    if (SyncOrRestore) {\r
-      CopyMem (Storage->Buffer, Storage->EditBuffer, Storage->Size);\r
-    } else {\r
-      CopyMem (Storage->EditBuffer, Storage->Buffer, Storage->Size);\r
-    }\r
-    break;\r
-\r
-  case EFI_HII_VARSTORE_NAME_VALUE:\r
-    Link = GetFirstNode (&Storage->NameValueListHead);\r
-    while (!IsNull (&Storage->NameValueListHead, Link)) {\r
-      Node = NAME_VALUE_NODE_FROM_LINK (Link);\r
-\r
-      if (SyncOrRestore) {\r
-        NewStringCpy (&Node->Value, Node->EditValue);\r
-      } else {\r
-        NewStringCpy (&Node->EditValue, Node->Value);\r
-      }\r
-\r
-      Link = GetNextNode (&Storage->NameValueListHead, Link);\r
-    }\r
-    break;\r
-\r
-  case EFI_HII_VARSTORE_EFI_VARIABLE:\r
-  default:\r
-    break;\r
-  }\r
-}\r
-\r
-\r
 /**\r
   Get Value for given Name from a NameValue Storage.\r
 \r
@@ -923,6 +1136,7 @@ GetValueByName (
   @param  Name                   The Name.\r
   @param  Value                  The Value to set.\r
   @param  SetValueTo             Whether update editValue or Value.\r
+  @param  ReturnNode             The node use the input name.\r
 \r
   @retval EFI_SUCCESS            Value found for given Name.\r
   @retval EFI_NOT_FOUND          No such Name found in NameValue storage.\r
@@ -930,10 +1144,11 @@ GetValueByName (
 **/\r
 EFI_STATUS\r
 SetValueByName (\r
-  IN BROWSER_STORAGE         *Storage,\r
-  IN CHAR16                  *Name,\r
-  IN CHAR16                  *Value,\r
-  IN GET_SET_QUESTION_VALUE_WITH SetValueTo\r
+  IN  BROWSER_STORAGE             *Storage,\r
+  IN  CHAR16                      *Name,\r
+  IN  CHAR16                      *Value,\r
+  IN  GET_SET_QUESTION_VALUE_WITH SetValueTo,\r
+  OUT NAME_VALUE_NODE             **ReturnNode\r
   )\r
 {\r
   LIST_ENTRY              *Link;\r
@@ -964,6 +1179,11 @@ SetValueByName (
       } else {\r
         Node->Value = Buffer;\r
       }\r
+\r
+      if (ReturnNode != NULL) {\r
+        *ReturnNode = Node;\r
+      }\r
+\r
       return EFI_SUCCESS;\r
     }\r
 \r
@@ -980,6 +1200,7 @@ SetValueByName (
   @param  Storage                The Storage to be conveted.\r
   @param  ConfigResp             The returned <ConfigResp>.\r
   @param  ConfigRequest          The ConfigRequest string.\r
+  @param  GetEditBuf             Get the data from editbuffer or buffer.\r
 \r
   @retval EFI_SUCCESS            Convert success.\r
   @retval EFI_INVALID_PARAMETER  Incorrect storage type.\r
@@ -989,23 +1210,26 @@ EFI_STATUS
 StorageToConfigResp (\r
   IN BROWSER_STORAGE         *Storage,\r
   IN CHAR16                  **ConfigResp,\r
-  IN CHAR16                  *ConfigRequest\r
+  IN CHAR16                  *ConfigRequest,\r
+  IN BOOLEAN                 GetEditBuf\r
   )\r
 {\r
   EFI_STATUS              Status;\r
   EFI_STRING              Progress;\r
   LIST_ENTRY              *Link;\r
   NAME_VALUE_NODE         *Node;\r
+  UINT8                   *SourceBuf;\r
 \r
   Status = EFI_SUCCESS;\r
 \r
   switch (Storage->Type) {\r
   case EFI_HII_VARSTORE_BUFFER:\r
   case EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER:\r
+    SourceBuf = GetEditBuf ? Storage->EditBuffer : Storage->Buffer;\r
     Status = mHiiConfigRouting->BlockToConfig (\r
                                   mHiiConfigRouting,\r
                                   ConfigRequest,\r
-                                  Storage->EditBuffer,\r
+                                  SourceBuf,\r
                                   Storage->Size,\r
                                   ConfigResp,\r
                                   &Progress\r
@@ -1024,7 +1248,11 @@ StorageToConfigResp (
         NewStringCat (ConfigResp, L"&");\r
         NewStringCat (ConfigResp, Node->Name);\r
         NewStringCat (ConfigResp, L"=");\r
-        NewStringCat (ConfigResp, Node->EditValue);\r
+        if (GetEditBuf) {\r
+          NewStringCat (ConfigResp, Node->EditValue);\r
+        } else {\r
+          NewStringCat (ConfigResp, Node->Value);\r
+        }\r
       }\r
       Link = GetNextNode (&Storage->NameValueListHead, Link);\r
     }\r
@@ -1105,7 +1333,7 @@ ConfigRespToStorage (
       if (StrPtr != NULL) {\r
         *StrPtr = 0;\r
       }\r
-      SetValueByName (Storage, Name, Value, GetSetValueWithEditBuffer);\r
+      SetValueByName (Storage, Name, Value, GetSetValueWithEditBuffer, NULL);\r
     }\r
     break;\r
 \r
@@ -1158,7 +1386,6 @@ GetQuestionValue (
   BOOLEAN             IsString;\r
   CHAR16              TemStr[5];\r
   UINT8               DigitUint8;\r
-  UINT8               *TemBuffer;\r
 \r
   Status = EFI_SUCCESS;\r
   Value  = NULL;\r
@@ -1169,14 +1396,7 @@ GetQuestionValue (
   }\r
 \r
   //\r
-  // Statement don't have storage, skip them\r
-  //\r
-  if (Question->QuestionId == 0) {\r
-    return Status;\r
-  }\r
-\r
-  //\r
-  // Question value is provided by an Expression, evaluate it\r
+  // Question value is provided by an Expression, evaluate it\r
   //\r
   if (Question->ValueExpression != NULL) {\r
     Status = EvaluateExpression (FormSet, Form, Question->ValueExpression);\r
@@ -1382,147 +1602,118 @@ GetQuestionValue (
       FreePool (Value);\r
     }\r
   } else {\r
-    if (Storage->Type == EFI_HII_VARSTORE_BUFFER || Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
-      //\r
-      // Request current settings from Configuration Driver\r
-      //\r
-      if (FormSet->ConfigAccess == NULL) {\r
-        return EFI_NOT_FOUND;\r
-      }\r
-\r
-      //\r
-      // <ConfigRequest> ::= <ConfigHdr> + <BlockName> ||\r
-      //                   <ConfigHdr> + "&" + <VariableName>\r
-      //\r
-      if (IsBufferStorage) {\r
-        Length = StrLen (Storage->ConfigHdr);\r
-        Length += StrLen (Question->BlockName);\r
-      } else {\r
-        Length = StrLen (Storage->ConfigHdr);\r
-        Length += StrLen (Question->VariableName) + 1;\r
-      }\r
-      ConfigRequest = AllocateZeroPool ((Length + 1) * sizeof (CHAR16));\r
-      ASSERT (ConfigRequest != NULL);\r
+    //\r
+    // <ConfigRequest> ::= <ConfigHdr> + <BlockName> ||\r
+    //                   <ConfigHdr> + "&" + <VariableName>\r
+    //\r
+    if (IsBufferStorage) {\r
+      Length = StrLen (Storage->ConfigHdr);\r
+      Length += StrLen (Question->BlockName);\r
+    } else {\r
+      Length = StrLen (Storage->ConfigHdr);\r
+      Length += StrLen (Question->VariableName) + 1;\r
+    }\r
+    ConfigRequest = AllocateZeroPool ((Length + 1) * sizeof (CHAR16));\r
+    ASSERT (ConfigRequest != NULL);\r
 \r
-      StrCpy (ConfigRequest, Storage->ConfigHdr);\r
-      if (IsBufferStorage) {\r
-        StrCat (ConfigRequest, Question->BlockName);\r
-      } else {\r
-        StrCat (ConfigRequest, L"&");\r
-        StrCat (ConfigRequest, Question->VariableName);\r
-      }\r
+    StrCpy (ConfigRequest, Storage->ConfigHdr);\r
+    if (IsBufferStorage) {\r
+      StrCat (ConfigRequest, Question->BlockName);\r
+    } else {\r
+      StrCat (ConfigRequest, L"&");\r
+      StrCat (ConfigRequest, Question->VariableName);\r
+    }\r
 \r
-      Status = FormSet->ConfigAccess->ExtractConfig (\r
-                                        FormSet->ConfigAccess,\r
-                                        ConfigRequest,\r
-                                        &Progress,\r
-                                        &Result\r
-                                        );\r
-      FreePool (ConfigRequest);\r
-      if (EFI_ERROR (Status)) {\r
-        return Status;\r
-      }\r
+    //\r
+    // Request current settings from Configuration Driver\r
+    //\r
+    Status = mHiiConfigRouting->ExtractConfig (\r
+                                      mHiiConfigRouting,\r
+                                      ConfigRequest,\r
+                                      &Progress,\r
+                                      &Result\r
+                                      );\r
+    FreePool (ConfigRequest);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
 \r
-      //\r
-      // Skip <ConfigRequest>\r
-      //\r
-      if (IsBufferStorage) {\r
-        Value = StrStr (Result, L"&VALUE");\r
-        if (Value == NULL) {\r
-          FreePool (Result);\r
-          return EFI_NOT_FOUND;\r
-        }\r
-        //\r
-        // Skip "&VALUE"\r
-        //\r
-        Value = Value + 6;\r
-      } else {\r
-        Value = Result + Length;\r
-      }\r
-      if (*Value != '=') {\r
+    //\r
+    // Skip <ConfigRequest>\r
+    //\r
+    if (IsBufferStorage) {\r
+      Value = StrStr (Result, L"&VALUE");\r
+      if (Value == NULL) {\r
         FreePool (Result);\r
         return EFI_NOT_FOUND;\r
       }\r
       //\r
-      // Skip '=', point to value\r
+      // Skip "&VALUE"\r
       //\r
-      Value = Value + 1;\r
+      Value = Value + 6;\r
+    } else {\r
+      Value = Result + Length;\r
+    }\r
+    if (*Value != '=') {\r
+      FreePool (Result);\r
+      return EFI_NOT_FOUND;\r
+    }\r
+    //\r
+    // Skip '=', point to value\r
+    //\r
+    Value = Value + 1;\r
+\r
+    //\r
+    // Suppress <AltResp> if any\r
+    //\r
+    StringPtr = Value;\r
+    while (*StringPtr != L'\0' && *StringPtr != L'&') {\r
+      StringPtr++;\r
+    }\r
+    *StringPtr = L'\0';\r
 \r
+    LengthStr = StrLen (Value);\r
+    Status    = EFI_SUCCESS;\r
+    if (!IsBufferStorage && IsString) {\r
       //\r
-      // Suppress <AltResp> if any\r
+      // Convert Config String to Unicode String, e.g "0041004200430044" => "ABCD"\r
+      // Add string tail char L'\0' into Length\r
       //\r
-      StringPtr = Value;\r
-      while (*StringPtr != L'\0' && *StringPtr != L'&') {\r
-        StringPtr++;\r
-      }\r
-      *StringPtr = L'\0';\r
-\r
-      LengthStr = StrLen (Value);\r
-      Status    = EFI_SUCCESS;\r
-      if (!IsBufferStorage && IsString) {\r
+      Length    = StorageWidth + sizeof (CHAR16);\r
+      if (Length < ((LengthStr / 4 + 1) * 2)) {\r
+        Status = EFI_BUFFER_TOO_SMALL;\r
+      } else {\r
+        StringPtr = (CHAR16 *) Dst;\r
+        ZeroMem (TemStr, sizeof (TemStr));\r
+        for (Index = 0; Index < LengthStr; Index += 4) {\r
+          StrnCpy (TemStr, Value + Index, 4);\r
+          StringPtr[Index/4] = (CHAR16) StrHexToUint64 (TemStr);\r
+        }\r
         //\r
-        // Convert Config String to Unicode String, e.g "0041004200430044" => "ABCD"\r
-        // Add string tail char L'\0' into Length\r
+        // Add tailing L'\0' character\r
         //\r
-        Length    = StorageWidth + sizeof (CHAR16);\r
-        if (Length < ((LengthStr / 4 + 1) * 2)) {\r
-          Status = EFI_BUFFER_TOO_SMALL;\r
-        } else {\r
-          StringPtr = (CHAR16 *) Dst;\r
-          ZeroMem (TemStr, sizeof (TemStr));\r
-          for (Index = 0; Index < LengthStr; Index += 4) {\r
-            StrnCpy (TemStr, Value + Index, 4);\r
-            StringPtr[Index/4] = (CHAR16) StrHexToUint64 (TemStr);\r
-          }\r
-          //\r
-          // Add tailing L'\0' character\r
-          //\r
-          StringPtr[Index/4] = L'\0';\r
-        }\r
+        StringPtr[Index/4] = L'\0';\r
+      }\r
+    } else {\r
+      if (StorageWidth < ((LengthStr + 1) / 2)) {\r
+        Status = EFI_BUFFER_TOO_SMALL;\r
       } else {\r
-        if (StorageWidth < ((LengthStr + 1) / 2)) {\r
-          Status = EFI_BUFFER_TOO_SMALL;\r
-        } else {\r
-          ZeroMem (TemStr, sizeof (TemStr));\r
-          for (Index = 0; Index < LengthStr; Index ++) {\r
-            TemStr[0] = Value[LengthStr - Index - 1];\r
-            DigitUint8 = (UINT8) StrHexToUint64 (TemStr);\r
-            if ((Index & 1) == 0) {\r
-              Dst [Index/2] = DigitUint8;\r
-            } else {\r
-              Dst [Index/2] = (UINT8) ((DigitUint8 << 4) + Dst [Index/2]);\r
-            }\r
+        ZeroMem (TemStr, sizeof (TemStr));\r
+        for (Index = 0; Index < LengthStr; Index ++) {\r
+          TemStr[0] = Value[LengthStr - Index - 1];\r
+          DigitUint8 = (UINT8) StrHexToUint64 (TemStr);\r
+          if ((Index & 1) == 0) {\r
+            Dst [Index/2] = DigitUint8;\r
+          } else {\r
+            Dst [Index/2] = (UINT8) ((DigitUint8 << 4) + Dst [Index/2]);\r
           }\r
         }\r
       }\r
+    }\r
 \r
-      if (EFI_ERROR (Status)) {\r
-        FreePool (Result);\r
-        return Status;\r
-      }\r
-    } else if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
-      TemBuffer = NULL;\r
-      TemBuffer = AllocateZeroPool (Storage->Size);\r
-      if (TemBuffer == NULL) {\r
-        Status = EFI_OUT_OF_RESOURCES;\r
-        return Status;\r
-      }\r
-      Length = Storage->Size;\r
-      Status = gRT->GetVariable (\r
-                       Storage->Name,\r
-                       &Storage->Guid,\r
-                       NULL,\r
-                       &Length,\r
-                       TemBuffer\r
-                       );\r
-      if (EFI_ERROR (Status)) {\r
-        FreePool (TemBuffer);\r
-        return Status;\r
-      }\r
-\r
-      CopyMem (Dst, TemBuffer + Question->VarStoreInfo.VarOffset, StorageWidth);\r
-\r
-      FreePool (TemBuffer);\r
+    if (EFI_ERROR (Status)) {\r
+      FreePool (Result);\r
+      return Status;\r
     }\r
 \r
     //\r
@@ -1531,7 +1722,7 @@ GetQuestionValue (
     if (IsBufferStorage) {\r
       CopyMem (Storage->EditBuffer + Question->VarStoreInfo.VarOffset, Dst, StorageWidth);\r
     } else {\r
-      SetValueByName (Storage, Question->VariableName, Value, GetSetValueWithEditBuffer);\r
+      SetValueByName (Storage, Question->VariableName, Value, GetSetValueWithEditBuffer, NULL);\r
     }\r
 \r
     if (Result != NULL) {\r
@@ -1581,20 +1772,15 @@ SetQuestionValue (
   CHAR16              *TemName;\r
   CHAR16              *TemString;\r
   UINTN               Index;\r
+  NAME_VALUE_NODE     *Node;\r
 \r
   Status = EFI_SUCCESS;\r
+  Node   = NULL;\r
 \r
   if (SetValueTo >= GetSetValueWithMax) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  //\r
-  // Statement don't have storage, skip them\r
-  //\r
-  if (Question->QuestionId == 0) {\r
-    return Status;\r
-  }\r
-\r
   //\r
   // If Question value is provided by an Expression, then it is read only\r
   //\r
@@ -1748,115 +1934,85 @@ SetQuestionValue (
         }\r
       }\r
 \r
-      Status = SetValueByName (Storage, Question->VariableName, Value, SetValueTo);\r
+      Status = SetValueByName (Storage, Question->VariableName, Value, SetValueTo, &Node);\r
       FreePool (Value);\r
+      if (EFI_ERROR (Status)) {\r
+        return Status;\r
+      }\r
     }\r
   } else if (SetValueTo == GetSetValueWithHiiDriver) {\r
-    if (Storage->Type == EFI_HII_VARSTORE_BUFFER || Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
-      //\r
-      // <ConfigResp> ::= <ConfigHdr> + <BlockName> + "&VALUE=" + "<HexCh>StorageWidth * 2" ||\r
-      //                <ConfigHdr> + "&" + <VariableName> + "=" + "<string>"\r
-      //\r
-      if (IsBufferStorage) {\r
-        Length = StrLen (Question->BlockName) + 7;\r
-      } else {\r
-        Length = StrLen (Question->VariableName) + 2;\r
-      }\r
-      if (!IsBufferStorage && IsString) {\r
-        Length += (StrLen ((CHAR16 *) Src) * 4);\r
-      } else {\r
-        Length += (StorageWidth * 2);\r
-      }\r
-      ConfigResp = AllocateZeroPool ((StrLen (Storage->ConfigHdr) + Length + 1) * sizeof (CHAR16));\r
-      ASSERT (ConfigResp != NULL);\r
-\r
-      StrCpy (ConfigResp, Storage->ConfigHdr);\r
-      if (IsBufferStorage) {\r
-        StrCat (ConfigResp, Question->BlockName);\r
-        StrCat (ConfigResp, L"&VALUE=");\r
-      } else {\r
-        StrCat (ConfigResp, L"&");\r
-        StrCat (ConfigResp, Question->VariableName);\r
-        StrCat (ConfigResp, L"=");\r
-      }\r
+    //\r
+    // <ConfigResp> ::= <ConfigHdr> + <BlockName> + "&VALUE=" + "<HexCh>StorageWidth * 2" ||\r
+    //                <ConfigHdr> + "&" + <VariableName> + "=" + "<string>"\r
+    //\r
+    if (IsBufferStorage) {\r
+      Length = StrLen (Question->BlockName) + 7;\r
+    } else {\r
+      Length = StrLen (Question->VariableName) + 2;\r
+    }\r
+    if (!IsBufferStorage && IsString) {\r
+      Length += (StrLen ((CHAR16 *) Src) * 4);\r
+    } else {\r
+      Length += (StorageWidth * 2);\r
+    }\r
+    ConfigResp = AllocateZeroPool ((StrLen (Storage->ConfigHdr) + Length + 1) * sizeof (CHAR16));\r
+    ASSERT (ConfigResp != NULL);\r
 \r
-      Value = ConfigResp + StrLen (ConfigResp);\r
+    StrCpy (ConfigResp, Storage->ConfigHdr);\r
+    if (IsBufferStorage) {\r
+      StrCat (ConfigResp, Question->BlockName);\r
+      StrCat (ConfigResp, L"&VALUE=");\r
+    } else {\r
+      StrCat (ConfigResp, L"&");\r
+      StrCat (ConfigResp, Question->VariableName);\r
+      StrCat (ConfigResp, L"=");\r
+    }\r
 \r
-      if (!IsBufferStorage && IsString) {\r
-        //\r
-        // Convert Unicode String to Config String, e.g. "ABCD" => "0041004200430044"\r
-        //\r
-        TemName = (CHAR16 *) Src;\r
-        TemString = Value;\r
-        for (; *TemName != L'\0'; TemName++) {\r
-          TemString += UnicodeValueToString (TemString, PREFIX_ZERO | RADIX_HEX, *TemName, 4);\r
-        }\r
-      } else {\r
-        //\r
-        // Convert Buffer to Hex String\r
-        //\r
-        TemBuffer = Src + StorageWidth - 1;\r
-        TemString = Value;\r
-        for (Index = 0; Index < StorageWidth; Index ++, TemBuffer --) {\r
-          TemString += UnicodeValueToString (TemString, PREFIX_ZERO | RADIX_HEX, *TemBuffer, 2);\r
-        }\r
-      }\r
+    Value = ConfigResp + StrLen (ConfigResp);\r
 \r
+    if (!IsBufferStorage && IsString) {\r
       //\r
-      // Convert to lower char.\r
+      // Convert Unicode String to Config String, e.g. "ABCD" => "0041004200430044"\r
       //\r
-      for (TemString = Value; *Value != L'\0'; Value++) {\r
-        if (*Value >= L'A' && *Value <= L'Z') {\r
-          *Value = (CHAR16) (*Value - L'A' + L'a');\r
-        }\r
+      TemName = (CHAR16 *) Src;\r
+      TemString = Value;\r
+      for (; *TemName != L'\0'; TemName++) {\r
+        TemString += UnicodeValueToString (TemString, PREFIX_ZERO | RADIX_HEX, *TemName, 4);\r
       }\r
-\r
+    } else {\r
       //\r
-      // Submit Question Value to Configuration Driver\r
+      // Convert Buffer to Hex String\r
       //\r
-      if (FormSet->ConfigAccess != NULL) {\r
-        Status = FormSet->ConfigAccess->RouteConfig (\r
-                                          FormSet->ConfigAccess,\r
-                                          ConfigResp,\r
-                                          &Progress\r
-                                          );\r
-        if (EFI_ERROR (Status)) {\r
-          FreePool (ConfigResp);\r
-          return Status;\r
-        }\r
-      }\r
-      FreePool (ConfigResp);\r
-      \r
-    } else if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
-      TemBuffer = NULL;\r
-      TemBuffer = AllocateZeroPool(Storage->Size);\r
-      if (TemBuffer == NULL) {\r
-        Status = EFI_OUT_OF_RESOURCES;\r
-        return Status;\r
+      TemBuffer = Src + StorageWidth - 1;\r
+      TemString = Value;\r
+      for (Index = 0; Index < StorageWidth; Index ++, TemBuffer --) {\r
+        TemString += UnicodeValueToString (TemString, PREFIX_ZERO | RADIX_HEX, *TemBuffer, 2);\r
       }\r
-      Length = Storage->Size;\r
-      Status = gRT->GetVariable (\r
-                       Storage->Name,\r
-                       &Storage->Guid,\r
-                       NULL,\r
-                       &Length,\r
-                       TemBuffer\r
-                       );\r
-\r
-      CopyMem (TemBuffer + Question->VarStoreInfo.VarOffset, Src, StorageWidth);\r
-      \r
-      Status = gRT->SetVariable (\r
-                       Storage->Name,\r
-                       &Storage->Guid,\r
-                       Storage->Attributes,\r
-                       Storage->Size,\r
-                       TemBuffer\r
-                       );\r
-      FreePool (TemBuffer);\r
-      if (EFI_ERROR (Status)){\r
-        return Status;\r
+    }\r
+\r
+    //\r
+    // Convert to lower char.\r
+    //\r
+    for (TemString = Value; *Value != L'\0'; Value++) {\r
+      if (*Value >= L'A' && *Value <= L'Z') {\r
+        *Value = (CHAR16) (*Value - L'A' + L'a');\r
       }\r
     }\r
+\r
+    //\r
+    // Submit Question Value to Configuration Driver\r
+    //\r
+    Status = mHiiConfigRouting->RouteConfig (\r
+                                      mHiiConfigRouting,\r
+                                      ConfigResp,\r
+                                      &Progress\r
+                                      );\r
+    if (EFI_ERROR (Status)) {\r
+      FreePool (ConfigResp);\r
+      return Status;\r
+    }\r
+    FreePool (ConfigResp);\r
+    \r
     //\r
     // Sync storage, from editbuffer to buffer.\r
     //\r
@@ -1868,12 +2024,12 @@ SetQuestionValue (
 \r
 \r
 /**\r
-  Perform inconsistent check for a Form.\r
+  Perform nosubmitif check for a Form.\r
 \r
   @param  FormSet                FormSet data structure.\r
   @param  Form                   Form data structure.\r
   @param  Question               The Question to be validated.\r
-  @param  Type                   Validation type: InConsistent or NoSubmit\r
+  @param  Type                   Validation type: NoSubmit\r
 \r
   @retval EFI_SUCCESS            Form validation pass.\r
   @retval other                  Form validation failed.\r
@@ -1890,15 +2046,28 @@ ValidateQuestion (
   EFI_STATUS              Status;\r
   LIST_ENTRY              *Link;\r
   LIST_ENTRY              *ListHead;\r
-  EFI_STRING              PopUp;\r
-  EFI_INPUT_KEY           Key;\r
   FORM_EXPRESSION         *Expression;\r
+  UINT32                  BrowserStatus;\r
+  CHAR16                  *ErrorStr;\r
 \r
-  if (Type == EFI_HII_EXPRESSION_INCONSISTENT_IF) {\r
+  BrowserStatus = BROWSER_SUCCESS;\r
+  ErrorStr      = NULL;\r
+\r
+  switch (Type) {\r
+  case EFI_HII_EXPRESSION_INCONSISTENT_IF:\r
     ListHead = &Question->InconsistentListHead;\r
-  } else if (Type == EFI_HII_EXPRESSION_NO_SUBMIT_IF) {\r
+    break;\r
+\r
+  case EFI_HII_EXPRESSION_WARNING_IF:\r
+    ListHead = &Question->WarningListHead;\r
+    break;\r
+\r
+  case EFI_HII_EXPRESSION_NO_SUBMIT_IF:\r
     ListHead = &Question->NoSubmitListHead;\r
-  } else {\r
+    break;\r
+\r
+  default:\r
+    ASSERT (FALSE);\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
@@ -1914,19 +2083,49 @@ ValidateQuestion (
       return Status;\r
     }\r
 \r
-    if ((Expression->Result.Type == EFI_IFR_TYPE_BOOLEAN) && Expression->Result.Value.b) {\r
-      //\r
-      // Condition meet, show up error message\r
-      //\r
-      if (Expression->Error != 0) {\r
-        PopUp = GetToken (Expression->Error, FormSet->HiiHandle);\r
-        do {\r
-          CreateDialog (4, TRUE, 0, NULL, &Key, gEmptyString, PopUp, gPressEnter, gEmptyString);\r
-        } while (Key.UnicodeChar != CHAR_CARRIAGE_RETURN);\r
-        FreePool (PopUp);\r
+    if (IsTrue (&Expression->Result)) {\r
+      switch (Type) {\r
+      case EFI_HII_EXPRESSION_INCONSISTENT_IF:\r
+        BrowserStatus = BROWSER_INCONSISTENT_IF;\r
+        break;\r
+\r
+      case EFI_HII_EXPRESSION_WARNING_IF:\r
+        BrowserStatus = BROWSER_WARNING_IF;\r
+        break;\r
+\r
+      case EFI_HII_EXPRESSION_NO_SUBMIT_IF:\r
+        BrowserStatus = BROWSER_NO_SUBMIT_IF;\r
+        //\r
+        // This code only used to compatible with old display engine,\r
+        // New display engine will not use this field.\r
+        //\r
+        if (Expression->Error != 0) {\r
+          ErrorStr = GetToken (Expression->Error, FormSet->HiiHandle);\r
+        }\r
+        break;\r
+\r
+      default:\r
+        ASSERT (FALSE);\r
+        break;\r
       }\r
 \r
-      return EFI_NOT_READY;\r
+      if (!((Type == EFI_HII_EXPRESSION_NO_SUBMIT_IF) && mSystemSubmit)) {\r
+        //\r
+        // If in system submit process and for no_submit_if check, not popup this error message.\r
+        // Will process this fail again later in not system submit process.\r
+        //\r
+        PopupErrorMessage(BrowserStatus, FormSet->HiiHandle, Expression->OpCode, ErrorStr);\r
+      }\r
+\r
+      if (ErrorStr != NULL) {\r
+        FreePool (ErrorStr);\r
+      }\r
+\r
+      if (Type == EFI_HII_EXPRESSION_WARNING_IF) {\r
+        return EFI_SUCCESS;\r
+      } else {\r
+        return EFI_NOT_READY;\r
+      }\r
     }\r
 \r
     Link = GetNextNode (ListHead, Link);\r
@@ -1935,12 +2134,57 @@ ValidateQuestion (
   return EFI_SUCCESS;\r
 }\r
 \r
+/**\r
+  Perform question check. \r
+  \r
+  If one question has more than one check, process form high priority to low. \r
+  Only one error info will be popup.\r
+\r
+  @param  FormSet                FormSet data structure.\r
+  @param  Form                   Form data structure.\r
+  @param  Question               The Question to be validated.\r
+\r
+  @retval EFI_SUCCESS            Form validation pass.\r
+  @retval other                  Form validation failed.\r
+\r
+**/\r
+EFI_STATUS\r
+ValueChangedValidation (\r
+  IN  FORM_BROWSER_FORMSET            *FormSet,\r
+  IN  FORM_BROWSER_FORM               *Form,\r
+  IN  FORM_BROWSER_STATEMENT          *Question\r
+  )\r
+{\r
+  EFI_STATUS   Status;\r
+\r
+  Status = EFI_SUCCESS;\r
+\r
+  //\r
+  // Do the inconsistentif check.\r
+  //\r
+  if (!IsListEmpty (&Question->InconsistentListHead)) {\r
+    Status = ValidateQuestion (FormSet, Form, Question, EFI_HII_EXPRESSION_INCONSISTENT_IF);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+  }\r
+\r
+  //\r
+  // Do the warningif check.\r
+  //\r
+  if (!IsListEmpty (&Question->WarningListHead)) {\r
+    Status = ValidateQuestion (FormSet, Form, Question, EFI_HII_EXPRESSION_WARNING_IF);\r
+  }\r
+\r
+  return Status;\r
+}\r
 \r
 /**\r
   Perform NoSubmit check for each Form in FormSet.\r
 \r
   @param  FormSet                FormSet data structure.\r
   @param  CurrentForm            Current input form data structure.\r
+  @param  Statement              The statement for this check.\r
 \r
   @retval EFI_SUCCESS            Form validation pass.\r
   @retval other                  Form validation failed.\r
@@ -1948,8 +2192,9 @@ ValidateQuestion (
 **/\r
 EFI_STATUS\r
 NoSubmitCheck (\r
-  IN  FORM_BROWSER_FORMSET            *FormSet,\r
-  IN  FORM_BROWSER_FORM               *CurrentForm\r
+  IN      FORM_BROWSER_FORMSET            *FormSet,\r
+  IN OUT  FORM_BROWSER_FORM               **CurrentForm,\r
+  OUT     FORM_BROWSER_STATEMENT          **Statement\r
   )\r
 {\r
   EFI_STATUS              Status;\r
@@ -1963,16 +2208,21 @@ NoSubmitCheck (
     Form = FORM_BROWSER_FORM_FROM_LINK (LinkForm);\r
     LinkForm = GetNextNode (&FormSet->FormListHead, LinkForm);\r
 \r
-    if (CurrentForm != NULL && CurrentForm != Form) {\r
+    if (*CurrentForm != NULL && *CurrentForm != Form) {\r
       continue;\r
     }\r
 \r
     Link = GetFirstNode (&Form->StatementListHead);\r
     while (!IsNull (&Form->StatementListHead, Link)) {\r
       Question = FORM_BROWSER_STATEMENT_FROM_LINK (Link);\r
-\r
       Status = ValidateQuestion (FormSet, Form, Question, EFI_HII_EXPRESSION_NO_SUBMIT_IF);\r
       if (EFI_ERROR (Status)) {\r
+        if (*CurrentForm == NULL) {\r
+          *CurrentForm = Form;\r
+        }\r
+        if (Statement != NULL) {\r
+          *Statement = Question;\r
+        }\r
         return Status;\r
       }\r
 \r
@@ -1986,8 +2236,8 @@ NoSubmitCheck (
 /**\r
   Fill storage's edit copy with settings requested from Configuration Driver.\r
 \r
-  @param  FormSet                FormSet data structure.\r
-  @param  ConfigInfo             The config info related to this form.\r
+  @param  Storage                The storage which need to sync.\r
+  @param  ConfigRequest          The config request string which used to sync storage.\r
   @param  SyncOrRestore          Sync the buffer to editbuffer or Restore  the \r
                                  editbuffer to buffer\r
                                  if TRUE, copy the editbuffer to the buffer.\r
@@ -1997,10 +2247,10 @@ NoSubmitCheck (
 \r
 **/\r
 EFI_STATUS\r
-SynchronizeStorageForForm (\r
-  IN FORM_BROWSER_FORMSET        *FormSet,\r
-  IN FORM_BROWSER_CONFIG_REQUEST *ConfigInfo,\r
-  IN BOOLEAN                     SyncOrRestore\r
+SynchronizeStorage (\r
+  OUT BROWSER_STORAGE             *Storage,\r
+  IN  CHAR16                      *ConfigRequest,\r
+  IN  BOOLEAN                     SyncOrRestore\r
   )\r
 {\r
   EFI_STATUS              Status;\r
@@ -2014,57 +2264,52 @@ SynchronizeStorageForForm (
 \r
   Status = EFI_SUCCESS;\r
   Result = NULL;\r
-  if (FormSet->ConfigAccess == NULL && ConfigInfo->Storage->Type != EFI_HII_VARSTORE_NAME_VALUE) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
 \r
-  if (ConfigInfo->ElementCount == 0) {\r
-    //\r
-    // Skip if there is no RequestElement\r
-    //\r
-    return EFI_SUCCESS;\r
-  }\r
-\r
-  if (ConfigInfo->Storage->Type == EFI_HII_VARSTORE_BUFFER || \r
-      (ConfigInfo->Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
-    BufferSize = ConfigInfo->Storage->Size;\r
+  if (Storage->Type == EFI_HII_VARSTORE_BUFFER || \r
+      (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+    BufferSize = Storage->Size;\r
 \r
     if (SyncOrRestore) {\r
-      Src = ConfigInfo->Storage->EditBuffer;\r
-      Dst = ConfigInfo->Storage->Buffer;\r
+      Src = Storage->EditBuffer;\r
+      Dst = Storage->Buffer;\r
     } else {\r
-      Src = ConfigInfo->Storage->Buffer;\r
-      Dst = ConfigInfo->Storage->EditBuffer;\r
+      Src = Storage->Buffer;\r
+      Dst = Storage->EditBuffer;\r
     }\r
 \r
-    Status = mHiiConfigRouting->BlockToConfig(\r
-                                  mHiiConfigRouting,\r
-                                  ConfigInfo->ConfigRequest,\r
-                                  Src,\r
-                                  BufferSize,\r
-                                  &Result,\r
-                                  &Progress\r
-                                  );\r
-    if (EFI_ERROR (Status)) {\r
-      return Status;\r
-    }\r
+    if (ConfigRequest != NULL) {\r
+      Status = mHiiConfigRouting->BlockToConfig(\r
+                                    mHiiConfigRouting,\r
+                                    ConfigRequest,\r
+                                    Src,\r
+                                    BufferSize,\r
+                                    &Result,\r
+                                    &Progress\r
+                                    );\r
+      if (EFI_ERROR (Status)) {\r
+        return Status;\r
+      }\r
 \r
-    Status = mHiiConfigRouting->ConfigToBlock (\r
-                                  mHiiConfigRouting,\r
-                                  Result,\r
-                                  Dst,\r
-                                  &BufferSize,\r
-                                  &Progress\r
-                                  );\r
-    if (Result != NULL) {\r
-      FreePool (Result);\r
+      Status = mHiiConfigRouting->ConfigToBlock (\r
+                                    mHiiConfigRouting,\r
+                                    Result,\r
+                                    Dst,\r
+                                    &BufferSize,\r
+                                    &Progress\r
+                                    );\r
+      if (Result != NULL) {\r
+        FreePool (Result);\r
+      }\r
+    } else {\r
+      CopyMem (Dst, Src, BufferSize);\r
     }\r
-  } else if (ConfigInfo->Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
-    Link = GetFirstNode (&ConfigInfo->Storage->NameValueListHead);\r
-    while (!IsNull (&ConfigInfo->Storage->NameValueListHead, Link)) {\r
+  } else if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    Link = GetFirstNode (&Storage->NameValueListHead);\r
+    while (!IsNull (&Storage->NameValueListHead, Link)) {\r
       Node = NAME_VALUE_NODE_FROM_LINK (Link);\r
 \r
-      if (StrStr (ConfigInfo->ConfigRequest, Node->Name) != NULL) {\r
+      if ((ConfigRequest != NULL && StrStr (ConfigRequest, Node->Name) != NULL) ||\r
+          (ConfigRequest == NULL)) {\r
         if (SyncOrRestore) {\r
           NewStringCpy (&Node->Value, Node->EditValue);\r
         } else {\r
@@ -2072,7 +2317,7 @@ SynchronizeStorageForForm (
         }\r
       }\r
 \r
-      Link = GetNextNode (&ConfigInfo->Storage->NameValueListHead, Link);\r
+      Link = GetNextNode (&Storage->NameValueListHead, Link);\r
     }\r
   }\r
 \r
@@ -2095,17 +2340,10 @@ SendDiscardInfoToDriver (
 {\r
   LIST_ENTRY                  *Link;\r
   FORM_BROWSER_STATEMENT      *Question;\r
-  EFI_STATUS                  Status;\r
-  EFI_HII_VALUE               HiiValue;\r
-  UINT8                       *BufferValue;\r
-  BOOLEAN                     ValueChanged;\r
   EFI_IFR_TYPE_VALUE          *TypeValue;\r
   EFI_BROWSER_ACTION_REQUEST  ActionRequest;\r
 \r
-  ValueChanged = FALSE;\r
-  BufferValue  = NULL;\r
-\r
-  if(!Form->NvUpdateRequired) {\r
+  if (FormSet->ConfigAccess == NULL) {\r
     return;\r
   }\r
 \r
@@ -2118,48 +2356,22 @@ SendDiscardInfoToDriver (
       continue;\r
     }\r
 \r
-    if (Question->Operand == EFI_IFR_PASSWORD_OP) {\r
+    if ((Question->QuestionFlags & EFI_IFR_FLAG_CALLBACK) != EFI_IFR_FLAG_CALLBACK) {\r
       continue;\r
     }\r
 \r
-    if (Question->BufferValue != NULL) {\r
-      BufferValue = AllocateZeroPool (Question->StorageWidth);\r
-      ASSERT (BufferValue != NULL);\r
-      CopyMem (BufferValue, Question->BufferValue, Question->StorageWidth);\r
-    } else {\r
-      HiiValue.Type = Question->HiiValue.Type;\r
-      CopyMem (&HiiValue.Value, &Question->HiiValue.Value, sizeof (EFI_IFR_TYPE_VALUE));\r
-    }\r
-\r
-    Status = GetQuestionValue (FormSet, Form, Question, GetSetValueWithBuffer);\r
-    if (EFI_ERROR (Status)) {\r
-      if (BufferValue != NULL) {\r
-        FreePool (BufferValue);\r
-        BufferValue = NULL;\r
-      }\r
+    if (Question->Operand == EFI_IFR_PASSWORD_OP) {\r
       continue;\r
     }\r
 \r
-    if (Question->BufferValue != NULL) {\r
-      if (CompareMem (BufferValue, Question->BufferValue, Question->StorageWidth)) {\r
-        ValueChanged = TRUE;\r
-      }\r
-    } else {\r
-      if (CompareMem (&HiiValue.Value, &Question->HiiValue.Value, sizeof (EFI_IFR_TYPE_VALUE))) {\r
-        ValueChanged = TRUE;\r
-      }\r
-    }\r
-\r
-    if (BufferValue != NULL) {\r
-      FreePool (BufferValue);\r
-      BufferValue = NULL;\r
-    }\r
-\r
-    if (!ValueChanged) {\r
+    if (!Question->ValueChanged) {\r
       continue;\r
     }\r
 \r
-    ValueChanged = FALSE;\r
+    //\r
+    // Restore the question value before call the CHANGED callback type.\r
+    //\r
+    GetQuestionValue (FormSet, Form, Question, GetSetValueWithEditBuffer);\r
 \r
     if (Question->HiiValue.Type == EFI_IFR_TYPE_BUFFER) {\r
       TypeValue = (EFI_IFR_TYPE_VALUE *) Question->BufferValue;\r
@@ -2180,50 +2392,378 @@ SendDiscardInfoToDriver (
 }\r
 \r
 /**\r
-  Validate the FormSet. If the formset is not validate, remove it from the list.\r
-\r
-  @param  FormSet                The input FormSet which need to validate.\r
+  Validate the FormSet. If the formset is not validate, remove it from the list.\r
+\r
+  @param  FormSet                The input FormSet which need to validate.\r
+\r
+  @retval TRUE                   The handle is validate.\r
+  @retval FALSE                  The handle is invalidate.\r
+\r
+**/\r
+BOOLEAN\r
+ValidateHiiHandle (\r
+  EFI_HII_HANDLE          HiiHandle\r
+  )\r
+{\r
+  EFI_HII_HANDLE          *HiiHandles;\r
+  UINTN                   Index;\r
+  BOOLEAN                 Find;\r
+\r
+  if (HiiHandle == NULL) {\r
+    return FALSE;\r
+  }\r
+\r
+  Find = FALSE;\r
+\r
+  HiiHandles = HiiGetHiiHandles (NULL);\r
+  ASSERT (HiiHandles != NULL);\r
+\r
+  for (Index = 0; HiiHandles[Index] != NULL; Index++) {\r
+    if (HiiHandles[Index] == HiiHandle) {\r
+      Find = TRUE;\r
+      break;\r
+    }\r
+  }\r
+\r
+  FreePool (HiiHandles);\r
+\r
+  return Find;\r
+}\r
+\r
+/**\r
+  Validate the FormSet. If the formset is not validate, remove it from the list.\r
+\r
+  @param  FormSet                The input FormSet which need to validate.\r
+\r
+  @retval TRUE                   The handle is validate.\r
+  @retval FALSE                  The handle is invalidate.\r
+\r
+**/\r
+BOOLEAN\r
+ValidateFormSet (\r
+  FORM_BROWSER_FORMSET    *FormSet\r
+  )\r
+{\r
+  BOOLEAN  Find;\r
+\r
+  ASSERT (FormSet != NULL);\r
+\r
+  Find = ValidateHiiHandle(FormSet->HiiHandle);\r
+  if (!Find) {\r
+    CleanBrowserStorage(FormSet);\r
+    RemoveEntryList (&FormSet->Link);\r
+    DestroyFormSet (FormSet);\r
+  }\r
+\r
+  return Find;\r
+}\r
+/**\r
+  Check whether need to enable the reset flag in form level.\r
+  Also clean all ValueChanged flag in question.\r
+\r
+  @param  SetFlag                Whether need to set the Reset Flag.\r
+  @param  FormSet                FormSet data structure.\r
+  @param  Form                   Form data structure.\r
+\r
+**/\r
+VOID\r
+UpdateFlagForForm (\r
+  IN BOOLEAN                          SetFlag,\r
+  IN FORM_BROWSER_FORMSET             *FormSet,\r
+  IN FORM_BROWSER_FORM                *Form\r
+  )\r
+{\r
+  LIST_ENTRY              *Link;\r
+  FORM_BROWSER_STATEMENT  *Question;\r
+  BOOLEAN                 OldValue;\r
+\r
+  Link = GetFirstNode (&Form->StatementListHead);\r
+  while (!IsNull (&Form->StatementListHead, Link)) {\r
+    Question = FORM_BROWSER_STATEMENT_FROM_LINK (Link);\r
+    Link = GetNextNode (&Form->StatementListHead, Link);\r
+\r
+    if (!Question->ValueChanged) {\r
+      continue;\r
+    }\r
+\r
+    OldValue = Question->ValueChanged;\r
+\r
+    //\r
+    // Compare the buffer and editbuffer data to see whether the data has been saved.\r
+    //\r
+    Question->ValueChanged = IsQuestionValueChanged(FormSet, Form, Question, GetSetValueWithBothBuffer);\r
+\r
+    //\r
+    // Only the changed data has been saved, then need to set the reset flag.\r
+    //\r
+    if (SetFlag && OldValue && !Question->ValueChanged && ((Question->QuestionFlags & EFI_IFR_FLAG_RESET_REQUIRED) != 0)) {\r
+      gResetRequired = TRUE;\r
+    } \r
+  }\r
+}\r
+\r
+/**\r
+  Check whether need to enable the reset flag.\r
+  Also clean ValueChanged flag for all statements.\r
+\r
+  Form level or formset level, only one.\r
+  \r
+  @param  SetFlag                Whether need to set the Reset Flag.\r
+  @param  FormSet                FormSet data structure.\r
+  @param  Form                   Form data structure.\r
+\r
+**/\r
+VOID\r
+ValueChangeResetFlagUpdate (\r
+  IN BOOLEAN                          SetFlag,\r
+  IN FORM_BROWSER_FORMSET             *FormSet,\r
+  IN FORM_BROWSER_FORM                *Form  \r
+  )\r
+{\r
+  FORM_BROWSER_FORM       *CurrentForm;\r
+  LIST_ENTRY              *Link;\r
+\r
+  if (Form != NULL) {\r
+    UpdateFlagForForm(SetFlag, FormSet, Form);\r
+    return;\r
+  }\r
+\r
+  Link = GetFirstNode (&FormSet->FormListHead);\r
+  while (!IsNull (&FormSet->FormListHead, Link)) {\r
+    CurrentForm = FORM_BROWSER_FORM_FROM_LINK (Link);\r
+    Link = GetNextNode (&FormSet->FormListHead, Link);\r
+\r
+    UpdateFlagForForm(SetFlag, FormSet, CurrentForm);\r
+  }\r
+}\r
+\r
+/**\r
+  Base on the return Progress string to find the form. \r
+  \r
+  Base on the first return Offset/Width (Name) string to find the form\r
+  which keep this string.\r
+\r
+  @param  FormSet                FormSet data structure.\r
+  @param  Storage                Storage which has this Progress string.\r
+  @param  Progress               The Progress string which has the first fail string.\r
+  @param  RetForm                The return form for this progress string.\r
+  @param  RetQuestion            The return question for the error progress string.\r
+\r
+  @retval TRUE                   Find the error form and statement for this error progress string.\r
+  @retval FALSE                  Not find the error form.\r
+\r
+**/\r
+BOOLEAN\r
+FindQuestionFromProgress (\r
+  IN FORM_BROWSER_FORMSET             *FormSet,\r
+  IN BROWSER_STORAGE                  *Storage,\r
+  IN EFI_STRING                       Progress,\r
+  OUT FORM_BROWSER_FORM               **RetForm,\r
+  OUT FORM_BROWSER_STATEMENT          **RetQuestion\r
+  )\r
+{\r
+  LIST_ENTRY                   *Link;\r
+  LIST_ENTRY                   *LinkStorage;\r
+  LIST_ENTRY                   *LinkStatement;\r
+  FORM_BROWSER_CONFIG_REQUEST  *ConfigInfo;\r
+  FORM_BROWSER_FORM            *Form;\r
+  EFI_STRING                   EndStr;\r
+  FORM_BROWSER_STATEMENT       *Statement;\r
+\r
+  ASSERT ((*Progress == '&') || (*Progress == 'G'));\r
+\r
+  ConfigInfo   = NULL;\r
+  *RetForm     = NULL;\r
+  *RetQuestion = NULL;\r
+\r
+  //\r
+  // Skip the first "&" or the ConfigHdr part.\r
+  //\r
+  if (*Progress == '&') {\r
+    Progress++;\r
+  } else {\r
+    //\r
+    // Prepare the "NAME" or "OFFSET=0x####&WIDTH=0x####" string.\r
+    //\r
+    if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+      //\r
+      // For Name/Value type, Skip the ConfigHdr part.\r
+      //\r
+      EndStr = StrStr (Progress, L"PATH=");\r
+      ASSERT (EndStr != NULL);\r
+      while (*EndStr != '&') {\r
+        EndStr++;\r
+      }\r
+\r
+      *EndStr = '\0';\r
+    } else {\r
+      //\r
+      // For Buffer type, Skip the ConfigHdr part.\r
+      //\r
+      EndStr = StrStr (Progress, L"&OFFSET=");\r
+      ASSERT (EndStr != NULL);\r
+      *EndStr = '\0';\r
+    }\r
+\r
+    Progress = EndStr + 1;\r
+  }\r
+\r
+  //\r
+  // Prepare the "NAME" or "OFFSET=0x####&WIDTH=0x####" string.\r
+  //\r
+  if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    //\r
+    // For Name/Value type, the data is "&Fred=16&George=16&Ron=12" formset,\r
+    // here, just keep the "Fred" string.\r
+    //\r
+    EndStr = StrStr (Progress, L"=");\r
+    ASSERT (EndStr != NULL);\r
+    *EndStr = '\0';\r
+  } else {\r
+    //\r
+    // For Buffer type, the data is "OFFSET=0x####&WIDTH=0x####&VALUE=0x####",\r
+    // here, just keep the "OFFSET=0x####&WIDTH=0x####" string.\r
+    //\r
+    EndStr = StrStr (Progress, L"&VALUE=");\r
+    ASSERT (EndStr != NULL);\r
+    *EndStr = '\0';\r
+  }\r
+\r
+  //\r
+  // Search in the form list.\r
+  //\r
+  Link = GetFirstNode (&FormSet->FormListHead);\r
+  while (!IsNull (&FormSet->FormListHead, Link)) {\r
+    Form = FORM_BROWSER_FORM_FROM_LINK (Link);\r
+    Link = GetNextNode (&FormSet->FormListHead, Link);\r
+\r
+    //\r
+    // Search in the ConfigReqeust list in this form.\r
+    //\r
+    LinkStorage = GetFirstNode (&Form->ConfigRequestHead);\r
+    while (!IsNull (&Form->ConfigRequestHead, LinkStorage)) {\r
+      ConfigInfo = FORM_BROWSER_CONFIG_REQUEST_FROM_LINK (LinkStorage);\r
+      LinkStorage = GetNextNode (&Form->ConfigRequestHead, LinkStorage);\r
+\r
+      if (Storage != ConfigInfo->Storage) {\r
+        continue;\r
+      }\r
+\r
+      if (StrStr (ConfigInfo->ConfigRequest, Progress) != NULL) {\r
+        //\r
+        // Find the OffsetWidth string in this form.\r
+        //\r
+        *RetForm = Form;\r
+        break;\r
+      }\r
+    }\r
+\r
+    if (*RetForm != NULL) {\r
+      LinkStatement = GetFirstNode (&Form->StatementListHead);\r
+      while (!IsNull (&Form->StatementListHead, LinkStatement)) {\r
+        Statement = FORM_BROWSER_STATEMENT_FROM_LINK (LinkStatement);\r
+        LinkStatement = GetNextNode (&Form->StatementListHead, LinkStatement);\r
+\r
+        if (Statement->BlockName != NULL && StrStr (Statement->BlockName, Progress) != NULL) {\r
+          *RetQuestion = Statement;\r
+          break;\r
+        }\r
+      }\r
+    }\r
+\r
+    if (*RetForm != NULL) {\r
+      break;\r
+    }\r
+  }\r
+\r
+  //\r
+  // restore the OffsetWidth string to the original format.\r
+  //\r
+  if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    *EndStr = '=';\r
+  } else {\r
+    *EndStr = '&';\r
+  }\r
+\r
+  return (BOOLEAN) (*RetForm != NULL);\r
+}\r
+\r
+/**\r
+  Popup an save error info and get user input.\r
+\r
+  @param  TitleId                The form title id.\r
+  @param  HiiHandle              The hii handle for this package.\r
+\r
+  @retval UINT32                 The user select option for the save fail.\r
+                                 BROWSER_ACTION_DISCARD or BROWSER_ACTION_JUMP_TO_FORMSET\r
+**/\r
+UINT32\r
+ConfirmSaveFail (\r
+  IN EFI_STRING_ID    TitleId,\r
+  IN EFI_HII_HANDLE   HiiHandle\r
+  )\r
+{\r
+  CHAR16                  *FormTitle;\r
+  CHAR16                  *StringBuffer;\r
+  UINT32                  RetVal;\r
+\r
+  FormTitle = GetToken (TitleId, HiiHandle);\r
+\r
+  StringBuffer = AllocateZeroPool (256 * sizeof (CHAR16));\r
+  ASSERT (StringBuffer != NULL);\r
+\r
+  UnicodeSPrint (\r
+    StringBuffer, \r
+    24 * sizeof (CHAR16) + StrSize (FormTitle), \r
+    L"Submit Fail For Form: %s.", \r
+    FormTitle\r
+    );\r
+\r
+  RetVal = PopupErrorMessage(BROWSER_SUBMIT_FAIL, NULL, NULL, StringBuffer);\r
+\r
+  FreePool (StringBuffer);\r
+  FreePool (FormTitle);\r
+\r
+  return RetVal;\r
+}\r
+\r
+/**\r
+  Popup an NO_SUBMIT_IF error info and get user input.\r
 \r
-  @retval TRUE                   The handle is validate.\r
-  @retval FALSE                  The handle is invalidate.\r
+  @param  TitleId                The form title id.\r
+  @param  HiiHandle              The hii handle for this package.\r
 \r
+  @retval UINT32                 The user select option for the save fail.\r
+                                 BROWSER_ACTION_DISCARD or BROWSER_ACTION_JUMP_TO_FORMSET\r
 **/\r
-BOOLEAN\r
-ValidateFormSet (\r
-  FORM_BROWSER_FORMSET    *FormSet\r
+UINT32\r
+ConfirmNoSubmitFail (\r
+  IN EFI_STRING_ID    TitleId,\r
+  IN EFI_HII_HANDLE   HiiHandle\r
   )\r
 {\r
-  EFI_HII_HANDLE          *HiiHandles;\r
-  UINTN                   Index;\r
-  BOOLEAN                 Find;\r
+  CHAR16                  *FormTitle;\r
+  CHAR16                  *StringBuffer;\r
+  UINT32                  RetVal;\r
 \r
-  ASSERT (FormSet != NULL);\r
-  Find = FALSE;\r
-  //\r
-  // Get all the Hii handles\r
-  //\r
-  HiiHandles = HiiGetHiiHandles (NULL);\r
-  ASSERT (HiiHandles != NULL);\r
+  FormTitle = GetToken (TitleId, HiiHandle);\r
 \r
-  //\r
-  // Search for formset of each class type\r
-  //\r
-  for (Index = 0; HiiHandles[Index] != NULL; Index++) {\r
-    if (HiiHandles[Index] == FormSet->HiiHandle) {\r
-      Find = TRUE;\r
-      break;\r
-    }\r
-  }\r
+  StringBuffer = AllocateZeroPool (256 * sizeof (CHAR16));\r
+  ASSERT (StringBuffer != NULL);\r
 \r
-  if (!Find) {\r
-    CleanBrowserStorage(FormSet);\r
-    RemoveEntryList (&FormSet->Link);\r
-    DestroyFormSet (FormSet);\r
-  }\r
+  UnicodeSPrint (\r
+    StringBuffer, \r
+    24 * sizeof (CHAR16) + StrSize (FormTitle), \r
+    L"NO_SUBMIT_IF error For Form: %s.", \r
+    FormTitle\r
+    );\r
 \r
-  FreePool (HiiHandles);\r
+  RetVal = PopupErrorMessage(BROWSER_SUBMIT_FAIL_NO_SUBMIT_IF, NULL, NULL, StringBuffer);\r
 \r
-  return Find;\r
+  FreePool (StringBuffer);\r
+  FreePool (FormTitle);\r
+\r
+  return RetVal;\r
 }\r
 \r
 /**\r
@@ -2247,7 +2787,8 @@ DiscardForm (
   LIST_ENTRY                   *Link;\r
   FORMSET_STORAGE              *Storage;\r
   FORM_BROWSER_CONFIG_REQUEST  *ConfigInfo;\r
-  FORM_BROWSER_FORMSET    *LocalFormSet;\r
+  FORM_BROWSER_FORMSET         *LocalFormSet;\r
+  FORM_BROWSER_FORMSET         *OldFormSet;\r
 \r
   //\r
   // Check the supported setting level.\r
@@ -2256,7 +2797,7 @@ DiscardForm (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  if (SettingScope == FormLevel && Form->NvUpdateRequired) {\r
+  if (SettingScope == FormLevel && IsNvUpdateRequiredForForm (Form)) {\r
     ConfigInfo = NULL;\r
     Link = GetFirstNode (&Form->ConfigRequestHead);\r
     while (!IsNull (&Form->ConfigRequestHead, Link)) {\r
@@ -2277,7 +2818,7 @@ DiscardForm (
       //\r
       // Prepare <ConfigResp>\r
       //\r
-      SynchronizeStorageForForm(FormSet, ConfigInfo, FALSE);\r
+      SynchronizeStorage(ConfigInfo->Storage, ConfigInfo->ConfigRequest, FALSE);\r
 \r
       //\r
       // Call callback with Changed type to inform the driver.\r
@@ -2285,8 +2826,8 @@ DiscardForm (
       SendDiscardInfoToDriver (FormSet, Form);\r
     }\r
 \r
-    Form->NvUpdateRequired = FALSE;\r
-  } else if (SettingScope == FormSetLevel && IsNvUpdateRequired(FormSet)) {\r
+    ValueChangeResetFlagUpdate (FALSE, FormSet, Form);\r
+  } else if (SettingScope == FormSetLevel && IsNvUpdateRequiredForFormSet (FormSet)) {\r
 \r
     //\r
     // Discard Buffer storage or Name/Value storage\r
@@ -2307,7 +2848,7 @@ DiscardForm (
         continue;\r
       }\r
 \r
-      SynchronizeStorage(Storage->BrowserStorage, FALSE);\r
+      SynchronizeStorage(Storage->BrowserStorage, Storage->ConfigRequest, FALSE);\r
     }\r
 \r
     Link = GetFirstNode (&FormSet->FormListHead);\r
@@ -2321,12 +2862,13 @@ DiscardForm (
       SendDiscardInfoToDriver (FormSet, Form);\r
     }\r
 \r
-    UpdateNvInfoInForm (FormSet, FALSE);   \r
+    ValueChangeResetFlagUpdate(FALSE, FormSet, NULL);\r
   } else if (SettingScope == SystemLevel) {\r
     //\r
     // System Level Discard.\r
     //\r
-    \r
+    OldFormSet = mSystemLevelFormSet;\r
+\r
     //\r
     // Discard changed value for each FormSet in the maintain list.\r
     //\r
@@ -2337,6 +2879,9 @@ DiscardForm (
       if (!ValidateFormSet(LocalFormSet)) {\r
         continue;\r
       }\r
+\r
+      mSystemLevelFormSet = LocalFormSet;\r
+\r
       DiscardForm (LocalFormSet, NULL, FormSetLevel);\r
       if (!IsHiiHandleInBrowserContext (LocalFormSet->HiiHandle)) {\r
         //\r
@@ -2347,27 +2892,27 @@ DiscardForm (
         DestroyFormSet (LocalFormSet);\r
       }\r
     }\r
+\r
+    mSystemLevelFormSet = OldFormSet;\r
   }\r
 \r
   return EFI_SUCCESS;  \r
 }\r
 \r
 /**\r
-  Submit data based on the input Setting level (Form, FormSet or System).\r
+  Submit data for a form.\r
 \r
   @param  FormSet                FormSet data structure.\r
   @param  Form                   Form data structure.\r
-  @param  SettingScope           Setting Scope for Submit action.\r
 \r
   @retval EFI_SUCCESS            The function completed successfully.\r
   @retval EFI_UNSUPPORTED        Unsupport SettingScope.\r
 \r
 **/\r
 EFI_STATUS\r
-SubmitForm (\r
+SubmitForForm (\r
   IN FORM_BROWSER_FORMSET             *FormSet,\r
-  IN FORM_BROWSER_FORM                *Form,\r
-  IN BROWSER_SETTING_SCOPE            SettingScope\r
+  IN FORM_BROWSER_FORM                *Form\r
   )\r
 {\r
   EFI_STATUS              Status;\r
@@ -2375,268 +2920,453 @@ SubmitForm (
   EFI_STRING              ConfigResp;\r
   EFI_STRING              Progress;\r
   BROWSER_STORAGE         *Storage;\r
-  FORMSET_STORAGE         *FormSetStorage;\r
-  UINTN                   BufferSize;\r
-  UINT8                   *TmpBuf;  \r
-  FORM_BROWSER_FORMSET    *LocalFormSet;\r
   FORM_BROWSER_CONFIG_REQUEST  *ConfigInfo;\r
 \r
+  if (!IsNvUpdateRequiredForForm (Form)) {\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  Status = NoSubmitCheck (FormSet, &Form, NULL);\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  Link = GetFirstNode (&Form->ConfigRequestHead);\r
+  while (!IsNull (&Form->ConfigRequestHead, Link)) {\r
+    ConfigInfo = FORM_BROWSER_CONFIG_REQUEST_FROM_LINK (Link);\r
+    Link = GetNextNode (&Form->ConfigRequestHead, Link);\r
+\r
+    Storage = ConfigInfo->Storage;\r
+    if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE) {\r
+      continue;\r
+    }\r
+\r
+    //\r
+    // Skip if there is no RequestElement\r
+    //\r
+    if (ConfigInfo->ElementCount == 0) {\r
+      continue;\r
+    }\r
+\r
+    //\r
+    // 1. Prepare <ConfigResp>\r
+    //\r
+    Status = StorageToConfigResp (ConfigInfo->Storage, &ConfigResp, ConfigInfo->ConfigRequest, TRUE);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+\r
+    //\r
+    // 2. Set value to hii config routine protocol.\r
+    //\r
+    Status = mHiiConfigRouting->RouteConfig (\r
+                                      mHiiConfigRouting,\r
+                                      ConfigResp,\r
+                                      &Progress\r
+                                      );\r
+    FreePool (ConfigResp);\r
+\r
+    if (EFI_ERROR (Status)) {\r
+      InsertTailList (&gBrowserSaveFailFormSetList, &ConfigInfo->SaveFailLink);\r
+      continue;\r
+    }\r
+\r
+    //\r
+    // 3. Config success, update storage shadow Buffer, only update the data belong to this form.\r
+    //\r
+    SynchronizeStorage (ConfigInfo->Storage, ConfigInfo->ConfigRequest, TRUE);\r
+  }\r
+\r
   //\r
-  // Check the supported setting level.\r
+  // 4. Process the save failed storage.\r
   //\r
-  if (SettingScope >= MaxLevel) {\r
-    return EFI_UNSUPPORTED;\r
+  if (!IsListEmpty (&gBrowserSaveFailFormSetList)) {\r
+    if (ConfirmSaveFail (Form->FormTitle, FormSet->HiiHandle) == BROWSER_ACTION_DISCARD) {\r
+      Link = GetFirstNode (&gBrowserSaveFailFormSetList);\r
+      while (!IsNull (&gBrowserSaveFailFormSetList, Link)) {\r
+        ConfigInfo = FORM_BROWSER_CONFIG_REQUEST_FROM_SAVE_FAIL_LINK (Link);\r
+        Link = GetNextNode (&gBrowserSaveFailFormSetList, Link);\r
+\r
+        SynchronizeStorage(ConfigInfo->Storage, ConfigInfo->ConfigRequest, FALSE);\r
+\r
+        Status = EFI_SUCCESS;\r
+      }\r
+    } else {\r
+      Status = EFI_UNSUPPORTED;\r
+    }\r
+\r
+    //\r
+    // Free Form save fail list.\r
+    //\r
+    while (!IsListEmpty (&gBrowserSaveFailFormSetList)) {\r
+      Link = GetFirstNode (&gBrowserSaveFailFormSetList);\r
+      ConfigInfo = FORM_BROWSER_CONFIG_REQUEST_FROM_SAVE_FAIL_LINK (Link);\r
+      RemoveEntryList (&ConfigInfo->SaveFailLink);\r
+    }\r
   }\r
 \r
   //\r
-  // Validate the Form by NoSubmit check\r
+  // 5. Update the NV flag.\r
   //\r
-  Status = EFI_SUCCESS;\r
-  if (SettingScope == FormLevel) {\r
-    Status = NoSubmitCheck (FormSet, Form);\r
-  } else if (SettingScope == FormSetLevel) {\r
-    Status = NoSubmitCheck (FormSet, NULL);\r
+  ValueChangeResetFlagUpdate(TRUE, FormSet, Form);\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Submit data for a formset.\r
+\r
+  @param  FormSet                FormSet data structure.\r
+  @param  SkipProcessFail        Whether skip to process the save failed storage.\r
+                                 If submit formset is called when do system level save, \r
+                                 set this value to true and process the failed formset \r
+                                 together. \r
+                                 if submit formset is called when do formset level save,\r
+                                 set the value to false and process the failed storage\r
+                                 right after process all storages for this formset.\r
+\r
+  @retval EFI_SUCCESS            The function completed successfully.\r
+  @retval EFI_UNSUPPORTED        Unsupport SettingScope.\r
+\r
+**/\r
+EFI_STATUS\r
+SubmitForFormSet (\r
+  IN FORM_BROWSER_FORMSET             *FormSet,\r
+  IN BOOLEAN                          SkipProcessFail\r
+  )\r
+{\r
+  EFI_STATUS              Status;\r
+  LIST_ENTRY              *Link;\r
+  EFI_STRING              ConfigResp;\r
+  EFI_STRING              Progress;\r
+  BROWSER_STORAGE         *Storage;\r
+  FORMSET_STORAGE         *FormSetStorage;\r
+  FORM_BROWSER_FORM       *Form;\r
+  BOOLEAN                 HasInserted;\r
+  FORM_BROWSER_STATEMENT  *Question;\r
+\r
+  HasInserted = FALSE;\r
+\r
+  if (!IsNvUpdateRequiredForFormSet (FormSet)) {\r
+    return EFI_SUCCESS;\r
   }\r
+\r
+  Form = NULL; \r
+  Status = NoSubmitCheck (FormSet, &Form, &Question);\r
   if (EFI_ERROR (Status)) {\r
+    if (SkipProcessFail) {\r
+      //\r
+      // Process NO_SUBMIT check first, so insert it at head.\r
+      //\r
+      FormSet->SaveFailForm = Form;\r
+      FormSet->SaveFailStatement = Question;\r
+      InsertHeadList (&gBrowserSaveFailFormSetList, &FormSet->SaveFailLink);\r
+    }\r
+\r
     return Status;\r
   }\r
 \r
-  if (SettingScope == FormLevel && Form->NvUpdateRequired) {\r
-    ConfigInfo = NULL;\r
-    Link = GetFirstNode (&Form->ConfigRequestHead);\r
-    while (!IsNull (&Form->ConfigRequestHead, Link)) {\r
-      ConfigInfo = FORM_BROWSER_CONFIG_REQUEST_FROM_LINK (Link);\r
-      Link = GetNextNode (&Form->ConfigRequestHead, Link);\r
-\r
-      Storage = ConfigInfo->Storage;\r
-      if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE) {\r
-        continue;\r
-      }\r
+  Form = NULL;\r
+  Question = NULL;\r
+  //\r
+  // Submit Buffer storage or Name/Value storage\r
+  //\r
+  Link = GetFirstNode (&FormSet->StorageListHead);\r
+  while (!IsNull (&FormSet->StorageListHead, Link)) {\r
+    FormSetStorage = FORMSET_STORAGE_FROM_LINK (Link);\r
+    Storage        = FormSetStorage->BrowserStorage;\r
+    Link = GetNextNode (&FormSet->StorageListHead, Link);\r
 \r
-      //\r
-      // Skip if there is no RequestElement\r
-      //\r
-      if (ConfigInfo->ElementCount == 0) {\r
-        continue;\r
-      }\r
+    if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE) {\r
+      continue;\r
+    }\r
 \r
-      //\r
-      // 1. Prepare <ConfigResp>\r
-      //\r
-      Status = StorageToConfigResp (ConfigInfo->Storage, &ConfigResp, ConfigInfo->ConfigRequest);\r
-      if (EFI_ERROR (Status)) {\r
-        return Status;\r
-      }\r
+    //\r
+    // Skip if there is no RequestElement\r
+    //\r
+    if (FormSetStorage->ElementCount == 0) {\r
+      continue;\r
+    }\r
 \r
-      //\r
-      // 2. Set value to hii driver or efi variable.\r
-      //\r
-      if (Storage->Type == EFI_HII_VARSTORE_BUFFER || \r
-          Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
-        //\r
-        // Send <ConfigResp> to Configuration Driver\r
-        //\r
-        if (FormSet->ConfigAccess != NULL) {\r
-          Status = FormSet->ConfigAccess->RouteConfig (\r
-                                            FormSet->ConfigAccess,\r
-                                            ConfigResp,\r
-                                            &Progress\r
-                                            );\r
-          if (EFI_ERROR (Status)) {\r
-            FreePool (ConfigResp);\r
-            return Status;\r
-          }\r
-        }\r
-      } else if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
-        TmpBuf = NULL;\r
-        TmpBuf = AllocateZeroPool(Storage->Size);\r
-        if (TmpBuf == NULL) {\r
-          Status = EFI_OUT_OF_RESOURCES;\r
-          return Status;\r
-        }\r
+    //\r
+    // 1. Prepare <ConfigResp>\r
+    //\r
+    Status = StorageToConfigResp (Storage, &ConfigResp, FormSetStorage->ConfigRequest, TRUE);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
 \r
-        BufferSize = Storage->Size;\r
-        Status = gRT->GetVariable (\r
-                         Storage->Name,\r
-                         &Storage->Guid,\r
-                         NULL,\r
-                         &BufferSize,\r
-                         TmpBuf\r
-                         );\r
-        if (EFI_ERROR (Status)) {\r
-          FreePool (TmpBuf);\r
-          FreePool (ConfigResp);\r
-          return Status;\r
-        }\r
-        ASSERT (BufferSize == Storage->Size);      \r
-        Status = mHiiConfigRouting->ConfigToBlock (\r
+    //\r
+    // 2. Send <ConfigResp> to Routine config Protocol.\r
+    //\r
+    Status = mHiiConfigRouting->RouteConfig (\r
                                       mHiiConfigRouting,\r
                                       ConfigResp,\r
-                                      TmpBuf,\r
-                                      &BufferSize,\r
                                       &Progress\r
                                       );\r
-        if (EFI_ERROR (Status)) {\r
-          FreePool (TmpBuf);\r
-          FreePool (ConfigResp);\r
-          return Status;\r
+    if (EFI_ERROR (Status)) {\r
+      InsertTailList (&FormSet->SaveFailStorageListHead, &FormSetStorage->SaveFailLink);\r
+      if (!HasInserted) {\r
+        //\r
+        // Call submit formset for system level, save the formset info\r
+        // and process later.\r
+        //\r
+        FindQuestionFromProgress(FormSet, Storage, Progress, &Form, &Question);\r
+        ASSERT (Form != NULL && Question != NULL);\r
+        FormSet->SaveFailForm = Form;\r
+        FormSet->SaveFailStatement = Question;\r
+        if (SkipProcessFail) {\r
+          InsertTailList (&gBrowserSaveFailFormSetList, &FormSet->SaveFailLink);\r
         }\r
+        HasInserted = TRUE;\r
+      }\r
 \r
-        Status = gRT->SetVariable (\r
-                         Storage->Name,\r
-                         &Storage->Guid,\r
-                         Storage->Attributes,\r
-                         Storage->Size,\r
-                         TmpBuf\r
-                         );\r
-        FreePool (TmpBuf);\r
-        if (EFI_ERROR (Status)) {\r
-          FreePool (ConfigResp);\r
-          return Status;\r
+      FreePool (ConfigResp);\r
+      continue;\r
+    }\r
+\r
+    FreePool (ConfigResp);\r
+    //\r
+    // 3. Config success, update storage shadow Buffer\r
+    //\r
+    SynchronizeStorage (Storage, FormSetStorage->ConfigRequest, TRUE);\r
+  }\r
+\r
+  //\r
+  // 4. Has save fail storage need to handle.\r
+  //\r
+  if (Form != NULL) {\r
+    if (!SkipProcessFail) {\r
+      //\r
+      // If not in system level, just handl the save failed storage here.\r
+      //\r
+      if (ConfirmSaveFail (Form->FormTitle, FormSet->HiiHandle) == BROWSER_ACTION_DISCARD) {\r
+        Link = GetFirstNode (&FormSet->SaveFailStorageListHead);\r
+        while (!IsNull (&FormSet->SaveFailStorageListHead, Link)) {\r
+          FormSetStorage = FORMSET_STORAGE_FROM_SAVE_FAIL_LINK (Link);\r
+          Storage        = FormSetStorage->BrowserStorage;\r
+          Link = GetNextNode (&FormSet->SaveFailStorageListHead, Link);\r
+\r
+          SynchronizeStorage(FormSetStorage->BrowserStorage, FormSetStorage->ConfigRequest, FALSE);\r
+\r
+          Status = EFI_SUCCESS;\r
         }\r
+      } else {\r
+        UiCopyMenuList(&mPrivateData.FormBrowserEx2.FormViewHistoryHead, &Form->FormViewListHead);\r
+\r
+        gCurrentSelection->Action = UI_ACTION_REFRESH_FORMSET;\r
+        gCurrentSelection->Handle = FormSet->HiiHandle;\r
+        CopyGuid (&gCurrentSelection->FormSetGuid, &FormSet->Guid);\r
+        gCurrentSelection->FormId = Form->FormId;\r
+        gCurrentSelection->QuestionId = Question->QuestionId;\r
+\r
+        Status = EFI_UNSUPPORTED;\r
       }\r
-      FreePool (ConfigResp);\r
+\r
+      //\r
+      // Free FormSet save fail list.\r
+      //\r
+      while (!IsListEmpty (&FormSet->SaveFailStorageListHead)) {\r
+        Link = GetFirstNode (&FormSet->SaveFailStorageListHead);\r
+        FormSetStorage = FORMSET_STORAGE_FROM_SAVE_FAIL_LINK (Link);\r
+        RemoveEntryList (&FormSetStorage->SaveFailLink);\r
+      }\r
+    } else {\r
       //\r
-      // 3. Config success, update storage shadow Buffer, only update the data belong to this form.\r
+      // If in system level, just return error and handle the failed formset later.\r
       //\r
-      SynchronizeStorageForForm(FormSet, ConfigInfo, TRUE);\r
+      Status = EFI_UNSUPPORTED;\r
+    }\r
+  }\r
+\r
+  //\r
+  // 5. Update the NV flag.\r
+  // \r
+  ValueChangeResetFlagUpdate(TRUE, FormSet, NULL);\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Submit data for all formsets.\r
+\r
+  @retval EFI_SUCCESS            The function completed successfully.\r
+  @retval EFI_UNSUPPORTED        Unsupport SettingScope.\r
+\r
+**/\r
+EFI_STATUS\r
+SubmitForSystem (\r
+  VOID\r
+  )\r
+{\r
+  EFI_STATUS              Status;\r
+  LIST_ENTRY              *Link;\r
+  LIST_ENTRY              *StorageLink;\r
+  BROWSER_STORAGE         *Storage;\r
+  FORMSET_STORAGE         *FormSetStorage;\r
+  FORM_BROWSER_FORM       *Form;\r
+  FORM_BROWSER_FORMSET    *LocalFormSet;\r
+  UINT32                  UserSelection;\r
+  FORM_BROWSER_STATEMENT  *Question;\r
+\r
+  mSystemSubmit = TRUE;\r
+  Link = GetFirstNode (&gBrowserFormSetList);\r
+  while (!IsNull (&gBrowserFormSetList, Link)) {\r
+    LocalFormSet = FORM_BROWSER_FORMSET_FROM_LINK (Link);\r
+    Link = GetNextNode (&gBrowserFormSetList, Link);\r
+    if (!ValidateFormSet(LocalFormSet)) {\r
+      continue;\r
+    }\r
+\r
+    Status = SubmitForFormSet (LocalFormSet, TRUE);\r
+    if (EFI_ERROR (Status)) {\r
+      continue;\r
     }\r
 \r
     //\r
-    // 4. Update the NV flag.\r
-    // \r
-    Form->NvUpdateRequired = FALSE;\r
-  } else if (SettingScope == FormSetLevel && IsNvUpdateRequired(FormSet)) {\r
-    //\r
-    // Submit Buffer storage or Name/Value storage\r
-    //\r
-    Link = GetFirstNode (&FormSet->StorageListHead);\r
-    while (!IsNull (&FormSet->StorageListHead, Link)) {\r
-      FormSetStorage = (FORMSET_STORAGE_FROM_LINK (Link));\r
-      Storage        = FormSetStorage->BrowserStorage;\r
-      Link = GetNextNode (&FormSet->StorageListHead, Link);\r
+    // Remove maintain backup list after save except for the current using FormSet.\r
+    //  \r
+    if (!IsHiiHandleInBrowserContext (LocalFormSet->HiiHandle)) {\r
+      CleanBrowserStorage(LocalFormSet);\r
+      RemoveEntryList (&LocalFormSet->Link);\r
+      DestroyFormSet (LocalFormSet);\r
+    }\r
+  }\r
+  mSystemSubmit = FALSE;\r
 \r
-      if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE) {\r
-        continue;\r
-      }\r
+  Status = EFI_SUCCESS;\r
 \r
-      //\r
-      // Skip if there is no RequestElement\r
-      //\r
-      if (FormSetStorage->ElementCount == 0) {\r
-        continue;\r
-      }\r
+  //\r
+  // Process the save failed formsets.\r
+  //\r
+  Link = GetFirstNode (&gBrowserSaveFailFormSetList);\r
+  while (!IsNull (&gBrowserSaveFailFormSetList, Link)) {\r
+    LocalFormSet = FORM_BROWSER_FORMSET_FROM_SAVE_FAIL_LINK (Link);\r
+    Link = GetNextNode (&gBrowserSaveFailFormSetList, Link);\r
+\r
+    if (!ValidateFormSet(LocalFormSet)) {\r
+      continue;\r
+    }\r
 \r
+    Form = LocalFormSet->SaveFailForm;\r
+    Question= LocalFormSet->SaveFailStatement;\r
+\r
+    //\r
+    // Confirm with user, get user input.\r
+    //\r
+    if (IsListEmpty (&LocalFormSet->SaveFailStorageListHead)) {\r
       //\r
-      // 1. Prepare <ConfigResp>\r
+      // NULL for SaveFailStorageListHead means error caused by NO_SUBMIT_IF check.\r
       //\r
-      Status = StorageToConfigResp (Storage, &ConfigResp, FormSetStorage->ConfigRequest);\r
-      if (EFI_ERROR (Status)) {\r
-        return Status;\r
-      }\r
+      UserSelection = ConfirmNoSubmitFail (Form->FormTitle, LocalFormSet->HiiHandle);\r
+    } else {\r
+      UserSelection = ConfirmSaveFail (Form->FormTitle, LocalFormSet->HiiHandle);\r
+    }\r
 \r
-      if (Storage->Type == EFI_HII_VARSTORE_BUFFER || \r
-          Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    if (UserSelection == BROWSER_ACTION_DISCARD) {\r
+      if (IsListEmpty (&LocalFormSet->SaveFailStorageListHead)) {\r
+        StorageLink = GetFirstNode (&LocalFormSet->StorageListHead);\r
+        while (!IsNull (&LocalFormSet->StorageListHead, StorageLink)) {\r
+          FormSetStorage = FORMSET_STORAGE_FROM_LINK (StorageLink);\r
+          Storage        = FormSetStorage->BrowserStorage;\r
+          StorageLink = GetNextNode (&LocalFormSet->StorageListHead, StorageLink);\r
 \r
-        //\r
-        // 2. Send <ConfigResp> to Configuration Driver\r
-        //\r
-        if (FormSet->ConfigAccess != NULL) {\r
-          Status = FormSet->ConfigAccess->RouteConfig (\r
-                                            FormSet->ConfigAccess,\r
-                                            ConfigResp,\r
-                                            &Progress\r
-                                            );\r
-          if (EFI_ERROR (Status)) {\r
-            FreePool (ConfigResp);\r
-            return Status;\r
-          }\r
-        }\r
-      } else if (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
-        //\r
-        // 1&2. Set the edit data to the variable.\r
-        //\r
-        TmpBuf = NULL;\r
-        TmpBuf = AllocateZeroPool (Storage->Size);\r
-        if (TmpBuf == NULL) {\r
-          Status = EFI_OUT_OF_RESOURCES;\r
-          return Status;\r
-        }        \r
-        BufferSize = Storage->Size;\r
-        Status = gRT->GetVariable (\r
-                       Storage->Name,\r
-                       &Storage->Guid,\r
-                       NULL,\r
-                       &BufferSize,\r
-                       TmpBuf\r
-                       );\r
-        ASSERT (BufferSize == Storage->Size);      \r
-        Status = mHiiConfigRouting->ConfigToBlock (\r
-                                      mHiiConfigRouting,\r
-                                      ConfigResp,\r
-                                      TmpBuf,\r
-                                      &BufferSize,\r
-                                      &Progress\r
-                                      );\r
-        if (EFI_ERROR (Status)) {\r
-          FreePool (TmpBuf);\r
-          FreePool (ConfigResp);\r
-          return Status;\r
+          SynchronizeStorage(FormSetStorage->BrowserStorage, FormSetStorage->ConfigRequest, FALSE);\r
         }\r
+      } else {\r
+        StorageLink = GetFirstNode (&LocalFormSet->SaveFailStorageListHead);\r
+        while (!IsNull (&LocalFormSet->SaveFailStorageListHead, StorageLink)) {\r
+          FormSetStorage = FORMSET_STORAGE_FROM_SAVE_FAIL_LINK (StorageLink);\r
+          Storage        = FormSetStorage->BrowserStorage;\r
+          StorageLink = GetNextNode (&LocalFormSet->SaveFailStorageListHead, StorageLink);\r
 \r
-        Status = gRT->SetVariable (\r
-                         Storage->Name,\r
-                         &Storage->Guid,\r
-                         Storage->Attributes,\r
-                         Storage->Size,\r
-                         TmpBuf\r
-                         );\r
-        if (EFI_ERROR (Status)) {\r
-          FreePool (TmpBuf);\r
-          FreePool (ConfigResp);\r
-          return Status;\r
+          SynchronizeStorage(FormSetStorage->BrowserStorage, FormSetStorage->ConfigRequest, FALSE);\r
         }\r
-        FreePool (TmpBuf);\r
       }\r
-      FreePool (ConfigResp);\r
-      //\r
-      // 3. Config success, update storage shadow Buffer\r
-      //\r
-      SynchronizeStorage (Storage, TRUE);\r
-    }\r
-\r
-    //\r
-    // 4. Update the NV flag.\r
-    // \r
-    UpdateNvInfoInForm (FormSet, FALSE);\r
-  } else if (SettingScope == SystemLevel) {\r
-    //\r
-    // System Level Save.\r
-    //\r
 \r
-    //\r
-    // Save changed value for each FormSet in the maintain list.\r
-    //\r
-    Link = GetFirstNode (&gBrowserFormSetList);\r
-    while (!IsNull (&gBrowserFormSetList, Link)) {\r
-      LocalFormSet = FORM_BROWSER_FORMSET_FROM_LINK (Link);\r
-      Link = GetNextNode (&gBrowserFormSetList, Link);\r
-      if (!ValidateFormSet(LocalFormSet)) {\r
-        continue;\r
-      }\r
-      SubmitForm (LocalFormSet, NULL, FormSetLevel);\r
       if (!IsHiiHandleInBrowserContext (LocalFormSet->HiiHandle)) {\r
-        //\r
-        // Remove maintain backup list after save except for the current using FormSet.\r
-        //\r
         CleanBrowserStorage(LocalFormSet);\r
         RemoveEntryList (&LocalFormSet->Link);\r
+        RemoveEntryList (&LocalFormSet->SaveFailLink);\r
         DestroyFormSet (LocalFormSet);\r
+      } else {\r
+        ValueChangeResetFlagUpdate(FALSE, LocalFormSet, NULL);\r
       }\r
+    } else {\r
+      if (IsListEmpty (&LocalFormSet->SaveFailStorageListHead)) {\r
+        NoSubmitCheck (LocalFormSet, &Form, &Question);\r
+      }\r
+\r
+      UiCopyMenuList(&mPrivateData.FormBrowserEx2.FormViewHistoryHead, &Form->FormViewListHead);\r
+\r
+      gCurrentSelection->Action = UI_ACTION_REFRESH_FORMSET;\r
+      gCurrentSelection->Handle = LocalFormSet->HiiHandle;\r
+      CopyGuid (&gCurrentSelection->FormSetGuid, &LocalFormSet->Guid);\r
+      gCurrentSelection->FormId = Form->FormId;\r
+      gCurrentSelection->QuestionId = Question->QuestionId;\r
+\r
+      Status = EFI_UNSUPPORTED;\r
+      break;\r
     }\r
   }\r
 \r
-  return EFI_SUCCESS;\r
+  //\r
+  // Clean the list which will not process.\r
+  //\r
+  while (!IsListEmpty (&gBrowserSaveFailFormSetList)) {\r
+    Link = GetFirstNode (&gBrowserSaveFailFormSetList);\r
+    LocalFormSet = FORM_BROWSER_FORMSET_FROM_SAVE_FAIL_LINK (Link);\r
+    RemoveEntryList (&LocalFormSet->SaveFailLink);\r
+\r
+    while (!IsListEmpty (&LocalFormSet->SaveFailStorageListHead)) {\r
+      StorageLink = GetFirstNode (&LocalFormSet->SaveFailStorageListHead);\r
+      FormSetStorage = FORMSET_STORAGE_FROM_SAVE_FAIL_LINK (StorageLink);\r
+      RemoveEntryList (&FormSetStorage->SaveFailLink);\r
+    }\r
+  }\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Submit data based on the input Setting level (Form, FormSet or System).\r
+\r
+  @param  FormSet                FormSet data structure.\r
+  @param  Form                   Form data structure.\r
+  @param  SettingScope           Setting Scope for Submit action.\r
+\r
+  @retval EFI_SUCCESS            The function completed successfully.\r
+  @retval EFI_UNSUPPORTED        Unsupport SettingScope.\r
+\r
+**/\r
+EFI_STATUS\r
+SubmitForm (\r
+  IN FORM_BROWSER_FORMSET             *FormSet,\r
+  IN FORM_BROWSER_FORM                *Form,\r
+  IN BROWSER_SETTING_SCOPE            SettingScope\r
+  )\r
+{\r
+  EFI_STATUS              Status;\r
+\r
+  switch (SettingScope) {\r
+  case FormLevel:\r
+    Status = SubmitForForm(FormSet, Form);\r
+    break;\r
+\r
+  case FormSetLevel:\r
+    Status = SubmitForFormSet (FormSet, FALSE);\r
+    break;\r
+\r
+  case SystemLevel:\r
+    Status = SubmitForSystem ();\r
+    break;\r
+\r
+  default:\r
+    Status = EFI_UNSUPPORTED;\r
+    break;\r
+  }\r
+\r
+  return Status;\r
 }\r
 \r
 /**\r
@@ -2682,9 +3412,7 @@ GetDefaultValueFromAltCfg (
   Value         = NULL;\r
   Storage       = Question->Storage;\r
 \r
-  if ((Storage == NULL) || \r
-      (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE) || \r
-      (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+  if ((Storage == NULL) || (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE)) {\r
     return Status;\r
   }\r
 \r
@@ -2703,7 +3431,11 @@ GetDefaultValueFromAltCfg (
     Dst = (UINT8 *) &Question->HiiValue.Value;\r
   }\r
 \r
-  IsBufferStorage = (BOOLEAN) ((Storage->Type == EFI_HII_VARSTORE_BUFFER) ? TRUE : FALSE);\r
+  if (Storage->Type == EFI_HII_VARSTORE_BUFFER || Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
+    IsBufferStorage = TRUE;\r
+  } else {\r
+    IsBufferStorage = FALSE;\r
+  }\r
   IsString = (BOOLEAN) ((Question->HiiValue.Type == EFI_IFR_TYPE_STRING) ?  TRUE : FALSE);\r
 \r
   //\r
@@ -2728,8 +3460,8 @@ GetDefaultValueFromAltCfg (
     StrCat (ConfigRequest, Question->VariableName);\r
   }\r
 \r
-  Status = FormSet->ConfigAccess->ExtractConfig (\r
-                                    FormSet->ConfigAccess,\r
+  Status = mHiiConfigRouting->ExtractConfig (\r
+                                    mHiiConfigRouting,\r
                                     ConfigRequest,\r
                                     &Progress,\r
                                     &Result\r
@@ -2759,6 +3491,11 @@ GetDefaultValueFromAltCfg (
     goto Done;\r
   }\r
 \r
+  if (ConfigResp == NULL) {\r
+    Status = EFI_NOT_FOUND;\r
+    goto Done;\r
+  }\r
+\r
   //\r
   // Skip <ConfigRequest>\r
   //\r
@@ -2864,6 +3601,138 @@ GetDefaultIdForCallBack (
   }\r
 }\r
 \r
+\r
+\r
+/**\r
+  Return data element in an Array by its Index.\r
+\r
+  @param  Array                  The data array.\r
+  @param  Type                   Type of the data in this array.\r
+  @param  Index                  Zero based index for data in this array.\r
+\r
+  @retval Value                  The data to be returned\r
+\r
+**/\r
+UINT64\r
+GetArrayData (\r
+  IN VOID                     *Array,\r
+  IN UINT8                    Type,\r
+  IN UINTN                    Index\r
+  )\r
+{\r
+  UINT64 Data;\r
+\r
+  ASSERT (Array != NULL);\r
+\r
+  Data = 0;\r
+  switch (Type) {\r
+  case EFI_IFR_TYPE_NUM_SIZE_8:\r
+    Data = (UINT64) *(((UINT8 *) Array) + Index);\r
+    break;\r
+\r
+  case EFI_IFR_TYPE_NUM_SIZE_16:\r
+    Data = (UINT64) *(((UINT16 *) Array) + Index);\r
+    break;\r
+\r
+  case EFI_IFR_TYPE_NUM_SIZE_32:\r
+    Data = (UINT64) *(((UINT32 *) Array) + Index);\r
+    break;\r
+\r
+  case EFI_IFR_TYPE_NUM_SIZE_64:\r
+    Data = (UINT64) *(((UINT64 *) Array) + Index);\r
+    break;\r
+\r
+  default:\r
+    break;\r
+  }\r
+\r
+  return Data;\r
+}\r
+\r
+\r
+/**\r
+  Set value of a data element in an Array by its Index.\r
+\r
+  @param  Array                  The data array.\r
+  @param  Type                   Type of the data in this array.\r
+  @param  Index                  Zero based index for data in this array.\r
+  @param  Value                  The value to be set.\r
+\r
+**/\r
+VOID\r
+SetArrayData (\r
+  IN VOID                     *Array,\r
+  IN UINT8                    Type,\r
+  IN UINTN                    Index,\r
+  IN UINT64                   Value\r
+  )\r
+{\r
+\r
+  ASSERT (Array != NULL);\r
+\r
+  switch (Type) {\r
+  case EFI_IFR_TYPE_NUM_SIZE_8:\r
+    *(((UINT8 *) Array) + Index) = (UINT8) Value;\r
+    break;\r
+\r
+  case EFI_IFR_TYPE_NUM_SIZE_16:\r
+    *(((UINT16 *) Array) + Index) = (UINT16) Value;\r
+    break;\r
+\r
+  case EFI_IFR_TYPE_NUM_SIZE_32:\r
+    *(((UINT32 *) Array) + Index) = (UINT32) Value;\r
+    break;\r
+\r
+  case EFI_IFR_TYPE_NUM_SIZE_64:\r
+    *(((UINT64 *) Array) + Index) = (UINT64) Value;\r
+    break;\r
+\r
+  default:\r
+    break;\r
+  }\r
+}\r
+\r
+/**\r
+  Search an Option of a Question by its value.\r
+\r
+  @param  Question               The Question\r
+  @param  OptionValue            Value for Option to be searched.\r
+\r
+  @retval Pointer                Pointer to the found Option.\r
+  @retval NULL                   Option not found.\r
+\r
+**/\r
+QUESTION_OPTION *\r
+ValueToOption (\r
+  IN FORM_BROWSER_STATEMENT   *Question,\r
+  IN EFI_HII_VALUE            *OptionValue\r
+  )\r
+{\r
+  LIST_ENTRY       *Link;\r
+  QUESTION_OPTION  *Option;\r
+  INTN             Result;\r
+\r
+  Link = GetFirstNode (&Question->OptionListHead);\r
+  while (!IsNull (&Question->OptionListHead, Link)) {\r
+    Option = QUESTION_OPTION_FROM_LINK (Link);\r
+\r
+    if ((CompareHiiValue (&Option->Value, OptionValue, &Result, NULL) == EFI_SUCCESS) && (Result == 0)) {\r
+      //\r
+      // Check the suppressif condition, only a valid option can be return.\r
+      //\r
+      if ((Option->SuppressExpression == NULL) ||\r
+          ((EvaluateExpressionList(Option->SuppressExpression, FALSE, NULL, NULL) == ExpressFalse))) {\r
+        return Option;\r
+      }\r
+    }\r
+\r
+    Link = GetNextNode (&Question->OptionListHead, Link);\r
+  }\r
+\r
+  return NULL;\r
+}\r
+\r
+\r
 /**\r
   Reset Question to its default value.\r
 \r
@@ -2893,6 +3762,7 @@ GetQuestionDefault (
   EFI_HII_CONFIG_ACCESS_PROTOCOL  *ConfigAccess;\r
   EFI_BROWSER_ACTION_REQUEST      ActionRequest;\r
   INTN                            Action;\r
+  CHAR16                          *NewString;\r
 \r
   Status   = EFI_NOT_FOUND;\r
   StrValue = NULL;\r
@@ -2930,6 +3800,19 @@ GetQuestionDefault (
                              &ActionRequest\r
                              );\r
     if (!EFI_ERROR (Status)) {\r
+      if (HiiValue->Type == EFI_IFR_TYPE_STRING) {\r
+        NewString = GetToken (Question->HiiValue.Value.string, FormSet->HiiHandle);\r
+        ASSERT (NewString != NULL);\r
+\r
+        ASSERT (StrLen (NewString) * sizeof (CHAR16) <= Question->StorageWidth);\r
+        if (StrLen (NewString) * sizeof (CHAR16) <= Question->StorageWidth) {\r
+          CopyMem (Question->BufferValue, NewString, StrSize (NewString));\r
+        } else {\r
+          CopyMem (Question->BufferValue, NewString, Question->StorageWidth);\r
+        }\r
+\r
+        FreePool (NewString);\r
+      }\r
       return Status;\r
     }\r
   }\r
@@ -3155,11 +4038,8 @@ ExtractDefault (
   LIST_ENTRY              *FormLink;\r
   LIST_ENTRY              *Link;\r
   FORM_BROWSER_STATEMENT  *Question;\r
-  FORM_BROWSER_FORMSET    *BackUpFormSet;\r
   FORM_BROWSER_FORMSET    *LocalFormSet;\r
-  EFI_HII_HANDLE          *HiiHandles;\r
-  UINTN                   Index;\r
-  EFI_GUID                ZeroGuid;\r
+  FORM_BROWSER_FORMSET    *OldFormSet;\r
 \r
   Status = EFI_SUCCESS;\r
 \r
@@ -3210,7 +4090,7 @@ ExtractDefault (
         //\r
         // Call the Retrieve call back to get the initial question value.\r
         //\r
-        Status = ProcessRetrieveForQuestion(FormSet->ConfigAccess, Question);\r
+        Status = ProcessRetrieveForQuestion(FormSet->ConfigAccess, Question, FormSet);\r
       }\r
 \r
       //\r
@@ -3229,10 +4109,6 @@ ExtractDefault (
       if ((Question->Storage != NULL) &&\r
           (Question->Storage->Type != EFI_HII_VARSTORE_EFI_VARIABLE)) {\r
         SetQuestionValue (FormSet, Form, Question, GetSetValueWithEditBuffer);\r
-        //\r
-        // Update Form NV flag.\r
-        //\r
-        Form->NvUpdateRequired = TRUE;\r
       }\r
     }\r
   } else if (SettingScope == FormSetLevel) {\r
@@ -3244,81 +4120,148 @@ ExtractDefault (
     }\r
   } else if (SettingScope == SystemLevel) {\r
     //\r
-    // Open all FormSet by locate HII packages.\r
-    // Initiliaze the maintain FormSet to store default data as back up data.\r
+    // Preload all Hii formset.\r
     //\r
-    BackUpFormSet    = gOldFormSet;\r
-    gOldFormSet      = NULL;\r
+    LoadAllHiiFormset();\r
 \r
-    //\r
-    // Get all the Hii handles\r
-    //\r
-    HiiHandles = HiiGetHiiHandles (NULL);\r
-    ASSERT (HiiHandles != NULL);\r
+    OldFormSet = mSystemLevelFormSet;\r
 \r
     //\r
-    // Search for formset of each class type\r
+    // Set Default Value for each FormSet in the maintain list.\r
     //\r
-    for (Index = 0; HiiHandles[Index] != NULL; Index++) {\r
-      //\r
-      // Check HiiHandles[Index] does exist in global maintain list. \r
-      //\r
-      if (GetFormSetFromHiiHandle (HiiHandles[Index]) != NULL) {\r
-        continue;\r
-      }\r
-      \r
-      //\r
-      // Initilize FormSet Setting\r
-      //\r
-      LocalFormSet = AllocateZeroPool (sizeof (FORM_BROWSER_FORMSET));\r
-      ASSERT (LocalFormSet != NULL);\r
-      ZeroMem (&ZeroGuid, sizeof (ZeroGuid));\r
-      Status = InitializeFormSet (HiiHandles[Index], &ZeroGuid, LocalFormSet, FALSE);\r
-      if (EFI_ERROR (Status) || IsListEmpty (&LocalFormSet->FormListHead)) {\r
-        DestroyFormSet (LocalFormSet);\r
-        continue;\r
-      }\r
-      Status = InitializeCurrentSetting (LocalFormSet);\r
-      if (EFI_ERROR (Status)) {\r
-        DestroyFormSet (LocalFormSet);\r
-        continue;\r
-      }\r
-      //\r
-      // Initilize Questions' Value\r
-      //\r
-      Status = LoadFormSetConfig (NULL, LocalFormSet);\r
-      if (EFI_ERROR (Status)) {\r
-        DestroyFormSet (LocalFormSet);\r
+    Link = GetFirstNode (&gBrowserFormSetList);\r
+    while (!IsNull (&gBrowserFormSetList, Link)) {\r
+      LocalFormSet = FORM_BROWSER_FORMSET_FROM_LINK (Link);\r
+      Link = GetNextNode (&gBrowserFormSetList, Link);\r
+      if (!ValidateFormSet(LocalFormSet)) {\r
         continue;\r
       }\r
 \r
-      //\r
-      // Add FormSet into the maintain list.\r
-      //\r
-      InsertTailList (&gBrowserFormSetList, &LocalFormSet->Link);\r
+      mSystemLevelFormSet = LocalFormSet;\r
+\r
+      ExtractDefault (LocalFormSet, NULL, DefaultId, FormSetLevel, GetDefaultValueScope, Storage, RetrieveValueFirst);\r
+    }\r
+\r
+    mSystemLevelFormSet = OldFormSet;\r
+  }\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+\r
+/**\r
+  Validate whether this question's value has changed.\r
+\r
+  @param  FormSet                FormSet data structure.\r
+  @param  Form                   Form data structure.\r
+  @param  Question               Question to be initialized.\r
+  @param  GetValueFrom           Where to get value, may from editbuffer, buffer or hii driver.\r
+\r
+  @retval TRUE                   Question's value has changed.\r
+  @retval FALSE                  Question's value has not changed\r
+\r
+**/\r
+BOOLEAN\r
+IsQuestionValueChanged (\r
+  IN FORM_BROWSER_FORMSET             *FormSet,\r
+  IN FORM_BROWSER_FORM                *Form,\r
+  IN OUT FORM_BROWSER_STATEMENT       *Question,\r
+  IN GET_SET_QUESTION_VALUE_WITH      GetValueFrom\r
+  )\r
+{\r
+  EFI_HII_VALUE    BackUpValue;\r
+  CHAR8            *BackUpBuffer;\r
+  EFI_HII_VALUE    BackUpValue2;\r
+  CHAR8            *BackUpBuffer2;\r
+  EFI_STATUS       Status;\r
+  BOOLEAN          ValueChanged;\r
+  UINTN            BufferWidth;\r
+\r
+  //\r
+  // For quetion without storage, always mark it as data not changed.\r
+  //\r
+  if (Question->Storage == NULL && Question->Operand != EFI_IFR_TIME_OP && Question->Operand != EFI_IFR_DATE_OP) {\r
+    return FALSE;\r
+  }\r
+\r
+  BackUpBuffer = NULL;\r
+  BackUpBuffer2 = NULL;\r
+  ValueChanged = FALSE;\r
+\r
+  switch (Question->Operand) {\r
+    case EFI_IFR_ORDERED_LIST_OP:\r
+      BufferWidth  = Question->StorageWidth;\r
+      BackUpBuffer = AllocateCopyPool (BufferWidth, Question->BufferValue);\r
+      ASSERT (BackUpBuffer != NULL);\r
+      break;\r
+\r
+    case EFI_IFR_STRING_OP:\r
+    case EFI_IFR_PASSWORD_OP:\r
+      BufferWidth  = (UINTN) Question->Maximum * sizeof (CHAR16);\r
+      BackUpBuffer = AllocateCopyPool (BufferWidth, Question->BufferValue);\r
+      ASSERT (BackUpBuffer != NULL);\r
+      break;\r
+\r
+    default:\r
+      BufferWidth = 0;\r
+      break;\r
+  }\r
+  CopyMem (&BackUpValue, &Question->HiiValue, sizeof (EFI_HII_VALUE));\r
+\r
+  if (GetValueFrom == GetSetValueWithBothBuffer) {\r
+    Status = GetQuestionValue (FormSet, Form, Question, GetSetValueWithEditBuffer);\r
+    ASSERT_EFI_ERROR(Status);\r
+\r
+    switch (Question->Operand) {\r
+      case EFI_IFR_ORDERED_LIST_OP:\r
+        BufferWidth  = Question->StorageWidth;\r
+        BackUpBuffer2 = AllocateCopyPool (BufferWidth, Question->BufferValue);\r
+        ASSERT (BackUpBuffer2 != NULL);\r
+        break;\r
+\r
+      case EFI_IFR_STRING_OP:\r
+      case EFI_IFR_PASSWORD_OP:\r
+        BufferWidth  = (UINTN) Question->Maximum * sizeof (CHAR16);\r
+        BackUpBuffer2 = AllocateCopyPool (BufferWidth, Question->BufferValue);\r
+        ASSERT (BackUpBuffer2 != NULL);\r
+        break;\r
+\r
+      default:\r
+        BufferWidth = 0;\r
+        break;\r
     }\r
-    \r
-    //\r
-    // Free resources, and restore gOldFormSet and gClassOfVfr\r
-    //\r
-    FreePool (HiiHandles);\r
-    gOldFormSet = BackUpFormSet;\r
-       \r
-    //\r
-    // Set Default Value for each FormSet in the maintain list.\r
-    //\r
-    Link = GetFirstNode (&gBrowserFormSetList);\r
-    while (!IsNull (&gBrowserFormSetList, Link)) {\r
-      LocalFormSet = FORM_BROWSER_FORMSET_FROM_LINK (Link);\r
-      Link = GetNextNode (&gBrowserFormSetList, Link);\r
-      if (!ValidateFormSet(LocalFormSet)) {\r
-        continue;\r
-      }\r
-      ExtractDefault (LocalFormSet, NULL, DefaultId, FormSetLevel, GetDefaultValueScope, Storage, RetrieveValueFirst);\r
+    CopyMem (&BackUpValue2, &Question->HiiValue, sizeof (EFI_HII_VALUE));\r
+\r
+    Status = GetQuestionValue (FormSet, Form, Question, GetSetValueWithBuffer);\r
+    ASSERT_EFI_ERROR(Status);\r
+\r
+    if (CompareMem (&BackUpValue2, &Question->HiiValue, sizeof (EFI_HII_VALUE)) != 0 ||\r
+        CompareMem (BackUpBuffer2, Question->BufferValue, BufferWidth) != 0) {\r
+      ValueChanged = TRUE;\r
+    }\r
+  } else {\r
+    Status = GetQuestionValue (FormSet, Form, Question, GetValueFrom);\r
+    ASSERT_EFI_ERROR(Status);\r
+\r
+    if (CompareMem (&BackUpValue, &Question->HiiValue, sizeof (EFI_HII_VALUE)) != 0 ||\r
+        CompareMem (BackUpBuffer, Question->BufferValue, BufferWidth) != 0) {\r
+      ValueChanged = TRUE;\r
     }\r
   }\r
 \r
-  return EFI_SUCCESS;\r
+  CopyMem (&Question->HiiValue, &BackUpValue, sizeof (EFI_HII_VALUE));\r
+  if (BackUpBuffer != NULL) {\r
+    CopyMem (Question->BufferValue, BackUpBuffer, BufferWidth);\r
+    FreePool (BackUpBuffer);\r
+  }\r
+\r
+  if (BackUpBuffer2 != NULL) {\r
+    FreePool (BackUpBuffer2);\r
+  }\r
+\r
+  Question->ValueChanged = ValueChanged;\r
+\r
+  return ValueChanged;\r
 }\r
 \r
 /**\r
@@ -3344,8 +4287,6 @@ LoadFormConfig (
   EFI_STATUS                  Status;\r
   LIST_ENTRY                  *Link;\r
   FORM_BROWSER_STATEMENT      *Question;\r
-  UINT8                       *BufferValue;\r
-  UINTN                       StorageWidth;\r
   \r
   Link = GetFirstNode (&Form->StatementListHead);\r
   while (!IsNull (&Form->StatementListHead, Link)) {\r
@@ -3354,7 +4295,11 @@ LoadFormConfig (
     //\r
     // Initialize local copy of Value for each Question\r
     //\r
-    Status = GetQuestionValue (FormSet, Form, Question, GetSetValueWithEditBuffer);\r
+    if (Question->Operand == EFI_IFR_PASSWORD_OP && (Question->QuestionFlags & EFI_IFR_FLAG_CALLBACK)== 0) {\r
+      Status = GetQuestionValue (FormSet, Form, Question, GetSetValueWithHiiDriver);\r
+    } else {\r
+      Status = GetQuestionValue (FormSet, Form, Question, GetSetValueWithEditBuffer);\r
+    }\r
     if (EFI_ERROR (Status)) {\r
       return Status;\r
     }\r
@@ -3363,37 +4308,6 @@ LoadFormConfig (
       HiiSetString (FormSet->HiiHandle, Question->HiiValue.Value.string, (CHAR16*)Question->BufferValue, NULL);\r
     }\r
 \r
-    //\r
-    // Call the Retrieve call back function for all questions.\r
-    //\r
-    if ((FormSet->ConfigAccess != NULL) && (Selection != NULL) &&\r
-        ((Question->QuestionFlags & EFI_IFR_FLAG_CALLBACK) == EFI_IFR_FLAG_CALLBACK)) {\r
-      //\r
-      // Check QuestionValue does exist.\r
-      //\r
-      StorageWidth = Question->StorageWidth;\r
-      if (Question->BufferValue != NULL) {\r
-        BufferValue  = Question->BufferValue;\r
-      } else {\r
-        BufferValue = (UINT8 *) &Question->HiiValue.Value;\r
-      }\r
-\r
-      //\r
-      // For efivarstore storage, initial question value first.\r
-      //\r
-      if ((Question->Storage != NULL) && (Question->Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE)) {\r
-        Status = gRT->GetVariable (\r
-                         Question->VariableName,\r
-                         &Question->Storage->Guid,\r
-                         NULL,\r
-                         &StorageWidth,\r
-                         BufferValue\r
-                         );\r
-      }\r
-\r
-      Status = ProcessCallBackFunction(Selection, Question, EFI_BROWSER_ACTION_RETRIEVE, TRUE);\r
-    }\r
-\r
     Link = GetNextNode (&Form->StatementListHead, Link);\r
   }\r
 \r
@@ -3437,6 +4351,11 @@ LoadFormSetConfig (
     Link = GetNextNode (&FormSet->FormListHead, Link);\r
   }\r
 \r
+  //\r
+  // Finished question initialization.\r
+  // \r
+  FormSet->QuestionInited = TRUE;\r
+\r
   return EFI_SUCCESS;\r
 }\r
 \r
@@ -3491,6 +4410,13 @@ RemoveConfigRequest (
   CHAR16       *NextRequestElement;\r
   CHAR16       *SearchKey;\r
 \r
+  //\r
+  // No request element in it, just return.\r
+  //\r
+  if (ConfigRequest == NULL) {\r
+    return;\r
+  }\r
+\r
   if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
     //\r
     // "&Name1&Name2" section for EFI_HII_VARSTORE_NAME_VALUE storage\r
@@ -3571,16 +4497,20 @@ CleanBrowserStorage (
     Storage = FORMSET_STORAGE_FROM_LINK (Link);\r
     Link = GetNextNode (&FormSet->StorageListHead, Link);\r
 \r
-    if ((Storage->BrowserStorage->Type != EFI_HII_VARSTORE_BUFFER) && \r
-        (Storage->BrowserStorage->Type != EFI_HII_VARSTORE_NAME_VALUE)) {\r
-      continue;\r
-    }\r
+    if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
+      if (Storage->ConfigRequest == NULL || Storage->BrowserStorage->ConfigRequest == NULL) {\r
+        continue;\r
+      }\r
 \r
-    if (Storage->ConfigRequest == NULL || Storage->BrowserStorage->ConfigRequest == NULL) {\r
-      continue;\r
+      RemoveConfigRequest (Storage->BrowserStorage, Storage->ConfigRequest);\r
+    } else if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_BUFFER ||\r
+               Storage->BrowserStorage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+      if (Storage->BrowserStorage->ConfigRequest != NULL) { \r
+        FreePool (Storage->BrowserStorage->ConfigRequest);\r
+        Storage->BrowserStorage->ConfigRequest = NULL;\r
+      }\r
+      Storage->BrowserStorage->Initialized = FALSE;\r
     }\r
-\r
-    RemoveConfigRequest (Storage->BrowserStorage, Storage->ConfigRequest);\r
   }\r
 }\r
 \r
@@ -3651,7 +4581,8 @@ AppendConfigRequest (
   Adjust the config request info, remove the request elements which already in AllConfigRequest string.\r
 \r
   @param  Storage                Form set Storage.\r
-  @param  ConfigRequest          Return the ConfigRequest info.\r
+  @param  Request                The input request string.\r
+  @param  RespString             Whether the input is ConfigRequest or ConfigResp format.\r
 \r
   @retval TRUE                   Has element not covered by current used elements, need to continue to call ExtractConfig\r
   @retval FALSE                  All elements covered by current used elements.\r
@@ -3659,28 +4590,37 @@ AppendConfigRequest (
 **/\r
 BOOLEAN \r
 ConfigRequestAdjust (\r
-  IN  FORMSET_STORAGE         *Storage,\r
-  OUT CHAR16                  **ConfigRequest\r
+  IN  BROWSER_STORAGE         *Storage,\r
+  IN  CHAR16                  *Request,\r
+  IN  BOOLEAN                 RespString\r
   )\r
 {\r
   CHAR16       *RequestElement;\r
   CHAR16       *NextRequestElement;\r
-  CHAR16       *RetBuf;\r
+  CHAR16       *NextElementBakup;\r
   UINTN        SpareBufLen;\r
   CHAR16       *SearchKey;\r
+  CHAR16       *ValueKey;\r
   BOOLEAN      RetVal;\r
+  CHAR16       *ConfigRequest;\r
 \r
   SpareBufLen    = 0;\r
-  RetBuf         = NULL;\r
   RetVal         = FALSE;\r
+  NextElementBakup = NULL;\r
+  ValueKey         = NULL;\r
+\r
+  if (Request != NULL) {\r
+    ConfigRequest = Request;\r
+  } else {\r
+    ConfigRequest = Storage->ConfigRequest;\r
+  }\r
 \r
-  if (Storage->BrowserStorage->ConfigRequest == NULL) {\r
-    Storage->BrowserStorage->ConfigRequest = AllocateCopyPool (StrSize (Storage->ConfigRequest), Storage->ConfigRequest);\r
-    *ConfigRequest = AllocateCopyPool (StrSize (Storage->ConfigRequest), Storage->ConfigRequest);\r
+  if (Storage->ConfigRequest == NULL) {\r
+    Storage->ConfigRequest = AllocateCopyPool (StrSize (ConfigRequest), ConfigRequest);\r
     return TRUE;\r
   }\r
 \r
-  if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+  if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
     //\r
     // "&Name1&Name2" section for EFI_HII_VARSTORE_NAME_VALUE storage\r
     //\r
@@ -3690,26 +4630,22 @@ ConfigRequestAdjust (
     // "&OFFSET=####&WIDTH=####" section for EFI_HII_VARSTORE_BUFFER storage\r
     //\r
     SearchKey = L"&OFFSET";\r
+    ValueKey  = L"&VALUE";\r
   }\r
 \r
-  //\r
-  // Prepare the config header.\r
-  // \r
-  RetBuf = AllocateCopyPool(StrSize (Storage->BrowserStorage->ConfigHdr), Storage->BrowserStorage->ConfigHdr);\r
-  ASSERT (RetBuf != NULL);\r
-\r
   //\r
   // Find SearchKey storage\r
   //\r
-  if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
-    RequestElement = StrStr (Storage->ConfigRequest, L"PATH");\r
+  if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    RequestElement = StrStr (ConfigRequest, L"PATH");\r
     ASSERT (RequestElement != NULL);\r
     RequestElement = StrStr (RequestElement, SearchKey);    \r
   } else {\r
-    RequestElement = StrStr (Storage->ConfigRequest, SearchKey);\r
+    RequestElement = StrStr (ConfigRequest, SearchKey);\r
   }\r
 \r
   while (RequestElement != NULL) {\r
+\r
     //\r
     // +1 to avoid find header itself.\r
     //\r
@@ -3719,18 +4655,32 @@ ConfigRequestAdjust (
     // The last Request element in configRequest string.\r
     //\r
     if (NextRequestElement != NULL) {\r
+      if (RespString && (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+        NextElementBakup = NextRequestElement;\r
+        NextRequestElement = StrStr (RequestElement, ValueKey);\r
+        ASSERT (NextRequestElement != NULL);\r
+      }\r
       //\r
       // Replace "&" with '\0'.\r
       //\r
       *NextRequestElement = L'\0';\r
+    } else {\r
+      if (RespString && (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+        NextElementBakup = NextRequestElement;\r
+        NextRequestElement = StrStr (RequestElement, ValueKey);\r
+        ASSERT (NextRequestElement != NULL);\r
+        //\r
+        // Replace "&" with '\0'.\r
+        //\r
+        *NextRequestElement = L'\0';\r
+      }\r
     }\r
  \r
-    if (!ElementValidation (Storage->BrowserStorage, RequestElement)) {\r
+    if (!ElementValidation (Storage, RequestElement)) {\r
       //\r
       // Add this element to the Storage->BrowserStorage->AllRequestElement.\r
       //\r
-      AppendConfigRequest(&Storage->BrowserStorage->ConfigRequest, &Storage->BrowserStorage->SpareStrLen, RequestElement);\r
-      AppendConfigRequest (&RetBuf, &SpareBufLen, RequestElement);\r
+      AppendConfigRequest(&Storage->ConfigRequest, &Storage->SpareStrLen, RequestElement);\r
       RetVal = TRUE;\r
     }\r
 \r
@@ -3741,16 +4691,166 @@ ConfigRequestAdjust (
       *NextRequestElement = L'&';\r
     }\r
 \r
-    RequestElement = NextRequestElement;\r
+    if (RespString && (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+      RequestElement = NextElementBakup;\r
+    } else {\r
+      RequestElement = NextRequestElement;\r
+    }\r
   }\r
 \r
-  if (RetVal) {\r
-    *ConfigRequest = RetBuf;\r
-  } else {\r
-    FreePool (RetBuf);\r
+  return RetVal;\r
+}\r
+\r
+/**\r
+\r
+  Base on ConfigRequest info to get default value for current formset. \r
+\r
+  ConfigRequest info include the info about which questions in current formset need to \r
+  get default value. This function only get these questions default value.\r
+  \r
+  @param  FormSet                FormSet data structure.\r
+  @param  Storage                Storage need to update value.\r
+  @param  ConfigRequest          The config request string.\r
+\r
+**/\r
+VOID\r
+GetDefaultForFormset (\r
+  IN FORM_BROWSER_FORMSET    *FormSet,\r
+  IN BROWSER_STORAGE         *Storage,\r
+  IN CHAR16                  *ConfigRequest\r
+  )\r
+{\r
+  UINT8             *BackUpBuf;\r
+  UINTN             BufferSize;\r
+  LIST_ENTRY        BackUpList;\r
+  NAME_VALUE_NODE   *Node;\r
+  LIST_ENTRY        *Link;\r
+  LIST_ENTRY        *NodeLink;\r
+  NAME_VALUE_NODE   *TmpNode;\r
+  EFI_STATUS        Status;\r
+  EFI_STRING        Progress;\r
+  EFI_STRING        Result;\r
+\r
+  BackUpBuf = NULL;\r
+  InitializeListHead(&BackUpList);\r
+\r
+  //\r
+  // Back update the edit buffer.\r
+  // \r
+  if (Storage->Type == EFI_HII_VARSTORE_BUFFER || \r
+      (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+    BackUpBuf = AllocateCopyPool (Storage->Size, Storage->EditBuffer);\r
+    ASSERT (BackUpBuf != NULL);\r
+  } else if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    Link = GetFirstNode (&Storage->NameValueListHead);\r
+    while (!IsNull (&Storage->NameValueListHead, Link)) {\r
+      Node = NAME_VALUE_NODE_FROM_LINK (Link);\r
+      Link = GetNextNode (&Storage->NameValueListHead, Link);\r
+\r
+      //\r
+      // Only back Node belong to this formset.\r
+      //\r
+      if (StrStr (Storage->ConfigRequest, Node->Name) == NULL) {\r
+        continue;\r
+      }\r
+\r
+      TmpNode = AllocateCopyPool (sizeof (NAME_VALUE_NODE), Node);\r
+      ASSERT (TmpNode != NULL);\r
+      TmpNode->Name = AllocateCopyPool (StrSize(Node->Name) * sizeof (CHAR16), Node->Name);\r
+      ASSERT (TmpNode->Name != NULL);\r
+      TmpNode->EditValue = AllocateCopyPool (StrSize(Node->EditValue) * sizeof (CHAR16), Node->EditValue);\r
+      ASSERT (TmpNode->EditValue != NULL);\r
+\r
+      InsertTailList(&BackUpList, &TmpNode->Link);\r
+    }\r
   }\r
 \r
-  return RetVal;\r
+  //\r
+  // Get default value.\r
+  //\r
+  ExtractDefault (FormSet, NULL, EFI_HII_DEFAULT_CLASS_STANDARD, FormSetLevel, GetDefaultForStorage, Storage, TRUE);\r
+\r
+  //\r
+  // Update the question value based on the input ConfigRequest.\r
+  //\r
+  if (Storage->Type == EFI_HII_VARSTORE_BUFFER || \r
+      (Storage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER)) {\r
+    ASSERT (BackUpBuf != NULL);\r
+    BufferSize = Storage->Size;\r
+    Status = mHiiConfigRouting->BlockToConfig(\r
+                                  mHiiConfigRouting,\r
+                                  ConfigRequest,\r
+                                  Storage->EditBuffer,\r
+                                  BufferSize,\r
+                                  &Result,\r
+                                  &Progress\r
+                                  );\r
+    ASSERT_EFI_ERROR (Status);\r
+    \r
+    Status = mHiiConfigRouting->ConfigToBlock (\r
+                                  mHiiConfigRouting,\r
+                                  Result,\r
+                                  BackUpBuf,\r
+                                  &BufferSize,\r
+                                  &Progress\r
+                                  );\r
+    ASSERT_EFI_ERROR (Status);\r
+\r
+    if (Result != NULL) {\r
+      FreePool (Result);\r
+    }\r
+    \r
+    CopyMem (Storage->EditBuffer, BackUpBuf, Storage->Size);\r
+    FreePool (BackUpBuf);\r
+  } else if (Storage->Type == EFI_HII_VARSTORE_NAME_VALUE) {\r
+    //\r
+    // Update question value, only element in ConfigReqeust will be update.\r
+    //\r
+    Link = GetFirstNode (&BackUpList);\r
+    while (!IsNull (&BackUpList, Link)) {\r
+      Node = NAME_VALUE_NODE_FROM_LINK (Link);\r
+      Link = GetNextNode (&BackUpList, Link);\r
+\r
+      if (StrStr (ConfigRequest, Node->Name) != NULL) {\r
+        continue;\r
+      }\r
+\r
+      NodeLink = GetFirstNode (&Storage->NameValueListHead);\r
+      while (!IsNull (&Storage->NameValueListHead, NodeLink)) {\r
+        TmpNode  = NAME_VALUE_NODE_FROM_LINK (NodeLink);\r
+        NodeLink = GetNextNode (&Storage->NameValueListHead, NodeLink);\r
+      \r
+        if (StrCmp (Node->Name, TmpNode->Name) != 0) {\r
+          continue;\r
+        }\r
+\r
+        FreePool (TmpNode->EditValue);\r
+        TmpNode->EditValue = AllocateCopyPool (StrSize(Node->EditValue) * sizeof (CHAR16), Node->EditValue);\r
+\r
+        RemoveEntryList (&Node->Link);\r
+        FreePool (Node->EditValue);\r
+        FreePool (Node->Name);\r
+        FreePool (Node);\r
+      }\r
+    }\r
+\r
+    //\r
+    // Restore the Name/Value node.\r
+    //  \r
+    Link = GetFirstNode (&BackUpList);\r
+    while (!IsNull (&BackUpList, Link)) {\r
+      Node = NAME_VALUE_NODE_FROM_LINK (Link);\r
+      Link = GetNextNode (&BackUpList, Link);\r
\r
+      //\r
+      // Free this node.\r
+      //\r
+      RemoveEntryList (&Node->Link);\r
+      FreePool (Node->EditValue);\r
+      FreePool (Node->Name);\r
+      FreePool (Node);\r
+    }\r
+  }\r
 }\r
 \r
 /**\r
@@ -3759,10 +4859,8 @@ ConfigRequestAdjust (
   @param  FormSet                FormSet data structure.\r
   @param  Storage                Buffer Storage.\r
 \r
-  @retval EFI_SUCCESS            The function completed successfully.\r
-\r
 **/\r
-EFI_STATUS\r
+VOID\r
 LoadStorage (\r
   IN FORM_BROWSER_FORMSET    *FormSet,\r
   IN FORMSET_STORAGE         *Storage\r
@@ -3772,74 +4870,193 @@ LoadStorage (
   EFI_STRING  Progress;\r
   EFI_STRING  Result;\r
   CHAR16      *StrPtr;\r
-  CHAR16      *ConfigRequest;\r
+  EFI_STRING  ConfigRequest;\r
+  UINTN       StrLen;\r
 \r
-  if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_EFI_VARIABLE) {\r
-    return EFI_SUCCESS;\r
+  ConfigRequest = NULL;\r
+\r
+  switch (Storage->BrowserStorage->Type) {\r
+    case EFI_HII_VARSTORE_EFI_VARIABLE:\r
+      return;\r
+\r
+    case EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER:\r
+      if (Storage->BrowserStorage->ConfigRequest != NULL) {\r
+        ConfigRequestAdjust(Storage->BrowserStorage, Storage->ConfigRequest, FALSE);\r
+        return;\r
+      }\r
+      break;\r
+\r
+    case EFI_HII_VARSTORE_BUFFER:\r
+    case EFI_HII_VARSTORE_NAME_VALUE:\r
+      //\r
+      // Skip if there is no RequestElement.\r
+      //\r
+      if (Storage->ElementCount == 0) {\r
+        return;\r
+      }\r
+\r
+      //\r
+      // Just update the ConfigRequest, if storage already initialized. \r
+      //\r
+      if (Storage->BrowserStorage->Initialized) {\r
+        ConfigRequestAdjust(Storage->BrowserStorage, Storage->ConfigRequest, FALSE);\r
+        return;\r
+      }\r
+\r
+      Storage->BrowserStorage->Initialized = TRUE;\r
+      break;\r
+\r
+    default:\r
+      return;\r
   }\r
 \r
-  if (Storage->BrowserStorage->Type == EFI_HII_VARSTORE_EFI_VARIABLE_BUFFER) {\r
-    Status = EFI_SUCCESS;\r
+  if (Storage->BrowserStorage->Type != EFI_HII_VARSTORE_NAME_VALUE) {\r
     //\r
-    // EFI varstore data all get from variable, so no need to get again.\r
+    // Create the config request string to get all fields for this storage.\r
+    // Allocate and fill a buffer large enough to hold the <ConfigHdr> template\r
+    // followed by "&OFFSET=0&WIDTH=WWWW"followed by a Null-terminator\r
     //\r
-    if (Storage->BrowserStorage->ReferenceCount == 1) {\r
-      Status = gRT->GetVariable (\r
-                       Storage->BrowserStorage->Name,\r
-                       &Storage->BrowserStorage->Guid,\r
-                       NULL,\r
-                       (UINTN*)&Storage->BrowserStorage->Size,\r
-                       Storage->BrowserStorage->EditBuffer\r
-                       );\r
-    }\r
-    return Status;\r
+    StrLen = StrSize (Storage->BrowserStorage->ConfigHdr) + 20 * sizeof (CHAR16);\r
+    ConfigRequest = AllocateZeroPool (StrLen);\r
+    ASSERT (ConfigRequest != NULL);\r
+    UnicodeSPrint (\r
+               ConfigRequest, \r
+               StrLen, \r
+               L"%s&OFFSET=0&WIDTH=%04x", \r
+               Storage->BrowserStorage->ConfigHdr,\r
+               Storage->BrowserStorage->Size);\r
+  } else {\r
+    ConfigRequest = Storage->ConfigRequest;\r
   }\r
 \r
-  if (FormSet->ConfigAccess == NULL) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
+  //\r
+  // Request current settings from Configuration Driver\r
+  //\r
+  Status = mHiiConfigRouting->ExtractConfig (\r
+                                    mHiiConfigRouting,\r
+                                    ConfigRequest,\r
+                                    &Progress,\r
+                                    &Result\r
+                                    );\r
 \r
-  if (Storage->ElementCount == 0) {\r
+  //\r
+  // If get value fail, extract default from IFR binary\r
+  //\r
+  if (EFI_ERROR (Status)) {\r
+    ExtractDefault (FormSet, NULL, EFI_HII_DEFAULT_CLASS_STANDARD, FormSetLevel, GetDefaultForStorage, Storage->BrowserStorage, TRUE);\r
+  } else {\r
     //\r
-    // Skip if there is no RequestElement\r
+    // Convert Result from <ConfigAltResp> to <ConfigResp>\r
     //\r
-    return EFI_SUCCESS;\r
+    StrPtr = StrStr (Result, L"&GUID=");\r
+    if (StrPtr != NULL) {\r
+      *StrPtr = L'\0';\r
+    }\r
+    \r
+    Status = ConfigRespToStorage (Storage->BrowserStorage, Result);\r
+    FreePool (Result);\r
   }\r
 \r
+  Storage->BrowserStorage->ConfigRequest = AllocateCopyPool (StrSize (Storage->ConfigRequest), Storage->ConfigRequest);\r
+\r
   //\r
-  // Adjust the ConfigRequest string, only the field not saved in BrowserStorage->AllConfig\r
-  // will used to call ExtractConfig.\r
+  // Input NULL for ConfigRequest field means sync all fields from editbuffer to buffer. \r
   //\r
-  if (!ConfigRequestAdjust(Storage, &ConfigRequest)) {\r
-    return EFI_SUCCESS;\r
+  SynchronizeStorage(Storage->BrowserStorage, NULL, TRUE);\r
+\r
+  if (Storage->BrowserStorage->Type != EFI_HII_VARSTORE_NAME_VALUE) {\r
+    if (ConfigRequest != NULL) {\r
+      FreePool (ConfigRequest);\r
+    }\r
   }\r
+}\r
+\r
+/**\r
+  Get Value changed status from old question.\r
+\r
+  @param  NewFormSet                FormSet data structure.\r
+  @param  OldQuestion               Old question which has value changed.\r
+\r
+**/\r
+VOID\r
+SyncStatusForQuestion (\r
+  IN OUT FORM_BROWSER_FORMSET             *NewFormSet,\r
+  IN     FORM_BROWSER_STATEMENT           *OldQuestion\r
+  )\r
+{\r
+  LIST_ENTRY                  *Link;\r
+  LIST_ENTRY                  *QuestionLink;\r
+  FORM_BROWSER_FORM           *Form;\r
+  FORM_BROWSER_STATEMENT      *Question;\r
 \r
   //\r
-  // Request current settings from Configuration Driver\r
+  // For each form in one formset.\r
   //\r
-  Status = FormSet->ConfigAccess->ExtractConfig (\r
-                                    FormSet->ConfigAccess,\r
-                                    ConfigRequest,\r
-                                    &Progress,\r
-                                    &Result\r
-                                    );\r
-  FreePool (ConfigRequest);\r
+  Link = GetFirstNode (&NewFormSet->FormListHead);\r
+  while (!IsNull (&NewFormSet->FormListHead, Link)) {\r
+    Form = FORM_BROWSER_FORM_FROM_LINK (Link);\r
+    Link = GetNextNode (&NewFormSet->FormListHead, Link);\r
 \r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
+    //\r
+    // for each question in one form.\r
+    //\r
+    QuestionLink = GetFirstNode (&Form->StatementListHead);\r
+    while (!IsNull (&Form->StatementListHead, QuestionLink)) {\r
+      Question = FORM_BROWSER_STATEMENT_FROM_LINK (QuestionLink);\r
+      QuestionLink = GetNextNode (&Form->StatementListHead, QuestionLink);\r
+\r
+      if (Question->QuestionId == OldQuestion->QuestionId) {\r
+        Question->ValueChanged = TRUE;\r
+        return;\r
+      }\r
+    }\r
   }\r
+}\r
+\r
+/**\r
+  Get Value changed status from old formset.\r
+\r
+  @param  NewFormSet                FormSet data structure.\r
+  @param  OldFormSet                FormSet data structure.\r
+\r
+**/\r
+VOID\r
+SyncStatusForFormSet (\r
+  IN OUT FORM_BROWSER_FORMSET             *NewFormSet,\r
+  IN     FORM_BROWSER_FORMSET             *OldFormSet\r
+  )\r
+{\r
+  LIST_ENTRY                  *Link;\r
+  LIST_ENTRY                  *QuestionLink;\r
+  FORM_BROWSER_FORM           *Form;\r
+  FORM_BROWSER_STATEMENT      *Question;\r
 \r
   //\r
-  // Convert Result from <ConfigAltResp> to <ConfigResp>\r
+  // For each form in one formset.\r
   //\r
-  StrPtr = StrStr (Result, L"&GUID=");\r
-  if (StrPtr != NULL) {\r
-    *StrPtr = L'\0';\r
-  }\r
+  Link = GetFirstNode (&OldFormSet->FormListHead);\r
+  while (!IsNull (&OldFormSet->FormListHead, Link)) {\r
+    Form = FORM_BROWSER_FORM_FROM_LINK (Link);\r
+    Link = GetNextNode (&OldFormSet->FormListHead, Link);\r
 \r
-  Status = ConfigRespToStorage (Storage->BrowserStorage, Result);\r
-  FreePool (Result);\r
-  return Status;\r
+    //\r
+    // for each question in one form.\r
+    //\r
+    QuestionLink = GetFirstNode (&Form->StatementListHead);\r
+    while (!IsNull (&Form->StatementListHead, QuestionLink)) {\r
+      Question = FORM_BROWSER_STATEMENT_FROM_LINK (QuestionLink);\r
+      QuestionLink = GetNextNode (&Form->StatementListHead, QuestionLink);\r
+\r
+      if (!Question->ValueChanged) {\r
+        continue;\r
+      }\r
+\r
+      //\r
+      // Find the same question in new formset and update the value changed flag.\r
+      //\r
+      SyncStatusForQuestion (NewFormSet, Question);\r
+    }\r
+  }\r
 }\r
 \r
 /**\r
@@ -3847,22 +5064,27 @@ LoadStorage (
 \r
   @param  FormSet                FormSet data structure.\r
 \r
-  @retval EFI_SUCCESS            The function completed successfully.\r
-\r
 **/\r
-EFI_STATUS\r
+VOID\r
 InitializeCurrentSetting (\r
   IN OUT FORM_BROWSER_FORMSET             *FormSet\r
   )\r
 {\r
   LIST_ENTRY              *Link;\r
-  LIST_ENTRY              *Link2;\r
   FORMSET_STORAGE         *Storage;\r
-  FORMSET_STORAGE         *StorageSrc;\r
-  FORMSET_STORAGE         *OldStorage;\r
-  FORM_BROWSER_FORM       *Form;\r
-  FORM_BROWSER_FORM       *Form2;\r
-  EFI_STATUS              Status;\r
+  FORM_BROWSER_FORMSET    *OldFormSet;\r
+\r
+  //\r
+  // Try to find pre FormSet in the maintain backup list.\r
+  // If old formset != NULL, destroy this formset. Add new formset to gBrowserFormSetList.\r
+  //\r
+  OldFormSet = GetFormSetFromHiiHandle (FormSet->HiiHandle);\r
+  if (OldFormSet != NULL) {\r
+    SyncStatusForFormSet (FormSet, OldFormSet);\r
+    RemoveEntryList (&OldFormSet->Link);\r
+    DestroyFormSet (OldFormSet);\r
+  }\r
+  InsertTailList (&gBrowserFormSetList, &FormSet->Link);\r
 \r
   //\r
   // Extract default from IFR binary for no storage questions.\r
@@ -3876,77 +5098,10 @@ InitializeCurrentSetting (
   while (!IsNull (&FormSet->StorageListHead, Link)) {\r
     Storage = FORMSET_STORAGE_FROM_LINK (Link);\r
 \r
-    OldStorage = NULL;\r
-    if (gOldFormSet != NULL) {\r
-      //\r
-      // Try to find the Storage in backup formset gOldFormSet\r
-      //\r
-      Link2 = GetFirstNode (&gOldFormSet->StorageListHead);\r
-      while (!IsNull (&gOldFormSet->StorageListHead, Link2)) {\r
-        StorageSrc = FORMSET_STORAGE_FROM_LINK (Link2);\r
-\r
-        if (StorageSrc->VarStoreId == Storage->VarStoreId) {\r
-          OldStorage = StorageSrc;\r
-          break;\r
-        }\r
-\r
-        Link2 = GetNextNode (&gOldFormSet->StorageListHead, Link2);\r
-      }\r
-    }\r
-\r
-    //\r
-    // Storage is not found in backup formset and current global storage not has other driver used,\r
-    // request it from ConfigDriver\r
-    //\r
-    if (OldStorage == NULL) {\r
-      Status = LoadStorage (FormSet, Storage);\r
-\r
-      if (EFI_ERROR (Status)) {\r
-        //\r
-        // If get last time changed value failed, extract default from IFR binary\r
-        //\r
-        ExtractDefault (FormSet, NULL, EFI_HII_DEFAULT_CLASS_STANDARD, FormSetLevel, GetDefaultForStorage, Storage->BrowserStorage, TRUE);\r
-        //\r
-        // ExtractDefault will set the NV flag to TRUE, so need this function to clean the flag\r
-        // in current situation.\r
-        //\r
-        UpdateNvInfoInForm (FormSet, FALSE);\r
-      }\r
-\r
-      //\r
-      // Now Edit Buffer is filled with default values(lower priority) or current\r
-      // settings(higher priority), sychronize it to shadow Buffer\r
-      //\r
-      SynchronizeStorage (Storage->BrowserStorage, TRUE);\r
-    }\r
+    LoadStorage (FormSet, Storage);\r
 \r
     Link = GetNextNode (&FormSet->StorageListHead, Link);\r
   }\r
-\r
-  //\r
-  // If has old formset, get the old nv update status.\r
-  //\r
-  if (gOldFormSet != NULL) {\r
-    Link = GetFirstNode (&FormSet->FormListHead);\r
-    while (!IsNull (&FormSet->FormListHead, Link)) {\r
-      Form = FORM_BROWSER_FORM_FROM_LINK (Link);\r
-\r
-      Link2 = GetFirstNode (&gOldFormSet->FormListHead);\r
-      while (!IsNull (&gOldFormSet->FormListHead, Link2)) {\r
-        Form2 = FORM_BROWSER_FORM_FROM_LINK (Link2);\r
-\r
-        if (Form->FormId == Form2->FormId) {\r
-          Form->NvUpdateRequired = Form2->NvUpdateRequired;\r
-          break;\r
-        }\r
-\r
-        Link2 = GetNextNode (&gOldFormSet->FormListHead, Link2);\r
-      }\r
-      Link = GetNextNode (&FormSet->FormListHead, Link);\r
-    }\r
-  }\r
-\r
-  return EFI_SUCCESS;\r
 }\r
 \r
 \r
@@ -4129,7 +5284,6 @@ GetIfrBinaryData (
                                  found in package list.\r
                                  On output, GUID of the formset found(if not NULL).\r
   @param  FormSet                FormSet data structure.\r
-  @param  UpdateGlobalVar        Whether need to update the global variable.\r
 \r
   @retval EFI_SUCCESS            The function completed successfully.\r
   @retval EFI_NOT_FOUND          The specified FormSet could not be found.\r
@@ -4139,13 +5293,11 @@ EFI_STATUS
 InitializeFormSet (\r
   IN  EFI_HII_HANDLE                   Handle,\r
   IN OUT EFI_GUID                      *FormSetGuid,\r
-  OUT FORM_BROWSER_FORMSET             *FormSet,\r
-  IN  BOOLEAN                          UpdateGlobalVar                   \r
+  OUT FORM_BROWSER_FORMSET             *FormSet\r
   )\r
 {\r
   EFI_STATUS                Status;\r
   EFI_HANDLE                DriverHandle;\r
-  UINT16                    Index;\r
 \r
   Status = GetIfrBinaryData (Handle, FormSetGuid, &FormSet->IfrBinaryLength, &FormSet->IfrBinaryData);\r
   if (EFI_ERROR (Status)) {\r
@@ -4155,6 +5307,7 @@ InitializeFormSet (
   FormSet->Signature = FORM_BROWSER_FORMSET_SIGNATURE;\r
   FormSet->HiiHandle = Handle;\r
   CopyMem (&FormSet->Guid, FormSetGuid, sizeof (EFI_GUID));\r
+  FormSet->QuestionInited = FALSE;\r
 \r
   //\r
   // Retrieve ConfigAccess Protocol associated with this HiiPackageList\r
@@ -4181,55 +5334,8 @@ InitializeFormSet (
   // Parse the IFR binary OpCodes\r
   //\r
   Status = ParseOpCodes (FormSet);\r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
-  }\r
-\r
-  // \r
-  // If not need to update the global variable, just return.\r
-  //\r
-  if (!UpdateGlobalVar) {\r
-    return Status;\r
-  }\r
-\r
-  //\r
-  // Set VFR type by FormSet SubClass field\r
-  //\r
-  gClassOfVfr = FORMSET_CLASS_PLATFORM_SETUP;\r
-  if (FormSet->SubClass == EFI_FRONT_PAGE_SUBCLASS) {\r
-    gClassOfVfr = FORMSET_CLASS_FRONT_PAGE;\r
-  }\r
-  \r
-  //\r
-  // Set VFR type by FormSet class guid\r
-  //\r
-  for (Index = 0; Index < 3; Index ++) {\r
-    if (CompareGuid (&FormSet->ClassGuid[Index], &gEfiHiiPlatformSetupFormsetGuid)) {\r
-      gClassOfVfr |= FORMSET_CLASS_PLATFORM_SETUP;\r
-      break;\r
-    }\r
-  }\r
-\r
-  gFunctionKeySetting = ENABLE_FUNCTION_KEY_SETTING;\r
-\r
-  if ((gClassOfVfr & FORMSET_CLASS_FRONT_PAGE) == FORMSET_CLASS_FRONT_PAGE) {\r
-    gFrontPageHandle = FormSet->HiiHandle;\r
-    gFunctionKeySetting = NONE_FUNCTION_KEY_SETTING;\r
-  }\r
-\r
-  //\r
-  // Match GUID to find out the function key setting. If match fail, use the default setting.\r
-  //\r
-  for (Index = 0; Index < sizeof (gFunctionKeySettingTable) / sizeof (FUNCTIION_KEY_SETTING); Index++) {\r
-    if (CompareGuid (&FormSet->Guid, &(gFunctionKeySettingTable[Index].FormSetGuid))) {\r
-      //\r
-      // Update the function key setting.\r
-      //\r
-      gFunctionKeySetting = gFunctionKeySettingTable[Index].KeySetting;\r
-    }\r
-  }\r
 \r
-  return EFI_SUCCESS;\r
+  return Status;\r
 }\r
 \r
 \r
@@ -4245,6 +5351,7 @@ SaveBrowserContext (
   )\r
 {\r
   BROWSER_CONTEXT  *Context;\r
+  FORM_ENTRY_INFO     *MenuList;\r
 \r
   gBrowserContextCount++;\r
   if (gBrowserContextCount == 1) {\r
@@ -4262,47 +5369,23 @@ SaveBrowserContext (
   //\r
   // Save FormBrowser context\r
   //\r
-  Context->BannerData           = gBannerData;\r
-  Context->ClassOfVfr           = gClassOfVfr;\r
-  Context->FunctionKeySetting   = gFunctionKeySetting;\r
+  Context->Selection            = gCurrentSelection;\r
   Context->ResetRequired        = gResetRequired;\r
-  Context->Direction            = gDirection;\r
-  Context->EnterString          = gEnterString;\r
-  Context->EnterCommitString    = gEnterCommitString;\r
-  Context->EnterEscapeString    = gEnterEscapeString;\r
-  Context->EscapeString         = gEscapeString;\r
-  Context->MoveHighlight        = gMoveHighlight;\r
-  Context->MakeSelection        = gMakeSelection;\r
-  Context->DecNumericInput      = gDecNumericInput;\r
-  Context->HexNumericInput      = gHexNumericInput;\r
-  Context->ToggleCheckBox       = gToggleCheckBox;\r
-  Context->PromptForData        = gPromptForData;\r
-  Context->PromptForPassword    = gPromptForPassword;\r
-  Context->PromptForNewPassword = gPromptForNewPassword;\r
-  Context->ConfirmPassword      = gConfirmPassword;\r
-  Context->ConfirmError         = gConfirmError;\r
-  Context->PassowordInvalid     = gPassowordInvalid;\r
-  Context->PressEnter           = gPressEnter;\r
-  Context->EmptyString          = gEmptyString;\r
-  Context->AreYouSure           = gAreYouSure;\r
-  Context->YesResponse          = gYesResponse;\r
-  Context->NoResponse           = gNoResponse;\r
-  Context->MiniString           = gMiniString;\r
-  Context->PlusString           = gPlusString;\r
-  Context->MinusString          = gMinusString;\r
-  Context->AdjustNumber         = gAdjustNumber;\r
-  Context->SaveChanges          = gSaveChanges;\r
-  Context->OptionMismatch       = gOptionMismatch;\r
-  Context->FormSuppress         = gFormSuppress;\r
-  Context->PromptBlockWidth     = gPromptBlockWidth;\r
-  Context->OptionBlockWidth     = gOptionBlockWidth;\r
-  Context->HelpBlockWidth       = gHelpBlockWidth;\r
-  Context->OldFormSet           = gOldFormSet;\r
-  Context->MenuRefreshHead      = gMenuRefreshHead;\r
-  Context->ProtocolNotFound     = gProtocolNotFound;\r
-\r
-  CopyMem (&Context->ScreenDimensions, &gScreenDimensions, sizeof (gScreenDimensions));\r
-  CopyMem (&Context->MenuOption, &gMenuOption, sizeof (gMenuOption));\r
+  Context->ExitRequired         = gExitRequired;\r
+  Context->HiiHandle            = mCurrentHiiHandle;\r
+  Context->FormId               = mCurrentFormId;\r
+  CopyGuid (&Context->FormSetGuid, &mCurrentFormSetGuid);\r
+\r
+  //\r
+  // Save the menu history data.\r
+  //\r
+  InitializeListHead(&Context->FormHistoryList);\r
+  while (!IsListEmpty (&mPrivateData.FormBrowserEx2.FormViewHistoryHead)) {\r
+    MenuList = FORM_ENTRY_INFO_FROM_LINK (mPrivateData.FormBrowserEx2.FormViewHistoryHead.ForwardLink);\r
+    RemoveEntryList (&MenuList->Link);\r
+\r
+    InsertTailList(&Context->FormHistoryList, &MenuList->Link);\r
+  }\r
 \r
   //\r
   // Insert to FormBrowser context list\r
@@ -4322,6 +5405,7 @@ RestoreBrowserContext (
 {\r
   LIST_ENTRY       *Link;\r
   BROWSER_CONTEXT  *Context;\r
+  FORM_ENTRY_INFO     *MenuList;\r
 \r
   ASSERT (gBrowserContextCount != 0);\r
   gBrowserContextCount--;\r
@@ -4340,47 +5424,22 @@ RestoreBrowserContext (
   //\r
   // Restore FormBrowser context\r
   //\r
-  gBannerData           = Context->BannerData;\r
-  gClassOfVfr           = Context->ClassOfVfr;\r
-  gFunctionKeySetting   = Context->FunctionKeySetting;\r
+  gCurrentSelection     = Context->Selection;\r
   gResetRequired        = Context->ResetRequired;\r
-  gDirection            = Context->Direction;\r
-  gEnterString          = Context->EnterString;\r
-  gEnterCommitString    = Context->EnterCommitString;\r
-  gEnterEscapeString    = Context->EnterEscapeString;\r
-  gEscapeString         = Context->EscapeString;\r
-  gMoveHighlight        = Context->MoveHighlight;\r
-  gMakeSelection        = Context->MakeSelection;\r
-  gDecNumericInput      = Context->DecNumericInput;\r
-  gHexNumericInput      = Context->HexNumericInput;\r
-  gToggleCheckBox       = Context->ToggleCheckBox;\r
-  gPromptForData        = Context->PromptForData;\r
-  gPromptForPassword    = Context->PromptForPassword;\r
-  gPromptForNewPassword = Context->PromptForNewPassword;\r
-  gConfirmPassword      = Context->ConfirmPassword;\r
-  gConfirmError         = Context->ConfirmError;\r
-  gPassowordInvalid     = Context->PassowordInvalid;\r
-  gPressEnter           = Context->PressEnter;\r
-  gEmptyString          = Context->EmptyString;\r
-  gAreYouSure           = Context->AreYouSure;\r
-  gYesResponse          = Context->YesResponse;\r
-  gNoResponse           = Context->NoResponse;\r
-  gMiniString           = Context->MiniString;\r
-  gPlusString           = Context->PlusString;\r
-  gMinusString          = Context->MinusString;\r
-  gAdjustNumber         = Context->AdjustNumber;\r
-  gSaveChanges          = Context->SaveChanges;\r
-  gOptionMismatch       = Context->OptionMismatch;\r
-  gFormSuppress         = Context->FormSuppress;\r
-  gPromptBlockWidth     = Context->PromptBlockWidth;\r
-  gOptionBlockWidth     = Context->OptionBlockWidth;\r
-  gHelpBlockWidth       = Context->HelpBlockWidth;\r
-  gOldFormSet           = Context->OldFormSet;\r
-  gMenuRefreshHead      = Context->MenuRefreshHead;\r
-  gProtocolNotFound     = Context->ProtocolNotFound;\r
-\r
-  CopyMem (&gScreenDimensions, &Context->ScreenDimensions, sizeof (gScreenDimensions));\r
-  CopyMem (&gMenuOption, &Context->MenuOption, sizeof (gMenuOption));\r
+  gExitRequired         = Context->ExitRequired;\r
+  mCurrentHiiHandle     = Context->HiiHandle;\r
+  mCurrentFormId        = Context->FormId;\r
+  CopyGuid (&mCurrentFormSetGuid, &Context->FormSetGuid);\r
+\r
+  //\r
+  // Restore the menu history data.\r
+  //\r
+  while (!IsListEmpty (&Context->FormHistoryList)) {\r
+    MenuList = FORM_ENTRY_INFO_FROM_LINK (Context->FormHistoryList.ForwardLink);\r
+    RemoveEntryList (&MenuList->Link);\r
+\r
+    InsertTailList(&mPrivateData.FormBrowserEx2.FormViewHistoryHead, &MenuList->Link);\r
+  }\r
 \r
   //\r
   // Remove from FormBrowser context list\r
@@ -4440,7 +5499,7 @@ IsHiiHandleInBrowserContext (
   //\r
   // HiiHandle is Current FormSet.\r
   //\r
-  if ((gOldFormSet != NULL) && (gOldFormSet->HiiHandle == Handle)) {\r
+  if (mCurrentHiiHandle == Handle) {\r
     return TRUE;\r
   }\r
 \r
@@ -4450,7 +5509,7 @@ IsHiiHandleInBrowserContext (
   Link = GetFirstNode (&gBrowserContextList);\r
   while (!IsNull (&gBrowserContextList, Link)) {\r
     Context = BROWSER_CONTEXT_FROM_LINK (Link);\r
-    if (Context->OldFormSet->HiiHandle == Handle) {\r
+    if (Context->HiiHandle == Handle) {\r
       //\r
       // HiiHandle is in BrowserContext\r
       //\r
@@ -4462,6 +5521,97 @@ IsHiiHandleInBrowserContext (
   return FALSE;\r
 }\r
 \r
+/**\r
+  Perform Password check. \r
+  Passwork may be encrypted by driver that requires the specific check.\r
+  \r
+  @param  Form             Form where Password Statement is in.\r
+  @param  Statement        Password statement\r
+  @param  PasswordString   Password string to be checked. It may be NULL.\r
+                           NULL means to restore password.\r
+                           "" string can be used to checked whether old password does exist.\r
+  \r
+  @return Status     Status of Password check.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+PasswordCheck (\r
+  IN FORM_DISPLAY_ENGINE_FORM      *Form,\r
+  IN FORM_DISPLAY_ENGINE_STATEMENT *Statement,\r
+  IN EFI_STRING                    PasswordString  OPTIONAL\r
+  )\r
+{\r
+  EFI_STATUS                      Status;\r
+  EFI_HII_CONFIG_ACCESS_PROTOCOL  *ConfigAccess;\r
+  EFI_BROWSER_ACTION_REQUEST      ActionRequest;\r
+  EFI_IFR_TYPE_VALUE              IfrTypeValue;\r
+  FORM_BROWSER_STATEMENT          *Question;\r
+\r
+  ConfigAccess = gCurrentSelection->FormSet->ConfigAccess;\r
+  Question = GetBrowserStatement(Statement);\r
+  ASSERT (Question != NULL);\r
+\r
+  if ((Question->QuestionFlags & EFI_IFR_FLAG_CALLBACK) == EFI_IFR_FLAG_CALLBACK) {\r
+    if (ConfigAccess == NULL) {\r
+      return EFI_UNSUPPORTED;\r
+    }\r
+  } else {\r
+    if (PasswordString == NULL) {\r
+      return EFI_SUCCESS;\r
+    } \r
+\r
+    //\r
+    // Check whether has preexisted password.\r
+    //\r
+    if (PasswordString[0] == 0) {\r
+      if (*((CHAR16 *) Question->BufferValue) == 0) {\r
+        return EFI_SUCCESS;\r
+      } else {\r
+        return EFI_NOT_READY;\r
+      }\r
+    }\r
+\r
+    //\r
+    // Check whether the input password is same as preexisted password.\r
+    //\r
+    if (StrnCmp (PasswordString, (CHAR16 *) Question->BufferValue, Question->StorageWidth/sizeof (CHAR16)) == 0) {\r
+      return EFI_SUCCESS;\r
+    } else {\r
+      return EFI_NOT_READY;\r
+    }\r
+  }\r
+    \r
+  //\r
+  // Prepare password string in HII database\r
+  //\r
+  if (PasswordString != NULL) {\r
+    IfrTypeValue.string = NewString (PasswordString, gCurrentSelection->FormSet->HiiHandle);\r
+  } else {\r
+    IfrTypeValue.string = 0;\r
+  }\r
+\r
+  //\r
+  // Send password to Configuration Driver for validation\r
+  //\r
+  Status = ConfigAccess->Callback (\r
+                           ConfigAccess,\r
+                           EFI_BROWSER_ACTION_CHANGING,\r
+                           Question->QuestionId,\r
+                           Question->HiiValue.Type,\r
+                           &IfrTypeValue,\r
+                           &ActionRequest\r
+                           );\r
+\r
+  //\r
+  // Remove password string from HII database\r
+  //\r
+  if (PasswordString != NULL) {\r
+    DeleteString (IfrTypeValue.string, gCurrentSelection->FormSet->HiiHandle);\r
+  }\r
+\r
+  return Status;\r
+}\r
+\r
 /**\r
   Find the registered HotKey based on KeyData.\r
   \r
@@ -4512,7 +5662,7 @@ SetScope (
   if (Scope >= MaxLevel) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
-  \r
+\r
   //\r
   // When no hot key registered in system or on the first setting,\r
   // Scope can be set.\r
@@ -4636,6 +5786,143 @@ RegiserExitHandler (
   return;\r
 }\r
 \r
+/**\r
+  Check whether the browser data has been modified.\r
+\r
+  @retval TRUE        Browser data is modified.\r
+  @retval FALSE       No browser data is modified.\r
+\r
+**/\r
+BOOLEAN\r
+EFIAPI\r
+IsBrowserDataModified (\r
+  VOID\r
+  )\r
+{\r
+  LIST_ENTRY              *Link;\r
+  FORM_BROWSER_FORMSET    *FormSet;\r
+\r
+  switch (gBrowserSettingScope) {\r
+    case FormLevel:\r
+      if (gCurrentSelection == NULL) {\r
+        return FALSE;\r
+      }\r
+      return IsNvUpdateRequiredForForm (gCurrentSelection->Form);\r
+\r
+    case FormSetLevel:\r
+      if (gCurrentSelection == NULL) {\r
+        return FALSE;\r
+      }\r
+      return IsNvUpdateRequiredForFormSet (gCurrentSelection->FormSet);\r
+\r
+    case SystemLevel:\r
+      Link = GetFirstNode (&gBrowserFormSetList);\r
+      while (!IsNull (&gBrowserFormSetList, Link)) {\r
+        FormSet = FORM_BROWSER_FORMSET_FROM_LINK (Link);\r
+        if (!ValidateFormSet(FormSet)) {\r
+          continue;\r
+        }\r
+\r
+        if (IsNvUpdateRequiredForFormSet (FormSet)) {\r
+          return TRUE;\r
+        }\r
+        Link = GetNextNode (&gBrowserFormSetList, Link);\r
+      }\r
+      return FALSE;\r
+\r
+    default:\r
+      return FALSE;\r
+  }\r
+}\r
+\r
+/**\r
+  Execute the action requested by the Action parameter.\r
+\r
+  @param[in] Action     Execute the request action.\r
+  @param[in] DefaultId  The default Id info when need to load default value. Only used when Action is BROWSER_ACTION_DEFAULT.\r
+\r
+  @retval EFI_SUCCESS              Execute the request action succss.\r
+  @retval EFI_INVALID_PARAMETER    The input action value is invalid.\r
+\r
+**/\r
+EFI_STATUS \r
+EFIAPI\r
+ExecuteAction (\r
+  IN UINT32        Action,\r
+  IN UINT16        DefaultId\r
+  )\r
+{\r
+  EFI_STATUS              Status;\r
+  FORM_BROWSER_FORMSET    *FormSet;\r
+  FORM_BROWSER_FORM       *Form;\r
+\r
+  if (gBrowserSettingScope < SystemLevel && gCurrentSelection == NULL) {\r
+    return EFI_NOT_READY;\r
+  }\r
+\r
+  Status  = EFI_SUCCESS;\r
+  FormSet = NULL;\r
+  Form    = NULL;\r
+  if (gBrowserSettingScope < SystemLevel) {\r
+    FormSet = gCurrentSelection->FormSet;\r
+    Form    = gCurrentSelection->Form; \r
+  }\r
+\r
+  //\r
+  // Executet the discard action.\r
+  //\r
+  if ((Action & BROWSER_ACTION_DISCARD) != 0) {\r
+    Status = DiscardForm (FormSet, Form, gBrowserSettingScope);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+  }\r
+\r
+  //\r
+  // Executet the difault action.\r
+  //\r
+  if ((Action & BROWSER_ACTION_DEFAULT) != 0) {\r
+    Status = ExtractDefault (FormSet, Form, DefaultId, gBrowserSettingScope, GetDefaultForAll, NULL, FALSE);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+    UpdateStatementStatus (FormSet, Form, gBrowserSettingScope);\r
+  }\r
+\r
+  //\r
+  // Executet the submit action.\r
+  //\r
+  if ((Action & BROWSER_ACTION_SUBMIT) != 0) {\r
+    Status = SubmitForm (FormSet, Form, gBrowserSettingScope);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+  }\r
+\r
+  //\r
+  // Executet the reset action.\r
+  //\r
+  if ((Action & BROWSER_ACTION_RESET) != 0) {\r
+    gResetRequired = TRUE;\r
+  }\r
+\r
+  //\r
+  // Executet the exit action.\r
+  //\r
+  if ((Action & BROWSER_ACTION_EXIT) != 0) {\r
+    DiscardForm (FormSet, Form, gBrowserSettingScope);\r
+    if (gBrowserSettingScope == SystemLevel) {\r
+      if (ExitHandlerFunction != NULL) {\r
+        ExitHandlerFunction ();\r
+      }\r
+    }\r
+\r
+    gExitRequired = TRUE;\r
+  }\r
+\r
+  return Status;\r
+}\r
+\r
 /**\r
   Create reminder to let user to choose save or discard the changed browser data.\r
   Caller can use it to actively check the changed browser data.\r
@@ -4643,6 +5930,7 @@ RegiserExitHandler (
   @retval BROWSER_NO_CHANGES       No browser data is changed.\r
   @retval BROWSER_SAVE_CHANGES     The changed browser data is saved.\r
   @retval BROWSER_DISCARD_CHANGES  The changed browser data is discard.\r
+  @retval BROWSER_KEEP_CURRENT     Browser keep current changes.\r
 \r
 **/\r
 UINT32\r
@@ -4655,12 +5943,7 @@ SaveReminder (
   FORM_BROWSER_FORMSET    *FormSet;\r
   BOOLEAN                 IsDataChanged;\r
   UINT32                  DataSavedAction;\r
-  CHAR16                  *YesResponse;\r
-  CHAR16                  *NoResponse;\r
-  CHAR16                  *EmptyString;\r
-  CHAR16                  *ChangeReminderString;\r
-  CHAR16                  *SaveConfirmString;\r
-  EFI_INPUT_KEY           Key;\r
+  UINT32                  ConfirmRet;\r
 \r
   DataSavedAction  = BROWSER_NO_CHANGES;\r
   IsDataChanged    = FALSE;\r
@@ -4671,7 +5954,7 @@ SaveReminder (
     if (!ValidateFormSet(FormSet)) {\r
       continue;\r
     }\r
-    if (IsNvUpdateRequired (FormSet)) {\r
+    if (IsNvUpdateRequiredForFormSet (FormSet)) {\r
       IsDataChanged = TRUE;\r
       break;\r
     }\r
@@ -4685,42 +5968,41 @@ SaveReminder (
   }\r
   \r
   //\r
-  // If data is changed, prompt user\r
+  // If data is changed, prompt user to save or discard it. \r
   //\r
-  gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
+  do {\r
+    ConfirmRet = (UINT32) mFormDisplay->ConfirmDataChange();\r
 \r
-  YesResponse          = GetToken (STRING_TOKEN (ARE_YOU_SURE_YES), gHiiHandle);\r
-  ASSERT (YesResponse != NULL);\r
-  NoResponse           = GetToken (STRING_TOKEN (ARE_YOU_SURE_NO), gHiiHandle);\r
-  ASSERT (NoResponse  != NULL);\r
-  EmptyString          = GetToken (STRING_TOKEN (EMPTY_STRING), gHiiHandle);\r
-  ChangeReminderString = GetToken (STRING_TOKEN (CHANGE_REMINDER), gHiiHandle);\r
-  SaveConfirmString    = GetToken (STRING_TOKEN (SAVE_CONFIRM), gHiiHandle);\r
+    if (ConfirmRet == BROWSER_ACTION_SUBMIT) {\r
+      SubmitForm (NULL, NULL, SystemLevel);\r
+      DataSavedAction = BROWSER_SAVE_CHANGES;\r
+      break;\r
+    } else if (ConfirmRet == BROWSER_ACTION_DISCARD) {\r
+      DiscardForm (NULL, NULL, SystemLevel);\r
+      DataSavedAction = BROWSER_DISCARD_CHANGES;\r
+      break;\r
+    } else if (ConfirmRet == BROWSER_ACTION_NONE) {\r
+      DataSavedAction = BROWSER_KEEP_CURRENT;\r
+      break;\r
+    }\r
+  } while (1);\r
 \r
-  do {\r
-    CreateDialog (4, TRUE, 0, NULL, &Key, EmptyString, ChangeReminderString, SaveConfirmString, EmptyString);\r
-  } while\r
-  (((Key.UnicodeChar | UPPER_LOWER_CASE_OFFSET) != (NoResponse[0] | UPPER_LOWER_CASE_OFFSET)) &&\r
-   ((Key.UnicodeChar | UPPER_LOWER_CASE_OFFSET) != (YesResponse[0] | UPPER_LOWER_CASE_OFFSET))\r
-  );\r
+  return DataSavedAction;\r
+}\r
 \r
-  //\r
-  // If the user hits the YesResponse key\r
-  //\r
-  if ((Key.UnicodeChar | UPPER_LOWER_CASE_OFFSET) == (YesResponse[0] | UPPER_LOWER_CASE_OFFSET)) {\r
-    SubmitForm (NULL, NULL, SystemLevel);\r
-    DataSavedAction = BROWSER_SAVE_CHANGES;\r
-  } else {\r
-    DiscardForm (NULL, NULL, SystemLevel);\r
-    DataSavedAction = BROWSER_DISCARD_CHANGES;\r
-    gResetRequired  = FALSE;\r
-  }\r
+/**\r
+  Check whether the Reset Required for the browser\r
 \r
-  FreePool (YesResponse);\r
-  FreePool (NoResponse);\r
-  FreePool (EmptyString);\r
-  FreePool (SaveConfirmString);\r
-  FreePool (ChangeReminderString);\r
+  @retval TRUE      Browser required to reset after exit.\r
+  @retval FALSE     Browser not need to reset after exit.\r
 \r
-  return DataSavedAction;\r
+**/\r
+BOOLEAN\r
+EFIAPI\r
+IsResetRequired (\r
+  VOID\r
+  )\r
+{\r
+  return gResetRequired;\r
 }\r
+\r