IntelFrameworkModulePkg: Replace BSD License with BSD+Patent License

[mirror_edk2.git] / IntelFrameworkModulePkg / Universal / BdsDxe / BootMaint / BootMaint.c

/** @file\r
  The functions for Boot Maintainence Main menu.\r
\r
Copyright (c) 2004 - 2018, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "BootMaint.h"\r
#include "FormGuid.h"\r
#include "Bds.h"\r
#include "FrontPage.h"\r
\r
EFI_DEVICE_PATH_PROTOCOL  EndDevicePath[] = {\r
  {\r
    END_DEVICE_PATH_TYPE,\r
    END_ENTIRE_DEVICE_PATH_SUBTYPE,\r
    {\r
      END_DEVICE_PATH_LENGTH,\r
      0\r
    }\r
  }\r
};\r
\r
HII_VENDOR_DEVICE_PATH  mBmmHiiVendorDevicePath = {\r
  {\r
    {\r
      HARDWARE_DEVICE_PATH,\r
      HW_VENDOR_DP,\r
      {\r
        (UINT8) (sizeof (VENDOR_DEVICE_PATH)),\r
        (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8)\r
      }\r
    },\r
    BOOT_MAINT_FORMSET_GUID\r
  },\r
  {\r
    END_DEVICE_PATH_TYPE,\r
    END_ENTIRE_DEVICE_PATH_SUBTYPE,\r
    {\r
      (UINT8) (END_DEVICE_PATH_LENGTH),\r
      (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8)\r
    }\r
  }\r
};\r
\r
HII_VENDOR_DEVICE_PATH  mFeHiiVendorDevicePath = {\r
  {\r
    {\r
      HARDWARE_DEVICE_PATH,\r
      HW_VENDOR_DP,\r
      {\r
        (UINT8) (sizeof (VENDOR_DEVICE_PATH)),\r
        (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8)\r
      }\r
    },\r
    FILE_EXPLORE_FORMSET_GUID\r
  },\r
  {\r
    END_DEVICE_PATH_TYPE,\r
    END_ENTIRE_DEVICE_PATH_SUBTYPE,\r
    {\r
      (UINT8) (END_DEVICE_PATH_LENGTH),\r
      (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8)\r
    }\r
  }\r
};\r
\r
CHAR16  mBootMaintStorageName[]     = L"BmmData";\r
CHAR16  mFileExplorerStorageName[]  = L"FeData";\r
BMM_CALLBACK_DATA *mBmmCallbackInfo = NULL;\r
\r
/**\r
  Init all memu.\r
\r
  @param CallbackData    The BMM context data.\r
\r
**/\r
VOID\r
InitAllMenu (\r
  IN  BMM_CALLBACK_DATA    *CallbackData\r
  );\r
\r
/**\r
  Free up all Menu Option list.\r
\r
**/\r
VOID\r
FreeAllMenu (\r
  VOID\r
  );\r
\r
/**\r
  Initialize all of BMM configuration data in BmmFakeNvData and BmmOldFakeNVData member\r
  in BMM context data and create all of dynamic OP code for BMM.\r
\r
  @param CallbackData    The BMM context data.\r
\r
**/\r
VOID\r
InitializeBmmConfig (\r
  IN  BMM_CALLBACK_DATA    *CallbackData\r
  )\r
{\r
  BM_MENU_ENTRY   *NewMenuEntry;\r
  BM_LOAD_CONTEXT *NewLoadContext;\r
  UINT16          Index;\r
\r
  ASSERT (CallbackData != NULL);\r
\r
  //\r
  // Initialize data which located in BMM main page\r
  //\r
  CallbackData->BmmFakeNvData.BootNext = (UINT16) (BootOptionMenu.MenuNumber);\r
  for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) {\r
    NewMenuEntry    = BOpt_GetMenuEntry (&BootOptionMenu, Index);\r
    NewLoadContext  = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
\r
    if (NewLoadContext->IsBootNext) {\r
      CallbackData->BmmFakeNvData.BootNext = Index;\r
      break;\r
    }\r
  }\r
\r
  CallbackData->BmmFakeNvData.BootTimeOut = PcdGet16 (PcdPlatformBootTimeOut);\r
\r
  //\r
  // Initialize data which located in Boot Options Menu\r
  //\r
  GetBootOrder (CallbackData);\r
  GetLegacyDeviceOrder (CallbackData);\r
\r
  //\r
  // Initialize data which located in Driver Options Menu\r
  //\r
  GetDriverOrder (CallbackData);\r
\r
  //\r
  // Initialize data which located in Console Options Menu\r
  //\r
  GetConsoleOutMode (CallbackData);\r
  GetConsoleInCheck (CallbackData);\r
  GetConsoleOutCheck (CallbackData);\r
  GetConsoleErrCheck (CallbackData);\r
  GetTerminalAttribute (CallbackData);\r
\r
  //\r
  // Backup Initialize BMM configuartion data to BmmOldFakeNVData\r
  //\r
  CopyMem (&CallbackData->BmmOldFakeNVData, &CallbackData->BmmFakeNvData, sizeof (BMM_FAKE_NV_DATA));\r
}\r
\r
/**\r
  Create string tokens for a menu from its help strings and display strings\r
\r
  @param CallbackData       The BMM context data.\r
  @param HiiHandle          Hii Handle of the package to be updated.\r
  @param MenuOption         The Menu whose string tokens need to be created\r
\r
  @retval  EFI_SUCCESS      String tokens created successfully\r
  @retval  others           contain some errors\r
**/\r
EFI_STATUS\r
CreateMenuStringToken (\r
  IN BMM_CALLBACK_DATA                *CallbackData,\r
  IN EFI_HII_HANDLE                   HiiHandle,\r
  IN BM_MENU_OPTION                   *MenuOption\r
  )\r
{\r
  BM_MENU_ENTRY *NewMenuEntry;\r
  UINTN         Index;\r
\r
  for (Index = 0; Index < MenuOption->MenuNumber; Index++) {\r
    NewMenuEntry = BOpt_GetMenuEntry (MenuOption, Index);\r
\r
    NewMenuEntry->DisplayStringToken = HiiSetString (\r
                                         HiiHandle,\r
                                         0,\r
                                         NewMenuEntry->DisplayString,\r
                                         NULL\r
                                         );\r
\r
    if (NULL == NewMenuEntry->HelpString) {\r
      NewMenuEntry->HelpStringToken = NewMenuEntry->DisplayStringToken;\r
    } else {\r
      NewMenuEntry->HelpStringToken = HiiSetString (\r
                                        HiiHandle,\r
                                        0,\r
                                        NewMenuEntry->HelpString,\r
                                        NULL\r
                                        );\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  This function allows a caller to extract the current configuration for one\r
  or more named elements from the target driver.\r
\r
\r
  @param This            Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
  @param Request         A null-terminated Unicode string in <ConfigRequest> format.\r
  @param Progress        On return, points to a character in the Request string.\r
                         Points to the string's null terminator if request was successful.\r
                         Points to the most recent '&' before the first failing name/value\r
                         pair (or the beginning of the string if the failure is in the\r
                         first name/value pair) if the request was not successful.\r
  @param Results         A null-terminated Unicode string in <ConfigAltResp> format which\r
                         has all values filled in for the names in the Request string.\r
                         String to be allocated by the called function.\r
\r
  @retval  EFI_SUCCESS            The Results is filled with the requested values.\r
  @retval  EFI_OUT_OF_RESOURCES   Not enough memory to store the results.\r
  @retval  EFI_INVALID_PARAMETER  Request is NULL, illegal syntax, or unknown name.\r
  @retval  EFI_NOT_FOUND          Routing data doesn't match any storage in this driver.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
BootMaintExtractConfig (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL   *This,\r
  IN  CONST EFI_STRING                       Request,\r
  OUT EFI_STRING                             *Progress,\r
  OUT EFI_STRING                             *Results\r
  )\r
{\r
  EFI_STATUS         Status;\r
  UINTN              BufferSize;\r
  BMM_CALLBACK_DATA  *Private;\r
  EFI_STRING                       ConfigRequestHdr;\r
  EFI_STRING                       ConfigRequest;\r
  BOOLEAN                          AllocatedRequest;\r
  UINTN                            Size;\r
\r
  if (Progress == NULL || Results == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  *Progress = Request;\r
  if ((Request != NULL) && !HiiIsConfigHdrMatch (Request, &gBootMaintFormSetGuid, mBootMaintStorageName)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  ConfigRequestHdr = NULL;\r
  ConfigRequest    = NULL;\r
  AllocatedRequest = FALSE;\r
  Size             = 0;\r
\r
  Private = BMM_CALLBACK_DATA_FROM_THIS (This);\r
  //\r
  // Convert buffer data to <ConfigResp> by helper function BlockToConfig()\r
  //\r
  BufferSize = sizeof (BMM_FAKE_NV_DATA);\r
  ConfigRequest = Request;\r
  if ((Request == NULL) || (StrStr (Request, L"OFFSET") == NULL)) {\r
    //\r
    // Request has no request element, construct full request string.\r
    // Allocate and fill a buffer large enough to hold the <ConfigHdr> template\r
    // followed by "&OFFSET=0&WIDTH=WWWWWWWWWWWWWWWW" followed by a Null-terminator\r
    //\r
    ConfigRequestHdr = HiiConstructConfigHdr (&gBootMaintFormSetGuid, mBootMaintStorageName, Private->BmmDriverHandle);\r
    Size = (StrLen (ConfigRequestHdr) + 32 + 1) * sizeof (CHAR16);\r
    ConfigRequest = AllocateZeroPool (Size);\r
    ASSERT (ConfigRequest != NULL);\r
    AllocatedRequest = TRUE;\r
    UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", ConfigRequestHdr, (UINT64)BufferSize);\r
    FreePool (ConfigRequestHdr);\r
  }\r
\r
  Status = gHiiConfigRouting->BlockToConfig (\r
                                gHiiConfigRouting,\r
                                ConfigRequest,\r
                                (UINT8 *) &Private->BmmFakeNvData,\r
                                BufferSize,\r
                                Results,\r
                                Progress\r
                                );\r
  //\r
  // Free the allocated config request string.\r
  //\r
  if (AllocatedRequest) {\r
    FreePool (ConfigRequest);\r
    ConfigRequest = NULL;\r
  }\r
  //\r
  // Set Progress string to the original request string.\r
  //\r
  if (Request == NULL) {\r
    *Progress = NULL;\r
  } else if (StrStr (Request, L"OFFSET") == NULL) {\r
    *Progress = Request + StrLen (Request);\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  This function applies changes in a driver's configuration.\r
  Input is a Configuration, which has the routing data for this\r
  driver followed by name / value configuration pairs. The driver\r
  must apply those pairs to its configurable storage. If the\r
  driver's configuration is stored in a linear block of data\r
  and the driver's name / value pairs are in <BlockConfig>\r
  format, it may use the ConfigToBlock helper function (above) to\r
  simplify the job. Currently not implemented.\r
\r
  @param[in]  This                Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
  @param[in]  Configuration       A null-terminated Unicode string in\r
                                  <ConfigString> format.\r
  @param[out] Progress            A pointer to a string filled in with the\r
                                  offset of the most recent '&' before the\r
                                  first failing name / value pair (or the\r
                                  beginn ing of the string if the failure\r
                                  is in the first name / value pair) or\r
                                  the terminating NULL if all was\r
                                  successful.\r
\r
  @retval EFI_SUCCESS             The results have been distributed or are\r
                                  awaiting distribution.\r
  @retval EFI_OUT_OF_RESOURCES    Not enough memory to store the\r
                                  parts of the results that must be\r
                                  stored awaiting possible future\r
                                  protocols.\r
  @retval EFI_INVALID_PARAMETERS  Passing in a NULL for the\r
                                  Results parameter would result\r
                                  in this type of error.\r
  @retval EFI_NOT_FOUND           Target for the specified routing data\r
                                  was not found.\r
**/\r
EFI_STATUS\r
EFIAPI\r
BootMaintRouteConfig (\r
  IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This,\r
  IN CONST EFI_STRING                     Configuration,\r
  OUT EFI_STRING                          *Progress\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  UINTN                           BufferSize;\r
  EFI_HII_CONFIG_ROUTING_PROTOCOL *ConfigRouting;\r
  BMM_FAKE_NV_DATA                *NewBmmData;\r
  BMM_FAKE_NV_DATA                *OldBmmData;\r
  BM_CONSOLE_CONTEXT              *NewConsoleContext;\r
  BM_TERMINAL_CONTEXT             *NewTerminalContext;\r
  BM_MENU_ENTRY                   *NewMenuEntry;\r
  BM_LOAD_CONTEXT                 *NewLoadContext;\r
  UINT16                          Index;\r
  BOOLEAN                         TerminalAttChange;\r
  BMM_CALLBACK_DATA               *Private;\r
\r
  if (Progress == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  *Progress = Configuration;\r
\r
  if (Configuration == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Check routing data in <ConfigHdr>.\r
  // Note: there is no name for Name/Value storage, only GUID will be checked\r
  //\r
  if (!HiiIsConfigHdrMatch (Configuration, &gBootMaintFormSetGuid, mBootMaintStorageName)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  Status = gBS->LocateProtocol (\r
                  &gEfiHiiConfigRoutingProtocolGuid,\r
                  NULL,\r
                  (VOID**) &ConfigRouting\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Private = BMM_CALLBACK_DATA_FROM_THIS (This);\r
  //\r
  // Get Buffer Storage data from EFI variable\r
  //\r
  BufferSize = sizeof (BMM_FAKE_NV_DATA);\r
  OldBmmData = &Private->BmmOldFakeNVData;\r
  NewBmmData = &Private->BmmFakeNvData;\r
  //\r
  // Convert <ConfigResp> to buffer data by helper function ConfigToBlock()\r
  //\r
  Status = ConfigRouting->ConfigToBlock (\r
                            ConfigRouting,\r
                            Configuration,\r
                            (UINT8 *) NewBmmData,\r
                            &BufferSize,\r
                            Progress\r
                            );\r
  ASSERT_EFI_ERROR (Status);\r
  //\r
  // Compare new and old BMM configuration data and only do action for modified item to\r
  // avoid setting unnecessary non-volatile variable\r
  //\r
\r
  //\r
  // Check data which located in BMM main page and save the settings if need\r
  //\r
  if (CompareMem (NewBmmData->LegacyFD, OldBmmData->LegacyFD, sizeof (NewBmmData->LegacyFD)) != 0) {\r
    Var_UpdateBBSOption (Private, FORM_SET_FD_ORDER_ID);\r
  }\r
\r
  if (CompareMem (NewBmmData->LegacyHD, OldBmmData->LegacyHD, sizeof (NewBmmData->LegacyHD)) != 0) {\r
    Var_UpdateBBSOption (Private, FORM_SET_HD_ORDER_ID);\r
  }\r
\r
  if (CompareMem (NewBmmData->LegacyCD, OldBmmData->LegacyCD, sizeof (NewBmmData->LegacyCD)) != 0) {\r
    Var_UpdateBBSOption (Private, FORM_SET_CD_ORDER_ID);\r
  }\r
\r
  if (CompareMem (NewBmmData->LegacyNET, OldBmmData->LegacyNET, sizeof (NewBmmData->LegacyNET)) != 0) {\r
    Var_UpdateBBSOption (Private, FORM_SET_NET_ORDER_ID);\r
  }\r
\r
  if (CompareMem (NewBmmData->LegacyBEV, OldBmmData->LegacyBEV, sizeof (NewBmmData->LegacyBEV)) != 0) {\r
    Var_UpdateBBSOption (Private, FORM_SET_BEV_ORDER_ID);\r
  }\r
\r
  //\r
  // Change for "delete boot option" page need update NewBmmData->BootOptionOrder, so process\r
  // NewBmmData->BootOptionOrder before NewBmmData->BootOptionDel\r
  //\r
  if (CompareMem (NewBmmData->BootOptionOrder, OldBmmData->BootOptionOrder, sizeof (NewBmmData->BootOptionOrder)) != 0) {\r
    Status = Var_UpdateBootOrder (Private);\r
  }\r
\r
  //\r
  // Change for "delete driver option" page need update NewBmmData->DriverOptionOrder, so process\r
  // NewBmmData->DriverOptionOrder before NewBmmData->DriverOptionDel\r
  //\r
  if (CompareMem (NewBmmData->DriverOptionOrder, OldBmmData->DriverOptionOrder, sizeof (NewBmmData->DriverOptionOrder)) != 0) {\r
    Status = Var_UpdateDriverOrder (Private);\r
  }\r
\r
  //\r
  // Check data which located in Boot Options Menu and save the settings if need\r
  //\r
  if (CompareMem (NewBmmData->BootOptionDel, OldBmmData->BootOptionDel, sizeof (NewBmmData->BootOptionDel)) != 0) {\r
    for (Index = 0;\r
         ((Index < BootOptionMenu.MenuNumber) && (Index < (sizeof (NewBmmData->BootOptionDel) / sizeof (NewBmmData->BootOptionDel[0]))));\r
         Index ++) {\r
      NewMenuEntry            = BOpt_GetMenuEntry (&BootOptionMenu, Index);\r
      NewLoadContext          = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
      NewLoadContext->Deleted = NewBmmData->BootOptionDel[Index];\r
      NewBmmData->BootOptionDel[Index] = FALSE;\r
      NewBmmData->BootOptionDelMark[Index] = FALSE;\r
    }\r
\r
    Var_DelBootOption ();\r
  }\r
\r
  //\r
  // Check data which located in Driver Options Menu and save the settings if need\r
  //\r
  if (CompareMem (NewBmmData->DriverOptionDel, OldBmmData->DriverOptionDel, sizeof (NewBmmData->DriverOptionDel)) != 0) {\r
    for (Index = 0;\r
         ((Index < DriverOptionMenu.MenuNumber) && (Index < (sizeof (NewBmmData->DriverOptionDel) / sizeof (NewBmmData->DriverOptionDel[0]))));\r
         Index++) {\r
      NewMenuEntry            = BOpt_GetMenuEntry (&DriverOptionMenu, Index);\r
      NewLoadContext          = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
      NewLoadContext->Deleted = NewBmmData->DriverOptionDel[Index];\r
      NewBmmData->DriverOptionDel[Index] = FALSE;\r
      NewBmmData->DriverOptionDelMark[Index] = FALSE;\r
    }\r
    Var_DelDriverOption ();\r
  }\r
\r
  if (CompareMem (&NewBmmData->BootTimeOut, &OldBmmData->BootTimeOut, sizeof (NewBmmData->BootTimeOut)) != 0) {\r
    Status = gRT->SetVariable (\r
                    L"Timeout",\r
                    &gEfiGlobalVariableGuid,\r
                    EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,\r
                    sizeof (UINT16),\r
                    &(NewBmmData->BootTimeOut)\r
                    );\r
    ASSERT_EFI_ERROR(Status);\r
\r
    //\r
    // Bugbug: code not exit in UiApp but in IntelFrameworkModulePkg, need do more check.\r
    //\r
    Private->BmmOldFakeNVData.BootTimeOut = NewBmmData->BootTimeOut;\r
  }\r
\r
  if (CompareMem (&NewBmmData->BootNext, &OldBmmData->BootNext, sizeof (NewBmmData->BootNext)) != 0) {\r
    Status = Var_UpdateBootNext (Private);\r
  }\r
\r
  if (CompareMem (&NewBmmData->ConsoleOutMode, &OldBmmData->ConsoleOutMode, sizeof (NewBmmData->ConsoleOutMode)) != 0) {\r
    Var_UpdateConMode (Private);\r
  }\r
\r
  TerminalAttChange = FALSE;\r
  for (Index = 0; Index < TerminalMenu.MenuNumber; Index++) {\r
\r
    //\r
    // only need update modified items\r
    //\r
    if (CompareMem (&NewBmmData->COMBaudRate[Index], &OldBmmData->COMBaudRate[Index], sizeof (NewBmmData->COMBaudRate[Index])) == 0 &&\r
         CompareMem (&NewBmmData->COMDataRate[Index], &OldBmmData->COMDataRate[Index], sizeof (NewBmmData->COMDataRate[Index])) == 0 &&\r
         CompareMem (&NewBmmData->COMStopBits[Index], &OldBmmData->COMStopBits[Index], sizeof (NewBmmData->COMStopBits[Index])) == 0 &&\r
         CompareMem (&NewBmmData->COMParity[Index], &OldBmmData->COMParity[Index], sizeof (NewBmmData->COMParity[Index])) == 0 &&\r
         CompareMem (&NewBmmData->COMTerminalType[Index], &OldBmmData->COMTerminalType[Index], sizeof (NewBmmData->COMTerminalType[Index])) == 0 &&\r
         CompareMem (&NewBmmData->COMFlowControl[Index], &OldBmmData->COMFlowControl[Index], sizeof (NewBmmData->COMFlowControl[Index])) == 0) {\r
      continue;\r
    }\r
\r
    NewMenuEntry = BOpt_GetMenuEntry (&TerminalMenu, Index);\r
    ASSERT (NewMenuEntry != NULL);\r
    NewTerminalContext = (BM_TERMINAL_CONTEXT *) NewMenuEntry->VariableContext;\r
    NewTerminalContext->BaudRateIndex = NewBmmData->COMBaudRate[Index];\r
    ASSERT (NewBmmData->COMBaudRate[Index] < (ARRAY_SIZE (BaudRateList)));\r
    NewTerminalContext->BaudRate      = BaudRateList[NewBmmData->COMBaudRate[Index]].Value;\r
    NewTerminalContext->DataBitsIndex = NewBmmData->COMDataRate[Index];\r
    ASSERT (NewBmmData->COMDataRate[Index] < (ARRAY_SIZE (DataBitsList)));\r
    NewTerminalContext->DataBits      = (UINT8) DataBitsList[NewBmmData->COMDataRate[Index]].Value;\r
    NewTerminalContext->StopBitsIndex = NewBmmData->COMStopBits[Index];\r
    ASSERT (NewBmmData->COMStopBits[Index] < (ARRAY_SIZE (StopBitsList)));\r
    NewTerminalContext->StopBits      = (UINT8) StopBitsList[NewBmmData->COMStopBits[Index]].Value;\r
    NewTerminalContext->ParityIndex   = NewBmmData->COMParity[Index];\r
    ASSERT (NewBmmData->COMParity[Index] < (ARRAY_SIZE (ParityList)));\r
    NewTerminalContext->Parity        = (UINT8) ParityList[NewBmmData->COMParity[Index]].Value;\r
    NewTerminalContext->TerminalType  = NewBmmData->COMTerminalType[Index];\r
    NewTerminalContext->FlowControl   = NewBmmData->COMFlowControl[Index];\r
    ChangeTerminalDevicePath (\r
      &(NewTerminalContext->DevicePath),\r
      FALSE\r
      );\r
    TerminalAttChange = TRUE;\r
  }\r
  if (TerminalAttChange) {\r
    Var_UpdateConsoleInpOption ();\r
    Var_UpdateConsoleOutOption ();\r
    Var_UpdateErrorOutOption ();\r
  }\r
\r
  //\r
  // Check data which located in Console Options Menu and save the settings if need\r
  //\r
  if (CompareMem (NewBmmData->ConsoleInCheck, OldBmmData->ConsoleInCheck, sizeof (NewBmmData->ConsoleInCheck)) != 0) {\r
    for (Index = 0; Index < ConsoleInpMenu.MenuNumber; Index++) {\r
      NewMenuEntry                = BOpt_GetMenuEntry (&ConsoleInpMenu, Index);\r
      NewConsoleContext           = (BM_CONSOLE_CONTEXT *) NewMenuEntry->VariableContext;\r
      ASSERT (Index < MAX_MENU_NUMBER);\r
      NewConsoleContext->IsActive = NewBmmData->ConsoleInCheck[Index];\r
    }\r
\r
    Var_UpdateConsoleInpOption ();\r
  }\r
\r
  if (CompareMem (NewBmmData->ConsoleOutCheck, OldBmmData->ConsoleOutCheck, sizeof (NewBmmData->ConsoleOutCheck)) != 0) {\r
    for (Index = 0; Index < ConsoleOutMenu.MenuNumber; Index++) {\r
      NewMenuEntry                = BOpt_GetMenuEntry (&ConsoleOutMenu, Index);\r
      NewConsoleContext           = (BM_CONSOLE_CONTEXT *) NewMenuEntry->VariableContext;\r
      ASSERT (Index < MAX_MENU_NUMBER);\r
      NewConsoleContext->IsActive = NewBmmData->ConsoleOutCheck[Index];\r
    }\r
\r
    Var_UpdateConsoleOutOption ();\r
  }\r
\r
  if (CompareMem (NewBmmData->ConsoleErrCheck, OldBmmData->ConsoleErrCheck, sizeof (NewBmmData->ConsoleErrCheck)) != 0) {\r
    for (Index = 0; Index < ConsoleErrMenu.MenuNumber; Index++) {\r
      NewMenuEntry                = BOpt_GetMenuEntry (&ConsoleErrMenu, Index);\r
      NewConsoleContext           = (BM_CONSOLE_CONTEXT *) NewMenuEntry->VariableContext;\r
      ASSERT (Index < MAX_MENU_NUMBER);\r
      NewConsoleContext->IsActive = NewBmmData->ConsoleErrCheck[Index];\r
    }\r
\r
    Var_UpdateErrorOutOption ();\r
  }\r
\r
  //\r
  // After user do the save action, need to update OldBmmData.\r
  //\r
  CopyMem (OldBmmData, NewBmmData, sizeof (BMM_FAKE_NV_DATA));\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Create GoTo OP code into FORM_BOOT_LEGACY_DEVICE label for legacy boot option.\r
\r
**/\r
EFI_STATUS\r
InitializeLegacyBootOption (\r
  VOID\r
  )\r
{\r
  RefreshUpdateData ();\r
  mStartLabel->Number = FORM_BOOT_LEGACY_DEVICE_ID;\r
\r
  //\r
  // If LegacyBios Protocol is installed, add 3 tags about legacy boot option\r
  // in BootOption form: legacy FD/HD/CD/NET/BEV\r
  //\r
  HiiCreateGotoOpCode (\r
    mStartOpCodeHandle,\r
    FORM_SET_FD_ORDER_ID,\r
    STRING_TOKEN (STR_FORM_SET_FD_ORDER_TITLE),\r
    STRING_TOKEN (STR_FORM_SET_FD_ORDER_TITLE),\r
    EFI_IFR_FLAG_CALLBACK,\r
    FORM_SET_FD_ORDER_ID\r
    );\r
\r
  HiiCreateGotoOpCode (\r
    mStartOpCodeHandle,\r
    FORM_SET_HD_ORDER_ID,\r
    STRING_TOKEN (STR_FORM_SET_HD_ORDER_TITLE),\r
    STRING_TOKEN (STR_FORM_SET_HD_ORDER_TITLE),\r
    EFI_IFR_FLAG_CALLBACK,\r
    FORM_SET_HD_ORDER_ID\r
    );\r
\r
  HiiCreateGotoOpCode (\r
    mStartOpCodeHandle,\r
    FORM_SET_CD_ORDER_ID,\r
    STRING_TOKEN (STR_FORM_SET_CD_ORDER_TITLE),\r
    STRING_TOKEN (STR_FORM_SET_CD_ORDER_TITLE),\r
    EFI_IFR_FLAG_CALLBACK,\r
    FORM_SET_CD_ORDER_ID\r
    );\r
\r
  HiiCreateGotoOpCode (\r
    mStartOpCodeHandle,\r
    FORM_SET_NET_ORDER_ID,\r
    STRING_TOKEN (STR_FORM_SET_NET_ORDER_TITLE),\r
    STRING_TOKEN (STR_FORM_SET_NET_ORDER_TITLE),\r
    EFI_IFR_FLAG_CALLBACK,\r
    FORM_SET_NET_ORDER_ID\r
    );\r
\r
  HiiCreateGotoOpCode (\r
    mStartOpCodeHandle,\r
    FORM_SET_BEV_ORDER_ID,\r
    STRING_TOKEN (STR_FORM_SET_BEV_ORDER_TITLE),\r
    STRING_TOKEN (STR_FORM_SET_BEV_ORDER_TITLE),\r
    EFI_IFR_FLAG_CALLBACK,\r
    FORM_SET_BEV_ORDER_ID\r
    );\r
\r
  HiiUpdateForm (\r
    mBmmCallbackInfo->BmmHiiHandle,\r
    &gBootMaintFormSetGuid,\r
    FORM_BOOT_SETUP_ID,\r
    mStartOpCodeHandle, // Label FORM_BOOT_LEGACY_DEVICE_ID\r
    mEndOpCodeHandle    // LABEL_END\r
    );\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  This function processes the results of changes in configuration.\r
\r
\r
  @param This               Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
  @param Action             Specifies the type of action taken by the browser.\r
  @param QuestionId         A unique value which is sent to the original exporting driver\r
                            so that it can identify the type of data to expect.\r
  @param Type               The type of value for the question.\r
  @param Value              A pointer to the data being sent to the original exporting driver.\r
  @param ActionRequest      On return, points to the action requested by the callback function.\r
\r
  @retval EFI_SUCCESS           The callback successfully handled the action.\r
  @retval EFI_OUT_OF_RESOURCES  Not enough storage is available to hold the variable and its data.\r
  @retval EFI_DEVICE_ERROR      The variable could not be saved.\r
  @retval EFI_UNSUPPORTED       The specified Action is not supported by the callback.\r
  @retval EFI_INVALID_PARAMETER The parameter of Value or ActionRequest is invalid.\r
**/\r
EFI_STATUS\r
EFIAPI\r
BootMaintCallback (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL         *This,\r
  IN        EFI_BROWSER_ACTION                     Action,\r
  IN        EFI_QUESTION_ID                        QuestionId,\r
  IN        UINT8                                  Type,\r
  IN        EFI_IFR_TYPE_VALUE                     *Value,\r
  OUT       EFI_BROWSER_ACTION_REQUEST             *ActionRequest\r
  )\r
{\r
  BMM_CALLBACK_DATA *Private;\r
  BM_MENU_ENTRY     *NewMenuEntry;\r
  BMM_FAKE_NV_DATA  *CurrentFakeNVMap;\r
  EFI_STATUS        Status;\r
  UINTN             OldValue;\r
  UINTN             NewValue;\r
  UINTN             Number;\r
  UINTN             Pos;\r
  UINTN             Bit;\r
  UINT16            NewValuePos;\r
  UINT16            Index3;\r
  UINT16            Index2;\r
  UINT16            Index;\r
  UINT8             *OldLegacyDev;\r
  UINT8             *NewLegacyDev;\r
  UINT8             *DisMap;\r
  EFI_LEGACY_BIOS_PROTOCOL    *LegacyBios;\r
\r
  Private = BMM_CALLBACK_DATA_FROM_THIS (This);\r
  if (Action == EFI_BROWSER_ACTION_FORM_OPEN && QuestionId == FORM_BOOT_SETUP_ID) {\r
    //\r
    // Initilize Form for legacy boot option.\r
    //\r
    Status = EfiLibLocateProtocol (&gEfiLegacyBiosProtocolGuid, (VOID **) &LegacyBios);\r
    if (!EFI_ERROR (Status)) {\r
      InitializeLegacyBootOption ();\r
    }\r
\r
    return EFI_SUCCESS;\r
  }\r
\r
  if (Action != EFI_BROWSER_ACTION_CHANGING && Action != EFI_BROWSER_ACTION_CHANGED) {\r
    //\r
    // All other action return unsupported.\r
    //\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  Status       = EFI_SUCCESS;\r
  OldValue     = 0;\r
  NewValue     = 0;\r
  Number       = 0;\r
  OldLegacyDev = NULL;\r
  NewLegacyDev = NULL;\r
  NewValuePos  = 0;\r
  DisMap       = NULL;\r
\r
  Private      = BMM_CALLBACK_DATA_FROM_THIS (This);\r
  //\r
  // Retrieve uncommitted data from Form Browser\r
  //\r
  CurrentFakeNVMap = &Private->BmmFakeNvData;\r
  HiiGetBrowserData (&gBootMaintFormSetGuid, mBootMaintStorageName, sizeof (BMM_FAKE_NV_DATA), (UINT8 *) CurrentFakeNVMap);\r
  if (Action == EFI_BROWSER_ACTION_CHANGING) {\r
    if (Value == NULL) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    UpdatePageId (Private, QuestionId);\r
\r
    if (QuestionId < FILE_OPTION_OFFSET) {\r
      if (QuestionId < CONFIG_OPTION_OFFSET) {\r
        switch (QuestionId) {\r
        case KEY_VALUE_BOOT_FROM_FILE:\r
          Private->FeCurrentState = FileExplorerStateBootFromFile;\r
          break;\r
\r
        case FORM_BOOT_ADD_ID:\r
          Private->FeCurrentState = FileExplorerStateAddBootOption;\r
          break;\r
\r
        case FORM_DRV_ADD_FILE_ID:\r
          Private->FeCurrentState = FileExplorerStateAddDriverOptionState;\r
          break;\r
\r
        case FORM_DRV_ADD_HANDLE_ID:\r
          CleanUpPage (FORM_DRV_ADD_HANDLE_ID, Private);\r
          UpdateDrvAddHandlePage (Private);\r
          break;\r
\r
        case FORM_BOOT_DEL_ID:\r
          CleanUpPage (FORM_BOOT_DEL_ID, Private);\r
          UpdateBootDelPage (Private);\r
          break;\r
\r
        case FORM_BOOT_CHG_ID:\r
        case FORM_DRV_CHG_ID:\r
          UpdatePageBody (QuestionId, Private);\r
          break;\r
\r
        case FORM_DRV_DEL_ID:\r
          CleanUpPage (FORM_DRV_DEL_ID, Private);\r
          UpdateDrvDelPage (Private);\r
          break;\r
\r
        case FORM_BOOT_NEXT_ID:\r
          CleanUpPage (FORM_BOOT_NEXT_ID, Private);\r
          UpdateBootNextPage (Private);\r
          break;\r
\r
        case FORM_TIME_OUT_ID:\r
          CleanUpPage (FORM_TIME_OUT_ID, Private);\r
          UpdateTimeOutPage (Private);\r
          break;\r
\r
        case FORM_CON_IN_ID:\r
        case FORM_CON_OUT_ID:\r
        case FORM_CON_ERR_ID:\r
          UpdatePageBody (QuestionId, Private);\r
          break;\r
\r
        case FORM_CON_MODE_ID:\r
          CleanUpPage (FORM_CON_MODE_ID, Private);\r
          UpdateConModePage (Private);\r
          break;\r
\r
        case FORM_CON_COM_ID:\r
          CleanUpPage (FORM_CON_COM_ID, Private);\r
          UpdateConCOMPage (Private);\r
          break;\r
\r
        case FORM_SET_FD_ORDER_ID:\r
        case FORM_SET_HD_ORDER_ID:\r
        case FORM_SET_CD_ORDER_ID:\r
        case FORM_SET_NET_ORDER_ID:\r
        case FORM_SET_BEV_ORDER_ID:\r
          CleanUpPage (QuestionId, Private);\r
          UpdateSetLegacyDeviceOrderPage (QuestionId, Private);\r
          break;\r
\r
        default:\r
          break;\r
        }\r
      } else if ((QuestionId >= TERMINAL_OPTION_OFFSET) && (QuestionId < CONSOLE_OPTION_OFFSET)) {\r
        Index2                    = (UINT16) (QuestionId - TERMINAL_OPTION_OFFSET);\r
        Private->CurrentTerminal  = Index2;\r
\r
        CleanUpPage (FORM_CON_COM_SETUP_ID, Private);\r
        UpdateTerminalPage (Private);\r
\r
      } else if (QuestionId >= HANDLE_OPTION_OFFSET) {\r
        Index2                  = (UINT16) (QuestionId - HANDLE_OPTION_OFFSET);\r
\r
        NewMenuEntry            = BOpt_GetMenuEntry (&DriverMenu, Index2);\r
        ASSERT (NewMenuEntry != NULL);\r
        Private->HandleContext  = (BM_HANDLE_CONTEXT *) NewMenuEntry->VariableContext;\r
\r
        CleanUpPage (FORM_DRV_ADD_HANDLE_DESC_ID, Private);\r
\r
        Private->MenuEntry                  = NewMenuEntry;\r
        Private->LoadContext->FilePathList  = Private->HandleContext->DevicePath;\r
\r
        UpdateDriverAddHandleDescPage (Private);\r
      }\r
    }\r
  } else if (Action == EFI_BROWSER_ACTION_CHANGED) {\r
    if ((Value == NULL) || (ActionRequest == NULL)) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
    if ((QuestionId >= BOOT_OPTION_DEL_QUESTION_ID) && (QuestionId < BOOT_OPTION_DEL_QUESTION_ID + MAX_MENU_NUMBER)) {\r
      if (Value->b){\r
        //\r
        // Means user try to delete this boot option but not press F10 or "Commit Changes and Exit" menu.\r
        //\r
        CurrentFakeNVMap->BootOptionDelMark[QuestionId - BOOT_OPTION_DEL_QUESTION_ID] = TRUE;\r
      } else {\r
        //\r
        // Means user remove the old check status.\r
        //\r
        CurrentFakeNVMap->BootOptionDelMark[QuestionId - BOOT_OPTION_DEL_QUESTION_ID] = FALSE;\r
      }\r
    } else if ((QuestionId >= DRIVER_OPTION_DEL_QUESTION_ID) && (QuestionId < DRIVER_OPTION_DEL_QUESTION_ID + MAX_MENU_NUMBER)) {\r
      if (Value->b){\r
        CurrentFakeNVMap->DriverOptionDelMark[QuestionId - DRIVER_OPTION_DEL_QUESTION_ID] = TRUE;\r
      } else {\r
        CurrentFakeNVMap->DriverOptionDelMark[QuestionId - DRIVER_OPTION_DEL_QUESTION_ID] = FALSE;\r
      }\r
    } else if ((QuestionId >= LEGACY_FD_QUESTION_ID) && (QuestionId < LEGACY_BEV_QUESTION_ID + MAX_MENU_NUMBER)) {\r
      //\r
      // Update Select FD/HD/CD/NET/BEV Order Form\r
      //\r
\r
      DisMap  = Private->BmmOldFakeNVData.DisableMap;\r
\r
      if (QuestionId >= LEGACY_FD_QUESTION_ID && QuestionId < LEGACY_FD_QUESTION_ID + MAX_MENU_NUMBER) {\r
        Number        = (UINT16) LegacyFDMenu.MenuNumber;\r
        OldLegacyDev  = Private->BmmOldFakeNVData.LegacyFD;\r
        NewLegacyDev  = CurrentFakeNVMap->LegacyFD;\r
      } else if (QuestionId >= LEGACY_HD_QUESTION_ID && QuestionId < LEGACY_HD_QUESTION_ID + MAX_MENU_NUMBER) {\r
        Number        = (UINT16) LegacyHDMenu.MenuNumber;\r
        OldLegacyDev  = Private->BmmOldFakeNVData.LegacyHD;\r
        NewLegacyDev  = CurrentFakeNVMap->LegacyHD;\r
      } else if (QuestionId >= LEGACY_CD_QUESTION_ID && QuestionId < LEGACY_CD_QUESTION_ID + MAX_MENU_NUMBER) {\r
        Number        = (UINT16) LegacyCDMenu.MenuNumber;\r
        OldLegacyDev  = Private->BmmOldFakeNVData.LegacyCD;\r
        NewLegacyDev  = CurrentFakeNVMap->LegacyCD;\r
      } else if (QuestionId >= LEGACY_NET_QUESTION_ID && QuestionId < LEGACY_NET_QUESTION_ID + MAX_MENU_NUMBER) {\r
        Number        = (UINT16) LegacyNETMenu.MenuNumber;\r
        OldLegacyDev  = Private->BmmOldFakeNVData.LegacyNET;\r
        NewLegacyDev  = CurrentFakeNVMap->LegacyNET;\r
      } else if (QuestionId >= LEGACY_BEV_QUESTION_ID && QuestionId < LEGACY_BEV_QUESTION_ID + MAX_MENU_NUMBER) {\r
        Number        = (UINT16) LegacyBEVMenu.MenuNumber;\r
        OldLegacyDev  = Private->BmmOldFakeNVData.LegacyBEV;\r
        NewLegacyDev  = CurrentFakeNVMap->LegacyBEV;\r
      }\r
      //\r
      //  First, find the different position\r
      //  if there is change, it should be only one\r
      //\r
      for (Index = 0; Index < Number; Index++) {\r
        if (OldLegacyDev[Index] != NewLegacyDev[Index]) {\r
          OldValue  = OldLegacyDev[Index];\r
          NewValue  = NewLegacyDev[Index];\r
          break;\r
        }\r
      }\r
\r
      if (Index != Number) {\r
        //\r
        // there is change, now process\r
        //\r
        if (0xFF == NewValue) {\r
          //\r
          // This item will be disable\r
          // Just move the items behind this forward to overlap it\r
          //\r
          Pos = OldValue / 8;\r
          Bit = 7 - (OldValue % 8);\r
          DisMap[Pos] = (UINT8) (DisMap[Pos] | (UINT8) (1 << Bit));\r
          for (Index2 = Index; Index2 < Number - 1; Index2++) {\r
            NewLegacyDev[Index2] = NewLegacyDev[Index2 + 1];\r
          }\r
\r
          NewLegacyDev[Index2] = 0xFF;\r
        } else {\r
          for (Index2 = 0; Index2 < Number; Index2++) {\r
            if (Index2 == Index) {\r
              continue;\r
            }\r
\r
            if (OldLegacyDev[Index2] == NewValue) {\r
              //\r
              // If NewValue is in OldLegacyDev array\r
              // remember its old position\r
              //\r
              NewValuePos = Index2;\r
              break;\r
            }\r
          }\r
\r
          if (Index2 != Number) {\r
            //\r
            // We will change current item to an existing item\r
            // (It's hard to describe here, please read code, it's like a cycle-moving)\r
            //\r
            for (Index2 = NewValuePos; Index2 != Index;) {\r
              if (NewValuePos < Index) {\r
                NewLegacyDev[Index2] = OldLegacyDev[Index2 + 1];\r
                Index2++;\r
              } else {\r
                NewLegacyDev[Index2] = OldLegacyDev[Index2 - 1];\r
                Index2--;\r
              }\r
            }\r
          } else {\r
            //\r
            // If NewValue is not in OldlegacyDev array, we are changing to a disabled item\r
            // so we should modify DisMap to reflect the change\r
            //\r
            Pos = NewValue / 8;\r
            Bit = 7 - (NewValue % 8);\r
            DisMap[Pos] = (UINT8) (DisMap[Pos] & (~ (UINT8) (1 << Bit)));\r
            if (0xFF != OldValue) {\r
              //\r
              // Because NewValue is a item that was disabled before\r
              // so after changing the OldValue should be disabled\r
              // actually we are doing a swap of enable-disable states of two items\r
              //\r
              Pos = OldValue / 8;\r
              Bit = 7 - (OldValue % 8);\r
              DisMap[Pos] = (UINT8) (DisMap[Pos] | (UINT8) (1 << Bit));\r
            }\r
          }\r
        }\r
        //\r
        // To prevent DISABLE appears in the middle of the list\r
        // we should perform a re-ordering\r
        //\r
        Index3 = Index;\r
        Index = 0;\r
        while (Index < Number) {\r
          if (0xFF != NewLegacyDev[Index]) {\r
            Index++;\r
            continue;\r
          }\r
\r
          Index2 = Index;\r
          Index2++;\r
          while (Index2 < Number) {\r
            if (0xFF != NewLegacyDev[Index2]) {\r
              break;\r
            }\r
\r
            Index2++;\r
          }\r
\r
          if (Index2 < Number) {\r
            NewLegacyDev[Index]   = NewLegacyDev[Index2];\r
            NewLegacyDev[Index2]  = 0xFF;\r
          }\r
\r
          Index++;\r
        }\r
\r
        //\r
        //  Return correct question value.\r
        //\r
        Value->u8 = NewLegacyDev[Index3];\r
      }\r
    } else {\r
      switch (QuestionId) {\r
      case KEY_VALUE_SAVE_AND_EXIT:\r
        *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_SUBMIT_EXIT;\r
        break;\r
\r
      case KEY_VALUE_NO_SAVE_AND_EXIT:\r
        //\r
        // Restore local maintain data.\r
        //\r
        DiscardChangeHandler (Private, CurrentFakeNVMap);\r
        *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_DISCARD_EXIT;\r
        break;\r
\r
      case FORM_RESET:\r
        gRT->ResetSystem (EfiResetCold, EFI_SUCCESS, 0, NULL);\r
        return EFI_UNSUPPORTED;\r
\r
      default:\r
        break;\r
      }\r
    }\r
  }\r
\r
  //\r
  // Pass changed uncommitted data back to Form Browser\r
  //\r
  HiiSetBrowserData (&gBootMaintFormSetGuid, mBootMaintStorageName, sizeof (BMM_FAKE_NV_DATA), (UINT8 *) CurrentFakeNVMap, NULL);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Discard all changes done to the BMM pages such as Boot Order change,\r
  Driver order change.\r
\r
  @param Private            The BMM context data.\r
  @param CurrentFakeNVMap   The current Fack NV Map.\r
\r
**/\r
VOID\r
DiscardChangeHandler (\r
  IN  BMM_CALLBACK_DATA               *Private,\r
  IN  BMM_FAKE_NV_DATA                *CurrentFakeNVMap\r
  )\r
{\r
  UINT16  Index;\r
\r
  switch (Private->BmmPreviousPageId) {\r
  case FORM_BOOT_CHG_ID:\r
    CopyMem (CurrentFakeNVMap->BootOptionOrder, Private->BmmOldFakeNVData.BootOptionOrder, sizeof (CurrentFakeNVMap->BootOptionOrder));\r
    break;\r
\r
  case FORM_DRV_CHG_ID:\r
    CopyMem (CurrentFakeNVMap->DriverOptionOrder, Private->BmmOldFakeNVData.DriverOptionOrder, sizeof (CurrentFakeNVMap->DriverOptionOrder));\r
    break;\r
\r
  case FORM_BOOT_DEL_ID:\r
    ASSERT (BootOptionMenu.MenuNumber <= (sizeof (CurrentFakeNVMap->BootOptionDel) / sizeof (CurrentFakeNVMap->BootOptionDel[0])));\r
    for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) {\r
      CurrentFakeNVMap->BootOptionDel[Index] = FALSE;\r
      CurrentFakeNVMap->BootOptionDelMark[Index] = FALSE;\r
    }\r
    break;\r
\r
  case FORM_DRV_DEL_ID:\r
    ASSERT (DriverOptionMenu.MenuNumber <= (sizeof (CurrentFakeNVMap->DriverOptionDel) / sizeof (CurrentFakeNVMap->DriverOptionDel[0])));\r
    for (Index = 0; Index < DriverOptionMenu.MenuNumber; Index++) {\r
      CurrentFakeNVMap->DriverOptionDel[Index] = FALSE;\r
      CurrentFakeNVMap->DriverOptionDelMark[Index] = FALSE;\r
    }\r
    break;\r
\r
  case FORM_BOOT_NEXT_ID:\r
    CurrentFakeNVMap->BootNext = Private->BmmOldFakeNVData.BootNext;\r
    break;\r
\r
  case FORM_TIME_OUT_ID:\r
    CurrentFakeNVMap->BootTimeOut = Private->BmmOldFakeNVData.BootTimeOut;\r
    break;\r
\r
  case FORM_DRV_ADD_HANDLE_DESC_ID:\r
  case FORM_DRV_ADD_FILE_ID:\r
  case FORM_DRV_ADD_HANDLE_ID:\r
    CurrentFakeNVMap->DriverAddHandleDesc[0]          = 0x0000;\r
    CurrentFakeNVMap->DriverAddHandleOptionalData[0]  = 0x0000;\r
    break;\r
\r
  default:\r
    break;\r
  }\r
}\r
\r
/**\r
  Initialize the Boot Maintenance Utitliy.\r
\r
\r
  @retval  EFI_SUCCESS      utility ended successfully\r
  @retval  others           contain some errors\r
\r
**/\r
EFI_STATUS\r
InitializeBM (\r
  VOID\r
  )\r
{\r
  BMM_CALLBACK_DATA           *BmmCallbackInfo;\r
  EFI_STATUS                  Status;\r
  EFI_HII_PACKAGE_LIST_HEADER *PackageListHeader;\r
  UINT32                      Length;\r
  UINT8                       *Data;\r
\r
  Status = EFI_SUCCESS;\r
  BmmCallbackInfo = mBmmCallbackInfo;\r
\r
  BmmCallbackInfo->BmmPreviousPageId             = FORM_MAIN_ID;\r
  BmmCallbackInfo->BmmCurrentPageId              = FORM_MAIN_ID;\r
  BmmCallbackInfo->FeCurrentState                = FileExplorerStateInActive;\r
  BmmCallbackInfo->FeDisplayContext              = FileExplorerDisplayUnknown;\r
\r
  //\r
  // Reinstall String packages to include more new strings.\r
  //\r
\r
  //\r
  // String package size\r
  //\r
  Length = ReadUnaligned32 ((UINT32 *) BdsDxeStrings) - sizeof (UINT32);\r
\r
  //\r
  // Add the length of the Package List Header and the terminating Package Header\r
  //\r
  Length += sizeof (EFI_HII_PACKAGE_LIST_HEADER) + sizeof (EFI_HII_PACKAGE_HEADER);\r
\r
  //\r
  // Allocate the storage for the entire Package List\r
  //\r
  PackageListHeader = AllocateZeroPool (Length);\r
\r
  //\r
  // If the Package List can not be allocated, then return a NULL HII Handle\r
  //\r
  if (PackageListHeader == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  //\r
  // Fill in the GUID and Length of the Package List Header\r
  //\r
  PackageListHeader->PackageLength = Length;\r
\r
  //\r
  // Copy String Data into Package list.\r
  //\r
  Data = (UINT8 *)(PackageListHeader + 1);\r
  Length = ReadUnaligned32 ((UINT32 *) BdsDxeStrings) - sizeof (UINT32);\r
  CopyMem (Data, (UINT8 *) BdsDxeStrings + sizeof (UINT32), Length);\r
\r
  //\r
  // Add End type HII package.\r
  //\r
  Data += Length;\r
  ((EFI_HII_PACKAGE_HEADER *) Data)->Type   = EFI_HII_PACKAGE_END;\r
  ((EFI_HII_PACKAGE_HEADER *) Data)->Length = sizeof (EFI_HII_PACKAGE_HEADER);\r
\r
  //\r
  // Update String package for BM\r
  //\r
  CopyGuid (&PackageListHeader->PackageListGuid, &gBootMaintFormSetGuid);\r
  Status = gHiiDatabase->UpdatePackageList (gHiiDatabase, BmmCallbackInfo->BmmHiiHandle, PackageListHeader);\r
\r
  //\r
  // Update String package for FE.\r
  //\r
  CopyGuid (&PackageListHeader->PackageListGuid, &gFileExploreFormSetGuid);\r
  Status = gHiiDatabase->UpdatePackageList (gHiiDatabase, BmmCallbackInfo->FeHiiHandle, PackageListHeader);\r
\r
  FreePool (PackageListHeader);\r
\r
  //\r
  // Init OpCode Handle and Allocate space for creation of Buffer\r
  //\r
  mStartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
  if (mStartOpCodeHandle == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Exit;\r
  }\r
\r
  mEndOpCodeHandle = HiiAllocateOpCodeHandle ();\r
  if (mEndOpCodeHandle == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Exit;\r
  }\r
\r
  //\r
  // Create Hii Extend Label OpCode as the start opcode\r
  //\r
  mStartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (mStartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
  mStartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
\r
  //\r
  // Create Hii Extend Label OpCode as the end opcode\r
  //\r
  mEndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (mEndOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
  mEndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
  mEndLabel->Number       = LABEL_END;\r
\r
  InitializeStringDepository ();\r
\r
  InitAllMenu (BmmCallbackInfo);\r
\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &ConsoleInpMenu);\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &ConsoleOutMenu);\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &ConsoleErrMenu);\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &BootOptionMenu);\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &DriverOptionMenu);\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &TerminalMenu);\r
  CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &DriverMenu);\r
\r
  InitializeBmmConfig (BmmCallbackInfo);\r
\r
  //\r
  // Dispatch BMM main formset and File Explorer formset.\r
  //\r
  FormSetDispatcher (BmmCallbackInfo);\r
\r
  //\r
  // Clean up.\r
  //\r
  CleanUpStringDepository ();\r
\r
  FreeAllMenu ();\r
\r
Exit:\r
  if (mStartOpCodeHandle != NULL) {\r
    HiiFreeOpCodeHandle (mStartOpCodeHandle);\r
  }\r
\r
  if (mEndOpCodeHandle != NULL) {\r
    HiiFreeOpCodeHandle (mEndOpCodeHandle);\r
  }\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  Initialized all Menu Option List.\r
\r
  @param CallbackData    The BMM context data.\r
\r
**/\r
VOID\r
InitAllMenu (\r
  IN  BMM_CALLBACK_DATA    *CallbackData\r
  )\r
{\r
  InitializeListHead (&BootOptionMenu.Head);\r
  InitializeListHead (&DriverOptionMenu.Head);\r
  BOpt_GetBootOptions (CallbackData);\r
  BOpt_GetDriverOptions (CallbackData);\r
  BOpt_GetLegacyOptions ();\r
  InitializeListHead (&FsOptionMenu.Head);\r
  BOpt_FindDrivers ();\r
  InitializeListHead (&DirectoryMenu.Head);\r
  InitializeListHead (&ConsoleInpMenu.Head);\r
  InitializeListHead (&ConsoleOutMenu.Head);\r
  InitializeListHead (&ConsoleErrMenu.Head);\r
  InitializeListHead (&TerminalMenu.Head);\r
  LocateSerialIo ();\r
  GetAllConsoles ();\r
}\r
\r
/**\r
  Free up all Menu Option list.\r
\r
**/\r
VOID\r
FreeAllMenu (\r
  VOID\r
  )\r
{\r
  BOpt_FreeMenu (&DirectoryMenu);\r
  BOpt_FreeMenu (&FsOptionMenu);\r
  BOpt_FreeMenu (&BootOptionMenu);\r
  BOpt_FreeMenu (&DriverOptionMenu);\r
  BOpt_FreeMenu (&DriverMenu);\r
  BOpt_FreeLegacyOptions ();\r
  FreeAllConsoles ();\r
}\r
\r
/**\r
  Initialize all the string depositories.\r
\r
**/\r
VOID\r
InitializeStringDepository (\r
  VOID\r
  )\r
{\r
  STRING_DEPOSITORY *StringDepository;\r
  StringDepository              = AllocateZeroPool (sizeof (STRING_DEPOSITORY) * STRING_DEPOSITORY_NUMBER);\r
  FileOptionStrDepository       = StringDepository++;\r
  ConsoleOptionStrDepository    = StringDepository++;\r
  BootOptionStrDepository       = StringDepository++;\r
  BootOptionHelpStrDepository   = StringDepository++;\r
  DriverOptionStrDepository     = StringDepository++;\r
  DriverOptionHelpStrDepository = StringDepository++;\r
  TerminalStrDepository         = StringDepository;\r
}\r
\r
/**\r
  Fetch a usable string node from the string depository and return the string token.\r
\r
  @param CallbackData       The BMM context data.\r
  @param StringDepository   The string repository.\r
\r
  @retval  EFI_STRING_ID           String token.\r
\r
**/\r
EFI_STRING_ID\r
GetStringTokenFromDepository (\r
  IN   BMM_CALLBACK_DATA     *CallbackData,\r
  IN   STRING_DEPOSITORY     *StringDepository\r
  )\r
{\r
  STRING_LIST_NODE  *CurrentListNode;\r
  STRING_LIST_NODE  *NextListNode;\r
\r
  CurrentListNode = StringDepository->CurrentNode;\r
\r
  if ((NULL != CurrentListNode) && (NULL != CurrentListNode->Next)) {\r
    //\r
    // Fetch one reclaimed node from the list.\r
    //\r
    NextListNode = StringDepository->CurrentNode->Next;\r
  } else {\r
    //\r
    // If there is no usable node in the list, update the list.\r
    //\r
    NextListNode = AllocateZeroPool (sizeof (STRING_LIST_NODE));\r
    ASSERT (NextListNode != NULL);\r
    NextListNode->StringToken = HiiSetString (CallbackData->BmmHiiHandle, 0, L" ", NULL);\r
    ASSERT (NextListNode->StringToken != 0);\r
\r
    StringDepository->TotalNodeNumber++;\r
\r
    if (NULL == CurrentListNode) {\r
      StringDepository->ListHead = NextListNode;\r
    } else {\r
      CurrentListNode->Next = NextListNode;\r
    }\r
  }\r
\r
  StringDepository->CurrentNode = NextListNode;\r
\r
  return StringDepository->CurrentNode->StringToken;\r
}\r
\r
/**\r
  Reclaim string depositories by moving the current node pointer to list head..\r
\r
**/\r
VOID\r
ReclaimStringDepository (\r
  VOID\r
  )\r
{\r
  UINTN             DepositoryIndex;\r
  STRING_DEPOSITORY *StringDepository;\r
\r
  StringDepository = FileOptionStrDepository;\r
  for (DepositoryIndex = 0; DepositoryIndex < STRING_DEPOSITORY_NUMBER; DepositoryIndex++) {\r
    StringDepository->CurrentNode = StringDepository->ListHead;\r
    StringDepository++;\r
  }\r
}\r
\r
/**\r
  Release resource for all the string depositories.\r
\r
**/\r
VOID\r
CleanUpStringDepository (\r
  VOID\r
  )\r
{\r
  UINTN             NodeIndex;\r
  UINTN             DepositoryIndex;\r
  STRING_LIST_NODE  *CurrentListNode;\r
  STRING_LIST_NODE  *NextListNode;\r
  STRING_DEPOSITORY *StringDepository;\r
\r
  //\r
  // Release string list nodes.\r
  //\r
  StringDepository = FileOptionStrDepository;\r
  for (DepositoryIndex = 0; DepositoryIndex < STRING_DEPOSITORY_NUMBER; DepositoryIndex++) {\r
    CurrentListNode = StringDepository->ListHead;\r
    for (NodeIndex = 0; NodeIndex < StringDepository->TotalNodeNumber; NodeIndex++) {\r
      NextListNode = CurrentListNode->Next;\r
      FreePool (CurrentListNode);\r
      CurrentListNode = NextListNode;\r
    }\r
\r
    StringDepository++;\r
  }\r
  //\r
  // Release string depository.\r
  //\r
  FreePool (FileOptionStrDepository);\r
}\r
\r
/**\r
  Start boot maintenance manager\r
\r
  @retval EFI_SUCCESS If BMM is invoked successfully.\r
  @return Other value if BMM return unsuccessfully.\r
\r
**/\r
EFI_STATUS\r
BdsStartBootMaint (\r
  VOID\r
  )\r
{\r
  EFI_STATUS      Status;\r
  LIST_ENTRY      BdsBootOptionList;\r
\r
  InitializeListHead (&BdsBootOptionList);\r
\r
  //\r
  // Connect all prior to entering the platform setup menu.\r
  //\r
  if (!gConnectAllHappened) {\r
    BdsLibConnectAllDriversToAllControllers ();\r
    gConnectAllHappened = TRUE;\r
  }\r
  //\r
  // Have chance to enumerate boot device\r
  //\r
  BdsLibEnumerateAllBootOption (&BdsBootOptionList);\r
\r
  //\r
  // Group the legacy boot options for the same device type\r
  //\r
  GroupMultipleLegacyBootOption4SameType ();\r
\r
  //\r
  // Init the BMM\r
  //\r
  Status = InitializeBM ();\r
\r
  return Status;\r
}\r
\r
/**\r
  Dispatch BMM formset and FileExplorer formset.\r
\r
\r
  @param CallbackData    The BMM context data.\r
\r
  @retval EFI_SUCCESS If function complete successfully.\r
  @return Other value if the Setup Browser process BMM's pages and\r
           return unsuccessfully.\r
\r
**/\r
EFI_STATUS\r
FormSetDispatcher (\r
  IN  BMM_CALLBACK_DATA    *CallbackData\r
  )\r
{\r
  EFI_STATUS                 Status;\r
  EFI_BROWSER_ACTION_REQUEST ActionRequest;\r
\r
  while (TRUE) {\r
    UpdatePageId (CallbackData, FORM_MAIN_ID);\r
\r
    ActionRequest = EFI_BROWSER_ACTION_REQUEST_NONE;\r
    Status = gFormBrowser2->SendForm (\r
                             gFormBrowser2,\r
                             &CallbackData->BmmHiiHandle,\r
                             1,\r
                             &gBootMaintFormSetGuid,\r
                             0,\r
                             NULL,\r
                             &ActionRequest\r
                             );\r
    if (ActionRequest == EFI_BROWSER_ACTION_REQUEST_RESET) {\r
      EnableResetRequired ();\r
    }\r
\r
    ReclaimStringDepository ();\r
\r
    //\r
    // When this Formset returns, check if we are going to explore files.\r
    //\r
    if (FileExplorerStateInActive != CallbackData->FeCurrentState) {\r
      UpdateFileExplorer (CallbackData, 0);\r
\r
      ActionRequest = EFI_BROWSER_ACTION_REQUEST_NONE;\r
      Status = gFormBrowser2->SendForm (\r
                               gFormBrowser2,\r
                               &CallbackData->FeHiiHandle,\r
                               1,\r
                               &gFileExploreFormSetGuid,\r
                               0,\r
                               NULL,\r
                               &ActionRequest\r
                               );\r
      if (ActionRequest == EFI_BROWSER_ACTION_REQUEST_RESET) {\r
        EnableResetRequired ();\r
      }\r
\r
      CallbackData->FeCurrentState    = FileExplorerStateInActive;\r
      CallbackData->FeDisplayContext  = FileExplorerDisplayUnknown;\r
      ReclaimStringDepository ();\r
    } else {\r
      break;\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Intall BootMaint and FileExplorer HiiPackages.\r
\r
**/\r
EFI_STATUS\r
InitBMPackage (\r
  VOID\r
  )\r
{\r
  BMM_CALLBACK_DATA           *BmmCallbackInfo;\r
  EFI_STATUS                  Status;\r
  UINT8                       *Ptr;\r
\r
  //\r
  // Create CallbackData structures for Driver Callback\r
  //\r
  BmmCallbackInfo = AllocateZeroPool (sizeof (BMM_CALLBACK_DATA));\r
  if (BmmCallbackInfo == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  //\r
  // Create LoadOption in BmmCallbackInfo for Driver Callback\r
  //\r
  Ptr = AllocateZeroPool (sizeof (BM_LOAD_CONTEXT) + sizeof (BM_FILE_CONTEXT) + sizeof (BM_HANDLE_CONTEXT) + sizeof (BM_MENU_ENTRY));\r
  if (Ptr == NULL) {\r
    FreePool (BmmCallbackInfo);\r
    BmmCallbackInfo = NULL;\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  //\r
  // Initialize Bmm callback data.\r
  //\r
  BmmCallbackInfo->LoadContext = (BM_LOAD_CONTEXT *) Ptr;\r
  Ptr += sizeof (BM_LOAD_CONTEXT);\r
\r
  BmmCallbackInfo->FileContext = (BM_FILE_CONTEXT *) Ptr;\r
  Ptr += sizeof (BM_FILE_CONTEXT);\r
\r
  BmmCallbackInfo->HandleContext = (BM_HANDLE_CONTEXT *) Ptr;\r
  Ptr += sizeof (BM_HANDLE_CONTEXT);\r
\r
  BmmCallbackInfo->MenuEntry      = (BM_MENU_ENTRY *) Ptr;\r
\r
  BmmCallbackInfo->Signature                     = BMM_CALLBACK_DATA_SIGNATURE;\r
  BmmCallbackInfo->BmmConfigAccess.ExtractConfig = BootMaintExtractConfig;\r
  BmmCallbackInfo->BmmConfigAccess.RouteConfig   = BootMaintRouteConfig;\r
  BmmCallbackInfo->BmmConfigAccess.Callback      = BootMaintCallback;\r
  BmmCallbackInfo->FeConfigAccess.ExtractConfig  = FakeExtractConfig;\r
  BmmCallbackInfo->FeConfigAccess.RouteConfig    = FileExplorerRouteConfig;\r
  BmmCallbackInfo->FeConfigAccess.Callback       = FileExplorerCallback;\r
\r
  //\r
  // Install Device Path Protocol and Config Access protocol to driver handle\r
  //\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &BmmCallbackInfo->BmmDriverHandle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  &mBmmHiiVendorDevicePath,\r
                  &gEfiHiiConfigAccessProtocolGuid,\r
                  &BmmCallbackInfo->BmmConfigAccess,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Install Device Path Protocol and Config Access protocol to driver handle\r
  //\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &BmmCallbackInfo->FeDriverHandle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  &mFeHiiVendorDevicePath,\r
                  &gEfiHiiConfigAccessProtocolGuid,\r
                  &BmmCallbackInfo->FeConfigAccess,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Post our Boot Maint VFR binary to the HII database.\r
  //\r
  BmmCallbackInfo->BmmHiiHandle = HiiAddPackages (\r
                                    &gBootMaintFormSetGuid,\r
                                    BmmCallbackInfo->BmmDriverHandle,\r
                                    BmBin,\r
                                    BdsDxeStrings,\r
                                    NULL\r
                                    );\r
  ASSERT (BmmCallbackInfo->BmmHiiHandle != NULL);\r
\r
  //\r
  // Post our File Explorer VFR binary to the HII database.\r
  //\r
  BmmCallbackInfo->FeHiiHandle = HiiAddPackages (\r
                                   &gFileExploreFormSetGuid,\r
                                   BmmCallbackInfo->FeDriverHandle,\r
                                   FEBin,\r
                                   BdsDxeStrings,\r
                                   NULL\r
                                   );\r
  ASSERT (BmmCallbackInfo->FeHiiHandle != NULL);\r
\r
  mBmmCallbackInfo = BmmCallbackInfo;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Remvoe the intalled BootMaint and FileExplorer HiiPackages.\r
\r
**/\r
VOID\r
FreeBMPackage (\r
  VOID\r
  )\r
{\r
  BMM_CALLBACK_DATA           *BmmCallbackInfo;\r
\r
  BmmCallbackInfo = mBmmCallbackInfo;\r
\r
  //\r
  // Remove our IFR data from HII database\r
  //\r
  HiiRemovePackages (BmmCallbackInfo->BmmHiiHandle);\r
  HiiRemovePackages (BmmCallbackInfo->FeHiiHandle);\r
\r
  if (BmmCallbackInfo->FeDriverHandle != NULL) {\r
    gBS->UninstallMultipleProtocolInterfaces (\r
           BmmCallbackInfo->FeDriverHandle,\r
           &gEfiDevicePathProtocolGuid,\r
           &mFeHiiVendorDevicePath,\r
           &gEfiHiiConfigAccessProtocolGuid,\r
           &BmmCallbackInfo->FeConfigAccess,\r
           NULL\r
           );\r
  }\r
\r
  if (BmmCallbackInfo->BmmDriverHandle != NULL) {\r
    gBS->UninstallMultipleProtocolInterfaces (\r
           BmmCallbackInfo->BmmDriverHandle,\r
           &gEfiDevicePathProtocolGuid,\r
           &mBmmHiiVendorDevicePath,\r
           &gEfiHiiConfigAccessProtocolGuid,\r
           &BmmCallbackInfo->BmmConfigAccess,\r
           NULL\r
           );\r
  }\r
\r
  FreePool (BmmCallbackInfo->LoadContext);\r
  FreePool (BmmCallbackInfo);\r
\r
  mBmmCallbackInfo = NULL;\r
\r
  return;\r
}\r
\r