878ddf1f |
1 | /*++\r |
2 | \r |
a8b194b9 |
3 | Copyright (c) 2006 - 2007, Intel Corporation\r |
878ddf1f |
4 | All rights reserved. This program and the accompanying materials\r |
5 | are licensed and made available under the terms and conditions of the BSD License\r |
6 | which accompanies this distribution. The full text of the license may be found at\r |
7 | http://opensource.org/licenses/bsd-license.php\r |
8 | \r |
9 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r |
10 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r |
11 | \r |
12 | Module Name:\r |
13 | \r |
14 | PeCoffGetEntryPoint.c\r |
15 | \r |
16 | Abstract:\r |
17 | \r |
18 | Tiano PE/COFF loader\r |
19 | \r |
20 | Revision History\r |
21 | \r |
22 | --*/\r |
23 | \r |
24 | \r |
25 | RETURN_STATUS\r |
26 | EFIAPI\r |
27 | PeCoffLoaderGetEntryPoint (\r |
28 | IN VOID *Pe32Data,\r |
29 | IN OUT VOID **EntryPoint\r |
30 | )\r |
31 | /*++\r |
32 | \r |
33 | Routine Description:\r |
34 | \r |
a8b194b9 |
35 | Loads a PE/COFF image into memory, this is not follow the original purpose of \r |
36 | PeCoffGetEntryPoint library class. But it's ok that Unix package not run on a real \r |
37 | platform and this is for source level debug.\r |
878ddf1f |
38 | \r |
39 | Arguments:\r |
40 | \r |
41 | Pe32Data - Pointer to a PE/COFF Image\r |
42 | \r |
43 | EntryPoint - Pointer to the entry point of the PE/COFF image\r |
44 | \r |
45 | Returns:\r |
46 | \r |
47 | EFI_SUCCESS if the EntryPoint was returned\r |
48 | EFI_INVALID_PARAMETER if the EntryPoint could not be found from Pe32Data\r |
49 | \r |
50 | --*/\r |
51 | {\r |
52 | EFI_STATUS Status;\r |
53 | EFI_PEI_PPI_DESCRIPTOR *PpiDescriptor;\r |
54 | NT_PEI_LOAD_FILE_PPI *PeiNtService;\r |
55 | EFI_PHYSICAL_ADDRESS ImageAddress;\r |
56 | UINT64 ImageSize;\r |
57 | EFI_PHYSICAL_ADDRESS ImageEntryPoint;\r |
58 | \r |
84a99d48 |
59 | Status = PeiServicesLocatePpi (\r |
878ddf1f |
60 | &gNtPeiLoadFilePpiGuid,\r |
61 | 0,\r |
62 | &PpiDescriptor,\r |
63 | &PeiNtService\r |
64 | );\r |
65 | if (EFI_ERROR (Status)) {\r |
66 | return Status;\r |
67 | }\r |
68 | \r |
69 | Status = PeiNtService->PeiLoadFileService (\r |
70 | Pe32Data,\r |
71 | &ImageAddress,\r |
72 | &ImageSize,\r |
73 | &ImageEntryPoint\r |
74 | );\r |
75 | *EntryPoint = (VOID*)(UINTN)ImageEntryPoint;\r |
76 | return Status;\r |
a8b194b9 |
77 | }\r |
78 | \r |
79 | /**\r |
80 | Returns the machine type of PE/COFF image. \r |
81 | This is copied from MDE BasePeCoffGetEntryPointLib, the code should be sync with it.\r |
82 | The reason is NT32 package needs to load the image to memory to support source\r |
83 | level debug.\r |
84 | \r |
85 | \r |
86 | @param Image Pointer to a PE/COFF header\r |
87 | \r |
88 | @return Machine type or zero if not a valid iamge\r |
89 | \r |
90 | **/\r |
91 | UINT16\r |
92 | EFIAPI\r |
93 | PeCoffLoaderGetMachineType (\r |
94 | IN VOID *Pe32Data\r |
95 | )\r |
96 | { \r |
97 | EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr;\r |
98 | EFI_IMAGE_DOS_HEADER *DosHdr;\r |
99 | \r |
100 | DosHdr = (EFI_IMAGE_DOS_HEADER *)Pe32Data;\r |
101 | if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r |
102 | Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN)Pe32Data + DosHdr->e_lfanew);\r |
103 | } else {\r |
104 | Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN)Pe32Data);\r |
105 | }\r |
106 | \r |
107 | if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE) {\r |
108 | return Hdr.Pe32->FileHeader.Machine;\r |
109 | }\r |
110 | \r |
111 | return 0x0000;\r |
112 | }\r |
113 | \r |