X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=IntelFrameworkModulePkg%2FUniversal%2FBdsDxe%2FBootMaint%2FVariable.c;h=c4b83ba0ffa4476d6c2d17a09e9fdb2d45cd9a31;hp=489827f9de92ecb85a0837db9442e33e44284af6;hb=847240774030f9da79140a9d515a64ec817eb6eb;hpb=6aa22a177c32344ff1d5f765183d0da416fcc252 diff --git a/IntelFrameworkModulePkg/Universal/BdsDxe/BootMaint/Variable.c b/IntelFrameworkModulePkg/Universal/BdsDxe/BootMaint/Variable.c index 489827f9de..c4b83ba0ff 100644 --- a/IntelFrameworkModulePkg/Universal/BdsDxe/BootMaint/Variable.c +++ b/IntelFrameworkModulePkg/Universal/BdsDxe/BootMaint/Variable.c @@ -1,8 +1,8 @@ /** @file Variable operation that will be used by bootmaint -Copyright (c) 2004 - 2008, Intel Corporation.
-All rights reserved. This program and the accompanying materials +Copyright (c) 2004 - 2011, Intel Corporation. All rights reserved.
+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 @@ -449,7 +449,7 @@ Var_UpdateConsoleOption ( (EFI_DEVICE_PATH_PROTOCOL *) &Vendor ); ASSERT (TerminalDevicePath != NULL); - ChangeTerminalDevicePath (TerminalDevicePath, TRUE); + ChangeTerminalDevicePath (&TerminalDevicePath, TRUE); ConDevicePath = AppendDevicePathInstance ( ConDevicePath, TerminalDevicePath @@ -642,13 +642,13 @@ Var_UpdateDriverOption ( CallbackData, DriverOptionStrDepository ); - HiiLibNewString (HiiHandle, &NewMenuEntry->DisplayStringToken, NewMenuEntry->DisplayString); + NewMenuEntry->DisplayStringToken = HiiSetString (HiiHandle, 0, NewMenuEntry->DisplayString, NULL); NewMenuEntry->HelpStringToken = GetStringTokenFromDepository ( CallbackData, DriverOptionHelpStrDepository ); - HiiLibNewString (HiiHandle, &NewMenuEntry->HelpStringToken, NewMenuEntry->HelpString); + NewMenuEntry->HelpStringToken = HiiSetString (HiiHandle, 0, NewMenuEntry->HelpString, NULL); if (OptionalDataExist) { Ptr += (UINT8) GetDevicePathSize (CallbackData->LoadContext->FilePathList); @@ -675,11 +675,11 @@ Var_UpdateDriverOption ( ); NewDriverOrderList = AllocateZeroPool (DriverOrderListSize + sizeof (UINT16)); ASSERT (NewDriverOrderList != NULL); - CopyMem (NewDriverOrderList, DriverOrderList, DriverOrderListSize); - NewDriverOrderList[DriverOrderListSize / sizeof (UINT16)] = Index; if (DriverOrderList != NULL) { + CopyMem (NewDriverOrderList, DriverOrderList, DriverOrderListSize); EfiLibDeleteVariable (L"DriverOrder", &gEfiGlobalVariableGuid); } + NewDriverOrderList[DriverOrderListSize / sizeof (UINT16)] = Index; Status = gRT->SetVariable ( L"DriverOrder", @@ -814,13 +814,13 @@ Var_UpdateBootOption ( CallbackData, BootOptionStrDepository ); - HiiLibNewString (CallbackData->FeHiiHandle, &NewMenuEntry->DisplayStringToken, NewMenuEntry->DisplayString); + NewMenuEntry->DisplayStringToken = HiiSetString (CallbackData->FeHiiHandle, 0, NewMenuEntry->DisplayString, NULL); NewMenuEntry->HelpStringToken = GetStringTokenFromDepository ( CallbackData, BootOptionHelpStrDepository ); - HiiLibNewString (CallbackData->FeHiiHandle, &NewMenuEntry->HelpStringToken, NewMenuEntry->HelpString); + NewMenuEntry->HelpStringToken = HiiSetString (CallbackData->FeHiiHandle, 0, NewMenuEntry->HelpString, NULL); if (OptionalDataExist) { Ptr += (UINT8) GetDevicePathSize (CallbackData->LoadContext->FilePathList); @@ -842,14 +842,13 @@ Var_UpdateBootOption ( &gEfiGlobalVariableGuid, &BootOrderListSize ); - + ASSERT (BootOrderList != NULL); NewBootOrderList = AllocateZeroPool (BootOrderListSize + sizeof (UINT16)); ASSERT (NewBootOrderList != NULL); CopyMem (NewBootOrderList, BootOrderList, BootOrderListSize); NewBootOrderList[BootOrderListSize / sizeof (UINT16)] = Index; if (BootOrderList != NULL) { - EfiLibDeleteVariable (L"BootOrder", &gEfiGlobalVariableGuid); FreePool (BootOrderList); } @@ -949,9 +948,10 @@ Var_UpdateBootOrder ( { EFI_STATUS Status; UINT16 Index; + UINT16 OrderIndex; UINT16 *BootOrderList; - UINT16 *NewBootOrderList; UINTN BootOrderListSize; + UINT16 OptionNumber; BootOrderList = NULL; BootOrderListSize = 0; @@ -964,41 +964,40 @@ Var_UpdateBootOrder ( &gEfiGlobalVariableGuid, &BootOrderListSize ); - - NewBootOrderList = AllocateZeroPool (BootOrderListSize); - if (NewBootOrderList == NULL) { + if (BootOrderList == NULL) { return EFI_OUT_OF_RESOURCES; } - // - // If exists, delete it to hold new BootOrder - // - if (BootOrderList != NULL) { - EfiLibDeleteVariable (L"BootOrder", &gEfiGlobalVariableGuid); - FreePool (BootOrderList); - } - ASSERT (BootOptionMenu.MenuNumber <= (sizeof (CallbackData->BmmFakeNvData.OptionOrder) / sizeof (CallbackData->BmmFakeNvData.OptionOrder[0]))); - for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) { - NewBootOrderList[Index] = (UINT16) (CallbackData->BmmFakeNvData.OptionOrder[Index] - 1); + + for (OrderIndex = 0; (OrderIndex < BootOptionMenu.MenuNumber) && (CallbackData->BmmFakeNvData.OptionOrder[OrderIndex] != 0); OrderIndex++) { + for (Index = OrderIndex; Index < BootOrderListSize / sizeof (UINT16); Index++) { + if ((BootOrderList[Index] == (UINT16) (CallbackData->BmmFakeNvData.OptionOrder[OrderIndex] - 1)) && (OrderIndex != Index)) { + OptionNumber = BootOrderList[Index]; + CopyMem (&BootOrderList[OrderIndex + 1], &BootOrderList[OrderIndex], (Index - OrderIndex) * sizeof (UINT16)); + BootOrderList[OrderIndex] = OptionNumber; + } + } } + GroupMultipleLegacyBootOption4SameType ( + BootOrderList, + BootOrderListSize / sizeof (UINT16) + ); + Status = gRT->SetVariable ( L"BootOrder", &gEfiGlobalVariableGuid, VAR_FLAG, BootOrderListSize, - NewBootOrderList + BootOrderList ); - FreePool (NewBootOrderList); - if (EFI_ERROR (Status)) { - return Status; - } + FreePool (BootOrderList); BOpt_FreeMenu (&BootOptionMenu); BOpt_GetBootOptions (CallbackData); - return EFI_SUCCESS; + return Status; } @@ -1050,8 +1049,8 @@ Var_UpdateDriverOrder ( FreePool (DriverOrderList); } - ASSERT (DriverOrderListSize <= (sizeof (CallbackData->BmmFakeNvData.OptionOrder) / sizeof (CallbackData->BmmFakeNvData.OptionOrder[0]))); - for (Index = 0; Index < DriverOrderListSize; Index++) { + ASSERT (DriverOptionMenu.MenuNumber <= (sizeof (CallbackData->BmmFakeNvData.OptionOrder) / sizeof (CallbackData->BmmFakeNvData.OptionOrder[0]))); + for (Index = 0; Index < DriverOptionMenu.MenuNumber; Index++) { NewDriverOrderList[Index] = (UINT16) (CallbackData->BmmFakeNvData.OptionOrder[Index] - 1); } @@ -1072,14 +1071,14 @@ Var_UpdateDriverOrder ( } /** - Update the legacy BBS boot option. L"LegacyDevOrder" and EfiLegacyDevOrderGuid EFI Variable + Update the legacy BBS boot option. VAR_LEGACY_DEV_ORDER and gEfiLegacyDevOrderVariableGuid EFI Variable is udpated with the new Legacy Boot order. The EFI Variable of "Boot####" and gEfiGlobalVariableGuid is also updated. @param CallbackData The context data for BMM. @return EFI_SUCCESS The function completed successfully. - @retval EFI_NOT_FOUND If L"LegacyDevOrder" and EfiLegacyDevOrderGuid EFI Variable can be found. + @retval EFI_NOT_FOUND If VAR_LEGACY_DEV_ORDER and gEfiLegacyDevOrderVariableGuid EFI Variable can be found. @retval EFI_OUT_OF_RESOURCES Fail to allocate memory resource **/ EFI_STATUS @@ -1092,82 +1091,71 @@ Var_UpdateBBSOption ( VOID *BootOptionVar; CHAR16 VarName[100]; UINTN OptionSize; - UINT8 *Ptr; EFI_STATUS Status; - CHAR16 DescString[100]; - CHAR8 DescAsciiString[100]; - UINTN NewOptionSize; - UINT8 *NewOptionPtr; - UINT8 *TempPtr; UINT32 *Attribute; BM_MENU_OPTION *OptionMenu; - BM_LEGACY_DEVICE_CONTEXT *LegacyDeviceContext; UINT8 *LegacyDev; UINT8 *VarData; UINTN VarSize; - BM_MENU_ENTRY *NewMenuEntry; - BM_LEGACY_DEV_ORDER_CONTEXT *DevOrder; + LEGACY_DEV_ORDER_ENTRY *DevOrder; UINT8 *OriginalPtr; UINT8 *DisMap; UINTN Pos; UINTN Bit; UINT16 *NewOrder; UINT16 Tmp; + UINT16 *EnBootOption; + UINTN EnBootOptionCount; + UINT16 *DisBootOption; + UINTN DisBootOptionCount; + UINT16 *BootOrder; - LegacyDeviceContext = NULL; DisMap = NULL; NewOrder = NULL; - if (FORM_SET_FD_ORDER_ID == CallbackData->BmmPreviousPageId) { - OptionMenu = (BM_MENU_OPTION *) &LegacyFDMenu; - LegacyDev = CallbackData->BmmFakeNvData.LegacyFD; - CallbackData->BbsType = BBS_FLOPPY; - } else { - if (FORM_SET_HD_ORDER_ID == CallbackData->BmmPreviousPageId) { + switch (CallbackData->BmmPreviousPageId) { + case FORM_SET_FD_ORDER_ID: + OptionMenu = (BM_MENU_OPTION *) &LegacyFDMenu; + LegacyDev = CallbackData->BmmFakeNvData.LegacyFD; + CallbackData->BbsType = BBS_FLOPPY; + break; + + case FORM_SET_HD_ORDER_ID: OptionMenu = (BM_MENU_OPTION *) &LegacyHDMenu; LegacyDev = CallbackData->BmmFakeNvData.LegacyHD; CallbackData->BbsType = BBS_HARDDISK; - } else { - if (FORM_SET_CD_ORDER_ID == CallbackData->BmmPreviousPageId) { - OptionMenu = (BM_MENU_OPTION *) &LegacyCDMenu; - LegacyDev = CallbackData->BmmFakeNvData.LegacyCD; - CallbackData->BbsType = BBS_CDROM; - } else { - if (FORM_SET_NET_ORDER_ID == CallbackData->BmmPreviousPageId) { - OptionMenu = (BM_MENU_OPTION *) &LegacyNETMenu; - LegacyDev = CallbackData->BmmFakeNvData.LegacyNET; - CallbackData->BbsType = BBS_EMBED_NETWORK; - } else { - OptionMenu = (BM_MENU_OPTION *) &LegacyBEVMenu; - LegacyDev = CallbackData->BmmFakeNvData.LegacyBEV; - CallbackData->BbsType = BBS_BEV_DEVICE; - } - } - } + break; + + case FORM_SET_CD_ORDER_ID: + OptionMenu = (BM_MENU_OPTION *) &LegacyCDMenu; + LegacyDev = CallbackData->BmmFakeNvData.LegacyCD; + CallbackData->BbsType = BBS_CDROM; + break; + + case FORM_SET_NET_ORDER_ID: + OptionMenu = (BM_MENU_OPTION *) &LegacyNETMenu; + LegacyDev = CallbackData->BmmFakeNvData.LegacyNET; + CallbackData->BbsType = BBS_EMBED_NETWORK; + break; + + default: + ASSERT (FORM_SET_BEV_ORDER_ID == CallbackData->BmmPreviousPageId); + OptionMenu = (BM_MENU_OPTION *) &LegacyBEVMenu; + LegacyDev = CallbackData->BmmFakeNvData.LegacyBEV; + CallbackData->BbsType = BBS_BEV_DEVICE; + break; } DisMap = CallbackData->BmmOldFakeNVData.DisableMap; Status = EFI_SUCCESS; - // - // Find the first device's context - // If all devices are disabled( 0xFF == LegacyDev[0]), LegacyDeviceContext can be set to any VariableContext - // because we just use it to fill the desc string, and user can not see the string in UI - // - for (Index = 0; Index < OptionMenu->MenuNumber; Index++) { - NewMenuEntry = BOpt_GetMenuEntry (OptionMenu, Index); - LegacyDeviceContext = (BM_LEGACY_DEVICE_CONTEXT *) NewMenuEntry->VariableContext; - if (0xFF != LegacyDev[0] && LegacyDev[0] == LegacyDeviceContext->Index) { - DEBUG ((DEBUG_ERROR, "DescStr: %s\n", LegacyDeviceContext->Description)); - break; - } - } + // // Update the Variable "LegacyDevOrder" // VarData = (UINT8 *) BdsLibGetVariableAndSize ( VAR_LEGACY_DEV_ORDER, - &EfiLegacyDevOrderGuid, + &gEfiLegacyDevOrderVariableGuid, &VarSize ); @@ -1176,26 +1164,25 @@ Var_UpdateBBSOption ( } OriginalPtr = VarData; - DevOrder = (BM_LEGACY_DEV_ORDER_CONTEXT *) VarData; + DevOrder = (LEGACY_DEV_ORDER_ENTRY *) VarData; - while (VarData < VarData + VarSize) { + while (VarData < OriginalPtr + VarSize) { if (DevOrder->BbsType == CallbackData->BbsType) { break; } - VarData += sizeof (BBS_TYPE); - VarData += *(UINT16 *) VarData; - DevOrder = (BM_LEGACY_DEV_ORDER_CONTEXT *) VarData; + VarData += sizeof (BBS_TYPE) + DevOrder->Length; + DevOrder = (LEGACY_DEV_ORDER_ENTRY *) VarData; } - if (VarData >= VarData + VarSize) { + if (VarData >= OriginalPtr + VarSize) { FreePool (OriginalPtr); return EFI_NOT_FOUND; } - NewOrder = (UINT16 *) AllocateZeroPool (DevOrder->Length - sizeof (UINT16)); + NewOrder = AllocateZeroPool (DevOrder->Length - sizeof (DevOrder->Length)); if (NewOrder == NULL) { - FreePool (VarData); + FreePool (OriginalPtr); return EFI_OUT_OF_RESOURCES; } @@ -1213,8 +1200,7 @@ Var_UpdateBBSOption ( // so we use DisMap to set en/dis state of each item in NewOrder array // for (Index2 = 0; Index2 < OptionMenu->MenuNumber; Index2++) { - Tmp = *(UINT16 *) ((UINT8 *) DevOrder + sizeof (BBS_TYPE) + sizeof (UINT16) + Index2 * sizeof (UINT16)); - Tmp &= 0xFF; + Tmp = (UINT16) (DevOrder->Data[Index2] & 0xFF); Pos = Tmp / 8; Bit = 7 - (Tmp % 8); if ((DisMap[Pos] & (1 << Bit)) != 0) { @@ -1224,157 +1210,115 @@ Var_UpdateBBSOption ( } CopyMem ( - (UINT8 *) DevOrder + sizeof (BBS_TYPE) + sizeof (UINT16), + DevOrder->Data, NewOrder, - DevOrder->Length - sizeof (UINT16) + DevOrder->Length - sizeof (DevOrder->Length) ); FreePool (NewOrder); Status = gRT->SetVariable ( VAR_LEGACY_DEV_ORDER, - &EfiLegacyDevOrderGuid, + &gEfiLegacyDevOrderVariableGuid, VAR_FLAG, VarSize, OriginalPtr ); - FreePool (OriginalPtr); // - // Update Optional Data of Boot#### + // Update BootOrder and Boot####.Attribute // - BootOptionVar = GetLegacyBootOptionVar (CallbackData->BbsType, &Index, &OptionSize); - - if (BootOptionVar != NULL) { - CopyMem ( - DescString, - LegacyDeviceContext->Description, - StrSize (LegacyDeviceContext->Description) - ); - - UnicodeStrToAsciiStr((CONST CHAR16*)&DescString, (CHAR8 *)&DescAsciiString); - - NewOptionSize = sizeof (UINT32) + sizeof (UINT16) + StrSize (DescString) + - sizeof (BBS_BBS_DEVICE_PATH); - NewOptionSize += AsciiStrLen (DescAsciiString) + - END_DEVICE_PATH_LENGTH + sizeof (BBS_TABLE) + sizeof (UINT16); - - UnicodeSPrint (VarName, 100, L"Boot%04x", Index); + // 1. Re-order the Option Number in BootOrder according to Legacy Dev Order + // + ASSERT (OptionMenu->MenuNumber == DevOrder->Length / sizeof (UINT16) - 1); + BootOrder = BdsLibGetVariableAndSize ( + L"BootOrder", + &gEfiGlobalVariableGuid, + &VarSize + ); + ASSERT (BootOrder != NULL); + + DisBootOption = AllocatePool (VarSize); + ASSERT (DisBootOption != NULL); + EnBootOption = AllocatePool (VarSize); + ASSERT (EnBootOption != NULL); + + OrderLegacyBootOption4SameType ( + BootOrder, + VarSize / sizeof (UINT16), + DevOrder->Data, + DevOrder->Length / sizeof (UINT16) - 1, + EnBootOption, + &EnBootOptionCount, + DisBootOption, + &DisBootOptionCount + ); + + Status = gRT->SetVariable ( + L"BootOrder", + &gEfiGlobalVariableGuid, + VAR_FLAG, + VarSize, + BootOrder + ); + ASSERT_EFI_ERROR (Status); - Ptr = BootOptionVar; + FreePool (BootOrder); - Attribute = (UINT32 *) Ptr; - *Attribute |= LOAD_OPTION_ACTIVE; - if (LegacyDev[0] == 0xFF) { - // - // Disable this legacy boot option - // + // + // 2. Deactivate the DisBootOption and activate the EnBootOption + // + for (Index = 0; Index < DisBootOptionCount; Index++) { + UnicodeSPrint (VarName, sizeof (VarName), L"Boot%04x", DisBootOption[Index]); + BootOptionVar = BdsLibGetVariableAndSize ( + VarName, + &gEfiGlobalVariableGuid, + &OptionSize + ); + if (BootOptionVar != NULL) { + Attribute = (UINT32 *) BootOptionVar; *Attribute &= ~LOAD_OPTION_ACTIVE; - } - Ptr += sizeof (UINT32); - - Ptr += sizeof (UINT16); - Ptr += StrSize ((CHAR16 *) Ptr); + Status = gRT->SetVariable ( + VarName, + &gEfiGlobalVariableGuid, + VAR_FLAG, + OptionSize, + BootOptionVar + ); - NewOptionPtr = AllocateZeroPool (NewOptionSize); - if (NewOptionPtr == NULL) { - return EFI_OUT_OF_RESOURCES; + FreePool (BootOptionVar); } + } - TempPtr = NewOptionPtr; - - // - // Attribute - // - CopyMem ( - TempPtr, - BootOptionVar, - sizeof (UINT32) - ); - - TempPtr += sizeof (UINT32); - - // - // BBS device path Length - // - *((UINT16 *) TempPtr) = (UINT16) (sizeof (BBS_BBS_DEVICE_PATH) + - AsciiStrLen (DescAsciiString) + - END_DEVICE_PATH_LENGTH); - - TempPtr += sizeof (UINT16); - - // - // Description string - // - CopyMem ( - TempPtr, - DescString, - StrSize (DescString) - ); - - TempPtr += StrSize (DescString); - - // - // BBS device path - // - CopyMem ( - TempPtr, - Ptr, - sizeof (BBS_BBS_DEVICE_PATH) - ); - - CopyMem ( - ((BBS_BBS_DEVICE_PATH*) TempPtr)->String, - DescAsciiString, - AsciiStrSize (DescAsciiString) - ); - - SetDevicePathNodeLength ( - (EFI_DEVICE_PATH_PROTOCOL *) TempPtr, - sizeof (BBS_BBS_DEVICE_PATH) + AsciiStrLen (DescAsciiString) - ); - - TempPtr += sizeof (BBS_BBS_DEVICE_PATH) + AsciiStrLen (DescAsciiString); - - // - // End node - // - CopyMem ( - TempPtr, - EndDevicePath, - END_DEVICE_PATH_LENGTH - ); - TempPtr += END_DEVICE_PATH_LENGTH; - - // - // Now TempPtr point to optional data, i.e. Bbs Table - // - CopyMem ( - TempPtr, - LegacyDeviceContext->BbsTable, - sizeof (BBS_TABLE) - ); - - // - // Now TempPtr point to BBS index - // - TempPtr += sizeof (BBS_TABLE); - *((UINT16 *) TempPtr) = (UINT16) LegacyDeviceContext->Index; + for (Index = 0; Index < EnBootOptionCount; Index++) { + UnicodeSPrint (VarName, sizeof (VarName), L"Boot%04x", EnBootOption[Index]); + BootOptionVar = BdsLibGetVariableAndSize ( + VarName, + &gEfiGlobalVariableGuid, + &OptionSize + ); + if (BootOptionVar != NULL) { + Attribute = (UINT32 *) BootOptionVar; + *Attribute |= LOAD_OPTION_ACTIVE; - Status = gRT->SetVariable ( - VarName, - &gEfiGlobalVariableGuid, - VAR_FLAG, - NewOptionSize, - NewOptionPtr - ); + Status = gRT->SetVariable ( + VarName, + &gEfiGlobalVariableGuid, + VAR_FLAG, + OptionSize, + BootOptionVar + ); - FreePool (NewOptionPtr); - FreePool (BootOptionVar); + FreePool (BootOptionVar); + } } BOpt_GetBootOptions (CallbackData); + + FreePool (OriginalPtr); + FreePool (EnBootOption); + FreePool (DisBootOption); return Status; } @@ -1400,8 +1344,8 @@ Var_UpdateConMode ( Status = gST->ConOut->QueryMode (gST->ConOut, Mode, &(ModeInfo.Column), &(ModeInfo.Row)); if (!EFI_ERROR(Status)) { - PcdSet32 (PcdConOutColumn, (UINT32) ModeInfo.Column); - PcdSet32 (PcdConOutRow, (UINT32) ModeInfo.Row); + PcdSet32 (PcdSetupConOutColumn, (UINT32) ModeInfo.Column); + PcdSet32 (PcdSetupConOutRow, (UINT32) ModeInfo.Row); } return EFI_SUCCESS;