OvmfPkg/IntelTdx: Enable separate-fv in IntelTdx/IntelTdxX64.fdf

[mirror_edk2.git] / OvmfPkg / Library / PeilessStartupLib / DxeLoad.c

/** @file\r
  Responsibility of this file is to load the DXE Core from a Firmware Volume.\r
\r
Copyright (c) 2016 HP Development Company, L.P.\r
Copyright (c) 2006 - 2020, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "PeilessStartupInternal.h"\r
#include <Library/DebugLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/PcdLib.h>\r
#include <Guid/MemoryTypeInformation.h>\r
#include <Guid/MemoryAllocationHob.h>\r
#include <Guid/PcdDataBaseSignatureGuid.h>\r
#include <Register/Intel/Cpuid.h>\r
#include <Library/PrePiLib.h>\r
#include "X64/PageTables.h"\r
#include <Library/ReportStatusCodeLib.h>\r
\r
#define STACK_SIZE  0x20000\r
\r
/**\r
   Transfers control to DxeCore.\r
\r
   This function performs a CPU architecture specific operations to execute\r
   the entry point of DxeCore\r
\r
   @param DxeCoreEntryPoint         The entry point of DxeCore.\r
\r
**/\r
VOID\r
HandOffToDxeCore (\r
  IN EFI_PHYSICAL_ADDRESS  DxeCoreEntryPoint\r
  )\r
{\r
  VOID   *BaseOfStack;\r
  VOID   *TopOfStack;\r
  UINTN  PageTables;\r
\r
  //\r
  // Clear page 0 and mark it as allocated if NULL pointer detection is enabled.\r
  //\r
  if (IsNullDetectionEnabled ()) {\r
    ClearFirst4KPage (GetHobList ());\r
    BuildMemoryAllocationHob (0, EFI_PAGES_TO_SIZE (1), EfiBootServicesData);\r
  }\r
\r
  //\r
  // Allocate 128KB for the Stack\r
  //\r
  BaseOfStack = AllocatePages (EFI_SIZE_TO_PAGES (STACK_SIZE));\r
  ASSERT (BaseOfStack != NULL);\r
\r
  //\r
  // Compute the top of the stack we were allocated. Pre-allocate a UINTN\r
  // for safety.\r
  //\r
  TopOfStack = (VOID *)((UINTN)BaseOfStack + EFI_SIZE_TO_PAGES (STACK_SIZE) * EFI_PAGE_SIZE - CPU_STACK_ALIGNMENT);\r
  TopOfStack = ALIGN_POINTER (TopOfStack, CPU_STACK_ALIGNMENT);\r
\r
  DEBUG ((DEBUG_INFO, "BaseOfStack=0x%x, TopOfStack=0x%x\n", BaseOfStack, TopOfStack));\r
\r
  //\r
  // Create page table and save PageMapLevel4 to CR3\r
  //\r
  PageTables = CreateIdentityMappingPageTables (\r
                 (EFI_PHYSICAL_ADDRESS)(UINTN)BaseOfStack,\r
                 STACK_SIZE\r
                 );\r
  if (PageTables == 0) {\r
    DEBUG ((DEBUG_ERROR, "Failed to create idnetity mapping page tables.\n"));\r
    CpuDeadLoop ();\r
  }\r
\r
  AsmWriteCr3 (PageTables);\r
\r
  //\r
  // Update the contents of BSP stack HOB to reflect the real stack info passed to DxeCore.\r
  //\r
  UpdateStackHob ((EFI_PHYSICAL_ADDRESS)(UINTN)BaseOfStack, STACK_SIZE);\r
\r
  DEBUG ((DEBUG_INFO, "SwitchStack then Jump to DxeCore\n"));\r
  //\r
  // Transfer the control to the entry point of DxeCore.\r
  //\r
  SwitchStack (\r
    (SWITCH_STACK_ENTRY_POINT)(UINTN)DxeCoreEntryPoint,\r
    GetHobList (),\r
    NULL,\r
    TopOfStack\r
    );\r
}\r
\r
/**\r
   Searches DxeCore in all firmware Volumes and loads the first\r
   instance that contains DxeCore.\r
\r
   @return FileHandle of DxeCore to load DxeCore.\r
\r
**/\r
EFI_STATUS\r
FindDxeCore (\r
  IN INTN                         FvInstance,\r
  IN OUT     EFI_PEI_FILE_HANDLE  *FileHandle\r
  )\r
{\r
  EFI_STATUS         Status;\r
  EFI_PEI_FV_HANDLE  VolumeHandle;\r
\r
  if (FileHandle == NULL) {\r
    ASSERT (FALSE);\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  *FileHandle = NULL;\r
\r
  //\r
  // Caller passed in a specific FV to try, so only try that one\r
  //\r
  Status = FfsFindNextVolume (FvInstance, &VolumeHandle);\r
  if (!EFI_ERROR (Status)) {\r
    Status = FfsFindNextFile (EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE, VolumeHandle, FileHandle);\r
    if (*FileHandle) {\r
      // Assume the FV that contains multiple compressed FVs.\r
      // So decompress the compressed FVs\r
      Status = FfsProcessFvFile (*FileHandle);\r
      ASSERT_EFI_ERROR (Status);\r
      Status = FfsAnyFvFindFirstFile (EFI_FV_FILETYPE_DXE_CORE, &VolumeHandle, FileHandle);\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
   This function finds DXE Core in the firmware volume and transfer the control to\r
   DXE core.\r
\r
   @return EFI_SUCCESS              DXE core was successfully loaded.\r
   @return EFI_OUT_OF_RESOURCES     There are not enough resources to load DXE core.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DxeLoadCore (\r
  IN INTN  FvInstance\r
  )\r
{\r
  EFI_STATUS            Status;\r
  EFI_FV_FILE_INFO      DxeCoreFileInfo;\r
  EFI_PHYSICAL_ADDRESS  DxeCoreAddress;\r
  UINT64                DxeCoreSize;\r
  EFI_PHYSICAL_ADDRESS  DxeCoreEntryPoint;\r
  EFI_PEI_FILE_HANDLE   FileHandle;\r
  VOID                  *PeCoffImage;\r
\r
  //\r
  // Look in all the FVs present and find the DXE Core FileHandle\r
  //\r
  Status = FindDxeCore (FvInstance, &FileHandle);\r
\r
  if (EFI_ERROR (Status)) {\r
    ASSERT (FALSE);\r
    return Status;\r
  }\r
\r
  //\r
  // Load the DXE Core from a Firmware Volume.\r
  //\r
  Status = FfsFindSectionData (EFI_SECTION_PE32, FileHandle, &PeCoffImage);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = LoadPeCoffImage (PeCoffImage, &DxeCoreAddress, &DxeCoreSize, &DxeCoreEntryPoint);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Extract the DxeCore GUID file name.\r
  //\r
  Status = FfsGetFileInfo (FileHandle, &DxeCoreFileInfo);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Add HOB for the DXE Core\r
  //\r
  BuildModuleHob (\r
    &DxeCoreFileInfo.FileName,\r
    DxeCoreAddress,\r
    ALIGN_VALUE (DxeCoreSize, EFI_PAGE_SIZE),\r
    DxeCoreEntryPoint\r
    );\r
\r
  DEBUG ((\r
    DEBUG_INFO | DEBUG_LOAD,\r
    "Loading DXE CORE at 0x%11p EntryPoint=0x%11p\n",\r
    (VOID *)(UINTN)DxeCoreAddress,\r
    FUNCTION_ENTRY_POINT (DxeCoreEntryPoint)\r
    ));\r
\r
  // Transfer control to the DXE Core\r
  // The hand off state is simply a pointer to the HOB list\r
  //\r
  HandOffToDxeCore (DxeCoreEntryPoint);\r
\r
  //\r
  // If we get here, then the DXE Core returned.  This is an error\r
  // DxeCore should not return.\r
  //\r
  ASSERT (FALSE);\r
  CpuDeadLoop ();\r
\r
  return EFI_OUT_OF_RESOURCES;\r
}\r