git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7105 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			716 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			716 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  Last PEIM.
 | 
						|
  Responsibility of this module is to load the DXE Core from a Firmware Volume.
 | 
						|
 | 
						|
Copyright (c) 2006 - 2008, Intel Corporation. <BR>
 | 
						|
All rights reserved. This program and the accompanying materials
 | 
						|
are licensed and made available under the terms and conditions of the BSD License
 | 
						|
which accompanies this distribution.  The full text of the license may be found at
 | 
						|
http://opensource.org/licenses/bsd-license.php
 | 
						|
 | 
						|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 | 
						|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
#include "DxeIpl.h"
 | 
						|
 | 
						|
 | 
						|
//
 | 
						|
// Module Globals used in the DXE to PEI hand off
 | 
						|
// These must be module globals, so the stack can be switched
 | 
						|
//
 | 
						|
CONST EFI_DXE_IPL_PPI mDxeIplPpi = {
 | 
						|
  DxeLoadCore
 | 
						|
};
 | 
						|
 | 
						|
CONST EFI_PEI_GUIDED_SECTION_EXTRACTION_PPI mCustomGuidedSectionExtractionPpi = {
 | 
						|
  CustomGuidedSectionExtract
 | 
						|
};
 | 
						|
 | 
						|
CONST EFI_PEI_DECOMPRESS_PPI mDecompressPpi = {
 | 
						|
  Decompress
 | 
						|
};
 | 
						|
 | 
						|
CONST EFI_PEI_PPI_DESCRIPTOR mPpiList[] = {
 | 
						|
  {
 | 
						|
    EFI_PEI_PPI_DESCRIPTOR_PPI,
 | 
						|
    &gEfiDxeIplPpiGuid,
 | 
						|
    (VOID *) &mDxeIplPpi
 | 
						|
  },
 | 
						|
  {
 | 
						|
    (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),
 | 
						|
    &gEfiPeiDecompressPpiGuid,
 | 
						|
    (VOID *) &mDecompressPpi
 | 
						|
  }
 | 
						|
};
 | 
						|
 | 
						|
CONST EFI_PEI_PPI_DESCRIPTOR gEndOfPeiSignalPpi = {
 | 
						|
  (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),
 | 
						|
  &gEfiEndOfPeiSignalPpiGuid,
 | 
						|
  NULL
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
  Entry point of DXE IPL PEIM.
 | 
						|
  
 | 
						|
  This function installs DXE IPL PPI and Decompress PPI.  It also reloads
 | 
						|
  itself to memory on non-S3 resume boot path.
 | 
						|
 | 
						|
  @param  FileHandle  Handle of the file being invoked.
 | 
						|
  @param  PeiServices Describes the list of possible PEI Services.
 | 
						|
 | 
						|
  @retval EFI_SUCESS  The entry point of DXE IPL PEIM executes successfully.
 | 
						|
  @retval Others      Some error occurs during the execution of this function. 
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
PeimInitializeDxeIpl (
 | 
						|
  IN       EFI_PEI_FILE_HANDLE  FileHandle,
 | 
						|
  IN CONST EFI_PEI_SERVICES     **PeiServices
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                                Status;
 | 
						|
  EFI_BOOT_MODE                             BootMode;
 | 
						|
  EFI_GUID                                  *ExtractHandlerGuidTable;
 | 
						|
  UINTN                                     ExtractHandlerNumber;
 | 
						|
  EFI_PEI_PPI_DESCRIPTOR                    *GuidPpi;
 | 
						|
  
 | 
						|
  BootMode = GetBootModeHob ();
 | 
						|
 | 
						|
  if (BootMode != BOOT_ON_S3_RESUME) {
 | 
						|
    Status = PeiServicesRegisterForShadow (FileHandle);
 | 
						|
    if (Status == EFI_SUCCESS) {
 | 
						|
      //
 | 
						|
      // EFI_SUCESS means it is the first time to call register for shadow. 
 | 
						|
      // 
 | 
						|
      return Status;
 | 
						|
    }
 | 
						|
    
 | 
						|
    //
 | 
						|
    // Ensure that DXE IPL is shadowed to permanent memory.
 | 
						|
    //
 | 
						|
    ASSERT (Status == EFI_ALREADY_STARTED);
 | 
						|
     
 | 
						|
    //
 | 
						|
    // Get custom extract guided section method guid list 
 | 
						|
    //
 | 
						|
    ExtractHandlerNumber = ExtractGuidedSectionGetGuidList (&ExtractHandlerGuidTable);
 | 
						|
    
 | 
						|
    //
 | 
						|
    // Install custom extraction guid PPI
 | 
						|
    //
 | 
						|
    if (ExtractHandlerNumber > 0) {
 | 
						|
      GuidPpi = (EFI_PEI_PPI_DESCRIPTOR *) AllocatePool (ExtractHandlerNumber * sizeof (EFI_PEI_PPI_DESCRIPTOR));
 | 
						|
      ASSERT (GuidPpi != NULL);
 | 
						|
      while (ExtractHandlerNumber-- > 0) {
 | 
						|
        GuidPpi->Flags = EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST;
 | 
						|
        GuidPpi->Ppi   = (VOID *) &mCustomGuidedSectionExtractionPpi;
 | 
						|
        GuidPpi->Guid  = &ExtractHandlerGuidTable[ExtractHandlerNumber];
 | 
						|
        Status = PeiServicesInstallPpi (GuidPpi++);
 | 
						|
        ASSERT_EFI_ERROR(Status);
 | 
						|
      }
 | 
						|
    }
 | 
						|
    
 | 
						|
  }
 | 
						|
  
 | 
						|
  //
 | 
						|
  // Install DxeIpl and Decompress PPIs.
 | 
						|
  //
 | 
						|
  Status = PeiServicesInstallPpi (mPpiList);
 | 
						|
  ASSERT_EFI_ERROR(Status);
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
   Main entry point to last PEIM. 
 | 
						|
 | 
						|
   This function finds DXE Core in the firmware volume and transfer the control to
 | 
						|
   DXE core.
 | 
						|
    
 | 
						|
   @param This          Entry point for DXE IPL PPI.
 | 
						|
   @param PeiServices   General purpose services available to every PEIM.
 | 
						|
   @param HobList       Address to the Pei HOB list.
 | 
						|
   
 | 
						|
   @return EFI_SUCCESS              DXE core was successfully loaded. 
 | 
						|
   @return EFI_OUT_OF_RESOURCES     There are not enough resources to load DXE core.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxeLoadCore (
 | 
						|
  IN CONST EFI_DXE_IPL_PPI *This,
 | 
						|
  IN EFI_PEI_SERVICES      **PeiServices,
 | 
						|
  IN EFI_PEI_HOB_POINTERS  HobList
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                                Status;
 | 
						|
  EFI_FV_FILE_INFO                          DxeCoreFileInfo;
 | 
						|
  EFI_PHYSICAL_ADDRESS                      DxeCoreAddress;
 | 
						|
  UINT64                                    DxeCoreSize;
 | 
						|
  EFI_PHYSICAL_ADDRESS                      DxeCoreEntryPoint;
 | 
						|
  EFI_BOOT_MODE                             BootMode;
 | 
						|
  EFI_PEI_FILE_HANDLE                       FileHandle;
 | 
						|
  EFI_PEI_READ_ONLY_VARIABLE2_PPI           *Variable;
 | 
						|
  UINTN                                     DataSize;
 | 
						|
  EFI_MEMORY_TYPE_INFORMATION               MemoryData[EfiMaxMemoryType + 1];
 | 
						|
 | 
						|
  //
 | 
						|
  // if in S3 Resume, restore configure
 | 
						|
  //
 | 
						|
  BootMode = GetBootModeHob ();
 | 
						|
 | 
						|
  if (BootMode == BOOT_ON_S3_RESUME) {
 | 
						|
    Status = AcpiS3ResumeOs();
 | 
						|
    ASSERT_EFI_ERROR (Status);
 | 
						|
  } else if (BootMode == BOOT_IN_RECOVERY_MODE) {
 | 
						|
    Status = PeiRecoverFirmware ();
 | 
						|
    if (EFI_ERROR (Status)) {
 | 
						|
      DEBUG ((DEBUG_ERROR, "Load Recovery Capsule Failed.(Status = %r)\n", Status));
 | 
						|
      CpuDeadLoop ();
 | 
						|
    }
 | 
						|
 | 
						|
    //
 | 
						|
    // Now should have a HOB with the DXE core
 | 
						|
    //
 | 
						|
  }
 | 
						|
 | 
						|
  Status = PeiServicesLocatePpi (
 | 
						|
             &gEfiPeiReadOnlyVariable2PpiGuid,
 | 
						|
             0,
 | 
						|
             NULL,
 | 
						|
             (VOID **)&Variable
 | 
						|
             );
 | 
						|
  if (!EFI_ERROR (Status)) {
 | 
						|
    DataSize = sizeof (MemoryData);
 | 
						|
    Status = Variable->GetVariable ( 
 | 
						|
                         Variable, 
 | 
						|
                         EFI_MEMORY_TYPE_INFORMATION_VARIABLE_NAME,
 | 
						|
                         &gEfiMemoryTypeInformationGuid,
 | 
						|
                         NULL,
 | 
						|
                         &DataSize,
 | 
						|
                         &MemoryData
 | 
						|
                         );
 | 
						|
    if (!EFI_ERROR (Status)) {
 | 
						|
      //
 | 
						|
      // Build the GUID'd HOB for DXE
 | 
						|
      //
 | 
						|
      BuildGuidDataHob (
 | 
						|
        &gEfiMemoryTypeInformationGuid,
 | 
						|
        MemoryData,
 | 
						|
        DataSize
 | 
						|
        );
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Look in all the FVs present in PEI and find the DXE Core FileHandle
 | 
						|
  //
 | 
						|
  FileHandle = DxeIplFindDxeCore ();
 | 
						|
 | 
						|
  //
 | 
						|
  // Load the DXE Core from a Firmware Volume, may use LoadFile PPI to do this to save code size.
 | 
						|
  //
 | 
						|
  Status = PeiLoadFile (
 | 
						|
             FileHandle,
 | 
						|
             &DxeCoreAddress,
 | 
						|
             &DxeCoreSize,
 | 
						|
             &DxeCoreEntryPoint
 | 
						|
             );
 | 
						|
  ASSERT_EFI_ERROR (Status);
 | 
						|
 | 
						|
  //
 | 
						|
  // Get the DxeCore File Info from the FileHandle for the DxeCore GUID file name.
 | 
						|
  //
 | 
						|
  Status = PeiServicesFfsGetFileInfo (FileHandle, &DxeCoreFileInfo);
 | 
						|
  ASSERT_EFI_ERROR (Status);
 | 
						|
 | 
						|
  //
 | 
						|
  // Add HOB for the DXE Core
 | 
						|
  //
 | 
						|
  BuildModuleHob (
 | 
						|
    &DxeCoreFileInfo.FileName,
 | 
						|
    DxeCoreAddress,
 | 
						|
    ALIGN_VALUE (DxeCoreSize, EFI_PAGE_SIZE),
 | 
						|
    DxeCoreEntryPoint
 | 
						|
    );
 | 
						|
 | 
						|
  //
 | 
						|
  // Report Status Code EFI_SW_PEI_PC_HANDOFF_TO_NEXT
 | 
						|
  //
 | 
						|
  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, PcdGet32 (PcdStatusCodeValuePeiHandoffToDxe));
 | 
						|
 | 
						|
  DEBUG ((DEBUG_INFO | DEBUG_LOAD, "Loading DXE CORE at 0x%11p EntryPoint=0x%11p\n", (VOID *)(UINTN)DxeCoreAddress, FUNCTION_ENTRY_POINT (DxeCoreEntryPoint)));
 | 
						|
 | 
						|
  //
 | 
						|
  // Transfer control to the DXE Core
 | 
						|
  // The hand off state is simply a pointer to the HOB list
 | 
						|
  //
 | 
						|
  HandOffToDxeCore (DxeCoreEntryPoint, HobList);
 | 
						|
  //
 | 
						|
  // If we get here, then the DXE Core returned.  This is an error
 | 
						|
  // DxeCore should not return.
 | 
						|
  //
 | 
						|
  ASSERT (FALSE);
 | 
						|
  CpuDeadLoop ();
 | 
						|
 | 
						|
  return EFI_OUT_OF_RESOURCES;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
   Searches DxeCore in all firmware Volumes and loads the first
 | 
						|
   instance that contains DxeCore.
 | 
						|
 | 
						|
   @return FileHandle of DxeCore to load DxeCore.
 | 
						|
   
 | 
						|
**/
 | 
						|
EFI_PEI_FILE_HANDLE
 | 
						|
DxeIplFindDxeCore (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS            Status;
 | 
						|
  UINTN                 Instance;
 | 
						|
  EFI_PEI_FV_HANDLE     VolumeHandle;
 | 
						|
  EFI_PEI_FILE_HANDLE   FileHandle;
 | 
						|
  
 | 
						|
  Instance    = 0;
 | 
						|
  while (TRUE) {
 | 
						|
    //
 | 
						|
    // Traverse all firmware volume instances
 | 
						|
    //
 | 
						|
    Status = PeiServicesFfsFindNextVolume (Instance, &VolumeHandle);
 | 
						|
    //
 | 
						|
    // If some error occurs here, then we cannot find any firmware
 | 
						|
    // volume that may contain DxeCore.
 | 
						|
    //
 | 
						|
    ASSERT_EFI_ERROR (Status);
 | 
						|
    
 | 
						|
    //
 | 
						|
    // Find the DxeCore file type from the beginning in this firmware volume.
 | 
						|
    //
 | 
						|
    FileHandle = NULL;
 | 
						|
    Status = PeiServicesFfsFindNextFile (EFI_FV_FILETYPE_DXE_CORE, VolumeHandle, &FileHandle);
 | 
						|
    if (!EFI_ERROR (Status)) {
 | 
						|
      //
 | 
						|
      // Find DxeCore FileHandle in this volume, then we skip other firmware volume and
 | 
						|
      // return the FileHandle.
 | 
						|
      //
 | 
						|
      return FileHandle;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // We cannot find DxeCore in this firmware volume, then search the next volume.
 | 
						|
    //
 | 
						|
    Instance++;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
   Loads and relocates a PE/COFF image into memory.
 | 
						|
 | 
						|
   @param FileHandle        The image file handle
 | 
						|
   @param ImageAddress      The base address of the relocated PE/COFF image
 | 
						|
   @param ImageSize         The size of the relocated PE/COFF image
 | 
						|
   @param EntryPoint        The entry point of the relocated PE/COFF image
 | 
						|
   
 | 
						|
   @return EFI_SUCCESS           The file was loaded and relocated
 | 
						|
   @return EFI_OUT_OF_RESOURCES  There was not enough memory to load and relocate the PE/COFF file
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
PeiLoadFile (
 | 
						|
  IN  EFI_PEI_FILE_HANDLE                       FileHandle,
 | 
						|
  OUT EFI_PHYSICAL_ADDRESS                      *ImageAddress,
 | 
						|
  OUT UINT64                                    *ImageSize,
 | 
						|
  OUT EFI_PHYSICAL_ADDRESS                      *EntryPoint
 | 
						|
  )
 | 
						|
{
 | 
						|
 | 
						|
  EFI_STATUS                        Status;
 | 
						|
  PE_COFF_LOADER_IMAGE_CONTEXT      ImageContext;
 | 
						|
  VOID                              *Pe32Data;
 | 
						|
 | 
						|
  //
 | 
						|
  // First try to find the PE32 section in this ffs file.
 | 
						|
  //
 | 
						|
  Status = PeiServicesFfsFindSectionData (
 | 
						|
             EFI_SECTION_PE32,
 | 
						|
             FileHandle,
 | 
						|
             &Pe32Data
 | 
						|
             );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    //
 | 
						|
    // NO image types we support so exit.
 | 
						|
    //
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  ZeroMem (&ImageContext, sizeof (ImageContext));
 | 
						|
  ImageContext.Handle = Pe32Data;
 | 
						|
  ImageContext.ImageRead = PeiImageRead;
 | 
						|
 | 
						|
 | 
						|
  Status = PeCoffLoaderGetImageInfo (&ImageContext);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
  //
 | 
						|
  // Allocate Memory for the image
 | 
						|
  //
 | 
						|
  Status = PeiServicesAllocatePages (
 | 
						|
             EfiBootServicesCode, 
 | 
						|
             EFI_SIZE_TO_PAGES ((UINT32) ImageContext.ImageSize), 
 | 
						|
             &ImageContext.ImageAddress
 | 
						|
             );
 | 
						|
  ASSERT_EFI_ERROR (Status);
 | 
						|
  ASSERT (ImageContext.ImageAddress != 0);
 | 
						|
 | 
						|
  //
 | 
						|
  // Load the image to our new buffer
 | 
						|
  //
 | 
						|
  Status = PeCoffLoaderLoadImage (&ImageContext);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
  //
 | 
						|
  // Relocate the image in our new buffer
 | 
						|
  //
 | 
						|
  Status = PeCoffLoaderRelocateImage (&ImageContext);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Flush the instruction cache so the image data are written before we execute it
 | 
						|
  //
 | 
						|
  InvalidateInstructionCacheRange ((VOID *)(UINTN) ImageContext.ImageAddress, (UINTN) ImageContext.ImageSize);
 | 
						|
 | 
						|
  *ImageAddress = ImageContext.ImageAddress;
 | 
						|
  *ImageSize    = ImageContext.ImageSize;
 | 
						|
  *EntryPoint   = ImageContext.EntryPoint;
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  The ExtractSection() function processes the input section and
 | 
						|
  returns a pointer to the section contents. If the section being
 | 
						|
  extracted does not require processing (if the section
 | 
						|
  GuidedSectionHeader.Attributes has the
 | 
						|
  EFI_GUIDED_SECTION_PROCESSING_REQUIRED field cleared), then
 | 
						|
  OutputBuffer is just updated to point to the start of the
 | 
						|
  section's contents. Otherwise, *Buffer must be allocated
 | 
						|
  from PEI permanent memory.
 | 
						|
 | 
						|
  @param This                   Indicates the
 | 
						|
                                EFI_PEI_GUIDED_SECTION_EXTRACTION_PPI instance.
 | 
						|
                                Buffer containing the input GUIDed section to be
 | 
						|
                                processed. OutputBuffer OutputBuffer is
 | 
						|
                                allocated from PEI permanent memory and contains
 | 
						|
                                the new section stream.
 | 
						|
  @param InputSection           A pointer to the input buffer, which contains
 | 
						|
                                the input section to be processed.
 | 
						|
  @param OutputBuffer           A pointer to a caller-allocated buffer, whose
 | 
						|
                                size is specified by the contents of OutputSize.
 | 
						|
  @param OutputSize             A pointer to a caller-allocated
 | 
						|
                                UINTN in which the size of *OutputBuffer
 | 
						|
                                allocation is stored. If the function
 | 
						|
                                returns anything other than EFI_SUCCESS,
 | 
						|
                                the value of OutputSize is undefined.
 | 
						|
  @param AuthenticationStatus   A pointer to a caller-allocated
 | 
						|
                                UINT32 that indicates the
 | 
						|
                                authentication status of the
 | 
						|
                                output buffer. If the input
 | 
						|
                                section's GuidedSectionHeader.
 | 
						|
                                Attributes field has the
 | 
						|
                                EFI_GUIDED_SECTION_AUTH_STATUS_VALID 
 | 
						|
                                bit as clear,
 | 
						|
                                AuthenticationStatus must return
 | 
						|
                                zero. These bits reflect the
 | 
						|
                                status of the extraction
 | 
						|
                                operation. If the function
 | 
						|
                                returns anything other than
 | 
						|
                                EFI_SUCCESS, the value of
 | 
						|
                                AuthenticationStatus is
 | 
						|
                                undefined.
 | 
						|
  
 | 
						|
  @retval EFI_SUCCESS           The InputSection was
 | 
						|
                                successfully processed and the
 | 
						|
                                section contents were returned.
 | 
						|
  
 | 
						|
  @retval EFI_OUT_OF_RESOURCES  The system has insufficient
 | 
						|
                                resources to process the request.
 | 
						|
  
 | 
						|
  @retval EFI_INVALID_PARAMETER The GUID in InputSection does
 | 
						|
                                not match this instance of the
 | 
						|
                                GUIDed Section Extraction PPI.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
CustomGuidedSectionExtract (
 | 
						|
  IN CONST  EFI_PEI_GUIDED_SECTION_EXTRACTION_PPI *This,
 | 
						|
  IN CONST  VOID                                  *InputSection,
 | 
						|
  OUT       VOID                                  **OutputBuffer,
 | 
						|
  OUT       UINTN                                 *OutputSize,
 | 
						|
  OUT       UINT32                                *AuthenticationStatus
 | 
						|
)
 | 
						|
{
 | 
						|
  EFI_STATUS      Status;
 | 
						|
  UINT8           *ScratchBuffer;
 | 
						|
  UINT32          ScratchBufferSize;
 | 
						|
  UINT32          OutputBufferSize;
 | 
						|
  UINT16          SectionAttribute;
 | 
						|
  
 | 
						|
  //
 | 
						|
  // Init local variable
 | 
						|
  //
 | 
						|
  ScratchBuffer = NULL;
 | 
						|
 | 
						|
  //
 | 
						|
  // Call GetInfo to get the size and attribute of input guided section data.
 | 
						|
  //
 | 
						|
  Status = ExtractGuidedSectionGetInfo (
 | 
						|
             InputSection,
 | 
						|
             &OutputBufferSize,
 | 
						|
             &ScratchBufferSize,
 | 
						|
             &SectionAttribute
 | 
						|
             );
 | 
						|
  
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    DEBUG ((DEBUG_ERROR, "GetInfo from guided section Failed - %r\n", Status));
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
  
 | 
						|
  if (ScratchBufferSize != 0) {
 | 
						|
    //
 | 
						|
    // Allocate scratch buffer
 | 
						|
    //
 | 
						|
    ScratchBuffer = AllocatePages (EFI_SIZE_TO_PAGES (ScratchBufferSize));
 | 
						|
    if (ScratchBuffer == NULL) {
 | 
						|
      return EFI_OUT_OF_RESOURCES;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (((SectionAttribute & EFI_GUIDED_SECTION_PROCESSING_REQUIRED) != 0) && OutputBufferSize > 0) {  
 | 
						|
    //
 | 
						|
    // Allocate output buffer
 | 
						|
    //
 | 
						|
    *OutputBuffer = AllocatePages (EFI_SIZE_TO_PAGES (OutputBufferSize) + 1);
 | 
						|
    if (*OutputBuffer == NULL) {
 | 
						|
      return EFI_OUT_OF_RESOURCES;
 | 
						|
    }
 | 
						|
    DEBUG ((DEBUG_INFO, "Customized Guided section Memory Size required is 0x%x and address is 0x%p\n", OutputBufferSize, *OutputBuffer));
 | 
						|
    //
 | 
						|
    // *OutputBuffer still is one section. Adjust *OutputBuffer offset, 
 | 
						|
    // skip EFI section header to make section data at page alignment.
 | 
						|
    //
 | 
						|
    *OutputBuffer = (VOID *)((UINT8 *) *OutputBuffer + EFI_PAGE_SIZE - sizeof (EFI_COMMON_SECTION_HEADER));
 | 
						|
  }
 | 
						|
  
 | 
						|
  Status = ExtractGuidedSectionDecode (
 | 
						|
             InputSection, 
 | 
						|
             OutputBuffer,
 | 
						|
             ScratchBuffer,
 | 
						|
             AuthenticationStatus
 | 
						|
             );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    //
 | 
						|
    // Decode failed
 | 
						|
    //
 | 
						|
    DEBUG ((DEBUG_ERROR, "Extract guided section Failed - %r\n", Status));
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
  
 | 
						|
  *OutputSize = (UINTN) OutputBufferSize;
 | 
						|
  
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
   Decompresses a section to the output buffer.
 | 
						|
 | 
						|
   This function looks up the compression type field in the input section and
 | 
						|
   applies the appropriate compression algorithm to compress the section to a
 | 
						|
   callee allocated buffer.
 | 
						|
    
 | 
						|
   @param  This                  Points to this instance of the
 | 
						|
                                 EFI_PEI_DECOMPRESS_PEI PPI.
 | 
						|
   @param  CompressionSection    Points to the compressed section.
 | 
						|
   @param  OutputBuffer          Holds the returned pointer to the decompressed
 | 
						|
                                 sections.
 | 
						|
   @param  OutputSize            Holds the returned size of the decompress
 | 
						|
                                 section streams.
 | 
						|
   
 | 
						|
   @retval EFI_SUCCESS           The section was decompressed successfully.
 | 
						|
                                 OutputBuffer contains the resulting data and
 | 
						|
                                 OutputSize contains the resulting size.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI 
 | 
						|
Decompress (
 | 
						|
  IN CONST  EFI_PEI_DECOMPRESS_PPI  *This,
 | 
						|
  IN CONST  EFI_COMPRESSION_SECTION *CompressionSection,
 | 
						|
  OUT       VOID                    **OutputBuffer,
 | 
						|
  OUT       UINTN                   *OutputSize
 | 
						|
 )
 | 
						|
{
 | 
						|
  EFI_STATUS                      Status;
 | 
						|
  UINT8                           *DstBuffer;
 | 
						|
  UINT8                           *ScratchBuffer;
 | 
						|
  UINTN                           DstBufferSize;
 | 
						|
  UINT32                          ScratchBufferSize;
 | 
						|
  EFI_COMMON_SECTION_HEADER       *Section;
 | 
						|
  UINTN                           SectionLength;
 | 
						|
 | 
						|
  if (CompressionSection->CommonHeader.Type != EFI_SECTION_COMPRESSION) {
 | 
						|
    ASSERT (FALSE);
 | 
						|
    return EFI_INVALID_PARAMETER;
 | 
						|
  }
 | 
						|
 | 
						|
  Section = (EFI_COMMON_SECTION_HEADER *) CompressionSection;
 | 
						|
  SectionLength = *(UINT32 *) (Section->Size) & 0x00ffffff;
 | 
						|
  
 | 
						|
  //
 | 
						|
  // This is a compression set, expand it
 | 
						|
  //
 | 
						|
  switch (CompressionSection->CompressionType) {
 | 
						|
  case EFI_STANDARD_COMPRESSION:
 | 
						|
    //
 | 
						|
    // Load EFI standard compression.
 | 
						|
    // For compressed data, decompress them to destination buffer.
 | 
						|
    //
 | 
						|
    Status = UefiDecompressGetInfo (
 | 
						|
               (UINT8 *) ((EFI_COMPRESSION_SECTION *) Section + 1),
 | 
						|
               (UINT32) SectionLength - sizeof (EFI_COMPRESSION_SECTION),
 | 
						|
               (UINT32 *) &DstBufferSize,
 | 
						|
               &ScratchBufferSize
 | 
						|
               );
 | 
						|
    if (EFI_ERROR (Status)) {
 | 
						|
      //
 | 
						|
      // GetInfo failed
 | 
						|
      //
 | 
						|
      DEBUG ((DEBUG_ERROR, "Decompress GetInfo Failed - %r\n", Status));
 | 
						|
      return EFI_NOT_FOUND;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Allocate scratch buffer
 | 
						|
    //
 | 
						|
    ScratchBuffer = AllocatePages (EFI_SIZE_TO_PAGES (ScratchBufferSize));
 | 
						|
    if (ScratchBuffer == NULL) {
 | 
						|
      return EFI_OUT_OF_RESOURCES;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Allocate destination buffer, extra one page for adjustment 
 | 
						|
    //
 | 
						|
    DstBuffer = AllocatePages (EFI_SIZE_TO_PAGES (DstBufferSize) + 1);
 | 
						|
    if (DstBuffer == NULL) {
 | 
						|
      return EFI_OUT_OF_RESOURCES;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // DstBuffer still is one section. Adjust DstBuffer offset, skip EFI section header
 | 
						|
    // to make section data at page alignment.
 | 
						|
    //
 | 
						|
    DstBuffer = DstBuffer + EFI_PAGE_SIZE - sizeof (EFI_COMMON_SECTION_HEADER);
 | 
						|
    //
 | 
						|
    // Call decompress function
 | 
						|
    //
 | 
						|
    Status = UefiDecompress (
 | 
						|
                (CHAR8 *) ((EFI_COMPRESSION_SECTION *) Section + 1),
 | 
						|
                DstBuffer,
 | 
						|
                ScratchBuffer
 | 
						|
                );
 | 
						|
    if (EFI_ERROR (Status)) {
 | 
						|
      //
 | 
						|
      // Decompress failed
 | 
						|
      //
 | 
						|
      DEBUG ((DEBUG_ERROR, "Decompress Failed - %r\n", Status));
 | 
						|
      return EFI_NOT_FOUND;
 | 
						|
    }
 | 
						|
    break;
 | 
						|
 | 
						|
  case EFI_NOT_COMPRESSED:
 | 
						|
    //
 | 
						|
    // Allocate destination buffer
 | 
						|
    //
 | 
						|
    DstBufferSize = CompressionSection->UncompressedLength;
 | 
						|
    DstBuffer     = AllocatePages (EFI_SIZE_TO_PAGES (DstBufferSize) + 1);
 | 
						|
    if (DstBuffer == NULL) {
 | 
						|
      return EFI_OUT_OF_RESOURCES;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Adjust DstBuffer offset, skip EFI section header
 | 
						|
    // to make section data at page alignment.
 | 
						|
    //
 | 
						|
    DstBuffer = DstBuffer + EFI_PAGE_SIZE - sizeof (EFI_COMMON_SECTION_HEADER);
 | 
						|
    //
 | 
						|
    // stream is not actually compressed, just encapsulated.  So just copy it.
 | 
						|
    //
 | 
						|
    CopyMem (DstBuffer, CompressionSection + 1, DstBufferSize);
 | 
						|
    break;
 | 
						|
 | 
						|
  default:
 | 
						|
    //
 | 
						|
    // Don't support other unknown compression type.
 | 
						|
    //
 | 
						|
    ASSERT (FALSE);
 | 
						|
    return EFI_NOT_FOUND;
 | 
						|
  }
 | 
						|
 | 
						|
  *OutputSize = DstBufferSize;
 | 
						|
  *OutputBuffer = DstBuffer;
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
   Updates the Stack HOB passed to DXE phase.
 | 
						|
 | 
						|
   This function traverses the whole HOB list and update the stack HOB to
 | 
						|
   reflect the real stack that is used by DXE core.
 | 
						|
 | 
						|
   @param BaseAddress           The lower address of stack used by DxeCore.
 | 
						|
   @param Length                The length of stack used by DxeCore.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
UpdateStackHob (
 | 
						|
  IN EFI_PHYSICAL_ADDRESS        BaseAddress,
 | 
						|
  IN UINT64                      Length
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_PEI_HOB_POINTERS           Hob;
 | 
						|
 | 
						|
  Hob.Raw = GetHobList ();
 | 
						|
  while ((Hob.Raw = GetNextHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, Hob.Raw)) != NULL) {
 | 
						|
    if (CompareGuid (&gEfiHobMemoryAllocStackGuid, &(Hob.MemoryAllocationStack->AllocDescriptor.Name))) {
 | 
						|
      //
 | 
						|
      // Build a new memory allocation HOB with old stack info with EfiConventionalMemory type
 | 
						|
      // to be reclaimed by DXE core.
 | 
						|
      //
 | 
						|
      BuildMemoryAllocationHob (
 | 
						|
        Hob.MemoryAllocationStack->AllocDescriptor.MemoryBaseAddress,
 | 
						|
        Hob.MemoryAllocationStack->AllocDescriptor.MemoryLength,
 | 
						|
        EfiConventionalMemory
 | 
						|
        );
 | 
						|
      //
 | 
						|
      // Update the BSP Stack Hob to reflect the new stack info.
 | 
						|
      //
 | 
						|
      Hob.MemoryAllocationStack->AllocDescriptor.MemoryBaseAddress = BaseAddress;
 | 
						|
      Hob.MemoryAllocationStack->AllocDescriptor.MemoryLength = Length;
 | 
						|
      break;
 | 
						|
    }
 | 
						|
    Hob.Raw = GET_NEXT_HOB (Hob);
 | 
						|
  }
 | 
						|
}
 |