]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Core/Pei/Image/Image.c
MdeModulePkg PeiCore: Always use PeiImageRead() function to load PEI image
[mirror_edk2.git] / MdeModulePkg / Core / Pei / Image / Image.c
index f07f48823117aa0373f7f6fb9f8abdac7819340e..315b551bfdb7f56b61761a8601c6caf181eb4338 100644 (file)
@@ -50,106 +50,25 @@ PeiImageRead (
 {\r
   CHAR8 *Destination8;\r
   CHAR8 *Source8;\r
-  \r
-  Destination8  = Buffer;\r
-  Source8       = (CHAR8 *) ((UINTN) FileHandle + FileOffset);\r
-  if (Destination8 != Source8) {\r
-    CopyMem (Destination8, Source8, *ReadSize);\r
-  }\r
-\r
-  return EFI_SUCCESS;\r
-}\r
-\r
-/**\r
-\r
-  Support routine for the PE/COFF Loader that reads a buffer from a PE/COFF file.\r
-  The function is implemented as PIC so as to support shadowing.\r
-\r
-  @param FileHandle      - The handle to the PE/COFF file\r
-  @param FileOffset      - The offset, in bytes, into the file to read\r
-  @param ReadSize        - The number of bytes to read from the file starting at FileOffset\r
-  @param Buffer          - A pointer to the buffer to read the data into.\r
-\r
-  @return EFI_SUCCESS - ReadSize bytes of data were read into Buffer from the PE/COFF file starting at FileOffset\r
-\r
-**/\r
-EFI_STATUS\r
-EFIAPI\r
-PeiImageReadForShadow (\r
-  IN     VOID    *FileHandle,\r
-  IN     UINTN   FileOffset,\r
-  IN     UINTN   *ReadSize,\r
-  OUT    VOID    *Buffer\r
-  )\r
-{\r
-  volatile CHAR8  *Destination8;\r
-  CHAR8           *Source8;\r
-  UINTN           Length;\r
 \r
   Destination8  = Buffer;\r
   Source8       = (CHAR8 *) ((UINTN) FileHandle + FileOffset);\r
   if (Destination8 != Source8) {\r
-    Length        = *ReadSize;\r
-    while ((Length--) > 0) {\r
-      *(Destination8++) = *(Source8++);\r
-    }\r
+    CopyMem (Destination8, Source8, *ReadSize);\r
   }\r
 \r
   return EFI_SUCCESS;\r
 }\r
 \r
-/**\r
-\r
-  Support routine to get the Image read file function.\r
-\r
-  @param ImageContext    - The context of the image being loaded\r
-\r
-  @retval EFI_SUCCESS - If Image function location is found\r
-\r
-**/\r
-EFI_STATUS\r
-GetImageReadFunction (\r
-  IN      PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext\r
-  )\r
-{\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
-  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
-      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
-    ImageContext->ImageRead = Private->ShadowedImageRead;\r
-  } else {\r
-    ImageContext->ImageRead = PeiImageRead;\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
@@ -167,7 +86,7 @@ CheckAndMarkFixLoadingMemoryUsageBitMap (
    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
@@ -176,19 +95,19 @@ CheckAndMarkFixLoadingMemoryUsageBitMap (
    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
@@ -196,17 +115,17 @@ CheckAndMarkFixLoadingMemoryUsageBitMap (
        //\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
@@ -236,7 +155,7 @@ GetPeCoffImageFixLoadingAssignedAddress(
    UINTN                              Size;\r
    UINT16                             NumberOfSections;\r
    UINT64                             ValueInSectionHeader;\r
\r
+\r
 \r
    FixLoadingAddress = 0;\r
    Status = EFI_NOT_FOUND;\r
@@ -338,7 +257,7 @@ GetPeCoffImageFixLoadingAssignedAddress(
 \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
@@ -369,15 +288,13 @@ LoadAndRelocatePeCoffImage (
   IsXipImage   = FALSE;\r
   ZeroMem (&ImageContext, sizeof (ImageContext));\r
   ImageContext.Handle = Pe32Data;\r
-  Status              = GetImageReadFunction (&ImageContext);\r
-\r
-  ASSERT_EFI_ERROR (Status);\r
+  ImageContext.ImageRead = PeiImageRead;\r
 \r
   Status = PeCoffLoaderGetImageInfo (&ImageContext);\r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
-  \r
+\r
   //\r
   // Initilize local IsS3Boot and IsRegisterForShadow variable\r
   //\r
@@ -386,7 +303,7 @@ LoadAndRelocatePeCoffImage (
     IsS3Boot = TRUE;\r
   }\r
   IsRegisterForShadow = FALSE;\r
-  if ((Private->CurrentFileHandle == FileHandle) \r
+  if ((Private->CurrentFileHandle == FileHandle)\r
     && (Private->Fv[Private->CurrentPeimFvCount].PeimState[Private->CurrentPeimCount] == PEIM_STATE_REGISTER_FOR_SHADOW)) {\r
     IsRegisterForShadow = TRUE;\r
   }\r
@@ -403,13 +320,13 @@ LoadAndRelocatePeCoffImage (
   //\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
@@ -548,7 +465,7 @@ LoadAndRelocatePeCoffImage (
   @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