[mirror_edk2.git] / OvmfPkg / Library / PlatformBootManagerLib / BdsPlatform.c

/** @file\r
  Platform BDS customizations.\r
\r
  Copyright (c) 2004 - 2016, Intel Corporation. All rights reserved.<BR>\r
  This program and the accompanying materials are licensed and made available\r
  under the terms and conditions of the BSD License which accompanies this\r
  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, WITHOUT\r
  WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "BdsPlatform.h"\r
#include <Guid/XenInfo.h>\r
#include <Guid/RootBridgesConnectedEventGroup.h>\r
#include <Protocol/FirmwareVolume2.h>\r
\r
\r
//\r
// Global data\r
//\r
\r
VOID          *mEfiDevPathNotifyReg;\r
EFI_EVENT     mEfiDevPathEvent;\r
VOID          *mEmuVariableEventReg;\r
EFI_EVENT     mEmuVariableEvent;\r
BOOLEAN       mDetectVgaOnly;\r
UINT16        mHostBridgeDevId;\r
\r
//\r
// Table of host IRQs matching PCI IRQs A-D\r
// (for configuring PCI Interrupt Line register)\r
//\r
CONST UINT8 PciHostIrqs[] = {\r
  0x0a, 0x0a, 0x0b, 0x0b\r
};\r
\r
//\r
// Type definitions\r
//\r
\r
typedef\r
EFI_STATUS\r
(EFIAPI *PROTOCOL_INSTANCE_CALLBACK)(\r
  IN EFI_HANDLE           Handle,\r
  IN VOID                 *Instance,\r
  IN VOID                 *Context\r
  );\r
\r
/**\r
  @param[in]  Handle - Handle of PCI device instance\r
  @param[in]  PciIo - PCI IO protocol instance\r
  @param[in]  Pci - PCI Header register block\r
**/\r
typedef\r
EFI_STATUS\r
(EFIAPI *VISIT_PCI_INSTANCE_CALLBACK)(\r
  IN EFI_HANDLE           Handle,\r
  IN EFI_PCI_IO_PROTOCOL  *PciIo,\r
  IN PCI_TYPE00           *Pci\r
  );\r
\r
\r
//\r
// Function prototypes\r
//\r
\r
EFI_STATUS\r
VisitAllInstancesOfProtocol (\r
  IN EFI_GUID                    *Id,\r
  IN PROTOCOL_INSTANCE_CALLBACK  CallBackFunction,\r
  IN VOID                        *Context\r
  );\r
\r
EFI_STATUS\r
VisitAllPciInstancesOfProtocol (\r
  IN VISIT_PCI_INSTANCE_CALLBACK CallBackFunction\r
  );\r
\r
VOID\r
InstallDevicePathCallback (\r
  VOID\r
  );\r
\r
VOID\r
PlatformRegisterFvBootOption (\r
  EFI_GUID                         *FileGuid,\r
  CHAR16                           *Description,\r
  UINT32                           Attributes\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  INTN                              OptionIndex;\r
  EFI_BOOT_MANAGER_LOAD_OPTION      NewOption;\r
  EFI_BOOT_MANAGER_LOAD_OPTION      *BootOptions;\r
  UINTN                             BootOptionCount;\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH FileNode;\r
  EFI_LOADED_IMAGE_PROTOCOL         *LoadedImage;\r
  EFI_DEVICE_PATH_PROTOCOL          *DevicePath;\r
\r
  Status = gBS->HandleProtocol (\r
                  gImageHandle,\r
                  &gEfiLoadedImageProtocolGuid,\r
                  (VOID **) &LoadedImage\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  EfiInitializeFwVolDevicepathNode (&FileNode, FileGuid);\r
  DevicePath = DevicePathFromHandle (LoadedImage->DeviceHandle);\r
  ASSERT (DevicePath != NULL);\r
  DevicePath = AppendDevicePathNode (\r
                 DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *) &FileNode\r
                 );\r
  ASSERT (DevicePath != NULL);\r
\r
  Status = EfiBootManagerInitializeLoadOption (\r
             &NewOption,\r
             LoadOptionNumberUnassigned,\r
             LoadOptionTypeBoot,\r
             Attributes,\r
             Description,\r
             DevicePath,\r
             NULL,\r
             0\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
  FreePool (DevicePath);\r
\r
  BootOptions = EfiBootManagerGetLoadOptions (\r
                  &BootOptionCount, LoadOptionTypeBoot\r
                  );\r
\r
  OptionIndex = EfiBootManagerFindLoadOption (\r
                  &NewOption, BootOptions, BootOptionCount\r
                  );\r
\r
  if (OptionIndex == -1) {\r
    Status = EfiBootManagerAddLoadOptionVariable (&NewOption, MAX_UINTN);\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
  EfiBootManagerFreeLoadOption (&NewOption);\r
  EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
}\r
\r
/**\r
  Remove all MemoryMapped(...)/FvFile(...) and Fv(...)/FvFile(...) boot options\r
  whose device paths do not resolve exactly to an FvFile in the system.\r
\r
  This removes any boot options that point to binaries built into the firmware\r
  and have become stale due to any of the following:\r
  - DXEFV's base address or size changed (historical),\r
  - DXEFV's FvNameGuid changed,\r
  - the FILE_GUID of the pointed-to binary changed,\r
  - the referenced binary is no longer built into the firmware.\r
\r
  EfiBootManagerFindLoadOption() used in PlatformRegisterFvBootOption() only\r
  avoids exact duplicates.\r
**/\r
VOID\r
RemoveStaleFvFileOptions (\r
  VOID\r
  )\r
{\r
  EFI_BOOT_MANAGER_LOAD_OPTION *BootOptions;\r
  UINTN                        BootOptionCount;\r
  UINTN                        Index;\r
\r
  BootOptions = EfiBootManagerGetLoadOptions (&BootOptionCount,\r
                  LoadOptionTypeBoot);\r
\r
  for (Index = 0; Index < BootOptionCount; ++Index) {\r
    EFI_DEVICE_PATH_PROTOCOL *Node1, *Node2, *SearchNode;\r
    EFI_STATUS               Status;\r
    EFI_HANDLE               FvHandle;\r
\r
    //\r
    // If the device path starts with neither MemoryMapped(...) nor Fv(...),\r
    // then keep the boot option.\r
    //\r
    Node1 = BootOptions[Index].FilePath;\r
    if (!(DevicePathType (Node1) == HARDWARE_DEVICE_PATH &&\r
          DevicePathSubType (Node1) == HW_MEMMAP_DP) &&\r
        !(DevicePathType (Node1) == MEDIA_DEVICE_PATH &&\r
          DevicePathSubType (Node1) == MEDIA_PIWG_FW_VOL_DP)) {\r
      continue;\r
    }\r
\r
    //\r
    // If the second device path node is not FvFile(...), then keep the boot\r
    // option.\r
    //\r
    Node2 = NextDevicePathNode (Node1);\r
    if (DevicePathType (Node2) != MEDIA_DEVICE_PATH ||\r
        DevicePathSubType (Node2) != MEDIA_PIWG_FW_FILE_DP) {\r
      continue;\r
    }\r
\r
    //\r
    // Locate the Firmware Volume2 protocol instance that is denoted by the\r
    // boot option. If this lookup fails (i.e., the boot option references a\r
    // firmware volume that doesn't exist), then we'll proceed to delete the\r
    // boot option.\r
    //\r
    SearchNode = Node1;\r
    Status = gBS->LocateDevicePath (&gEfiFirmwareVolume2ProtocolGuid,\r
                    &SearchNode, &FvHandle);\r
\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // The firmware volume was found; now let's see if it contains the FvFile\r
      // identified by GUID.\r
      //\r
      EFI_FIRMWARE_VOLUME2_PROTOCOL     *FvProtocol;\r
      MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FvFileNode;\r
      UINTN                             BufferSize;\r
      EFI_FV_FILETYPE                   FoundType;\r
      EFI_FV_FILE_ATTRIBUTES            FileAttributes;\r
      UINT32                            AuthenticationStatus;\r
\r
      Status = gBS->HandleProtocol (FvHandle, &gEfiFirmwareVolume2ProtocolGuid,\r
                      (VOID **)&FvProtocol);\r
      ASSERT_EFI_ERROR (Status);\r
\r
      FvFileNode = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)Node2;\r
      //\r
      // Buffer==NULL means we request metadata only: BufferSize, FoundType,\r
      // FileAttributes.\r
      //\r
      Status = FvProtocol->ReadFile (\r
                             FvProtocol,\r
                             &FvFileNode->FvFileName, // NameGuid\r
                             NULL,                    // Buffer\r
                             &BufferSize,\r
                             &FoundType,\r
                             &FileAttributes,\r
                             &AuthenticationStatus\r
                             );\r
      if (!EFI_ERROR (Status)) {\r
        //\r
        // The FvFile was found. Keep the boot option.\r
        //\r
        continue;\r
      }\r
    }\r
\r
    //\r
    // Delete the boot option.\r
    //\r
    Status = EfiBootManagerDeleteLoadOptionVariable (\r
               BootOptions[Index].OptionNumber, LoadOptionTypeBoot);\r
    DEBUG_CODE (\r
      CHAR16 *DevicePathString;\r
\r
      DevicePathString = ConvertDevicePathToText(BootOptions[Index].FilePath,\r
                           FALSE, FALSE);\r
      DEBUG ((\r
        EFI_ERROR (Status) ? EFI_D_WARN : EFI_D_VERBOSE,\r
        "%a: removing stale Boot#%04x %s: %r\n",\r
        __FUNCTION__,\r
        (UINT32)BootOptions[Index].OptionNumber,\r
        DevicePathString == NULL ? L"<unavailable>" : DevicePathString,\r
        Status\r
        ));\r
      if (DevicePathString != NULL) {\r
        FreePool (DevicePathString);\r
      }\r
      );\r
  }\r
\r
  EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
}\r
\r
VOID\r
PlatformRegisterOptionsAndKeys (\r
  VOID\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  EFI_INPUT_KEY                Enter;\r
  EFI_INPUT_KEY                F2;\r
  EFI_INPUT_KEY                Esc;\r
  EFI_BOOT_MANAGER_LOAD_OPTION BootOption;\r
\r
  //\r
  // Register ENTER as CONTINUE key\r
  //\r
  Enter.ScanCode    = SCAN_NULL;\r
  Enter.UnicodeChar = CHAR_CARRIAGE_RETURN;\r
  Status = EfiBootManagerRegisterContinueKeyOption (0, &Enter, NULL);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Map F2 to Boot Manager Menu\r
  //\r
  F2.ScanCode     = SCAN_F2;\r
  F2.UnicodeChar  = CHAR_NULL;\r
  Esc.ScanCode    = SCAN_ESC;\r
  Esc.UnicodeChar = CHAR_NULL;\r
  Status = EfiBootManagerGetBootManagerMenu (&BootOption);\r
  ASSERT_EFI_ERROR (Status);\r
  Status = EfiBootManagerAddKeyOptionVariable (\r
             NULL, (UINT16) BootOption.OptionNumber, 0, &F2, NULL\r
             );\r
  ASSERT (Status == EFI_SUCCESS || Status == EFI_ALREADY_STARTED);\r
  Status = EfiBootManagerAddKeyOptionVariable (\r
             NULL, (UINT16) BootOption.OptionNumber, 0, &Esc, NULL\r
             );\r
  ASSERT (Status == EFI_SUCCESS || Status == EFI_ALREADY_STARTED);\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
ConnectRootBridge (\r
  IN EFI_HANDLE  RootBridgeHandle,\r
  IN VOID        *Instance,\r
  IN VOID        *Context\r
  );\r
\r
STATIC\r
VOID\r
SaveS3BootScript (\r
  VOID\r
  );\r
\r
//\r
// BDS Platform Functions\r
//\r
VOID\r
EFIAPI\r
PlatformBootManagerBeforeConsole (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Platform Bds init. Include the platform firmware vendor, revision\r
  and so crc check.\r
\r
Arguments:\r
\r
Returns:\r
\r
  None.\r
\r
--*/\r
{\r
  EFI_HANDLE    Handle;\r
  EFI_STATUS    Status;\r
  RETURN_STATUS PcdStatus;\r
\r
  DEBUG ((EFI_D_INFO, "PlatformBootManagerBeforeConsole\n"));\r
  InstallDevicePathCallback ();\r
\r
  VisitAllInstancesOfProtocol (&gEfiPciRootBridgeIoProtocolGuid,\r
    ConnectRootBridge, NULL);\r
\r
  //\r
  // Signal the ACPI platform driver that it can download QEMU ACPI tables.\r
  //\r
  EfiEventGroupSignal (&gRootBridgesConnectedEventGroupGuid);\r
\r
  //\r
  // We can't signal End-of-Dxe earlier than this. Namely, End-of-Dxe triggers\r
  // the preparation of S3 system information. That logic has a hard dependency\r
  // on the presence of the FACS ACPI table. Since our ACPI tables are only\r
  // installed after PCI enumeration completes, we must not trigger the S3 save\r
  // earlier, hence we can't signal End-of-Dxe earlier.\r
  //\r
  EfiEventGroupSignal (&gEfiEndOfDxeEventGroupGuid);\r
\r
  if (QemuFwCfgS3Enabled ()) {\r
    //\r
    // Save the boot script too. Note that this will require us to emit the\r
    // DxeSmmReadyToLock event just below, which in turn locks down SMM.\r
    //\r
    SaveS3BootScript ();\r
  }\r
\r
  //\r
  // Prevent further changes to LockBoxes or SMRAM.\r
  //\r
  Handle = NULL;\r
  Status = gBS->InstallProtocolInterface (&Handle,\r
                  &gEfiDxeSmmReadyToLockProtocolGuid, EFI_NATIVE_INTERFACE,\r
                  NULL);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Dispatch deferred images after EndOfDxe event and ReadyToLock\r
  // installation.\r
  //\r
  EfiBootManagerDispatchDeferredImages ();\r
\r
  PlatformInitializeConsole (gPlatformConsole);\r
  PcdStatus = PcdSet16S (PcdPlatformBootTimeOut,\r
                GetFrontPageTimeoutFromQemu ());\r
  ASSERT_RETURN_ERROR (PcdStatus);\r
\r
  PlatformRegisterOptionsAndKeys ();\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
ConnectRootBridge (\r
  IN EFI_HANDLE  RootBridgeHandle,\r
  IN VOID        *Instance,\r
  IN VOID        *Context\r
  )\r
{\r
  EFI_STATUS Status;\r
\r
  //\r
  // Make the PCI bus driver connect the root bridge, non-recursively. This\r
  // will produce a number of child handles with PciIo on them.\r
  //\r
  Status = gBS->ConnectController (\r
                  RootBridgeHandle, // ControllerHandle\r
                  NULL,             // DriverImageHandle\r
                  NULL,             // RemainingDevicePath -- produce all\r
                                    //   children\r
                  FALSE             // Recursive\r
                  );\r
  return Status;\r
}\r
\r
\r
EFI_STATUS\r
PrepareLpcBridgeDevicePath (\r
  IN EFI_HANDLE                DeviceHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Add IsaKeyboard to ConIn,\r
  add IsaSerial to ConOut, ConIn, ErrOut.\r
  LPC Bridge: 06 01 00\r
\r
Arguments:\r
\r
  DeviceHandle            - Handle of PCIIO protocol.\r
\r
Returns:\r
\r
  EFI_SUCCESS             - LPC bridge is added to ConOut, ConIn, and ErrOut.\r
  EFI_STATUS              - No LPC bridge is added.\r
\r
--*/\r
{\r
  EFI_STATUS                Status;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *TempDevicePath;\r
  CHAR16                    *DevPathStr;\r
\r
  DevicePath = NULL;\r
  Status = gBS->HandleProtocol (\r
                  DeviceHandle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  (VOID*)&DevicePath\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  TempDevicePath = DevicePath;\r
\r
  //\r
  // Register Keyboard\r
  //\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gPnpPs2KeyboardDeviceNode);\r
\r
  EfiBootManagerUpdateConsoleVariable (ConIn, DevicePath, NULL);\r
\r
  //\r
  // Register COM1\r
  //\r
  DevicePath = TempDevicePath;\r
  gPnp16550ComPortDeviceNode.UID = 0;\r
\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gPnp16550ComPortDeviceNode);\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gUartDeviceNode);\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gTerminalTypeDeviceNode);\r
\r
  //\r
  // Print Device Path\r
  //\r
  DevPathStr = ConvertDevicePathToText (DevicePath, FALSE, FALSE);\r
  if (DevPathStr != NULL) {\r
    DEBUG((\r
      EFI_D_INFO,\r
      "BdsPlatform.c+%d: COM%d DevPath: %s\n",\r
      __LINE__,\r
      gPnp16550ComPortDeviceNode.UID + 1,\r
      DevPathStr\r
      ));\r
    FreePool(DevPathStr);\r
  }\r
\r
  EfiBootManagerUpdateConsoleVariable (ConOut, DevicePath, NULL);\r
  EfiBootManagerUpdateConsoleVariable (ConIn, DevicePath, NULL);\r
  EfiBootManagerUpdateConsoleVariable (ErrOut, DevicePath, NULL);\r
\r
  //\r
  // Register COM2\r
  //\r
  DevicePath = TempDevicePath;\r
  gPnp16550ComPortDeviceNode.UID = 1;\r
\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gPnp16550ComPortDeviceNode);\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gUartDeviceNode);\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gTerminalTypeDeviceNode);\r
\r
  //\r
  // Print Device Path\r
  //\r
  DevPathStr = ConvertDevicePathToText (DevicePath, FALSE, FALSE);\r
  if (DevPathStr != NULL) {\r
    DEBUG((\r
      EFI_D_INFO,\r
      "BdsPlatform.c+%d: COM%d DevPath: %s\n",\r
      __LINE__,\r
      gPnp16550ComPortDeviceNode.UID + 1,\r
      DevPathStr\r
      ));\r
    FreePool(DevPathStr);\r
  }\r
\r
  EfiBootManagerUpdateConsoleVariable (ConOut, DevicePath, NULL);\r
  EfiBootManagerUpdateConsoleVariable (ConIn, DevicePath, NULL);\r
  EfiBootManagerUpdateConsoleVariable (ErrOut, DevicePath, NULL);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
GetGopDevicePath (\r
   IN  EFI_DEVICE_PATH_PROTOCOL *PciDevicePath,\r
   OUT EFI_DEVICE_PATH_PROTOCOL **GopDevicePath\r
   )\r
{\r
  UINTN                           Index;\r
  EFI_STATUS                      Status;\r
  EFI_HANDLE                      PciDeviceHandle;\r
  EFI_DEVICE_PATH_PROTOCOL        *TempDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL        *TempPciDevicePath;\r
  UINTN                           GopHandleCount;\r
  EFI_HANDLE                      *GopHandleBuffer;\r
\r
  if (PciDevicePath == NULL || GopDevicePath == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Initialize the GopDevicePath to be PciDevicePath\r
  //\r
  *GopDevicePath    = PciDevicePath;\r
  TempPciDevicePath = PciDevicePath;\r
\r
  Status = gBS->LocateDevicePath (\r
                  &gEfiDevicePathProtocolGuid,\r
                  &TempPciDevicePath,\r
                  &PciDeviceHandle\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Try to connect this handle, so that GOP driver could start on this\r
  // device and create child handles with GraphicsOutput Protocol installed\r
  // on them, then we get device paths of these child handles and select\r
  // them as possible console device.\r
  //\r
  gBS->ConnectController (PciDeviceHandle, NULL, NULL, FALSE);\r
\r
  Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  &gEfiGraphicsOutputProtocolGuid,\r
                  NULL,\r
                  &GopHandleCount,\r
                  &GopHandleBuffer\r
                  );\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Add all the child handles as possible Console Device\r
    //\r
    for (Index = 0; Index < GopHandleCount; Index++) {\r
      Status = gBS->HandleProtocol (GopHandleBuffer[Index],\r
                      &gEfiDevicePathProtocolGuid, (VOID*)&TempDevicePath);\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
      if (CompareMem (\r
            PciDevicePath,\r
            TempDevicePath,\r
            GetDevicePathSize (PciDevicePath) - END_DEVICE_PATH_LENGTH\r
            ) == 0) {\r
        //\r
        // In current implementation, we only enable one of the child handles\r
        // as console device, i.e. sotre one of the child handle's device\r
        // path to variable "ConOut"\r
        // In future, we could select all child handles to be console device\r
        //\r
\r
        *GopDevicePath = TempDevicePath;\r
\r
        //\r
        // Delete the PCI device's path that added by\r
        // GetPlugInPciVgaDevicePath(). Add the integrity GOP device path.\r
        //\r
        EfiBootManagerUpdateConsoleVariable (ConOutDev, NULL, PciDevicePath);\r
        EfiBootManagerUpdateConsoleVariable (ConOutDev, TempDevicePath, NULL);\r
      }\r
    }\r
    gBS->FreePool (GopHandleBuffer);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
PreparePciDisplayDevicePath (\r
  IN EFI_HANDLE                DeviceHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Add PCI VGA to ConOut.\r
  PCI VGA: 03 00 00\r
\r
Arguments:\r
\r
  DeviceHandle            - Handle of PCIIO protocol.\r
\r
Returns:\r
\r
  EFI_SUCCESS             - PCI VGA is added to ConOut.\r
  EFI_STATUS              - No PCI VGA device is added.\r
\r
--*/\r
{\r
  EFI_STATUS                Status;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *GopDevicePath;\r
\r
  DevicePath    = NULL;\r
  GopDevicePath = NULL;\r
  Status = gBS->HandleProtocol (\r
                  DeviceHandle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  (VOID*)&DevicePath\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  GetGopDevicePath (DevicePath, &GopDevicePath);\r
  DevicePath = GopDevicePath;\r
\r
  EfiBootManagerUpdateConsoleVariable (ConOut, DevicePath, NULL);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
PreparePciSerialDevicePath (\r
  IN EFI_HANDLE                DeviceHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Add PCI Serial to ConOut, ConIn, ErrOut.\r
  PCI Serial: 07 00 02\r
\r
Arguments:\r
\r
  DeviceHandle            - Handle of PCIIO protocol.\r
\r
Returns:\r
\r
  EFI_SUCCESS             - PCI Serial is added to ConOut, ConIn, and ErrOut.\r
  EFI_STATUS              - No PCI Serial device is added.\r
\r
--*/\r
{\r
  EFI_STATUS                Status;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
\r
  DevicePath = NULL;\r
  Status = gBS->HandleProtocol (\r
                  DeviceHandle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  (VOID*)&DevicePath\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gUartDeviceNode);\r
  DevicePath = AppendDevicePathNode (DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&gTerminalTypeDeviceNode);\r
\r
  EfiBootManagerUpdateConsoleVariable (ConOut, DevicePath, NULL);\r
  EfiBootManagerUpdateConsoleVariable (ConIn, DevicePath, NULL);\r
  EfiBootManagerUpdateConsoleVariable (ErrOut, DevicePath, NULL);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
VisitAllInstancesOfProtocol (\r
  IN EFI_GUID                    *Id,\r
  IN PROTOCOL_INSTANCE_CALLBACK  CallBackFunction,\r
  IN VOID                        *Context\r
  )\r
{\r
  EFI_STATUS                Status;\r
  UINTN                     HandleCount;\r
  EFI_HANDLE                *HandleBuffer;\r
  UINTN                     Index;\r
  VOID                      *Instance;\r
\r
  //\r
  // Start to check all the PciIo to find all possible device\r
  //\r
  HandleCount = 0;\r
  HandleBuffer = NULL;\r
  Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  Id,\r
                  NULL,\r
                  &HandleCount,\r
                  &HandleBuffer\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  for (Index = 0; Index < HandleCount; Index++) {\r
    Status = gBS->HandleProtocol (HandleBuffer[Index], Id, &Instance);\r
    if (EFI_ERROR (Status)) {\r
      continue;\r
    }\r
\r
    Status = (*CallBackFunction) (\r
               HandleBuffer[Index],\r
               Instance,\r
               Context\r
               );\r
  }\r
\r
  gBS->FreePool (HandleBuffer);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
VisitingAPciInstance (\r
  IN EFI_HANDLE  Handle,\r
  IN VOID        *Instance,\r
  IN VOID        *Context\r
  )\r
{\r
  EFI_STATUS                Status;\r
  EFI_PCI_IO_PROTOCOL       *PciIo;\r
  PCI_TYPE00                Pci;\r
\r
  PciIo = (EFI_PCI_IO_PROTOCOL*) Instance;\r
\r
  //\r
  // Check for all PCI device\r
  //\r
  Status = PciIo->Pci.Read (\r
                    PciIo,\r
                    EfiPciIoWidthUint32,\r
                    0,\r
                    sizeof (Pci) / sizeof (UINT32),\r
                    &Pci\r
                    );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  return (*(VISIT_PCI_INSTANCE_CALLBACK)(UINTN) Context) (\r
           Handle,\r
           PciIo,\r
           &Pci\r
           );\r
\r
}\r
\r
\r
\r
EFI_STATUS\r
VisitAllPciInstances (\r
  IN VISIT_PCI_INSTANCE_CALLBACK CallBackFunction\r
  )\r
{\r
  return VisitAllInstancesOfProtocol (\r
           &gEfiPciIoProtocolGuid,\r
           VisitingAPciInstance,\r
           (VOID*)(UINTN) CallBackFunction\r
           );\r
}\r
\r
\r
/**\r
  Do platform specific PCI Device check and add them to\r
  ConOut, ConIn, ErrOut.\r
\r
  @param[in]  Handle - Handle of PCI device instance\r
  @param[in]  PciIo - PCI IO protocol instance\r
  @param[in]  Pci - PCI Header register block\r
\r
  @retval EFI_SUCCESS - PCI Device check and Console variable update\r
                        successfully.\r
  @retval EFI_STATUS - PCI Device check or Console variable update fail.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DetectAndPreparePlatformPciDevicePath (\r
  IN EFI_HANDLE           Handle,\r
  IN EFI_PCI_IO_PROTOCOL  *PciIo,\r
  IN PCI_TYPE00           *Pci\r
  )\r
{\r
  EFI_STATUS                Status;\r
\r
  Status = PciIo->Attributes (\r
    PciIo,\r
    EfiPciIoAttributeOperationEnable,\r
    EFI_PCI_DEVICE_ENABLE,\r
    NULL\r
    );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  if (!mDetectVgaOnly) {\r
    //\r
    // Here we decide whether it is LPC Bridge\r
    //\r
    if ((IS_PCI_LPC (Pci)) ||\r
        ((IS_PCI_ISA_PDECODE (Pci)) &&\r
         (Pci->Hdr.VendorId == 0x8086) &&\r
         (Pci->Hdr.DeviceId == 0x7000)\r
        )\r
       ) {\r
      //\r
      // Add IsaKeyboard to ConIn,\r
      // add IsaSerial to ConOut, ConIn, ErrOut\r
      //\r
      DEBUG ((EFI_D_INFO, "Found LPC Bridge device\n"));\r
      PrepareLpcBridgeDevicePath (Handle);\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // Here we decide which Serial device to enable in PCI bus\r
    //\r
    if (IS_PCI_16550SERIAL (Pci)) {\r
      //\r
      // Add them to ConOut, ConIn, ErrOut.\r
      //\r
      DEBUG ((EFI_D_INFO, "Found PCI 16550 SERIAL device\n"));\r
      PreparePciSerialDevicePath (Handle);\r
      return EFI_SUCCESS;\r
    }\r
  }\r
\r
  //\r
  // Here we decide which display device to enable in PCI bus\r
  //\r
  if (IS_PCI_DISPLAY (Pci)) {\r
    //\r
    // Add them to ConOut.\r
    //\r
    DEBUG ((EFI_D_INFO, "Found PCI display device\n"));\r
    PreparePciDisplayDevicePath (Handle);\r
    return EFI_SUCCESS;\r
  }\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  Do platform specific PCI Device check and add them to ConOut, ConIn, ErrOut\r
\r
  @param[in]  DetectVgaOnly - Only detect VGA device if it's TRUE.\r
\r
  @retval EFI_SUCCESS - PCI Device check and Console variable update\r
                        successfully.\r
  @retval EFI_STATUS - PCI Device check or Console variable update fail.\r
\r
**/\r
EFI_STATUS\r
DetectAndPreparePlatformPciDevicePaths (\r
  BOOLEAN DetectVgaOnly\r
  )\r
{\r
  mDetectVgaOnly = DetectVgaOnly;\r
  return VisitAllPciInstances (DetectAndPreparePlatformPciDevicePath);\r
}\r
\r
\r
VOID\r
PlatformInitializeConsole (\r
  IN PLATFORM_CONSOLE_CONNECT_ENTRY   *PlatformConsole\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Connect the predefined platform default console device. Always try to find\r
  and enable the vga device if have.\r
\r
Arguments:\r
\r
  PlatformConsole         - Predefined platform default console device array.\r
--*/\r
{\r
  UINTN                              Index;\r
  EFI_DEVICE_PATH_PROTOCOL           *VarConout;\r
  EFI_DEVICE_PATH_PROTOCOL           *VarConin;\r
\r
  //\r
  // Connect RootBridge\r
  //\r
  GetEfiGlobalVariable2 (EFI_CON_OUT_VARIABLE_NAME, (VOID **) &VarConout,\r
    NULL);\r
  GetEfiGlobalVariable2 (EFI_CON_IN_VARIABLE_NAME, (VOID **) &VarConin, NULL);\r
\r
  if (VarConout == NULL || VarConin == NULL) {\r
    //\r
    // Do platform specific PCI Device check and add them to ConOut, ConIn,\r
    // ErrOut\r
    //\r
    DetectAndPreparePlatformPciDevicePaths (FALSE);\r
\r
    //\r
    // Have chance to connect the platform default console,\r
    // the platform default console is the minimum device group\r
    // the platform should support\r
    //\r
    for (Index = 0; PlatformConsole[Index].DevicePath != NULL; ++Index) {\r
      //\r
      // Update the console variable with the connect type\r
      //\r
      if ((PlatformConsole[Index].ConnectType & CONSOLE_IN) == CONSOLE_IN) {\r
        EfiBootManagerUpdateConsoleVariable (ConIn,\r
          PlatformConsole[Index].DevicePath, NULL);\r
      }\r
      if ((PlatformConsole[Index].ConnectType & CONSOLE_OUT) == CONSOLE_OUT) {\r
        EfiBootManagerUpdateConsoleVariable (ConOut,\r
          PlatformConsole[Index].DevicePath, NULL);\r
      }\r
      if ((PlatformConsole[Index].ConnectType & STD_ERROR) == STD_ERROR) {\r
        EfiBootManagerUpdateConsoleVariable (ErrOut,\r
          PlatformConsole[Index].DevicePath, NULL);\r
      }\r
    }\r
  } else {\r
    //\r
    // Only detect VGA device and add them to ConOut\r
    //\r
    DetectAndPreparePlatformPciDevicePaths (TRUE);\r
  }\r
}\r
\r
\r
/**\r
  Configure PCI Interrupt Line register for applicable devices\r
  Ported from SeaBIOS, src/fw/pciinit.c, *_pci_slot_get_irq()\r
\r
  @param[in]  Handle - Handle of PCI device instance\r
  @param[in]  PciIo - PCI IO protocol instance\r
  @param[in]  PciHdr - PCI Header register block\r
\r
  @retval EFI_SUCCESS - PCI Interrupt Line register configured successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SetPciIntLine (\r
  IN EFI_HANDLE           Handle,\r
  IN EFI_PCI_IO_PROTOCOL  *PciIo,\r
  IN PCI_TYPE00           *PciHdr\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL  *DevPathNode;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevPath;\r
  UINTN                     RootSlot;\r
  UINTN                     Idx;\r
  UINT8                     IrqLine;\r
  EFI_STATUS                Status;\r
  UINT32                    RootBusNumber;\r
\r
  Status = EFI_SUCCESS;\r
\r
  if (PciHdr->Device.InterruptPin != 0) {\r
\r
    DevPathNode = DevicePathFromHandle (Handle);\r
    ASSERT (DevPathNode != NULL);\r
    DevPath = DevPathNode;\r
\r
    RootBusNumber = 0;\r
    if (DevicePathType (DevPathNode) == ACPI_DEVICE_PATH &&\r
        DevicePathSubType (DevPathNode) == ACPI_DP &&\r
        ((ACPI_HID_DEVICE_PATH *)DevPathNode)->HID == EISA_PNP_ID(0x0A03)) {\r
      RootBusNumber = ((ACPI_HID_DEVICE_PATH *)DevPathNode)->UID;\r
    }\r
\r
    //\r
    // Compute index into PciHostIrqs[] table by walking\r
    // the device path and adding up all device numbers\r
    //\r
    Status = EFI_NOT_FOUND;\r
    RootSlot = 0;\r
    Idx = PciHdr->Device.InterruptPin - 1;\r
    while (!IsDevicePathEnd (DevPathNode)) {\r
      if (DevicePathType (DevPathNode) == HARDWARE_DEVICE_PATH &&\r
          DevicePathSubType (DevPathNode) == HW_PCI_DP) {\r
\r
        Idx += ((PCI_DEVICE_PATH *)DevPathNode)->Device;\r
\r
        //\r
        // Unlike SeaBIOS, which starts climbing from the leaf device\r
        // up toward the root, we traverse the device path starting at\r
        // the root moving toward the leaf node.\r
        // The slot number of the top-level parent bridge is needed for\r
        // Q35 cases with more than 24 slots on the root bus.\r
        //\r
        if (Status != EFI_SUCCESS) {\r
          Status = EFI_SUCCESS;\r
          RootSlot = ((PCI_DEVICE_PATH *)DevPathNode)->Device;\r
        }\r
      }\r
\r
      DevPathNode = NextDevicePathNode (DevPathNode);\r
    }\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    if (RootBusNumber == 0 && RootSlot == 0) {\r
      DEBUG((\r
        EFI_D_ERROR,\r
        "%a: PCI host bridge (00:00.0) should have no interrupts!\n",\r
        __FUNCTION__\r
        ));\r
      ASSERT (FALSE);\r
    }\r
\r
    //\r
    // Final PciHostIrqs[] index calculation depends on the platform\r
    // and should match SeaBIOS src/fw/pciinit.c *_pci_slot_get_irq()\r
    //\r
    switch (mHostBridgeDevId) {\r
      case INTEL_82441_DEVICE_ID:\r
        Idx -= 1;\r
        break;\r
      case INTEL_Q35_MCH_DEVICE_ID:\r
        //\r
        // SeaBIOS contains the following comment:\r
        // "Slots 0-24 rotate slot:pin mapping similar to piix above, but\r
        //  with a different starting index - see q35-acpi-dsdt.dsl.\r
        //\r
        //  Slots 25-31 all use LNKA mapping (or LNKE, but A:D = E:H)"\r
        //\r
        if (RootSlot > 24) {\r
          //\r
          // in this case, subtract back out RootSlot from Idx\r
          // (SeaBIOS never adds it to begin with, but that would make our\r
          //  device path traversal loop above too awkward)\r
          //\r
          Idx -= RootSlot;\r
        }\r
        break;\r
      default:\r
        ASSERT (FALSE); // should never get here\r
    }\r
    Idx %= ARRAY_SIZE (PciHostIrqs);\r
    IrqLine = PciHostIrqs[Idx];\r
\r
    DEBUG_CODE_BEGIN ();\r
    {\r
      CHAR16        *DevPathString;\r
      STATIC CHAR16 Fallback[] = L"<failed to convert>";\r
      UINTN         Segment, Bus, Device, Function;\r
\r
      DevPathString = ConvertDevicePathToText (DevPath, FALSE, FALSE);\r
      if (DevPathString == NULL) {\r
        DevPathString = Fallback;\r
      }\r
      Status = PciIo->GetLocation (PciIo, &Segment, &Bus, &Device, &Function);\r
      ASSERT_EFI_ERROR (Status);\r
\r
      DEBUG ((EFI_D_VERBOSE, "%a: [%02x:%02x.%x] %s -> 0x%02x\n", __FUNCTION__,\r
        (UINT32)Bus, (UINT32)Device, (UINT32)Function, DevPathString,\r
        IrqLine));\r
\r
      if (DevPathString != Fallback) {\r
        FreePool (DevPathString);\r
      }\r
    }\r
    DEBUG_CODE_END ();\r
\r
    //\r
    // Set PCI Interrupt Line register for this device to PciHostIrqs[Idx]\r
    //\r
    Status = PciIo->Pci.Write (\r
               PciIo,\r
               EfiPciIoWidthUint8,\r
               PCI_INT_LINE_OFFSET,\r
               1,\r
               &IrqLine\r
               );\r
  }\r
\r
  return Status;\r
}\r
\r
\r
VOID\r
PciAcpiInitialization (\r
  )\r
{\r
  UINTN  Pmba;\r
\r
  //\r
  // Query Host Bridge DID to determine platform type\r
  //\r
  mHostBridgeDevId = PcdGet16 (PcdOvmfHostBridgePciDevId);\r
  switch (mHostBridgeDevId) {\r
    case INTEL_82441_DEVICE_ID:\r
      Pmba = POWER_MGMT_REGISTER_PIIX4 (PIIX4_PMBA);\r
      //\r
      // 00:01.0 ISA Bridge (PIIX4) LNK routing targets\r
      //\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 1, 0, 0x60), 0x0b); // A\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 1, 0, 0x61), 0x0b); // B\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 1, 0, 0x62), 0x0a); // C\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 1, 0, 0x63), 0x0a); // D\r
      break;\r
    case INTEL_Q35_MCH_DEVICE_ID:\r
      Pmba = POWER_MGMT_REGISTER_Q35 (ICH9_PMBASE);\r
      //\r
      // 00:1f.0 LPC Bridge (Q35) LNK routing targets\r
      //\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x60), 0x0a); // A\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x61), 0x0a); // B\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x62), 0x0b); // C\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x63), 0x0b); // D\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x68), 0x0a); // E\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x69), 0x0a); // F\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x6a), 0x0b); // G\r
      PciWrite8 (PCI_LIB_ADDRESS (0, 0x1f, 0, 0x6b), 0x0b); // H\r
      break;\r
    default:\r
      DEBUG ((EFI_D_ERROR, "%a: Unknown Host Bridge Device ID: 0x%04x\n",\r
        __FUNCTION__, mHostBridgeDevId));\r
      ASSERT (FALSE);\r
      return;\r
  }\r
\r
  //\r
  // Initialize PCI_INTERRUPT_LINE for applicable present PCI devices\r
  //\r
  VisitAllPciInstances (SetPciIntLine);\r
\r
  //\r
  // Set ACPI SCI_EN bit in PMCNTRL\r
  //\r
  IoOr16 ((PciRead32 (Pmba) & ~BIT0) + 4, BIT0);\r
}\r
\r
/**\r
  This function detects if OVMF is running on Xen.\r
\r
**/\r
STATIC\r
BOOLEAN\r
XenDetected (\r
  VOID\r
  )\r
{\r
  EFI_HOB_GUID_TYPE         *GuidHob;\r
  STATIC INTN               FoundHob = -1;\r
\r
  if (FoundHob == 0) {\r
    return FALSE;\r
  } else if (FoundHob == 1) {\r
    return TRUE;\r
  }\r
\r
  //\r
  // See if a XenInfo HOB is available\r
  //\r
  GuidHob = GetFirstGuidHob (&gEfiXenInfoGuid);\r
  if (GuidHob == NULL) {\r
    FoundHob = 0;\r
    return FALSE;\r
  }\r
\r
  FoundHob = 1;\r
  return TRUE;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
ConnectRecursivelyIfPciMassStorage (\r
  IN EFI_HANDLE           Handle,\r
  IN EFI_PCI_IO_PROTOCOL  *Instance,\r
  IN PCI_TYPE00           *PciHeader\r
  )\r
{\r
  EFI_STATUS                Status;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  CHAR16                    *DevPathStr;\r
\r
  //\r
  // Recognize PCI Mass Storage, and Xen PCI devices\r
  //\r
  if (IS_CLASS1 (PciHeader, PCI_CLASS_MASS_STORAGE) ||\r
      (XenDetected() && IS_CLASS2 (PciHeader, 0xFF, 0x80))) {\r
    DevicePath = NULL;\r
    Status = gBS->HandleProtocol (\r
                    Handle,\r
                    &gEfiDevicePathProtocolGuid,\r
                    (VOID*)&DevicePath\r
                    );\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    // Print Device Path\r
    //\r
    DevPathStr = ConvertDevicePathToText (DevicePath, FALSE, FALSE);\r
    if (DevPathStr != NULL) {\r
      DEBUG((\r
        EFI_D_INFO,\r
        "Found %s device: %s\n",\r
        (IS_CLASS1 (PciHeader, PCI_CLASS_MASS_STORAGE) ?\r
         L"Mass Storage" :\r
         L"Xen"\r
         ),\r
        DevPathStr\r
        ));\r
      FreePool(DevPathStr);\r
    }\r
\r
    Status = gBS->ConnectController (Handle, NULL, NULL, TRUE);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  This notification function is invoked when the\r
  EMU Variable FVB has been changed.\r
\r
  @param  Event                 The event that occurred\r
  @param  Context               For EFI compatibility.  Not used.\r
\r
**/\r
VOID\r
EFIAPI\r
EmuVariablesUpdatedCallback (\r
  IN  EFI_EVENT Event,\r
  IN  VOID      *Context\r
  )\r
{\r
  DEBUG ((EFI_D_INFO, "EmuVariablesUpdatedCallback\n"));\r
  UpdateNvVarsOnFileSystem ();\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
VisitingFileSystemInstance (\r
  IN EFI_HANDLE  Handle,\r
  IN VOID        *Instance,\r
  IN VOID        *Context\r
  )\r
{\r
  EFI_STATUS      Status;\r
  STATIC BOOLEAN  ConnectedToFileSystem = FALSE;\r
  RETURN_STATUS   PcdStatus;\r
\r
  if (ConnectedToFileSystem) {\r
    return EFI_ALREADY_STARTED;\r
  }\r
\r
  Status = ConnectNvVarsToFileSystem (Handle);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  ConnectedToFileSystem = TRUE;\r
  mEmuVariableEvent =\r
    EfiCreateProtocolNotifyEvent (\r
      &gEfiDevicePathProtocolGuid,\r
      TPL_CALLBACK,\r
      EmuVariablesUpdatedCallback,\r
      NULL,\r
      &mEmuVariableEventReg\r
      );\r
  PcdStatus = PcdSet64S (PcdEmuVariableEvent,\r
                (UINT64)(UINTN) mEmuVariableEvent);\r
  ASSERT_RETURN_ERROR (PcdStatus);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
\r
VOID\r
PlatformBdsRestoreNvVarsFromHardDisk (\r
  )\r
{\r
  VisitAllPciInstances (ConnectRecursivelyIfPciMassStorage);\r
  VisitAllInstancesOfProtocol (\r
    &gEfiSimpleFileSystemProtocolGuid,\r
    VisitingFileSystemInstance,\r
    NULL\r
    );\r
\r
}\r
\r
VOID\r
PlatformBdsConnectSequence (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Connect with predefined platform connect sequence,\r
  the OEM/IBV can customize with their own connect sequence.\r
\r
Arguments:\r
\r
  None.\r
\r
Returns:\r
\r
  None.\r
\r
--*/\r
{\r
  UINTN         Index;\r
  RETURN_STATUS Status;\r
\r
  DEBUG ((EFI_D_INFO, "PlatformBdsConnectSequence\n"));\r
\r
  Index = 0;\r
\r
  //\r
  // Here we can get the customized platform connect sequence\r
  // Notes: we can connect with new variable which record the\r
  // last time boots connect device path sequence\r
  //\r
  while (gPlatformConnectSequence[Index] != NULL) {\r
    //\r
    // Build the platform boot option\r
    //\r
    EfiBootManagerConnectDevicePath (gPlatformConnectSequence[Index], NULL);\r
    Index++;\r
  }\r
\r
  Status = ConnectDevicesFromQemu ();\r
  if (RETURN_ERROR (Status)) {\r
    //\r
    // Just use the simple policy to connect all devices\r
    //\r
    DEBUG ((DEBUG_INFO, "EfiBootManagerConnectAll\n"));\r
    EfiBootManagerConnectAll ();\r
  }\r
\r
  PciAcpiInitialization ();\r
}\r
\r
/**\r
  Save the S3 boot script.\r
\r
  Note that DxeSmmReadyToLock must be signaled after this function returns;\r
  otherwise the script wouldn't be saved actually.\r
**/\r
STATIC\r
VOID\r
SaveS3BootScript (\r
  VOID\r
  )\r
{\r
  EFI_STATUS                 Status;\r
  EFI_S3_SAVE_STATE_PROTOCOL *BootScript;\r
  STATIC CONST UINT8         Info[] = { 0xDE, 0xAD, 0xBE, 0xEF };\r
\r
  Status = gBS->LocateProtocol (&gEfiS3SaveStateProtocolGuid, NULL,\r
                  (VOID **) &BootScript);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Despite the opcode documentation in the PI spec, the protocol\r
  // implementation embeds a deep copy of the info in the boot script, rather\r
  // than storing just a pointer to runtime or NVS storage.\r
  //\r
  Status = BootScript->Write(BootScript, EFI_BOOT_SCRIPT_INFORMATION_OPCODE,\r
                         (UINT32) sizeof Info,\r
                         (EFI_PHYSICAL_ADDRESS)(UINTN) &Info);\r
  ASSERT_EFI_ERROR (Status);\r
}\r
\r
\r
VOID\r
EFIAPI\r
PlatformBootManagerAfterConsole (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  The function will execute with as the platform policy, current policy\r
  is driven by boot mode. IBV/OEM can customize this code for their specific\r
  policy action.\r
\r
--*/\r
{\r
  EFI_BOOT_MODE                      BootMode;\r
\r
  DEBUG ((EFI_D_INFO, "PlatformBootManagerAfterConsole\n"));\r
\r
  if (PcdGetBool (PcdOvmfFlashVariablesEnable)) {\r
    DEBUG ((EFI_D_INFO, "PlatformBdsPolicyBehavior: not restoring NvVars "\r
      "from disk since flash variables appear to be supported.\n"));\r
  } else {\r
    //\r
    // Try to restore variables from the hard disk early so\r
    // they can be used for the other BDS connect operations.\r
    //\r
    PlatformBdsRestoreNvVarsFromHardDisk ();\r
  }\r
\r
  //\r
  // Get current Boot Mode\r
  //\r
  BootMode = GetBootModeHob ();\r
  DEBUG ((DEBUG_INFO, "Boot Mode:%x\n", BootMode));\r
\r
  //\r
  // Go the different platform policy with different boot mode\r
  // Notes: this part code can be change with the table policy\r
  //\r
  ASSERT (BootMode == BOOT_WITH_FULL_CONFIGURATION);\r
\r
  //\r
  // Logo show\r
  //\r
  BootLogoEnableLogo ();\r
\r
  //\r
  // Perform some platform specific connect sequence\r
  //\r
  PlatformBdsConnectSequence ();\r
\r
  //\r
  // Process QEMU's -kernel command line option\r
  //\r
  TryRunningQemuKernel ();\r
\r
  EfiBootManagerRefreshAllBootOption ();\r
\r
  //\r
  // Register UEFI Shell\r
  //\r
  PlatformRegisterFvBootOption (\r
    PcdGetPtr (PcdShellFile), L"EFI Internal Shell", LOAD_OPTION_ACTIVE\r
    );\r
\r
  RemoveStaleFvFileOptions ();\r
  SetBootOrderFromQemu ();\r
}\r
\r
/**\r
  This notification function is invoked when an instance of the\r
  EFI_DEVICE_PATH_PROTOCOL is produced.\r
\r
  @param  Event                 The event that occurred\r
  @param  Context               For EFI compatibility.  Not used.\r
\r
**/\r
VOID\r
EFIAPI\r
NotifyDevPath (\r
  IN  EFI_EVENT Event,\r
  IN  VOID      *Context\r
  )\r
{\r
  EFI_HANDLE                            Handle;\r
  EFI_STATUS                            Status;\r
  UINTN                                 BufferSize;\r
  EFI_DEVICE_PATH_PROTOCOL             *DevPathNode;\r
  ATAPI_DEVICE_PATH                    *Atapi;\r
\r
  //\r
  // Examine all new handles\r
  //\r
  for (;;) {\r
    //\r
    // Get the next handle\r
    //\r
    BufferSize = sizeof (Handle);\r
    Status = gBS->LocateHandle (\r
              ByRegisterNotify,\r
              NULL,\r
              mEfiDevPathNotifyReg,\r
              &BufferSize,\r
              &Handle\r
              );\r
\r
    //\r
    // If not found, we're done\r
    //\r
    if (EFI_NOT_FOUND == Status) {\r
      break;\r
    }\r
\r
    if (EFI_ERROR (Status)) {\r
      continue;\r
    }\r
\r
    //\r
    // Get the DevicePath protocol on that handle\r
    //\r
    Status = gBS->HandleProtocol (Handle, &gEfiDevicePathProtocolGuid,\r
                    (VOID **)&DevPathNode);\r
    ASSERT_EFI_ERROR (Status);\r
\r
    while (!IsDevicePathEnd (DevPathNode)) {\r
      //\r
      // Find the handler to dump this device path node\r
      //\r
      if (\r
           (DevicePathType(DevPathNode) == MESSAGING_DEVICE_PATH) &&\r
           (DevicePathSubType(DevPathNode) == MSG_ATAPI_DP)\r
         ) {\r
        Atapi = (ATAPI_DEVICE_PATH*) DevPathNode;\r
        PciOr16 (\r
          PCI_LIB_ADDRESS (\r
            0,\r
            1,\r
            1,\r
            (Atapi->PrimarySecondary == 1) ? 0x42: 0x40\r
            ),\r
          BIT15\r
          );\r
      }\r
\r
      //\r
      // Next device path node\r
      //\r
      DevPathNode = NextDevicePathNode (DevPathNode);\r
    }\r
  }\r
\r
  return;\r
}\r
\r
\r
VOID\r
InstallDevicePathCallback (\r
  VOID\r
  )\r
{\r
  DEBUG ((EFI_D_INFO, "Registered NotifyDevPath Event\n"));\r
  mEfiDevPathEvent = EfiCreateProtocolNotifyEvent (\r
                          &gEfiDevicePathProtocolGuid,\r
                          TPL_CALLBACK,\r
                          NotifyDevPath,\r
                          NULL,\r
                          &mEfiDevPathNotifyReg\r
                          );\r
}\r
\r
/**\r
  This function is called each second during the boot manager waits the\r
  timeout.\r
\r
  @param TimeoutRemain  The remaining timeout.\r
**/\r
VOID\r
EFIAPI\r
PlatformBootManagerWaitCallback (\r
  UINT16          TimeoutRemain\r
  )\r
{\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL_UNION Black;\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL_UNION White;\r
  UINT16                              Timeout;\r
\r
  Timeout = PcdGet16 (PcdPlatformBootTimeOut);\r
\r
  Black.Raw = 0x00000000;\r
  White.Raw = 0x00FFFFFF;\r
\r
  BootLogoUpdateProgress (\r
    White.Pixel,\r
    Black.Pixel,\r
    L"Start boot option",\r
    White.Pixel,\r
    (Timeout - TimeoutRemain) * 100 / Timeout,\r
    0\r
    );\r
}\r
\r