[mirror_edk2.git] / MdeModulePkg / Library / UefiBootManagerLib / BmBoot.c

/** @file\r
  Library functions which relates with booting.\r
\r
Copyright (c) 2011 - 2016, Intel Corporation. All rights reserved.<BR>\r
(C) Copyright 2015 Hewlett Packard Enterprise Development LP<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution.  The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "InternalBm.h"\r
\r
EFI_BOOT_MANAGER_REFRESH_LEGACY_BOOT_OPTION  mBmRefreshLegacyBootOption = NULL;\r
EFI_BOOT_MANAGER_LEGACY_BOOT                 mBmLegacyBoot              = NULL;\r
\r
///\r
/// This GUID is used for an EFI Variable that stores the front device pathes\r
/// for a partial device path that starts with the HD node.\r
///\r
EFI_GUID mBmHardDriveBootVariableGuid = { 0xfab7e9e1, 0x39dd, 0x4f2b, { 0x84, 0x08, 0xe2, 0x0e, 0x90, 0x6c, 0xb6, 0xde } };\r
EFI_GUID mBmAutoCreateBootOptionGuid  = { 0x8108ac4e, 0x9f11, 0x4d59, { 0x85, 0x0e, 0xe2, 0x1a, 0x52, 0x2c, 0x59, 0xb2 } };\r
\r
/**\r
  The function registers the legacy boot support capabilities.\r
\r
  @param RefreshLegacyBootOption The function pointer to create all the legacy boot options.\r
  @param LegacyBoot              The function pointer to boot the legacy boot option.\r
**/\r
VOID\r
EFIAPI\r
EfiBootManagerRegisterLegacyBootSupport (\r
  EFI_BOOT_MANAGER_REFRESH_LEGACY_BOOT_OPTION   RefreshLegacyBootOption,\r
  EFI_BOOT_MANAGER_LEGACY_BOOT                  LegacyBoot\r
  )\r
{\r
  mBmRefreshLegacyBootOption = RefreshLegacyBootOption;\r
  mBmLegacyBoot              = LegacyBoot;\r
}\r
\r
/**\r
  Return TRUE when the boot option is auto-created instead of manually added.\r
\r
  @param BootOption Pointer to the boot option to check.\r
\r
  @retval TRUE  The boot option is auto-created.\r
  @retval FALSE The boot option is manually added.\r
**/\r
BOOLEAN\r
BmIsAutoCreateBootOption (\r
  EFI_BOOT_MANAGER_LOAD_OPTION    *BootOption\r
  )\r
{\r
  if ((BootOption->OptionalDataSize == sizeof (EFI_GUID)) &&\r
      CompareGuid ((EFI_GUID *) BootOption->OptionalData, &mBmAutoCreateBootOptionGuid)\r
      ) {\r
    return TRUE;\r
  } else {\r
    return FALSE;\r
  }\r
}\r
\r
/**\r
  Find the boot option in the NV storage and return the option number.\r
\r
  @param OptionToFind  Boot option to be checked.\r
\r
  @return   The option number of the found boot option.\r
\r
**/\r
UINTN\r
BmFindBootOptionInVariable (\r
  IN  EFI_BOOT_MANAGER_LOAD_OPTION             *OptionToFind\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  EFI_BOOT_MANAGER_LOAD_OPTION BootOption;\r
  UINTN                        OptionNumber;\r
  CHAR16                       OptionName[BM_OPTION_NAME_LEN];\r
  EFI_BOOT_MANAGER_LOAD_OPTION *BootOptions;\r
  UINTN                        BootOptionCount;\r
  UINTN                        Index;\r
  \r
  OptionNumber = LoadOptionNumberUnassigned;\r
\r
  //\r
  // Try to match the variable exactly if the option number is assigned\r
  //\r
  if (OptionToFind->OptionNumber != LoadOptionNumberUnassigned) {\r
    UnicodeSPrint (\r
      OptionName, sizeof (OptionName), L"%s%04x",\r
      mBmLoadOptionName[OptionToFind->OptionType], OptionToFind->OptionNumber\r
      );\r
    Status = EfiBootManagerVariableToLoadOption (OptionName, &BootOption);\r
\r
    if (!EFI_ERROR (Status)) {\r
      ASSERT (OptionToFind->OptionNumber == BootOption.OptionNumber);\r
      if ((OptionToFind->Attributes == BootOption.Attributes) &&\r
          (StrCmp (OptionToFind->Description, BootOption.Description) == 0) &&\r
          (CompareMem (OptionToFind->FilePath, BootOption.FilePath, GetDevicePathSize (OptionToFind->FilePath)) == 0) &&\r
          (OptionToFind->OptionalDataSize == BootOption.OptionalDataSize) &&\r
          (CompareMem (OptionToFind->OptionalData, BootOption.OptionalData, OptionToFind->OptionalDataSize) == 0)\r
         ) {\r
        OptionNumber = OptionToFind->OptionNumber;\r
      }\r
      EfiBootManagerFreeLoadOption (&BootOption);\r
    }\r
  }\r
\r
  //\r
  // The option number assigned is either incorrect or unassigned.\r
  //\r
  if (OptionNumber == LoadOptionNumberUnassigned) {\r
    BootOptions = EfiBootManagerGetLoadOptions (&BootOptionCount, LoadOptionTypeBoot);\r
\r
    Index = EfiBootManagerFindLoadOption (OptionToFind, BootOptions, BootOptionCount);\r
    if (Index != -1) {\r
      OptionNumber = BootOptions[Index].OptionNumber;\r
    }\r
\r
    EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
  }\r
\r
  return OptionNumber;\r
}\r
\r
/**\r
  Get the file buffer using a Memory Mapped Device Path.\r
\r
  FV address may change across reboot. This routine promises the FV file device path is right.\r
\r
  @param  FilePath     The Memory Mapped Device Path to get the file buffer.\r
  @param  FullPath     Receive the updated FV Device Path pointint to the file.\r
  @param  FileSize     Receive the file buffer size.\r
\r
  @return  The file buffer.\r
**/\r
VOID *\r
BmGetFileBufferByFvFilePath (\r
  IN EFI_DEVICE_PATH_PROTOCOL      *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL     **FullPath,\r
  OUT UINTN                        *FileSize\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  UINTN                         Index;\r
  EFI_DEVICE_PATH_PROTOCOL      *FvFileNode;\r
  EFI_HANDLE                    FvHandle;\r
  EFI_LOADED_IMAGE_PROTOCOL     *LoadedImage;\r
  UINT32                        AuthenticationStatus;\r
  UINTN                         FvHandleCount;\r
  EFI_HANDLE                    *FvHandles;\r
  EFI_DEVICE_PATH_PROTOCOL      *NewDevicePath;\r
  VOID                          *FileBuffer;\r
\r
  //\r
  // Get the file buffer by using the exactly FilePath.\r
  //\r
  FvFileNode = FilePath;\r
  Status = gBS->LocateDevicePath (&gEfiFirmwareVolume2ProtocolGuid, &FvFileNode, &FvHandle);\r
  if (!EFI_ERROR (Status)) {\r
    FileBuffer = GetFileBufferByFilePath (TRUE, FilePath, FileSize, &AuthenticationStatus);\r
    if (FileBuffer != NULL) {\r
      *FullPath = DuplicateDevicePath (FilePath);\r
    }\r
    return FileBuffer;\r
  }\r
\r
  //\r
  // Only wide match other FVs if it's a memory mapped FV file path.\r
  //\r
  if ((DevicePathType (FilePath) != HARDWARE_DEVICE_PATH) || (DevicePathSubType (FilePath) != HW_MEMMAP_DP)) {\r
    return NULL;\r
  }\r
\r
  FvFileNode = NextDevicePathNode (FilePath);\r
\r
  //\r
  // Firstly find the FV file in current FV\r
  //\r
  gBS->HandleProtocol (\r
         gImageHandle,\r
         &gEfiLoadedImageProtocolGuid,\r
         (VOID **) &LoadedImage\r
         );\r
  NewDevicePath = AppendDevicePathNode (DevicePathFromHandle (LoadedImage->DeviceHandle), FvFileNode);\r
  FileBuffer = BmGetFileBufferByFvFilePath (NewDevicePath, FullPath, FileSize);\r
  FreePool (NewDevicePath);\r
\r
  if (FileBuffer != NULL) {\r
    return FileBuffer;\r
  }\r
\r
  //\r
  // Secondly find the FV file in all other FVs\r
  //\r
  gBS->LocateHandleBuffer (\r
         ByProtocol,\r
         &gEfiFirmwareVolume2ProtocolGuid,\r
         NULL,\r
         &FvHandleCount,\r
         &FvHandles\r
         );\r
  for (Index = 0; (Index < FvHandleCount) && (FileBuffer == NULL); Index++) {\r
    if (FvHandles[Index] == LoadedImage->DeviceHandle) {\r
      //\r
      // Skip current FV\r
      //\r
      continue;\r
    }\r
    NewDevicePath = AppendDevicePathNode (DevicePathFromHandle (FvHandles[Index]), FvFileNode);\r
    FileBuffer = BmGetFileBufferByFvFilePath (NewDevicePath, FullPath, FileSize);\r
    FreePool (NewDevicePath);\r
  }\r
  \r
  if (FvHandles != NULL) {\r
    FreePool (FvHandles);\r
  }\r
  return FileBuffer;\r
}\r
\r
/**\r
  Check if it's a Device Path pointing to FV file.\r
  \r
  The function doesn't garentee the device path points to existing FV file.\r
\r
  @param  DevicePath     Input device path.\r
\r
  @retval TRUE   The device path is a FV File Device Path.\r
  @retval FALSE  The device path is NOT a FV File Device Path.\r
**/\r
BOOLEAN\r
BmIsFvFilePath (\r
  IN EFI_DEVICE_PATH_PROTOCOL    *DevicePath\r
  )\r
{\r
  EFI_STATUS                     Status;\r
  EFI_HANDLE                     Handle;\r
  EFI_DEVICE_PATH_PROTOCOL       *Node;\r
\r
  Node = DevicePath;\r
  Status = gBS->LocateDevicePath (&gEfiFirmwareVolume2ProtocolGuid, &Node, &Handle);\r
  if (!EFI_ERROR (Status)) {\r
    return TRUE;\r
  }\r
\r
  if ((DevicePathType (DevicePath) == HARDWARE_DEVICE_PATH) && (DevicePathSubType (DevicePath) == HW_MEMMAP_DP)) {\r
    DevicePath = NextDevicePathNode (DevicePath);\r
    if ((DevicePathType (DevicePath) == MEDIA_DEVICE_PATH) && (DevicePathSubType (DevicePath) == MEDIA_PIWG_FW_FILE_DP)) {\r
      return IsDevicePathEnd (NextDevicePathNode (DevicePath));\r
    }\r
  }\r
  return FALSE;\r
}\r
\r
/**\r
  Check whether a USB device match the specified USB Class device path. This\r
  function follows "Load Option Processing" behavior in UEFI specification.\r
\r
  @param UsbIo       USB I/O protocol associated with the USB device.\r
  @param UsbClass    The USB Class device path to match.\r
\r
  @retval TRUE       The USB device match the USB Class device path.\r
  @retval FALSE      The USB device does not match the USB Class device path.\r
\r
**/\r
BOOLEAN\r
BmMatchUsbClass (\r
  IN EFI_USB_IO_PROTOCOL        *UsbIo,\r
  IN USB_CLASS_DEVICE_PATH      *UsbClass\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  EFI_USB_DEVICE_DESCRIPTOR     DevDesc;\r
  EFI_USB_INTERFACE_DESCRIPTOR  IfDesc;\r
  UINT8                         DeviceClass;\r
  UINT8                         DeviceSubClass;\r
  UINT8                         DeviceProtocol;\r
\r
  if ((DevicePathType (UsbClass) != MESSAGING_DEVICE_PATH) ||\r
      (DevicePathSubType (UsbClass) != MSG_USB_CLASS_DP)){\r
    return FALSE;\r
  }\r
\r
  //\r
  // Check Vendor Id and Product Id.\r
  //\r
  Status = UsbIo->UsbGetDeviceDescriptor (UsbIo, &DevDesc);\r
  if (EFI_ERROR (Status)) {\r
    return FALSE;\r
  }\r
\r
  if ((UsbClass->VendorId != 0xffff) &&\r
      (UsbClass->VendorId != DevDesc.IdVendor)) {\r
    return FALSE;\r
  }\r
\r
  if ((UsbClass->ProductId != 0xffff) &&\r
      (UsbClass->ProductId != DevDesc.IdProduct)) {\r
    return FALSE;\r
  }\r
\r
  DeviceClass    = DevDesc.DeviceClass;\r
  DeviceSubClass = DevDesc.DeviceSubClass;\r
  DeviceProtocol = DevDesc.DeviceProtocol;\r
  if (DeviceClass == 0) {\r
    //\r
    // If Class in Device Descriptor is set to 0, use the Class, SubClass and\r
    // Protocol in Interface Descriptor instead.\r
    //\r
    Status = UsbIo->UsbGetInterfaceDescriptor (UsbIo, &IfDesc);\r
    if (EFI_ERROR (Status)) {\r
      return FALSE;\r
    }\r
\r
    DeviceClass    = IfDesc.InterfaceClass;\r
    DeviceSubClass = IfDesc.InterfaceSubClass;\r
    DeviceProtocol = IfDesc.InterfaceProtocol;\r
  }\r
\r
  //\r
  // Check Class, SubClass and Protocol.\r
  //\r
  if ((UsbClass->DeviceClass != 0xff) &&\r
      (UsbClass->DeviceClass != DeviceClass)) {\r
    return FALSE;\r
  }\r
\r
  if ((UsbClass->DeviceSubClass != 0xff) &&\r
      (UsbClass->DeviceSubClass != DeviceSubClass)) {\r
    return FALSE;\r
  }\r
\r
  if ((UsbClass->DeviceProtocol != 0xff) &&\r
      (UsbClass->DeviceProtocol != DeviceProtocol)) {\r
    return FALSE;\r
  }\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Check whether a USB device match the specified USB WWID device path. This\r
  function follows "Load Option Processing" behavior in UEFI specification.\r
\r
  @param UsbIo       USB I/O protocol associated with the USB device.\r
  @param UsbWwid     The USB WWID device path to match.\r
\r
  @retval TRUE       The USB device match the USB WWID device path.\r
  @retval FALSE      The USB device does not match the USB WWID device path.\r
\r
**/\r
BOOLEAN\r
BmMatchUsbWwid (\r
  IN EFI_USB_IO_PROTOCOL        *UsbIo,\r
  IN USB_WWID_DEVICE_PATH       *UsbWwid\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  EFI_USB_DEVICE_DESCRIPTOR    DevDesc;\r
  EFI_USB_INTERFACE_DESCRIPTOR IfDesc;\r
  UINT16                       *LangIdTable;\r
  UINT16                       TableSize;\r
  UINT16                       Index;\r
  CHAR16                       *CompareStr;\r
  UINTN                        CompareLen;\r
  CHAR16                       *SerialNumberStr;\r
  UINTN                        Length;\r
\r
  if ((DevicePathType (UsbWwid) != MESSAGING_DEVICE_PATH) ||\r
      (DevicePathSubType (UsbWwid) != MSG_USB_WWID_DP)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Check Vendor Id and Product Id.\r
  //\r
  Status = UsbIo->UsbGetDeviceDescriptor (UsbIo, &DevDesc);\r
  if (EFI_ERROR (Status)) {\r
    return FALSE;\r
  }\r
  if ((DevDesc.IdVendor != UsbWwid->VendorId) ||\r
      (DevDesc.IdProduct != UsbWwid->ProductId)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Check Interface Number.\r
  //\r
  Status = UsbIo->UsbGetInterfaceDescriptor (UsbIo, &IfDesc);\r
  if (EFI_ERROR (Status)) {\r
    return FALSE;\r
  }\r
  if (IfDesc.InterfaceNumber != UsbWwid->InterfaceNumber) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Check Serial Number.\r
  //\r
  if (DevDesc.StrSerialNumber == 0) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Get all supported languages.\r
  //\r
  TableSize = 0;\r
  LangIdTable = NULL;\r
  Status = UsbIo->UsbGetSupportedLanguages (UsbIo, &LangIdTable, &TableSize);\r
  if (EFI_ERROR (Status) || (TableSize == 0) || (LangIdTable == NULL)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Serial number in USB WWID device path is the last 64-or-less UTF-16 characters.\r
  //\r
  CompareStr = (CHAR16 *) (UINTN) (UsbWwid + 1);\r
  CompareLen = (DevicePathNodeLength (UsbWwid) - sizeof (USB_WWID_DEVICE_PATH)) / sizeof (CHAR16);\r
  if (CompareStr[CompareLen - 1] == L'\0') {\r
    CompareLen--;\r
  }\r
\r
  //\r
  // Compare serial number in each supported language.\r
  //\r
  for (Index = 0; Index < TableSize / sizeof (UINT16); Index++) {\r
    SerialNumberStr = NULL;\r
    Status = UsbIo->UsbGetStringDescriptor (\r
                      UsbIo,\r
                      LangIdTable[Index],\r
                      DevDesc.StrSerialNumber,\r
                      &SerialNumberStr\r
                      );\r
    if (EFI_ERROR (Status) || (SerialNumberStr == NULL)) {\r
      continue;\r
    }\r
\r
    Length = StrLen (SerialNumberStr);\r
    if ((Length >= CompareLen) &&\r
        (CompareMem (SerialNumberStr + Length - CompareLen, CompareStr, CompareLen * sizeof (CHAR16)) == 0)) {\r
      FreePool (SerialNumberStr);\r
      return TRUE;\r
    }\r
\r
    FreePool (SerialNumberStr);\r
  }\r
\r
  return FALSE;\r
}\r
\r
/**\r
  Find a USB device which match the specified short-form device path start with \r
  USB Class or USB WWID device path. If ParentDevicePath is NULL, this function\r
  will search in all USB devices of the platform. If ParentDevicePath is not NULL,\r
  this function will only search in its child devices.\r
\r
  @param DevicePath           The device path that contains USB Class or USB WWID device path.\r
  @param ParentDevicePathSize The length of the device path before the USB Class or \r
                              USB WWID device path.\r
  @param UsbIoHandleCount     A pointer to the count of the returned USB IO handles.\r
\r
  @retval NULL       The matched USB IO handles cannot be found.\r
  @retval other      The matched USB IO handles.\r
\r
**/\r
EFI_HANDLE *\r
BmFindUsbDevice (\r
  IN  EFI_DEVICE_PATH_PROTOCOL  *DevicePath,\r
  IN  UINTN                     ParentDevicePathSize,\r
  OUT UINTN                     *UsbIoHandleCount\r
  )\r
{\r
  EFI_STATUS                Status;\r
  EFI_HANDLE                *UsbIoHandles;\r
  EFI_DEVICE_PATH_PROTOCOL  *UsbIoDevicePath;\r
  EFI_USB_IO_PROTOCOL       *UsbIo;\r
  UINTN                     Index;\r
  BOOLEAN                   Matched;\r
\r
  ASSERT (UsbIoHandleCount != NULL);  \r
\r
  //\r
  // Get all UsbIo Handles.\r
  //\r
  Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  &gEfiUsbIoProtocolGuid,\r
                  NULL,\r
                  UsbIoHandleCount,\r
                  &UsbIoHandles\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    *UsbIoHandleCount = 0;\r
    UsbIoHandles      = NULL;\r
  }\r
\r
  for (Index = 0; Index < *UsbIoHandleCount; ) {\r
    //\r
    // Get the Usb IO interface.\r
    //\r
    Status = gBS->HandleProtocol(\r
                    UsbIoHandles[Index],\r
                    &gEfiUsbIoProtocolGuid,\r
                    (VOID **) &UsbIo\r
                    );\r
    UsbIoDevicePath = DevicePathFromHandle (UsbIoHandles[Index]);\r
    Matched         = FALSE;\r
    if (!EFI_ERROR (Status) && (UsbIoDevicePath != NULL)) {\r
\r
      //\r
      // Compare starting part of UsbIoHandle's device path with ParentDevicePath.\r
      //\r
      if (CompareMem (UsbIoDevicePath, DevicePath, ParentDevicePathSize) == 0) {\r
        if (BmMatchUsbClass (UsbIo, (USB_CLASS_DEVICE_PATH *) ((UINTN) DevicePath + ParentDevicePathSize)) ||\r
            BmMatchUsbWwid (UsbIo, (USB_WWID_DEVICE_PATH *) ((UINTN) DevicePath + ParentDevicePathSize))) {\r
          Matched = TRUE;\r
        }\r
      }\r
    }\r
\r
    if (!Matched) {\r
      (*UsbIoHandleCount) --;\r
      CopyMem (&UsbIoHandles[Index], &UsbIoHandles[Index + 1], (*UsbIoHandleCount - Index) * sizeof (EFI_HANDLE));\r
    } else {\r
      Index++;\r
    }\r
  }\r
\r
  return UsbIoHandles;\r
}\r
\r
/**\r
  Expand USB Class or USB WWID device path node to be full device path of a USB\r
  device in platform.\r
\r
  This function support following 4 cases:\r
  1) Boot Option device path starts with a USB Class or USB WWID device path,\r
     and there is no Media FilePath device path in the end.\r
     In this case, it will follow Removable Media Boot Behavior.\r
  2) Boot Option device path starts with a USB Class or USB WWID device path,\r
     and ended with Media FilePath device path.\r
  3) Boot Option device path starts with a full device path to a USB Host Controller,\r
     contains a USB Class or USB WWID device path node, while not ended with Media\r
     FilePath device path. In this case, it will follow Removable Media Boot Behavior.\r
  4) Boot Option device path starts with a full device path to a USB Host Controller,\r
     contains a USB Class or USB WWID device path node, and ended with Media\r
     FilePath device path.\r
\r
  @param FilePath      The device path pointing to a load option.\r
                       It could be a short-form device path.\r
  @param FullPath      Return the full device path of the load option after\r
                       short-form device path expanding.\r
                       Caller is responsible to free it.\r
  @param FileSize      Return the load option size.\r
  @param ShortformNode Pointer to the USB short-form device path node in the FilePath buffer.\r
\r
  @return The load option buffer. Caller is responsible to free the memory.\r
**/\r
VOID *\r
BmExpandUsbDevicePath (\r
  IN  EFI_DEVICE_PATH_PROTOCOL  *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL  **FullPath,\r
  OUT UINTN                     *FileSize,\r
  IN EFI_DEVICE_PATH_PROTOCOL   *ShortformNode\r
  )\r
{\r
  UINTN                             ParentDevicePathSize;\r
  EFI_DEVICE_PATH_PROTOCOL          *RemainingDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL          *FullDevicePath;\r
  EFI_HANDLE                        *Handles;\r
  UINTN                             HandleCount;\r
  UINTN                             Index;\r
  VOID                              *FileBuffer;\r
\r
  ParentDevicePathSize = (UINTN) ShortformNode - (UINTN) FilePath;\r
  RemainingDevicePath = NextDevicePathNode (ShortformNode);\r
  FileBuffer = NULL;\r
  Handles = BmFindUsbDevice (FilePath, ParentDevicePathSize, &HandleCount);\r
\r
  for (Index = 0; (Index < HandleCount) && (FileBuffer == NULL); Index++) {\r
    FullDevicePath = AppendDevicePath (DevicePathFromHandle (Handles[Index]), RemainingDevicePath);\r
    FileBuffer = BmGetLoadOptionBuffer (FullDevicePath, FullPath, FileSize);\r
    FreePool (FullDevicePath);\r
  }\r
\r
  if (Handles != NULL) {\r
    FreePool (Handles);\r
  }\r
\r
  return FileBuffer;\r
}\r
\r
/**\r
  Expand File-path device path node to be full device path in platform.\r
\r
  @param FilePath      The device path pointing to a load option.\r
                       It could be a short-form device path.\r
  @param FullPath      Return the full device path of the load option after\r
                       short-form device path expanding.\r
                       Caller is responsible to free it.\r
  @param FileSize      Return the load option size.\r
\r
  @return The load option buffer. Caller is responsible to free the memory.\r
**/\r
VOID *\r
BmExpandFileDevicePath (\r
  IN  EFI_DEVICE_PATH_PROTOCOL    *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL    **FullPath,\r
  OUT UINTN                       *FileSize\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  UINTN                           Index;\r
  UINTN                           HandleCount;\r
  EFI_HANDLE                      *Handles;\r
  EFI_BLOCK_IO_PROTOCOL           *BlockIo;\r
  UINTN                           MediaType;\r
  EFI_DEVICE_PATH_PROTOCOL        *FullDevicePath;\r
  VOID                            *FileBuffer;\r
  UINT32                          AuthenticationStatus;\r
  \r
  EfiBootManagerConnectAll ();\r
  Status = gBS->LocateHandleBuffer (ByProtocol, &gEfiSimpleFileSystemProtocolGuid, NULL, &HandleCount, &Handles);\r
  if (EFI_ERROR (Status)) {\r
    HandleCount = 0;\r
    Handles = NULL;\r
  }\r
\r
  //\r
  // Enumerate all removable media devices followed by all fixed media devices,\r
  //   followed by media devices which don't layer on block io.\r
  //\r
  for (MediaType = 0; MediaType < 3; MediaType++) {\r
    for (Index = 0; Index < HandleCount; Index++) {\r
      Status = gBS->HandleProtocol (Handles[Index], &gEfiBlockIoProtocolGuid, (VOID *) &BlockIo);\r
      if (EFI_ERROR (Status)) {\r
        BlockIo = NULL;\r
      }\r
      if ((MediaType == 0 && BlockIo != NULL && BlockIo->Media->RemovableMedia) ||\r
          (MediaType == 1 && BlockIo != NULL && !BlockIo->Media->RemovableMedia) ||\r
          (MediaType == 2 && BlockIo == NULL)\r
          ) {\r
        FullDevicePath = AppendDevicePath (DevicePathFromHandle (Handles[Index]), FilePath);\r
        FileBuffer = GetFileBufferByFilePath (TRUE, FullDevicePath, FileSize, &AuthenticationStatus);\r
        if (FileBuffer != NULL) {\r
          *FullPath = FullDevicePath;\r
          FreePool (Handles);\r
          return FileBuffer;\r
        }\r
        FreePool (FullDevicePath);\r
      }\r
    }\r
  }\r
\r
  if (Handles != NULL) {\r
    FreePool (Handles);\r
  }\r
\r
  *FullPath = NULL;\r
  return NULL;\r
}\r
\r
/**\r
  Expand URI device path node to be full device path in platform.\r
\r
  @param FilePath      The device path pointing to a load option.\r
                       It could be a short-form device path.\r
  @param FullPath      Return the full device path of the load option after\r
                       short-form device path expanding.\r
                       Caller is responsible to free it.\r
  @param FileSize      Return the load option size.\r
\r
  @return The load option buffer. Caller is responsible to free the memory.\r
**/\r
VOID *\r
BmExpandUriDevicePath (\r
  IN  EFI_DEVICE_PATH_PROTOCOL    *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL    **FullPath,\r
  OUT UINTN                       *FileSize\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  UINTN                           Index;\r
  UINTN                           HandleCount;\r
  EFI_HANDLE                      *Handles;\r
  EFI_LOAD_FILE_PROTOCOL          *LoadFile;\r
  VOID                            *FileBuffer;\r
\r
  EfiBootManagerConnectAll ();\r
  Status = gBS->LocateHandleBuffer (ByProtocol, &gEfiLoadFileProtocolGuid, NULL, &HandleCount, &Handles);\r
  if (EFI_ERROR (Status)) {\r
    HandleCount = 0;\r
    Handles = NULL;\r
  }\r
\r
  FileBuffer = NULL;\r
\r
  for (Index = 0; Index < HandleCount; Index++) {\r
    Status = gBS->HandleProtocol (Handles[Index], &gEfiLoadFileProtocolGuid, (VOID *) &LoadFile);\r
    ASSERT_EFI_ERROR (Status);\r
\r
    FileBuffer = NULL;\r
    Status = LoadFile->LoadFile (LoadFile, FilePath, TRUE, FileSize, FileBuffer);\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      FileBuffer = AllocatePool (*FileSize);\r
      if (FileBuffer == NULL) {\r
        break;\r
      }\r
      Status = LoadFile->LoadFile (LoadFile, FilePath, TRUE, FileSize, FileBuffer);\r
    }\r
\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // LoadFile() returns a file buffer mapping to a file system.\r
      //\r
      if (Status == EFI_WARN_FILE_SYSTEM) {\r
        return BmGetFileBufferFromLoadFileFileSystem (Handles[Index], FullPath, FileSize);\r
      }\r
\r
      ASSERT (Status == EFI_SUCCESS);\r
      *FullPath = DuplicateDevicePath (DevicePathFromHandle (Handles[Index]));\r
      break;\r
    }\r
\r
    if (FileBuffer != NULL) {\r
      FreePool (FileBuffer);\r
    }\r
  }\r
\r
  if (Handles != NULL) {\r
    FreePool (Handles);\r
  }\r
\r
  return FileBuffer;\r
}\r
\r
/**\r
  Save the partition DevicePath to the CachedDevicePath as the first instance.\r
\r
  @param CachedDevicePath  The device path cache.\r
  @param DevicePath        The partition device path to be cached.\r
**/\r
VOID\r
BmCachePartitionDevicePath (\r
  IN OUT EFI_DEVICE_PATH_PROTOCOL **CachedDevicePath,\r
  IN EFI_DEVICE_PATH_PROTOCOL     *DevicePath\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL        *TempDevicePath;\r
  UINTN                           Count;\r
  \r
  if (BmMatchDevicePaths (*CachedDevicePath, DevicePath)) {\r
    TempDevicePath = *CachedDevicePath;\r
    *CachedDevicePath = BmDelPartMatchInstance (*CachedDevicePath, DevicePath);\r
    FreePool (TempDevicePath);\r
  }\r
\r
  if (*CachedDevicePath == NULL) {\r
    *CachedDevicePath = DuplicateDevicePath (DevicePath);\r
    return;\r
  }\r
\r
  TempDevicePath = *CachedDevicePath;\r
  *CachedDevicePath = AppendDevicePathInstance (DevicePath, *CachedDevicePath);\r
  if (TempDevicePath != NULL) {\r
    FreePool (TempDevicePath);\r
  }\r
\r
  //\r
  // Here limit the device path instance number to 12, which is max number for a system support 3 IDE controller\r
  // If the user try to boot many OS in different HDs or partitions, in theory, the 'HDDP' variable maybe become larger and larger.\r
  //\r
  Count = 0;\r
  TempDevicePath = *CachedDevicePath;\r
  while (!IsDevicePathEnd (TempDevicePath)) {\r
    TempDevicePath = NextDevicePathNode (TempDevicePath);\r
    //\r
    // Parse one instance\r
    //\r
    while (!IsDevicePathEndType (TempDevicePath)) {\r
      TempDevicePath = NextDevicePathNode (TempDevicePath);\r
    }\r
    Count++;\r
    //\r
    // If the CachedDevicePath variable contain too much instance, only remain 12 instances.\r
    //\r
    if (Count == 12) {\r
      SetDevicePathEndNode (TempDevicePath);\r
      break;\r
    }\r
  }\r
}\r
\r
/**\r
  Expand a device path that starts with a hard drive media device path node to be a\r
  full device path that includes the full hardware path to the device. We need\r
  to do this so it can be booted. As an optimization the front match (the part point\r
  to the partition node. E.g. ACPI() /PCI()/ATA()/Partition() ) is saved in a variable\r
  so a connect all is not required on every boot. All successful history device path\r
  which point to partition node (the front part) will be saved.\r
\r
  @param FilePath      The device path pointing to a load option.\r
                       It could be a short-form device path.\r
  @param FullPath      Return the full device path of the load option after\r
                       short-form device path expanding.\r
                       Caller is responsible to free it.\r
  @param FileSize      Return the load option size.\r
\r
  @return The load option buffer. Caller is responsible to free the memory.\r
**/\r
VOID *\r
BmExpandPartitionDevicePath (\r
  IN  EFI_DEVICE_PATH_PROTOCOL  *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL  **FullPath,\r
  OUT UINTN                     *FileSize\r
  )\r
{\r
  EFI_STATUS                Status;\r
  UINTN                     BlockIoHandleCount;\r
  EFI_HANDLE                *BlockIoBuffer;\r
  VOID                      *FileBuffer;\r
  EFI_DEVICE_PATH_PROTOCOL  *BlockIoDevicePath;\r
  UINTN                     Index;\r
  EFI_DEVICE_PATH_PROTOCOL  *CachedDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *TempNewDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *TempDevicePath;\r
  UINTN                     CachedDevicePathSize;\r
  BOOLEAN                   NeedAdjust;\r
  EFI_DEVICE_PATH_PROTOCOL  *Instance;\r
  UINTN                     Size;\r
\r
  FileBuffer = NULL;\r
  //\r
  // Check if there is prestore 'HDDP' variable.\r
  // If exist, search the front path which point to partition node in the variable instants.\r
  // If fail to find or 'HDDP' not exist, reconnect all and search in all system\r
  //\r
  GetVariable2 (L"HDDP", &mBmHardDriveBootVariableGuid, (VOID **) &CachedDevicePath, &CachedDevicePathSize);\r
\r
  //\r
  // Delete the invalid 'HDDP' variable.\r
  //\r
  if ((CachedDevicePath != NULL) && !IsDevicePathValid (CachedDevicePath, CachedDevicePathSize)) {\r
    FreePool (CachedDevicePath);\r
    CachedDevicePath = NULL;\r
    Status = gRT->SetVariable (\r
                    L"HDDP",\r
                    &mBmHardDriveBootVariableGuid,\r
                    0,\r
                    0,\r
                    NULL\r
                    );\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
\r
  if (CachedDevicePath != NULL) {\r
    TempNewDevicePath = CachedDevicePath;\r
    NeedAdjust = FALSE;\r
    do {\r
      //\r
      // Check every instance of the variable\r
      // First, check whether the instance contain the partition node, which is needed for distinguishing  multi\r
      // partial partition boot option. Second, check whether the instance could be connected.\r
      //\r
      Instance  = GetNextDevicePathInstance (&TempNewDevicePath, &Size);\r
      if (BmMatchPartitionDevicePathNode (Instance, (HARDDRIVE_DEVICE_PATH *) FilePath)) {\r
        //\r
        // Connect the device path instance, the device path point to hard drive media device path node\r
        // e.g. ACPI() /PCI()/ATA()/Partition()\r
        //\r
        Status = EfiBootManagerConnectDevicePath (Instance, NULL);\r
        if (!EFI_ERROR (Status)) {\r
          TempDevicePath = AppendDevicePath (Instance, NextDevicePathNode (FilePath));\r
          FileBuffer = BmGetLoadOptionBuffer (TempDevicePath, FullPath, FileSize);\r
          FreePool (TempDevicePath);\r
\r
          if (FileBuffer != NULL) {\r
            //\r
            // Adjust the 'HDDP' instances sequence if the matched one is not first one.\r
            //\r
            if (NeedAdjust) {\r
              BmCachePartitionDevicePath (&CachedDevicePath, Instance);\r
              //\r
              // Save the matching Device Path so we don't need to do a connect all next time\r
              // Failing to save only impacts performance next time expanding the short-form device path\r
              //\r
              Status = gRT->SetVariable (\r
                L"HDDP",\r
                &mBmHardDriveBootVariableGuid,\r
                EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_NON_VOLATILE,\r
                GetDevicePathSize (CachedDevicePath),\r
                CachedDevicePath\r
                );\r
            }\r
\r
            FreePool (Instance);\r
            FreePool (CachedDevicePath);\r
            return FileBuffer;\r
          }\r
        }\r
      }\r
      //\r
      // Come here means the first instance is not matched\r
      //\r
      NeedAdjust = TRUE;\r
      FreePool(Instance);\r
    } while (TempNewDevicePath != NULL);\r
  }\r
\r
  //\r
  // If we get here we fail to find or 'HDDP' not exist, and now we need\r
  // to search all devices in the system for a matched partition\r
  //\r
  EfiBootManagerConnectAll ();\r
  Status = gBS->LocateHandleBuffer (ByProtocol, &gEfiBlockIoProtocolGuid, NULL, &BlockIoHandleCount, &BlockIoBuffer);\r
  if (EFI_ERROR (Status)) {\r
    BlockIoHandleCount = 0;\r
    BlockIoBuffer      = NULL;\r
  }\r
  //\r
  // Loop through all the device handles that support the BLOCK_IO Protocol\r
  //\r
  for (Index = 0; Index < BlockIoHandleCount; Index++) {\r
    BlockIoDevicePath = DevicePathFromHandle (BlockIoBuffer[Index]);\r
    if (BlockIoDevicePath == NULL) {\r
      continue;\r
    }\r
\r
    if (BmMatchPartitionDevicePathNode (BlockIoDevicePath, (HARDDRIVE_DEVICE_PATH *) FilePath)) {\r
      //\r
      // Find the matched partition device path\r
      //\r
      TempDevicePath = AppendDevicePath (BlockIoDevicePath, NextDevicePathNode (FilePath));\r
      FileBuffer = BmGetLoadOptionBuffer (TempDevicePath, FullPath, FileSize);\r
      FreePool (TempDevicePath);\r
\r
      if (FileBuffer != NULL) {\r
        BmCachePartitionDevicePath (&CachedDevicePath, BlockIoDevicePath);\r
\r
        //\r
        // Save the matching Device Path so we don't need to do a connect all next time\r
        // Failing to save only impacts performance next time expanding the short-form device path\r
        //\r
        Status = gRT->SetVariable (\r
                        L"HDDP",\r
                        &mBmHardDriveBootVariableGuid,\r
                        EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_NON_VOLATILE,\r
                        GetDevicePathSize (CachedDevicePath),\r
                        CachedDevicePath\r
                        );\r
\r
        break;\r
      }\r
    }\r
  }\r
\r
  if (CachedDevicePath != NULL) {\r
    FreePool (CachedDevicePath);\r
  }\r
  if (BlockIoBuffer != NULL) {\r
    FreePool (BlockIoBuffer);\r
  }\r
  return FileBuffer;\r
}\r
\r
/**\r
  Expand the media device path which points to a BlockIo or SimpleFileSystem instance\r
  by appending EFI_REMOVABLE_MEDIA_FILE_NAME.\r
\r
  @param DevicePath  The media device path pointing to a BlockIo or SimpleFileSystem instance.\r
  @param FullPath    Return the full device path pointing to the load option.\r
  @param FileSize    Return the size of the load option.\r
\r
  @return  The load option buffer.\r
**/\r
VOID *\r
BmExpandMediaDevicePath (\r
  IN  EFI_DEVICE_PATH_PROTOCOL        *DevicePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL        **FullPath,\r
  OUT UINTN                           *FileSize\r
  )\r
{\r
  EFI_STATUS                          Status;\r
  EFI_HANDLE                          Handle;\r
  EFI_BLOCK_IO_PROTOCOL               *BlockIo;\r
  VOID                                *Buffer;\r
  EFI_DEVICE_PATH_PROTOCOL            *TempDevicePath;\r
  UINTN                               Size;\r
  UINTN                               TempSize;\r
  EFI_HANDLE                          *SimpleFileSystemHandles;\r
  UINTN                               NumberSimpleFileSystemHandles;\r
  UINTN                               Index;\r
  VOID                                *FileBuffer;\r
  UINT32                              AuthenticationStatus;\r
\r
  //\r
  // Check whether the device is connected\r
  //\r
  TempDevicePath = DevicePath;\r
  Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid, &TempDevicePath, &Handle);\r
  if (!EFI_ERROR (Status)) {\r
    ASSERT (IsDevicePathEnd (TempDevicePath));\r
\r
    TempDevicePath = FileDevicePath (Handle, EFI_REMOVABLE_MEDIA_FILE_NAME);\r
    FileBuffer = GetFileBufferByFilePath (TRUE, TempDevicePath, FileSize, &AuthenticationStatus);\r
    if (FileBuffer == NULL) {\r
      FreePool (TempDevicePath);\r
      TempDevicePath = NULL;\r
    }\r
    *FullPath = TempDevicePath;\r
    return FileBuffer;\r
  }\r
\r
  //\r
  // For device boot option only pointing to the removable device handle, \r
  // should make sure all its children handles (its child partion or media handles) are created and connected. \r
  //\r
  gBS->ConnectController (Handle, NULL, NULL, TRUE);\r
\r
  //\r
  // Issue a dummy read to the device to check for media change.\r
  // When the removable media is changed, any Block IO read/write will\r
  // cause the BlockIo protocol be reinstalled and EFI_MEDIA_CHANGED is\r
  // returned. After the Block IO protocol is reinstalled, subsequent\r
  // Block IO read/write will success.\r
  //\r
  Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &TempDevicePath, &Handle);\r
  ASSERT_EFI_ERROR (Status);\r
  Status = gBS->HandleProtocol (Handle, &gEfiBlockIoProtocolGuid, (VOID **) &BlockIo);\r
  ASSERT_EFI_ERROR (Status);\r
  Buffer = AllocatePool (BlockIo->Media->BlockSize);\r
  if (Buffer != NULL) {\r
    BlockIo->ReadBlocks (\r
      BlockIo,\r
      BlockIo->Media->MediaId,\r
      0,\r
      BlockIo->Media->BlockSize,\r
      Buffer\r
      );\r
    FreePool (Buffer);\r
  }\r
\r
  //\r
  // Detect the the default boot file from removable Media\r
  //\r
  FileBuffer = NULL;\r
  *FullPath = NULL;\r
  Size = GetDevicePathSize (DevicePath) - END_DEVICE_PATH_LENGTH;\r
  gBS->LocateHandleBuffer (\r
         ByProtocol,\r
         &gEfiSimpleFileSystemProtocolGuid,\r
         NULL,\r
         &NumberSimpleFileSystemHandles,\r
         &SimpleFileSystemHandles\r
         );\r
  for (Index = 0; Index < NumberSimpleFileSystemHandles; Index++) {\r
    //\r
    // Get the device path size of SimpleFileSystem handle\r
    //\r
    TempDevicePath = DevicePathFromHandle (SimpleFileSystemHandles[Index]);\r
    TempSize = GetDevicePathSize (TempDevicePath) - END_DEVICE_PATH_LENGTH;\r
    //\r
    // Check whether the device path of boot option is part of the SimpleFileSystem handle's device path\r
    //\r
    if ((Size <= TempSize) && (CompareMem (TempDevicePath, DevicePath, Size) == 0)) {\r
      TempDevicePath = FileDevicePath (SimpleFileSystemHandles[Index], EFI_REMOVABLE_MEDIA_FILE_NAME);\r
      FileBuffer = GetFileBufferByFilePath (TRUE, TempDevicePath, FileSize, &AuthenticationStatus);\r
      if (FileBuffer != NULL) {\r
        *FullPath = TempDevicePath;\r
        break;\r
      }\r
      FreePool (TempDevicePath);\r
    }\r
  }\r
\r
  if (SimpleFileSystemHandles != NULL) {\r
    FreePool (SimpleFileSystemHandles);\r
  }\r
\r
  return FileBuffer;\r
}\r
\r
/**\r
  Check whether Left and Right are the same without matching the specific\r
  device path data in IP device path and URI device path node.\r
\r
  @retval TRUE  Left and Right are the same.\r
  @retval FALSE Left and Right are the different.\r
**/\r
BOOLEAN\r
BmMatchHttpBootDevicePath (\r
  IN EFI_DEVICE_PATH_PROTOCOL *Left,\r
  IN EFI_DEVICE_PATH_PROTOCOL *Right\r
  )\r
{\r
  for (;  !IsDevicePathEnd (Left) && !IsDevicePathEnd (Right)\r
       ;  Left = NextDevicePathNode (Left), Right = NextDevicePathNode (Right)\r
       ) {\r
    if (CompareMem (Left, Right, DevicePathNodeLength (Left)) != 0) {\r
      if ((DevicePathType (Left) != MESSAGING_DEVICE_PATH) || (DevicePathType (Right) != MESSAGING_DEVICE_PATH)) {\r
        return FALSE;\r
      }\r
\r
      if (((DevicePathSubType (Left) != MSG_IPv4_DP) || (DevicePathSubType (Right) != MSG_IPv4_DP)) &&\r
          ((DevicePathSubType (Left) != MSG_IPv6_DP) || (DevicePathSubType (Right) != MSG_IPv6_DP)) &&\r
          ((DevicePathSubType (Left) != MSG_URI_DP)  || (DevicePathSubType (Right) != MSG_URI_DP))\r
          ) {\r
        return FALSE;\r
      }\r
    }\r
  }\r
  return (BOOLEAN) (IsDevicePathEnd (Left) && IsDevicePathEnd (Right));\r
}\r
\r
/**\r
  Get the file buffer from the file system produced by Load File instance.\r
\r
  @param LoadFileHandle The handle of LoadFile instance.\r
  @param FullPath       Return the full device path pointing to the load option.\r
  @param FileSize       Return the size of the load option.\r
\r
  @return  The load option buffer.\r
**/\r
VOID *\r
BmGetFileBufferFromLoadFileFileSystem (\r
  IN  EFI_HANDLE                      LoadFileHandle,\r
  OUT EFI_DEVICE_PATH_PROTOCOL        **FullPath,\r
  OUT UINTN                           *FileSize\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  EFI_HANDLE                      Handle;\r
  EFI_HANDLE                      *Handles;\r
  UINTN                           HandleCount;\r
  UINTN                           Index;\r
  EFI_DEVICE_PATH_PROTOCOL        *Node;\r
\r
  Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  &gEfiBlockIoProtocolGuid,\r
                  NULL,\r
                  &HandleCount,\r
                  &Handles\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    Handles = NULL;\r
    HandleCount = 0;\r
  }\r
\r
  Handle = NULL;\r
  for (Index = 0; Index < HandleCount; Index++) {\r
    Node = DevicePathFromHandle (Handles[Index]);\r
    Status = gBS->LocateDevicePath (&gEfiLoadFileProtocolGuid, &Node, &Handle);\r
    if (!EFI_ERROR (Status) &&\r
        (Handle == LoadFileHandle) &&\r
        (DevicePathType (Node) == MEDIA_DEVICE_PATH) && (DevicePathSubType (Node) == MEDIA_RAM_DISK_DP)) {\r
      Handle = Handles[Index];\r
      break;\r
    }\r
  }\r
\r
  if (Handles != NULL) {\r
    FreePool (Handles);\r
  }\r
\r
  if (Index != HandleCount) {\r
    return BmExpandMediaDevicePath (DevicePathFromHandle (Handle), FullPath, FileSize);\r
  } else {\r
    return NULL;\r
  }\r
}\r
\r
/**\r
  Get the file buffer from Load File instance.\r
\r
  @param FilePath    The media device path pointing to a LoadFile instance.\r
  @param FullPath    Return the full device path pointing to the load option.\r
  @param FileSize    Return the size of the load option.\r
\r
  @return  The load option buffer.\r
**/\r
VOID *\r
BmGetFileBufferFromLoadFile (\r
  IN  EFI_DEVICE_PATH_PROTOCOL        *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL        **FullPath,\r
  OUT UINTN                           *FileSize\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  EFI_HANDLE                      Handle;\r
  VOID                            *FileBuffer;\r
  EFI_HANDLE                      *Handles;\r
  UINTN                           HandleCount;\r
  UINTN                           Index;\r
  EFI_DEVICE_PATH_PROTOCOL        *Node;\r
  EFI_LOAD_FILE_PROTOCOL          *LoadFile;\r
  UINTN                           BufferSize;\r
\r
  //\r
  // Get file buffer from load file instance.\r
  //\r
  Node = FilePath;\r
  Status = gBS->LocateDevicePath (&gEfiLoadFileProtocolGuid, &Node, &Handle);\r
  if (!EFI_ERROR (Status) && IsDevicePathEnd (Node)) {\r
    //\r
    // When wide match happens, pass full device path to LoadFile (),\r
    // otherwise, pass remaining device path to LoadFile ().\r
    //\r
    FilePath = Node;\r
  } else {\r
    Handle = NULL;\r
    //\r
    // Use wide match algorithm to find one when\r
    //  cannot find a LoadFile instance to exactly match the FilePath\r
    //\r
    Status = gBS->LocateHandleBuffer (\r
                    ByProtocol,\r
                    &gEfiLoadFileProtocolGuid,\r
                    NULL,\r
                    &HandleCount,\r
                    &Handles\r
                    );\r
    if (EFI_ERROR (Status)) {\r
      Handles = NULL;\r
      HandleCount = 0;\r
    }\r
    for (Index = 0; Index < HandleCount; Index++) {\r
      if (BmMatchHttpBootDevicePath (DevicePathFromHandle (Handles[Index]), FilePath)) {\r
        Handle = Handles[Index];\r
        break;\r
      }\r
    }\r
    if (Handles != NULL) {\r
      FreePool (Handles);\r
    }\r
  }\r
\r
  if (Handle == NULL) {\r
    return NULL;\r
  }\r
\r
  Status = gBS->HandleProtocol (Handle, &gEfiLoadFileProtocolGuid, (VOID **) &LoadFile);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  BufferSize = 0;\r
  FileBuffer = NULL;\r
  Status = LoadFile->LoadFile (LoadFile, FilePath, TRUE, &BufferSize, FileBuffer);\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
    FileBuffer = AllocatePool (BufferSize);\r
    if (FileBuffer != NULL) {\r
      Status = EFI_SUCCESS;\r
    }\r
  }\r
\r
  if (!EFI_ERROR (Status)) {\r
    Status = LoadFile->LoadFile (LoadFile, FilePath, TRUE, &BufferSize, FileBuffer);\r
  }\r
\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // LoadFile() returns a file buffer mapping to a file system.\r
    //\r
    if (Status == EFI_WARN_FILE_SYSTEM) {\r
      return BmGetFileBufferFromLoadFileFileSystem (Handle, FullPath, FileSize);\r
    }\r
\r
    ASSERT (Status == EFI_SUCCESS);\r
    //\r
    // LoadFile () may cause the device path of the Handle be updated.\r
    //\r
    *FullPath = DuplicateDevicePath (DevicePathFromHandle (Handle));\r
    *FileSize = BufferSize;\r
    return FileBuffer;\r
  } else {\r
    return NULL;\r
  }\r
}\r
\r
/**\r
  Get the load option by its device path.\r
\r
  @param FilePath  The device path pointing to a load option.\r
                   It could be a short-form device path.\r
  @param FullPath  Return the full device path of the load option after\r
                   short-form device path expanding.\r
                   Caller is responsible to free it.\r
  @param FileSize  Return the load option size.\r
\r
  @return The load option buffer. Caller is responsible to free the memory.\r
**/\r
VOID *\r
BmGetLoadOptionBuffer (\r
  IN  EFI_DEVICE_PATH_PROTOCOL          *FilePath,\r
  OUT EFI_DEVICE_PATH_PROTOCOL          **FullPath,\r
  OUT UINTN                             *FileSize\r
  )\r
{\r
  EFI_HANDLE                      Handle;\r
  VOID                            *FileBuffer;\r
  UINT32                          AuthenticationStatus;\r
  EFI_DEVICE_PATH_PROTOCOL        *Node;\r
  EFI_STATUS                      Status;\r
\r
  ASSERT ((FilePath != NULL) && (FullPath != NULL) && (FileSize != NULL));\r
\r
  EfiBootManagerConnectDevicePath (FilePath, NULL);\r
\r
  *FullPath  = NULL;\r
  *FileSize  = 0;\r
  FileBuffer = NULL;\r
\r
  //\r
  // Boot from media device by adding a default file name \EFI\BOOT\BOOT{machine type short-name}.EFI\r
  //\r
  Node = FilePath;\r
  Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid, &Node, &Handle);\r
  if (EFI_ERROR (Status)) {\r
    Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &Node, &Handle);\r
  }\r
\r
  if (!EFI_ERROR (Status) && IsDevicePathEnd (Node)) {\r
    return BmExpandMediaDevicePath (FilePath, FullPath, FileSize);\r
  }\r
\r
  //\r
  // Expand the short-form device path to full device path\r
  //\r
  if ((DevicePathType (FilePath) == MEDIA_DEVICE_PATH) &&\r
      (DevicePathSubType (FilePath) == MEDIA_HARDDRIVE_DP)) {\r
    //\r
    // Expand the Harddrive device path\r
    //\r
    return BmExpandPartitionDevicePath (FilePath, FullPath, FileSize);\r
  } else if ((DevicePathType (FilePath) == MEDIA_DEVICE_PATH) &&\r
             (DevicePathSubType (FilePath) == MEDIA_FILEPATH_DP)) {\r
    //\r
    // Expand the File-path device path\r
    //\r
    return BmExpandFileDevicePath (FilePath, FullPath, FileSize);\r
  } else if ((DevicePathType (FilePath) == MESSAGING_DEVICE_PATH) &&\r
             (DevicePathSubType (FilePath) == MSG_URI_DP)) {\r
    //\r
    // Expand the URI device path\r
    //\r
    return BmExpandUriDevicePath (FilePath, FullPath, FileSize);\r
  } else {\r
    for (Node = FilePath; !IsDevicePathEnd (Node); Node = NextDevicePathNode (Node)) {\r
      if ((DevicePathType (Node) == MESSAGING_DEVICE_PATH) &&\r
          ((DevicePathSubType (Node) == MSG_USB_CLASS_DP) || (DevicePathSubType (Node) == MSG_USB_WWID_DP))) {\r
        break;\r
      }\r
    }\r
\r
    if (!IsDevicePathEnd (Node)) {\r
      //\r
      // Expand the USB WWID/Class device path\r
      //\r
      FileBuffer = BmExpandUsbDevicePath (FilePath, FullPath, FileSize, Node);\r
      if ((FileBuffer == NULL) && (FilePath == Node)) {\r
        //\r
        // Boot Option device path starts with USB Class or USB WWID device path.\r
        // For Boot Option device path which doesn't begin with the USB Class or\r
        // USB WWID device path, it's not needed to connect again here.\r
        //\r
        BmConnectUsbShortFormDevicePath (FilePath);\r
        FileBuffer = BmExpandUsbDevicePath (FilePath, FullPath, FileSize, Node);\r
      }\r
      return FileBuffer;\r
    }\r
  }\r
\r
  //\r
  // Get file buffer from FV file path.\r
  //\r
  if (BmIsFvFilePath (FilePath)) {\r
    return BmGetFileBufferByFvFilePath (FilePath, FullPath, FileSize);\r
  }\r
\r
  //\r
  // Get file buffer from simple file system.\r
  //\r
  Node   = FilePath;\r
  Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid, &Node, &Handle);\r
  if (!EFI_ERROR (Status)) {\r
    FileBuffer = GetFileBufferByFilePath (TRUE, FilePath, FileSize, &AuthenticationStatus);\r
    if (FileBuffer != NULL) {\r
      *FullPath = DuplicateDevicePath (FilePath);\r
    }\r
    return FileBuffer;\r
  }\r
\r
  return BmGetFileBufferFromLoadFile (FilePath, FullPath, FileSize);\r
}\r
\r
/**\r
  Attempt to boot the EFI boot option. This routine sets L"BootCurent" and\r
  also signals the EFI ready to boot event. If the device path for the option\r
  starts with a BBS device path a legacy boot is attempted via the registered \r
  gLegacyBoot function. Short form device paths are also supported via this \r
  rountine. A device path starting with MEDIA_HARDDRIVE_DP, MSG_USB_WWID_DP,\r
  MSG_USB_CLASS_DP gets expaned out to find the first device that matches.\r
  If the BootOption Device Path fails the removable media boot algorithm \r
  is attempted (\EFI\BOOTIA32.EFI, \EFI\BOOTX64.EFI,... only one file type \r
  is tried per processor type)\r
\r
  @param  BootOption    Boot Option to try and boot.\r
                        On return, BootOption->Status contains the boot status.\r
                        EFI_SUCCESS     BootOption was booted\r
                        EFI_UNSUPPORTED A BBS device path was found with no valid callback\r
                                        registered via EfiBootManagerInitialize().\r
                        EFI_NOT_FOUND   The BootOption was not found on the system\r
                        !EFI_SUCCESS    BootOption failed with this error status\r
\r
**/\r
VOID\r
EFIAPI\r
EfiBootManagerBoot (\r
  IN  EFI_BOOT_MANAGER_LOAD_OPTION             *BootOption\r
  )\r
{\r
  EFI_STATUS                Status;\r
  EFI_HANDLE                ImageHandle;\r
  EFI_LOADED_IMAGE_PROTOCOL *ImageInfo;\r
  UINT16                    Uint16;\r
  UINTN                     OptionNumber;\r
  UINTN                     OriginalOptionNumber;\r
  EFI_DEVICE_PATH_PROTOCOL  *FilePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *Node;\r
  EFI_HANDLE                FvHandle;\r
  VOID                      *FileBuffer;\r
  UINTN                     FileSize;\r
  EFI_BOOT_LOGO_PROTOCOL    *BootLogo;\r
  EFI_EVENT                 LegacyBootEvent;\r
\r
  if (BootOption == NULL) {\r
    return;\r
  }\r
\r
  if (BootOption->FilePath == NULL || BootOption->OptionType != LoadOptionTypeBoot) {\r
    BootOption->Status = EFI_INVALID_PARAMETER;\r
    return;\r
  }\r
\r
  //\r
  // 1. Create Boot#### for a temporary boot if there is no match Boot#### (i.e. a boot by selected a EFI Shell using "Boot From File")\r
  //\r
  OptionNumber = BmFindBootOptionInVariable (BootOption);\r
  if (OptionNumber == LoadOptionNumberUnassigned) {\r
    Status = BmGetFreeOptionNumber (LoadOptionTypeBoot, &Uint16);\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // Save the BootOption->OptionNumber to restore later\r
      //\r
      OptionNumber             = Uint16;\r
      OriginalOptionNumber     = BootOption->OptionNumber;\r
      BootOption->OptionNumber = OptionNumber;\r
      Status = EfiBootManagerLoadOptionToVariable (BootOption);\r
      BootOption->OptionNumber = OriginalOptionNumber;\r
    }\r
\r
    if (EFI_ERROR (Status)) {\r
      DEBUG ((EFI_D_ERROR, "[Bds] Failed to create Boot#### for a temporary boot - %r!\n", Status));\r
      BootOption->Status = Status;\r
      return ;\r
    }\r
  }\r
\r
  //\r
  // 2. Set BootCurrent\r
  //\r
  Uint16 = (UINT16) OptionNumber;\r
  BmSetVariableAndReportStatusCodeOnError (\r
    L"BootCurrent",\r
    &gEfiGlobalVariableGuid,\r
    EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,\r
    sizeof (UINT16),\r
    &Uint16\r
    );\r
\r
  //\r
  // 3. Signal the EVT_SIGNAL_READY_TO_BOOT event when we are about to load and execute\r
  //    the boot option.\r
  //\r
  Node   = BootOption->FilePath;\r
  Status = gBS->LocateDevicePath (&gEfiFirmwareVolume2ProtocolGuid, &Node, &FvHandle);\r
  if (!EFI_ERROR (Status) && CompareGuid (\r
        EfiGetNameGuidFromFwVolDevicePathNode ((CONST MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) Node),\r
        PcdGetPtr (PcdBootManagerMenuFile)\r
        )) {\r
    DEBUG ((EFI_D_INFO, "[Bds] Booting Boot Manager Menu.\n"));\r
    BmStopHotkeyService (NULL, NULL);\r
  } else {\r
    EfiSignalEventReadyToBoot();\r
    //\r
    // Report Status Code to indicate ReadyToBoot was signalled\r
    //\r
    REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_DXE_BS_PC_READY_TO_BOOT_EVENT));\r
    //\r
    // 4. Repair system through DriverHealth protocol\r
    //\r
    BmRepairAllControllers ();\r
  }\r
\r
  PERF_START_EX (gImageHandle, "BdsAttempt", NULL, 0, (UINT32) OptionNumber);\r
\r
  //\r
  // 5. Load EFI boot option to ImageHandle\r
  //\r
  ImageHandle = NULL;\r
  if (DevicePathType (BootOption->FilePath) != BBS_DEVICE_PATH) {\r
    Status     = EFI_NOT_FOUND;\r
    FileBuffer = BmGetLoadOptionBuffer (BootOption->FilePath, &FilePath, &FileSize);\r
    DEBUG_CODE (\r
      if (FileBuffer != NULL && CompareMem (BootOption->FilePath, FilePath, GetDevicePathSize (FilePath)) != 0) {\r
        DEBUG ((EFI_D_INFO, "[Bds] DevicePath expand: "));\r
        BmPrintDp (BootOption->FilePath);\r
        DEBUG ((EFI_D_INFO, " -> "));\r
        BmPrintDp (FilePath);\r
        DEBUG ((EFI_D_INFO, "\n"));\r
      }\r
    );\r
    if (BmIsLoadOptionPeHeaderValid (BootOption->OptionType, FileBuffer, FileSize)) {\r
      REPORT_STATUS_CODE (EFI_PROGRESS_CODE, PcdGet32 (PcdProgressCodeOsLoaderLoad));\r
      Status = gBS->LoadImage (\r
                      TRUE,\r
                      gImageHandle,\r
                      FilePath,\r
                      FileBuffer,\r
                      FileSize,\r
                      &ImageHandle\r
                      );\r
    }\r
    if (FileBuffer != NULL) {\r
      FreePool (FileBuffer);\r
    }\r
    if (FilePath != NULL) {\r
      FreePool (FilePath);\r
    }\r
\r
    if (EFI_ERROR (Status)) {\r
      //\r
      // Report Status Code to indicate that the failure to load boot option\r
      //\r
      REPORT_STATUS_CODE (\r
        EFI_ERROR_CODE | EFI_ERROR_MINOR,\r
        (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_DXE_BS_EC_BOOT_OPTION_LOAD_ERROR)\r
        );\r
      BootOption->Status = Status;\r
      return;\r
    }\r
  }\r
\r
  //\r
  // 6. Adjust the different type memory page number just before booting\r
  //    and save the updated info into the variable for next boot to use\r
  //\r
  BmSetMemoryTypeInformationVariable (\r
    (BOOLEAN) ((BootOption->Attributes & LOAD_OPTION_CATEGORY) == LOAD_OPTION_CATEGORY_BOOT)\r
    );\r
\r
  DEBUG_CODE_BEGIN();\r
    if (BootOption->Description == NULL) {\r
      DEBUG ((DEBUG_INFO | DEBUG_LOAD, "[Bds]Booting from unknown device path\n"));\r
    } else {\r
      DEBUG ((DEBUG_INFO | DEBUG_LOAD, "[Bds]Booting %s\n", BootOption->Description));\r
    }\r
  DEBUG_CODE_END();\r
\r
  //\r
  // Check to see if we should legacy BOOT. If yes then do the legacy boot\r
  // Write boot to OS performance data for Legacy boot\r
  //\r
  if ((DevicePathType (BootOption->FilePath) == BBS_DEVICE_PATH) && (DevicePathSubType (BootOption->FilePath) == BBS_BBS_DP)) {\r
    if (mBmLegacyBoot != NULL) {\r
      //\r
      // Write boot to OS performance data for legacy boot.\r
      //\r
      PERF_CODE (\r
        //\r
        // Create an event to be signalled when Legacy Boot occurs to write performance data.\r
        //\r
        Status = EfiCreateEventLegacyBootEx(\r
                   TPL_NOTIFY,\r
                   BmWriteBootToOsPerformanceData,\r
                   NULL, \r
                   &LegacyBootEvent\r
                   );\r
        ASSERT_EFI_ERROR (Status);\r
      );\r
\r
      mBmLegacyBoot (BootOption);\r
    } else {\r
      BootOption->Status = EFI_UNSUPPORTED;\r
    }\r
\r
    PERF_END_EX (gImageHandle, "BdsAttempt", NULL, 0, (UINT32) OptionNumber);\r
    return;\r
  }\r
 \r
  //\r
  // Provide the image with its load options\r
  //\r
  Status = gBS->HandleProtocol (ImageHandle, &gEfiLoadedImageProtocolGuid, (VOID **) &ImageInfo);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  if (!BmIsAutoCreateBootOption (BootOption)) {\r
    ImageInfo->LoadOptionsSize = BootOption->OptionalDataSize;\r
    ImageInfo->LoadOptions     = BootOption->OptionalData;\r
  }\r
\r
  //\r
  // Clean to NULL because the image is loaded directly from the firmwares boot manager.\r
  //\r
  ImageInfo->ParentHandle = NULL;\r
\r
  //\r
  // Before calling the image, enable the Watchdog Timer for 5 minutes period\r
  //\r
  gBS->SetWatchdogTimer (5 * 60, 0x0000, 0x00, NULL);\r
\r
  //\r
  // Write boot to OS performance data for UEFI boot\r
  //\r
  PERF_CODE (\r
    BmWriteBootToOsPerformanceData (NULL, NULL);\r
  );\r
\r
  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, PcdGet32 (PcdProgressCodeOsLoaderStart));\r
\r
  Status = gBS->StartImage (ImageHandle, &BootOption->ExitDataSize, &BootOption->ExitData);\r
  DEBUG ((DEBUG_INFO | DEBUG_LOAD, "Image Return Status = %r\n", Status));\r
  BootOption->Status = Status;\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Report Status Code to indicate that boot failure\r
    //\r
    REPORT_STATUS_CODE (\r
      EFI_ERROR_CODE | EFI_ERROR_MINOR,\r
      (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_DXE_BS_EC_BOOT_OPTION_FAILED)\r
      );\r
  }\r
  PERF_END_EX (gImageHandle, "BdsAttempt", NULL, 0, (UINT32) OptionNumber);\r
\r
  //\r
  // Clear the Watchdog Timer after the image returns\r
  //\r
  gBS->SetWatchdogTimer (0x0000, 0x0000, 0x0000, NULL);\r
\r
  //\r
  // Set Logo status invalid after trying one boot option\r
  //\r
  BootLogo = NULL;\r
  Status = gBS->LocateProtocol (&gEfiBootLogoProtocolGuid, NULL, (VOID **) &BootLogo);\r
  if (!EFI_ERROR (Status) && (BootLogo != NULL)) {\r
    Status = BootLogo->SetBootLogo (BootLogo, NULL, 0, 0, 0, 0);\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
\r
  //\r
  // Clear Boot Current\r
  //\r
  Status = gRT->SetVariable (\r
                  L"BootCurrent",\r
                  &gEfiGlobalVariableGuid,\r
                  0,\r
                  0,\r
                  NULL\r
                  );\r
  //\r
  // Deleting variable with current variable implementation shouldn't fail.\r
  // When BootXXXX (e.g.: BootManagerMenu) boots BootYYYY, exiting BootYYYY causes BootCurrent deleted,\r
  // exiting BootXXXX causes deleting BootCurrent returns EFI_NOT_FOUND.\r
  //\r
  ASSERT (Status == EFI_SUCCESS || Status == EFI_NOT_FOUND);\r
}\r
\r
/**\r
  Check whether there is a instance in BlockIoDevicePath, which contain multi device path\r
  instances, has the same partition node with HardDriveDevicePath device path\r
\r
  @param  BlockIoDevicePath      Multi device path instances which need to check\r
  @param  HardDriveDevicePath    A device path which starts with a hard drive media\r
                                 device path.\r
\r
  @retval TRUE                   There is a matched device path instance.\r
  @retval FALSE                  There is no matched device path instance.\r
\r
**/\r
BOOLEAN\r
BmMatchPartitionDevicePathNode (\r
  IN  EFI_DEVICE_PATH_PROTOCOL   *BlockIoDevicePath,\r
  IN  HARDDRIVE_DEVICE_PATH      *HardDriveDevicePath\r
  )\r
{\r
  HARDDRIVE_DEVICE_PATH     *Node;\r
\r
  if ((BlockIoDevicePath == NULL) || (HardDriveDevicePath == NULL)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // find the partition device path node\r
  //\r
  while (!IsDevicePathEnd (BlockIoDevicePath)) {\r
    if ((DevicePathType (BlockIoDevicePath) == MEDIA_DEVICE_PATH) &&\r
        (DevicePathSubType (BlockIoDevicePath) == MEDIA_HARDDRIVE_DP)\r
        ) {\r
      break;\r
    }\r
\r
    BlockIoDevicePath = NextDevicePathNode (BlockIoDevicePath);\r
  }\r
\r
  if (IsDevicePathEnd (BlockIoDevicePath)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // See if the harddrive device path in blockio matches the orig Hard Drive Node\r
  //\r
  Node = (HARDDRIVE_DEVICE_PATH *) BlockIoDevicePath;\r
\r
  //\r
  // Match Signature and PartitionNumber.\r
  // Unused bytes in Signature are initiaized with zeros.\r
  //\r
  return (BOOLEAN) (\r
    (Node->PartitionNumber == HardDriveDevicePath->PartitionNumber) &&\r
    (Node->MBRType == HardDriveDevicePath->MBRType) &&\r
    (Node->SignatureType == HardDriveDevicePath->SignatureType) &&\r
    (CompareMem (Node->Signature, HardDriveDevicePath->Signature, sizeof (Node->Signature)) == 0)\r
    );\r
}\r
\r
/**\r
  Emuerate all possible bootable medias in the following order:\r
  1. Removable BlockIo            - The boot option only points to the removable media\r
                                    device, like USB key, DVD, Floppy etc.\r
  2. Fixed BlockIo                - The boot option only points to a Fixed blockIo device,\r
                                    like HardDisk.\r
  3. Non-BlockIo SimpleFileSystem - The boot option points to a device supporting\r
                                    SimpleFileSystem Protocol, but not supporting BlockIo\r
                                    protocol.\r
  4. LoadFile                     - The boot option points to the media supporting \r
                                    LoadFile protocol.\r
  Reference: UEFI Spec chapter 3.3 Boot Option Variables Default Boot Behavior\r
\r
  @param BootOptionCount   Return the boot option count which has been found.\r
\r
  @retval   Pointer to the boot option array.\r
**/\r
EFI_BOOT_MANAGER_LOAD_OPTION *\r
BmEnumerateBootOptions (\r
  UINTN                                 *BootOptionCount\r
  )\r
{\r
  EFI_STATUS                            Status;\r
  EFI_BOOT_MANAGER_LOAD_OPTION          *BootOptions;\r
  UINTN                                 HandleCount;\r
  EFI_HANDLE                            *Handles;\r
  EFI_BLOCK_IO_PROTOCOL                 *BlkIo;\r
  UINTN                                 Removable;\r
  UINTN                                 Index;\r
  CHAR16                                *Description;\r
\r
  ASSERT (BootOptionCount != NULL);\r
\r
  *BootOptionCount = 0;\r
  BootOptions      = NULL;\r
\r
  //\r
  // Parse removable block io followed by fixed block io\r
  //\r
  gBS->LocateHandleBuffer (\r
         ByProtocol,\r
         &gEfiBlockIoProtocolGuid,\r
         NULL,\r
         &HandleCount,\r
         &Handles\r
         );\r
\r
  for (Removable = 0; Removable < 2; Removable++) {\r
    for (Index = 0; Index < HandleCount; Index++) {\r
      Status = gBS->HandleProtocol (\r
                      Handles[Index],\r
                      &gEfiBlockIoProtocolGuid,\r
                      (VOID **) &BlkIo\r
                      );\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
\r
      //\r
      // Skip the logical partitions\r
      //\r
      if (BlkIo->Media->LogicalPartition) {\r
        continue;\r
      }\r
\r
      //\r
      // Skip the fixed block io then the removable block io\r
      //\r
      if (BlkIo->Media->RemovableMedia == ((Removable == 0) ? FALSE : TRUE)) {\r
        continue;\r
      }\r
\r
      Description = BmGetBootDescription (Handles[Index]);\r
      BootOptions = ReallocatePool (\r
                      sizeof (EFI_BOOT_MANAGER_LOAD_OPTION) * (*BootOptionCount),\r
                      sizeof (EFI_BOOT_MANAGER_LOAD_OPTION) * (*BootOptionCount + 1),\r
                      BootOptions\r
                      );\r
      ASSERT (BootOptions != NULL);\r
\r
      Status = EfiBootManagerInitializeLoadOption (\r
                 &BootOptions[(*BootOptionCount)++],\r
                 LoadOptionNumberUnassigned,\r
                 LoadOptionTypeBoot,\r
                 LOAD_OPTION_ACTIVE,\r
                 Description,\r
                 DevicePathFromHandle (Handles[Index]),\r
                 NULL,\r
                 0\r
                 );\r
      ASSERT_EFI_ERROR (Status);\r
\r
      FreePool (Description);\r
    }\r
  }\r
\r
  if (HandleCount != 0) {\r
    FreePool (Handles);\r
  }\r
\r
  //\r
  // Parse simple file system not based on block io\r
  //\r
  gBS->LocateHandleBuffer (\r
         ByProtocol,\r
         &gEfiSimpleFileSystemProtocolGuid,\r
         NULL,\r
         &HandleCount,\r
         &Handles\r
         );\r
  for (Index = 0; Index < HandleCount; Index++) {\r
    Status = gBS->HandleProtocol (\r
                    Handles[Index],\r
                    &gEfiBlockIoProtocolGuid,\r
                    (VOID **) &BlkIo\r
                    );\r
     if (!EFI_ERROR (Status)) {\r
      //\r
      //  Skip if the file system handle supports a BlkIo protocol, which we've handled in above\r
      //\r
      continue;\r
    }\r
    Description = BmGetBootDescription (Handles[Index]);\r
    BootOptions = ReallocatePool (\r
                    sizeof (EFI_BOOT_MANAGER_LOAD_OPTION) * (*BootOptionCount),\r
                    sizeof (EFI_BOOT_MANAGER_LOAD_OPTION) * (*BootOptionCount + 1),\r
                    BootOptions\r
                    );\r
    ASSERT (BootOptions != NULL);\r
\r
    Status = EfiBootManagerInitializeLoadOption (\r
               &BootOptions[(*BootOptionCount)++],\r
               LoadOptionNumberUnassigned,\r
               LoadOptionTypeBoot,\r
               LOAD_OPTION_ACTIVE,\r
               Description,\r
               DevicePathFromHandle (Handles[Index]),\r
               NULL,\r
               0\r
               );\r
    ASSERT_EFI_ERROR (Status);\r
    FreePool (Description);\r
  }\r
\r
  if (HandleCount != 0) {\r
    FreePool (Handles);\r
  }\r
\r
  //\r
  // Parse load file, assuming UEFI Network boot option\r
  //\r
  gBS->LocateHandleBuffer (\r
         ByProtocol,\r
         &gEfiLoadFileProtocolGuid,\r
         NULL,\r
         &HandleCount,\r
         &Handles\r
         );\r
  for (Index = 0; Index < HandleCount; Index++) {\r
\r
    Description = BmGetBootDescription (Handles[Index]);\r
    BootOptions = ReallocatePool (\r
                    sizeof (EFI_BOOT_MANAGER_LOAD_OPTION) * (*BootOptionCount),\r
                    sizeof (EFI_BOOT_MANAGER_LOAD_OPTION) * (*BootOptionCount + 1),\r
                    BootOptions\r
                    );\r
    ASSERT (BootOptions != NULL);\r
\r
    Status = EfiBootManagerInitializeLoadOption (\r
               &BootOptions[(*BootOptionCount)++],\r
               LoadOptionNumberUnassigned,\r
               LoadOptionTypeBoot,\r
               LOAD_OPTION_ACTIVE,\r
               Description,\r
               DevicePathFromHandle (Handles[Index]),\r
               NULL,\r
               0\r
               );\r
    ASSERT_EFI_ERROR (Status);\r
    FreePool (Description);\r
  }\r
\r
  if (HandleCount != 0) {\r
    FreePool (Handles);\r
  }\r
\r
  BmMakeBootOptionDescriptionUnique (BootOptions, *BootOptionCount);\r
  return BootOptions;\r
}\r
\r
/**\r
  The function enumerates all boot options, creates them and registers them in the BootOrder variable.\r
**/\r
VOID\r
EFIAPI\r
EfiBootManagerRefreshAllBootOption (\r
  VOID\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  EFI_BOOT_MANAGER_LOAD_OPTION  *NvBootOptions;\r
  UINTN                         NvBootOptionCount;\r
  EFI_BOOT_MANAGER_LOAD_OPTION  *BootOptions;\r
  UINTN                         BootOptionCount;\r
  UINTN                         Index;\r
\r
  //\r
  // Optionally refresh the legacy boot option\r
  //\r
  if (mBmRefreshLegacyBootOption != NULL) {\r
    mBmRefreshLegacyBootOption ();\r
  }\r
\r
  BootOptions   = BmEnumerateBootOptions (&BootOptionCount);\r
  NvBootOptions = EfiBootManagerGetLoadOptions (&NvBootOptionCount, LoadOptionTypeBoot);\r
\r
  //\r
  // Mark the boot option as added by BDS by setting OptionalData to a special GUID\r
  //\r
  for (Index = 0; Index < BootOptionCount; Index++) {\r
    BootOptions[Index].OptionalData     = AllocateCopyPool (sizeof (EFI_GUID), &mBmAutoCreateBootOptionGuid);\r
    BootOptions[Index].OptionalDataSize = sizeof (EFI_GUID);\r
  }\r
\r
  //\r
  // Remove invalid EFI boot options from NV\r
  //\r
  for (Index = 0; Index < NvBootOptionCount; Index++) {\r
    if (((DevicePathType (NvBootOptions[Index].FilePath) != BBS_DEVICE_PATH) || \r
         (DevicePathSubType (NvBootOptions[Index].FilePath) != BBS_BBS_DP)\r
        ) && BmIsAutoCreateBootOption (&NvBootOptions[Index])\r
       ) {\r
      //\r
      // Only check those added by BDS\r
      // so that the boot options added by end-user or OS installer won't be deleted\r
      //\r
      if (EfiBootManagerFindLoadOption (&NvBootOptions[Index], BootOptions, BootOptionCount) == (UINTN) -1) {\r
        Status = EfiBootManagerDeleteLoadOptionVariable (NvBootOptions[Index].OptionNumber, LoadOptionTypeBoot);\r
        //\r
        // Deleting variable with current variable implementation shouldn't fail.\r
        //\r
        ASSERT_EFI_ERROR (Status);\r
      }\r
    }\r
  }\r
\r
  //\r
  // Add new EFI boot options to NV\r
  //\r
  for (Index = 0; Index < BootOptionCount; Index++) {\r
    if (EfiBootManagerFindLoadOption (&BootOptions[Index], NvBootOptions, NvBootOptionCount) == (UINTN) -1) {\r
      EfiBootManagerAddLoadOptionVariable (&BootOptions[Index], (UINTN) -1);\r
      //\r
      // Try best to add the boot options so continue upon failure.\r
      //\r
    }\r
  }\r
\r
  EfiBootManagerFreeLoadOptions (BootOptions,   BootOptionCount);\r
  EfiBootManagerFreeLoadOptions (NvBootOptions, NvBootOptionCount);\r
}\r
\r
/**\r
  This function is called to create the boot option for the Boot Manager Menu.\r
\r
  The Boot Manager Menu is shown after successfully booting a boot option.\r
  Assume the BootManagerMenuFile is in the same FV as the module links to this library.\r
\r
  @param  BootOption    Return the boot option of the Boot Manager Menu\r
\r
  @retval EFI_SUCCESS   Successfully register the Boot Manager Menu.\r
  @retval Status        Return status of gRT->SetVariable (). BootOption still points\r
                        to the Boot Manager Menu even the Status is not EFI_SUCCESS.\r
**/\r
EFI_STATUS\r
BmRegisterBootManagerMenu (\r
  OUT EFI_BOOT_MANAGER_LOAD_OPTION   *BootOption\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  CHAR16                             *Description;\r
  UINTN                              DescriptionLength;\r
  EFI_DEVICE_PATH_PROTOCOL           *DevicePath;\r
  EFI_LOADED_IMAGE_PROTOCOL          *LoadedImage;\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  FileNode;\r
\r
  Status = GetSectionFromFv (\r
             PcdGetPtr (PcdBootManagerMenuFile),\r
             EFI_SECTION_USER_INTERFACE,\r
             0,\r
             (VOID **) &Description,\r
             &DescriptionLength\r
             );\r
  if (EFI_ERROR (Status)) {\r
    Description = NULL;\r
  }\r
\r
  EfiInitializeFwVolDevicepathNode (&FileNode, PcdGetPtr (PcdBootManagerMenuFile));\r
  Status = gBS->HandleProtocol (\r
                  gImageHandle,\r
                  &gEfiLoadedImageProtocolGuid,\r
                  (VOID **) &LoadedImage\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
  DevicePath = AppendDevicePathNode (\r
                 DevicePathFromHandle (LoadedImage->DeviceHandle),\r
                 (EFI_DEVICE_PATH_PROTOCOL *) &FileNode\r
                 );\r
  ASSERT (DevicePath != NULL);\r
\r
  Status = EfiBootManagerInitializeLoadOption (\r
             BootOption,\r
             LoadOptionNumberUnassigned,\r
             LoadOptionTypeBoot,\r
             LOAD_OPTION_CATEGORY_APP | LOAD_OPTION_ACTIVE | LOAD_OPTION_HIDDEN,\r
             (Description != NULL) ? Description : L"Boot Manager Menu",\r
             DevicePath,\r
             NULL,\r
             0\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
  FreePool (DevicePath);\r
  if (Description != NULL) {\r
    FreePool (Description);\r
  }\r
\r
  DEBUG_CODE (\r
    EFI_BOOT_MANAGER_LOAD_OPTION    *BootOptions;\r
    UINTN                           BootOptionCount;\r
\r
    BootOptions = EfiBootManagerGetLoadOptions (&BootOptionCount, LoadOptionTypeBoot);\r
    ASSERT (EfiBootManagerFindLoadOption (BootOption, BootOptions, BootOptionCount) == -1);\r
    EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
    );\r
\r
  return EfiBootManagerAddLoadOptionVariable (BootOption, 0);\r
}\r
\r
/**\r
  Return the boot option corresponding to the Boot Manager Menu.\r
  It may automatically create one if the boot option hasn't been created yet.\r
  \r
  @param BootOption    Return the Boot Manager Menu.\r
\r
  @retval EFI_SUCCESS   The Boot Manager Menu is successfully returned.\r
  @retval Status        Return status of gRT->SetVariable (). BootOption still points\r
                        to the Boot Manager Menu even the Status is not EFI_SUCCESS.\r
**/\r
EFI_STATUS\r
EFIAPI\r
EfiBootManagerGetBootManagerMenu (\r
  EFI_BOOT_MANAGER_LOAD_OPTION *BootOption\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  UINTN                        BootOptionCount;\r
  EFI_BOOT_MANAGER_LOAD_OPTION *BootOptions;\r
  UINTN                        Index;\r
  EFI_DEVICE_PATH_PROTOCOL     *Node;\r
  EFI_HANDLE                   FvHandle;\r
  \r
  BootOptions = EfiBootManagerGetLoadOptions (&BootOptionCount, LoadOptionTypeBoot);\r
\r
  for (Index = 0; Index < BootOptionCount; Index++) {\r
    Node   = BootOptions[Index].FilePath;\r
    Status = gBS->LocateDevicePath (&gEfiFirmwareVolume2ProtocolGuid, &Node, &FvHandle);\r
    if (!EFI_ERROR (Status)) {\r
      if (CompareGuid (\r
            EfiGetNameGuidFromFwVolDevicePathNode ((CONST MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) Node),\r
            PcdGetPtr (PcdBootManagerMenuFile)\r
            )\r
          ) {        \r
        Status = EfiBootManagerInitializeLoadOption (\r
                   BootOption,\r
                   BootOptions[Index].OptionNumber,\r
                   BootOptions[Index].OptionType,\r
                   BootOptions[Index].Attributes,\r
                   BootOptions[Index].Description,\r
                   BootOptions[Index].FilePath,\r
                   BootOptions[Index].OptionalData,\r
                   BootOptions[Index].OptionalDataSize\r
                   );\r
        ASSERT_EFI_ERROR (Status);\r
        break;\r
      }\r
    }\r
  }\r
\r
  EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
\r
  //\r
  // Automatically create the Boot#### for Boot Manager Menu when not found.\r
  //\r
  if (Index == BootOptionCount) {\r
    return BmRegisterBootManagerMenu (BootOption);\r
  } else {\r
    return EFI_SUCCESS;\r
  }\r
}\r
\r