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

[mirror_edk2.git] / StandaloneMmPkg / Core / FwVol.c

/** @file\r
    Firmware volume helper interfaces.\r
\r
  Copyright (c) 2015, Intel Corporation. All rights reserved.<BR>\r
  Copyright (c) 2016 - 2021, Arm Limited. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "StandaloneMmCore.h"\r
#include <Library/FvLib.h>\r
#include <Library/ExtractGuidedSectionLib.h>\r
\r
//\r
// List of file types supported by dispatcher\r
//\r
EFI_FV_FILETYPE  mMmFileTypes[] = {\r
  EFI_FV_FILETYPE_MM,\r
  0xE, // EFI_FV_FILETYPE_MM_STANDALONE,\r
       //\r
       // Note: DXE core will process the FV image file, so skip it in MM core\r
       // EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE\r
       //\r
};\r
\r
EFI_STATUS\r
MmAddToDriverList (\r
  IN EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader,\r
  IN VOID                        *Pe32Data,\r
  IN UINTN                       Pe32DataSize,\r
  IN VOID                        *Depex,\r
  IN UINTN                       DepexSize,\r
  IN EFI_GUID                    *DriverName\r
  );\r
\r
BOOLEAN\r
FvHasBeenProcessed (\r
  IN EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader\r
  );\r
\r
VOID\r
FvIsBeingProcessed (\r
  IN EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader\r
  );\r
\r
/**\r
  Given the pointer to the Firmware Volume Header find the\r
  MM driver and return its PE32 image.\r
\r
  @param [in] FwVolHeader   Pointer to memory mapped FV\r
\r
  @retval  EFI_SUCCESS            Success.\r
  @retval  EFI_INVALID_PARAMETER  Invalid parameter.\r
  @retval  EFI_NOT_FOUND          Could not find section data.\r
  @retval  EFI_OUT_OF_RESOURCES   Out of resources.\r
  @retval  EFI_VOLUME_CORRUPTED   Firmware volume is corrupted.\r
  @retval  EFI_UNSUPPORTED        Operation not supported.\r
\r
**/\r
EFI_STATUS\r
MmCoreFfsFindMmDriver (\r
  IN  EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader\r
  )\r
{\r
  EFI_STATUS                  Status;\r
  EFI_STATUS                  DepexStatus;\r
  EFI_FFS_FILE_HEADER         *FileHeader;\r
  EFI_FV_FILETYPE             FileType;\r
  VOID                        *Pe32Data;\r
  UINTN                       Pe32DataSize;\r
  VOID                        *Depex;\r
  UINTN                       DepexSize;\r
  UINTN                       Index;\r
  EFI_COMMON_SECTION_HEADER   *Section;\r
  VOID                        *SectionData;\r
  UINTN                       SectionDataSize;\r
  UINT32                      DstBufferSize;\r
  VOID                        *ScratchBuffer;\r
  UINT32                      ScratchBufferSize;\r
  VOID                        *DstBuffer;\r
  UINT16                      SectionAttribute;\r
  UINT32                      AuthenticationStatus;\r
  EFI_FIRMWARE_VOLUME_HEADER  *InnerFvHeader;\r
\r
  DEBUG ((DEBUG_INFO, "MmCoreFfsFindMmDriver - 0x%x\n", FwVolHeader));\r
\r
  if (FvHasBeenProcessed (FwVolHeader)) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  FvIsBeingProcessed (FwVolHeader);\r
\r
  //\r
  // First check for encapsulated compressed firmware volumes\r
  //\r
  FileHeader = NULL;\r
  do {\r
    Status = FfsFindNextFile (\r
               EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE,\r
               FwVolHeader,\r
               &FileHeader\r
               );\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    Status = FfsFindSectionData (\r
               EFI_SECTION_GUID_DEFINED,\r
               FileHeader,\r
               &SectionData,\r
               &SectionDataSize\r
               );\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    Section = (EFI_COMMON_SECTION_HEADER *)(FileHeader + 1);\r
    Status  = ExtractGuidedSectionGetInfo (\r
                Section,\r
                &DstBufferSize,\r
                &ScratchBufferSize,\r
                &SectionAttribute\r
                );\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    //\r
    // Allocate scratch buffer\r
    //\r
    ScratchBuffer = (VOID *)(UINTN)AllocatePages (EFI_SIZE_TO_PAGES (ScratchBufferSize));\r
    if (ScratchBuffer == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    //\r
    // Allocate destination buffer, extra one page for adjustment\r
    //\r
    DstBuffer = (VOID *)(UINTN)AllocatePages (EFI_SIZE_TO_PAGES (DstBufferSize));\r
    if (DstBuffer == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    //\r
    // Call decompress function\r
    //\r
    Status = ExtractGuidedSectionDecode (\r
               Section,\r
               &DstBuffer,\r
               ScratchBuffer,\r
               &AuthenticationStatus\r
               );\r
    FreePages (ScratchBuffer, EFI_SIZE_TO_PAGES (ScratchBufferSize));\r
    if (EFI_ERROR (Status)) {\r
      goto FreeDstBuffer;\r
    }\r
\r
    DEBUG ((\r
      DEBUG_INFO,\r
      "Processing compressed firmware volume (AuthenticationStatus == %x)\n",\r
      AuthenticationStatus\r
      ));\r
\r
    Status = FindFfsSectionInSections (\r
               DstBuffer,\r
               DstBufferSize,\r
               EFI_SECTION_FIRMWARE_VOLUME_IMAGE,\r
               &Section\r
               );\r
    if (EFI_ERROR (Status)) {\r
      goto FreeDstBuffer;\r
    }\r
\r
    InnerFvHeader = (VOID *)(Section + 1);\r
    Status        = MmCoreFfsFindMmDriver (InnerFvHeader);\r
    if (EFI_ERROR (Status)) {\r
      goto FreeDstBuffer;\r
    }\r
  } while (TRUE);\r
\r
  for (Index = 0; Index < sizeof (mMmFileTypes) / sizeof (mMmFileTypes[0]); Index++) {\r
    DEBUG ((DEBUG_INFO, "Check MmFileTypes - 0x%x\n", mMmFileTypes[Index]));\r
    FileType   = mMmFileTypes[Index];\r
    FileHeader = NULL;\r
    do {\r
      Status = FfsFindNextFile (FileType, FwVolHeader, &FileHeader);\r
      if (!EFI_ERROR (Status)) {\r
        Status = FfsFindSectionData (EFI_SECTION_PE32, FileHeader, &Pe32Data, &Pe32DataSize);\r
        DEBUG ((DEBUG_INFO, "Find PE data - 0x%x\n", Pe32Data));\r
        DepexStatus = FfsFindSectionData (EFI_SECTION_MM_DEPEX, FileHeader, &Depex, &DepexSize);\r
        if (!EFI_ERROR (DepexStatus)) {\r
          MmAddToDriverList (FwVolHeader, Pe32Data, Pe32DataSize, Depex, DepexSize, &FileHeader->Name);\r
        }\r
      }\r
    } while (!EFI_ERROR (Status));\r
  }\r
\r
  return EFI_SUCCESS;\r
\r
FreeDstBuffer:\r
  FreePages (DstBuffer, EFI_SIZE_TO_PAGES (DstBufferSize));\r
\r
  return Status;\r
}\r