]>
Commit | Line | Data |
---|---|---|
878ddf1f | 1 | /** @file\r |
2 | Tiano PE/COFF loader.\r | |
3 | \r | |
4 | Copyright (c) 2006, Intel Corporation<BR>\r | |
5 | All rights reserved. This program and the accompanying materials\r | |
6 | are licensed and made available under the terms and conditions of the BSD License\r | |
7 | which accompanies this distribution. The full text of the license may be found at\r | |
8 | http://opensource.org/licenses/bsd-license.php\r | |
9 | \r | |
10 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r | |
11 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r | |
12 | \r | |
13 | Module Name: PeCoffGetEntryPoint.c\r | |
14 | \r | |
15 | **/\r | |
16 | \r | |
17 | \r | |
18 | \r | |
19 | /**\r | |
cd14fe3d | 20 | Retrieves and returns a pointer to the entry point to a PE/COFF image that has been loaded\r |
21 | into system memory with the PE/COFF Loader Library functions.\r | |
878ddf1f | 22 | \r |
cd14fe3d | 23 | Retrieves the entry point to the PE/COFF image specified by Pe32Data and returns this entry\r |
24 | point in EntryPoint. If the entry point could not be retrieved from the PE/COFF image, then\r | |
25 | return RETURN_INVALID_PARAMETER. Otherwise return RETURN_SUCCESS.\r | |
26 | If Pe32Data is NULL, then ASSERT().\r | |
27 | If EntryPoint is NULL, then ASSERT().\r | |
878ddf1f | 28 | \r |
cd14fe3d | 29 | @param Pe32Data Pointer to the PE/COFF image that is loaded in system memory.\r |
30 | @param EntryPoint Pointer to entry point to the PE/COFF image to return.\r | |
31 | \r | |
32 | @retval RETURN_SUCCESS EntryPoint was returned.\r | |
33 | @retval RETURN_INVALID_PARAMETER The entry point could not be found in the PE/COFF image.\r | |
878ddf1f | 34 | \r |
35 | **/\r | |
36 | RETURN_STATUS\r | |
37 | EFIAPI\r | |
38 | PeCoffLoaderGetEntryPoint (\r | |
cd14fe3d | 39 | IN VOID *Pe32Data,\r |
40 | OUT VOID **EntryPoint\r | |
878ddf1f | 41 | )\r |
42 | {\r | |
2ce31132 | 43 | EFI_IMAGE_DOS_HEADER *DosHeader;\r |
44 | EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Header;\r | |
878ddf1f | 45 | \r |
cd14fe3d | 46 | ASSERT (Pe32Data != NULL);\r |
47 | ASSERT (EntryPoint != NULL);\r | |
48 | \r | |
878ddf1f | 49 | DosHeader = (EFI_IMAGE_DOS_HEADER *)Pe32Data;\r |
50 | if (DosHeader->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r | |
51 | //\r | |
cd14fe3d | 52 | // DOS image header is present, so read the PE header after the DOS image header.\r |
878ddf1f | 53 | //\r |
2ce31132 | 54 | Header.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN) Pe32Data + (UINTN) ((DosHeader->e_lfanew) & 0x0ffff));\r |
878ddf1f | 55 | } else {\r |
56 | //\r | |
cd14fe3d | 57 | // DOS image header is not present, so PE header is at the image base.\r |
878ddf1f | 58 | //\r |
2ce31132 | 59 | Header.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)Pe32Data;\r |
878ddf1f | 60 | }\r |
cd14fe3d | 61 | \r |
2ce31132 | 62 | //\r |
63 | // Calculate the entry point relative to the start of the image. \r | |
64 | // AddressOfEntryPoint is common for PE32 & PE32+\r | |
65 | //\r | |
66 | *EntryPoint = (VOID *)((UINTN)Pe32Data + (UINTN)(Header.Pe32->OptionalHeader.AddressOfEntryPoint & 0x0ffffffff));\r | |
878ddf1f | 67 | return RETURN_SUCCESS;\r |
68 | }\r | |
2ce31132 | 69 | \r |
70 | \r | |
71 | /**\r | |
72 | Returns the machine type of PE/COFF image. \r | |
73 | \r | |
74 | @param Image Pointer to a PE/COFF header\r | |
75 | \r | |
76 | @return Machine type or zero if not a valid iamge\r | |
77 | \r | |
78 | **/\r | |
79 | UINT16\r | |
80 | EFIAPI\r | |
81 | PeCoffLoaderGetMachineType (\r | |
82 | IN VOID *Pe32Data\r | |
83 | )\r | |
84 | {\r | |
85 | EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr;\r | |
86 | EFI_IMAGE_DOS_HEADER *DosHdr;\r | |
87 | \r | |
88 | DosHdr = (EFI_IMAGE_DOS_HEADER *)Pe32Data;\r | |
89 | if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r | |
90 | Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN)Pe32Data + DosHdr->e_lfanew);\r | |
91 | } else {\r | |
92 | Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN)Pe32Data);\r | |
93 | }\r | |
94 | \r | |
95 | if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE) {\r | |
96 | return Hdr.Pe32->FileHeader.Machine;\r | |
97 | }\r | |
98 | \r | |
99 | return 0x0000;\r | |
100 | }\r | |
101 | \r | |
102 | \r |