EdkNt32Pkg/Library/EdkNt32PeiPeCoffGetEntryPointLib/PeCoffGetEntryPoint.c

   1 /*++
   2
   3 Copyright (c) 2006 - 2007, 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   PeCoffGetEntryPoint.c
  15
  16 Abstract:
  17
  18   Tiano PE/COFF loader
  19
  20 Revision History
  21
  22 --*/
  23
  24
  25 RETURN_STATUS
  26 EFIAPI
  27 PeCoffLoaderGetEntryPoint (
  28   IN     VOID  *Pe32Data,
  29   IN OUT VOID  **EntryPoint
  30   )
  31 /*++
  32
  33 Routine Description:
  34
  35   Loads a PE/COFF image into memory, this is not follow the original purpose of
  36   PeCoffGetEntryPoint library class.  But it's ok that Unix package not run on a real
  37   platform and this is for source level debug.
  38
  39 Arguments:
  40
  41   Pe32Data   - Pointer to a PE/COFF Image
  42
  43   EntryPoint - Pointer to the entry point of the PE/COFF image
  44
  45 Returns:
  46
  47   EFI_SUCCESS            if the EntryPoint was returned
  48   EFI_INVALID_PARAMETER  if the EntryPoint could not be found from Pe32Data
  49
  50 --*/
  51 {
  52   EFI_STATUS              Status;
  53   EFI_PEI_PPI_DESCRIPTOR  *PpiDescriptor;
  54   NT_PEI_LOAD_FILE_PPI    *PeiNtService;
  55   EFI_PHYSICAL_ADDRESS    ImageAddress;
  56   UINT64                  ImageSize;
  57   EFI_PHYSICAL_ADDRESS    ImageEntryPoint;
  58
  59   Status = PeiServicesLocatePpi (
  60              &gNtPeiLoadFilePpiGuid,
  61              0,
  62              &PpiDescriptor,
  63              &PeiNtService
  64              );
  65   if (EFI_ERROR (Status)) {
  66     return Status;
  67   }
  68
  69   Status = PeiNtService->PeiLoadFileService (
  70                            Pe32Data,
  71                            &ImageAddress,
  72                            &ImageSize,
  73                            &ImageEntryPoint
  74                            );
  75   *EntryPoint = (VOID*)(UINTN)ImageEntryPoint;
  76   return Status;
  77 }
  78
  79 /**
  80   Returns the machine type of PE/COFF image.
  81   This is copied from MDE BasePeCoffGetEntryPointLib, the code should be sync with it.
  82   The reason is NT32 package needs to load the image to memory to support source
  83   level debug.
  84
  85
  86   @param  Image   Pointer to a PE/COFF header
  87
  88   @return         Machine type or zero if not a valid iamge
  89
  90 **/
  91 UINT16
  92 EFIAPI
  93 PeCoffLoaderGetMachineType (
  94   IN  VOID  *Pe32Data
  95   )
  96 {
  97   EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION  Hdr;
  98   EFI_IMAGE_DOS_HEADER                 *DosHdr;
  99
 100   DosHdr = (EFI_IMAGE_DOS_HEADER  *)Pe32Data;
 101   if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {
 102     Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN)Pe32Data + DosHdr->e_lfanew);
 103   } else {
 104     Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN)Pe32Data);
 105   }
 106
 107   if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE)  {
 108     return Hdr.Pe32->FileHeader.Machine;
 109   }
 110
 111   return 0x0000;
 112 }
 113