ldrkStuff.cpp@ 46087

Last change on this file since 46087 was 46083, checked in by vboxsync, 11 years ago
Made it possible to find symbols for windows nt using a image-in-guest-memory loader fallback.
Property svn:eol-style set to `native` Property svn:keywords set to `Id`
File size: 29.9 KB

Line
1	/* $Id: ldrkStuff.cpp 46083 2013-05-14 23:39:28Z vboxsync $ */
2	/** @file
3	* IPRT - Binary Image Loader, kLdr Interface.
4	*/
5
6	/*
7	* Copyright (C) 2006-2011 Oracle Corporation
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
27
28	/*******************************************************************************
29	* Header Files *
30	*******************************************************************************/
31	#define LOG_GROUP RTLOGGROUP_LDR
32	#include <iprt/ldr.h>
33	#include "internal/iprt.h"
34
35	#include <iprt/file.h>
36	#include <iprt/alloc.h>
37	#include <iprt/alloca.h>
38	#include <iprt/assert.h>
39	#include <iprt/string.h>
40	#include <iprt/path.h>
41	#include <iprt/log.h>
42	#include <iprt/param.h>
43	#include <iprt/err.h>
44	#include "internal/ldr.h"
45	#define KLDR_ALREADY_INCLUDE_STD_TYPES
46	#define KLDR_NO_KLDR_H_INCLUDES
47	#include <k/kLdr.h>
48	#include <k/kRdrAll.h>
49	#include <k/kErr.h>
50	#include <k/kErrors.h>
51	#include <k/kMagics.h>
52
53
54	/*******************************************************************************
55	* Structures and Typedefs *
56	*******************************************************************************/
57	/**
58	* kLdr file provider.
59	*/
60	typedef struct RTKLDRRDR
61	{
62	/** The core. */
63	KRDR Core;
64	/** The IPRT bit reader. */
65	PRTLDRREADER pReader;
66	} RTKLDRRDR, *PRTKLDRRDR;
67
68	/**
69	* IPRT module.
70	*/
71	typedef struct RTLDRMODKLDR
72	{
73	/** The Core module structure. */
74	RTLDRMODINTERNAL Core;
75	/** The kLdr module. */
76	PKLDRMOD pMod;
77	} RTLDRMODKLDR, *PRTLDRMODKLDR;
78
79
80	/**
81	* Arguments for a RTLDRMODKLDR callback wrapper.
82	*/
83	typedef struct RTLDRMODKLDRARGS
84	{
85	union
86	{
87	PFNRT pfn;
88	PFNRTLDRENUMDBG pfnEnumDbgInfo;
89	PFNRTLDRENUMSYMS pfnEnumSyms;
90	PFNRTLDRIMPORT pfnGetImport;
91	} u;
92	void *pvUser;
93	const void *pvBits;
94	PRTLDRMODKLDR pMod;
95	int rc;
96	} RTLDRMODKLDRARGS, *PRTLDRMODKLDRARGS;
97
98
99	/**
100	* Converts a kLdr error code to an IPRT one.
101	*/
102	static int rtkldrConvertError(int krc)
103	{
104	if (!krc)
105	return VINF_SUCCESS;
106	switch (krc)
107	{
108	case KERR_INVALID_PARAMETER: return VERR_INVALID_PARAMETER;
109	case KERR_INVALID_HANDLE: return VERR_INVALID_HANDLE;
110	case KERR_NO_MEMORY: return VERR_NO_MEMORY;
111
112
113	case KLDR_ERR_UNKNOWN_FORMAT:
114	case KLDR_ERR_MZ_NOT_SUPPORTED: return VERR_MZ_EXE_NOT_SUPPORTED;
115	case KLDR_ERR_NE_NOT_SUPPORTED: return VERR_NE_EXE_NOT_SUPPORTED;
116	case KLDR_ERR_LX_NOT_SUPPORTED: return VERR_LX_EXE_NOT_SUPPORTED;
117	case KLDR_ERR_LE_NOT_SUPPORTED: return VERR_LE_EXE_NOT_SUPPORTED;
118	case KLDR_ERR_PE_NOT_SUPPORTED: return VERR_PE_EXE_NOT_SUPPORTED;
119	case KLDR_ERR_ELF_NOT_SUPPORTED: return VERR_ELF_EXE_NOT_SUPPORTED;
120	case KLDR_ERR_MACHO_NOT_SUPPORTED: return VERR_INVALID_EXE_SIGNATURE;
121	case KLDR_ERR_AOUT_NOT_SUPPORTED: return VERR_AOUT_EXE_NOT_SUPPORTED;
122
123	case KLDR_ERR_MODULE_NOT_FOUND: return VERR_MODULE_NOT_FOUND;
124	case KLDR_ERR_PREREQUISITE_MODULE_NOT_FOUND: return VERR_MODULE_NOT_FOUND;
125	case KLDR_ERR_MAIN_STACK_ALLOC_FAILED: return VERR_NO_MEMORY;
126	case KERR_BUFFER_OVERFLOW: return VERR_BUFFER_OVERFLOW;
127	case KLDR_ERR_SYMBOL_NOT_FOUND: return VERR_SYMBOL_NOT_FOUND;
128	case KLDR_ERR_FORWARDER_SYMBOL: return VERR_BAD_EXE_FORMAT;
129	case KLDR_ERR_BAD_FIXUP: AssertMsgFailedReturn(("KLDR_ERR_BAD_FIXUP\n"), VERR_BAD_EXE_FORMAT);
130	case KLDR_ERR_IMPORT_ORDINAL_OUT_OF_BOUNDS: return VERR_BAD_EXE_FORMAT;
131	case KLDR_ERR_NO_DEBUG_INFO: return VERR_FILE_NOT_FOUND;
132	case KLDR_ERR_ALREADY_MAPPED: return VERR_WRONG_ORDER;
133	case KLDR_ERR_NOT_MAPPED: return VERR_WRONG_ORDER;
134	case KLDR_ERR_ADDRESS_OVERFLOW: return VERR_NUMBER_TOO_BIG;
135	case KLDR_ERR_TODO: return VERR_NOT_IMPLEMENTED;
136
137	case KLDR_ERR_NOT_LOADED_DYNAMICALLY:
138	case KCPU_ERR_ARCH_CPU_NOT_COMPATIBLE:
139	case KLDR_ERR_TOO_LONG_FORWARDER_CHAIN:
140	case KLDR_ERR_MODULE_TERMINATING:
141	case KLDR_ERR_PREREQUISITE_MODULE_TERMINATING:
142	case KLDR_ERR_MODULE_INIT_FAILED:
143	case KLDR_ERR_PREREQUISITE_MODULE_INIT_FAILED:
144	case KLDR_ERR_MODULE_INIT_FAILED_ALREADY:
145	case KLDR_ERR_PREREQUISITE_MODULE_INIT_FAILED_ALREADY:
146	case KLDR_ERR_PREREQUISITE_RECURSED_TOO_DEEPLY:
147	case KLDR_ERR_THREAD_ATTACH_FAILED:
148	case KRDR_ERR_TOO_MANY_MAPPINGS:
149	case KLDR_ERR_NOT_DLL:
150	case KLDR_ERR_NOT_EXE:
151	AssertMsgFailedReturn(("krc=%d (%#x): %s\n", krc, krc, kErrName(krc)), VERR_GENERAL_FAILURE);
152
153
154	case KLDR_ERR_PE_UNSUPPORTED_MACHINE:
155	case KLDR_ERR_PE_BAD_FILE_HEADER:
156	case KLDR_ERR_PE_BAD_OPTIONAL_HEADER:
157	case KLDR_ERR_PE_BAD_SECTION_HEADER:
158	case KLDR_ERR_PE_BAD_FORWARDER:
159	case KLDR_ERR_PE_FORWARDER_IMPORT_NOT_FOUND:
160	case KLDR_ERR_PE_BAD_FIXUP:
161	case KLDR_ERR_PE_BAD_IMPORT:
162	AssertMsgFailedReturn(("krc=%d (%#x): %s\n", krc, krc, kErrName(krc)), VERR_GENERAL_FAILURE);
163
164	case KLDR_ERR_LX_BAD_HEADER:
165	case KLDR_ERR_LX_BAD_LOADER_SECTION:
166	case KLDR_ERR_LX_BAD_FIXUP_SECTION:
167	case KLDR_ERR_LX_BAD_OBJECT_TABLE:
168	case KLDR_ERR_LX_BAD_PAGE_MAP:
169	case KLDR_ERR_LX_BAD_ITERDATA:
170	case KLDR_ERR_LX_BAD_ITERDATA2:
171	case KLDR_ERR_LX_BAD_BUNDLE:
172	case KLDR_ERR_LX_NO_SONAME:
173	case KLDR_ERR_LX_BAD_SONAME:
174	case KLDR_ERR_LX_BAD_FORWARDER:
175	case KLDR_ERR_LX_NRICHAIN_NOT_SUPPORTED:
176	AssertMsgFailedReturn(("krc=%d (%#x): %s\n", krc, krc, kErrName(krc)), VERR_GENERAL_FAILURE);
177
178	case KLDR_ERR_MACHO_OTHER_ENDIAN_NOT_SUPPORTED:
179	case KLDR_ERR_MACHO_BAD_HEADER:
180	case KLDR_ERR_MACHO_UNSUPPORTED_FILE_TYPE:
181	case KLDR_ERR_MACHO_UNSUPPORTED_MACHINE:
182	case KLDR_ERR_MACHO_BAD_LOAD_COMMAND:
183	case KLDR_ERR_MACHO_UNKNOWN_LOAD_COMMAND:
184	case KLDR_ERR_MACHO_UNSUPPORTED_LOAD_COMMAND:
185	case KLDR_ERR_MACHO_BAD_SECTION:
186	case KLDR_ERR_MACHO_UNSUPPORTED_SECTION:
187	#ifdef KLDR_ERR_MACHO_UNSUPPORTED_INIT_SECTION
188	case KLDR_ERR_MACHO_UNSUPPORTED_INIT_SECTION:
189	case KLDR_ERR_MACHO_UNSUPPORTED_TERM_SECTION:
190	#endif
191	case KLDR_ERR_MACHO_UNKNOWN_SECTION:
192	case KLDR_ERR_MACHO_BAD_SECTION_ORDER:
193	case KLDR_ERR_MACHO_BIT_MIX:
194	case KLDR_ERR_MACHO_BAD_OBJECT_FILE:
195	case KLDR_ERR_MACHO_BAD_SYMBOL:
196	case KLDR_ERR_MACHO_UNSUPPORTED_FIXUP_TYPE:
197	AssertMsgFailedReturn(("krc=%d (%#x): %s\n", krc, krc, kErrName(krc)), VERR_GENERAL_FAILURE);
198
199	default:
200	if (RT_FAILURE(krc))
201	return krc;
202	AssertMsgFailedReturn(("krc=%d (%#x): %s\n", krc, krc, kErrName(krc)), VERR_NO_TRANSLATION);
203	}
204	}
205
206
207	/**
208	* Converts a IPRT error code to an kLdr one.
209	*/
210	static int rtkldrConvertErrorFromIPRT(int rc)
211	{
212	if (RT_SUCCESS(rc))
213	return 0;
214	switch (rc)
215	{
216	case VERR_NO_MEMORY: return KERR_NO_MEMORY;
217	case VERR_INVALID_PARAMETER: return KERR_INVALID_PARAMETER;
218	case VERR_INVALID_HANDLE: return KERR_INVALID_HANDLE;
219	case VERR_BUFFER_OVERFLOW: return KERR_BUFFER_OVERFLOW;
220	default:
221	return rc;
222	}
223	}
224
225
226
227
228
229
230	/** @copydoc KLDRRDROPS::pfnCreate
231	* @remark This is a dummy which isn't used. */
232	static int rtkldrRdr_Create( PPKRDR ppRdr, const char *pszFilename)
233	{
234	NOREF(ppRdr); NOREF(pszFilename);
235	AssertReleaseFailed();
236	return -1;
237	}
238
239
240	/** @copydoc KLDRRDROPS::pfnDestroy */
241	static int rtkldrRdr_Destroy( PKRDR pRdr)
242	{
243	PRTKLDRRDR pThis = (PRTKLDRRDR)pRdr;
244	pThis->pReader = NULL;
245	RTMemFree(pThis);
246	return 0;
247	}
248
249
250	/** @copydoc KLDRRDROPS::pfnRead */
251	static int rtkldrRdr_Read( PKRDR pRdr, void *pvBuf, KSIZE cb, KFOFF off)
252	{
253	PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
254	int rc = pReader->pfnRead(pReader, pvBuf, cb, off);
255	return rtkldrConvertErrorFromIPRT(rc);
256	}
257
258
259	/** @copydoc KLDRRDROPS::pfnAllMap */
260	static int rtkldrRdr_AllMap( PKRDR pRdr, const void **ppvBits)
261	{
262	PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
263	int rc = pReader->pfnMap(pReader, ppvBits);
264	return rtkldrConvertErrorFromIPRT(rc);
265	}
266
267
268	/** @copydoc KLDRRDROPS::pfnAllUnmap */
269	static int rtkldrRdr_AllUnmap(PKRDR pRdr, const void *pvBits)
270	{
271	PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
272	int rc = pReader->pfnUnmap(pReader, pvBits);
273	return rtkldrConvertErrorFromIPRT(rc);
274	}
275
276
277	/** @copydoc KLDRRDROPS::pfnSize */
278	static KFOFF rtkldrRdr_Size( PKRDR pRdr)
279	{
280	PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
281	return (KFOFF)pReader->pfnSize(pReader);
282	}
283
284
285	/** @copydoc KLDRRDROPS::pfnTell */
286	static KFOFF rtkldrRdr_Tell( PKRDR pRdr)
287	{
288	PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
289	return (KFOFF)pReader->pfnTell(pReader);
290	}
291
292
293	/** @copydoc KLDRRDROPS::pfnName */
294	static const char * rtkldrRdr_Name(PKRDR pRdr)
295	{
296	PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
297	return pReader->pfnLogName(pReader);
298	}
299
300
301	/** @copydoc KLDRRDROPS::pfnNativeFH */
302	static KIPTR rtkldrRdr_NativeFH(PKRDR pRdr)
303	{
304	NOREF(pRdr);
305	AssertFailed();
306	return -1;
307	}
308
309
310	/** @copydoc KLDRRDROPS::pfnPageSize */
311	static KSIZE rtkldrRdr_PageSize(PKRDR pRdr)
312	{
313	NOREF(pRdr);
314	return PAGE_SIZE;
315	}
316
317
318	/** @copydoc KLDRRDROPS::pfnMap */
319	static int rtkldrRdr_Map( PKRDR pRdr, void **ppvBase, KU32 cSegments, PCKLDRSEG paSegments, KBOOL fFixed)
320	{
321	//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
322	NOREF(pRdr); NOREF(ppvBase); NOREF(cSegments); NOREF(paSegments); NOREF(fFixed);
323	AssertFailed();
324	return -1;
325	}
326
327
328	/** @copydoc KLDRRDROPS::pfnRefresh */
329	static int rtkldrRdr_Refresh( PKRDR pRdr, void *pvBase, KU32 cSegments, PCKLDRSEG paSegments)
330	{
331	//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
332	NOREF(pRdr); NOREF(pvBase); NOREF(cSegments); NOREF(paSegments);
333	AssertFailed();
334	return -1;
335	}
336
337
338	/** @copydoc KLDRRDROPS::pfnProtect */
339	static int rtkldrRdr_Protect( PKRDR pRdr, void *pvBase, KU32 cSegments, PCKLDRSEG paSegments, KBOOL fUnprotectOrProtect)
340	{
341	//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
342	NOREF(pRdr); NOREF(pvBase); NOREF(cSegments); NOREF(paSegments); NOREF(fUnprotectOrProtect);
343	AssertFailed();
344	return -1;
345	}
346
347
348	/** @copydoc KLDRRDROPS::pfnUnmap */
349	static int rtkldrRdr_Unmap( PKRDR pRdr, void *pvBase, KU32 cSegments, PCKLDRSEG paSegments)
350	{
351	//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
352	NOREF(pRdr); NOREF(pvBase); NOREF(cSegments); NOREF(paSegments);
353	AssertFailed();
354	return -1;
355	}
356
357	/** @copydoc KLDRRDROPS::pfnDone */
358	static void rtkldrRdr_Done( PKRDR pRdr)
359	{
360	NOREF(pRdr);
361	//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
362	}
363
364
365	/**
366	* The file reader operations.
367	* We provide our own based on IPRT instead of using the kLdr ones.
368	*/
369	extern "C" const KRDROPS g_kLdrRdrFileOps;
370	extern "C" const KRDROPS g_kLdrRdrFileOps =
371	{
372	/* .pszName = */ "IPRT",
373	/* .pNext = */ NULL,
374	/* .pfnCreate = */ rtkldrRdr_Create,
375	/* .pfnDestroy = */ rtkldrRdr_Destroy,
376	/* .pfnRead = */ rtkldrRdr_Read,
377	/* .pfnAllMap = */ rtkldrRdr_AllMap,
378	/* .pfnAllUnmap = */ rtkldrRdr_AllUnmap,
379	/* .pfnSize = */ rtkldrRdr_Size,
380	/* .pfnTell = */ rtkldrRdr_Tell,
381	/* .pfnName = */ rtkldrRdr_Name,
382	/* .pfnNativeFH = */ rtkldrRdr_NativeFH,
383	/* .pfnPageSize = */ rtkldrRdr_PageSize,
384	/* .pfnMap = */ rtkldrRdr_Map,
385	/* .pfnRefresh = */ rtkldrRdr_Refresh,
386	/* .pfnProtect = */ rtkldrRdr_Protect,
387	/* .pfnUnmap = */ rtkldrRdr_Unmap,
388	/* .pfnDone = */ rtkldrRdr_Done,
389	/* .u32Dummy = */ 42
390	};
391
392
393
394
395
396	/** @copydoc RTLDROPS::pfnClose */
397	static DECLCALLBACK(int) rtkldr_Close(PRTLDRMODINTERNAL pMod)
398	{
399	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
400	int rc = kLdrModClose(pModkLdr);
401	return rtkldrConvertError(rc);
402	}
403
404
405	/** @copydoc RTLDROPS::pfnDone */
406	static DECLCALLBACK(int) rtkldr_Done(PRTLDRMODINTERNAL pMod)
407	{
408	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
409	int rc = kLdrModMostlyDone(pModkLdr);
410	return rtkldrConvertError(rc);
411	}
412
413
414	/** @copydoc FNKLDRMODENUMSYMS */
415	static int rtkldrEnumSymbolsWrapper(PKLDRMOD pMod, uint32_t iSymbol,
416	const char pchSymbol, KSIZE cchSymbol, const char pszVersion,
417	KLDRADDR uValue, uint32_t fKind, void *pvUser)
418	{
419	PRTLDRMODKLDRARGS pArgs = (PRTLDRMODKLDRARGS)pvUser;
420	NOREF(pMod); NOREF(pszVersion); NOREF(fKind);
421
422	/* If not zero terminated we'll have to use a temporary buffer. */
423	const char *pszSymbol = pchSymbol;
424	if (pchSymbol && pchSymbol[cchSymbol])
425	{
426	char psz = (char )alloca(cchSymbol + 1);
427	memcpy(psz, pchSymbol, cchSymbol);
428	psz[cchSymbol] = '\0';
429	pszSymbol = psz;
430	}
431
432	#if defined(RT_OS_OS2) \|\| defined(RT_OS_DARWIN)
433	/* skip the underscore prefix. */
434	if (*pszSymbol == '_')
435	pszSymbol++;
436	#endif
437
438	int rc = pArgs->u.pfnEnumSyms(&pArgs->pMod->Core, pszSymbol, iSymbol, uValue, pArgs->pvUser);
439	if (RT_FAILURE(rc))
440	return rc; /* don't bother converting. */
441	return 0;
442	}
443
444
445	/** @copydoc RTLDROPS::pfnEnumSymbols */
446	static DECLCALLBACK(int) rtkldr_EnumSymbols(PRTLDRMODINTERNAL pMod, unsigned fFlags, const void *pvBits, RTUINTPTR BaseAddress,
447	PFNRTLDRENUMSYMS pfnCallback, void *pvUser)
448	{
449	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
450	RTLDRMODKLDRARGS Args;
451	Args.pvUser = pvUser;
452	Args.u.pfnEnumSyms = pfnCallback;
453	Args.pMod = (PRTLDRMODKLDR)pMod;
454	Args.pvBits = pvBits;
455	Args.rc = VINF_SUCCESS;
456	int rc = kLdrModEnumSymbols(pModkLdr, pvBits, BaseAddress,
457	fFlags & RTLDR_ENUM_SYMBOL_FLAGS_ALL ? KLDRMOD_ENUM_SYMS_FLAGS_ALL : 0,
458	rtkldrEnumSymbolsWrapper, &Args);
459	if (Args.rc != VINF_SUCCESS)
460	rc = Args.rc;
461	else
462	rc = rtkldrConvertError(rc);
463	return rc;
464	}
465
466
467	/** @copydoc RTLDROPS::pfnGetImageSize */
468	static DECLCALLBACK(size_t) rtkldr_GetImageSize(PRTLDRMODINTERNAL pMod)
469	{
470	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
471	return kLdrModSize(pModkLdr);
472	}
473
474
475	/** @copydoc FNKLDRMODGETIMPORT */
476	static int rtkldrGetImportWrapper(PKLDRMOD pMod, uint32_t iImport, uint32_t iSymbol, const char *pchSymbol, KSIZE cchSymbol,
477	const char pszVersion, PKLDRADDR puValue, uint32_t pfKind, void *pvUser)
478	{
479	PRTLDRMODKLDRARGS pArgs = (PRTLDRMODKLDRARGS)pvUser;
480	NOREF(pMod); NOREF(pszVersion); NOREF(pfKind);
481
482	/* If not zero terminated we'll have to use a temporary buffer. */
483	const char *pszSymbol = pchSymbol;
484	if (pchSymbol && pchSymbol[cchSymbol])
485	{
486	char psz = (char )alloca(cchSymbol + 1);
487	memcpy(psz, pchSymbol, cchSymbol);
488	psz[cchSymbol] = '\0';
489	pszSymbol = psz;
490	}
491
492	#if defined(RT_OS_OS2) \|\| defined(RT_OS_DARWIN)
493	/* skip the underscore prefix. */
494	if (*pszSymbol == '_')
495	pszSymbol++;
496	#endif
497
498	/* get the import module name - TODO: cache this */
499	const char *pszModule = NULL;
500	if (iImport != NIL_KLDRMOD_IMPORT)
501	{
502	char pszBuf = (char )alloca(64);
503	int rc = kLdrModGetImport(pMod, pArgs->pvBits, iImport, pszBuf, 64);
504	if (rc)
505	return rc;
506	pszModule = pszBuf;
507	}
508
509	/* do the query */
510	RTUINTPTR Value;
511	int rc = pArgs->u.pfnGetImport(&pArgs->pMod->Core, pszModule, pszSymbol, pszSymbol ? ~0 : iSymbol, &Value, pArgs->pvUser);
512	if (RT_SUCCESS(rc))
513	{
514	*puValue = Value;
515	return 0;
516	}
517	return rtkldrConvertErrorFromIPRT(rc);
518	}
519
520
521	/** @copydoc RTLDROPS::pfnGetBits */
522	static DECLCALLBACK(int) rtkldr_GetBits(PRTLDRMODINTERNAL pMod, void *pvBits, RTUINTPTR BaseAddress,
523	PFNRTLDRIMPORT pfnGetImport, void *pvUser)
524	{
525	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
526	RTLDRMODKLDRARGS Args;
527	Args.pvUser = pvUser;
528	Args.u.pfnGetImport = pfnGetImport;
529	Args.pMod = (PRTLDRMODKLDR)pMod;
530	Args.pvBits = pvBits;
531	Args.rc = VINF_SUCCESS;
532	int rc = kLdrModGetBits(pModkLdr, pvBits, BaseAddress, rtkldrGetImportWrapper, &Args);
533	if (Args.rc != VINF_SUCCESS)
534	rc = Args.rc;
535	else
536	rc = rtkldrConvertError(rc);
537	return rc;
538	}
539
540
541	/** @copydoc RTLDROPS::pfnRelocate */
542	static DECLCALLBACK(int) rtkldr_Relocate(PRTLDRMODINTERNAL pMod, void *pvBits, RTUINTPTR NewBaseAddress,
543	RTUINTPTR OldBaseAddress, PFNRTLDRIMPORT pfnGetImport, void *pvUser)
544	{
545	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
546	RTLDRMODKLDRARGS Args;
547	Args.pvUser = pvUser;
548	Args.u.pfnGetImport = pfnGetImport;
549	Args.pMod = (PRTLDRMODKLDR)pMod;
550	Args.pvBits = pvBits;
551	Args.rc = VINF_SUCCESS;
552	int rc = kLdrModRelocateBits(pModkLdr, pvBits, NewBaseAddress, OldBaseAddress, rtkldrGetImportWrapper, &Args);
553	if (Args.rc != VINF_SUCCESS)
554	rc = Args.rc;
555	else
556	rc = rtkldrConvertError(rc);
557	return rc;
558	}
559
560
561	/** @copydoc RTLDROPS::pfnGetSymbolEx */
562	static DECLCALLBACK(int) rtkldr_GetSymbolEx(PRTLDRMODINTERNAL pMod, const void *pvBits, RTUINTPTR BaseAddress,
563	const char pszSymbol, RTUINTPTR pValue)
564	{
565	PKLDRMOD pModkLdr = ((PRTLDRMODKLDR)pMod)->pMod;
566	KLDRADDR uValue;
567
568	#if defined(RT_OS_OS2) \|\| defined(RT_OS_DARWIN)
569	/*
570	* Add underscore prefix.
571	*/
572	if (pszSymbol)
573	{
574	size_t cch = strlen(pszSymbol);
575	char psz = (char )alloca(cch + 2);
576	memcpy(psz + 1, pszSymbol, cch + 1);
577	*psz = '_';
578	pszSymbol = psz;
579	}
580	#endif
581
582	int rc = kLdrModQuerySymbol(pModkLdr, pvBits, BaseAddress,
583	NIL_KLDRMOD_SYM_ORDINAL, pszSymbol, strlen(pszSymbol), NULL,
584	NULL, NULL, &uValue, NULL);
585	if (!rc)
586	{
587	*pValue = uValue;
588	return VINF_SUCCESS;
589	}
590	return rtkldrConvertError(rc);
591	}
592
593
594	/** @copydoc FNKLDRENUMDBG */
595	static int rtkldrEnumDbgInfoWrapper(PKLDRMOD pMod, KU32 iDbgInfo, KLDRDBGINFOTYPE enmType, KI16 iMajorVer, KI16 iMinorVer,
596	const char *pszPartNm, KLDRFOFF offFile, KLDRADDR LinkAddress, KLDRSIZE cb,
597	const char pszExtFile, void pvUser)
598	{
599	PRTLDRMODKLDRARGS pArgs = (PRTLDRMODKLDRARGS)pvUser;
600	NOREF(pMod);
601
602	RTLDRDBGINFO DbgInfo;
603	RT_ZERO(DbgInfo.u);
604	DbgInfo.iDbgInfo = iDbgInfo;
605	DbgInfo.offFile = offFile;
606	DbgInfo.LinkAddress = LinkAddress;
607	DbgInfo.cb = cb;
608	DbgInfo.pszExtFile = pszExtFile;
609
610	switch (enmType)
611	{
612	case KLDRDBGINFOTYPE_UNKNOWN:
613	DbgInfo.enmType = RTLDRDBGINFOTYPE_UNKNOWN;
614	break;
615	case KLDRDBGINFOTYPE_STABS:
616	DbgInfo.enmType = RTLDRDBGINFOTYPE_STABS;
617	break;
618	case KLDRDBGINFOTYPE_DWARF:
619	DbgInfo.enmType = RTLDRDBGINFOTYPE_DWARF;
620	if (pszExtFile)
621	DbgInfo.enmType = RTLDRDBGINFOTYPE_DWARF_DWO;
622	break;
623	case KLDRDBGINFOTYPE_CODEVIEW:
624	DbgInfo.enmType = RTLDRDBGINFOTYPE_CODEVIEW;
625	/* Should make some more effort here... Assume the IPRT loader will kick in before we get here! */
626	break;
627	case KLDRDBGINFOTYPE_WATCOM:
628	DbgInfo.enmType = RTLDRDBGINFOTYPE_WATCOM;
629	break;
630	case KLDRDBGINFOTYPE_HLL:
631	DbgInfo.enmType = RTLDRDBGINFOTYPE_HLL;
632	break;
633	default:
634	AssertFailed();
635	DbgInfo.enmType = RTLDRDBGINFOTYPE_UNKNOWN;
636	break;
637	}
638
639	int rc = pArgs->u.pfnEnumDbgInfo(&pArgs->pMod->Core, &DbgInfo, pArgs->pvUser);
640	if (RT_FAILURE(rc))
641	return rc; /* don't bother converting. */
642	return 0;
643	}
644
645
646	/** @copydoc RTLDROPS::pfnEnumDbgInfo */
647	static DECLCALLBACK(int) rtkldr_EnumDbgInfo(PRTLDRMODINTERNAL pMod, const void *pvBits,
648	PFNRTLDRENUMDBG pfnCallback, void *pvUser)
649	{
650	PRTLDRMODKLDR pThis = (PRTLDRMODKLDR)pMod;
651	RTLDRMODKLDRARGS Args;
652	Args.pvUser = pvUser;
653	Args.u.pfnEnumDbgInfo = pfnCallback;
654	Args.pvBits = pvBits;
655	Args.pMod = pThis;
656	Args.rc = VINF_SUCCESS;
657	int rc = kLdrModEnumDbgInfo(pThis->pMod, pvBits, rtkldrEnumDbgInfoWrapper, &Args);
658	if (Args.rc != VINF_SUCCESS)
659	rc = Args.rc;
660	return rc;
661	}
662
663
664	/** @copydoc RTLDROPS::pfnEnumSegments. */
665	static DECLCALLBACK(int) rtkldr_EnumSegments(PRTLDRMODINTERNAL pMod, PFNRTLDRENUMSEGS pfnCallback, void *pvUser)
666	{
667	PRTLDRMODKLDR pThis = (PRTLDRMODKLDR)pMod;
668	uint32_t const cSegments = pThis->pMod->cSegments;
669	PCKLDRSEG paSegments = &pThis->pMod->aSegments[0];
670
671	for (uint32_t iSeg = 0; iSeg < cSegments; iSeg++)
672	{
673	RTLDRSEG Seg;
674
675	Seg.pchName = paSegments[iSeg].pchName;
676	Seg.cchName = paSegments[iSeg].cchName;
677	Seg.SelFlat = paSegments[iSeg].SelFlat;
678	Seg.Sel16bit = paSegments[iSeg].Sel16bit;
679	Seg.fFlags = paSegments[iSeg].fFlags;
680	AssertCompile(KLDRSEG_FLAG_16BIT == RTLDRSEG_FLAG_16BIT );
681	AssertCompile(KLDRSEG_FLAG_OS2_ALIAS16 == RTLDRSEG_FLAG_OS2_ALIAS16);
682	AssertCompile(KLDRSEG_FLAG_OS2_CONFORM == RTLDRSEG_FLAG_OS2_CONFORM);
683	AssertCompile(KLDRSEG_FLAG_OS2_IOPL == RTLDRSEG_FLAG_OS2_IOPL );
684
685	switch (paSegments[iSeg].enmProt)
686	{
687	default:
688	AssertMsgFailed(("%d\n", paSegments[iSeg].enmProt));
689	case KPROT_NOACCESS:
690	Seg.fProt = 0;
691	break;
692
693	case KPROT_READONLY: Seg.fProt = RTMEM_PROT_READ; break;
694	case KPROT_READWRITE: Seg.fProt = RTMEM_PROT_READ \| RTMEM_PROT_WRITE; break;
695	case KPROT_WRITECOPY: Seg.fProt = RTMEM_PROT_WRITE; break;
696	case KPROT_EXECUTE: Seg.fProt = RTMEM_PROT_EXEC; break;
697	case KPROT_EXECUTE_READ: Seg.fProt = RTMEM_PROT_EXEC \| RTMEM_PROT_READ; break;
698	case KPROT_EXECUTE_READWRITE: Seg.fProt = RTMEM_PROT_EXEC \| RTMEM_PROT_READ \| RTMEM_PROT_WRITE; break;
699	case KPROT_EXECUTE_WRITECOPY: Seg.fProt = RTMEM_PROT_EXEC \| RTMEM_PROT_WRITE; break;
700	}
701	Seg.cb = paSegments[iSeg].cb;
702	Seg.Alignment = paSegments[iSeg].Alignment;
703	Seg.LinkAddress = paSegments[iSeg].LinkAddress;
704	Seg.offFile = paSegments[iSeg].offFile;
705	Seg.cbFile = paSegments[iSeg].cbFile;
706	Seg.RVA = paSegments[iSeg].RVA;
707	Seg.cbMapped = paSegments[iSeg].cbMapped;
708
709	int rc = pfnCallback(pMod, &Seg, pvUser);
710	if (rc != VINF_SUCCESS)
711	return rc;
712	}
713
714	return VINF_SUCCESS;
715	}
716
717
718	/** @copydoc RTLDROPS::pfnLinkAddressToSegOffset. */
719	static DECLCALLBACK(int) rtkldr_LinkAddressToSegOffset(PRTLDRMODINTERNAL pMod, RTLDRADDR LinkAddress,
720	uint32_t *piSeg, PRTLDRADDR poffSeg)
721	{
722	PRTLDRMODKLDR pThis = (PRTLDRMODKLDR)pMod;
723	uint32_t const cSegments = pThis->pMod->cSegments;
724	PCKLDRSEG paSegments = &pThis->pMod->aSegments[0];
725
726	for (uint32_t iSeg = 0; iSeg < cSegments; iSeg++)
727	{
728	KLDRADDR offSeg = LinkAddress - paSegments[iSeg].LinkAddress;
729	if ( offSeg < paSegments[iSeg].cbMapped
730	\|\| offSeg < paSegments[iSeg].cb)
731	{
732	*piSeg = iSeg;
733	*poffSeg = offSeg;
734	return VINF_SUCCESS;
735	}
736	}
737
738	return VERR_LDR_INVALID_LINK_ADDRESS;
739	}
740
741
742	/** @copydoc RTLDROPS::pfnLinkAddressToRva. */
743	static DECLCALLBACK(int) rtkldr_LinkAddressToRva(PRTLDRMODINTERNAL pMod, RTLDRADDR LinkAddress, PRTLDRADDR pRva)
744	{
745	PRTLDRMODKLDR pThis = (PRTLDRMODKLDR)pMod;
746	uint32_t const cSegments = pThis->pMod->cSegments;
747	PCKLDRSEG paSegments = &pThis->pMod->aSegments[0];
748
749	for (uint32_t iSeg = 0; iSeg < cSegments; iSeg++)
750	{
751	KLDRADDR offSeg = LinkAddress - paSegments[iSeg].LinkAddress;
752	if ( offSeg < paSegments[iSeg].cbMapped
753	\|\| offSeg < paSegments[iSeg].cb)
754	{
755	*pRva = paSegments[iSeg].RVA + offSeg;
756	return VINF_SUCCESS;
757	}
758	}
759
760	return VERR_LDR_INVALID_RVA;
761	}
762
763
764	/** @copydoc RTLDROPS::pfnSegOffsetToRva. */
765	static DECLCALLBACK(int) rtkldr_SegOffsetToRva(PRTLDRMODINTERNAL pMod, uint32_t iSeg, RTLDRADDR offSeg,
766	PRTLDRADDR pRva)
767	{
768	PRTLDRMODKLDR pThis = (PRTLDRMODKLDR)pMod;
769
770	if (iSeg >= pThis->pMod->cSegments)
771	return VERR_LDR_INVALID_SEG_OFFSET;
772	PCKLDRSEG const pSegment = &pThis->pMod->aSegments[iSeg];
773
774	if ( offSeg > pSegment->cbMapped
775	&& offSeg > pSegment->cb
776	&& ( pSegment->cbFile < 0
777	\|\| offSeg > (uint64_t)pSegment->cbFile))
778	return VERR_LDR_INVALID_SEG_OFFSET;
779
780	*pRva = pSegment->RVA + offSeg;
781	return VINF_SUCCESS;
782	}
783
784
785	/** @copydoc RTLDROPS::pfnRvaToSegOffset. */
786	static DECLCALLBACK(int) rtkldr_RvaToSegOffset(PRTLDRMODINTERNAL pMod, RTLDRADDR Rva,
787	uint32_t *piSeg, PRTLDRADDR poffSeg)
788	{
789	PRTLDRMODKLDR pThis = (PRTLDRMODKLDR)pMod;
790	uint32_t const cSegments = pThis->pMod->cSegments;
791	PCKLDRSEG paSegments = &pThis->pMod->aSegments[0];
792
793	for (uint32_t iSeg = 0; iSeg < cSegments; iSeg++)
794	{
795	KLDRADDR offSeg = Rva - paSegments[iSeg].RVA;
796	if ( offSeg < paSegments[iSeg].cbMapped
797	\|\| offSeg < paSegments[iSeg].cb)
798	{
799	*piSeg = iSeg;
800	*poffSeg = offSeg;
801	return VINF_SUCCESS;
802	}
803	}
804
805	return VERR_LDR_INVALID_RVA;
806	}
807
808
809	/**
810	* Operations for a kLdr module.
811	*/
812	static const RTLDROPS g_rtkldrOps =
813	{
814	"kLdr",
815	rtkldr_Close,
816	NULL,
817	rtkldr_Done,
818	rtkldr_EnumSymbols,
819	/* ext */
820	rtkldr_GetImageSize,
821	rtkldr_GetBits,
822	rtkldr_Relocate,
823	rtkldr_GetSymbolEx,
824	rtkldr_EnumDbgInfo,
825	rtkldr_EnumSegments,
826	rtkldr_LinkAddressToSegOffset,
827	rtkldr_LinkAddressToRva,
828	rtkldr_SegOffsetToRva,
829	rtkldr_RvaToSegOffset,
830	42
831	};
832
833
834	/**
835	* Open a image using kLdr.
836	*
837	* @returns iprt status code.
838	* @param pReader The loader reader instance which will provide the raw image bits.
839	* @param fFlags Reserved, MBZ.
840	* @param enmArch CPU architecture specifier for the image to be loaded.
841	* @param phLdrMod Where to store the handle.
842	*/
843	int rtldrkLdrOpen(PRTLDRREADER pReader, uint32_t fFlags, RTLDRARCH enmArch, PRTLDRMOD phLdrMod)
844	{
845	/* Convert enmArch to k-speak. */
846	KCPUARCH enmCpuArch;
847	switch (enmArch)
848	{
849	case RTLDRARCH_WHATEVER:
850	enmCpuArch = KCPUARCH_UNKNOWN;
851	break;
852	case RTLDRARCH_X86_32:
853	enmCpuArch = KCPUARCH_X86_32;
854	break;
855	case RTLDRARCH_AMD64:
856	enmCpuArch = KCPUARCH_AMD64;
857	break;
858	default:
859	return VERR_INVALID_PARAMETER;
860	}
861
862	/* Create a rtkldrRdr_ instance. */
863	PRTKLDRRDR pRdr = (PRTKLDRRDR)RTMemAllocZ(sizeof(*pRdr));
864	if (!pRdr)
865	return VERR_NO_MEMORY;
866	pRdr->Core.u32Magic = KRDR_MAGIC;
867	pRdr->Core.pOps = &g_kLdrRdrFileOps;
868	pRdr->pReader = pReader;
869
870	/* Try open it. */
871	PKLDRMOD pMod;
872	int krc = kLdrModOpenFromRdr(&pRdr->Core, fFlags, enmCpuArch, &pMod);
873	if (!krc)
874	{
875	/* Create a module wrapper for it. */
876	PRTLDRMODKLDR pNewMod = (PRTLDRMODKLDR)RTMemAllocZ(sizeof(*pNewMod));
877	if (pNewMod)
878	{
879	pNewMod->Core.u32Magic = RTLDRMOD_MAGIC;
880	pNewMod->Core.eState = LDR_STATE_OPENED;
881	pNewMod->Core.pOps = &g_rtkldrOps;
882	pNewMod->Core.pReader = pReader;
883	pNewMod->pMod = pMod;
884	*phLdrMod = &pNewMod->Core;
885
886	#ifdef LOG_ENABLED
887	Log(("rtldrkLdrOpen: '%s' (%s) %u segments\n",
888	pMod->pszName, pMod->pszFilename, pMod->cSegments));
889	for (unsigned iSeg = 0; iSeg < pMod->cSegments; iSeg++)
890	{
891	Log(("Segment #%-2u: RVA=%08llx cb=%08llx '%.*s'\n", iSeg,
892	pMod->aSegments[iSeg].RVA,
893	pMod->aSegments[iSeg].cb,
894	pMod->aSegments[iSeg].cchName,
895	pMod->aSegments[iSeg].pchName));
896	}
897	#endif
898	return VINF_SUCCESS;
899	}
900
901	/* bail out */
902	kLdrModClose(pMod);
903	krc = KERR_NO_MEMORY;
904	}
905	else
906	RTMemFree(pRdr);
907
908	return rtkldrConvertError(krc);
909	}
910

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Runtime/common/ldr/ldrkStuff.cpp@ 46087

Download in other formats: