mirror of https://github.com/acidanthera/audk.git
200 lines
6.3 KiB
C
200 lines
6.3 KiB
C
|
/** @file
|
||
|
Locate the entry point for the PEI Core
|
||
|
|
||
|
Copyright (c) 2014, Intel Corporation. All rights reserved.<BR>
|
||
|
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 <PiPei.h>
|
||
|
#include <Library/BaseLib.h>
|
||
|
#include <Library/PeCoffGetEntryPointLib.h>
|
||
|
|
||
|
#include "SecMain.h"
|
||
|
|
||
|
/**
|
||
|
Find core image base.
|
||
|
|
||
|
@param[in] BootFirmwareVolumePtr Point to the boot firmware volume.
|
||
|
@param[out] SecCoreImageBase The base address of the SEC core image.
|
||
|
@param[out] PeiCoreImageBase The base address of the PEI core image.
|
||
|
|
||
|
**/
|
||
|
EFI_STATUS
|
||
|
EFIAPI
|
||
|
FindImageBase (
|
||
|
IN EFI_FIRMWARE_VOLUME_HEADER *BootFirmwareVolumePtr,
|
||
|
OUT EFI_PHYSICAL_ADDRESS *SecCoreImageBase,
|
||
|
OUT EFI_PHYSICAL_ADDRESS *PeiCoreImageBase
|
||
|
)
|
||
|
{
|
||
|
EFI_PHYSICAL_ADDRESS CurrentAddress;
|
||
|
EFI_PHYSICAL_ADDRESS EndOfFirmwareVolume;
|
||
|
EFI_FFS_FILE_HEADER *File;
|
||
|
UINT32 Size;
|
||
|
EFI_PHYSICAL_ADDRESS EndOfFile;
|
||
|
EFI_COMMON_SECTION_HEADER *Section;
|
||
|
EFI_PHYSICAL_ADDRESS EndOfSection;
|
||
|
|
||
|
*SecCoreImageBase = 0;
|
||
|
*PeiCoreImageBase = 0;
|
||
|
|
||
|
CurrentAddress = (EFI_PHYSICAL_ADDRESS)(UINTN) BootFirmwareVolumePtr;
|
||
|
EndOfFirmwareVolume = CurrentAddress + BootFirmwareVolumePtr->FvLength;
|
||
|
|
||
|
//
|
||
|
// Loop through the FFS files in the Boot Firmware Volume
|
||
|
//
|
||
|
for (EndOfFile = CurrentAddress + BootFirmwareVolumePtr->HeaderLength; ; ) {
|
||
|
|
||
|
CurrentAddress = (EndOfFile + 7) & 0xfffffffffffffff8ULL;
|
||
|
if (CurrentAddress > EndOfFirmwareVolume) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
|
||
|
File = (EFI_FFS_FILE_HEADER*)(UINTN) CurrentAddress;
|
||
|
if (IS_FFS_FILE2 (File)) {
|
||
|
Size = FFS_FILE2_SIZE (File);
|
||
|
if (Size <= 0x00FFFFFF) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
} else {
|
||
|
Size = FFS_FILE_SIZE (File);
|
||
|
if (Size < sizeof (EFI_FFS_FILE_HEADER)) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
EndOfFile = CurrentAddress + Size;
|
||
|
if (EndOfFile > EndOfFirmwareVolume) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Look for SEC Core / PEI Core files
|
||
|
//
|
||
|
if (File->Type != EFI_FV_FILETYPE_SECURITY_CORE &&
|
||
|
File->Type != EFI_FV_FILETYPE_PEI_CORE) {
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Loop through the FFS file sections within the FFS file
|
||
|
//
|
||
|
if (IS_FFS_FILE2 (File)) {
|
||
|
EndOfSection = (EFI_PHYSICAL_ADDRESS) (UINTN) ((UINT8 *) File + sizeof (EFI_FFS_FILE_HEADER2));
|
||
|
} else {
|
||
|
EndOfSection = (EFI_PHYSICAL_ADDRESS) (UINTN) ((UINT8 *) File + sizeof (EFI_FFS_FILE_HEADER));
|
||
|
}
|
||
|
for (;;) {
|
||
|
CurrentAddress = (EndOfSection + 3) & 0xfffffffffffffffcULL;
|
||
|
Section = (EFI_COMMON_SECTION_HEADER*)(UINTN) CurrentAddress;
|
||
|
|
||
|
if (IS_SECTION2 (Section)) {
|
||
|
Size = SECTION2_SIZE (Section);
|
||
|
if (Size <= 0x00FFFFFF) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
} else {
|
||
|
Size = SECTION_SIZE (Section);
|
||
|
if (Size < sizeof (EFI_COMMON_SECTION_HEADER)) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
EndOfSection = CurrentAddress + Size;
|
||
|
if (EndOfSection > EndOfFile) {
|
||
|
return EFI_NOT_FOUND;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Look for executable sections
|
||
|
//
|
||
|
if (Section->Type == EFI_SECTION_PE32 || Section->Type == EFI_SECTION_TE) {
|
||
|
if (File->Type == EFI_FV_FILETYPE_SECURITY_CORE) {
|
||
|
if (IS_SECTION2 (Section)) {
|
||
|
*SecCoreImageBase = (PHYSICAL_ADDRESS) (UINTN) ((UINT8 *) Section + sizeof (EFI_COMMON_SECTION_HEADER2));
|
||
|
} else {
|
||
|
*SecCoreImageBase = (PHYSICAL_ADDRESS) (UINTN) ((UINT8 *) Section + sizeof (EFI_COMMON_SECTION_HEADER));
|
||
|
}
|
||
|
} else {
|
||
|
if (IS_SECTION2 (Section)) {
|
||
|
*PeiCoreImageBase = (PHYSICAL_ADDRESS) (UINTN) ((UINT8 *) Section + sizeof (EFI_COMMON_SECTION_HEADER2));
|
||
|
} else {
|
||
|
*PeiCoreImageBase = (PHYSICAL_ADDRESS) (UINTN) ((UINT8 *) Section + sizeof (EFI_COMMON_SECTION_HEADER));
|
||
|
}
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Both SEC Core and PEI Core images found
|
||
|
//
|
||
|
if (*SecCoreImageBase != 0 && *PeiCoreImageBase != 0) {
|
||
|
return EFI_SUCCESS;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
Find and return Pei Core entry point.
|
||
|
|
||
|
It also find SEC and PEI Core file debug inforamtion. It will report them if
|
||
|
remote debug is enabled.
|
||
|
|
||
|
@param[in] BootFirmwareVolumePtr Point to the boot firmware volume.
|
||
|
@param[out] PeiCoreEntryPoint The entry point of the PEI core.
|
||
|
|
||
|
**/
|
||
|
VOID
|
||
|
EFIAPI
|
||
|
FindAndReportEntryPoints (
|
||
|
IN EFI_FIRMWARE_VOLUME_HEADER *BootFirmwareVolumePtr,
|
||
|
OUT EFI_PEI_CORE_ENTRY_POINT *PeiCoreEntryPoint
|
||
|
)
|
||
|
{
|
||
|
EFI_STATUS Status;
|
||
|
EFI_PHYSICAL_ADDRESS SecCoreImageBase;
|
||
|
EFI_PHYSICAL_ADDRESS PeiCoreImageBase;
|
||
|
PE_COFF_LOADER_IMAGE_CONTEXT ImageContext;
|
||
|
|
||
|
//
|
||
|
// Find SEC Core and PEI Core image base
|
||
|
//
|
||
|
Status = FindImageBase (BootFirmwareVolumePtr, &SecCoreImageBase, &PeiCoreImageBase);
|
||
|
ASSERT_EFI_ERROR (Status);
|
||
|
|
||
|
ZeroMem ((VOID *) &ImageContext, sizeof (PE_COFF_LOADER_IMAGE_CONTEXT));
|
||
|
//
|
||
|
// Report SEC Core debug information when remote debug is enabled
|
||
|
//
|
||
|
ImageContext.ImageAddress = SecCoreImageBase;
|
||
|
ImageContext.PdbPointer = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageContext.ImageAddress);
|
||
|
PeCoffLoaderRelocateImageExtraAction (&ImageContext);
|
||
|
|
||
|
//
|
||
|
// Report PEI Core debug information when remote debug is enabled
|
||
|
//
|
||
|
ImageContext.ImageAddress = PeiCoreImageBase;
|
||
|
ImageContext.PdbPointer = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageContext.ImageAddress);
|
||
|
PeCoffLoaderRelocateImageExtraAction (&ImageContext);
|
||
|
|
||
|
//
|
||
|
// Find PEI Core entry point
|
||
|
//
|
||
|
Status = PeCoffLoaderGetEntryPoint ((VOID *) (UINTN) PeiCoreImageBase, (VOID**) PeiCoreEntryPoint);
|
||
|
if (EFI_ERROR (Status)) {
|
||
|
*PeiCoreEntryPoint = 0;
|
||
|
}
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|