UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm

[mirror_edk2.git] / MdeModulePkg / Core / RuntimeDxe / Runtime.c

/** @file\r
  This file implements Runtime Architectural Protocol as defined in the\r
  Platform Initialization specification 1.0 VOLUME 2 DXE Core Interface.\r
\r
  This code is used to produce the EFI runtime virtual switch over\r
\r
  THIS IS VERY DANGEROUS CODE BE VERY CAREFUL IF YOU CHANGE IT\r
\r
  The transition for calling EFI Runtime functions in physical mode to calling\r
  them in virtual mode is very very complex. Every pointer in needs to be\r
  converted from physical mode to virtual mode. Be very careful walking linked\r
  lists! Then to make it really hard the code it's self needs be relocated into\r
  the new virtual address space.\r
\r
  So here is the concept. The code in this module will never ever be called in\r
  virtual mode. This is the code that collects the information needed to convert\r
  to virtual mode (DXE core registers runtime stuff with this code). Since this\r
  code is used to fix up all runtime images, it CAN NOT fix it's self up. So some\r
  code has to stay behind and that is us.\r
\r
  Also you need to be careful about when you allocate memory, as once we are in\r
  runtime (including our EVT_SIGNAL_EXIT_BOOT_SERVICES event) you can no longer\r
  allocate memory.\r
\r
  Any runtime driver that gets loaded before us will not be callable in virtual\r
  mode. This is due to the fact that the DXE core can not register the info\r
  needed with us. This is good, since it keeps the code in this file from\r
  getting registered.\r
\r
\r
Revision History:\r
\r
  - Move the CalculateCrc32 function from Runtime Arch Protocol to Boot Service.\r
  Runtime Arch Protocol definition no longer contains CalculateCrc32. Boot Service\r
  Table now contains an item named CalculateCrc32.\r
\r
\r
Copyright (c) 2006 - 2015, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "Runtime.h"\r
\r
//\r
// Global Variables\r
//\r
EFI_MEMORY_DESCRIPTOR  *mVirtualMap = NULL;\r
UINTN                  mVirtualMapDescriptorSize;\r
UINTN                  mVirtualMapMaxIndex;\r
VOID                   *mMyImageBase;\r
\r
//\r
// The handle onto which the Runtime Architectural Protocol instance is installed\r
//\r
EFI_HANDLE  mRuntimeHandle = NULL;\r
\r
//\r
// The Runtime Architectural Protocol instance produced by this driver\r
//\r
EFI_RUNTIME_ARCH_PROTOCOL  mRuntime = {\r
  INITIALIZE_LIST_HEAD_VARIABLE (mRuntime.ImageHead),\r
  INITIALIZE_LIST_HEAD_VARIABLE (mRuntime.EventHead),\r
\r
  //\r
  // Make sure Size != sizeof (EFI_MEMORY_DESCRIPTOR). This will\r
  // prevent people from having pointer math bugs in their code.\r
  // now you have to use *DescriptorSize to make things work.\r
  //\r
  sizeof (EFI_MEMORY_DESCRIPTOR) + sizeof (UINT64) - (sizeof (EFI_MEMORY_DESCRIPTOR) % sizeof (UINT64)),\r
  EFI_MEMORY_DESCRIPTOR_VERSION,\r
  0,\r
  NULL,\r
  NULL,\r
  FALSE,\r
  FALSE\r
};\r
\r
//\r
// Worker Functions\r
//\r
\r
/**\r
\r
  Calculate the 32-bit CRC in a EFI table using the Runtime Drivers\r
  internal function.  The EFI Boot Services Table can not be used because\r
  the EFI Boot Services Table was destroyed at ExitBootServices().\r
  This is a internal function.\r
\r
\r
  @param Hdr             Pointer to an EFI standard header\r
\r
**/\r
VOID\r
RuntimeDriverCalculateEfiHdrCrc (\r
  IN OUT EFI_TABLE_HEADER  *Hdr\r
  )\r
{\r
  UINT32  Crc;\r
\r
  Hdr->CRC32 = 0;\r
\r
  Crc = 0;\r
  RuntimeDriverCalculateCrc32 ((UINT8 *)Hdr, Hdr->HeaderSize, &Crc);\r
  Hdr->CRC32 = Crc;\r
}\r
\r
/**\r
\r
  Determines the new virtual address that is to be used on subsequent memory accesses.\r
\r
\r
  @param DebugDisposition Supplies type information for the pointer being converted.\r
  @param ConvertAddress  A pointer to a pointer that is to be fixed to be the value needed\r
                         for the new virtual address mappings being applied.\r
\r
  @retval  EFI_SUCCESS              The pointer pointed to by Address was modified.\r
  @retval  EFI_NOT_FOUND            The pointer pointed to by Address was not found to be part\r
                                    of the current memory map. This is normally fatal.\r
  @retval  EFI_INVALID_PARAMETER    One of the parameters has an invalid value.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RuntimeDriverConvertPointer (\r
  IN     UINTN  DebugDisposition,\r
  IN OUT VOID   **ConvertAddress\r
  )\r
{\r
  UINTN                  Address;\r
  UINT64                 VirtEndOfRange;\r
  EFI_MEMORY_DESCRIPTOR  *VirtEntry;\r
  UINTN                  Index;\r
\r
  //\r
  // Make sure ConvertAddress is a valid pointer\r
  //\r
  if (ConvertAddress == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Get the address to convert\r
  //\r
  Address = (UINTN)*ConvertAddress;\r
\r
  //\r
  // If this is a null pointer, return if it's allowed\r
  //\r
  if (Address == 0) {\r
    if ((DebugDisposition & EFI_OPTIONAL_PTR) != 0) {\r
      return EFI_SUCCESS;\r
    }\r
\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  VirtEntry = mVirtualMap;\r
  for (Index = 0; Index < mVirtualMapMaxIndex; Index++) {\r
    //\r
    //  To prevent the inclusion of 64-bit math functions a UINTN was placed in\r
    //  front of VirtEntry->NumberOfPages to cast it to a 32-bit thing on IA-32\r
    //  platforms. If you get this ASSERT remove the UINTN and do a 64-bit\r
    //  multiply.\r
    //\r
    ASSERT (((UINTN)VirtEntry->NumberOfPages < 0xffffffff) || (sizeof (UINTN) > 4));\r
\r
    if ((VirtEntry->Attribute & EFI_MEMORY_RUNTIME) == EFI_MEMORY_RUNTIME) {\r
      if (Address >= VirtEntry->PhysicalStart) {\r
        VirtEndOfRange = VirtEntry->PhysicalStart + (((UINTN)VirtEntry->NumberOfPages) * EFI_PAGE_SIZE);\r
        if (Address < VirtEndOfRange) {\r
          //\r
          // Compute new address\r
          //\r
          *ConvertAddress = (VOID *)(Address - (UINTN)VirtEntry->PhysicalStart + (UINTN)VirtEntry->VirtualStart);\r
          return EFI_SUCCESS;\r
        }\r
      }\r
    }\r
\r
    VirtEntry = NEXT_MEMORY_DESCRIPTOR (VirtEntry, mVirtualMapDescriptorSize);\r
  }\r
\r
  return EFI_NOT_FOUND;\r
}\r
\r
/**\r
\r
  Determines the new virtual address that is to be used on subsequent memory accesses\r
  for internal pointers.\r
  This is a internal function.\r
\r
\r
  @param ConvertAddress  A pointer to a pointer that is to be fixed to be the value needed\r
                         for the new virtual address mappings being applied.\r
\r
  @retval  EFI_SUCCESS              The pointer pointed to by Address was modified.\r
  @retval  EFI_NOT_FOUND            The pointer pointed to by Address was not found to be part\r
                                    of the current memory map. This is normally fatal.\r
  @retval  EFI_INVALID_PARAMETER    One of the parameters has an invalid value.\r
\r
**/\r
EFI_STATUS\r
RuntimeDriverConvertInternalPointer (\r
  IN OUT VOID  **ConvertAddress\r
  )\r
{\r
  return RuntimeDriverConvertPointer (0x0, ConvertAddress);\r
}\r
\r
/**\r
\r
  Changes the runtime addressing mode of EFI firmware from physical to virtual.\r
\r
\r
  @param MemoryMapSize   The size in bytes of VirtualMap.\r
  @param DescriptorSize  The size in bytes of an entry in the VirtualMap.\r
  @param DescriptorVersion The version of the structure entries in VirtualMap.\r
  @param VirtualMap      An array of memory descriptors which contain new virtual\r
                         address mapping information for all runtime ranges.\r
\r
  @retval  EFI_SUCCESS            The virtual address map has been applied.\r
  @retval  EFI_UNSUPPORTED        EFI firmware is not at runtime, or the EFI firmware is already in\r
                                  virtual address mapped mode.\r
  @retval  EFI_INVALID_PARAMETER  DescriptorSize or DescriptorVersion is invalid.\r
  @retval  EFI_NO_MAPPING         A virtual address was not supplied for a range in the memory\r
                                  map that requires a mapping.\r
  @retval  EFI_NOT_FOUND          A virtual address was supplied for an address that is not found\r
                                  in the memory map.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RuntimeDriverSetVirtualAddressMap (\r
  IN UINTN                  MemoryMapSize,\r
  IN UINTN                  DescriptorSize,\r
  IN UINT32                 DescriptorVersion,\r
  IN EFI_MEMORY_DESCRIPTOR  *VirtualMap\r
  )\r
{\r
  EFI_STATUS               Status;\r
  EFI_RUNTIME_EVENT_ENTRY  *RuntimeEvent;\r
  EFI_RUNTIME_IMAGE_ENTRY  *RuntimeImage;\r
  LIST_ENTRY               *Link;\r
  EFI_PHYSICAL_ADDRESS     VirtImageBase;\r
\r
  //\r
  // Can only switch to virtual addresses once the memory map is locked down,\r
  // and can only set it once\r
  //\r
  if (!mRuntime.AtRuntime || mRuntime.VirtualMode) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  //\r
  // Only understand the original descriptor format\r
  //\r
  if ((DescriptorVersion != EFI_MEMORY_DESCRIPTOR_VERSION) || (DescriptorSize < sizeof (EFI_MEMORY_DESCRIPTOR))) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // We are now committed to go to virtual mode, so lets get to it!\r
  //\r
  mRuntime.VirtualMode = TRUE;\r
\r
  //\r
  // ConvertPointer() needs this mVirtualMap to do the conversion. So set up\r
  // globals we need to parse the virtual address map.\r
  //\r
  mVirtualMapDescriptorSize = DescriptorSize;\r
  mVirtualMapMaxIndex       = MemoryMapSize / DescriptorSize;\r
  mVirtualMap               = VirtualMap;\r
\r
  //\r
  // ReporstStatusCodeLib will check and make sure this service can be called in runtime mode.\r
  //\r
  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_EFI_RUNTIME_SERVICE | EFI_SW_RS_PC_SET_VIRTUAL_ADDRESS_MAP));\r
\r
  //\r
  // Report Status Code here since EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE event will be signalled.\r
  //\r
  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_DXE_BS_PC_VIRTUAL_ADDRESS_CHANGE_EVENT));\r
\r
  //\r
  // Signal all the EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE events.\r
  // All runtime events are stored in a list in Runtime AP.\r
  //\r
  for (Link = mRuntime.EventHead.ForwardLink; Link != &mRuntime.EventHead; Link = Link->ForwardLink) {\r
    RuntimeEvent = BASE_CR (Link, EFI_RUNTIME_EVENT_ENTRY, Link);\r
    if ((RuntimeEvent->Type & EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE) == EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE) {\r
      //\r
      // Work around the bug in the Platform Init specification (v1.7),\r
      // reported as Mantis#2017: "EFI_RUNTIME_EVENT_ENTRY.Event" should have\r
      // type EFI_EVENT, not (EFI_EVENT*). The PI spec documents the field\r
      // correctly as "The EFI_EVENT returned by CreateEvent()", but the type\r
      // of the field doesn't match the natural language description. Therefore\r
      // we need an explicit cast here.\r
      //\r
      RuntimeEvent->NotifyFunction (\r
                      (EFI_EVENT)RuntimeEvent->Event,\r
                      RuntimeEvent->NotifyContext\r
                      );\r
    }\r
  }\r
\r
  //\r
  // Relocate runtime images. All runtime images are stored in a list in Runtime AP.\r
  //\r
  for (Link = mRuntime.ImageHead.ForwardLink; Link != &mRuntime.ImageHead; Link = Link->ForwardLink) {\r
    RuntimeImage = BASE_CR (Link, EFI_RUNTIME_IMAGE_ENTRY, Link);\r
    //\r
    // We don't want to relocate our selves, as we only run in physical mode.\r
    //\r
    if (mMyImageBase != RuntimeImage->ImageBase) {\r
      VirtImageBase = (EFI_PHYSICAL_ADDRESS)(UINTN)RuntimeImage->ImageBase;\r
      Status        = RuntimeDriverConvertPointer (0, (VOID **)&VirtImageBase);\r
      ASSERT_EFI_ERROR (Status);\r
\r
      PeCoffLoaderRelocateImageForRuntime (\r
        (EFI_PHYSICAL_ADDRESS)(UINTN)RuntimeImage->ImageBase,\r
        VirtImageBase,\r
        (UINTN)RuntimeImage->ImageSize,\r
        RuntimeImage->RelocationData\r
        );\r
\r
      InvalidateInstructionCacheRange (RuntimeImage->ImageBase, (UINTN)RuntimeImage->ImageSize);\r
    }\r
  }\r
\r
  //\r
  // Convert all the Runtime Services except ConvertPointer() and SetVirtualAddressMap()\r
  // and recompute the CRC-32\r
  //\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->GetTime);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->SetTime);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->GetWakeupTime);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->SetWakeupTime);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->ResetSystem);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->GetNextHighMonotonicCount);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->GetVariable);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->SetVariable);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->GetNextVariableName);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->QueryVariableInfo);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->UpdateCapsule);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gRT->QueryCapsuleCapabilities);\r
  RuntimeDriverCalculateEfiHdrCrc (&gRT->Hdr);\r
\r
  //\r
  // UEFI don't require System Configuration Tables Conversion.\r
  //\r
\r
  //\r
  // Convert the runtime fields of the EFI System Table and recompute the CRC-32\r
  //\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gST->FirmwareVendor);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gST->ConfigurationTable);\r
  RuntimeDriverConvertInternalPointer ((VOID **)&gST->RuntimeServices);\r
  RuntimeDriverCalculateEfiHdrCrc (&gST->Hdr);\r
\r
  //\r
  // At this point, gRT and gST are physical pointers, but the contents of these tables\r
  // have been converted to runtime.\r
  //\r
  //\r
  // mVirtualMap is only valid during SetVirtualAddressMap() call\r
  //\r
  mVirtualMap         = NULL;\r
  mVirtualMapMaxIndex = 0;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Entry Point for Runtime driver.\r
\r
  This function installs Runtime Architectural Protocol and registers CalculateCrc32 boot services table,\r
  SetVirtualAddressMap & ConvertPointer runtime services table.\r
\r
  @param ImageHandle     Image handle of this driver.\r
  @param SystemTable     a Pointer to the EFI System Table.\r
\r
  @retval  EFI_SUCEESS  Runtime Driver Architectural Protocol is successfully installed\r
  @return  Others       Some error occurs when installing Runtime Driver Architectural Protocol.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RuntimeDriverInitialize (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS                 Status;\r
  EFI_LOADED_IMAGE_PROTOCOL  *MyLoadedImage;\r
\r
  //\r
  // This image needs to be excluded from relocation for virtual mode, so cache\r
  // a copy of the Loaded Image protocol to test later.\r
  //\r
  Status = gBS->HandleProtocol (\r
                  ImageHandle,\r
                  &gEfiLoadedImageProtocolGuid,\r
                  (VOID **)&MyLoadedImage\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
  mMyImageBase = MyLoadedImage->ImageBase;\r
\r
  //\r
  // Fill in the entries of the EFI Boot Services and EFI Runtime Services Tables\r
  //\r
  gBS->CalculateCrc32       = RuntimeDriverCalculateCrc32;\r
  gRT->SetVirtualAddressMap = RuntimeDriverSetVirtualAddressMap;\r
  gRT->ConvertPointer       = RuntimeDriverConvertPointer;\r
\r
  //\r
  // Install the Runtime Architectural Protocol onto a new handle\r
  //\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &mRuntimeHandle,\r
                  &gEfiRuntimeArchProtocolGuid,\r
                  &mRuntime,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return Status;\r
}\r