OvmfPkg/AcpiPlatformDxe: Assert if AcpiTable protocol is not found

[mirror_edk2.git] / OvmfPkg / AcpiPlatformDxe / AcpiPlatform.c

/** @file\r
  OVMF ACPI Platform Driver\r
\r
  Copyright (c) 2008 - 2012, Intel Corporation. All rights reserved.<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 "AcpiPlatform.h"\r
\r
EFI_STATUS\r
EFIAPI\r
InstallAcpiTable (\r
  IN   EFI_ACPI_TABLE_PROTOCOL       *AcpiProtocol,\r
  IN   VOID                          *AcpiTableBuffer,\r
  IN   UINTN                         AcpiTableBufferSize,\r
  OUT  UINTN                         *TableKey\r
  )\r
{\r
  return AcpiProtocol->InstallAcpiTable (\r
                         AcpiProtocol,\r
                         AcpiTableBuffer,\r
                         AcpiTableBufferSize,\r
                         TableKey\r
                         );\r
}\r
\r
\r
/**\r
  Locate the first instance of a protocol.  If the protocol requested is an\r
  FV protocol, then it will return the first FV that contains the ACPI table\r
  storage file.\r
\r
  @param  Instance      Return pointer to the first instance of the protocol\r
\r
  @return EFI_SUCCESS           The function completed successfully.\r
  @return EFI_NOT_FOUND         The protocol could not be located.\r
  @return EFI_OUT_OF_RESOURCES  There are not enough resources to find the protocol.\r
\r
**/\r
EFI_STATUS\r
LocateFvInstanceWithTables (\r
  OUT EFI_FIRMWARE_VOLUME2_PROTOCOL **Instance\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  EFI_HANDLE                    *HandleBuffer;\r
  UINTN                         NumberOfHandles;\r
  EFI_FV_FILETYPE               FileType;\r
  UINT32                        FvStatus;\r
  EFI_FV_FILE_ATTRIBUTES        Attributes;\r
  UINTN                         Size;\r
  UINTN                         Index;\r
  EFI_FIRMWARE_VOLUME2_PROTOCOL *FvInstance;\r
\r
  FvStatus = 0;\r
\r
  //\r
  // Locate protocol.\r
  //\r
  Status = gBS->LocateHandleBuffer (\r
                   ByProtocol,\r
                   &gEfiFirmwareVolume2ProtocolGuid,\r
                   NULL,\r
                   &NumberOfHandles,\r
                   &HandleBuffer\r
                   );\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Defined errors at this time are not found and out of resources.\r
    //\r
    return Status;\r
  }\r
\r
  //\r
  // Looking for FV with ACPI storage file\r
  //\r
  for (Index = 0; Index < NumberOfHandles; Index++) {\r
    //\r
    // Get the protocol on this handle\r
    // This should not fail because of LocateHandleBuffer\r
    //\r
    Status = gBS->HandleProtocol (\r
                     HandleBuffer[Index],\r
                     &gEfiFirmwareVolume2ProtocolGuid,\r
                     (VOID**) &FvInstance\r
                     );\r
    ASSERT_EFI_ERROR (Status);\r
\r
    //\r
    // See if it has the ACPI storage file\r
    //\r
    Status = FvInstance->ReadFile (\r
                           FvInstance,\r
                           (EFI_GUID*)PcdGetPtr (PcdAcpiTableStorageFile),\r
                           NULL,\r
                           &Size,\r
                           &FileType,\r
                           &Attributes,\r
                           &FvStatus\r
                           );\r
\r
    //\r
    // If we found it, then we are done\r
    //\r
    if (Status == EFI_SUCCESS) {\r
      *Instance = FvInstance;\r
      break;\r
    }\r
  }\r
\r
  //\r
  // Our exit status is determined by the success of the previous operations\r
  // If the protocol was found, Instance already points to it.\r
  //\r
\r
  //\r
  // Free any allocated buffers\r
  //\r
  gBS->FreePool (HandleBuffer);\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  Find ACPI tables in an FV and parses them. This function is useful for QEMU and KVM.\r
\r
  @param  AcpiTable     Protocol instance pointer    \r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FindAcpiTablesInFv (\r
  IN  EFI_ACPI_TABLE_PROTOCOL     *AcpiTable\r
  )\r
{\r
  EFI_STATUS                           Status;\r
  EFI_FIRMWARE_VOLUME2_PROTOCOL        *FwVol;\r
  INTN                                 Instance;\r
  EFI_ACPI_COMMON_HEADER               *CurrentTable;\r
  UINTN                                TableHandle;\r
  UINT32                               FvStatus;\r
  UINTN                                TableSize;\r
  UINTN                                Size;\r
  EFI_ACPI_TABLE_INSTALL_ACPI_TABLE    TableInstallFunction;\r
\r
  Instance     = 0;\r
  CurrentTable = NULL;\r
  TableHandle  = 0;\r
\r
  if (QemuDetected ()) {\r
    TableInstallFunction = QemuInstallAcpiTable;\r
  } else {\r
    TableInstallFunction = InstallAcpiTable;\r
  }\r
\r
  //\r
  // Locate the firmware volume protocol\r
  //\r
  Status = LocateFvInstanceWithTables (&FwVol);\r
  if (EFI_ERROR (Status)) {\r
    return EFI_ABORTED;\r
  }\r
  //\r
  // Read tables from the storage file.\r
  //\r
  while (Status == EFI_SUCCESS) {\r
\r
    Status = FwVol->ReadSection (\r
                      FwVol,\r
                      (EFI_GUID*)PcdGetPtr (PcdAcpiTableStorageFile),\r
                      EFI_SECTION_RAW,\r
                      Instance,\r
                      (VOID**) &CurrentTable,\r
                      &Size,\r
                      &FvStatus\r
                      );\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // Add the table\r
      //\r
      TableHandle = 0;\r
\r
      TableSize = ((EFI_ACPI_DESCRIPTION_HEADER *) CurrentTable)->Length;\r
      ASSERT (Size >= TableSize);\r
\r
      //\r
      // Install ACPI table\r
      //\r
      Status = TableInstallFunction (\r
                 AcpiTable,\r
                 CurrentTable,\r
                 TableSize,\r
                 &TableHandle\r
                 );\r
\r
      //\r
      // Free memory allocated by ReadSection\r
      //\r
      gBS->FreePool (CurrentTable);\r
\r
      if (EFI_ERROR (Status)) {\r
        return EFI_ABORTED;\r
      }\r
\r
      //\r
      // Increment the instance\r
      //\r
      Instance++;\r
      CurrentTable = NULL;\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Entrypoint of Acpi Platform driver.\r
\r
  @param  ImageHandle\r
  @param  SystemTable\r
\r
  @return EFI_SUCCESS\r
  @return EFI_LOAD_ERROR\r
  @return EFI_OUT_OF_RESOURCES\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
AcpiPlatformEntryPoint (\r
  IN EFI_HANDLE         ImageHandle,\r
  IN EFI_SYSTEM_TABLE   *SystemTable\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  EFI_ACPI_TABLE_PROTOCOL            *AcpiTable;\r
\r
  //\r
  // Find the AcpiTable protocol\r
  //\r
  Status = gBS->LocateProtocol (\r
                  &gEfiAcpiTableProtocolGuid,\r
                  NULL,\r
                  (VOID**)&AcpiTable\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  if (XenDetected ()) {\r
    Status = InstallXenTables (AcpiTable);\r
  } else {\r
    Status = InstallAllQemuLinkedTables (AcpiTable);\r
  }\r
\r
  if (EFI_ERROR (Status)) {\r
    Status = FindAcpiTablesInFv (AcpiTable);\r
  }\r
\r
  return Status;\r
}\r
\r