]>
Commit | Line | Data |
---|---|---|
cb149c66 AG |
1 | /* |
2 | * EFI image loader | |
3 | * | |
4 | * based partly on wine code | |
5 | * | |
6 | * Copyright (c) 2016 Alexander Graf | |
7 | * | |
8 | * SPDX-License-Identifier: GPL-2.0+ | |
9 | */ | |
10 | ||
11 | #include <common.h> | |
12 | #include <efi_loader.h> | |
13 | #include <pe.h> | |
14 | #include <asm/global_data.h> | |
15 | ||
16 | DECLARE_GLOBAL_DATA_PTR; | |
17 | ||
18 | const efi_guid_t efi_guid_device_path = DEVICE_PATH_GUID; | |
19 | const efi_guid_t efi_guid_loaded_image = LOADED_IMAGE_GUID; | |
20 | ||
da684a64 | 21 | static efi_status_t efi_loader_relocate(const IMAGE_BASE_RELOCATION *rel, |
cb149c66 AG |
22 | unsigned long rel_size, void *efi_reloc) |
23 | { | |
24 | const IMAGE_BASE_RELOCATION *end; | |
25 | int i; | |
26 | ||
27 | end = (const IMAGE_BASE_RELOCATION *)((const char *)rel + rel_size); | |
28 | while (rel < end - 1 && rel->SizeOfBlock) { | |
29 | const uint16_t *relocs = (const uint16_t *)(rel + 1); | |
30 | i = (rel->SizeOfBlock - sizeof(*rel)) / sizeof(uint16_t); | |
31 | while (i--) { | |
b1237c6e | 32 | uint32_t offset = (uint32_t)(*relocs & 0xfff) + |
cb149c66 AG |
33 | rel->VirtualAddress; |
34 | int type = *relocs >> EFI_PAGE_SHIFT; | |
35 | unsigned long delta = (unsigned long)efi_reloc; | |
36 | uint64_t *x64 = efi_reloc + offset; | |
37 | uint32_t *x32 = efi_reloc + offset; | |
38 | uint16_t *x16 = efi_reloc + offset; | |
39 | ||
40 | switch (type) { | |
41 | case IMAGE_REL_BASED_ABSOLUTE: | |
42 | break; | |
43 | case IMAGE_REL_BASED_HIGH: | |
44 | *x16 += ((uint32_t)delta) >> 16; | |
45 | break; | |
46 | case IMAGE_REL_BASED_LOW: | |
47 | *x16 += (uint16_t)delta; | |
48 | break; | |
49 | case IMAGE_REL_BASED_HIGHLOW: | |
50 | *x32 += (uint32_t)delta; | |
51 | break; | |
52 | case IMAGE_REL_BASED_DIR64: | |
53 | *x64 += (uint64_t)delta; | |
54 | break; | |
55 | default: | |
56 | printf("Unknown Relocation off %x type %x\n", | |
57 | offset, type); | |
da684a64 | 58 | return EFI_LOAD_ERROR; |
cb149c66 AG |
59 | } |
60 | relocs++; | |
61 | } | |
62 | rel = (const IMAGE_BASE_RELOCATION *)relocs; | |
63 | } | |
da684a64 | 64 | return EFI_SUCCESS; |
cb149c66 AG |
65 | } |
66 | ||
67 | void __weak invalidate_icache_all(void) | |
68 | { | |
69 | /* If the system doesn't support icache_all flush, cross our fingers */ | |
70 | } | |
71 | ||
72 | /* | |
73 | * This function loads all sections from a PE binary into a newly reserved | |
74 | * piece of memory. On successful load it then returns the entry point for | |
75 | * the binary. Otherwise NULL. | |
76 | */ | |
77 | void *efi_load_pe(void *efi, struct efi_loaded_image *loaded_image_info) | |
78 | { | |
79 | IMAGE_NT_HEADERS32 *nt; | |
80 | IMAGE_DOS_HEADER *dos; | |
81 | IMAGE_SECTION_HEADER *sections; | |
82 | int num_sections; | |
83 | void *efi_reloc; | |
84 | int i; | |
85 | const IMAGE_BASE_RELOCATION *rel; | |
86 | unsigned long rel_size; | |
87 | int rel_idx = IMAGE_DIRECTORY_ENTRY_BASERELOC; | |
88 | void *entry; | |
89 | uint64_t image_size; | |
90 | unsigned long virt_size = 0; | |
91 | bool can_run_nt64 = true; | |
92 | bool can_run_nt32 = true; | |
93 | ||
94 | #if defined(CONFIG_ARM64) | |
95 | can_run_nt32 = false; | |
96 | #elif defined(CONFIG_ARM) | |
97 | can_run_nt64 = false; | |
98 | #endif | |
99 | ||
100 | dos = efi; | |
101 | if (dos->e_magic != IMAGE_DOS_SIGNATURE) { | |
102 | printf("%s: Invalid DOS Signature\n", __func__); | |
103 | return NULL; | |
104 | } | |
105 | ||
106 | nt = (void *) ((char *)efi + dos->e_lfanew); | |
107 | if (nt->Signature != IMAGE_NT_SIGNATURE) { | |
108 | printf("%s: Invalid NT Signature\n", __func__); | |
109 | return NULL; | |
110 | } | |
111 | ||
112 | /* Calculate upper virtual address boundary */ | |
113 | num_sections = nt->FileHeader.NumberOfSections; | |
114 | sections = (void *)&nt->OptionalHeader + | |
115 | nt->FileHeader.SizeOfOptionalHeader; | |
116 | ||
117 | for (i = num_sections - 1; i >= 0; i--) { | |
118 | IMAGE_SECTION_HEADER *sec = §ions[i]; | |
119 | virt_size = max_t(unsigned long, virt_size, | |
120 | sec->VirtualAddress + sec->Misc.VirtualSize); | |
121 | } | |
122 | ||
123 | /* Read 32/64bit specific header bits */ | |
124 | if (can_run_nt64 && | |
125 | (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)) { | |
126 | IMAGE_NT_HEADERS64 *nt64 = (void *)nt; | |
127 | IMAGE_OPTIONAL_HEADER64 *opt = &nt64->OptionalHeader; | |
128 | image_size = opt->SizeOfImage; | |
129 | efi_reloc = efi_alloc(virt_size, EFI_LOADER_DATA); | |
130 | if (!efi_reloc) { | |
131 | printf("%s: Could not allocate %ld bytes\n", | |
132 | __func__, virt_size); | |
133 | return NULL; | |
134 | } | |
135 | entry = efi_reloc + opt->AddressOfEntryPoint; | |
136 | rel_size = opt->DataDirectory[rel_idx].Size; | |
137 | rel = efi_reloc + opt->DataDirectory[rel_idx].VirtualAddress; | |
138 | } else if (can_run_nt32 && | |
139 | (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC)) { | |
140 | IMAGE_OPTIONAL_HEADER32 *opt = &nt->OptionalHeader; | |
141 | image_size = opt->SizeOfImage; | |
142 | efi_reloc = efi_alloc(virt_size, EFI_LOADER_DATA); | |
143 | if (!efi_reloc) { | |
144 | printf("%s: Could not allocate %ld bytes\n", | |
145 | __func__, virt_size); | |
146 | return NULL; | |
147 | } | |
148 | entry = efi_reloc + opt->AddressOfEntryPoint; | |
149 | rel_size = opt->DataDirectory[rel_idx].Size; | |
150 | rel = efi_reloc + opt->DataDirectory[rel_idx].VirtualAddress; | |
151 | } else { | |
152 | printf("%s: Invalid optional header magic %x\n", __func__, | |
153 | nt->OptionalHeader.Magic); | |
154 | return NULL; | |
155 | } | |
156 | ||
157 | /* Load sections into RAM */ | |
158 | for (i = num_sections - 1; i >= 0; i--) { | |
159 | IMAGE_SECTION_HEADER *sec = §ions[i]; | |
160 | memset(efi_reloc + sec->VirtualAddress, 0, | |
161 | sec->Misc.VirtualSize); | |
162 | memcpy(efi_reloc + sec->VirtualAddress, | |
163 | efi + sec->PointerToRawData, | |
164 | sec->SizeOfRawData); | |
165 | } | |
166 | ||
167 | /* Run through relocations */ | |
da684a64 | 168 | if (efi_loader_relocate(rel, rel_size, efi_reloc) != EFI_SUCCESS) { |
169 | efi_free_pages((uintptr_t) efi_reloc, | |
170 | (virt_size + EFI_PAGE_MASK) >> EFI_PAGE_SHIFT); | |
171 | return NULL; | |
172 | } | |
cb149c66 AG |
173 | |
174 | /* Flush cache */ | |
d0d90993 SG |
175 | flush_cache((ulong)efi_reloc, |
176 | ALIGN(virt_size, CONFIG_SYS_CACHELINE_SIZE)); | |
cb149c66 AG |
177 | invalidate_icache_all(); |
178 | ||
179 | /* Populate the loaded image interface bits */ | |
180 | loaded_image_info->image_base = efi; | |
181 | loaded_image_info->image_size = image_size; | |
182 | ||
183 | return entry; | |
184 | } |