/** @file\r
Pei Core Load Image Support\r
\r
-Copyright (c) 2006 - 2016, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2006 - 2018, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution. The full text of the license may be found at\r
{\r
CHAR8 *Destination8;\r
CHAR8 *Source8;\r
- \r
+\r
Destination8 = Buffer;\r
Source8 = (CHAR8 *) ((UINTN) FileHandle + FileOffset);\r
if (Destination8 != Source8) {\r
IN PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r
)\r
{\r
- PEI_CORE_INSTANCE *Private;\r
- VOID* MemoryBuffer;\r
+#if defined (MDE_CPU_IA32) || defined (MDE_CPU_X64)\r
+ PEI_CORE_INSTANCE *Private;\r
+ EFI_PHYSICAL_ADDRESS MemoryBuffer;\r
\r
Private = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer ());\r
- \r
- if (Private->PeiMemoryInstalled && (((Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME) && PcdGetBool (PcdShadowPeimOnBoot)) || \r
- ((Private->HobList.HandoffInformationTable->BootMode == BOOT_ON_S3_RESUME) && PcdGetBool (PcdShadowPeimOnS3Boot))) &&\r
- (EFI_IMAGE_MACHINE_TYPE_SUPPORTED(EFI_IMAGE_MACHINE_X64) || EFI_IMAGE_MACHINE_TYPE_SUPPORTED(EFI_IMAGE_MACHINE_IA32))) {\r
- // \r
- // Shadow algorithm makes lots of non ANSI C assumptions and only works for IA32 and X64 \r
+ MemoryBuffer = 0;\r
+\r
+ if (Private->PeiMemoryInstalled && (((Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME) && PcdGetBool (PcdShadowPeimOnBoot)) ||\r
+ ((Private->HobList.HandoffInformationTable->BootMode == BOOT_ON_S3_RESUME) && PcdGetBool (PcdShadowPeimOnS3Boot)))) {\r
+ //\r
+ // Shadow algorithm makes lots of non ANSI C assumptions and only works for IA32 and X64\r
// compilers that have been tested\r
//\r
if (Private->ShadowedImageRead == NULL) {\r
- MemoryBuffer = AllocatePages (0x400 / EFI_PAGE_SIZE + 1);\r
- ASSERT (MemoryBuffer != NULL);\r
- CopyMem (MemoryBuffer, (CONST VOID *) (UINTN) PeiImageReadForShadow, 0x400);\r
+ PeiServicesAllocatePages (EfiBootServicesCode, 0x400 / EFI_PAGE_SIZE + 1, &MemoryBuffer);\r
+ ASSERT (MemoryBuffer != 0);\r
+ CopyMem ((VOID *)(UINTN)MemoryBuffer, (CONST VOID *) (UINTN) PeiImageReadForShadow, 0x400);\r
Private->ShadowedImageRead = (PE_COFF_LOADER_READ_FILE) (UINTN) MemoryBuffer;\r
}\r
\r
} else {\r
ImageContext->ImageRead = PeiImageRead;\r
}\r
-\r
+#else\r
+ ImageContext->ImageRead = PeiImageRead;\r
+#endif\r
return EFI_SUCCESS;\r
}\r
/**\r
To check memory usage bit map array to figure out if the memory range the image will be loaded in is available or not. If\r
memory range is available, the function will mark the corresponding bits to 1 which indicates the memory range is used.\r
- The function is only invoked when load modules at fixed address feature is enabled. \r
- \r
+ The function is only invoked when load modules at fixed address feature is enabled.\r
+\r
@param Private Pointer to the private data passed in from caller\r
@param ImageBase The base address the image will be loaded at.\r
@param ImageSize The size of the image\r
- \r
+\r
@retval EFI_SUCCESS The memory range the image will be loaded in is available\r
@retval EFI_NOT_FOUND The memory range the image will be loaded in is not available\r
**/\r
UINT32 TopOffsetPageNumber;\r
UINT32 Index;\r
UINT64 *MemoryUsageBitMap;\r
- \r
+\r
\r
//\r
// The reserved code range includes RuntimeCodePage range, Boot time code range and PEI code range.\r
DxeCodePageNumber += PcdGet32(PcdLoadFixAddressRuntimeCodePageNumber);\r
ReservedCodeSize = EFI_PAGES_TO_SIZE(DxeCodePageNumber + PcdGet32(PcdLoadFixAddressPeiCodePageNumber));\r
PeiCodeBase = Private->LoadModuleAtFixAddressTopAddress - ReservedCodeSize;\r
- \r
+\r
//\r
// Test the memory range for loading the image in the PEI code range.\r
//\r
if ((Private->LoadModuleAtFixAddressTopAddress - EFI_PAGES_TO_SIZE(DxeCodePageNumber)) < (ImageBase + ImageSize) ||\r
- (PeiCodeBase > ImageBase)) { \r
- return EFI_NOT_FOUND; \r
+ (PeiCodeBase > ImageBase)) {\r
+ return EFI_NOT_FOUND;\r
}\r
- \r
+\r
//\r
// Test if the memory is avalaible or not.\r
//\r
- MemoryUsageBitMap = Private->PeiCodeMemoryRangeUsageBitMap; \r
+ MemoryUsageBitMap = Private->PeiCodeMemoryRangeUsageBitMap;\r
BaseOffsetPageNumber = EFI_SIZE_TO_PAGES((UINT32)(ImageBase - PeiCodeBase));\r
TopOffsetPageNumber = EFI_SIZE_TO_PAGES((UINT32)(ImageBase + ImageSize - PeiCodeBase));\r
for (Index = BaseOffsetPageNumber; Index < TopOffsetPageNumber; Index ++) {\r
//\r
// This page is already used.\r
//\r
- return EFI_NOT_FOUND; \r
+ return EFI_NOT_FOUND;\r
}\r
}\r
- \r
+\r
//\r
// Being here means the memory range is available. So mark the bits for the memory range\r
- // \r
+ //\r
for (Index = BaseOffsetPageNumber; Index < TopOffsetPageNumber; Index ++) {\r
MemoryUsageBitMap[Index / 64] |= LShiftU64(1, (Index % 64));\r
}\r
- return EFI_SUCCESS; \r
+ return EFI_SUCCESS;\r
}\r
/**\r
\r
UINTN Size;\r
UINT16 NumberOfSections;\r
UINT64 ValueInSectionHeader;\r
- \r
+\r
\r
FixLoadingAddress = 0;\r
Status = EFI_NOT_FOUND;\r
SectionHeaderOffset = sizeof (EFI_TE_IMAGE_HEADER);\r
NumberOfSections = ImgHdr->Te.NumberOfSections;\r
} else {\r
- SectionHeaderOffset = (UINTN)(\r
- ImageContext->PeCoffHeaderOffset +\r
- sizeof (UINT32) +\r
- sizeof (EFI_IMAGE_FILE_HEADER) +\r
- ImgHdr->Pe32.FileHeader.SizeOfOptionalHeader\r
- );\r
+ SectionHeaderOffset = ImageContext->PeCoffHeaderOffset +\r
+ sizeof (UINT32) +\r
+ sizeof (EFI_IMAGE_FILE_HEADER) +\r
+ ImgHdr->Pe32.FileHeader.SizeOfOptionalHeader;\r
NumberOfSections = ImgHdr->Pe32.FileHeader.NumberOfSections;\r
}\r
//\r
\r
@retval EFI_SUCCESS The file was loaded and relocated\r
@retval EFI_OUT_OF_RESOURCES There was not enough memory to load and relocate the PE/COFF file\r
- @retval EFI_WARN_BUFFER_TOO_SMALL \r
+ @retval EFI_WARN_BUFFER_TOO_SMALL\r
There is not enough heap to allocate the requested size.\r
This will not prevent the XIP image from being invoked.\r
\r
if (EFI_ERROR (Status)) {\r
return Status;\r
}\r
- \r
+\r
//\r
// Initilize local IsS3Boot and IsRegisterForShadow variable\r
//\r
IsS3Boot = TRUE;\r
}\r
IsRegisterForShadow = FALSE;\r
- if ((Private->CurrentFileHandle == FileHandle) \r
- && (Private->Fv[Private->CurrentPeimFvCount].PeimState[Private->CurrentPeimCount] == PEIM_STATE_REGISITER_FOR_SHADOW)) {\r
+ if ((Private->CurrentFileHandle == FileHandle)\r
+ && (Private->Fv[Private->CurrentPeimFvCount].PeimState[Private->CurrentPeimCount] == PEIM_STATE_REGISTER_FOR_SHADOW)) {\r
IsRegisterForShadow = TRUE;\r
}\r
\r
//\r
Status = PeiServicesFfsGetFileInfo (FileHandle, &FileInfo);\r
ASSERT_EFI_ERROR (Status);\r
- \r
+\r
//\r
// Check whether the file type is PEI module.\r
//\r
IsPeiModule = FALSE;\r
- if (FileInfo.FileType == EFI_FV_FILETYPE_PEI_CORE || \r
- FileInfo.FileType == EFI_FV_FILETYPE_PEIM || \r
+ if (FileInfo.FileType == EFI_FV_FILETYPE_PEI_CORE ||\r
+ FileInfo.FileType == EFI_FV_FILETYPE_PEIM ||\r
FileInfo.FileType == EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER) {\r
IsPeiModule = TRUE;\r
}\r
//\r
// The PEIM is not assiged valid address, try to allocate page to load it.\r
//\r
- ImageContext.ImageAddress = (EFI_PHYSICAL_ADDRESS)(UINTN) AllocatePages (EFI_SIZE_TO_PAGES ((UINT32) AlignImageSize));\r
+ Status = PeiServicesAllocatePages (EfiBootServicesCode,\r
+ EFI_SIZE_TO_PAGES ((UINT32) AlignImageSize),\r
+ &ImageContext.ImageAddress);\r
}\r
} else {\r
- ImageContext.ImageAddress = (EFI_PHYSICAL_ADDRESS)(UINTN) AllocatePages (EFI_SIZE_TO_PAGES ((UINT32) AlignImageSize));\r
+ Status = PeiServicesAllocatePages (EfiBootServicesCode,\r
+ EFI_SIZE_TO_PAGES ((UINT32) AlignImageSize),\r
+ &ImageContext.ImageAddress);\r
}\r
- if (ImageContext.ImageAddress != 0) {\r
+ if (!EFI_ERROR (Status)) {\r
//\r
// Adjust the Image Address to make sure it is section alignment.\r
//\r
//\r
Status = PeCoffLoaderLoadImage (&ImageContext);\r
if (EFI_ERROR (Status)) {\r
+ if (ImageContext.ImageError == IMAGE_ERROR_INVALID_SECTION_ALIGNMENT) {\r
+ DEBUG ((DEBUG_ERROR, "PEIM Image Address 0x%11p doesn't meet with section alignment 0x%x.\n", (VOID*)(UINTN)ImageContext.ImageAddress, ImageContext.SectionAlignment));\r
+ }\r
return Status;\r
}\r
//\r
@retval EFI_SUCCESS Image is successfully loaded.\r
@retval EFI_NOT_FOUND Fail to locate necessary PPI.\r
@retval EFI_UNSUPPORTED Image Machine Type is not supported.\r
- @retval EFI_WARN_BUFFER_TOO_SMALL \r
+ @retval EFI_WARN_BUFFER_TOO_SMALL\r
There is not enough heap to allocate the requested size.\r
This will not prevent the XIP image from being invoked.\r
\r
}\r
}\r
\r
+ DEBUG ((DEBUG_INFO, "Loading PEIM %g\n", FileHandle));\r
+\r
//\r
// If memory is installed, perform the shadow operations\r
//\r
//\r
// The shadowed PEIM must be relocatable.\r
//\r
- if (PeimState == PEIM_STATE_REGISITER_FOR_SHADOW) {\r
+ if (PeimState == PEIM_STATE_REGISTER_FOR_SHADOW) {\r
IsStrip = RelocationIsStrip ((VOID *) (UINTN) ImageAddress);\r
ASSERT (!IsStrip);\r
if (IsStrip) {\r