| /** @file | |
| The functions for Boot Maintainence Main menu. | |
| Copyright (c) 2004 - 2015, Intel Corporation. All rights reserved.<BR> | |
| This program and the accompanying materials | |
| are licensed and made available under the terms and conditions of the BSD License | |
| which accompanies this distribution. The full text of the license may be found at | |
| http://opensource.org/licenses/bsd-license.php | |
| THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, | |
| WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. | |
| **/ | |
| #include "BootMaint.h" | |
| EFI_DEVICE_PATH_PROTOCOL EndDevicePath[] = { | |
| { | |
| END_DEVICE_PATH_TYPE, | |
| END_ENTIRE_DEVICE_PATH_SUBTYPE, | |
| { | |
| END_DEVICE_PATH_LENGTH, | |
| 0 | |
| } | |
| } | |
| }; | |
| HII_VENDOR_DEVICE_PATH mBmmHiiVendorDevicePath = { | |
| { | |
| { | |
| HARDWARE_DEVICE_PATH, | |
| HW_VENDOR_DP, | |
| { | |
| (UINT8) (sizeof (VENDOR_DEVICE_PATH)), | |
| (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8) | |
| } | |
| }, | |
| // | |
| // {165A028F-0BB2-4b5f-8747-77592E3F6499} | |
| // | |
| { 0x165a028f, 0xbb2, 0x4b5f, { 0x87, 0x47, 0x77, 0x59, 0x2e, 0x3f, 0x64, 0x99 } } | |
| }, | |
| { | |
| END_DEVICE_PATH_TYPE, | |
| END_ENTIRE_DEVICE_PATH_SUBTYPE, | |
| { | |
| (UINT8) (END_DEVICE_PATH_LENGTH), | |
| (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8) | |
| } | |
| } | |
| }; | |
| HII_VENDOR_DEVICE_PATH mFeHiiVendorDevicePath = { | |
| { | |
| { | |
| HARDWARE_DEVICE_PATH, | |
| HW_VENDOR_DP, | |
| { | |
| (UINT8) (sizeof (VENDOR_DEVICE_PATH)), | |
| (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8) | |
| } | |
| }, | |
| // | |
| // {91DB4238-B0C8-472e-BBCF-F3A6541010F4} | |
| // | |
| { 0x91db4238, 0xb0c8, 0x472e, { 0xbb, 0xcf, 0xf3, 0xa6, 0x54, 0x10, 0x10, 0xf4 } } | |
| }, | |
| { | |
| END_DEVICE_PATH_TYPE, | |
| END_ENTIRE_DEVICE_PATH_SUBTYPE, | |
| { | |
| (UINT8) (END_DEVICE_PATH_LENGTH), | |
| (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8) | |
| } | |
| } | |
| }; | |
| EFI_GUID mBootMaintGuid = BOOT_MAINT_FORMSET_GUID; | |
| EFI_GUID mFileExplorerGuid = FILE_EXPLORE_FORMSET_GUID; | |
| CHAR16 mBootMaintStorageName[] = L"BmmData"; | |
| CHAR16 mFileExplorerStorageName[] = L"FeData"; | |
| BMM_CALLBACK_DATA *mBmmCallbackInfo = NULL; | |
| BOOLEAN mAllMenuInit = FALSE; | |
| BOOLEAN mEnterFileExplorer = FALSE; | |
| /** | |
| Init all memu. | |
| @param CallbackData The BMM context data. | |
| **/ | |
| VOID | |
| InitAllMenu ( | |
| IN BMM_CALLBACK_DATA *CallbackData | |
| ); | |
| /** | |
| Free up all Menu Option list. | |
| **/ | |
| VOID | |
| FreeAllMenu ( | |
| VOID | |
| ); | |
| /** | |
| Create string tokens for a menu from its help strings and display strings | |
| @param CallbackData The BMM context data. | |
| @param HiiHandle Hii Handle of the package to be updated. | |
| @param MenuOption The Menu whose string tokens need to be created | |
| @retval EFI_SUCCESS String tokens created successfully | |
| @retval others contain some errors | |
| **/ | |
| EFI_STATUS | |
| CreateMenuStringToken ( | |
| IN BMM_CALLBACK_DATA *CallbackData, | |
| IN EFI_HII_HANDLE HiiHandle, | |
| IN BM_MENU_OPTION *MenuOption | |
| ) | |
| { | |
| BM_MENU_ENTRY *NewMenuEntry; | |
| UINTN Index; | |
| for (Index = 0; Index < MenuOption->MenuNumber; Index++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (MenuOption, Index); | |
| NewMenuEntry->DisplayStringToken = HiiSetString ( | |
| HiiHandle, | |
| 0, | |
| NewMenuEntry->DisplayString, | |
| NULL | |
| ); | |
| if (NULL == NewMenuEntry->HelpString) { | |
| NewMenuEntry->HelpStringToken = NewMenuEntry->DisplayStringToken; | |
| } else { | |
| NewMenuEntry->HelpStringToken = HiiSetString ( | |
| HiiHandle, | |
| 0, | |
| NewMenuEntry->HelpString, | |
| NULL | |
| ); | |
| } | |
| } | |
| return EFI_SUCCESS; | |
| } | |
| /** | |
| This function allows a caller to extract the current configuration for one | |
| or more named elements from the target driver. | |
| @param This Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL. | |
| @param Request A null-terminated Unicode string in <ConfigRequest> format. | |
| @param Progress On return, points to a character in the Request string. | |
| Points to the string's null terminator if request was successful. | |
| Points to the most recent '&' before the first failing name/value | |
| pair (or the beginning of the string if the failure is in the | |
| first name/value pair) if the request was not successful. | |
| @param Results A null-terminated Unicode string in <ConfigAltResp> format which | |
| has all values filled in for the names in the Request string. | |
| String to be allocated by the called function. | |
| @retval EFI_SUCCESS The Results is filled with the requested values. | |
| @retval EFI_OUT_OF_RESOURCES Not enough memory to store the results. | |
| @retval EFI_INVALID_PARAMETER Request is NULL, illegal syntax, or unknown name. | |
| @retval EFI_NOT_FOUND Routing data doesn't match any storage in this driver. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| BootMaintExtractConfig ( | |
| IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This, | |
| IN CONST EFI_STRING Request, | |
| OUT EFI_STRING *Progress, | |
| OUT EFI_STRING *Results | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| UINTN BufferSize; | |
| BMM_CALLBACK_DATA *Private; | |
| EFI_STRING ConfigRequestHdr; | |
| EFI_STRING ConfigRequest; | |
| BOOLEAN AllocatedRequest; | |
| UINTN Size; | |
| if (Progress == NULL || Results == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| *Progress = Request; | |
| if ((Request != NULL) && !HiiIsConfigHdrMatch (Request, &mBootMaintGuid, mBootMaintStorageName)) { | |
| return EFI_NOT_FOUND; | |
| } | |
| ConfigRequestHdr = NULL; | |
| ConfigRequest = NULL; | |
| AllocatedRequest = FALSE; | |
| Size = 0; | |
| Private = BMM_CALLBACK_DATA_FROM_THIS (This); | |
| // | |
| // Convert buffer data to <ConfigResp> by helper function BlockToConfig() | |
| // | |
| BufferSize = sizeof (BMM_FAKE_NV_DATA); | |
| ConfigRequest = Request; | |
| if ((Request == NULL) || (StrStr (Request, L"OFFSET") == NULL)) { | |
| // | |
| // Request has no request element, construct full request string. | |
| // Allocate and fill a buffer large enough to hold the <ConfigHdr> template | |
| // followed by "&OFFSET=0&WIDTH=WWWWWWWWWWWWWWWW" followed by a Null-terminator | |
| // | |
| ConfigRequestHdr = HiiConstructConfigHdr (&mBootMaintGuid, mBootMaintStorageName, Private->BmmDriverHandle); | |
| Size = (StrLen (ConfigRequestHdr) + 32 + 1) * sizeof (CHAR16); | |
| ConfigRequest = AllocateZeroPool (Size); | |
| ASSERT (ConfigRequest != NULL); | |
| AllocatedRequest = TRUE; | |
| UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", ConfigRequestHdr, (UINT64)BufferSize); | |
| FreePool (ConfigRequestHdr); | |
| } | |
| Status = gHiiConfigRouting->BlockToConfig ( | |
| gHiiConfigRouting, | |
| ConfigRequest, | |
| (UINT8 *) &Private->BmmFakeNvData, | |
| BufferSize, | |
| Results, | |
| Progress | |
| ); | |
| // | |
| // Free the allocated config request string. | |
| // | |
| if (AllocatedRequest) { | |
| FreePool (ConfigRequest); | |
| ConfigRequest = NULL; | |
| } | |
| // | |
| // Set Progress string to the original request string. | |
| // | |
| if (Request == NULL) { | |
| *Progress = NULL; | |
| } else if (StrStr (Request, L"OFFSET") == NULL) { | |
| *Progress = Request + StrLen (Request); | |
| } | |
| return Status; | |
| } | |
| /** | |
| This function applies changes in a driver's configuration. | |
| Input is a Configuration, which has the routing data for this | |
| driver followed by name / value configuration pairs. The driver | |
| must apply those pairs to its configurable storage. If the | |
| driver's configuration is stored in a linear block of data | |
| and the driver's name / value pairs are in <BlockConfig> | |
| format, it may use the ConfigToBlock helper function (above) to | |
| simplify the job. Currently not implemented. | |
| @param[in] This Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL. | |
| @param[in] Configuration A null-terminated Unicode string in | |
| <ConfigString> format. | |
| @param[out] Progress A pointer to a string filled in with the | |
| offset of the most recent '&' before the | |
| first failing name / value pair (or the | |
| beginn ing of the string if the failure | |
| is in the first name / value pair) or | |
| the terminating NULL if all was | |
| successful. | |
| @retval EFI_SUCCESS The results have been distributed or are | |
| awaiting distribution. | |
| @retval EFI_OUT_OF_RESOURCES Not enough memory to store the | |
| parts of the results that must be | |
| stored awaiting possible future | |
| protocols. | |
| @retval EFI_INVALID_PARAMETERS Passing in a NULL for the | |
| Results parameter would result | |
| in this type of error. | |
| @retval EFI_NOT_FOUND Target for the specified routing data | |
| was not found. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| BootMaintRouteConfig ( | |
| IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This, | |
| IN CONST EFI_STRING Configuration, | |
| OUT EFI_STRING *Progress | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| UINTN BufferSize; | |
| EFI_HII_CONFIG_ROUTING_PROTOCOL *ConfigRouting; | |
| BMM_FAKE_NV_DATA *NewBmmData; | |
| BMM_FAKE_NV_DATA *OldBmmData; | |
| BM_CONSOLE_CONTEXT *NewConsoleContext; | |
| BM_TERMINAL_CONTEXT *NewTerminalContext; | |
| BM_MENU_ENTRY *NewMenuEntry; | |
| BM_LOAD_CONTEXT *NewLoadContext; | |
| UINT16 Index; | |
| BOOLEAN TerminalAttChange; | |
| BMM_CALLBACK_DATA *Private; | |
| if (Progress == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| *Progress = Configuration; | |
| if (Configuration == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| // | |
| // Check routing data in <ConfigHdr>. | |
| // Note: there is no name for Name/Value storage, only GUID will be checked | |
| // | |
| if (!HiiIsConfigHdrMatch (Configuration, &mBootMaintGuid, mBootMaintStorageName)) { | |
| return EFI_NOT_FOUND; | |
| } | |
| Status = gBS->LocateProtocol ( | |
| &gEfiHiiConfigRoutingProtocolGuid, | |
| NULL, | |
| (VOID **)&ConfigRouting | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| return Status; | |
| } | |
| Private = BMM_CALLBACK_DATA_FROM_THIS (This); | |
| // | |
| // Get Buffer Storage data from EFI variable | |
| // | |
| BufferSize = sizeof (BMM_FAKE_NV_DATA); | |
| OldBmmData = &Private->BmmOldFakeNVData; | |
| NewBmmData = &Private->BmmFakeNvData; | |
| // | |
| // Convert <ConfigResp> to buffer data by helper function ConfigToBlock() | |
| // | |
| Status = ConfigRouting->ConfigToBlock ( | |
| ConfigRouting, | |
| Configuration, | |
| (UINT8 *) NewBmmData, | |
| &BufferSize, | |
| Progress | |
| ); | |
| ASSERT_EFI_ERROR (Status); | |
| // | |
| // Compare new and old BMM configuration data and only do action for modified item to | |
| // avoid setting unnecessary non-volatile variable | |
| // | |
| // | |
| // Check data which located in BMM main page and save the settings if need | |
| // | |
| if (CompareMem (&NewBmmData->BootNext, &OldBmmData->BootNext, sizeof (NewBmmData->BootNext)) != 0) { | |
| Status = Var_UpdateBootNext (Private); | |
| } | |
| // | |
| // Check data which located in Boot Options Menu and save the settings if need | |
| // | |
| if (CompareMem (NewBmmData->BootOptionDel, OldBmmData->BootOptionDel, sizeof (NewBmmData->BootOptionDel)) != 0) { | |
| for (Index = 0; | |
| ((Index < BootOptionMenu.MenuNumber) && (Index < (sizeof (NewBmmData->BootOptionDel) / sizeof (NewBmmData->BootOptionDel[0])))); | |
| Index ++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (&BootOptionMenu, Index); | |
| NewLoadContext = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext; | |
| NewLoadContext->Deleted = NewBmmData->BootOptionDel[Index]; | |
| NewBmmData->BootOptionDel[Index] = FALSE; | |
| NewBmmData->BootOptionDelMark[Index] = FALSE; | |
| } | |
| Var_DelBootOption (); | |
| } | |
| if (CompareMem (NewBmmData->BootOptionOrder, OldBmmData->BootOptionOrder, sizeof (NewBmmData->BootOptionOrder)) != 0) { | |
| Status = Var_UpdateBootOrder (Private); | |
| } | |
| if (CompareMem (&NewBmmData->BootTimeOut, &OldBmmData->BootTimeOut, sizeof (NewBmmData->BootTimeOut)) != 0){ | |
| Status = gRT->SetVariable( | |
| L"Timeout", | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| sizeof(UINT16), | |
| &(NewBmmData->BootTimeOut) | |
| ); | |
| ASSERT_EFI_ERROR(Status); | |
| Private->BmmOldFakeNVData.BootTimeOut = NewBmmData->BootTimeOut; | |
| } | |
| // | |
| // Check data which located in Driver Options Menu and save the settings if need | |
| // | |
| if (CompareMem (NewBmmData->DriverOptionDel, OldBmmData->DriverOptionDel, sizeof (NewBmmData->DriverOptionDel)) != 0) { | |
| for (Index = 0; | |
| ((Index < DriverOptionMenu.MenuNumber) && (Index < (sizeof (NewBmmData->DriverOptionDel) / sizeof (NewBmmData->DriverOptionDel[0])))); | |
| Index++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (&DriverOptionMenu, Index); | |
| NewLoadContext = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext; | |
| NewLoadContext->Deleted = NewBmmData->DriverOptionDel[Index]; | |
| NewBmmData->DriverOptionDel[Index] = FALSE; | |
| NewBmmData->DriverOptionDelMark[Index] = FALSE; | |
| } | |
| Var_DelDriverOption (); | |
| } | |
| if (CompareMem (NewBmmData->DriverOptionOrder, OldBmmData->DriverOptionOrder, sizeof (NewBmmData->DriverOptionOrder)) != 0) { | |
| Status = Var_UpdateDriverOrder (Private); | |
| } | |
| if (CompareMem (&NewBmmData->ConsoleOutMode, &OldBmmData->ConsoleOutMode, sizeof (NewBmmData->ConsoleOutMode)) != 0){ | |
| Var_UpdateConMode(Private); | |
| } | |
| TerminalAttChange = FALSE; | |
| for (Index = 0; Index < TerminalMenu.MenuNumber; Index++) { | |
| // | |
| // only need update modified items | |
| // | |
| if (CompareMem (&NewBmmData->COMBaudRate[Index], &OldBmmData->COMBaudRate[Index], sizeof (NewBmmData->COMBaudRate[Index])) == 0 && | |
| CompareMem (&NewBmmData->COMDataRate[Index], &OldBmmData->COMDataRate[Index], sizeof (NewBmmData->COMDataRate[Index])) == 0 && | |
| CompareMem (&NewBmmData->COMStopBits[Index], &OldBmmData->COMStopBits[Index], sizeof (NewBmmData->COMStopBits[Index])) == 0 && | |
| CompareMem (&NewBmmData->COMParity[Index], &OldBmmData->COMParity[Index], sizeof (NewBmmData->COMParity[Index])) == 0 && | |
| CompareMem (&NewBmmData->COMTerminalType[Index], &OldBmmData->COMTerminalType[Index], sizeof (NewBmmData->COMTerminalType[Index])) == 0 && | |
| CompareMem (&NewBmmData->COMFlowControl[Index], &OldBmmData->COMFlowControl[Index], sizeof (NewBmmData->COMFlowControl[Index])) == 0) { | |
| continue; | |
| } | |
| NewMenuEntry = BOpt_GetMenuEntry (&TerminalMenu, Index); | |
| ASSERT (NewMenuEntry != NULL); | |
| NewTerminalContext = (BM_TERMINAL_CONTEXT *) NewMenuEntry->VariableContext; | |
| NewTerminalContext->BaudRateIndex = NewBmmData->COMBaudRate[Index]; | |
| ASSERT (NewBmmData->COMBaudRate[Index] < (sizeof (BaudRateList) / sizeof (BaudRateList[0]))); | |
| NewTerminalContext->BaudRate = BaudRateList[NewBmmData->COMBaudRate[Index]].Value; | |
| NewTerminalContext->DataBitsIndex = NewBmmData->COMDataRate[Index]; | |
| ASSERT (NewBmmData->COMDataRate[Index] < (sizeof (DataBitsList) / sizeof (DataBitsList[0]))); | |
| NewTerminalContext->DataBits = (UINT8) DataBitsList[NewBmmData->COMDataRate[Index]].Value; | |
| NewTerminalContext->StopBitsIndex = NewBmmData->COMStopBits[Index]; | |
| ASSERT (NewBmmData->COMStopBits[Index] < (sizeof (StopBitsList) / sizeof (StopBitsList[0]))); | |
| NewTerminalContext->StopBits = (UINT8) StopBitsList[NewBmmData->COMStopBits[Index]].Value; | |
| NewTerminalContext->ParityIndex = NewBmmData->COMParity[Index]; | |
| ASSERT (NewBmmData->COMParity[Index] < (sizeof (ParityList) / sizeof (ParityList[0]))); | |
| NewTerminalContext->Parity = (UINT8) ParityList[NewBmmData->COMParity[Index]].Value; | |
| NewTerminalContext->TerminalType = NewBmmData->COMTerminalType[Index]; | |
| NewTerminalContext->FlowControl = NewBmmData->COMFlowControl[Index]; | |
| ChangeTerminalDevicePath ( | |
| NewTerminalContext->DevicePath, | |
| FALSE | |
| ); | |
| TerminalAttChange = TRUE; | |
| } | |
| if (TerminalAttChange) { | |
| Var_UpdateConsoleInpOption (); | |
| Var_UpdateConsoleOutOption (); | |
| Var_UpdateErrorOutOption (); | |
| } | |
| // | |
| // Check data which located in Console Options Menu and save the settings if need | |
| // | |
| if (CompareMem (NewBmmData->ConsoleInCheck, OldBmmData->ConsoleInCheck, sizeof (NewBmmData->ConsoleInCheck)) != 0){ | |
| for (Index = 0; Index < ConsoleInpMenu.MenuNumber; Index++){ | |
| NewMenuEntry = BOpt_GetMenuEntry(&ConsoleInpMenu, Index); | |
| NewConsoleContext = (BM_CONSOLE_CONTEXT *)NewMenuEntry->VariableContext; | |
| ASSERT (Index < MAX_MENU_NUMBER); | |
| NewConsoleContext->IsActive = NewBmmData->ConsoleInCheck[Index]; | |
| } | |
| for (Index = 0; Index < TerminalMenu.MenuNumber; Index++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (&TerminalMenu, Index); | |
| NewTerminalContext = (BM_TERMINAL_CONTEXT *) NewMenuEntry->VariableContext; | |
| ASSERT (Index + ConsoleInpMenu.MenuNumber < MAX_MENU_NUMBER); | |
| NewTerminalContext->IsConIn = NewBmmData->ConsoleInCheck[Index + ConsoleInpMenu.MenuNumber]; | |
| } | |
| Var_UpdateConsoleInpOption(); | |
| } | |
| if (CompareMem (NewBmmData->ConsoleOutCheck, OldBmmData->ConsoleOutCheck, sizeof (NewBmmData->ConsoleOutCheck)) != 0){ | |
| for (Index = 0; Index < ConsoleOutMenu.MenuNumber; Index++){ | |
| NewMenuEntry = BOpt_GetMenuEntry(&ConsoleOutMenu, Index); | |
| NewConsoleContext = (BM_CONSOLE_CONTEXT *)NewMenuEntry->VariableContext; | |
| ASSERT (Index < MAX_MENU_NUMBER); | |
| NewConsoleContext->IsActive = NewBmmData->ConsoleOutCheck[Index]; | |
| } | |
| for (Index = 0; Index < TerminalMenu.MenuNumber; Index++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (&TerminalMenu, Index); | |
| NewTerminalContext = (BM_TERMINAL_CONTEXT *) NewMenuEntry->VariableContext; | |
| ASSERT (Index + ConsoleOutMenu.MenuNumber < MAX_MENU_NUMBER); | |
| NewTerminalContext->IsConOut = NewBmmData->ConsoleOutCheck[Index + ConsoleOutMenu.MenuNumber]; | |
| } | |
| Var_UpdateConsoleOutOption(); | |
| } | |
| if (CompareMem (NewBmmData->ConsoleErrCheck, OldBmmData->ConsoleErrCheck, sizeof (NewBmmData->ConsoleErrCheck)) != 0){ | |
| for (Index = 0; Index < ConsoleErrMenu.MenuNumber; Index++){ | |
| NewMenuEntry = BOpt_GetMenuEntry(&ConsoleErrMenu, Index); | |
| NewConsoleContext = (BM_CONSOLE_CONTEXT *)NewMenuEntry->VariableContext; | |
| ASSERT (Index < MAX_MENU_NUMBER); | |
| NewConsoleContext->IsActive = NewBmmData->ConsoleErrCheck[Index]; | |
| } | |
| for (Index = 0; Index < TerminalMenu.MenuNumber; Index++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (&TerminalMenu, Index); | |
| NewTerminalContext = (BM_TERMINAL_CONTEXT *) NewMenuEntry->VariableContext; | |
| ASSERT (Index + ConsoleErrMenu.MenuNumber < MAX_MENU_NUMBER); | |
| NewTerminalContext->IsStdErr = NewBmmData->ConsoleErrCheck[Index + ConsoleErrMenu.MenuNumber]; | |
| } | |
| Var_UpdateErrorOutOption(); | |
| } | |
| // | |
| // After user do the save action, need to update OldBmmData. | |
| // | |
| CopyMem (OldBmmData, NewBmmData, sizeof (BMM_FAKE_NV_DATA)); | |
| return EFI_SUCCESS; | |
| } | |
| /** | |
| This function processes the results of changes in configuration. | |
| @param This Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL. | |
| @param Action Specifies the type of action taken by the browser. | |
| @param QuestionId A unique value which is sent to the original exporting driver | |
| so that it can identify the type of data to expect. | |
| @param Type The type of value for the question. | |
| @param Value A pointer to the data being sent to the original exporting driver. | |
| @param ActionRequest On return, points to the action requested by the callback function. | |
| @retval EFI_SUCCESS The callback successfully handled the action. | |
| @retval EFI_OUT_OF_RESOURCES Not enough storage is available to hold the variable and its data. | |
| @retval EFI_DEVICE_ERROR The variable could not be saved. | |
| @retval EFI_UNSUPPORTED The specified Action is not supported by the callback. | |
| @retval EFI_INVALID_PARAMETER The parameter of Value or ActionRequest is invalid. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| BootMaintCallback ( | |
| IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This, | |
| IN EFI_BROWSER_ACTION Action, | |
| IN EFI_QUESTION_ID QuestionId, | |
| IN UINT8 Type, | |
| IN EFI_IFR_TYPE_VALUE *Value, | |
| OUT EFI_BROWSER_ACTION_REQUEST *ActionRequest | |
| ) | |
| { | |
| BMM_CALLBACK_DATA *Private; | |
| BM_MENU_ENTRY *NewMenuEntry; | |
| BMM_FAKE_NV_DATA *CurrentFakeNVMap; | |
| UINTN Index; | |
| // | |
| //Chech whether exit from FileExplorer and reenter BM,if yes,reclaim string depositories | |
| // | |
| if (Action == EFI_BROWSER_ACTION_FORM_OPEN){ | |
| if(mEnterFileExplorer ){ | |
| ReclaimStringDepository(); | |
| mEnterFileExplorer = FALSE; | |
| } | |
| } | |
| if (Action != EFI_BROWSER_ACTION_CHANGING && Action != EFI_BROWSER_ACTION_CHANGED) { | |
| // | |
| // Do nothing for other UEFI Action. Only do call back when data is changed. | |
| // | |
| return EFI_UNSUPPORTED; | |
| } | |
| Private = BMM_CALLBACK_DATA_FROM_THIS (This); | |
| // | |
| // Retrive uncommitted data from Form Browser | |
| // | |
| CurrentFakeNVMap = &Private->BmmFakeNvData; | |
| HiiGetBrowserData (&mBootMaintGuid, mBootMaintStorageName, sizeof (BMM_FAKE_NV_DATA), (UINT8 *) CurrentFakeNVMap); | |
| if (Action == EFI_BROWSER_ACTION_CHANGING) { | |
| if (Value == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| UpdatePageId (Private, QuestionId); | |
| if (QuestionId < FILE_OPTION_OFFSET) { | |
| if (QuestionId < CONFIG_OPTION_OFFSET) { | |
| switch (QuestionId) { | |
| case FORM_BOOT_ADD_ID: | |
| // Leave Bm and enter FileExplorer. | |
| Private->FeCurrentState = FileExplorerStateAddBootOption; | |
| Private->FeDisplayContext = FileExplorerDisplayUnknown; | |
| ReclaimStringDepository (); | |
| UpdateFileExplorer(Private, 0); | |
| mEnterFileExplorer = TRUE; | |
| break; | |
| case FORM_DRV_ADD_FILE_ID: | |
| // Leave Bm and enter FileExplorer. | |
| Private->FeCurrentState = FileExplorerStateAddDriverOptionState; | |
| Private->FeDisplayContext = FileExplorerDisplayUnknown; | |
| ReclaimStringDepository (); | |
| UpdateFileExplorer(Private, 0); | |
| mEnterFileExplorer = TRUE; | |
| break; | |
| case FORM_DRV_ADD_HANDLE_ID: | |
| CleanUpPage (FORM_DRV_ADD_HANDLE_ID, Private); | |
| UpdateDrvAddHandlePage (Private); | |
| break; | |
| case FORM_BOOT_DEL_ID: | |
| CleanUpPage (FORM_BOOT_DEL_ID, Private); | |
| UpdateBootDelPage (Private); | |
| break; | |
| case FORM_BOOT_CHG_ID: | |
| case FORM_DRV_CHG_ID: | |
| UpdatePageBody (QuestionId, Private); | |
| break; | |
| case FORM_DRV_DEL_ID: | |
| CleanUpPage (FORM_DRV_DEL_ID, Private); | |
| UpdateDrvDelPage (Private); | |
| break; | |
| case FORM_BOOT_NEXT_ID: | |
| CleanUpPage (FORM_BOOT_NEXT_ID, Private); | |
| UpdateBootNextPage (Private); | |
| break; | |
| case FORM_TIME_OUT_ID: | |
| CleanUpPage (FORM_TIME_OUT_ID, Private); | |
| UpdateTimeOutPage (Private); | |
| break; | |
| case FORM_CON_IN_ID: | |
| case FORM_CON_OUT_ID: | |
| case FORM_CON_ERR_ID: | |
| UpdatePageBody (QuestionId, Private); | |
| break; | |
| case FORM_CON_MODE_ID: | |
| CleanUpPage (FORM_CON_MODE_ID, Private); | |
| UpdateConModePage (Private); | |
| break; | |
| case FORM_CON_COM_ID: | |
| CleanUpPage (FORM_CON_COM_ID, Private); | |
| UpdateConCOMPage (Private); | |
| break; | |
| default: | |
| break; | |
| } | |
| } else if ((QuestionId >= TERMINAL_OPTION_OFFSET) && (QuestionId < CONSOLE_OPTION_OFFSET)) { | |
| Index = (UINT16) (QuestionId - TERMINAL_OPTION_OFFSET); | |
| Private->CurrentTerminal = Index; | |
| CleanUpPage (FORM_CON_COM_SETUP_ID, Private); | |
| UpdateTerminalPage (Private); | |
| } else if (QuestionId >= HANDLE_OPTION_OFFSET) { | |
| Index = (UINT16) (QuestionId - HANDLE_OPTION_OFFSET); | |
| NewMenuEntry = BOpt_GetMenuEntry (&DriverMenu, Index); | |
| ASSERT (NewMenuEntry != NULL); | |
| Private->HandleContext = (BM_HANDLE_CONTEXT *) NewMenuEntry->VariableContext; | |
| CleanUpPage (FORM_DRV_ADD_HANDLE_DESC_ID, Private); | |
| Private->MenuEntry = NewMenuEntry; | |
| Private->LoadContext->FilePathList = Private->HandleContext->DevicePath; | |
| UpdateDriverAddHandleDescPage (Private); | |
| } | |
| } | |
| if (QuestionId == KEY_VALUE_BOOT_FROM_FILE){ | |
| // Leave Bm and enter FileExplorer. | |
| Private->FeCurrentState = FileExplorerStateBootFromFile; | |
| Private->FeDisplayContext = FileExplorerDisplayUnknown; | |
| ReclaimStringDepository (); | |
| UpdateFileExplorer(Private, 0); | |
| mEnterFileExplorer = TRUE; | |
| } | |
| } else if (Action == EFI_BROWSER_ACTION_CHANGED) { | |
| if ((Value == NULL) || (ActionRequest == NULL)) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| if ((QuestionId >= BOOT_OPTION_DEL_QUESTION_ID) && (QuestionId < BOOT_OPTION_DEL_QUESTION_ID + MAX_MENU_NUMBER)) { | |
| if (Value->b){ | |
| // | |
| // Means user try to delete this boot option but not press F10 or "Commit Changes and Exit" menu. | |
| // | |
| CurrentFakeNVMap->BootOptionDelMark[QuestionId - BOOT_OPTION_DEL_QUESTION_ID] = TRUE; | |
| } else { | |
| // | |
| // Means user remove the old check status. | |
| // | |
| CurrentFakeNVMap->BootOptionDelMark[QuestionId - BOOT_OPTION_DEL_QUESTION_ID] = FALSE; | |
| } | |
| } else if ((QuestionId >= DRIVER_OPTION_DEL_QUESTION_ID) && (QuestionId < DRIVER_OPTION_DEL_QUESTION_ID + MAX_MENU_NUMBER)) { | |
| if (Value->b){ | |
| CurrentFakeNVMap->DriverOptionDelMark[QuestionId - DRIVER_OPTION_DEL_QUESTION_ID] = TRUE; | |
| } else { | |
| CurrentFakeNVMap->DriverOptionDelMark[QuestionId - DRIVER_OPTION_DEL_QUESTION_ID] = FALSE; | |
| } | |
| } else { | |
| switch (QuestionId) { | |
| case KEY_VALUE_SAVE_AND_EXIT: | |
| case KEY_VALUE_NO_SAVE_AND_EXIT: | |
| if (QuestionId == KEY_VALUE_SAVE_AND_EXIT) { | |
| *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_SUBMIT_EXIT; | |
| } else if (QuestionId == KEY_VALUE_NO_SAVE_AND_EXIT) { | |
| DiscardChangeHandler (Private, CurrentFakeNVMap); | |
| *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_DISCARD_EXIT; | |
| } | |
| break; | |
| case FORM_RESET: | |
| gRT->ResetSystem (EfiResetCold, EFI_SUCCESS, 0, NULL); | |
| return EFI_UNSUPPORTED; | |
| default: | |
| break; | |
| } | |
| } | |
| } | |
| // | |
| // Pass changed uncommitted data back to Form Browser | |
| // | |
| HiiSetBrowserData (&mBootMaintGuid, mBootMaintStorageName, sizeof (BMM_FAKE_NV_DATA), (UINT8 *) CurrentFakeNVMap, NULL); | |
| return EFI_SUCCESS; | |
| } | |
| /** | |
| Discard all changes done to the BMM pages such as Boot Order change, | |
| Driver order change. | |
| @param Private The BMM context data. | |
| @param CurrentFakeNVMap The current Fack NV Map. | |
| **/ | |
| VOID | |
| DiscardChangeHandler ( | |
| IN BMM_CALLBACK_DATA *Private, | |
| IN BMM_FAKE_NV_DATA *CurrentFakeNVMap | |
| ) | |
| { | |
| UINT16 Index; | |
| switch (Private->BmmPreviousPageId) { | |
| case FORM_BOOT_CHG_ID: | |
| CopyMem (CurrentFakeNVMap->BootOptionOrder, Private->BmmOldFakeNVData.BootOptionOrder, sizeof (CurrentFakeNVMap->BootOptionOrder)); | |
| break; | |
| case FORM_DRV_CHG_ID: | |
| CopyMem (CurrentFakeNVMap->DriverOptionOrder, Private->BmmOldFakeNVData.DriverOptionOrder, sizeof (CurrentFakeNVMap->DriverOptionOrder)); | |
| break; | |
| case FORM_BOOT_DEL_ID: | |
| ASSERT (BootOptionMenu.MenuNumber <= (sizeof (CurrentFakeNVMap->BootOptionDel) / sizeof (CurrentFakeNVMap->BootOptionDel[0]))); | |
| for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) { | |
| CurrentFakeNVMap->BootOptionDel[Index] = FALSE; | |
| } | |
| break; | |
| case FORM_DRV_DEL_ID: | |
| ASSERT (DriverOptionMenu.MenuNumber <= (sizeof (CurrentFakeNVMap->DriverOptionDel) / sizeof (CurrentFakeNVMap->DriverOptionDel[0]))); | |
| for (Index = 0; Index < DriverOptionMenu.MenuNumber; Index++) { | |
| CurrentFakeNVMap->DriverOptionDel[Index] = FALSE; | |
| } | |
| break; | |
| case FORM_BOOT_NEXT_ID: | |
| CurrentFakeNVMap->BootNext = Private->BmmOldFakeNVData.BootNext; | |
| break; | |
| case FORM_TIME_OUT_ID: | |
| CurrentFakeNVMap->BootTimeOut = Private->BmmOldFakeNVData.BootTimeOut; | |
| break; | |
| case FORM_DRV_ADD_HANDLE_DESC_ID: | |
| case FORM_DRV_ADD_FILE_ID: | |
| case FORM_DRV_ADD_HANDLE_ID: | |
| CurrentFakeNVMap->DriverAddHandleDesc[0] = 0x0000; | |
| CurrentFakeNVMap->DriverAddHandleOptionalData[0] = 0x0000; | |
| break; | |
| default: | |
| break; | |
| } | |
| } | |
| /** | |
| Create dynamic code for BMM. | |
| @param BmmCallbackInfo The BMM context data. | |
| **/ | |
| VOID | |
| InitializeDrivers( | |
| IN BMM_CALLBACK_DATA *BmmCallbackInfo | |
| ) | |
| { | |
| EFI_HII_HANDLE HiiHandle; | |
| VOID *StartOpCodeHandle; | |
| VOID *EndOpCodeHandle; | |
| EFI_IFR_GUID_LABEL *StartLabel; | |
| EFI_IFR_GUID_LABEL *EndLabel; | |
| UINTN Index; | |
| EFI_STRING_ID FormSetTitle; | |
| EFI_STRING_ID FormSetHelp; | |
| EFI_STRING String; | |
| EFI_STRING_ID Token; | |
| EFI_STRING_ID TokenHelp; | |
| EFI_HII_HANDLE *HiiHandles; | |
| UINTN SkipCount; | |
| EFI_GUID FormSetGuid; | |
| CHAR16 *DevicePathStr; | |
| EFI_STRING_ID DevicePathId; | |
| HiiHandle = BmmCallbackInfo->BmmHiiHandle; | |
| // | |
| // Allocate space for creation of UpdateData Buffer | |
| // | |
| StartOpCodeHandle = HiiAllocateOpCodeHandle (); | |
| ASSERT (StartOpCodeHandle != NULL); | |
| EndOpCodeHandle = HiiAllocateOpCodeHandle (); | |
| ASSERT (EndOpCodeHandle != NULL); | |
| // | |
| // Create Hii Extend Label OpCode as the start opcode | |
| // | |
| StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (StartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL)); | |
| StartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL; | |
| StartLabel->Number = LABEL_BMM_PLATFORM_INFORMATION; | |
| // | |
| // Create Hii Extend Label OpCode as the end opcode | |
| // | |
| EndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (EndOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL)); | |
| EndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL; | |
| EndLabel->Number = LABEL_END; | |
| // | |
| // Get all the Hii handles | |
| // | |
| HiiHandles = HiiGetHiiHandles (NULL); | |
| ASSERT (HiiHandles != NULL); | |
| // | |
| // Search for formset of each class type | |
| // | |
| SkipCount = 0; | |
| for (Index = 0; HiiHandles[Index] != NULL; Index++) { | |
| if (!ExtractDisplayedHiiFormFromHiiHandle (HiiHandles[Index], &gEfiIfrBootMaintenanceGuid, SkipCount, &FormSetTitle, &FormSetHelp, &FormSetGuid)) { | |
| SkipCount = 0; | |
| continue; | |
| } | |
| String = HiiGetString (HiiHandles[Index], FormSetTitle, NULL); | |
| if (String == NULL) { | |
| String = HiiGetString (HiiHandle, STR_MISSING_STRING, NULL); | |
| ASSERT (String != NULL); | |
| } | |
| Token = HiiSetString (HiiHandle, 0, String, NULL); | |
| FreePool (String); | |
| String = HiiGetString (HiiHandles[Index], FormSetHelp, NULL); | |
| if (String == NULL) { | |
| String = HiiGetString (HiiHandle, STR_MISSING_STRING, NULL); | |
| ASSERT (String != NULL); | |
| } | |
| TokenHelp = HiiSetString (HiiHandle, 0, String, NULL); | |
| FreePool (String); | |
| DevicePathStr = ExtractDevicePathFromHiiHandle(HiiHandles[Index]); | |
| DevicePathId = 0; | |
| if (DevicePathStr != NULL){ | |
| DevicePathId = HiiSetString (HiiHandle, 0, DevicePathStr, NULL); | |
| FreePool (DevicePathStr); | |
| } | |
| HiiCreateGotoExOpCode ( | |
| StartOpCodeHandle, | |
| 0, | |
| Token, | |
| TokenHelp, | |
| 0, | |
| (EFI_QUESTION_ID) (Index + FRONT_PAGE_KEY_OFFSET), | |
| 0, | |
| &FormSetGuid, | |
| DevicePathId | |
| ); | |
| // | |
| //One packagelist may has more than one form package, | |
| //Index-- means keep current HiiHandle and still extract from the packagelist, | |
| //SkipCount++ means skip the formset which was found before in the same form package. | |
| // | |
| SkipCount++; | |
| Index--; | |
| } | |
| HiiUpdateForm ( | |
| HiiHandle, | |
| &mBootMaintGuid, | |
| FORM_MAIN_ID, | |
| StartOpCodeHandle, | |
| EndOpCodeHandle | |
| ); | |
| HiiFreeOpCodeHandle (StartOpCodeHandle); | |
| HiiFreeOpCodeHandle (EndOpCodeHandle); | |
| FreePool (HiiHandles); | |
| } | |
| /** | |
| Create dynamic code for BMM and initialize all of BMM configuration data in BmmFakeNvData and | |
| BmmOldFakeNVData member in BMM context data. | |
| @param CallbackData The BMM context data. | |
| **/ | |
| VOID | |
| InitializeBmmConfig ( | |
| IN BMM_CALLBACK_DATA *CallbackData | |
| ) | |
| { | |
| BM_MENU_ENTRY *NewMenuEntry; | |
| BM_LOAD_CONTEXT *NewLoadContext; | |
| UINT16 Index; | |
| ASSERT (CallbackData != NULL); | |
| InitializeDrivers (CallbackData); | |
| // | |
| // Initialize data which located in BMM main page | |
| // | |
| CallbackData->BmmFakeNvData.BootNext = (UINT16) (BootOptionMenu.MenuNumber); | |
| for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) { | |
| NewMenuEntry = BOpt_GetMenuEntry (&BootOptionMenu, Index); | |
| NewLoadContext = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext; | |
| if (NewLoadContext->IsBootNext) { | |
| CallbackData->BmmFakeNvData.BootNext = Index; | |
| break; | |
| } | |
| } | |
| CallbackData->BmmFakeNvData.BootTimeOut = PcdGet16 (PcdPlatformBootTimeOut); | |
| // | |
| // Initialize data which located in Boot Options Menu | |
| // | |
| GetBootOrder (CallbackData); | |
| // | |
| // Initialize data which located in Driver Options Menu | |
| // | |
| GetDriverOrder (CallbackData); | |
| // | |
| // Initialize data which located in Console Options Menu | |
| // | |
| GetConsoleOutMode (CallbackData); | |
| GetConsoleInCheck (CallbackData); | |
| GetConsoleOutCheck (CallbackData); | |
| GetConsoleErrCheck (CallbackData); | |
| GetTerminalAttribute (CallbackData); | |
| // | |
| // Backup Initialize BMM configuartion data to BmmOldFakeNVData | |
| // | |
| CopyMem (&CallbackData->BmmOldFakeNVData, &CallbackData->BmmFakeNvData, sizeof (BMM_FAKE_NV_DATA)); | |
| } | |
| /** | |
| Initialize the Boot Maintenance Utitliy. | |
| **/ | |
| VOID | |
| InitializeBM ( | |
| VOID | |
| ) | |
| { | |
| BMM_CALLBACK_DATA *BmmCallbackInfo; | |
| BmmCallbackInfo = mBmmCallbackInfo; | |
| BmmCallbackInfo->BmmPreviousPageId = FORM_MAIN_ID; | |
| BmmCallbackInfo->BmmCurrentPageId = FORM_MAIN_ID; | |
| BmmCallbackInfo->FeCurrentState = FileExplorerStateInActive; | |
| BmmCallbackInfo->FeDisplayContext = FileExplorerDisplayUnknown; | |
| InitAllMenu (BmmCallbackInfo); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &ConsoleInpMenu); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &ConsoleOutMenu); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &ConsoleErrMenu); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &BootOptionMenu); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &DriverOptionMenu); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &TerminalMenu); | |
| CreateMenuStringToken (BmmCallbackInfo, BmmCallbackInfo->BmmHiiHandle, &DriverMenu); | |
| InitializeBmmConfig(BmmCallbackInfo); | |
| } | |
| /** | |
| Initialized all Menu Option List. | |
| @param CallbackData The BMM context data. | |
| **/ | |
| VOID | |
| InitAllMenu ( | |
| IN BMM_CALLBACK_DATA *CallbackData | |
| ) | |
| { | |
| InitializeListHead (&BootOptionMenu.Head); | |
| InitializeListHead (&DriverOptionMenu.Head); | |
| BOpt_GetBootOptions (CallbackData); | |
| BOpt_GetDriverOptions (CallbackData); | |
| InitializeListHead (&FsOptionMenu.Head); | |
| BOpt_FindDrivers (); | |
| InitializeListHead (&DirectoryMenu.Head); | |
| InitializeListHead (&ConsoleInpMenu.Head); | |
| InitializeListHead (&ConsoleOutMenu.Head); | |
| InitializeListHead (&ConsoleErrMenu.Head); | |
| InitializeListHead (&TerminalMenu.Head); | |
| LocateSerialIo (); | |
| GetAllConsoles (); | |
| mAllMenuInit = TRUE; | |
| } | |
| /** | |
| Free up all Menu Option list. | |
| **/ | |
| VOID | |
| FreeAllMenu ( | |
| VOID | |
| ) | |
| { | |
| if (!mAllMenuInit){ | |
| return; | |
| } | |
| BOpt_FreeMenu (&DirectoryMenu); | |
| BOpt_FreeMenu (&FsOptionMenu); | |
| BOpt_FreeMenu (&BootOptionMenu); | |
| BOpt_FreeMenu (&DriverOptionMenu); | |
| BOpt_FreeMenu (&DriverMenu); | |
| FreeAllConsoles (); | |
| mAllMenuInit = FALSE; | |
| } | |
| /** | |
| Initialize all the string depositories. | |
| **/ | |
| VOID | |
| InitializeStringDepository ( | |
| VOID | |
| ) | |
| { | |
| STRING_DEPOSITORY *StringDepository; | |
| StringDepository = AllocateZeroPool (sizeof (STRING_DEPOSITORY) * STRING_DEPOSITORY_NUMBER); | |
| FileOptionStrDepository = StringDepository++; | |
| ConsoleOptionStrDepository = StringDepository++; | |
| BootOptionStrDepository = StringDepository++; | |
| BootOptionHelpStrDepository = StringDepository++; | |
| DriverOptionStrDepository = StringDepository++; | |
| DriverOptionHelpStrDepository = StringDepository++; | |
| TerminalStrDepository = StringDepository; | |
| } | |
| /** | |
| Fetch a usable string node from the string depository and return the string token. | |
| @param CallbackData The BMM context data. | |
| @param StringDepository The string repository. | |
| @retval EFI_STRING_ID String token. | |
| **/ | |
| EFI_STRING_ID | |
| GetStringTokenFromDepository ( | |
| IN BMM_CALLBACK_DATA *CallbackData, | |
| IN STRING_DEPOSITORY *StringDepository | |
| ) | |
| { | |
| STRING_LIST_NODE *CurrentListNode; | |
| STRING_LIST_NODE *NextListNode; | |
| CurrentListNode = StringDepository->CurrentNode; | |
| if ((NULL != CurrentListNode) && (NULL != CurrentListNode->Next)) { | |
| // | |
| // Fetch one reclaimed node from the list. | |
| // | |
| NextListNode = StringDepository->CurrentNode->Next; | |
| } else { | |
| // | |
| // If there is no usable node in the list, update the list. | |
| // | |
| NextListNode = AllocateZeroPool (sizeof (STRING_LIST_NODE)); | |
| ASSERT (NextListNode != NULL); | |
| NextListNode->StringToken = HiiSetString (CallbackData->BmmHiiHandle, 0, L" ", NULL); | |
| ASSERT (NextListNode->StringToken != 0); | |
| StringDepository->TotalNodeNumber++; | |
| if (NULL == CurrentListNode) { | |
| StringDepository->ListHead = NextListNode; | |
| } else { | |
| CurrentListNode->Next = NextListNode; | |
| } | |
| } | |
| StringDepository->CurrentNode = NextListNode; | |
| return StringDepository->CurrentNode->StringToken; | |
| } | |
| /** | |
| Reclaim string depositories by moving the current node pointer to list head.. | |
| **/ | |
| VOID | |
| ReclaimStringDepository ( | |
| VOID | |
| ) | |
| { | |
| UINTN DepositoryIndex; | |
| STRING_DEPOSITORY *StringDepository; | |
| StringDepository = FileOptionStrDepository; | |
| for (DepositoryIndex = 0; DepositoryIndex < STRING_DEPOSITORY_NUMBER; DepositoryIndex++) { | |
| StringDepository->CurrentNode = StringDepository->ListHead; | |
| StringDepository++; | |
| } | |
| } | |
| /** | |
| Release resource for all the string depositories. | |
| **/ | |
| VOID | |
| CleanUpStringDepository ( | |
| VOID | |
| ) | |
| { | |
| UINTN NodeIndex; | |
| UINTN DepositoryIndex; | |
| STRING_LIST_NODE *CurrentListNode; | |
| STRING_LIST_NODE *NextListNode; | |
| STRING_DEPOSITORY *StringDepository; | |
| // | |
| // Release string list nodes. | |
| // | |
| StringDepository = FileOptionStrDepository; | |
| for (DepositoryIndex = 0; DepositoryIndex < STRING_DEPOSITORY_NUMBER; DepositoryIndex++) { | |
| CurrentListNode = StringDepository->ListHead; | |
| for (NodeIndex = 0; NodeIndex < StringDepository->TotalNodeNumber; NodeIndex++) { | |
| NextListNode = CurrentListNode->Next; | |
| FreePool (CurrentListNode); | |
| CurrentListNode = NextListNode; | |
| } | |
| StringDepository++; | |
| } | |
| // | |
| // Release string depository. | |
| // | |
| FreePool (FileOptionStrDepository); | |
| } | |
| /** | |
| Install BootMaint and FileExplorer HiiPackages. | |
| **/ | |
| VOID | |
| InitBootMaintenance( | |
| VOID | |
| ) | |
| { | |
| BMM_CALLBACK_DATA *BmmCallbackInfo; | |
| EFI_STATUS Status; | |
| UINT8 *Ptr; | |
| Status = EFI_SUCCESS; | |
| if (!gConnectAllHappened){ | |
| EfiBootManagerConnectAll(); | |
| gConnectAllHappened = TRUE; | |
| } | |
| EfiBootManagerRefreshAllBootOption (); | |
| // | |
| // Create CallbackData structures for Driver Callback | |
| // | |
| BmmCallbackInfo = AllocateZeroPool (sizeof (BMM_CALLBACK_DATA)); | |
| ASSERT (BmmCallbackInfo != NULL); | |
| // | |
| // Create LoadOption in BmmCallbackInfo for Driver Callback | |
| // | |
| Ptr = AllocateZeroPool (sizeof (BM_LOAD_CONTEXT) + sizeof (BM_FILE_CONTEXT) + sizeof (BM_HANDLE_CONTEXT) + sizeof (BM_MENU_ENTRY)); | |
| ASSERT (Ptr != NULL); | |
| // | |
| // Initialize Bmm callback data. | |
| // | |
| BmmCallbackInfo->LoadContext = (BM_LOAD_CONTEXT *) Ptr; | |
| Ptr += sizeof (BM_LOAD_CONTEXT); | |
| BmmCallbackInfo->FileContext = (BM_FILE_CONTEXT *) Ptr; | |
| Ptr += sizeof (BM_FILE_CONTEXT); | |
| BmmCallbackInfo->HandleContext = (BM_HANDLE_CONTEXT *) Ptr; | |
| Ptr += sizeof (BM_HANDLE_CONTEXT); | |
| BmmCallbackInfo->MenuEntry = (BM_MENU_ENTRY *) Ptr; | |
| BmmCallbackInfo->Signature = BMM_CALLBACK_DATA_SIGNATURE; | |
| BmmCallbackInfo->BmmConfigAccess.ExtractConfig = BootMaintExtractConfig; | |
| BmmCallbackInfo->BmmConfigAccess.RouteConfig = BootMaintRouteConfig; | |
| BmmCallbackInfo->BmmConfigAccess.Callback = BootMaintCallback; | |
| BmmCallbackInfo->BmmPreviousPageId = FORM_MAIN_ID; | |
| BmmCallbackInfo->BmmCurrentPageId = FORM_MAIN_ID; | |
| BmmCallbackInfo->FeConfigAccess.ExtractConfig = FakeExtractConfig; | |
| BmmCallbackInfo->FeConfigAccess.RouteConfig = FileExplorerRouteConfig; | |
| BmmCallbackInfo->FeConfigAccess.Callback = FileExplorerCallback; | |
| BmmCallbackInfo->FeCurrentState = FileExplorerStateInActive; | |
| BmmCallbackInfo->FeDisplayContext = FileExplorerDisplayUnknown; | |
| // | |
| // Install Device Path Protocol and Config Access protocol to driver handle | |
| // | |
| Status = gBS->InstallMultipleProtocolInterfaces ( | |
| &BmmCallbackInfo->BmmDriverHandle, | |
| &gEfiDevicePathProtocolGuid, | |
| &mBmmHiiVendorDevicePath, | |
| &gEfiHiiConfigAccessProtocolGuid, | |
| &BmmCallbackInfo->BmmConfigAccess, | |
| NULL | |
| ); | |
| ASSERT_EFI_ERROR (Status); | |
| // | |
| // Install Device Path Protocol and Config Access protocol to driver handle | |
| // | |
| Status = gBS->InstallMultipleProtocolInterfaces ( | |
| &BmmCallbackInfo->FeDriverHandle, | |
| &gEfiDevicePathProtocolGuid, | |
| &mFeHiiVendorDevicePath, | |
| &gEfiHiiConfigAccessProtocolGuid, | |
| &BmmCallbackInfo->FeConfigAccess, | |
| NULL | |
| ); | |
| ASSERT_EFI_ERROR (Status); | |
| // | |
| // Post our Boot Maint VFR binary to the HII database. | |
| // | |
| BmmCallbackInfo->BmmHiiHandle = HiiAddPackages ( | |
| &mBootMaintGuid, | |
| BmmCallbackInfo->BmmDriverHandle, | |
| BmBin, | |
| UiAppStrings, | |
| NULL | |
| ); | |
| ASSERT (BmmCallbackInfo->BmmHiiHandle != NULL); | |
| // | |
| // Post our File Explorer VFR binary to the HII database. | |
| // | |
| BmmCallbackInfo->FeHiiHandle = HiiAddPackages ( | |
| &mFileExplorerGuid, | |
| BmmCallbackInfo->FeDriverHandle, | |
| FEBin, | |
| UiAppStrings, | |
| NULL | |
| ); | |
| ASSERT (BmmCallbackInfo->FeHiiHandle != NULL); | |
| // | |
| // Init OpCode Handle and Allocate space for creation of Buffer | |
| // | |
| mStartOpCodeHandle = HiiAllocateOpCodeHandle (); | |
| ASSERT (mStartOpCodeHandle != NULL); | |
| mEndOpCodeHandle = HiiAllocateOpCodeHandle (); | |
| ASSERT (mEndOpCodeHandle != NULL); | |
| // | |
| // Create Hii Extend Label OpCode as the start opcode | |
| // | |
| mStartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (mStartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL)); | |
| mStartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL; | |
| // | |
| // Create Hii Extend Label OpCode as the end opcode | |
| // | |
| mEndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (mEndOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL)); | |
| mEndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL; | |
| mEndLabel->Number = LABEL_END; | |
| mBmmCallbackInfo = BmmCallbackInfo; | |
| InitializeStringDepository (); | |
| } | |
| /** | |
| Remove the installed BootMaint and FileExplorer HiiPackages. | |
| **/ | |
| VOID | |
| FreeBMPackage( | |
| VOID | |
| ) | |
| { | |
| BMM_CALLBACK_DATA *BmmCallbackInfo; | |
| if (mStartOpCodeHandle != NULL) { | |
| HiiFreeOpCodeHandle (mStartOpCodeHandle); | |
| } | |
| if (mEndOpCodeHandle != NULL) { | |
| HiiFreeOpCodeHandle (mEndOpCodeHandle); | |
| } | |
| FreeAllMenu (); | |
| CleanUpStringDepository (); | |
| BmmCallbackInfo = mBmmCallbackInfo; | |
| // | |
| // Remove our IFR data from HII database | |
| // | |
| HiiRemovePackages (BmmCallbackInfo->BmmHiiHandle); | |
| HiiRemovePackages (BmmCallbackInfo->FeHiiHandle); | |
| gBS->UninstallMultipleProtocolInterfaces ( | |
| BmmCallbackInfo->FeDriverHandle, | |
| &gEfiDevicePathProtocolGuid, | |
| &mFeHiiVendorDevicePath, | |
| &gEfiHiiConfigAccessProtocolGuid, | |
| &BmmCallbackInfo->FeConfigAccess, | |
| NULL | |
| ); | |
| gBS->UninstallMultipleProtocolInterfaces ( | |
| BmmCallbackInfo->BmmDriverHandle, | |
| &gEfiDevicePathProtocolGuid, | |
| &mBmmHiiVendorDevicePath, | |
| &gEfiHiiConfigAccessProtocolGuid, | |
| &BmmCallbackInfo->BmmConfigAccess, | |
| NULL | |
| ); | |
| FreePool (BmmCallbackInfo->LoadContext); | |
| FreePool (BmmCallbackInfo); | |
| } | |