| /** @file | |
| BDS Lib functions which relate with create or process the boot option. | |
| Copyright (c) 2004 - 2009, Intel Corporation. <BR> | |
| 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 | |
| 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 "InternalBdsLib.h" | |
| BOOLEAN mEnumBootDevice = FALSE; | |
| /// | |
| /// This GUID is used for an EFI Variable that stores the front device pathes | |
| /// for a partial device path that starts with the HD node. | |
| /// | |
| EFI_GUID mHdBootVariablePrivateGuid = { 0xfab7e9e1, 0x39dd, 0x4f2b, { 0x84, 0x8, 0xe2, 0xe, 0x90, 0x6c, 0xb6, 0xde } }; | |
| /** | |
| Boot the legacy system with the boot option | |
| @param Option The legacy boot option which have BBS device path | |
| @retval EFI_UNSUPPORTED There is no legacybios protocol, do not support | |
| legacy boot. | |
| @retval EFI_STATUS Return the status of LegacyBios->LegacyBoot (). | |
| **/ | |
| EFI_STATUS | |
| BdsLibDoLegacyBoot ( | |
| IN BDS_COMMON_OPTION *Option | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_LEGACY_BIOS_PROTOCOL *LegacyBios; | |
| Status = gBS->LocateProtocol (&gEfiLegacyBiosProtocolGuid, NULL, (VOID **) &LegacyBios); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // If no LegacyBios protocol we do not support legacy boot | |
| // | |
| return EFI_UNSUPPORTED; | |
| } | |
| // | |
| // Notes: if we separate the int 19, then we don't need to refresh BBS | |
| // | |
| BdsRefreshBbsTableForBoot (Option); | |
| // | |
| // Write boot to OS performance data for legacy boot. | |
| // | |
| PERF_CODE ( | |
| WriteBootToOsPerformanceData (); | |
| ); | |
| DEBUG ((DEBUG_INFO | DEBUG_LOAD, "Legacy Boot: %S\n", Option->Description)); | |
| return LegacyBios->LegacyBoot ( | |
| LegacyBios, | |
| (BBS_BBS_DEVICE_PATH *) Option->DevicePath, | |
| Option->LoadOptionsSize, | |
| Option->LoadOptions | |
| ); | |
| } | |
| /** | |
| Internal function to check if the input boot option is a valid EFI NV Boot####. | |
| @param OptionToCheck Boot option to be checked. | |
| @retval TRUE This boot option matches a valid EFI NV Boot####. | |
| @retval FALSE If not. | |
| **/ | |
| BOOLEAN | |
| IsBootOptionValidNVVarialbe ( | |
| IN BDS_COMMON_OPTION *OptionToCheck | |
| ) | |
| { | |
| LIST_ENTRY TempList; | |
| BDS_COMMON_OPTION *BootOption; | |
| BOOLEAN Valid; | |
| CHAR16 OptionName[20]; | |
| Valid = FALSE; | |
| InitializeListHead (&TempList); | |
| UnicodeSPrint (OptionName, sizeof (OptionName), L"Boot%04x", OptionToCheck->BootCurrent); | |
| BootOption = BdsLibVariableToOption (&TempList, OptionName); | |
| if (BootOption == NULL) { | |
| return FALSE; | |
| } | |
| // | |
| // If the Boot Option Number and Device Path matches, OptionToCheck matches a | |
| // valid EFI NV Boot####. | |
| // | |
| if ((OptionToCheck->BootCurrent == BootOption->BootCurrent) && | |
| (CompareMem (OptionToCheck->DevicePath, BootOption->DevicePath, GetDevicePathSize (OptionToCheck->DevicePath)) == 0)) | |
| { | |
| Valid = TRUE; | |
| } | |
| FreePool (BootOption); | |
| return Valid; | |
| } | |
| /** | |
| Process the boot option follow the UEFI specification and | |
| special treat the legacy boot option with BBS_DEVICE_PATH. | |
| @param Option The boot option need to be processed | |
| @param DevicePath The device path which describe where to load the | |
| boot image or the legacy BBS device path to boot | |
| the legacy OS | |
| @param ExitDataSize The size of exit data. | |
| @param ExitData Data returned when Boot image failed. | |
| @retval EFI_SUCCESS Boot from the input boot option successfully. | |
| @retval EFI_NOT_FOUND If the Device Path is not found in the system | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| BdsLibBootViaBootOption ( | |
| IN BDS_COMMON_OPTION *Option, | |
| IN EFI_DEVICE_PATH_PROTOCOL *DevicePath, | |
| OUT UINTN *ExitDataSize, | |
| OUT CHAR16 **ExitData OPTIONAL | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_HANDLE Handle; | |
| EFI_HANDLE ImageHandle; | |
| EFI_DEVICE_PATH_PROTOCOL *FilePath; | |
| EFI_LOADED_IMAGE_PROTOCOL *ImageInfo; | |
| EFI_DEVICE_PATH_PROTOCOL *WorkingDevicePath; | |
| EFI_ACPI_S3_SAVE_PROTOCOL *AcpiS3Save; | |
| LIST_ENTRY TempBootLists; | |
| EFI_SECURITY_ARCH_PROTOCOL *SecurityProtocol; | |
| // | |
| // Record the performance data for End of BDS | |
| // | |
| PERF_END(NULL, "BDS", NULL, 0); | |
| *ExitDataSize = 0; | |
| *ExitData = NULL; | |
| // | |
| // Notes: put EFI64 ROM Shadow Solution | |
| // | |
| EFI64_SHADOW_ALL_LEGACY_ROM (); | |
| // | |
| // Notes: this code can be remove after the s3 script table | |
| // hook on the event EVT_SIGNAL_READY_TO_BOOT or | |
| // EVT_SIGNAL_LEGACY_BOOT | |
| // | |
| Status = gBS->LocateProtocol (&gEfiAcpiS3SaveProtocolGuid, NULL, (VOID **) &AcpiS3Save); | |
| if (!EFI_ERROR (Status)) { | |
| AcpiS3Save->S3Save (AcpiS3Save, NULL); | |
| } | |
| // | |
| // If it's Device Path that starts with a hard drive path, append it with the front part to compose a | |
| // full device path | |
| // | |
| WorkingDevicePath = NULL; | |
| if ((DevicePathType (DevicePath) == MEDIA_DEVICE_PATH) && | |
| (DevicePathSubType (DevicePath) == MEDIA_HARDDRIVE_DP)) { | |
| WorkingDevicePath = BdsExpandPartitionPartialDevicePathToFull ( | |
| (HARDDRIVE_DEVICE_PATH *)DevicePath | |
| ); | |
| if (WorkingDevicePath != NULL) { | |
| DevicePath = WorkingDevicePath; | |
| } | |
| } | |
| // | |
| // Signal the EVT_SIGNAL_READY_TO_BOOT event | |
| // | |
| EfiSignalEventReadyToBoot(); | |
| // | |
| // Set Boot Current | |
| // | |
| if (IsBootOptionValidNVVarialbe (Option)) { | |
| // | |
| // For a temporary boot (i.e. a boot by selected a EFI Shell using "Boot From File"), Boot Current is actually not valid. | |
| // In this case, "BootCurrent" is not created. | |
| // Only create the BootCurrent variable when it points to a valid Boot#### variable. | |
| // | |
| gRT->SetVariable ( | |
| L"BootCurrent", | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS, | |
| sizeof (UINT16), | |
| &Option->BootCurrent | |
| ); | |
| } | |
| ASSERT (Option->DevicePath != NULL); | |
| if ((DevicePathType (Option->DevicePath) == BBS_DEVICE_PATH) && | |
| (DevicePathSubType (Option->DevicePath) == BBS_BBS_DP) | |
| ) { | |
| // | |
| // Check to see if we should legacy BOOT. If yes then do the legacy boot | |
| // | |
| return BdsLibDoLegacyBoot (Option); | |
| } | |
| // | |
| // If the boot option point to Internal FV shell, make sure it is valid | |
| // | |
| Status = BdsLibUpdateFvFileDevicePath (&DevicePath, PcdGetPtr(PcdShellFile)); | |
| if (!EFI_ERROR(Status)) { | |
| if (Option->DevicePath != NULL) { | |
| FreePool(Option->DevicePath); | |
| } | |
| Option->DevicePath = AllocateZeroPool (GetDevicePathSize (DevicePath)); | |
| ASSERT(Option->DevicePath != NULL); | |
| CopyMem (Option->DevicePath, DevicePath, GetDevicePathSize (DevicePath)); | |
| // | |
| // Update the shell boot option | |
| // | |
| InitializeListHead (&TempBootLists); | |
| BdsLibRegisterNewOption (&TempBootLists, DevicePath, L"EFI Internal Shell", L"BootOrder"); | |
| // | |
| // free the temporary device path created by BdsLibUpdateFvFileDevicePath() | |
| // | |
| FreePool (DevicePath); | |
| DevicePath = Option->DevicePath; | |
| } | |
| // | |
| // Measure GPT Table by SAP protocol. | |
| // | |
| Status = gBS->LocateProtocol ( | |
| &gEfiSecurityArchProtocolGuid, | |
| NULL, | |
| (VOID**) &SecurityProtocol | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| Status = SecurityProtocol->FileAuthenticationState (SecurityProtocol, 0, DevicePath); | |
| } | |
| DEBUG_CODE_BEGIN(); | |
| UINTN DevicePathTypeValue; | |
| CHAR16 *HiiString; | |
| CHAR16 *BootStringNumber; | |
| UINTN BufferSize; | |
| DevicePathTypeValue = BdsGetBootTypeFromDevicePath (Option->DevicePath); | |
| // | |
| // store number string of boot option temporary. | |
| // | |
| HiiString = NULL; | |
| switch (DevicePathTypeValue) { | |
| case BDS_EFI_ACPI_FLOPPY_BOOT: | |
| HiiString = L"EFI Floppy"; | |
| break; | |
| case BDS_EFI_MEDIA_CDROM_BOOT: | |
| case BDS_EFI_MESSAGE_SATA_BOOT: | |
| case BDS_EFI_MESSAGE_ATAPI_BOOT: | |
| HiiString = L"EFI DVD/CDROM"; | |
| break; | |
| case BDS_EFI_MESSAGE_USB_DEVICE_BOOT: | |
| HiiString = L"EFI USB Device"; | |
| break; | |
| case BDS_EFI_MESSAGE_SCSI_BOOT: | |
| HiiString = L"EFI SCSI Device"; | |
| break; | |
| case BDS_EFI_MESSAGE_MISC_BOOT: | |
| HiiString = L"EFI Misc Device"; | |
| break; | |
| case BDS_EFI_MESSAGE_MAC_BOOT: | |
| HiiString = L"EFI Network"; | |
| break; | |
| case BBS_DEVICE_PATH: | |
| // | |
| // Do nothing for legacy boot option. | |
| // | |
| break; | |
| default: | |
| DEBUG((EFI_D_INFO, "Can not find HiiString for given device path type 0x%x\n", DevicePathTypeValue)); | |
| } | |
| // | |
| // If found Hii description string then cat Hii string with original description. | |
| // | |
| if (HiiString != NULL) { | |
| BootStringNumber = Option->Description; | |
| BufferSize = StrSize(BootStringNumber); | |
| BufferSize += StrSize(HiiString); | |
| Option->Description = AllocateZeroPool(BufferSize); | |
| StrCpy (Option->Description, HiiString); | |
| if (StrnCmp (BootStringNumber, L"0", 1) != 0) { | |
| StrCat (Option->Description, L" "); | |
| StrCat (Option->Description, BootStringNumber); | |
| } | |
| FreePool (BootStringNumber); | |
| } | |
| DEBUG ((DEBUG_INFO | DEBUG_LOAD, "Booting %S\n", Option->Description)); | |
| DEBUG_CODE_END(); | |
| Status = gBS->LoadImage ( | |
| TRUE, | |
| mBdsImageHandle, | |
| DevicePath, | |
| NULL, | |
| 0, | |
| &ImageHandle | |
| ); | |
| // | |
| // If we didn't find an image directly, we need to try as if it is a removable device boot opotion | |
| // and load the image according to the default boot behavior for removable device. | |
| // | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // check if there is a bootable removable media could be found in this device path , | |
| // and get the bootable media handle | |
| // | |
| Handle = BdsLibGetBootableHandle(DevicePath); | |
| if (Handle == NULL) { | |
| goto Done; | |
| } | |
| // | |
| // Load the default boot file \EFI\BOOT\boot{machinename}.EFI from removable Media | |
| // machinename is ia32, ia64, x64, ... | |
| // | |
| FilePath = FileDevicePath (Handle, EFI_REMOVABLE_MEDIA_FILE_NAME); | |
| if (FilePath != NULL) { | |
| Status = gBS->LoadImage ( | |
| TRUE, | |
| mBdsImageHandle, | |
| FilePath, | |
| NULL, | |
| 0, | |
| &ImageHandle | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // The DevicePath failed, and it's not a valid | |
| // removable media device. | |
| // | |
| goto Done; | |
| } | |
| } | |
| } | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // It there is any error from the Boot attempt exit now. | |
| // | |
| goto Done; | |
| } | |
| // | |
| // Provide the image with it's load options | |
| // | |
| Status = gBS->HandleProtocol (ImageHandle, &gEfiLoadedImageProtocolGuid, (VOID **) &ImageInfo); | |
| ASSERT_EFI_ERROR (Status); | |
| if (Option->LoadOptionsSize != 0) { | |
| ImageInfo->LoadOptionsSize = Option->LoadOptionsSize; | |
| ImageInfo->LoadOptions = Option->LoadOptions; | |
| } | |
| // | |
| // Before calling the image, enable the Watchdog Timer for | |
| // the 5 Minute period | |
| // | |
| gBS->SetWatchdogTimer (5 * 60, 0x0000, 0x00, NULL); | |
| // | |
| // Write boot to OS performance data for UEFI boot | |
| // | |
| PERF_CODE ( | |
| WriteBootToOsPerformanceData (); | |
| ); | |
| Status = gBS->StartImage (ImageHandle, ExitDataSize, ExitData); | |
| DEBUG ((DEBUG_INFO | DEBUG_LOAD, "Image Return Status = %r\n", Status)); | |
| // | |
| // Clear the Watchdog Timer after the image returns | |
| // | |
| gBS->SetWatchdogTimer (0x0000, 0x0000, 0x0000, NULL); | |
| Done: | |
| // | |
| // Clear Boot Current | |
| // | |
| gRT->SetVariable ( | |
| L"BootCurrent", | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS, | |
| 0, | |
| &Option->BootCurrent | |
| ); | |
| return Status; | |
| } | |
| /** | |
| Expand a device path that starts with a hard drive media device path node to be a | |
| full device path that includes the full hardware path to the device. We need | |
| to do this so it can be booted. As an optimization the front match (the part point | |
| to the partition node. E.g. ACPI() /PCI()/ATA()/Partition() ) is saved in a variable | |
| so a connect all is not required on every boot. All successful history device path | |
| which point to partition node (the front part) will be saved. | |
| @param HardDriveDevicePath EFI Device Path to boot, if it starts with a hard | |
| drive media device path. | |
| @return A Pointer to the full device path or NULL if a valid Hard Drive devic path | |
| cannot be found. | |
| **/ | |
| EFI_DEVICE_PATH_PROTOCOL * | |
| EFIAPI | |
| BdsExpandPartitionPartialDevicePathToFull ( | |
| IN HARDDRIVE_DEVICE_PATH *HardDriveDevicePath | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| UINTN BlockIoHandleCount; | |
| EFI_HANDLE *BlockIoBuffer; | |
| EFI_DEVICE_PATH_PROTOCOL *FullDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *BlockIoDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *DevicePath; | |
| UINTN Index; | |
| UINTN InstanceNum; | |
| EFI_DEVICE_PATH_PROTOCOL *CachedDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *TempNewDevicePath; | |
| UINTN CachedDevicePathSize; | |
| BOOLEAN DeviceExist; | |
| BOOLEAN NeedAdjust; | |
| EFI_DEVICE_PATH_PROTOCOL *Instance; | |
| UINTN Size; | |
| FullDevicePath = NULL; | |
| // | |
| // Check if there is prestore 'HDDP' variable. | |
| // If exist, search the front path which point to partition node in the variable instants. | |
| // If fail to find or 'HDDP' not exist, reconnect all and search in all system | |
| // | |
| CachedDevicePath = BdsLibGetVariableAndSize ( | |
| L"HDDP", | |
| &mHdBootVariablePrivateGuid, | |
| &CachedDevicePathSize | |
| ); | |
| if (CachedDevicePath != NULL) { | |
| TempNewDevicePath = CachedDevicePath; | |
| DeviceExist = FALSE; | |
| NeedAdjust = FALSE; | |
| do { | |
| // | |
| // Check every instance of the variable | |
| // First, check whether the instance contain the partition node, which is needed for distinguishing multi | |
| // partial partition boot option. Second, check whether the instance could be connected. | |
| // | |
| Instance = GetNextDevicePathInstance (&TempNewDevicePath, &Size); | |
| if (MatchPartitionDevicePathNode (Instance, HardDriveDevicePath)) { | |
| // | |
| // Connect the device path instance, the device path point to hard drive media device path node | |
| // e.g. ACPI() /PCI()/ATA()/Partition() | |
| // | |
| Status = BdsLibConnectDevicePath (Instance); | |
| if (!EFI_ERROR (Status)) { | |
| DeviceExist = TRUE; | |
| break; | |
| } | |
| } | |
| // | |
| // Come here means the first instance is not matched | |
| // | |
| NeedAdjust = TRUE; | |
| FreePool(Instance); | |
| } while (TempNewDevicePath != NULL); | |
| if (DeviceExist) { | |
| // | |
| // Find the matched device path. | |
| // Append the file path information from the boot option and return the fully expanded device path. | |
| // | |
| DevicePath = NextDevicePathNode ((EFI_DEVICE_PATH_PROTOCOL *) HardDriveDevicePath); | |
| FullDevicePath = AppendDevicePath (Instance, DevicePath); | |
| // | |
| // Adjust the 'HDDP' instances sequence if the matched one is not first one. | |
| // | |
| if (NeedAdjust) { | |
| // | |
| // First delete the matched instance. | |
| // | |
| TempNewDevicePath = CachedDevicePath; | |
| CachedDevicePath = BdsLibDelPartMatchInstance (CachedDevicePath, Instance ); | |
| FreePool (TempNewDevicePath); | |
| // | |
| // Second, append the remaining path after the matched instance | |
| // | |
| TempNewDevicePath = CachedDevicePath; | |
| CachedDevicePath = AppendDevicePathInstance (Instance, CachedDevicePath ); | |
| FreePool (TempNewDevicePath); | |
| // | |
| // Save the matching Device Path so we don't need to do a connect all next time | |
| // | |
| Status = gRT->SetVariable ( | |
| L"HDDP", | |
| &mHdBootVariablePrivateGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| GetDevicePathSize (CachedDevicePath), | |
| CachedDevicePath | |
| ); | |
| } | |
| FreePool (Instance); | |
| FreePool (CachedDevicePath); | |
| return FullDevicePath; | |
| } | |
| } | |
| // | |
| // If we get here we fail to find or 'HDDP' not exist, and now we need | |
| // to search all devices in the system for a matched partition | |
| // | |
| BdsLibConnectAllDriversToAllControllers (); | |
| Status = gBS->LocateHandleBuffer (ByProtocol, &gEfiBlockIoProtocolGuid, NULL, &BlockIoHandleCount, &BlockIoBuffer); | |
| if (EFI_ERROR (Status) || BlockIoHandleCount == 0 || BlockIoBuffer == NULL) { | |
| // | |
| // If there was an error or there are no device handles that support | |
| // the BLOCK_IO Protocol, then return. | |
| // | |
| return NULL; | |
| } | |
| // | |
| // Loop through all the device handles that support the BLOCK_IO Protocol | |
| // | |
| for (Index = 0; Index < BlockIoHandleCount; Index++) { | |
| Status = gBS->HandleProtocol (BlockIoBuffer[Index], &gEfiDevicePathProtocolGuid, (VOID *) &BlockIoDevicePath); | |
| if (EFI_ERROR (Status) || BlockIoDevicePath == NULL) { | |
| continue; | |
| } | |
| if (MatchPartitionDevicePathNode (BlockIoDevicePath, HardDriveDevicePath)) { | |
| // | |
| // Find the matched partition device path | |
| // | |
| DevicePath = NextDevicePathNode ((EFI_DEVICE_PATH_PROTOCOL *) HardDriveDevicePath); | |
| FullDevicePath = AppendDevicePath (BlockIoDevicePath, DevicePath); | |
| // | |
| // Save the matched partition device path in 'HDDP' variable | |
| // | |
| if (CachedDevicePath != NULL) { | |
| // | |
| // Save the matched partition device path as first instance of 'HDDP' variable | |
| // | |
| if (BdsLibMatchDevicePaths (CachedDevicePath, BlockIoDevicePath)) { | |
| TempNewDevicePath = CachedDevicePath; | |
| CachedDevicePath = BdsLibDelPartMatchInstance (CachedDevicePath, BlockIoDevicePath); | |
| FreePool(TempNewDevicePath); | |
| TempNewDevicePath = CachedDevicePath; | |
| CachedDevicePath = AppendDevicePathInstance (BlockIoDevicePath, CachedDevicePath); | |
| FreePool(TempNewDevicePath); | |
| } else { | |
| TempNewDevicePath = CachedDevicePath; | |
| CachedDevicePath = AppendDevicePathInstance (BlockIoDevicePath, CachedDevicePath); | |
| FreePool(TempNewDevicePath); | |
| } | |
| // | |
| // Here limit the device path instance number to 12, which is max number for a system support 3 IDE controller | |
| // If the user try to boot many OS in different HDs or partitions, in theory, the 'HDDP' variable maybe become larger and larger. | |
| // | |
| InstanceNum = 0; | |
| ASSERT (CachedDevicePath != NULL); | |
| TempNewDevicePath = CachedDevicePath; | |
| while (!IsDevicePathEnd (TempNewDevicePath)) { | |
| TempNewDevicePath = NextDevicePathNode (TempNewDevicePath); | |
| // | |
| // Parse one instance | |
| // | |
| while (!IsDevicePathEndType (TempNewDevicePath)) { | |
| TempNewDevicePath = NextDevicePathNode (TempNewDevicePath); | |
| } | |
| InstanceNum++; | |
| // | |
| // If the CachedDevicePath variable contain too much instance, only remain 12 instances. | |
| // | |
| if (InstanceNum >= 12) { | |
| SetDevicePathEndNode (TempNewDevicePath); | |
| break; | |
| } | |
| } | |
| } else { | |
| CachedDevicePath = DuplicateDevicePath (BlockIoDevicePath); | |
| } | |
| // | |
| // Save the matching Device Path so we don't need to do a connect all next time | |
| // | |
| Status = gRT->SetVariable ( | |
| L"HDDP", | |
| &mHdBootVariablePrivateGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| GetDevicePathSize (CachedDevicePath), | |
| CachedDevicePath | |
| ); | |
| break; | |
| } | |
| } | |
| if (CachedDevicePath != NULL) { | |
| FreePool (CachedDevicePath); | |
| } | |
| if (BlockIoBuffer != NULL) { | |
| FreePool (BlockIoBuffer); | |
| } | |
| return FullDevicePath; | |
| } | |
| /** | |
| Check whether there is a instance in BlockIoDevicePath, which contain multi device path | |
| instances, has the same partition node with HardDriveDevicePath device path | |
| @param BlockIoDevicePath Multi device path instances which need to check | |
| @param HardDriveDevicePath A device path which starts with a hard drive media | |
| device path. | |
| @retval TRUE There is a matched device path instance. | |
| @retval FALSE There is no matched device path instance. | |
| **/ | |
| BOOLEAN | |
| EFIAPI | |
| MatchPartitionDevicePathNode ( | |
| IN EFI_DEVICE_PATH_PROTOCOL *BlockIoDevicePath, | |
| IN HARDDRIVE_DEVICE_PATH *HardDriveDevicePath | |
| ) | |
| { | |
| HARDDRIVE_DEVICE_PATH *TmpHdPath; | |
| EFI_DEVICE_PATH_PROTOCOL *DevicePath; | |
| BOOLEAN Match; | |
| EFI_DEVICE_PATH_PROTOCOL *BlockIoHdDevicePathNode; | |
| if ((BlockIoDevicePath == NULL) || (HardDriveDevicePath == NULL)) { | |
| return FALSE; | |
| } | |
| // | |
| // Make PreviousDevicePath == the device path node before the end node | |
| // | |
| DevicePath = BlockIoDevicePath; | |
| BlockIoHdDevicePathNode = NULL; | |
| // | |
| // find the partition device path node | |
| // | |
| while (!IsDevicePathEnd (DevicePath)) { | |
| if ((DevicePathType (DevicePath) == MEDIA_DEVICE_PATH) && | |
| (DevicePathSubType (DevicePath) == MEDIA_HARDDRIVE_DP) | |
| ) { | |
| BlockIoHdDevicePathNode = DevicePath; | |
| break; | |
| } | |
| DevicePath = NextDevicePathNode (DevicePath); | |
| } | |
| if (BlockIoHdDevicePathNode == NULL) { | |
| return FALSE; | |
| } | |
| // | |
| // See if the harddrive device path in blockio matches the orig Hard Drive Node | |
| // | |
| TmpHdPath = (HARDDRIVE_DEVICE_PATH *) BlockIoHdDevicePathNode; | |
| Match = FALSE; | |
| // | |
| // Check for the match | |
| // | |
| if ((TmpHdPath->MBRType == HardDriveDevicePath->MBRType) && | |
| (TmpHdPath->SignatureType == HardDriveDevicePath->SignatureType)) { | |
| switch (TmpHdPath->SignatureType) { | |
| case SIGNATURE_TYPE_GUID: | |
| Match = CompareGuid ((EFI_GUID *)TmpHdPath->Signature, (EFI_GUID *)HardDriveDevicePath->Signature); | |
| break; | |
| case SIGNATURE_TYPE_MBR: | |
| Match = (BOOLEAN)(*((UINT32 *)(&(TmpHdPath->Signature[0]))) == ReadUnaligned32((UINT32 *)(&(HardDriveDevicePath->Signature[0])))); | |
| break; | |
| default: | |
| Match = FALSE; | |
| break; | |
| } | |
| } | |
| return Match; | |
| } | |
| /** | |
| Delete the boot option associated with the handle passed in. | |
| @param Handle The handle which present the device path to create | |
| boot option | |
| @retval EFI_SUCCESS Delete the boot option success | |
| @retval EFI_NOT_FOUND If the Device Path is not found in the system | |
| @retval EFI_OUT_OF_RESOURCES Lack of memory resource | |
| @retval Other Error return value from SetVariable() | |
| **/ | |
| EFI_STATUS | |
| BdsLibDeleteOptionFromHandle ( | |
| IN EFI_HANDLE Handle | |
| ) | |
| { | |
| UINT16 *BootOrder; | |
| UINT8 *BootOptionVar; | |
| UINTN BootOrderSize; | |
| UINTN BootOptionSize; | |
| EFI_STATUS Status; | |
| UINTN Index; | |
| UINT16 BootOption[BOOT_OPTION_MAX_CHAR]; | |
| UINTN DevicePathSize; | |
| UINTN OptionDevicePathSize; | |
| EFI_DEVICE_PATH_PROTOCOL *DevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *OptionDevicePath; | |
| UINT8 *TempPtr; | |
| Status = EFI_SUCCESS; | |
| BootOrder = NULL; | |
| BootOrderSize = 0; | |
| // | |
| // Check "BootOrder" variable, if no, means there is no any boot order. | |
| // | |
| BootOrder = BdsLibGetVariableAndSize ( | |
| L"BootOrder", | |
| &gEfiGlobalVariableGuid, | |
| &BootOrderSize | |
| ); | |
| if (BootOrder == NULL) { | |
| return EFI_NOT_FOUND; | |
| } | |
| // | |
| // Convert device handle to device path protocol instance | |
| // | |
| DevicePath = DevicePathFromHandle (Handle); | |
| if (DevicePath == NULL) { | |
| return EFI_NOT_FOUND; | |
| } | |
| DevicePathSize = GetDevicePathSize (DevicePath); | |
| // | |
| // Loop all boot order variable and find the matching device path | |
| // | |
| Index = 0; | |
| while (Index < BootOrderSize / sizeof (UINT16)) { | |
| UnicodeSPrint (BootOption, sizeof (BootOption), L"Boot%04x", BootOrder[Index]); | |
| BootOptionVar = BdsLibGetVariableAndSize ( | |
| BootOption, | |
| &gEfiGlobalVariableGuid, | |
| &BootOptionSize | |
| ); | |
| if (BootOptionVar == NULL) { | |
| FreePool (BootOrder); | |
| return EFI_OUT_OF_RESOURCES; | |
| } | |
| TempPtr = BootOptionVar; | |
| TempPtr += sizeof (UINT32) + sizeof (UINT16); | |
| TempPtr += StrSize ((CHAR16 *) TempPtr); | |
| OptionDevicePath = (EFI_DEVICE_PATH_PROTOCOL *) TempPtr; | |
| OptionDevicePathSize = GetDevicePathSize (OptionDevicePath); | |
| // | |
| // Check whether the device path match | |
| // | |
| if ((OptionDevicePathSize == DevicePathSize) && | |
| (CompareMem (DevicePath, OptionDevicePath, DevicePathSize) == 0)) { | |
| BdsDeleteBootOption (BootOrder[Index], BootOrder, &BootOrderSize); | |
| FreePool (BootOptionVar); | |
| break; | |
| } | |
| FreePool (BootOptionVar); | |
| Index++; | |
| } | |
| // | |
| // Adjust number of boot option for "BootOrder" variable. | |
| // | |
| Status = gRT->SetVariable ( | |
| L"BootOrder", | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| BootOrderSize, | |
| BootOrder | |
| ); | |
| FreePool (BootOrder); | |
| return Status; | |
| } | |
| /** | |
| Delete all invalid EFI boot options. | |
| @retval EFI_SUCCESS Delete all invalid boot option success | |
| @retval EFI_NOT_FOUND Variable "BootOrder" is not found | |
| @retval EFI_OUT_OF_RESOURCES Lack of memory resource | |
| @retval Other Error return value from SetVariable() | |
| **/ | |
| EFI_STATUS | |
| BdsDeleteAllInvalidEfiBootOption ( | |
| VOID | |
| ) | |
| { | |
| UINT16 *BootOrder; | |
| UINT8 *BootOptionVar; | |
| UINTN BootOrderSize; | |
| UINTN BootOptionSize; | |
| EFI_STATUS Status; | |
| UINTN Index; | |
| UINTN Index2; | |
| UINT16 BootOption[BOOT_OPTION_MAX_CHAR]; | |
| EFI_DEVICE_PATH_PROTOCOL *OptionDevicePath; | |
| UINT8 *TempPtr; | |
| CHAR16 *Description; | |
| Status = EFI_SUCCESS; | |
| BootOrder = NULL; | |
| BootOrderSize = 0; | |
| // | |
| // Check "BootOrder" variable firstly, this variable hold the number of boot options | |
| // | |
| BootOrder = BdsLibGetVariableAndSize ( | |
| L"BootOrder", | |
| &gEfiGlobalVariableGuid, | |
| &BootOrderSize | |
| ); | |
| if (NULL == BootOrder) { | |
| return EFI_NOT_FOUND; | |
| } | |
| Index = 0; | |
| while (Index < BootOrderSize / sizeof (UINT16)) { | |
| UnicodeSPrint (BootOption, sizeof (BootOption), L"Boot%04x", BootOrder[Index]); | |
| BootOptionVar = BdsLibGetVariableAndSize ( | |
| BootOption, | |
| &gEfiGlobalVariableGuid, | |
| &BootOptionSize | |
| ); | |
| if (NULL == BootOptionVar) { | |
| FreePool (BootOrder); | |
| return EFI_OUT_OF_RESOURCES; | |
| } | |
| TempPtr = BootOptionVar; | |
| TempPtr += sizeof (UINT32) + sizeof (UINT16); | |
| Description = (CHAR16 *) TempPtr; | |
| TempPtr += StrSize ((CHAR16 *) TempPtr); | |
| OptionDevicePath = (EFI_DEVICE_PATH_PROTOCOL *) TempPtr; | |
| // | |
| // Skip legacy boot option (BBS boot device) | |
| // | |
| if ((DevicePathType (OptionDevicePath) == BBS_DEVICE_PATH) && | |
| (DevicePathSubType (OptionDevicePath) == BBS_BBS_DP)) { | |
| FreePool (BootOptionVar); | |
| Index++; | |
| continue; | |
| } | |
| if (!BdsLibIsValidEFIBootOptDevicePathExt (OptionDevicePath, FALSE, Description)) { | |
| // | |
| // Delete this invalid boot option "Boot####" | |
| // | |
| Status = gRT->SetVariable ( | |
| BootOption, | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| 0, | |
| NULL | |
| ); | |
| // | |
| // Mark this boot option in boot order as deleted | |
| // | |
| BootOrder[Index] = 0xffff; | |
| } | |
| FreePool (BootOptionVar); | |
| Index++; | |
| } | |
| // | |
| // Adjust boot order array | |
| // | |
| Index2 = 0; | |
| for (Index = 0; Index < BootOrderSize / sizeof (UINT16); Index++) { | |
| if (BootOrder[Index] != 0xffff) { | |
| BootOrder[Index2] = BootOrder[Index]; | |
| Index2 ++; | |
| } | |
| } | |
| Status = gRT->SetVariable ( | |
| L"BootOrder", | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| Index2 * sizeof (UINT16), | |
| BootOrder | |
| ); | |
| FreePool (BootOrder); | |
| return Status; | |
| } | |
| /** | |
| For EFI boot option, BDS separate them as six types: | |
| 1. Network - The boot option points to the SimpleNetworkProtocol device. | |
| Bds will try to automatically create this type boot option when enumerate. | |
| 2. Shell - The boot option points to internal flash shell. | |
| Bds will try to automatically create this type boot option when enumerate. | |
| 3. Removable BlockIo - The boot option only points to the removable media | |
| device, like USB flash disk, DVD, Floppy etc. | |
| These device should contain a *removable* blockIo | |
| protocol in their device handle. | |
| Bds will try to automatically create this type boot option | |
| when enumerate. | |
| 4. Fixed BlockIo - The boot option only points to a Fixed blockIo device, | |
| like HardDisk. | |
| These device should contain a *fixed* blockIo | |
| protocol in their device handle. | |
| BDS will skip fixed blockIo devices, and NOT | |
| automatically create boot option for them. But BDS | |
| will help to delete those fixed blockIo boot option, | |
| whose description rule conflict with other auto-created | |
| boot options. | |
| 5. Non-BlockIo Simplefile - The boot option points to a device whose handle | |
| has SimpleFileSystem Protocol, but has no blockio | |
| protocol. These devices do not offer blockIo | |
| protocol, but BDS still can get the | |
| \EFI\BOOT\boot{machinename}.EFI by SimpleFileSystem | |
| Protocol. | |
| 6. File - The boot option points to a file. These boot options are usually | |
| created by user manually or OS loader. BDS will not delete or modify | |
| these boot options. | |
| This function will enumerate all possible boot device in the system, and | |
| automatically create boot options for Network, Shell, Removable BlockIo, | |
| and Non-BlockIo Simplefile devices. | |
| It will only execute once of every boot. | |
| @param BdsBootOptionList The header of the link list which indexed all | |
| current boot options | |
| @retval EFI_SUCCESS Finished all the boot device enumerate and create | |
| the boot option base on that boot device | |
| @retval EFI_OUT_OF_RESOURCES Failed to enumerate the boot device and create the boot option list | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| BdsLibEnumerateAllBootOption ( | |
| IN OUT LIST_ENTRY *BdsBootOptionList | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| UINT16 FloppyNumber; | |
| UINT16 CdromNumber; | |
| UINT16 UsbNumber; | |
| UINT16 MiscNumber; | |
| UINT16 ScsiNumber; | |
| UINT16 NonBlockNumber; | |
| UINTN NumberBlockIoHandles; | |
| EFI_HANDLE *BlockIoHandles; | |
| EFI_BLOCK_IO_PROTOCOL *BlkIo; | |
| UINTN Index; | |
| UINTN NumberSimpleNetworkHandles; | |
| EFI_HANDLE *SimpleNetworkHandles; | |
| UINTN FvHandleCount; | |
| EFI_HANDLE *FvHandleBuffer; | |
| EFI_FV_FILETYPE Type; | |
| UINTN Size; | |
| EFI_FV_FILE_ATTRIBUTES Attributes; | |
| UINT32 AuthenticationStatus; | |
| EFI_FIRMWARE_VOLUME2_PROTOCOL *Fv; | |
| EFI_DEVICE_PATH_PROTOCOL *DevicePath; | |
| UINTN DevicePathType; | |
| CHAR16 Buffer[40]; | |
| EFI_HANDLE *FileSystemHandles; | |
| UINTN NumberFileSystemHandles; | |
| BOOLEAN NeedDelete; | |
| EFI_IMAGE_DOS_HEADER DosHeader; | |
| EFI_IMAGE_OPTIONAL_HEADER_UNION HdrData; | |
| EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr; | |
| FloppyNumber = 0; | |
| CdromNumber = 0; | |
| UsbNumber = 0; | |
| MiscNumber = 0; | |
| ScsiNumber = 0; | |
| ZeroMem (Buffer, sizeof (Buffer)); | |
| // | |
| // If the boot device enumerate happened, just get the boot | |
| // device from the boot order variable | |
| // | |
| if (mEnumBootDevice) { | |
| Status = BdsLibBuildOptionFromVar (BdsBootOptionList, L"BootOrder"); | |
| return Status; | |
| } | |
| // | |
| // Notes: this dirty code is to get the legacy boot option from the | |
| // BBS table and create to variable as the EFI boot option, it should | |
| // be removed after the CSM can provide legacy boot option directly | |
| // | |
| REFRESH_LEGACY_BOOT_OPTIONS; | |
| // | |
| // Delete invalid boot option | |
| // | |
| BdsDeleteAllInvalidEfiBootOption (); | |
| // | |
| // Parse removable media | |
| // | |
| gBS->LocateHandleBuffer ( | |
| ByProtocol, | |
| &gEfiBlockIoProtocolGuid, | |
| NULL, | |
| &NumberBlockIoHandles, | |
| &BlockIoHandles | |
| ); | |
| for (Index = 0; Index < NumberBlockIoHandles; Index++) { | |
| Status = gBS->HandleProtocol ( | |
| BlockIoHandles[Index], | |
| &gEfiBlockIoProtocolGuid, | |
| (VOID **) &BlkIo | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| if (!BlkIo->Media->RemovableMedia) { | |
| // | |
| // skip the non-removable block devices | |
| // | |
| continue; | |
| } | |
| } | |
| DevicePath = DevicePathFromHandle (BlockIoHandles[Index]); | |
| DevicePathType = BdsGetBootTypeFromDevicePath (DevicePath); | |
| switch (DevicePathType) { | |
| case BDS_EFI_ACPI_FLOPPY_BOOT: | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"%d", FloppyNumber); | |
| BdsLibBuildOptionFromHandle (BlockIoHandles[Index], BdsBootOptionList, Buffer); | |
| FloppyNumber++; | |
| break; | |
| // | |
| // Assume a removable SATA device should be the DVD/CD device | |
| // | |
| case BDS_EFI_MESSAGE_ATAPI_BOOT: | |
| case BDS_EFI_MESSAGE_SATA_BOOT: | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"%d", CdromNumber); | |
| BdsLibBuildOptionFromHandle (BlockIoHandles[Index], BdsBootOptionList, Buffer); | |
| CdromNumber++; | |
| break; | |
| case BDS_EFI_MESSAGE_USB_DEVICE_BOOT: | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"%d", UsbNumber); | |
| BdsLibBuildOptionFromHandle (BlockIoHandles[Index], BdsBootOptionList, Buffer); | |
| UsbNumber++; | |
| break; | |
| case BDS_EFI_MESSAGE_SCSI_BOOT: | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"%d", ScsiNumber); | |
| BdsLibBuildOptionFromHandle (BlockIoHandles[Index], BdsBootOptionList, Buffer); | |
| ScsiNumber++; | |
| break; | |
| case BDS_EFI_MESSAGE_MISC_BOOT: | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"%d", MiscNumber); | |
| BdsLibBuildOptionFromHandle (BlockIoHandles[Index], BdsBootOptionList, Buffer); | |
| MiscNumber++; | |
| break; | |
| default: | |
| break; | |
| } | |
| } | |
| if (NumberBlockIoHandles != 0) { | |
| FreePool (BlockIoHandles); | |
| } | |
| // | |
| // If there is simple file protocol which does not consume block Io protocol, create a boot option for it here. | |
| // | |
| NonBlockNumber = 0; | |
| gBS->LocateHandleBuffer ( | |
| ByProtocol, | |
| &gEfiSimpleFileSystemProtocolGuid, | |
| NULL, | |
| &NumberFileSystemHandles, | |
| &FileSystemHandles | |
| ); | |
| for (Index = 0; Index < NumberFileSystemHandles; Index++) { | |
| Status = gBS->HandleProtocol ( | |
| FileSystemHandles[Index], | |
| &gEfiBlockIoProtocolGuid, | |
| (VOID **) &BlkIo | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| // | |
| // Skip if the file system handle supports a BlkIo protocol, | |
| // | |
| continue; | |
| } | |
| // | |
| // Do the removable Media thing. \EFI\BOOT\boot{machinename}.EFI | |
| // machinename is ia32, ia64, x64, ... | |
| // | |
| Hdr.Union = &HdrData; | |
| NeedDelete = TRUE; | |
| Status = BdsLibGetImageHeader ( | |
| FileSystemHandles[Index], | |
| EFI_REMOVABLE_MEDIA_FILE_NAME, | |
| &DosHeader, | |
| Hdr | |
| ); | |
| if (!EFI_ERROR (Status) && | |
| EFI_IMAGE_MACHINE_TYPE_SUPPORTED (Hdr.Pe32->FileHeader.Machine) && | |
| Hdr.Pe32->OptionalHeader.Subsystem == EFI_IMAGE_SUBSYSTEM_EFI_APPLICATION) { | |
| NeedDelete = FALSE; | |
| } | |
| if (NeedDelete) { | |
| // | |
| // No such file or the file is not a EFI application, delete this boot option | |
| // | |
| BdsLibDeleteOptionFromHandle (FileSystemHandles[Index]); | |
| } else { | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"EFI Non-Block Boot Device %d", NonBlockNumber); | |
| BdsLibBuildOptionFromHandle (FileSystemHandles[Index], BdsBootOptionList, Buffer); | |
| NonBlockNumber++; | |
| } | |
| } | |
| if (NumberFileSystemHandles != 0) { | |
| FreePool (FileSystemHandles); | |
| } | |
| // | |
| // Parse Network Boot Device | |
| // | |
| gBS->LocateHandleBuffer ( | |
| ByProtocol, | |
| &gEfiSimpleNetworkProtocolGuid, | |
| NULL, | |
| &NumberSimpleNetworkHandles, | |
| &SimpleNetworkHandles | |
| ); | |
| for (Index = 0; Index < NumberSimpleNetworkHandles; Index++) { | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"%d", Index); | |
| BdsLibBuildOptionFromHandle (SimpleNetworkHandles[Index], BdsBootOptionList, Buffer); | |
| } | |
| if (NumberSimpleNetworkHandles != 0) { | |
| FreePool (SimpleNetworkHandles); | |
| } | |
| // | |
| // Check if we have on flash shell | |
| // | |
| gBS->LocateHandleBuffer ( | |
| ByProtocol, | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| NULL, | |
| &FvHandleCount, | |
| &FvHandleBuffer | |
| ); | |
| for (Index = 0; Index < FvHandleCount; Index++) { | |
| gBS->HandleProtocol ( | |
| FvHandleBuffer[Index], | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| (VOID **) &Fv | |
| ); | |
| Status = Fv->ReadFile ( | |
| Fv, | |
| PcdGetPtr(PcdShellFile), | |
| NULL, | |
| &Size, | |
| &Type, | |
| &Attributes, | |
| &AuthenticationStatus | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Skip if no shell file in the FV | |
| // | |
| continue; | |
| } | |
| // | |
| // Build the shell boot option | |
| // | |
| BdsLibBuildOptionFromShell (FvHandleBuffer[Index], BdsBootOptionList); | |
| } | |
| if (FvHandleCount != 0) { | |
| FreePool (FvHandleBuffer); | |
| } | |
| // | |
| // Make sure every boot only have one time | |
| // boot device enumerate | |
| // | |
| Status = BdsLibBuildOptionFromVar (BdsBootOptionList, L"BootOrder"); | |
| mEnumBootDevice = TRUE; | |
| return Status; | |
| } | |
| /** | |
| Build the boot option with the handle parsed in | |
| @param Handle The handle which present the device path to create | |
| boot option | |
| @param BdsBootOptionList The header of the link list which indexed all | |
| current boot options | |
| @param String The description of the boot option. | |
| **/ | |
| VOID | |
| EFIAPI | |
| BdsLibBuildOptionFromHandle ( | |
| IN EFI_HANDLE Handle, | |
| IN LIST_ENTRY *BdsBootOptionList, | |
| IN CHAR16 *String | |
| ) | |
| { | |
| EFI_DEVICE_PATH_PROTOCOL *DevicePath; | |
| DevicePath = DevicePathFromHandle (Handle); | |
| // | |
| // Create and register new boot option | |
| // | |
| BdsLibRegisterNewOption (BdsBootOptionList, DevicePath, String, L"BootOrder"); | |
| } | |
| /** | |
| Build the on flash shell boot option with the handle parsed in. | |
| @param Handle The handle which present the device path to create | |
| on flash shell boot option | |
| @param BdsBootOptionList The header of the link list which indexed all | |
| current boot options | |
| **/ | |
| VOID | |
| EFIAPI | |
| BdsLibBuildOptionFromShell ( | |
| IN EFI_HANDLE Handle, | |
| IN OUT LIST_ENTRY *BdsBootOptionList | |
| ) | |
| { | |
| EFI_DEVICE_PATH_PROTOCOL *DevicePath; | |
| MEDIA_FW_VOL_FILEPATH_DEVICE_PATH ShellNode; | |
| DevicePath = DevicePathFromHandle (Handle); | |
| // | |
| // Build the shell device path | |
| // | |
| EfiInitializeFwVolDevicepathNode (&ShellNode, PcdGetPtr(PcdShellFile)); | |
| DevicePath = AppendDevicePathNode (DevicePath, (EFI_DEVICE_PATH_PROTOCOL *) &ShellNode); | |
| // | |
| // Create and register the shell boot option | |
| // | |
| BdsLibRegisterNewOption (BdsBootOptionList, DevicePath, L"EFI Internal Shell", L"BootOrder"); | |
| } | |
| /** | |
| Boot from the UEFI spec defined "BootNext" variable. | |
| **/ | |
| VOID | |
| EFIAPI | |
| BdsLibBootNext ( | |
| VOID | |
| ) | |
| { | |
| UINT16 *BootNext; | |
| UINTN BootNextSize; | |
| CHAR16 Buffer[20]; | |
| BDS_COMMON_OPTION *BootOption; | |
| LIST_ENTRY TempList; | |
| UINTN ExitDataSize; | |
| CHAR16 *ExitData; | |
| // | |
| // Init the boot option name buffer and temp link list | |
| // | |
| InitializeListHead (&TempList); | |
| ZeroMem (Buffer, sizeof (Buffer)); | |
| BootNext = BdsLibGetVariableAndSize ( | |
| L"BootNext", | |
| &gEfiGlobalVariableGuid, | |
| &BootNextSize | |
| ); | |
| // | |
| // Clear the boot next variable first | |
| // | |
| if (BootNext != NULL) { | |
| gRT->SetVariable ( | |
| L"BootNext", | |
| &gEfiGlobalVariableGuid, | |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE, | |
| 0, | |
| BootNext | |
| ); | |
| // | |
| // Start to build the boot option and try to boot | |
| // | |
| UnicodeSPrint (Buffer, sizeof (Buffer), L"Boot%04x", *BootNext); | |
| BootOption = BdsLibVariableToOption (&TempList, Buffer); | |
| ASSERT (BootOption != NULL); | |
| BdsLibConnectDevicePath (BootOption->DevicePath); | |
| BdsLibBootViaBootOption (BootOption, BootOption->DevicePath, &ExitDataSize, &ExitData); | |
| } | |
| } | |
| /** | |
| Return the bootable media handle. | |
| First, check the device is connected | |
| Second, check whether the device path point to a device which support SimpleFileSystemProtocol, | |
| Third, detect the the default boot file in the Media, and return the removable Media handle. | |
| @param DevicePath Device Path to a bootable device | |
| @return The bootable media handle. If the media on the DevicePath is not bootable, NULL will return. | |
| **/ | |
| EFI_HANDLE | |
| EFIAPI | |
| BdsLibGetBootableHandle ( | |
| IN EFI_DEVICE_PATH_PROTOCOL *DevicePath | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_DEVICE_PATH_PROTOCOL *UpdatedDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *DupDevicePath; | |
| EFI_HANDLE Handle; | |
| EFI_BLOCK_IO_PROTOCOL *BlockIo; | |
| VOID *Buffer; | |
| EFI_DEVICE_PATH_PROTOCOL *TempDevicePath; | |
| UINTN Size; | |
| UINTN TempSize; | |
| EFI_HANDLE ReturnHandle; | |
| EFI_HANDLE *SimpleFileSystemHandles; | |
| UINTN NumberSimpleFileSystemHandles; | |
| UINTN Index; | |
| EFI_IMAGE_DOS_HEADER DosHeader; | |
| EFI_IMAGE_OPTIONAL_HEADER_UNION HdrData; | |
| EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr; | |
| UpdatedDevicePath = DevicePath; | |
| // | |
| // Check whether the device is connected | |
| // | |
| Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &UpdatedDevicePath, &Handle); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Skip the case that the boot option point to a simple file protocol which does not consume block Io protocol, | |
| // | |
| Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid, &UpdatedDevicePath, &Handle); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Fail to find the proper BlockIo and simple file protocol, maybe because device not present, we need to connect it firstly | |
| // | |
| UpdatedDevicePath = DevicePath; | |
| Status = gBS->LocateDevicePath (&gEfiDevicePathProtocolGuid, &UpdatedDevicePath, &Handle); | |
| gBS->ConnectController (Handle, NULL, NULL, TRUE); | |
| } | |
| } else { | |
| // | |
| // Get BlockIo protocol and check removable attribute | |
| // | |
| Status = gBS->HandleProtocol (Handle, &gEfiBlockIoProtocolGuid, (VOID **)&BlockIo); | |
| // | |
| // Issue a dummy read to the device to check for media change. | |
| // When the removable media is changed, any Block IO read/write will | |
| // cause the BlockIo protocol be reinstalled and EFI_MEDIA_CHANGED is | |
| // returned. After the Block IO protocol is reinstalled, subsequent | |
| // Block IO read/write will success. | |
| // | |
| Buffer = AllocatePool (BlockIo->Media->BlockSize); | |
| if (Buffer != NULL) { | |
| BlockIo->ReadBlocks ( | |
| BlockIo, | |
| BlockIo->Media->MediaId, | |
| 0, | |
| BlockIo->Media->BlockSize, | |
| Buffer | |
| ); | |
| FreePool(Buffer); | |
| } | |
| } | |
| // | |
| // Detect the the default boot file from removable Media | |
| // | |
| // | |
| // If fail to get bootable handle specified by a USB boot option, the BDS should try to find other bootable device in the same USB bus | |
| // Try to locate the USB node device path first, if fail then use its previous PCI node to search | |
| // | |
| DupDevicePath = DuplicateDevicePath (DevicePath); | |
| ASSERT (DupDevicePath != NULL); | |
| UpdatedDevicePath = DupDevicePath; | |
| Status = gBS->LocateDevicePath (&gEfiDevicePathProtocolGuid, &UpdatedDevicePath, &Handle); | |
| // | |
| // if the resulting device path point to a usb node, and the usb node is a dummy node, should only let device path only point to the previous Pci node | |
| // Acpi()/Pci()/Usb() --> Acpi()/Pci() | |
| // | |
| if ((DevicePathType (UpdatedDevicePath) == MESSAGING_DEVICE_PATH) && | |
| (DevicePathSubType (UpdatedDevicePath) == MSG_USB_DP)) { | |
| // | |
| // Remove the usb node, let the device path only point to PCI node | |
| // | |
| SetDevicePathEndNode (UpdatedDevicePath); | |
| UpdatedDevicePath = DupDevicePath; | |
| } else { | |
| UpdatedDevicePath = DevicePath; | |
| } | |
| // | |
| // Get the device path size of boot option | |
| // | |
| Size = GetDevicePathSize(UpdatedDevicePath) - sizeof (EFI_DEVICE_PATH_PROTOCOL); // minus the end node | |
| ReturnHandle = NULL; | |
| gBS->LocateHandleBuffer ( | |
| ByProtocol, | |
| &gEfiSimpleFileSystemProtocolGuid, | |
| NULL, | |
| &NumberSimpleFileSystemHandles, | |
| &SimpleFileSystemHandles | |
| ); | |
| for (Index = 0; Index < NumberSimpleFileSystemHandles; Index++) { | |
| // | |
| // Get the device path size of SimpleFileSystem handle | |
| // | |
| TempDevicePath = DevicePathFromHandle (SimpleFileSystemHandles[Index]); | |
| TempSize = GetDevicePathSize (TempDevicePath)- sizeof (EFI_DEVICE_PATH_PROTOCOL); // minus the end node | |
| // | |
| // Check whether the device path of boot option is part of the SimpleFileSystem handle's device path | |
| // | |
| if (Size <= TempSize && CompareMem (TempDevicePath, UpdatedDevicePath, Size)==0) { | |
| // | |
| // Load the default boot file \EFI\BOOT\boot{machinename}.EFI from removable Media | |
| // machinename is ia32, ia64, x64, ... | |
| // | |
| Hdr.Union = &HdrData; | |
| Status = BdsLibGetImageHeader ( | |
| SimpleFileSystemHandles[Index], | |
| EFI_REMOVABLE_MEDIA_FILE_NAME, | |
| &DosHeader, | |
| Hdr | |
| ); | |
| if (!EFI_ERROR (Status) && | |
| EFI_IMAGE_MACHINE_TYPE_SUPPORTED (Hdr.Pe32->FileHeader.Machine) && | |
| Hdr.Pe32->OptionalHeader.Subsystem == EFI_IMAGE_SUBSYSTEM_EFI_APPLICATION) { | |
| ReturnHandle = SimpleFileSystemHandles[Index]; | |
| break; | |
| } | |
| } | |
| } | |
| FreePool(DupDevicePath); | |
| if (SimpleFileSystemHandles != NULL) { | |
| FreePool(SimpleFileSystemHandles); | |
| } | |
| return ReturnHandle; | |
| } | |
| /** | |
| Check to see if the network cable is plugged in. If the DevicePath is not | |
| connected it will be connected. | |
| @param DevicePath Device Path to check | |
| @retval TRUE DevicePath points to an Network that is connected | |
| @retval FALSE DevicePath does not point to a bootable network | |
| **/ | |
| BOOLEAN | |
| BdsLibNetworkBootWithMediaPresent ( | |
| IN EFI_DEVICE_PATH_PROTOCOL *DevicePath | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_DEVICE_PATH_PROTOCOL *UpdatedDevicePath; | |
| EFI_HANDLE Handle; | |
| EFI_SIMPLE_NETWORK_PROTOCOL *Snp; | |
| BOOLEAN MediaPresent; | |
| MediaPresent = FALSE; | |
| UpdatedDevicePath = DevicePath; | |
| Status = gBS->LocateDevicePath (&gEfiSimpleNetworkProtocolGuid, &UpdatedDevicePath, &Handle); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Device not present so see if we need to connect it | |
| // | |
| Status = BdsLibConnectDevicePath (DevicePath); | |
| if (!EFI_ERROR (Status)) { | |
| // | |
| // This one should work after we did the connect | |
| // | |
| Status = gBS->LocateDevicePath (&gEfiSimpleNetworkProtocolGuid, &UpdatedDevicePath, &Handle); | |
| } | |
| } | |
| if (!EFI_ERROR (Status)) { | |
| Status = gBS->HandleProtocol (Handle, &gEfiSimpleNetworkProtocolGuid, (VOID **)&Snp); | |
| if (!EFI_ERROR (Status)) { | |
| if (Snp->Mode->MediaPresentSupported) { | |
| if (Snp->Mode->State == EfiSimpleNetworkInitialized) { | |
| // | |
| // In case some one else is using the SNP check to see if it's connected | |
| // | |
| MediaPresent = Snp->Mode->MediaPresent; | |
| } else { | |
| // | |
| // No one is using SNP so we need to Start and Initialize so | |
| // MediaPresent will be valid. | |
| // | |
| Status = Snp->Start (Snp); | |
| if (!EFI_ERROR (Status)) { | |
| Status = Snp->Initialize (Snp, 0, 0); | |
| if (!EFI_ERROR (Status)) { | |
| MediaPresent = Snp->Mode->MediaPresent; | |
| Snp->Shutdown (Snp); | |
| } | |
| Snp->Stop (Snp); | |
| } | |
| } | |
| } else { | |
| MediaPresent = TRUE; | |
| } | |
| } | |
| } | |
| return MediaPresent; | |
| } | |
| /** | |
| For a bootable Device path, return its boot type. | |
| @param DevicePath The bootable device Path to check | |
| @retval BDS_EFI_MEDIA_HD_BOOT If given device path contains MEDIA_DEVICE_PATH type device path node | |
| which subtype is MEDIA_HARDDRIVE_DP | |
| @retval BDS_EFI_MEDIA_CDROM_BOOT If given device path contains MEDIA_DEVICE_PATH type device path node | |
| which subtype is MEDIA_CDROM_DP | |
| @retval BDS_EFI_ACPI_FLOPPY_BOOT If given device path contains ACPI_DEVICE_PATH type device path node | |
| which HID is floppy device. | |
| @retval BDS_EFI_MESSAGE_ATAPI_BOOT If given device path contains MESSAGING_DEVICE_PATH type device path node | |
| and its last device path node's subtype is MSG_ATAPI_DP. | |
| @retval BDS_EFI_MESSAGE_SCSI_BOOT If given device path contains MESSAGING_DEVICE_PATH type device path node | |
| and its last device path node's subtype is MSG_SCSI_DP. | |
| @retval BDS_EFI_MESSAGE_USB_DEVICE_BOOT If given device path contains MESSAGING_DEVICE_PATH type device path node | |
| and its last device path node's subtype is MSG_USB_DP. | |
| @retval BDS_EFI_MESSAGE_MISC_BOOT If the device path not contains any media device path node, and | |
| its last device path node point to a message device path node. | |
| @retval BDS_LEGACY_BBS_BOOT If given device path contains BBS_DEVICE_PATH type device path node. | |
| @retval BDS_EFI_UNSUPPORT An EFI Removable BlockIO device path not point to a media and message device, | |
| **/ | |
| UINT32 | |
| EFIAPI | |
| BdsGetBootTypeFromDevicePath ( | |
| IN EFI_DEVICE_PATH_PROTOCOL *DevicePath | |
| ) | |
| { | |
| ACPI_HID_DEVICE_PATH *Acpi; | |
| EFI_DEVICE_PATH_PROTOCOL *TempDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *LastDeviceNode; | |
| if (NULL == DevicePath) { | |
| return BDS_EFI_UNSUPPORT; | |
| } | |
| TempDevicePath = DevicePath; | |
| while (!IsDevicePathEndType (TempDevicePath)) { | |
| switch (DevicePathType (TempDevicePath)) { | |
| case BBS_DEVICE_PATH: | |
| return BDS_LEGACY_BBS_BOOT; | |
| case MEDIA_DEVICE_PATH: | |
| if (DevicePathSubType (TempDevicePath) == MEDIA_HARDDRIVE_DP) { | |
| return BDS_EFI_MEDIA_HD_BOOT; | |
| } else if (DevicePathSubType (TempDevicePath) == MEDIA_CDROM_DP) { | |
| return BDS_EFI_MEDIA_CDROM_BOOT; | |
| } | |
| break; | |
| case ACPI_DEVICE_PATH: | |
| Acpi = (ACPI_HID_DEVICE_PATH *) TempDevicePath; | |
| if (EISA_ID_TO_NUM (Acpi->HID) == 0x0604) { | |
| return BDS_EFI_ACPI_FLOPPY_BOOT; | |
| } | |
| break; | |
| case MESSAGING_DEVICE_PATH: | |
| // | |
| // Get the last device path node | |
| // | |
| LastDeviceNode = NextDevicePathNode (TempDevicePath); | |
| if (DevicePathSubType(LastDeviceNode) == MSG_DEVICE_LOGICAL_UNIT_DP) { | |
| // | |
| // if the next node type is Device Logical Unit, which specify the Logical Unit Number (LUN), | |
| // skit it | |
| // | |
| LastDeviceNode = NextDevicePathNode (LastDeviceNode); | |
| } | |
| // | |
| // if the device path not only point to driver device, it is not a messaging device path, | |
| // | |
| if (!IsDevicePathEndType (LastDeviceNode)) { | |
| break; | |
| } | |
| if (DevicePathSubType(TempDevicePath) == MSG_ATAPI_DP) { | |
| return BDS_EFI_MESSAGE_ATAPI_BOOT; | |
| } else if (DevicePathSubType(TempDevicePath) == MSG_USB_DP) { | |
| return BDS_EFI_MESSAGE_USB_DEVICE_BOOT; | |
| } else if (DevicePathSubType(TempDevicePath) == MSG_SCSI_DP) { | |
| return BDS_EFI_MESSAGE_SCSI_BOOT; | |
| } else if (DevicePathSubType(TempDevicePath) == MSG_SATA_DP) { | |
| return BDS_EFI_MESSAGE_SATA_BOOT; | |
| } else if (DevicePathSubType(TempDevicePath) == MSG_MAC_ADDR_DP) { | |
| return BDS_EFI_MESSAGE_MAC_BOOT; | |
| } | |
| return BDS_EFI_MESSAGE_MISC_BOOT; | |
| default: | |
| break; | |
| } | |
| TempDevicePath = NextDevicePathNode (TempDevicePath); | |
| } | |
| return BDS_EFI_UNSUPPORT; | |
| } | |
| /** | |
| Check whether the Device path in a boot option point to a valid bootable device, | |
| And if CheckMedia is true, check the device is ready to boot now. | |
| @param DevPath the Device path in a boot option | |
| @param CheckMedia if true, check the device is ready to boot now. | |
| @retval TRUE the Device path is valid | |
| @retval FALSE the Device path is invalid . | |
| **/ | |
| BOOLEAN | |
| EFIAPI | |
| BdsLibIsValidEFIBootOptDevicePath ( | |
| IN EFI_DEVICE_PATH_PROTOCOL *DevPath, | |
| IN BOOLEAN CheckMedia | |
| ) | |
| { | |
| return BdsLibIsValidEFIBootOptDevicePathExt (DevPath, CheckMedia, NULL); | |
| } | |
| /** | |
| Check whether the Device path in a boot option point to a valid bootable device, | |
| And if CheckMedia is true, check the device is ready to boot now. | |
| If Description is not NULL and the device path point to a fixed BlockIo | |
| device, check the description whether conflict with other auto-created | |
| boot options. | |
| @param DevPath the Device path in a boot option | |
| @param CheckMedia if true, check the device is ready to boot now. | |
| @param Description the description in a boot option | |
| @retval TRUE the Device path is valid | |
| @retval FALSE the Device path is invalid . | |
| **/ | |
| BOOLEAN | |
| EFIAPI | |
| BdsLibIsValidEFIBootOptDevicePathExt ( | |
| IN EFI_DEVICE_PATH_PROTOCOL *DevPath, | |
| IN BOOLEAN CheckMedia, | |
| IN CHAR16 *Description | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_HANDLE Handle; | |
| EFI_DEVICE_PATH_PROTOCOL *TempDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *LastDeviceNode; | |
| EFI_BLOCK_IO_PROTOCOL *BlockIo; | |
| TempDevicePath = DevPath; | |
| LastDeviceNode = DevPath; | |
| // | |
| // Check if it's a valid boot option for network boot device | |
| // Only check if there is SimpleNetworkProtocol installed. If yes, that means | |
| // there is the network card there. | |
| // | |
| Status = gBS->LocateDevicePath ( | |
| &gEfiSimpleNetworkProtocolGuid, | |
| &TempDevicePath, | |
| &Handle | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Device not present so see if we need to connect it | |
| // | |
| TempDevicePath = DevPath; | |
| BdsLibConnectDevicePath (TempDevicePath); | |
| Status = gBS->LocateDevicePath ( | |
| &gEfiSimpleNetworkProtocolGuid, | |
| &TempDevicePath, | |
| &Handle | |
| ); | |
| } | |
| if (!EFI_ERROR (Status)) { | |
| if (CheckMedia) { | |
| // | |
| // Test if it is ready to boot now | |
| // | |
| if (BdsLibNetworkBootWithMediaPresent(DevPath)) { | |
| return TRUE; | |
| } | |
| } else { | |
| return TRUE; | |
| } | |
| } | |
| // | |
| // If the boot option point to a file, it is a valid EFI boot option, | |
| // and assume it is ready to boot now | |
| // | |
| while (!IsDevicePathEnd (TempDevicePath)) { | |
| LastDeviceNode = TempDevicePath; | |
| TempDevicePath = NextDevicePathNode (TempDevicePath); | |
| } | |
| if ((DevicePathType (LastDeviceNode) == MEDIA_DEVICE_PATH) && | |
| (DevicePathSubType (LastDeviceNode) == MEDIA_FILEPATH_DP)) { | |
| return TRUE; | |
| } | |
| // | |
| // Check if it's a valid boot option for internal Shell | |
| // | |
| if (EfiGetNameGuidFromFwVolDevicePathNode ((MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) LastDeviceNode) != NULL) { | |
| // | |
| // If the boot option point to Internal FV shell, make sure it is valid | |
| // | |
| TempDevicePath = DevPath; | |
| Status = BdsLibUpdateFvFileDevicePath (&TempDevicePath, PcdGetPtr(PcdShellFile)); | |
| if (Status == EFI_ALREADY_STARTED) { | |
| return TRUE; | |
| } else { | |
| if (Status == EFI_SUCCESS) { | |
| FreePool (TempDevicePath); | |
| } | |
| return FALSE; | |
| } | |
| } | |
| // | |
| // If the boot option point to a blockIO device: | |
| // if it is a removable blockIo device, it is valid. | |
| // if it is a fixed blockIo device, check its description confliction. | |
| // | |
| TempDevicePath = DevPath; | |
| Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &TempDevicePath, &Handle); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Device not present so see if we need to connect it | |
| // | |
| Status = BdsLibConnectDevicePath (DevPath); | |
| if (!EFI_ERROR (Status)) { | |
| // | |
| // Try again to get the Block Io protocol after we did the connect | |
| // | |
| TempDevicePath = DevPath; | |
| Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &TempDevicePath, &Handle); | |
| } | |
| } | |
| if (!EFI_ERROR (Status)) { | |
| Status = gBS->HandleProtocol (Handle, &gEfiBlockIoProtocolGuid, (VOID **)&BlockIo); | |
| if (!EFI_ERROR (Status)) { | |
| if (CheckMedia) { | |
| // | |
| // Test if it is ready to boot now | |
| // | |
| if (BdsLibGetBootableHandle (DevPath) != NULL) { | |
| return TRUE; | |
| } | |
| } else { | |
| return TRUE; | |
| } | |
| } | |
| } else { | |
| // | |
| // if the boot option point to a simple file protocol which does not consume block Io protocol, it is also a valid EFI boot option, | |
| // | |
| Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid, &TempDevicePath, &Handle); | |
| if (!EFI_ERROR (Status)) { | |
| if (CheckMedia) { | |
| // | |
| // Test if it is ready to boot now | |
| // | |
| if (BdsLibGetBootableHandle (DevPath) != NULL) { | |
| return TRUE; | |
| } | |
| } else { | |
| return TRUE; | |
| } | |
| } | |
| } | |
| return FALSE; | |
| } | |
| /** | |
| According to a file guild, check a Fv file device path is valid. If it is invalid, | |
| try to return the valid device path. | |
| FV address maybe changes for memory layout adjust from time to time, use this function | |
| could promise the Fv file device path is right. | |
| @param DevicePath on input, the Fv file device path need to check on | |
| output, the updated valid Fv file device path | |
| @param FileGuid the Fv file guild | |
| @retval EFI_INVALID_PARAMETER the input DevicePath or FileGuid is invalid | |
| parameter | |
| @retval EFI_UNSUPPORTED the input DevicePath does not contain Fv file | |
| guild at all | |
| @retval EFI_ALREADY_STARTED the input DevicePath has pointed to Fv file, it is | |
| valid | |
| @retval EFI_SUCCESS has successfully updated the invalid DevicePath, | |
| and return the updated device path in DevicePath | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| BdsLibUpdateFvFileDevicePath ( | |
| IN OUT EFI_DEVICE_PATH_PROTOCOL ** DevicePath, | |
| IN EFI_GUID *FileGuid | |
| ) | |
| { | |
| EFI_DEVICE_PATH_PROTOCOL *TempDevicePath; | |
| EFI_DEVICE_PATH_PROTOCOL *LastDeviceNode; | |
| EFI_STATUS Status; | |
| EFI_GUID *GuidPoint; | |
| UINTN Index; | |
| UINTN FvHandleCount; | |
| EFI_HANDLE *FvHandleBuffer; | |
| EFI_FV_FILETYPE Type; | |
| UINTN Size; | |
| EFI_FV_FILE_ATTRIBUTES Attributes; | |
| UINT32 AuthenticationStatus; | |
| BOOLEAN FindFvFile; | |
| EFI_LOADED_IMAGE_PROTOCOL *LoadedImage; | |
| EFI_FIRMWARE_VOLUME2_PROTOCOL *Fv; | |
| MEDIA_FW_VOL_FILEPATH_DEVICE_PATH FvFileNode; | |
| EFI_HANDLE FoundFvHandle; | |
| EFI_DEVICE_PATH_PROTOCOL *NewDevicePath; | |
| if ((DevicePath == NULL) || (*DevicePath == NULL)) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| if (FileGuid == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| // | |
| // Check whether the device path point to the default the input Fv file | |
| // | |
| TempDevicePath = *DevicePath; | |
| LastDeviceNode = TempDevicePath; | |
| while (!IsDevicePathEnd (TempDevicePath)) { | |
| LastDeviceNode = TempDevicePath; | |
| TempDevicePath = NextDevicePathNode (TempDevicePath); | |
| } | |
| GuidPoint = EfiGetNameGuidFromFwVolDevicePathNode ( | |
| (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) LastDeviceNode | |
| ); | |
| if (GuidPoint == NULL) { | |
| // | |
| // if this option does not points to a Fv file, just return EFI_UNSUPPORTED | |
| // | |
| return EFI_UNSUPPORTED; | |
| } | |
| if (!CompareGuid (GuidPoint, FileGuid)) { | |
| // | |
| // If the Fv file is not the input file guid, just return EFI_UNSUPPORTED | |
| // | |
| return EFI_UNSUPPORTED; | |
| } | |
| // | |
| // Check whether the input Fv file device path is valid | |
| // | |
| TempDevicePath = *DevicePath; | |
| FoundFvHandle = NULL; | |
| Status = gBS->LocateDevicePath ( | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| &TempDevicePath, | |
| &FoundFvHandle | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| Status = gBS->HandleProtocol ( | |
| FoundFvHandle, | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| (VOID **) &Fv | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| // | |
| // Set FV ReadFile Buffer as NULL, only need to check whether input Fv file exist there | |
| // | |
| Status = Fv->ReadFile ( | |
| Fv, | |
| FileGuid, | |
| NULL, | |
| &Size, | |
| &Type, | |
| &Attributes, | |
| &AuthenticationStatus | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| return EFI_ALREADY_STARTED; | |
| } | |
| } | |
| } | |
| // | |
| // Look for the input wanted FV file in current FV | |
| // First, try to look for in Bds own FV. Bds and input wanted FV file usually are in the same FV | |
| // | |
| FindFvFile = FALSE; | |
| FoundFvHandle = NULL; | |
| Status = gBS->HandleProtocol ( | |
| mBdsImageHandle, | |
| &gEfiLoadedImageProtocolGuid, | |
| (VOID **) &LoadedImage | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| Status = gBS->HandleProtocol ( | |
| LoadedImage->DeviceHandle, | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| (VOID **) &Fv | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| Status = Fv->ReadFile ( | |
| Fv, | |
| FileGuid, | |
| NULL, | |
| &Size, | |
| &Type, | |
| &Attributes, | |
| &AuthenticationStatus | |
| ); | |
| if (!EFI_ERROR (Status)) { | |
| FindFvFile = TRUE; | |
| FoundFvHandle = LoadedImage->DeviceHandle; | |
| } | |
| } | |
| } | |
| // | |
| // Second, if fail to find, try to enumerate all FV | |
| // | |
| if (!FindFvFile) { | |
| FvHandleBuffer = NULL; | |
| gBS->LocateHandleBuffer ( | |
| ByProtocol, | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| NULL, | |
| &FvHandleCount, | |
| &FvHandleBuffer | |
| ); | |
| for (Index = 0; Index < FvHandleCount; Index++) { | |
| gBS->HandleProtocol ( | |
| FvHandleBuffer[Index], | |
| &gEfiFirmwareVolume2ProtocolGuid, | |
| (VOID **) &Fv | |
| ); | |
| Status = Fv->ReadFile ( | |
| Fv, | |
| FileGuid, | |
| NULL, | |
| &Size, | |
| &Type, | |
| &Attributes, | |
| &AuthenticationStatus | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // Skip if input Fv file not in the FV | |
| // | |
| continue; | |
| } | |
| FindFvFile = TRUE; | |
| FoundFvHandle = FvHandleBuffer[Index]; | |
| break; | |
| } | |
| if (FvHandleBuffer != NULL) { | |
| FreePool (FvHandleBuffer); | |
| } | |
| } | |
| if (FindFvFile) { | |
| // | |
| // Build the shell device path | |
| // | |
| NewDevicePath = DevicePathFromHandle (FoundFvHandle); | |
| EfiInitializeFwVolDevicepathNode (&FvFileNode, FileGuid); | |
| NewDevicePath = AppendDevicePathNode (NewDevicePath, (EFI_DEVICE_PATH_PROTOCOL *) &FvFileNode); | |
| *DevicePath = NewDevicePath; | |
| return EFI_SUCCESS; | |
| } | |
| return EFI_NOT_FOUND; | |
| } |