EdkModulePkg/Core/Pei/Image/Image.c

   1 /*++
   2
   3 Copyright (c) 2006, Intel Corporation
   4 All rights reserved. This program and the accompanying materials
   5 are licensed and made available under the terms and conditions of the BSD License
   6 which accompanies this distribution.  The full text of the license may be found at
   7 http://opensource.org/licenses/bsd-license.php
   8
   9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  11
  12 Module Name:
  13
  14   Image.c
  15
  16 Abstract:
  17
  18   Pei Core Load Image Support
  19
  20 --*/
  21
  22 #include <PeiMain.h>
  23
  24
  25
  26 EFI_STATUS
  27 PeiLoadImage (
  28   IN EFI_PEI_SERVICES         **PeiServices,
  29   IN EFI_FFS_FILE_HEADER      *PeimFileHeader,
  30   OUT VOID                    **EntryPoint
  31   )
  32 /*++
  33
  34 Routine Description:
  35
  36   Routine for loading file image.
  37
  38 Arguments:
  39
  40   PeiServices     - The PEI core services table.
  41   PeimFileHeader  - Pointer to the FFS file header of the image.
  42   EntryPoint      - Pointer to entry point of specified image file for output.
  43
  44 Returns:
  45
  46   Status - EFI_SUCCESS    - Image is successfully loaded.
  47            EFI_NOT_FOUND  - Fail to locate necessary PPI
  48            Others         - Fail to load file.
  49
  50 --*/
  51 {
  52   EFI_STATUS                  Status;
  53   VOID                        *Pe32Data;
  54   EFI_PEI_FV_FILE_LOADER_PPI  *FvLoadFilePpi;
  55   EFI_PHYSICAL_ADDRESS        ImageAddress;
  56   UINT64                      ImageSize;
  57   EFI_PHYSICAL_ADDRESS        ImageEntryPoint;
  58   EFI_TE_IMAGE_HEADER         *TEImageHeader;
  59
  60   *EntryPoint   = NULL;
  61   TEImageHeader = NULL;
  62
  63   //
  64   // Try to find a PE32 section.
  65   //
  66   Status = PeiServicesFfsFindSectionData (
  67              EFI_SECTION_PE32,
  68              PeimFileHeader,
  69              &Pe32Data
  70              );
  71   //
  72   // If we didn't find a PE32 section, try to find a TE section.
  73   //
  74   if (EFI_ERROR (Status)) {
  75     Status = PeiServicesFfsFindSectionData (
  76                EFI_SECTION_TE,
  77                PeimFileHeader,
  78                (VOID **) &TEImageHeader
  79                );
  80     if (EFI_ERROR (Status) || TEImageHeader == NULL) {
  81       //
  82       // There was not a PE32 or a TE section, so assume that it's a Compressed section
  83       // and use the LoadFile
  84       //
  85       Status = PeiServicesLocatePpi (
  86                 &gEfiPeiFvFileLoaderPpiGuid,
  87                 0,
  88                 NULL,
  89                 (VOID **)&FvLoadFilePpi
  90                 );
  91       if (EFI_ERROR (Status)) {
  92         return EFI_NOT_FOUND;
  93       }
  94
  95       Status = FvLoadFilePpi->FvLoadFile (
  96                                 FvLoadFilePpi,
  97                                 PeimFileHeader,
  98                                 &ImageAddress,
  99                                 &ImageSize,
 100                                 &ImageEntryPoint
 101                                 );
 102
 103       if (EFI_ERROR (Status)) {
 104         return EFI_NOT_FOUND;
 105       }
 106
 107       //
 108       // Got the entry point from ImageEntryPoint
 109       //
 110       *EntryPoint = (VOID *) ((UINTN) ImageEntryPoint);
 111       return EFI_SUCCESS;
 112     } else {
 113       //
 114       // Retrieve the entry point from the TE image header
 115       //
 116       ImageAddress = (EFI_PHYSICAL_ADDRESS) (UINTN) TEImageHeader;
 117       *EntryPoint = (VOID *)((UINTN) TEImageHeader + sizeof (EFI_TE_IMAGE_HEADER) +
 118                     TEImageHeader->AddressOfEntryPoint - TEImageHeader->StrippedSize);
 119     }
 120   } else {
 121     //
 122     // Retrieve the entry point from the PE/COFF image header
 123     //
 124     ImageAddress = (EFI_PHYSICAL_ADDRESS) (UINTN) Pe32Data;
 125     Status = PeCoffLoaderGetEntryPoint (Pe32Data, EntryPoint);
 126     if (EFI_ERROR (Status)) {
 127       return EFI_NOT_FOUND;
 128     }
 129   }
 130
 131   //
 132   // Print debug message: Loading PEIM at 0x12345678 EntryPoint=0x12345688 Driver.efi
 133   //
 134   DEBUG ((EFI_D_INFO | EFI_D_LOAD, "Loading PEIM at 0x%08x EntryPoint=0x%08x ", Pe32Data, *EntryPoint));
 135   DEBUG_CODE_BEGIN ();
 136     PE_COFF_LOADER_IMAGE_CONTEXT          ImageContext;
 137     UINTN                                 Index;
 138     CHAR8                                 *PdbStr;
 139     CHAR8                                 AsciiBuffer[512];
 140
 141
 142     ZeroMem (&ImageContext, sizeof (ImageContext));
 143     ImageContext.Handle    = Pe32Data;
 144     ImageContext.ImageRead = PeCoffLoaderImageReadFromMemory;
 145
 146     PeCoffLoaderGetImageInfo (&ImageContext);
 147
 148     if (ImageContext.PdbPointer != NULL) {
 149       //
 150       // Copy PDB pointer to AsciiBuffer and replace .PDB with .EFI
 151       //
 152       PdbStr = ImageContext.PdbPointer;
 153       for (Index = 0; PdbStr != 0; Index++, PdbStr++) {
 154         AsciiBuffer[Index] = *PdbStr;
 155         if (*PdbStr == '.') {
 156           AsciiBuffer[Index] = '\0';
 157         }
 158       }
 159
 160       DEBUG ((EFI_D_INFO | EFI_D_LOAD, "%a.efi", AsciiBuffer));
 161     }
 162
 163   DEBUG_CODE_END ();
 164
 165   DEBUG ((EFI_D_INFO | EFI_D_LOAD, "\n"));
 166
 167   return EFI_SUCCESS;
 168 }