1 | /* $Id: ldrPE.cpp 8245 2008-04-21 17:24:28Z vboxsync $ */
|
---|
2 | /** @file
|
---|
3 | * IPRT - Binary Image Loader, Portable Executable (PE).
|
---|
4 | */
|
---|
5 |
|
---|
6 | /*
|
---|
7 | * Copyright (C) 2006-2007 Sun Microsystems, Inc.
|
---|
8 | *
|
---|
9 | * This file is part of VirtualBox Open Source Edition (OSE), as
|
---|
10 | * available from http://www.virtualbox.org. This file is free software;
|
---|
11 | * you can redistribute it and/or modify it under the terms of the GNU
|
---|
12 | * General Public License (GPL) as published by the Free Software
|
---|
13 | * Foundation, in version 2 as it comes in the "COPYING" file of the
|
---|
14 | * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
|
---|
15 | * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
|
---|
16 | *
|
---|
17 | * The contents of this file may alternatively be used under the terms
|
---|
18 | * of the Common Development and Distribution License Version 1.0
|
---|
19 | * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
|
---|
20 | * VirtualBox OSE distribution, in which case the provisions of the
|
---|
21 | * CDDL are applicable instead of those of the GPL.
|
---|
22 | *
|
---|
23 | * You may elect to license modified versions of this file under the
|
---|
24 | * terms and conditions of either the GPL or the CDDL or both.
|
---|
25 | *
|
---|
26 | * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
|
---|
27 | * Clara, CA 95054 USA or visit http://www.sun.com if you need
|
---|
28 | * additional information or have any questions.
|
---|
29 | */
|
---|
30 |
|
---|
31 |
|
---|
32 | /*******************************************************************************
|
---|
33 | * Header Files *
|
---|
34 | *******************************************************************************/
|
---|
35 | #define LOG_GROUP RTLOGGROUP_LDR
|
---|
36 | #include <iprt/ldr.h>
|
---|
37 | #include <iprt/alloc.h>
|
---|
38 | #include <iprt/assert.h>
|
---|
39 | #include <iprt/log.h>
|
---|
40 | #include <iprt/string.h>
|
---|
41 | #include <iprt/err.h>
|
---|
42 | #include "internal/ldrPE.h"
|
---|
43 | #include "internal/ldr.h"
|
---|
44 |
|
---|
45 |
|
---|
46 |
|
---|
47 | /*******************************************************************************
|
---|
48 | * Defined Constants And Macros *
|
---|
49 | *******************************************************************************/
|
---|
50 | /** Converts rva to a type.
|
---|
51 | * @param pvBits Pointer to base of image bits.
|
---|
52 | * @param rva Relative virtual address.
|
---|
53 | * @param type Type.
|
---|
54 | */
|
---|
55 | #define PE_RVA2TYPE(pvBits, rva, type) ((type) ((uintptr_t)pvBits + (uintptr_t)(rva)) )
|
---|
56 |
|
---|
57 |
|
---|
58 | /*******************************************************************************
|
---|
59 | * Structures and Typedefs *
|
---|
60 | *******************************************************************************/
|
---|
61 | /**
|
---|
62 | * The PE loader structure.
|
---|
63 | */
|
---|
64 | typedef struct RTLDRMODPE
|
---|
65 | {
|
---|
66 | /** Core module structure. */
|
---|
67 | RTLDRMODINTERNAL Core;
|
---|
68 | /** Pointer to the reader instance. */
|
---|
69 | PRTLDRREADER pReader;
|
---|
70 | /** Pointer to internal copy of image bits.
|
---|
71 | * @todo the reader should take care of this. */
|
---|
72 | void *pvBits;
|
---|
73 | /** The offset of the NT headers. */
|
---|
74 | RTFOFF offNtHdrs;
|
---|
75 |
|
---|
76 | /** The machine type (IMAGE_FILE_HEADER::Machine). */
|
---|
77 | uint16_t u16Machine;
|
---|
78 | /** The file flags (IMAGE_FILE_HEADER::Characteristics). */
|
---|
79 | uint16_t fFile;
|
---|
80 | /** Number of sections (IMAGE_FILE_HEADER::NumberOfSections). */
|
---|
81 | unsigned cSections;
|
---|
82 | /** Pointer to an array of the section headers related to the file. */
|
---|
83 | PIMAGE_SECTION_HEADER paSections;
|
---|
84 |
|
---|
85 | /** The RVA of the entry point (IMAGE_OPTIONAL_HEADER32::AddressOfEntryPoint). */
|
---|
86 | RTUINTPTR uEntryPointRVA;
|
---|
87 | /** The base address of the image at link time (IMAGE_OPTIONAL_HEADER32::ImageBase). */
|
---|
88 | RTUINTPTR uImageBase;
|
---|
89 | /** The size of the loaded image (IMAGE_OPTIONAL_HEADER32::SizeOfImage). */
|
---|
90 | uint32_t cbImage;
|
---|
91 | /** Size of the header (IMAGE_OPTIONAL_HEADER32::SizeOfHeaders). */
|
---|
92 | uint32_t cbHeaders;
|
---|
93 | /** The import data directory entry. */
|
---|
94 | IMAGE_DATA_DIRECTORY ImportDir;
|
---|
95 | /** The base relocation data directory entry. */
|
---|
96 | IMAGE_DATA_DIRECTORY RelocDir;
|
---|
97 | /** The export data directory entry. */
|
---|
98 | IMAGE_DATA_DIRECTORY ExportDir;
|
---|
99 | } RTLDRMODPE, *PRTLDRMODPE;
|
---|
100 |
|
---|
101 | /**
|
---|
102 | * PE Loader module operations.
|
---|
103 | *
|
---|
104 | * The PE loader has one operation which is a bit different between 32-bit and 64-bit PE images,
|
---|
105 | * and for historical and performance reasons have been split into separate functions. Thus the
|
---|
106 | * PE loader extends the RTLDROPS structure with this one entry.
|
---|
107 | */
|
---|
108 | typedef struct RTLDROPSPE
|
---|
109 | {
|
---|
110 | /** The usual ops. */
|
---|
111 | RTLDROPS Core;
|
---|
112 |
|
---|
113 | /**
|
---|
114 | * Resolves all imports.
|
---|
115 | *
|
---|
116 | * @returns iprt status code.
|
---|
117 | * @param pModPe Pointer to the PE loader module structure.
|
---|
118 | * @param pvBitsR Where to read raw image bits. (optional)
|
---|
119 | * @param pvBitsW Where to store the imports. The size of this buffer is equal or
|
---|
120 | * larger to the value returned by pfnGetImageSize().
|
---|
121 | * @param pfnGetImport The callback function to use to resolve imports (aka unresolved externals).
|
---|
122 | * @param pvUser User argument to pass to the callback.
|
---|
123 | */
|
---|
124 | DECLCALLBACKMEMBER(int, pfnResolveImports)(PRTLDRMODPE pModPe, const void *pvBitsR, void *pvBitsW, PFNRTLDRIMPORT pfnGetImport, void *pvUser);
|
---|
125 |
|
---|
126 | /** Dummy entry to make sure we've initialized it all. */
|
---|
127 | RTUINT uDummy;
|
---|
128 | } RTLDROPSPE, *PRTLDROPSPE;
|
---|
129 |
|
---|
130 |
|
---|
131 | /*******************************************************************************
|
---|
132 | * Internal Functions *
|
---|
133 | *******************************************************************************/
|
---|
134 | static void rtldrPEConvert32BitOptionalHeaderTo64Bit(PIMAGE_OPTIONAL_HEADER64 pOptHdr);
|
---|
135 | static void rtldrPEConvert32BitLoadConfigTo64Bit(PIMAGE_LOAD_CONFIG_DIRECTORY64 pLoadCfg);
|
---|
136 | static int rtldrPEApplyFixups(PRTLDRMODPE pModPe, const void *pvBitsR, void *pvBitsW, RTUINTPTR BaseAddress, RTUINTPTR OldBaseAddress);
|
---|
137 |
|
---|
138 |
|
---|
139 | /** @copydoc RTLDROPS::pfnGetImageSize */
|
---|
140 | static DECLCALLBACK(size_t) rtldrPEGetImageSize(PRTLDRMODINTERNAL pMod)
|
---|
141 | {
|
---|
142 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
143 | return pModPe->cbImage;
|
---|
144 | }
|
---|
145 |
|
---|
146 |
|
---|
147 | /**
|
---|
148 | * Reads the image into memory.
|
---|
149 | *
|
---|
150 | * @returns iprt status code.
|
---|
151 | * @param pModPe The PE module.
|
---|
152 | * @param pvBits Where to store the bits, this buffer is at least pItem->Core.cbImage in size.
|
---|
153 | */
|
---|
154 | static int rtldrPEGetBitsNoImportsNorFixups(PRTLDRMODPE pModPe, void *pvBits)
|
---|
155 | {
|
---|
156 | /*
|
---|
157 | * Both these checks are related to pfnDone().
|
---|
158 | */
|
---|
159 | PRTLDRREADER pReader = pModPe->pReader;
|
---|
160 | if (!pReader)
|
---|
161 | {
|
---|
162 | AssertMsgFailed(("You've called done!\n"));
|
---|
163 | return VERR_WRONG_ORDER;
|
---|
164 | }
|
---|
165 | if (!pvBits)
|
---|
166 | return VERR_NO_MEMORY;
|
---|
167 |
|
---|
168 | /*
|
---|
169 | * Zero everything (could be done per section).
|
---|
170 | */
|
---|
171 | memset(pvBits, 0, pModPe->cbImage);
|
---|
172 |
|
---|
173 | #ifdef PE_FILE_OFFSET_EQUALS_RVA
|
---|
174 | /*
|
---|
175 | * Read the entire image / file.
|
---|
176 | */
|
---|
177 | const RTFOFF cbRawImage = pReader->pfnSize(pReader)
|
---|
178 | rc = pReader->pfnRead(pReader, pvBits, RT_MIN(pModPe->cbImage, cbRawImage), 0);
|
---|
179 | if (RT_FAILURE(rc))
|
---|
180 | Log(("rtldrPE: %s: Reading %#x bytes at offset %#x failed, %Rrc!!! (the entire image)\n",
|
---|
181 | pReader->pfnLogName(pReader), RT_MIN(pModPe->cbImage, cbRawImage), 0, rc));
|
---|
182 | #else
|
---|
183 |
|
---|
184 | /*
|
---|
185 | * Read the headers.
|
---|
186 | */
|
---|
187 | int rc = pReader->pfnRead(pReader, pvBits, pModPe->cbHeaders, 0);
|
---|
188 | if (RT_SUCCESS(rc))
|
---|
189 | {
|
---|
190 | /*
|
---|
191 | * Read the sections.
|
---|
192 | */
|
---|
193 | PIMAGE_SECTION_HEADER pSH = pModPe->paSections;
|
---|
194 | for (unsigned cLeft = pModPe->cSections; cLeft > 0; cLeft--, pSH++)
|
---|
195 | if (pSH->SizeOfRawData && pSH->Misc.VirtualSize)
|
---|
196 | {
|
---|
197 | rc = pReader->pfnRead(pReader, (uint8_t *)pvBits + pSH->VirtualAddress, pSH->SizeOfRawData, pSH->PointerToRawData);
|
---|
198 | if (RT_FAILURE(rc))
|
---|
199 | {
|
---|
200 | Log(("rtldrPE: %s: Reading %#x bytes at offset %#x failed, %Rrc - section #%d '%.*s'!!!\n",
|
---|
201 | pReader->pfnLogName(pReader), pSH->SizeOfRawData, pSH->PointerToRawData, rc,
|
---|
202 | pSH - pModPe->paSections, sizeof(pSH->Name), pSH->Name));
|
---|
203 | break;
|
---|
204 | }
|
---|
205 | }
|
---|
206 | }
|
---|
207 | else
|
---|
208 | Log(("rtldrPE: %s: Reading %#x bytes at offset %#x failed, %Rrc!!!\n",
|
---|
209 | pReader->pfnLogName(pReader), pModPe->cbHeaders, 0, rc));
|
---|
210 | #endif
|
---|
211 | return rc;
|
---|
212 | }
|
---|
213 |
|
---|
214 |
|
---|
215 | /**
|
---|
216 | * Reads the bits into the internal buffer pointed to by PRTLDRMODPE::pvBits.
|
---|
217 | *
|
---|
218 | * @returns iprt status code.
|
---|
219 | * @param pModPe The PE module.
|
---|
220 | */
|
---|
221 | static int rtldrPEReadBits(PRTLDRMODPE pModPe)
|
---|
222 | {
|
---|
223 | Assert(!pModPe->pvBits);
|
---|
224 | void *pvBitsW = RTMemAllocZ(pModPe->cbImage);
|
---|
225 | if (!pvBitsW)
|
---|
226 | return VERR_NO_MEMORY;
|
---|
227 | int rc = rtldrPEGetBitsNoImportsNorFixups(pModPe, pvBitsW);
|
---|
228 | if (RT_SUCCESS(rc))
|
---|
229 | pModPe->pvBits = pvBitsW;
|
---|
230 | else
|
---|
231 | RTMemFree(pvBitsW);
|
---|
232 | return rc;
|
---|
233 | }
|
---|
234 |
|
---|
235 |
|
---|
236 | /** @copydoc RTLDROPS::pfnGetBits */
|
---|
237 | static DECLCALLBACK(int) rtldrPEGetBits(PRTLDRMODINTERNAL pMod, void *pvBits, RTUINTPTR BaseAddress, PFNRTLDRIMPORT pfnGetImport, void *pvUser)
|
---|
238 | {
|
---|
239 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
240 |
|
---|
241 | /*
|
---|
242 | * Read the image.
|
---|
243 | */
|
---|
244 | int rc = rtldrPEGetBitsNoImportsNorFixups(pModPe, pvBits);
|
---|
245 | if (RT_SUCCESS(rc))
|
---|
246 | {
|
---|
247 | /*
|
---|
248 | * Resolve imports.
|
---|
249 | */
|
---|
250 | rc = ((PRTLDROPSPE)pMod->pOps)->pfnResolveImports(pModPe, pvBits, pvBits, pfnGetImport, pvUser);
|
---|
251 | if (RT_SUCCESS(rc))
|
---|
252 | {
|
---|
253 | /*
|
---|
254 | * Apply relocations.
|
---|
255 | */
|
---|
256 | rc = rtldrPEApplyFixups(pModPe, pvBits, pvBits, BaseAddress, pModPe->uImageBase);
|
---|
257 | if (RT_SUCCESS(rc))
|
---|
258 | return rc;
|
---|
259 | AssertMsgFailed(("Failed to apply fixups. rc=%Rrc\n", rc));
|
---|
260 | }
|
---|
261 | else
|
---|
262 | AssertMsgFailed(("Failed to resolve imports. rc=%Rrc\n", rc));
|
---|
263 | }
|
---|
264 | return rc;
|
---|
265 | }
|
---|
266 |
|
---|
267 |
|
---|
268 | /** @copydoc RTLDROPSPE::pfnResolveImports */
|
---|
269 | static DECLCALLBACK(int) rtldrPEResolveImports32(PRTLDRMODPE pModPe, const void *pvBitsR, void *pvBitsW, PFNRTLDRIMPORT pfnGetImport, void *pvUser)
|
---|
270 | {
|
---|
271 | /*
|
---|
272 | * Check if there is actually anything to work on.
|
---|
273 | */
|
---|
274 | if ( !pModPe->ImportDir.VirtualAddress
|
---|
275 | || !pModPe->ImportDir.Size)
|
---|
276 | return 0;
|
---|
277 |
|
---|
278 | /*
|
---|
279 | * Walk the IMAGE_IMPORT_DESCRIPTOR table.
|
---|
280 | */
|
---|
281 | int rc = VINF_SUCCESS;
|
---|
282 | PIMAGE_IMPORT_DESCRIPTOR pImps;
|
---|
283 | for (pImps = PE_RVA2TYPE(pvBitsR, pModPe->ImportDir.VirtualAddress, PIMAGE_IMPORT_DESCRIPTOR);
|
---|
284 | !rc && pImps->Name != 0 && pImps->FirstThunk != 0;
|
---|
285 | pImps++)
|
---|
286 | {
|
---|
287 | const char *pszModName = PE_RVA2TYPE(pvBitsR, pImps->Name, const char *);
|
---|
288 | PIMAGE_THUNK_DATA32 pFirstThunk; /* update this. */
|
---|
289 | PIMAGE_THUNK_DATA32 pThunk; /* read from this. */
|
---|
290 | Log3(("RTLdrPE: Import descriptor: %s\n", pszModName));
|
---|
291 | Log4(("RTLdrPE: OriginalFirstThunk = %#RX32\n"
|
---|
292 | "RTLdrPE: TimeDateStamp = %#RX32\n"
|
---|
293 | "RTLdrPE: ForwarderChain = %#RX32\n"
|
---|
294 | "RTLdrPE: Name = %#RX32\n"
|
---|
295 | "RTLdrPE: FirstThunk = %#RX32\n",
|
---|
296 | pImps->u.OriginalFirstThunk, pImps->TimeDateStamp,
|
---|
297 | pImps->ForwarderChain, pImps->Name, pImps->FirstThunk));
|
---|
298 |
|
---|
299 | /*
|
---|
300 | * Walk the thunks table(s).
|
---|
301 | */
|
---|
302 | pFirstThunk = PE_RVA2TYPE(pvBitsW, pImps->FirstThunk, PIMAGE_THUNK_DATA32);
|
---|
303 | pThunk = pImps->u.OriginalFirstThunk == 0
|
---|
304 | ? PE_RVA2TYPE(pvBitsR, pImps->FirstThunk, PIMAGE_THUNK_DATA32)
|
---|
305 | : PE_RVA2TYPE(pvBitsR, pImps->u.OriginalFirstThunk, PIMAGE_THUNK_DATA32);
|
---|
306 | while (!rc && pThunk->u1.Ordinal != 0)
|
---|
307 | {
|
---|
308 | RTUINTPTR Value = 0;
|
---|
309 | if (pThunk->u1.Ordinal & IMAGE_ORDINAL_FLAG32)
|
---|
310 | {
|
---|
311 | rc = pfnGetImport(&pModPe->Core, pszModName, NULL, IMAGE_ORDINAL32(pThunk->u1.Ordinal), &Value, pvUser);
|
---|
312 | Log4((RT_SUCCESS(rc) ? "RTLdrPE: %RTptr #%u\n" : "RTLdrPE: %08RX32 #%u rc=%Vrc\n",
|
---|
313 | (uint32_t)Value, IMAGE_ORDINAL32(pThunk->u1.Ordinal), rc));
|
---|
314 | }
|
---|
315 | else if ( pThunk->u1.Ordinal > 0
|
---|
316 | && pThunk->u1.Ordinal < pModPe->cbImage)
|
---|
317 | {
|
---|
318 | rc = pfnGetImport(&pModPe->Core, pszModName, PE_RVA2TYPE(pvBitsR, (char*)pThunk->u1.AddressOfData + 2, const char *),
|
---|
319 | ~0, &Value, pvUser);
|
---|
320 | Log4((RT_SUCCESS(rc) ? "RTLdrPE: %RTptr %s\n" : "RTLdrPE: %08RX32 %s rc=%Vrc\n",
|
---|
321 | (uint32_t)Value, PE_RVA2TYPE(pvBitsR, (char*)pThunk->u1.AddressOfData + 2, const char *), rc));
|
---|
322 | }
|
---|
323 | else
|
---|
324 | {
|
---|
325 | AssertMsgFailed(("bad import data thunk!\n"));
|
---|
326 | rc = VERR_BAD_EXE_FORMAT;
|
---|
327 | }
|
---|
328 | pFirstThunk->u1.Function = Value;
|
---|
329 | if (pFirstThunk->u1.Function != Value)
|
---|
330 | {
|
---|
331 | AssertMsgFailed(("external symbol address to big!\n"));
|
---|
332 | rc = VERR_ADDRESS_CONFLICT; /** @todo get me a better error status code. */
|
---|
333 | }
|
---|
334 | pThunk++;
|
---|
335 | pFirstThunk++;
|
---|
336 | }
|
---|
337 | }
|
---|
338 |
|
---|
339 | return rc;
|
---|
340 | }
|
---|
341 |
|
---|
342 |
|
---|
343 | /** @copydoc RTLDROPSPE::pfnResolveImports */
|
---|
344 | static DECLCALLBACK(int) rtldrPEResolveImports64(PRTLDRMODPE pModPe, const void *pvBitsR, void *pvBitsW, PFNRTLDRIMPORT pfnGetImport, void *pvUser)
|
---|
345 | {
|
---|
346 | /*
|
---|
347 | * Check if there is actually anything to work on.
|
---|
348 | */
|
---|
349 | if ( !pModPe->ImportDir.VirtualAddress
|
---|
350 | || !pModPe->ImportDir.Size)
|
---|
351 | return 0;
|
---|
352 |
|
---|
353 | /*
|
---|
354 | * Walk the IMAGE_IMPORT_DESCRIPTOR table.
|
---|
355 | */
|
---|
356 | int rc = VINF_SUCCESS;
|
---|
357 | PIMAGE_IMPORT_DESCRIPTOR pImps;
|
---|
358 | for (pImps = PE_RVA2TYPE(pvBitsR, pModPe->ImportDir.VirtualAddress, PIMAGE_IMPORT_DESCRIPTOR);
|
---|
359 | !rc && pImps->Name != 0 && pImps->FirstThunk != 0;
|
---|
360 | pImps++)
|
---|
361 | {
|
---|
362 | const char * pszModName = PE_RVA2TYPE(pvBitsR, pImps->Name, const char *);
|
---|
363 | PIMAGE_THUNK_DATA64 pFirstThunk; /* update this. */
|
---|
364 | PIMAGE_THUNK_DATA64 pThunk; /* read from this. */
|
---|
365 | Log3(("RTLdrPE: Import descriptor: %s\n", pszModName));
|
---|
366 | Log4(("RTLdrPE: OriginalFirstThunk = %#RX32\n"
|
---|
367 | "RTLdrPE: TimeDateStamp = %#RX32\n"
|
---|
368 | "RTLdrPE: ForwarderChain = %#RX32\n"
|
---|
369 | "RTLdrPE: Name = %#RX32\n"
|
---|
370 | "RTLdrPE: FirstThunk = %#RX32\n",
|
---|
371 | pImps->u.OriginalFirstThunk, pImps->TimeDateStamp,
|
---|
372 | pImps->ForwarderChain, pImps->Name, pImps->FirstThunk));
|
---|
373 |
|
---|
374 | /*
|
---|
375 | * Walk the thunks table(s).
|
---|
376 | */
|
---|
377 | pFirstThunk = PE_RVA2TYPE(pvBitsW, pImps->FirstThunk, PIMAGE_THUNK_DATA64);
|
---|
378 | pThunk = pImps->u.OriginalFirstThunk == 0
|
---|
379 | ? PE_RVA2TYPE(pvBitsR, pImps->FirstThunk, PIMAGE_THUNK_DATA64)
|
---|
380 | : PE_RVA2TYPE(pvBitsR, pImps->u.OriginalFirstThunk, PIMAGE_THUNK_DATA64);
|
---|
381 | while (!rc && pThunk->u1.Ordinal != 0)
|
---|
382 | {
|
---|
383 | RTUINTPTR Value = 0;
|
---|
384 | if (pThunk->u1.Ordinal & IMAGE_ORDINAL_FLAG64)
|
---|
385 | {
|
---|
386 | rc = pfnGetImport(&pModPe->Core, pszModName, NULL, (unsigned)IMAGE_ORDINAL64(pThunk->u1.Ordinal), &Value, pvUser);
|
---|
387 | Log4((RT_SUCCESS(rc) ? "RTLdrPE: %016RX64 #%u\n" : "RTLdrPE: %016RX64 #%u rc=%Vrc\n",
|
---|
388 | (uint64_t)Value, (unsigned)IMAGE_ORDINAL64(pThunk->u1.Ordinal), rc));
|
---|
389 | }
|
---|
390 | else if ( pThunk->u1.Ordinal > 0
|
---|
391 | && pThunk->u1.Ordinal < pModPe->cbImage)
|
---|
392 | {
|
---|
393 | /** @todo add validation of the string pointer! */
|
---|
394 | rc = pfnGetImport(&pModPe->Core, pszModName, PE_RVA2TYPE(pvBitsR, (uintptr_t)pThunk->u1.AddressOfData + 2, const char *),
|
---|
395 | ~0, &Value, pvUser);
|
---|
396 | Log4((RT_SUCCESS(rc) ? "RTLdrPE: %016RX64 %s\n" : "RTLdrPE: %016RX64 %s rc=%Vrc\n",
|
---|
397 | (uint64_t)Value, PE_RVA2TYPE(pvBitsR, (uintptr_t)pThunk->u1.AddressOfData + 2, const char *), rc));
|
---|
398 | }
|
---|
399 | else
|
---|
400 | {
|
---|
401 | AssertMsgFailed(("bad import data thunk!\n"));
|
---|
402 | rc = VERR_BAD_EXE_FORMAT;
|
---|
403 | }
|
---|
404 | pFirstThunk->u1.Function = Value;
|
---|
405 | pThunk++;
|
---|
406 | pFirstThunk++;
|
---|
407 | }
|
---|
408 | }
|
---|
409 |
|
---|
410 | return rc;
|
---|
411 | }
|
---|
412 |
|
---|
413 |
|
---|
414 | /**
|
---|
415 | * Applies fixups.
|
---|
416 | */
|
---|
417 | static int rtldrPEApplyFixups(PRTLDRMODPE pModPe, const void *pvBitsR, void *pvBitsW, RTUINTPTR BaseAddress, RTUINTPTR OldBaseAddress)
|
---|
418 | {
|
---|
419 | if ( !pModPe->RelocDir.VirtualAddress
|
---|
420 | || !pModPe->RelocDir.Size)
|
---|
421 | return 0;
|
---|
422 |
|
---|
423 | /*
|
---|
424 | * Apply delta fixups iterating fixup chunks.
|
---|
425 | */
|
---|
426 | PIMAGE_BASE_RELOCATION pbr = PE_RVA2TYPE(pvBitsR, pModPe->RelocDir.VirtualAddress, PIMAGE_BASE_RELOCATION);
|
---|
427 | PIMAGE_BASE_RELOCATION pBaseRelocs = pbr;
|
---|
428 | unsigned cbBaseRelocs = pModPe->RelocDir.Size;
|
---|
429 | RTUINTPTR uDelta = BaseAddress - OldBaseAddress;
|
---|
430 | Log2(("RTLdrPE: Fixups: uDelta=%#RTptr BaseAddress=%#RTptr OldBaseAddress=%#RTptr\n", uDelta, BaseAddress, OldBaseAddress));
|
---|
431 | Log4(("RTLdrPE: BASERELOC: VirtualAddres=%RX32 Size=%RX32\n", pModPe->RelocDir.VirtualAddress, pModPe->RelocDir.Size));
|
---|
432 | Assert(sizeof(*pbr) == sizeof(uint32_t) * 2);
|
---|
433 |
|
---|
434 | while ( (uintptr_t)pbr - (uintptr_t)pBaseRelocs + 8 < cbBaseRelocs /* 8= VirtualAddress and SizeOfBlock members */
|
---|
435 | && pbr->SizeOfBlock >= 8)
|
---|
436 | {
|
---|
437 | uint16_t *pwoffFixup = (uint16_t *)(pbr + 1);
|
---|
438 | uint32_t cRelocations = (pbr->SizeOfBlock - sizeof(IMAGE_BASE_RELOCATION)) / sizeof(uint16_t);
|
---|
439 | Log3(("RTLdrPE: base relocs for %#010x, size %#06x (%d relocs)\n", pbr->VirtualAddress, pbr->SizeOfBlock, cRelocations));
|
---|
440 |
|
---|
441 | /* Some bound checking just to be sure it works... */
|
---|
442 | if ((uintptr_t)pbr - (uintptr_t)pBaseRelocs + pbr->SizeOfBlock > cbBaseRelocs)
|
---|
443 | cRelocations = (((uintptr_t)pBaseRelocs + cbBaseRelocs) - (uintptr_t)pbr - sizeof(IMAGE_BASE_RELOCATION)) / sizeof(uint16_t);
|
---|
444 |
|
---|
445 | /*
|
---|
446 | * Loop thru the fixups in this chunk.
|
---|
447 | */
|
---|
448 | while (cRelocations != 0)
|
---|
449 | {
|
---|
450 | /*
|
---|
451 | * Common fixup
|
---|
452 | */
|
---|
453 | static const char * const s_apszReloc[16] =
|
---|
454 | {
|
---|
455 | "ABS", "HIGH", "LOW", "HIGHLOW", "HIGHADJ", "MIPS_JMPADDR", "RES6", "RES7",
|
---|
456 | "RES8", "IA64_IMM64", "DIR64", "HIGH3ADJ", "RES12", "RES13", "RES14", "RES15"
|
---|
457 | }; NOREF(s_apszReloc);
|
---|
458 | union
|
---|
459 | {
|
---|
460 | uint16_t *pu16;
|
---|
461 | uint32_t *pu32;
|
---|
462 | uint64_t *pu64;
|
---|
463 | } u;
|
---|
464 | const int offFixup = *pwoffFixup & 0xfff;
|
---|
465 | u.pu32 = PE_RVA2TYPE(pvBitsW, offFixup + pbr->VirtualAddress, uint32_t *);
|
---|
466 | const int fType = *pwoffFixup >> 12;
|
---|
467 | Log4(("RTLdrPE: %08x %s\n", offFixup + pbr->VirtualAddress, s_apszReloc[fType]));
|
---|
468 | switch (fType)
|
---|
469 | {
|
---|
470 | case IMAGE_REL_BASED_HIGHLOW: /* 32-bit, add delta. */
|
---|
471 | *u.pu32 += uDelta;
|
---|
472 | break;
|
---|
473 | case IMAGE_REL_BASED_DIR64: /* 64-bit, add delta. */
|
---|
474 | *u.pu64 += (RTINTPTR)uDelta;
|
---|
475 | break;
|
---|
476 | case IMAGE_REL_BASED_ABSOLUTE: /* Alignment placeholder. */
|
---|
477 | break;
|
---|
478 | /* odd ones */
|
---|
479 | case IMAGE_REL_BASED_LOW: /* 16-bit, add 1st 16-bit part of the delta. */
|
---|
480 | *u.pu16 += (uint16_t)uDelta;
|
---|
481 | break;
|
---|
482 | case IMAGE_REL_BASED_HIGH: /* 16-bit, add 2nd 16-bit part of the delta. */
|
---|
483 | *u.pu16 += (uint16_t)(uDelta >> 16);
|
---|
484 | break;
|
---|
485 | /* never ever seen these next two, and I'm not 100% sure they are correctly implemented here. */
|
---|
486 | case IMAGE_REL_BASED_HIGHADJ:
|
---|
487 | {
|
---|
488 | if (cRelocations <= 1)
|
---|
489 | {
|
---|
490 | AssertMsgFailed(("HIGHADJ missing 2nd record!\n"));
|
---|
491 | return VERR_BAD_EXE_FORMAT;
|
---|
492 | }
|
---|
493 | cRelocations--;
|
---|
494 | pwoffFixup++;
|
---|
495 | int32_t i32 = (uint32_t)(*u.pu16 << 16) | *pwoffFixup;
|
---|
496 | i32 += uDelta;
|
---|
497 | i32 += 0x8000; //??
|
---|
498 | *u.pu16 = (uint16_t)(i32 >> 16);
|
---|
499 | break;
|
---|
500 | }
|
---|
501 | case IMAGE_REL_BASED_HIGH3ADJ:
|
---|
502 | {
|
---|
503 | if (cRelocations <= 2)
|
---|
504 | {
|
---|
505 | AssertMsgFailed(("HIGHADJ3 missing 2nd record!\n"));
|
---|
506 | return VERR_BAD_EXE_FORMAT;
|
---|
507 | }
|
---|
508 | cRelocations -= 2;
|
---|
509 | pwoffFixup++;
|
---|
510 | int64_t i64 = ((uint64_t)*u.pu16 << 32) | *(uint32_t *)pwoffFixup++;
|
---|
511 | i64 += (int64_t)uDelta << 16; //??
|
---|
512 | i64 += 0x80000000;//??
|
---|
513 | *u.pu16 = (uint16_t)(i64 >> 32);
|
---|
514 | break;
|
---|
515 | }
|
---|
516 | default:
|
---|
517 | AssertMsgFailed(("Unknown fixup type %d offset=%#x\n", fType, offFixup));
|
---|
518 | break;
|
---|
519 | }
|
---|
520 |
|
---|
521 | /*
|
---|
522 | * Next offset/type
|
---|
523 | */
|
---|
524 | pwoffFixup++;
|
---|
525 | cRelocations--;
|
---|
526 | } /* while loop */
|
---|
527 |
|
---|
528 | /*
|
---|
529 | * Next Fixup chunk. (i.e. next page)
|
---|
530 | */
|
---|
531 | pbr = (PIMAGE_BASE_RELOCATION)((uintptr_t)pbr + pbr->SizeOfBlock);
|
---|
532 | } /* while loop */
|
---|
533 |
|
---|
534 | return 0;
|
---|
535 | }
|
---|
536 |
|
---|
537 |
|
---|
538 | /** @copydoc RTLDROPS::pfnRelocate. */
|
---|
539 | static int rtldrPERelocate(PRTLDRMODINTERNAL pMod, void *pvBits, RTUINTPTR NewBaseAddress, RTUINTPTR OldBaseAddress, PFNRTLDRIMPORT pfnGetImport, void *pvUser)
|
---|
540 | {
|
---|
541 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
542 |
|
---|
543 | /*
|
---|
544 | * Do we have to read the image bits?
|
---|
545 | */
|
---|
546 | if (!pModPe->pvBits)
|
---|
547 | {
|
---|
548 | int rc = rtldrPEReadBits(pModPe);
|
---|
549 | if (RT_FAILURE(rc))
|
---|
550 | return rc;
|
---|
551 | }
|
---|
552 |
|
---|
553 | /*
|
---|
554 | * Process imports.
|
---|
555 | */
|
---|
556 | int rc = ((PRTLDROPSPE)pMod->pOps)->pfnResolveImports(pModPe, pModPe->pvBits, pvBits, pfnGetImport, pvUser);
|
---|
557 | if (RT_SUCCESS(rc))
|
---|
558 | {
|
---|
559 | /*
|
---|
560 | * Apply relocations.
|
---|
561 | */
|
---|
562 | rc = rtldrPEApplyFixups(pModPe, pModPe->pvBits, pvBits, NewBaseAddress, OldBaseAddress);
|
---|
563 | AssertRC(rc);
|
---|
564 | }
|
---|
565 | return rc;
|
---|
566 | }
|
---|
567 |
|
---|
568 |
|
---|
569 | /** @copydoc RTLDROPS::pfnGetSymbolEx. */
|
---|
570 | static DECLCALLBACK(int) rtldrPEGetSymbolEx(PRTLDRMODINTERNAL pMod, const void *pvBits, RTUINTPTR BaseAddress, const char *pszSymbol, RTUINTPTR *pValue)
|
---|
571 | {
|
---|
572 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
573 |
|
---|
574 | /*
|
---|
575 | * Check if there is actually anything to work on.
|
---|
576 | */
|
---|
577 | if ( !pModPe->ExportDir.VirtualAddress
|
---|
578 | || !pModPe->ExportDir.Size)
|
---|
579 | return VERR_SYMBOL_NOT_FOUND;
|
---|
580 |
|
---|
581 | /*
|
---|
582 | * No bits supplied? Do we need to read the bits?
|
---|
583 | */
|
---|
584 | if (!pvBits)
|
---|
585 | {
|
---|
586 | if (!pModPe->pvBits)
|
---|
587 | {
|
---|
588 | int rc = rtldrPEReadBits(pModPe);
|
---|
589 | if (RT_FAILURE(rc))
|
---|
590 | return rc;
|
---|
591 | }
|
---|
592 | pvBits = pModPe->pvBits;
|
---|
593 | }
|
---|
594 |
|
---|
595 | PIMAGE_EXPORT_DIRECTORY pExpDir = PE_RVA2TYPE(pvBits, pModPe->ExportDir.VirtualAddress, PIMAGE_EXPORT_DIRECTORY);
|
---|
596 | int iExpOrdinal = 0; /* index into address table. */
|
---|
597 | if ((uintptr_t)pszSymbol <= 0xffff)
|
---|
598 | {
|
---|
599 | /*
|
---|
600 | * Find ordinal export: Simple table lookup.
|
---|
601 | */
|
---|
602 | unsigned uOrdinal = (uintptr_t)pszSymbol & 0xffff;
|
---|
603 | if ( uOrdinal >= pExpDir->Base + RT_MAX(pExpDir->NumberOfNames, pExpDir->NumberOfFunctions)
|
---|
604 | || uOrdinal < pExpDir->Base)
|
---|
605 | return VERR_SYMBOL_NOT_FOUND;
|
---|
606 | iExpOrdinal = uOrdinal - pExpDir->Base;
|
---|
607 | }
|
---|
608 | else
|
---|
609 | {
|
---|
610 | /*
|
---|
611 | * Find Named Export: Do binary search on the name table.
|
---|
612 | */
|
---|
613 | uint32_t *paRVANames = PE_RVA2TYPE(pvBits, pExpDir->AddressOfNames, uint32_t *);
|
---|
614 | uint16_t *paOrdinals = PE_RVA2TYPE(pvBits, pExpDir->AddressOfNameOrdinals, uint16_t *);
|
---|
615 | int iStart = 1;
|
---|
616 | int iEnd = pExpDir->NumberOfNames;
|
---|
617 |
|
---|
618 | for (;;)
|
---|
619 | {
|
---|
620 | /* end of search? */
|
---|
621 | if (iStart > iEnd)
|
---|
622 | {
|
---|
623 | #ifdef RT_STRICT
|
---|
624 | /* do a linear search just to verify the correctness of the above algorithm */
|
---|
625 | for (unsigned i = 0; i < pExpDir->NumberOfNames; i++)
|
---|
626 | {
|
---|
627 | AssertMsg(i == 0 || strcmp(PE_RVA2TYPE(pvBits, paRVANames[i], const char *), PE_RVA2TYPE(pvBits, paRVANames[i - 1], const char *)) > 0,
|
---|
628 | ("bug in binary export search!!!\n"));
|
---|
629 | AssertMsg(strcmp(PE_RVA2TYPE(pvBits, paRVANames[i], const char *), pszSymbol) != 0,
|
---|
630 | ("bug in binary export search!!!\n"));
|
---|
631 | }
|
---|
632 | #endif
|
---|
633 | return VERR_SYMBOL_NOT_FOUND;
|
---|
634 | }
|
---|
635 |
|
---|
636 | int i = (iEnd - iStart) / 2 + iStart;
|
---|
637 | const char *pszExpName = PE_RVA2TYPE(pvBits, paRVANames[i - 1], const char *);
|
---|
638 | int diff = strcmp(pszExpName, pszSymbol);
|
---|
639 | if (diff > 0) /* pszExpName > pszSymbol: search chunck before i */
|
---|
640 | iEnd = i - 1;
|
---|
641 | else if (diff) /* pszExpName < pszSymbol: search chunk after i */
|
---|
642 | iStart = i + 1;
|
---|
643 | else /* pszExpName == pszSymbol */
|
---|
644 | {
|
---|
645 | iExpOrdinal = paOrdinals[i - 1];
|
---|
646 | break;
|
---|
647 | }
|
---|
648 | } /* binary search thru name table */
|
---|
649 | }
|
---|
650 |
|
---|
651 | /*
|
---|
652 | * Found export (iExpOrdinal).
|
---|
653 | */
|
---|
654 | uint32_t * paAddress = PE_RVA2TYPE(pvBits, pExpDir->AddressOfFunctions, uint32_t *);
|
---|
655 | unsigned uRVAExport = paAddress[iExpOrdinal];
|
---|
656 |
|
---|
657 | if ( uRVAExport > pModPe->ExportDir.VirtualAddress
|
---|
658 | && uRVAExport < pModPe->ExportDir.VirtualAddress + pModPe->ExportDir.Size)
|
---|
659 | {
|
---|
660 | /* Resolve forwarder. */
|
---|
661 | AssertMsgFailed(("Forwarders are not supported!\n"));
|
---|
662 | return VERR_SYMBOL_NOT_FOUND;
|
---|
663 | }
|
---|
664 |
|
---|
665 | /* Get plain export address */
|
---|
666 | *pValue = PE_RVA2TYPE(BaseAddress, uRVAExport, RTUINTPTR);
|
---|
667 |
|
---|
668 | return VINF_SUCCESS;
|
---|
669 | }
|
---|
670 |
|
---|
671 |
|
---|
672 | /** @copydoc RTLDROPS::pfnEnumSymbols */
|
---|
673 | static DECLCALLBACK(int) rtldrPEEnumSymbols(PRTLDRMODINTERNAL pMod, unsigned fFlags, const void *pvBits, RTUINTPTR BaseAddress,
|
---|
674 | PFNRTLDRENUMSYMS pfnCallback, void *pvUser)
|
---|
675 | {
|
---|
676 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
677 |
|
---|
678 | /*
|
---|
679 | * Check if there is actually anything to work on.
|
---|
680 | */
|
---|
681 | if ( !pModPe->ExportDir.VirtualAddress
|
---|
682 | || !pModPe->ExportDir.Size)
|
---|
683 | return VERR_SYMBOL_NOT_FOUND;
|
---|
684 |
|
---|
685 | /*
|
---|
686 | * No bits supplied? Do we need to read the bits?
|
---|
687 | */
|
---|
688 | if (!pvBits)
|
---|
689 | {
|
---|
690 | if (!pModPe->pvBits)
|
---|
691 | {
|
---|
692 | int rc = rtldrPEReadBits(pModPe);
|
---|
693 | if (RT_FAILURE(rc))
|
---|
694 | return rc;
|
---|
695 | }
|
---|
696 | pvBits = pModPe->pvBits;
|
---|
697 | }
|
---|
698 |
|
---|
699 | /*
|
---|
700 | * We enumerates by ordinal, which means using a slow linear search for
|
---|
701 | * getting any name
|
---|
702 | */
|
---|
703 | PIMAGE_EXPORT_DIRECTORY pExpDir = PE_RVA2TYPE(pvBits, pModPe->ExportDir.VirtualAddress, PIMAGE_EXPORT_DIRECTORY);
|
---|
704 | uint32_t *paAddress = PE_RVA2TYPE(pvBits, pExpDir->AddressOfFunctions, uint32_t *);
|
---|
705 | uint32_t *paRVANames = PE_RVA2TYPE(pvBits, pExpDir->AddressOfNames, uint32_t *);
|
---|
706 | uint16_t *paOrdinals = PE_RVA2TYPE(pvBits, pExpDir->AddressOfNameOrdinals, uint16_t *);
|
---|
707 | uintptr_t uNamePrev = 0;
|
---|
708 | unsigned cOrdinals = RT_MAX(pExpDir->NumberOfNames, pExpDir->NumberOfFunctions);
|
---|
709 | for (unsigned uOrdinal = 0; uOrdinal < cOrdinals; uOrdinal++)
|
---|
710 | {
|
---|
711 | if (paAddress[uOrdinal] /* needed? */)
|
---|
712 | {
|
---|
713 | /*
|
---|
714 | * Look for name.
|
---|
715 | */
|
---|
716 | const char *pszName = NULL;
|
---|
717 | /* Search from previous + 1 to the end. */
|
---|
718 | unsigned uName = uNamePrev + 1;
|
---|
719 | while (uName < pExpDir->NumberOfNames)
|
---|
720 | {
|
---|
721 | if (paOrdinals[uName] == uOrdinal)
|
---|
722 | {
|
---|
723 | pszName = PE_RVA2TYPE(pvBits, paRVANames[uName], const char *);
|
---|
724 | uNamePrev = uName;
|
---|
725 | break;
|
---|
726 | }
|
---|
727 | uName++;
|
---|
728 | }
|
---|
729 | if (!pszName)
|
---|
730 | {
|
---|
731 | /* Search from start to the previous. */
|
---|
732 | uName = 0;
|
---|
733 | for (uName = 0 ; uName <= uNamePrev; uName++)
|
---|
734 | {
|
---|
735 | if (paOrdinals[uName] == uOrdinal)
|
---|
736 | {
|
---|
737 | pszName = PE_RVA2TYPE(pvBits, paRVANames[uName], const char *);
|
---|
738 | uNamePrev = uName;
|
---|
739 | break;
|
---|
740 | }
|
---|
741 | }
|
---|
742 | }
|
---|
743 |
|
---|
744 | /*
|
---|
745 | * Get address.
|
---|
746 | */
|
---|
747 | uintptr_t uRVAExport = paAddress[uOrdinal];
|
---|
748 | RTUINTPTR Value;
|
---|
749 | if ( uRVAExport - (uintptr_t)pModPe->ExportDir.VirtualAddress
|
---|
750 | < pModPe->ExportDir.Size)
|
---|
751 | {
|
---|
752 | /* Resolve forwarder. */
|
---|
753 | AssertMsgFailed(("Forwarders are not supported!\n"));
|
---|
754 | continue;
|
---|
755 | }
|
---|
756 | else
|
---|
757 | /* Get plain export address */
|
---|
758 | Value = PE_RVA2TYPE(BaseAddress, uRVAExport, RTUINTPTR);
|
---|
759 |
|
---|
760 | /*
|
---|
761 | * Call back.
|
---|
762 | */
|
---|
763 | int rc = pfnCallback(pMod, pszName, uOrdinal + pExpDir->Base, Value, pvUser);
|
---|
764 | if (rc)
|
---|
765 | return rc;
|
---|
766 | }
|
---|
767 | }
|
---|
768 |
|
---|
769 | return VINF_SUCCESS;
|
---|
770 | }
|
---|
771 |
|
---|
772 |
|
---|
773 | /** @copydoc RTLDROPS::pfnDone */
|
---|
774 | static DECLCALLBACK(int) rtldrPEDone(PRTLDRMODINTERNAL pMod)
|
---|
775 | {
|
---|
776 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
777 | if (pModPe->pvBits)
|
---|
778 | {
|
---|
779 | RTMemFree(pModPe->pvBits);
|
---|
780 | pModPe->pvBits = NULL;
|
---|
781 | }
|
---|
782 | if (pModPe->pReader)
|
---|
783 | {
|
---|
784 | int rc = pModPe->pReader->pfnDestroy(pModPe->pReader);
|
---|
785 | AssertRC(rc);
|
---|
786 | pModPe->pReader = NULL;
|
---|
787 | }
|
---|
788 | return VINF_SUCCESS;
|
---|
789 | }
|
---|
790 |
|
---|
791 | /** @copydoc RTLDROPS::pfnClose */
|
---|
792 | static DECLCALLBACK(int) rtldrPEClose(PRTLDRMODINTERNAL pMod)
|
---|
793 | {
|
---|
794 | PRTLDRMODPE pModPe = (PRTLDRMODPE)pMod;
|
---|
795 | if (pModPe->paSections)
|
---|
796 | {
|
---|
797 | RTMemFree(pModPe->paSections);
|
---|
798 | pModPe->paSections = NULL;
|
---|
799 | }
|
---|
800 | if (pModPe->pvBits)
|
---|
801 | {
|
---|
802 | RTMemFree(pModPe->pvBits);
|
---|
803 | pModPe->pvBits = NULL;
|
---|
804 | }
|
---|
805 | if (pModPe->pReader)
|
---|
806 | {
|
---|
807 | int rc = pModPe->pReader->pfnDestroy(pModPe->pReader);
|
---|
808 | AssertRC(rc);
|
---|
809 | pModPe->pReader = NULL;
|
---|
810 | }
|
---|
811 | return VINF_SUCCESS;
|
---|
812 | }
|
---|
813 |
|
---|
814 |
|
---|
815 | /**
|
---|
816 | * Operations for a 32-bit PE module.
|
---|
817 | */
|
---|
818 | static const RTLDROPSPE s_rtldrPE32Ops =
|
---|
819 | {
|
---|
820 | {
|
---|
821 | "pe32",
|
---|
822 | rtldrPEClose,
|
---|
823 | NULL,
|
---|
824 | rtldrPEDone,
|
---|
825 | rtldrPEEnumSymbols,
|
---|
826 | /* ext */
|
---|
827 | rtldrPEGetImageSize,
|
---|
828 | rtldrPEGetBits,
|
---|
829 | rtldrPERelocate,
|
---|
830 | rtldrPEGetSymbolEx,
|
---|
831 | 42
|
---|
832 | },
|
---|
833 | rtldrPEResolveImports32,
|
---|
834 | 42
|
---|
835 | };
|
---|
836 |
|
---|
837 |
|
---|
838 | /**
|
---|
839 | * Operations for a 64-bit PE module.
|
---|
840 | */
|
---|
841 | static const RTLDROPSPE s_rtldrPE64Ops =
|
---|
842 | {
|
---|
843 | {
|
---|
844 | "pe64",
|
---|
845 | rtldrPEClose,
|
---|
846 | NULL,
|
---|
847 | rtldrPEDone,
|
---|
848 | rtldrPEEnumSymbols,
|
---|
849 | /* ext */
|
---|
850 | rtldrPEGetImageSize,
|
---|
851 | rtldrPEGetBits,
|
---|
852 | rtldrPERelocate,
|
---|
853 | rtldrPEGetSymbolEx,
|
---|
854 | 42
|
---|
855 | },
|
---|
856 | rtldrPEResolveImports64,
|
---|
857 | 42
|
---|
858 | };
|
---|
859 |
|
---|
860 |
|
---|
861 | /**
|
---|
862 | * Converts the optional header from 32 bit to 64 bit.
|
---|
863 | * This is a rather simple task, if you start from the right end.
|
---|
864 | *
|
---|
865 | * @param pOptHdr On input this is a PIMAGE_OPTIONAL_HEADER32.
|
---|
866 | * On output this will be a PIMAGE_OPTIONAL_HEADER64.
|
---|
867 | */
|
---|
868 | static void rtldrPEConvert32BitOptionalHeaderTo64Bit(PIMAGE_OPTIONAL_HEADER64 pOptHdr)
|
---|
869 | {
|
---|
870 | /*
|
---|
871 | * volatile everywhere! Trying to prevent the compiler being a smarta$$ and reorder stuff.
|
---|
872 | */
|
---|
873 | IMAGE_OPTIONAL_HEADER32 volatile *pOptHdr32 = (IMAGE_OPTIONAL_HEADER32 volatile *)pOptHdr;
|
---|
874 | IMAGE_OPTIONAL_HEADER64 volatile *pOptHdr64 = pOptHdr;
|
---|
875 |
|
---|
876 | /* from LoaderFlags and out the difference is 4 * 32-bits. */
|
---|
877 | Assert(RT_OFFSETOF(IMAGE_OPTIONAL_HEADER32, LoaderFlags) + 16 == RT_OFFSETOF(IMAGE_OPTIONAL_HEADER64, LoaderFlags));
|
---|
878 | Assert( RT_OFFSETOF(IMAGE_OPTIONAL_HEADER32, DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES]) + 16
|
---|
879 | == RT_OFFSETOF(IMAGE_OPTIONAL_HEADER64, DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES]));
|
---|
880 | uint32_t volatile *pu32Dst = (uint32_t *)&pOptHdr64->DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES] - 1;
|
---|
881 | const uint32_t volatile *pu32Src = (uint32_t *)&pOptHdr32->DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES] - 1;
|
---|
882 | const uint32_t volatile *pu32SrcLast = (uint32_t *)&pOptHdr32->LoaderFlags;
|
---|
883 | while (pu32Src >= pu32SrcLast)
|
---|
884 | *pu32Dst-- = *pu32Src--;
|
---|
885 |
|
---|
886 | /* the previous 4 fields are 32/64 and needs special attention. */
|
---|
887 | pOptHdr64->SizeOfHeapCommit = pOptHdr32->SizeOfHeapCommit;
|
---|
888 | pOptHdr64->SizeOfHeapReserve = pOptHdr32->SizeOfHeapReserve;
|
---|
889 | pOptHdr64->SizeOfStackCommit = pOptHdr32->SizeOfStackCommit;
|
---|
890 | uint32_t u32SizeOfStackReserve = pOptHdr32->SizeOfStackReserve;
|
---|
891 | pOptHdr64->SizeOfStackReserve = u32SizeOfStackReserve;
|
---|
892 |
|
---|
893 | /* The rest matches except for BaseOfData which has been merged into ImageBase in the 64-bit version..
|
---|
894 | * Thus, ImageBase needs some special treatement. It will probably work fine assigning one to the
|
---|
895 | * other since this is all declared volatile, but taking now chances, we'll use a temp variable.
|
---|
896 | */
|
---|
897 | Assert(RT_OFFSETOF(IMAGE_OPTIONAL_HEADER32, SizeOfStackReserve) == RT_OFFSETOF(IMAGE_OPTIONAL_HEADER64, SizeOfStackReserve));
|
---|
898 | Assert(RT_OFFSETOF(IMAGE_OPTIONAL_HEADER32, BaseOfData) == RT_OFFSETOF(IMAGE_OPTIONAL_HEADER64, ImageBase));
|
---|
899 | Assert(RT_OFFSETOF(IMAGE_OPTIONAL_HEADER32, SectionAlignment) == RT_OFFSETOF(IMAGE_OPTIONAL_HEADER64, SectionAlignment));
|
---|
900 | uint32_t u32ImageBase = pOptHdr32->ImageBase;
|
---|
901 | pOptHdr64->ImageBase = u32ImageBase;
|
---|
902 | }
|
---|
903 |
|
---|
904 |
|
---|
905 | /**
|
---|
906 | * Converts the load config directory from 32 bit to 64 bit.
|
---|
907 | * This is a rather simple task, if you start from the right end.
|
---|
908 | *
|
---|
909 | * @param pLoadCfg On input this is a PIMAGE_LOAD_CONFIG_DIRECTORY32.
|
---|
910 | * On output this will be a PIMAGE_LOAD_CONFIG_DIRECTORY64.
|
---|
911 | */
|
---|
912 | static void rtldrPEConvert32BitLoadConfigTo64Bit(PIMAGE_LOAD_CONFIG_DIRECTORY64 pLoadCfg)
|
---|
913 | {
|
---|
914 | /*
|
---|
915 | * volatile everywhere! Trying to prevent the compiler being a smarta$$ and reorder stuff.
|
---|
916 | */
|
---|
917 | IMAGE_LOAD_CONFIG_DIRECTORY32 volatile *pLoadCfg32 = (IMAGE_LOAD_CONFIG_DIRECTORY32 volatile *)pLoadCfg;
|
---|
918 | IMAGE_LOAD_CONFIG_DIRECTORY64 volatile *pLoadCfg64 = pLoadCfg;
|
---|
919 |
|
---|
920 | pLoadCfg64->SEHandlerCount = pLoadCfg32->SEHandlerCount;
|
---|
921 | pLoadCfg64->SEHandlerTable = pLoadCfg32->SEHandlerTable;
|
---|
922 | pLoadCfg64->SecurityCookie = pLoadCfg32->SecurityCookie;
|
---|
923 | pLoadCfg64->EditList = pLoadCfg32->EditList;
|
---|
924 | pLoadCfg64->Reserved1 = pLoadCfg32->Reserved1;
|
---|
925 | pLoadCfg64->CSDVersion = pLoadCfg32->CSDVersion;
|
---|
926 | pLoadCfg64->ProcessHeapFlags = pLoadCfg32->ProcessHeapFlags; /* switched place with ProcessAffinityMask, but we're more than 16 byte off by now so it doesn't matter. */
|
---|
927 | pLoadCfg64->ProcessAffinityMask = pLoadCfg32->ProcessAffinityMask;
|
---|
928 | pLoadCfg64->VirtualMemoryThreshold = pLoadCfg32->VirtualMemoryThreshold;
|
---|
929 | pLoadCfg64->MaximumAllocationSize = pLoadCfg32->MaximumAllocationSize;
|
---|
930 | pLoadCfg64->LockPrefixTable = pLoadCfg32->LockPrefixTable;
|
---|
931 | pLoadCfg64->DeCommitTotalFreeThreshold = pLoadCfg32->DeCommitTotalFreeThreshold;
|
---|
932 | uint32_t u32DeCommitFreeBlockThreshold = pLoadCfg32->DeCommitFreeBlockThreshold;
|
---|
933 | pLoadCfg64->DeCommitFreeBlockThreshold = u32DeCommitFreeBlockThreshold;
|
---|
934 | /* the rest is equal. */
|
---|
935 | Assert( RT_OFFSETOF(IMAGE_LOAD_CONFIG_DIRECTORY32, DeCommitFreeBlockThreshold)
|
---|
936 | == RT_OFFSETOF(IMAGE_LOAD_CONFIG_DIRECTORY64, DeCommitFreeBlockThreshold));
|
---|
937 | }
|
---|
938 |
|
---|
939 |
|
---|
940 | /**
|
---|
941 | * Validates the file header.
|
---|
942 | *
|
---|
943 | * @returns iprt status code.
|
---|
944 | * @param pFileHdr Pointer to the file header that needs validating.
|
---|
945 | * @param pszLogName The log name to prefix the errors with.
|
---|
946 | */
|
---|
947 | int rtldrPEValidateFileHeader(PIMAGE_FILE_HEADER pFileHdr, const char *pszLogName)
|
---|
948 | {
|
---|
949 | size_t cbOptionalHeader;
|
---|
950 | switch (pFileHdr->Machine)
|
---|
951 | {
|
---|
952 | case IMAGE_FILE_MACHINE_I386:
|
---|
953 | cbOptionalHeader = sizeof(IMAGE_OPTIONAL_HEADER32);
|
---|
954 | break;
|
---|
955 | case IMAGE_FILE_MACHINE_AMD64:
|
---|
956 | cbOptionalHeader = sizeof(IMAGE_OPTIONAL_HEADER64);
|
---|
957 | break;
|
---|
958 |
|
---|
959 | default:
|
---|
960 | Log(("rtldrPEOpen: %s: Unsupported Machine=%#x\n",
|
---|
961 | pszLogName, pFileHdr->Machine));
|
---|
962 | return VERR_BAD_EXE_FORMAT;
|
---|
963 | }
|
---|
964 | if (pFileHdr->SizeOfOptionalHeader != cbOptionalHeader)
|
---|
965 | {
|
---|
966 | Log(("rtldrPEOpen: %s: SizeOfOptionalHeader=%#x expected %#x\n",
|
---|
967 | pszLogName, pFileHdr->SizeOfOptionalHeader, cbOptionalHeader));
|
---|
968 | return VERR_BAD_EXE_FORMAT;
|
---|
969 | }
|
---|
970 | /* This restriction needs to be implemented elsewhere. */
|
---|
971 | if (pFileHdr->Characteristics & IMAGE_FILE_RELOCS_STRIPPED)
|
---|
972 | {
|
---|
973 | Log(("rtldrPEOpen: %s: IMAGE_FILE_RELOCS_STRIPPED\n", pszLogName));
|
---|
974 | return VERR_BAD_EXE_FORMAT;
|
---|
975 | }
|
---|
976 | if (pFileHdr->NumberOfSections > 42)
|
---|
977 | {
|
---|
978 | Log(("rtldrPEOpen: %s: NumberOfSections=%d - our limit is 42, please raise it if the binary makes sense.(!!!)\n",
|
---|
979 | pszLogName, pFileHdr->NumberOfSections));
|
---|
980 | return VERR_BAD_EXE_FORMAT;
|
---|
981 | }
|
---|
982 | if (pFileHdr->NumberOfSections < 1)
|
---|
983 | {
|
---|
984 | Log(("rtldrPEOpen: %s: NumberOfSections=%d - we can't have an image without sections (!!!)\n",
|
---|
985 | pszLogName, pFileHdr->NumberOfSections));
|
---|
986 | return VERR_BAD_EXE_FORMAT;
|
---|
987 | }
|
---|
988 | return VINF_SUCCESS;
|
---|
989 | }
|
---|
990 |
|
---|
991 |
|
---|
992 | /**
|
---|
993 | * Validates the optional header (64/32-bit)
|
---|
994 | *
|
---|
995 | * @returns iprt status code.
|
---|
996 | * @param pOptHdr Pointer to the optional header which needs validation.
|
---|
997 | * @param pszLogName The log name to prefix the errors with.
|
---|
998 | * @param offNtHdrs The offset of the NT headers from teh start of the file.
|
---|
999 | * @param pFileHdr Pointer to the file header (valid).
|
---|
1000 | * @param cbRawImage The raw image size.
|
---|
1001 | */
|
---|
1002 | static int rtldrPEValidateOptionalHeader(const IMAGE_OPTIONAL_HEADER64 *pOptHdr, const char *pszLogName, RTFOFF offNtHdrs,
|
---|
1003 | const IMAGE_FILE_HEADER *pFileHdr, RTFOFF cbRawImage)
|
---|
1004 | {
|
---|
1005 | const uint16_t CorrectMagic = pFileHdr->SizeOfOptionalHeader == sizeof(IMAGE_OPTIONAL_HEADER32)
|
---|
1006 | ? IMAGE_NT_OPTIONAL_HDR32_MAGIC : IMAGE_NT_OPTIONAL_HDR64_MAGIC;
|
---|
1007 | if (pOptHdr->Magic != CorrectMagic)
|
---|
1008 | {
|
---|
1009 | Log(("rtldrPEOpen: %s: Magic=%#x - expected %#x!!!\n", pszLogName, pOptHdr->Magic, CorrectMagic));
|
---|
1010 | return VERR_BAD_EXE_FORMAT;
|
---|
1011 | }
|
---|
1012 | const uint32_t cbImage = pOptHdr->SizeOfImage;
|
---|
1013 | if (cbImage > _1G)
|
---|
1014 | {
|
---|
1015 | Log(("rtldrPEOpen: %s: SizeOfImage=%#x - Our limit is 1GB (%#x)!!!\n", pszLogName, cbImage, _1G));
|
---|
1016 | return VERR_BAD_EXE_FORMAT;
|
---|
1017 | }
|
---|
1018 | const uint32_t cbMinImageSize = pFileHdr->SizeOfOptionalHeader + sizeof(*pFileHdr) + 4 + (uint32_t)offNtHdrs;
|
---|
1019 | if (cbImage < cbMinImageSize)
|
---|
1020 | {
|
---|
1021 | Log(("rtldrPEOpen: %s: SizeOfImage=%#x to small, minimum %#x!!!\n", pszLogName, cbImage, cbMinImageSize));
|
---|
1022 | return VERR_BAD_EXE_FORMAT;
|
---|
1023 | }
|
---|
1024 | if (pOptHdr->AddressOfEntryPoint >= cbImage)
|
---|
1025 | {
|
---|
1026 | Log(("rtldrPEOpen: %s: AddressOfEntryPoint=%#x - beyond image size (%#x)!!!\n",
|
---|
1027 | pszLogName, pOptHdr->AddressOfEntryPoint, cbImage));
|
---|
1028 | return VERR_BAD_EXE_FORMAT;
|
---|
1029 | }
|
---|
1030 | if (pOptHdr->BaseOfCode >= cbImage)
|
---|
1031 | {
|
---|
1032 | Log(("rtldrPEOpen: %s: BaseOfCode=%#x - beyond image size (%#x)!!!\n",
|
---|
1033 | pszLogName, pOptHdr->BaseOfCode, cbImage));
|
---|
1034 | return VERR_BAD_EXE_FORMAT;
|
---|
1035 | }
|
---|
1036 | #if 0/* only in 32-bit header */
|
---|
1037 | if (pOptHdr->BaseOfData >= cbImage)
|
---|
1038 | {
|
---|
1039 | Log(("rtldrPEOpen: %s: BaseOfData=%#x - beyond image size (%#x)!!!\n",
|
---|
1040 | pszLogName, pOptHdr->BaseOfData, cbImage));
|
---|
1041 | return VERR_BAD_EXE_FORMAT;
|
---|
1042 | }
|
---|
1043 | #endif
|
---|
1044 | if (pOptHdr->SizeOfHeaders >= cbImage)
|
---|
1045 | {
|
---|
1046 | Log(("rtldrPEOpen: %s: SizeOfHeaders=%#x - beyond image size (%#x)!!!\n",
|
---|
1047 | pszLogName, pOptHdr->SizeOfHeaders, cbImage));
|
---|
1048 | return VERR_BAD_EXE_FORMAT;
|
---|
1049 | }
|
---|
1050 | /* don't know how to do the checksum, so ignore it. */
|
---|
1051 | if (pOptHdr->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN)
|
---|
1052 | {
|
---|
1053 | Log(("rtldrPEOpen: %s: Subsystem=%#x (unknown)!!!\n", pszLogName, pOptHdr->Subsystem));
|
---|
1054 | return VERR_BAD_EXE_FORMAT;
|
---|
1055 | }
|
---|
1056 | if (pOptHdr->SizeOfHeaders < cbMinImageSize + pFileHdr->NumberOfSections * sizeof(IMAGE_SECTION_HEADER))
|
---|
1057 | {
|
---|
1058 | Log(("rtldrPEOpen: %s: SizeOfHeaders=%#x - cbMinImageSize %#x + sections %#x = %#llx!!!\n",
|
---|
1059 | pszLogName, pOptHdr->SizeOfHeaders,
|
---|
1060 | cbImage, cbMinImageSize, pFileHdr->NumberOfSections * sizeof(IMAGE_SECTION_HEADER),
|
---|
1061 | cbMinImageSize + pFileHdr->NumberOfSections * sizeof(IMAGE_SECTION_HEADER)));
|
---|
1062 | return VERR_BAD_EXE_FORMAT;
|
---|
1063 | }
|
---|
1064 | if (pOptHdr->SizeOfStackReserve < pOptHdr->SizeOfStackCommit)
|
---|
1065 | {
|
---|
1066 | Log(("rtldrPEOpen: %s: SizeOfStackReserve %#x < SizeOfStackCommit %#x!!!\n",
|
---|
1067 | pszLogName, pOptHdr->SizeOfStackReserve, pOptHdr->SizeOfStackCommit));
|
---|
1068 | return VERR_BAD_EXE_FORMAT;
|
---|
1069 | }
|
---|
1070 | if (pOptHdr->SizeOfHeapReserve < pOptHdr->SizeOfHeapCommit)
|
---|
1071 | {
|
---|
1072 | Log(("rtldrPEOpen: %s: SizeOfStackReserve %#x < SizeOfStackCommit %#x!!!\n",
|
---|
1073 | pszLogName, pOptHdr->SizeOfStackReserve, pOptHdr->SizeOfStackCommit));
|
---|
1074 | return VERR_BAD_EXE_FORMAT;
|
---|
1075 | }
|
---|
1076 |
|
---|
1077 | /* DataDirectory */
|
---|
1078 | if (pOptHdr->NumberOfRvaAndSizes != ELEMENTS(pOptHdr->DataDirectory))
|
---|
1079 | {
|
---|
1080 | Log(("rtldrPEOpen: %s: NumberOfRvaAndSizes=%d!!!\n", pszLogName, pOptHdr->NumberOfRvaAndSizes));
|
---|
1081 | return VERR_BAD_EXE_FORMAT;
|
---|
1082 | }
|
---|
1083 | for (unsigned i = 0; i < ELEMENTS(pOptHdr->DataDirectory); i++)
|
---|
1084 | {
|
---|
1085 | IMAGE_DATA_DIRECTORY const *pDir = &pOptHdr->DataDirectory[i];
|
---|
1086 | if (!pDir->Size)
|
---|
1087 | continue;
|
---|
1088 | size_t cb = cbImage;
|
---|
1089 | switch (i)
|
---|
1090 | {
|
---|
1091 | case IMAGE_DIRECTORY_ENTRY_EXPORT: // 0
|
---|
1092 | case IMAGE_DIRECTORY_ENTRY_IMPORT: // 1
|
---|
1093 | case IMAGE_DIRECTORY_ENTRY_RESOURCE: // 2
|
---|
1094 | case IMAGE_DIRECTORY_ENTRY_EXCEPTION: // 3
|
---|
1095 | case IMAGE_DIRECTORY_ENTRY_BASERELOC: // 5
|
---|
1096 | case IMAGE_DIRECTORY_ENTRY_DEBUG: // 6
|
---|
1097 | case IMAGE_DIRECTORY_ENTRY_COPYRIGHT: // 7
|
---|
1098 | case IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT: // 11
|
---|
1099 | case IMAGE_DIRECTORY_ENTRY_IAT: // 12 /* Import Address Table */
|
---|
1100 | break;
|
---|
1101 | case IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG: // 10 - need to check for lock prefixes.
|
---|
1102 | /* Delay inspection after section table is validated. */
|
---|
1103 | break;
|
---|
1104 |
|
---|
1105 | case IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT: // 13
|
---|
1106 | Log(("rtldrPEOpen: %s: dir no. %d (DELAY_IMPORT) VirtualAddress=%#x Size=%#x is not supported!!!\n",
|
---|
1107 | pszLogName, i, pDir->VirtualAddress, pDir->Size));
|
---|
1108 | return VERR_LDRPE_DELAY_IMPORT;
|
---|
1109 |
|
---|
1110 | /* The security directory seems to be some kind of hack, and the rva is a fileoffset or something. */
|
---|
1111 | case IMAGE_DIRECTORY_ENTRY_SECURITY: // 4
|
---|
1112 | cb = (size_t)cbRawImage; Assert((RTFOFF)cb == cbRawImage);
|
---|
1113 | Log(("rtldrPEOpen: %s: dir no. %d (SECURITY) VirtualAddress=%#x Size=%#x is not supported!!!\n",
|
---|
1114 | pszLogName, i, pDir->VirtualAddress, pDir->Size));
|
---|
1115 | return VERR_LDRPE_SECURITY;
|
---|
1116 |
|
---|
1117 | case IMAGE_DIRECTORY_ENTRY_GLOBALPTR: // 8 /* (MIPS GP) */
|
---|
1118 | Log(("rtldrPEOpen: %s: dir no. %d (GLOBALPTR) VirtualAddress=%#x Size=%#x is not supported!!!\n",
|
---|
1119 | pszLogName, i, pDir->VirtualAddress, pDir->Size));
|
---|
1120 | return VERR_LDRPE_GLOBALPTR;
|
---|
1121 |
|
---|
1122 | case IMAGE_DIRECTORY_ENTRY_TLS: // 9
|
---|
1123 | Log(("rtldrPEOpen: %s: dir no. %d (TLS) VirtualAddress=%#x Size=%#x is not supported!!!\n",
|
---|
1124 | pszLogName, i, pDir->VirtualAddress, pDir->Size));
|
---|
1125 | return VERR_LDRPE_TLS;
|
---|
1126 |
|
---|
1127 | case IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR: // 14
|
---|
1128 | Log(("rtldrPEOpen: %s: dir no. %d (COM_DESCRIPTOR) VirtualAddress=%#x Size=%#x is not supported!!!\n",
|
---|
1129 | pszLogName, i, pDir->VirtualAddress, pDir->Size));
|
---|
1130 | return VERR_LDRPE_COM_DESCRIPTOR;
|
---|
1131 |
|
---|
1132 | default:
|
---|
1133 | Log(("rtldrPEOpen: %s: dir no. %d VirtualAddress=%#x Size=%#x is not supported!!!\n",
|
---|
1134 | pszLogName, i, pDir->VirtualAddress, pDir->Size));
|
---|
1135 | return VERR_BAD_EXE_FORMAT;
|
---|
1136 | }
|
---|
1137 | if (pDir->VirtualAddress >= cb)
|
---|
1138 | {
|
---|
1139 | Log(("rtldrPEOpen: %s: dir no. %d VirtualAddress=%#x is invalid (limit %#x)!!!\n",
|
---|
1140 | pszLogName, i, pDir->VirtualAddress, cb));
|
---|
1141 | return VERR_BAD_EXE_FORMAT;
|
---|
1142 | }
|
---|
1143 | if (pDir->Size > cb - pDir->VirtualAddress)
|
---|
1144 | {
|
---|
1145 | Log(("rtldrPEOpen: %s: dir no. %d Size=%#x is invalid (rva=%#x, limit=%#x)!!!\n",
|
---|
1146 | pszLogName, i, pDir->Size, pDir->VirtualAddress, cb));
|
---|
1147 | return VERR_BAD_EXE_FORMAT;
|
---|
1148 | }
|
---|
1149 | }
|
---|
1150 | return VINF_SUCCESS;
|
---|
1151 | }
|
---|
1152 |
|
---|
1153 |
|
---|
1154 | /**
|
---|
1155 | * Validates the section headers.
|
---|
1156 | *
|
---|
1157 | * @returns iprt status code.
|
---|
1158 | * @param paSections Pointer to the array of sections that is to be validated.
|
---|
1159 | * @param cSections Number of sections in that array.
|
---|
1160 | * @param pszLogName The log name to prefix the errors with.
|
---|
1161 | * @param pOptHdr Pointer to the optional header (valid).
|
---|
1162 | * @param cbRawImage The raw image size.
|
---|
1163 | */
|
---|
1164 | int rtldrPEValidateSectionHeaders(const IMAGE_SECTION_HEADER *paSections, unsigned cSections, const char *pszLogName,
|
---|
1165 | const IMAGE_OPTIONAL_HEADER64 *pOptHdr, RTFOFF cbRawImage)
|
---|
1166 | {
|
---|
1167 | const uint32_t cbImage = pOptHdr->SizeOfImage;
|
---|
1168 | const IMAGE_SECTION_HEADER *pSH = &paSections[0];
|
---|
1169 | uint32_t uRvaPrev = pOptHdr->SizeOfHeaders;
|
---|
1170 | Log3(("RTLdrPE: Section Headers:\n"));
|
---|
1171 | for (unsigned cSHdrsLeft = cSections; cSHdrsLeft > 0; cSHdrsLeft--, pSH++)
|
---|
1172 | {
|
---|
1173 | const unsigned iSH = pSH - &paSections[0]; NOREF(iSH);
|
---|
1174 | Log3(("RTLdrPE: #%d '%-8.8s' Characteristics: %08RX32\n"
|
---|
1175 | "RTLdrPE: VirtAddr: %08RX32 VirtSize: %08RX32\n"
|
---|
1176 | "RTLdrPE: FileOff: %08RX32 FileSize: %08RX32\n"
|
---|
1177 | "RTLdrPE: RelocOff: %08RX32 #Relocs: %08RX32\n"
|
---|
1178 | "RTLdrPE: LineOff: %08RX32 #Lines: %08RX32\n",
|
---|
1179 | iSH, pSH->Name, pSH->Characteristics,
|
---|
1180 | pSH->VirtualAddress, pSH->Misc.VirtualSize,
|
---|
1181 | pSH->PointerToRawData, pSH->SizeOfRawData,
|
---|
1182 | pSH->PointerToRelocations, pSH->NumberOfRelocations,
|
---|
1183 | pSH->PointerToLinenumbers, pSH->NumberOfLinenumbers));
|
---|
1184 | if (pSH->Characteristics & (IMAGE_SCN_MEM_16BIT | IMAGE_SCN_MEM_FARDATA | IMAGE_SCN_MEM_PURGEABLE | IMAGE_SCN_MEM_PRELOAD))
|
---|
1185 | {
|
---|
1186 | Log(("rtldrPEOpen: %s: Unsupported section flag(s) %#x section #%d '%.*s'!!!\n",
|
---|
1187 | pszLogName, pSH->Characteristics, iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1188 | return VERR_BAD_EXE_FORMAT;
|
---|
1189 | }
|
---|
1190 |
|
---|
1191 | if ( pSH->Misc.VirtualSize
|
---|
1192 | && !(pSH->Characteristics & IMAGE_SCN_TYPE_NOLOAD)) /* binutils uses this for '.stab' even if it's reserved/obsoleted by MS. */
|
---|
1193 | {
|
---|
1194 | if (pSH->VirtualAddress < uRvaPrev)
|
---|
1195 | {
|
---|
1196 | Log(("rtldrPEOpen: %s: Overlaps previous section or sections aren't in ascending order, VirtualAddress=%#x uRvaPrev=%#x - section #%d '%.*s'!!!\n",
|
---|
1197 | pszLogName, pSH->VirtualAddress, uRvaPrev, iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1198 | return VERR_BAD_EXE_FORMAT;
|
---|
1199 | }
|
---|
1200 | if (pSH->VirtualAddress > cbImage)
|
---|
1201 | {
|
---|
1202 | Log(("rtldrPEOpen: %s: VirtualAddress=%#x - beyond image size (%#x) - section #%d '%.*s'!!!\n",
|
---|
1203 | pszLogName, pSH->VirtualAddress, cbImage, iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1204 | return VERR_BAD_EXE_FORMAT;
|
---|
1205 | }
|
---|
1206 |
|
---|
1207 | if (pSH->VirtualAddress & (pOptHdr->SectionAlignment - 1)) //ASSUMES power of 2 alignment.
|
---|
1208 | {
|
---|
1209 | Log(("rtldrPEOpen: %s: VirtualAddress=%#x missaligned (%#x) - section #%d '%.*s'!!!\n",
|
---|
1210 | pszLogName, pSH->VirtualAddress, pOptHdr->SectionAlignment, iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1211 | return VERR_BAD_EXE_FORMAT;
|
---|
1212 | }
|
---|
1213 |
|
---|
1214 | #ifdef PE_FILE_OFFSET_EQUALS_RVA
|
---|
1215 | /* Our loader code assume rva matches the file offset. */
|
---|
1216 | if ( pSH->SizeOfRawData
|
---|
1217 | && pSH->PointerToRawData != pSH->VirtualAddress)
|
---|
1218 | {
|
---|
1219 | Log(("rtldrPEOpen: %s: ASSUMPTION FAILED: file offset %#x != RVA %#x - section #%d '%.*s'!!!\n",
|
---|
1220 | pszLogName, pSH->PointerToRawData, pSH->VirtualAddress, iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1221 | return VERR_BAD_EXE_FORMAT;
|
---|
1222 | }
|
---|
1223 | #endif
|
---|
1224 | }
|
---|
1225 |
|
---|
1226 | ///@todo only if SizeOfRawData > 0 ?
|
---|
1227 | if ( pSH->PointerToRawData > cbRawImage /// @todo pSH->PointerToRawData >= cbRawImage ?
|
---|
1228 | || pSH->SizeOfRawData > cbRawImage
|
---|
1229 | || pSH->PointerToRawData + pSH->SizeOfRawData > cbRawImage)
|
---|
1230 | {
|
---|
1231 | Log(("rtldrPEOpen: %s: PointerToRawData=%#x SizeOfRawData=%#x - beyond end of file (%#x) - section #%d '%.*s'!!!\n",
|
---|
1232 | pszLogName, pSH->PointerToRawData, pSH->SizeOfRawData, cbRawImage,
|
---|
1233 | iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1234 | return VERR_BAD_EXE_FORMAT;
|
---|
1235 | }
|
---|
1236 |
|
---|
1237 | if (pSH->PointerToRawData & (pOptHdr->FileAlignment - 1)) //ASSUMES power of 2 alignment.
|
---|
1238 | {
|
---|
1239 | Log(("rtldrPEOpen: %s: PointerToRawData=%#x missaligned (%#x) - section #%d '%.*s'!!!\n",
|
---|
1240 | pszLogName, pSH->PointerToRawData, pOptHdr->FileAlignment, iSH, sizeof(pSH->Name), pSH->Name));
|
---|
1241 | return VERR_BAD_EXE_FORMAT;
|
---|
1242 | }
|
---|
1243 |
|
---|
1244 | /* ignore the relocations and linenumbers. */
|
---|
1245 |
|
---|
1246 | uRvaPrev = pSH->VirtualAddress + pSH->Misc.VirtualSize;
|
---|
1247 | }
|
---|
1248 |
|
---|
1249 | /** @todo r=bird: more sanity checks! */
|
---|
1250 | return VINF_SUCCESS;
|
---|
1251 | }
|
---|
1252 |
|
---|
1253 |
|
---|
1254 | /**
|
---|
1255 | * Reads image data by RVA using the section headers.
|
---|
1256 | *
|
---|
1257 | * @returns iprt status code.
|
---|
1258 | * @param pModPe The PE module instance.
|
---|
1259 | * @param pvBuf Where to store the bits.
|
---|
1260 | * @param cb Number of bytes to tread.
|
---|
1261 | * @param RVA Where to read from.
|
---|
1262 | */
|
---|
1263 | static int rtldrPEReadRVA(PRTLDRMODPE pModPe, void *pvBuf, uint32_t cb, uint32_t RVA)
|
---|
1264 | {
|
---|
1265 | const IMAGE_SECTION_HEADER *pSH = pModPe->paSections;
|
---|
1266 | PRTLDRREADER pReader = pModPe->pReader;
|
---|
1267 | uint32_t cbRead;
|
---|
1268 | int rc;
|
---|
1269 |
|
---|
1270 | /*
|
---|
1271 | * Is it the headers, i.e. prior to the first section.
|
---|
1272 | */
|
---|
1273 | if (RVA < pModPe->cbHeaders)
|
---|
1274 | {
|
---|
1275 | cbRead = RT_MIN(pModPe->cbHeaders - RVA, cb);
|
---|
1276 | rc = pReader->pfnRead(pReader, pvBuf, cbRead, RVA);
|
---|
1277 | if ( cbRead == cb
|
---|
1278 | || RT_FAILURE(rc))
|
---|
1279 | return rc;
|
---|
1280 | cb -= cbRead;
|
---|
1281 | RVA += cbRead;
|
---|
1282 | pvBuf = (uint8_t *)pvBuf + cbRead;
|
---|
1283 | }
|
---|
1284 |
|
---|
1285 | /* In the zero space between headers and the first section? */
|
---|
1286 | if (RVA < pSH->VirtualAddress)
|
---|
1287 | {
|
---|
1288 | cbRead = RT_MIN(pSH->VirtualAddress - RVA, cb);
|
---|
1289 | memset(pvBuf, 0, cbRead);
|
---|
1290 | if (cbRead == cb)
|
---|
1291 | return VINF_SUCCESS;
|
---|
1292 | cb -= cbRead;
|
---|
1293 | RVA += cbRead;
|
---|
1294 | pvBuf = (uint8_t *)pvBuf + cbRead;
|
---|
1295 | }
|
---|
1296 |
|
---|
1297 | /*
|
---|
1298 | * Iterate the sections.
|
---|
1299 | */
|
---|
1300 | for (unsigned cLeft = pModPe->cSections;
|
---|
1301 | cLeft > 0;
|
---|
1302 | cLeft--, pSH++)
|
---|
1303 | {
|
---|
1304 | uint32_t off = RVA - pSH->VirtualAddress;
|
---|
1305 | if (off < pSH->Misc.VirtualSize)
|
---|
1306 | {
|
---|
1307 | cbRead = RT_MIN(pSH->Misc.VirtualSize - off, cb);
|
---|
1308 | rc = pReader->pfnRead(pReader, pvBuf, cbRead, pSH->PointerToRawData + off);
|
---|
1309 | if ( cbRead == cb
|
---|
1310 | || RT_FAILURE(rc))
|
---|
1311 | return rc;
|
---|
1312 | cb -= cbRead;
|
---|
1313 | RVA += cbRead;
|
---|
1314 | pvBuf = (uint8_t *)pvBuf + cbRead;
|
---|
1315 | }
|
---|
1316 | uint32_t RVANext = cLeft ? pSH[1].VirtualAddress : pModPe->cbImage;
|
---|
1317 | if (RVA < RVANext)
|
---|
1318 | {
|
---|
1319 | cbRead = RT_MIN(RVANext - RVA, cb);
|
---|
1320 | memset(pvBuf, 0, cbRead);
|
---|
1321 | if (cbRead == cb)
|
---|
1322 | return VINF_SUCCESS;
|
---|
1323 | cb -= cbRead;
|
---|
1324 | RVA += cbRead;
|
---|
1325 | pvBuf = (uint8_t *)pvBuf + cbRead;
|
---|
1326 | }
|
---|
1327 | }
|
---|
1328 |
|
---|
1329 | AssertFailed();
|
---|
1330 | return VERR_INTERNAL_ERROR;
|
---|
1331 | }
|
---|
1332 |
|
---|
1333 |
|
---|
1334 | /**
|
---|
1335 | * Validates the data of some selected data directories entries.
|
---|
1336 | *
|
---|
1337 | * This requires a valid section table and thus has to wait
|
---|
1338 | * till after we've read and validated it.
|
---|
1339 | *
|
---|
1340 | * @returns iprt status code.
|
---|
1341 | * @param pModPe The PE module instance.
|
---|
1342 | * @param pOptHdr Pointer to the optional header (valid).
|
---|
1343 | */
|
---|
1344 | int rtldrPEValidateDirectories(PRTLDRMODPE pModPe, const IMAGE_OPTIONAL_HEADER64 *pOptHdr)
|
---|
1345 | {
|
---|
1346 | const char *pszLogName = pModPe->pReader->pfnLogName(pModPe->pReader); NOREF(pszLogName);
|
---|
1347 | union /* combine stuff we're reading to help reduce stack usage. */
|
---|
1348 | {
|
---|
1349 | IMAGE_LOAD_CONFIG_DIRECTORY64 Cfg64;
|
---|
1350 | } u;
|
---|
1351 |
|
---|
1352 | /*
|
---|
1353 | * The load config entry may include lock prefix tables and whatnot which we don't implement.
|
---|
1354 | * It does also include a lot of stuff which we can ignore, so we'll have to inspect the
|
---|
1355 | * actual data before we can make up our mind about it all.
|
---|
1356 | */
|
---|
1357 | IMAGE_DATA_DIRECTORY Dir = pOptHdr->DataDirectory[IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG];
|
---|
1358 | if (Dir.Size)
|
---|
1359 | {
|
---|
1360 | const size_t cbExpect = pOptHdr->Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC
|
---|
1361 | ? sizeof(IMAGE_LOAD_CONFIG_DIRECTORY32)
|
---|
1362 | : sizeof(IMAGE_LOAD_CONFIG_DIRECTORY64);
|
---|
1363 | if ( Dir.Size != cbExpect
|
---|
1364 | && ( cbExpect == sizeof(IMAGE_LOAD_CONFIG_DIRECTORY32)
|
---|
1365 | && Dir.Size != (uint32_t)RT_OFFSETOF(IMAGE_LOAD_CONFIG_DIRECTORY32, SEHandlerTable))
|
---|
1366 | )
|
---|
1367 | {
|
---|
1368 | Log(("rtldrPEOpen: %s: load cfg dir: unexpected dir size of %d bytes, expected %d.\n",
|
---|
1369 | pszLogName, Dir.Size, cbExpect));
|
---|
1370 | return VERR_LDRPE_LOAD_CONFIG_SIZE;
|
---|
1371 | }
|
---|
1372 |
|
---|
1373 | /*
|
---|
1374 | * Read and convert to 64-bit.
|
---|
1375 | */
|
---|
1376 | memset(&u.Cfg64, 0, sizeof(u.Cfg64));
|
---|
1377 | int rc = rtldrPEReadRVA(pModPe, &u.Cfg64, Dir.Size, Dir.VirtualAddress);
|
---|
1378 | if (RT_FAILURE(rc))
|
---|
1379 | return rc;
|
---|
1380 | rtldrPEConvert32BitLoadConfigTo64Bit(&u.Cfg64);
|
---|
1381 |
|
---|
1382 | if (u.Cfg64.Size != cbExpect)
|
---|
1383 | {
|
---|
1384 | Log(("rtldrPEOpen: %s: load cfg dir: unexpected header size of %d bytes, expected %d.\n",
|
---|
1385 | pszLogName, u.Cfg64.Size, cbExpect));
|
---|
1386 | return VERR_LDRPE_LOAD_CONFIG_SIZE;
|
---|
1387 | }
|
---|
1388 | if (u.Cfg64.LockPrefixTable)
|
---|
1389 | {
|
---|
1390 | Log(("rtldrPEOpen: %s: load cfg dir: lock prefix table at %RX64. We don't support lock prefix tables!\n",
|
---|
1391 | pszLogName, u.Cfg64.LockPrefixTable));
|
---|
1392 | return VERR_LDRPE_LOCK_PREFIX_TABLE;
|
---|
1393 | }
|
---|
1394 | #if 0/* this seems to be safe to ignore. */
|
---|
1395 | if ( u.Cfg64.SEHandlerTable
|
---|
1396 | || u.Cfg64.SEHandlerCount)
|
---|
1397 | {
|
---|
1398 | Log(("rtldrPEOpen: %s: load cfg dir: SEHandlerTable=%RX64 and SEHandlerCount=%RX64 are unsupported!\n",
|
---|
1399 | pszLogName, u.Cfg64.SEHandlerTable, u.Cfg64.SEHandlerCount));
|
---|
1400 | return VERR_BAD_EXE_FORMAT;
|
---|
1401 | }
|
---|
1402 | #endif
|
---|
1403 | if (u.Cfg64.EditList)
|
---|
1404 | {
|
---|
1405 | Log(("rtldrPEOpen: %s: load cfg dir: EditList=%RX64 is unsupported!\n",
|
---|
1406 | pszLogName, u.Cfg64.EditList));
|
---|
1407 | return VERR_BAD_EXE_FORMAT;
|
---|
1408 | }
|
---|
1409 | }
|
---|
1410 | return VINF_SUCCESS;
|
---|
1411 | }
|
---|
1412 |
|
---|
1413 |
|
---|
1414 | /**
|
---|
1415 | * Open a PE image.
|
---|
1416 | *
|
---|
1417 | * @returns iprt status code.
|
---|
1418 | * @param pReader The loader reader instance which will provide the raw image bits.
|
---|
1419 | * @param offNtHdrs The offset of the NT headers (where you find "PE\0\0").
|
---|
1420 | * @param phLdrMod Where to store the handle.
|
---|
1421 | */
|
---|
1422 | int rtldrPEOpen(PRTLDRREADER pReader, RTFOFF offNtHdrs, PRTLDRMOD phLdrMod)
|
---|
1423 | {
|
---|
1424 | /*
|
---|
1425 | * Read and validate the file header.
|
---|
1426 | */
|
---|
1427 | IMAGE_FILE_HEADER FileHdr;
|
---|
1428 | int rc = pReader->pfnRead(pReader, &FileHdr, sizeof(FileHdr), offNtHdrs + 4);
|
---|
1429 | if (RT_FAILURE(rc))
|
---|
1430 | return rc;
|
---|
1431 | const char *pszLogName = pReader->pfnLogName(pReader);
|
---|
1432 | rc = rtldrPEValidateFileHeader(&FileHdr, pszLogName);
|
---|
1433 | if (RT_FAILURE(rc))
|
---|
1434 | return rc;
|
---|
1435 |
|
---|
1436 | /*
|
---|
1437 | * Read and validate the "optional" header. Convert 32->64 if necessary.
|
---|
1438 | */
|
---|
1439 | IMAGE_OPTIONAL_HEADER64 OptHdr;
|
---|
1440 | rc = pReader->pfnRead(pReader, &OptHdr, FileHdr.SizeOfOptionalHeader, offNtHdrs + 4 + sizeof(IMAGE_FILE_HEADER));
|
---|
1441 | if (RT_FAILURE(rc))
|
---|
1442 | return rc;
|
---|
1443 | if (FileHdr.SizeOfOptionalHeader != sizeof(OptHdr))
|
---|
1444 | rtldrPEConvert32BitOptionalHeaderTo64Bit(&OptHdr);
|
---|
1445 | rc = rtldrPEValidateOptionalHeader(&OptHdr, pszLogName, offNtHdrs, &FileHdr, pReader->pfnSize(pReader));
|
---|
1446 | if (RT_FAILURE(rc))
|
---|
1447 | return rc;
|
---|
1448 |
|
---|
1449 | /*
|
---|
1450 | * Read and validate section headers.
|
---|
1451 | */
|
---|
1452 | const size_t cbSections = sizeof(IMAGE_SECTION_HEADER) * FileHdr.NumberOfSections;
|
---|
1453 | PIMAGE_SECTION_HEADER paSections = (PIMAGE_SECTION_HEADER)RTMemAlloc(cbSections);
|
---|
1454 | if (!paSections)
|
---|
1455 | return VERR_NO_MEMORY;
|
---|
1456 | rc = pReader->pfnRead(pReader, paSections, cbSections, offNtHdrs + 4 + sizeof(IMAGE_FILE_HEADER) + FileHdr.SizeOfOptionalHeader);
|
---|
1457 | if (RT_SUCCESS(rc))
|
---|
1458 | {
|
---|
1459 | rc = rtldrPEValidateSectionHeaders(paSections, FileHdr.NumberOfSections, pszLogName,
|
---|
1460 | &OptHdr, pReader->pfnSize(pReader));
|
---|
1461 | if (RT_SUCCESS(rc))
|
---|
1462 | {
|
---|
1463 | /*
|
---|
1464 | * Allocate and initialize the PE module structure.
|
---|
1465 | */
|
---|
1466 | PRTLDRMODPE pModPe = (PRTLDRMODPE)RTMemAllocZ(sizeof(*pModPe));
|
---|
1467 | if (pModPe)
|
---|
1468 | {
|
---|
1469 | pModPe->Core.u32Magic = RTLDRMOD_MAGIC;
|
---|
1470 | pModPe->Core.eState = LDR_STATE_OPENED;
|
---|
1471 | if (FileHdr.SizeOfOptionalHeader == sizeof(OptHdr))
|
---|
1472 | pModPe->Core.pOps = &s_rtldrPE64Ops.Core;
|
---|
1473 | else
|
---|
1474 | pModPe->Core.pOps = &s_rtldrPE32Ops.Core;
|
---|
1475 | pModPe->pReader = pReader;
|
---|
1476 | pModPe->pvBits = NULL;
|
---|
1477 | pModPe->offNtHdrs = offNtHdrs;
|
---|
1478 | pModPe->u16Machine = FileHdr.Machine;
|
---|
1479 | pModPe->fFile = FileHdr.Characteristics;
|
---|
1480 | pModPe->cSections = FileHdr.NumberOfSections;
|
---|
1481 | pModPe->paSections = paSections;
|
---|
1482 | pModPe->uEntryPointRVA= OptHdr.AddressOfEntryPoint;
|
---|
1483 | pModPe->uImageBase = (RTUINTPTR)OptHdr.ImageBase;
|
---|
1484 | pModPe->cbImage = OptHdr.SizeOfImage;
|
---|
1485 | pModPe->cbHeaders = OptHdr.SizeOfHeaders;
|
---|
1486 | pModPe->ImportDir = OptHdr.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT];
|
---|
1487 | pModPe->RelocDir = OptHdr.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC];
|
---|
1488 | pModPe->ExportDir = OptHdr.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT];
|
---|
1489 |
|
---|
1490 | /*
|
---|
1491 | * Perform validation of some selected data directories which requires
|
---|
1492 | * inspection of the actual data.
|
---|
1493 | */
|
---|
1494 | rc = rtldrPEValidateDirectories(pModPe, &OptHdr);
|
---|
1495 | if (RT_SUCCESS(rc))
|
---|
1496 | {
|
---|
1497 | *phLdrMod = &pModPe->Core;
|
---|
1498 | return VINF_SUCCESS;
|
---|
1499 | }
|
---|
1500 | RTMemFree(pModPe);
|
---|
1501 | }
|
---|
1502 | else
|
---|
1503 | rc = VERR_NO_MEMORY;
|
---|
1504 | }
|
---|
1505 | }
|
---|
1506 | RTMemFree(paSections);
|
---|
1507 | return rc;
|
---|
1508 | }
|
---|
1509 |
|
---|
1510 |
|
---|
1511 |
|
---|