Split Nt32OemHookStatusCodeLib to PEI/DXE instances
[mirror_edk2.git] / 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