]>
Commit | Line | Data |
---|---|---|
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 | |
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 | |
22 | \r | |
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 | |
28 | \r | |
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 | |
34 | \r | |
35 | **/\r | |
36 | RETURN_STATUS\r | |
37 | EFIAPI\r | |
38 | PeCoffLoaderGetEntryPoint (\r | |
39 | IN VOID *Pe32Data,\r | |
40 | OUT VOID **EntryPoint\r | |
41 | )\r | |
42 | {\r | |
43 | EFI_IMAGE_DOS_HEADER *DosHeader;\r | |
44 | EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Header;\r | |
45 | \r | |
46 | ASSERT (Pe32Data != NULL);\r | |
47 | ASSERT (EntryPoint != NULL);\r | |
48 | \r | |
49 | DosHeader = (EFI_IMAGE_DOS_HEADER *)Pe32Data;\r | |
50 | if (DosHeader->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r | |
51 | //\r | |
52 | // DOS image header is present, so read the PE header after the DOS image header.\r | |
53 | //\r | |
54 | Header.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINTN) Pe32Data + (UINTN) ((DosHeader->e_lfanew) & 0x0ffff));\r | |
55 | } else {\r | |
56 | //\r | |
57 | // DOS image header is not present, so PE header is at the image base.\r | |
58 | //\r | |
59 | Header.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)Pe32Data;\r | |
60 | }\r | |
61 | \r | |
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 | |
67 | return RETURN_SUCCESS;\r | |
68 | }\r | |
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 |