tstLdr-3.cpp@ 8061

Last change on this file since 8061 was 7016, checked in by vboxsync, 17 years ago
Compile fixes
Property svn:eol-style set to `native` Property svn:keywords set to `Id`
File size: 8.3 KB

Line
1	/* $Id: tstLdr-3.cpp 7016 2008-02-19 13:32:39Z vboxsync $ */
2	/** @file
3	* innotek Portable Runtime - Testcase for parts of RTLdr*, manual inspection.
4	*/
5
6	/*
7	* Copyright (C) 2006-2007 innotek GmbH
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	#include <iprt/ldr.h>
32	#include <iprt/alloc.h>
33	#include <iprt/stream.h>
34	#include <iprt/assert.h>
35	#include <iprt/runtime.h>
36	#include <iprt/err.h>
37	#include <iprt/string.h>
38	#include <VBox/dis.h>
39
40
41	static bool MyDisBlock(PDISCPUSTATE pCpu, RTHCUINTPTR pvCodeBlock, int32_t cbMax, RTUINTPTR off, RTUINTPTR Addr)
42	{
43	int32_t i = 0;
44	while (i < cbMax)
45	{
46	char szOutput[256];
47	uint32_t cbInstr;
48	if (RT_FAILURE(DISInstr(pCpu, pvCodeBlock + i, off, &cbInstr, szOutput)))
49	return false;
50
51	RTPrintf("%s", szOutput);
52	if (pvCodeBlock + i + off == Addr)
53	RTPrintf("^^^^^^^^\n");
54
55	/* next */
56	i += cbInstr;
57	}
58	return true;
59	}
60
61
62
63	/**
64	* Resolve an external symbol during RTLdrGetBits().
65	*
66	* @returns iprt status code.
67	* @param hLdrMod The loader module handle.
68	* @param pszModule Module name.
69	* @param pszSymbol Symbol name, NULL if uSymbol should be used.
70	* @param uSymbol Symbol ordinal, ~0 if pszSymbol should be used.
71	* @param pValue Where to store the symbol value (address).
72	* @param pvUser User argument.
73	*/
74	static DECLCALLBACK(int) testGetImport(RTLDRMOD hLdrMod, const char pszModule, const char pszSymbol, unsigned uSymbol, RTUINTPTR pValue, void pvUser)
75	{
76	/* check the name format and only permit certain names */
77	*pValue = 0xf0f0f0f0;
78	return VINF_SUCCESS;
79	}
80
81
82	/**
83	* Enumeration callback function used by RTLdrEnumSymbols().
84	*
85	* @returns iprt status code. Failure will stop the enumeration.
86	* @param hLdrMod The loader module handle.
87	* @param pszSymbol Symbol name. NULL if ordinal only.
88	* @param uSymbol Symbol ordinal, ~0 if not used.
89	* @param Value Symbol value.
90	* @param pvUser The user argument specified to RTLdrEnumSymbols().
91	*/
92	static DECLCALLBACK(int) testEnumSymbol1(RTLDRMOD hLdrMod, const char pszSymbol, unsigned uSymbol, RTUINTPTR Value, void pvUser)
93	{
94	RTPrintf(" %RTptr %s (%d)\n", Value, pszSymbol, uSymbol);
95	return VINF_SUCCESS;
96	}
97
98	/**
99	* Current nearest symbol.
100	*/
101	typedef struct TESTNEARSYM
102	{
103	RTUINTPTR Addr;
104	struct TESTSYM
105	{
106	RTUINTPTR Value;
107	unsigned uSymbol;
108	char szName[512];
109	} aSyms[2];
110	} TESTNEARSYM, *PTESTNEARSYM;
111
112	/**
113	* Enumeration callback function used by RTLdrEnumSymbols().
114	*
115	* @returns iprt status code. Failure will stop the enumeration.
116	* @param hLdrMod The loader module handle.
117	* @param pszSymbol Symbol name. NULL if ordinal only.
118	* @param uSymbol Symbol ordinal, ~0 if not used.
119	* @param Value Symbol value.
120	* @param pvUser The user argument specified to RTLdrEnumSymbols().
121	*/
122	static DECLCALLBACK(int) testEnumSymbol2(RTLDRMOD hLdrMod, const char pszSymbol, unsigned uSymbol, RTUINTPTR Value, void pvUser)
123	{
124	PTESTNEARSYM pSym = (PTESTNEARSYM)pvUser;
125
126	/* less or equal */
127	if ( Value <= pSym->Addr
128	&& ( Value > pSym->aSyms[0].Value
129	\|\| ( Value == pSym->aSyms[0].Value
130	&& !pSym->aSyms[0].szName[0]
131	&& pszSymbol
132	&& *pszSymbol
133	)
134	)
135	)
136	{
137	pSym->aSyms[0].Value = Value;
138	pSym->aSyms[0].uSymbol = uSymbol;
139	pSym->aSyms[0].szName[0] = '\0';
140	if (pszSymbol)
141	strncat(pSym->aSyms[0].szName, pszSymbol, sizeof(pSym->aSyms[0].szName));
142	}
143
144	/* above */
145	if ( Value > pSym->Addr
146	&& ( Value < pSym->aSyms[1].Value
147	\|\| ( Value == pSym->aSyms[1].Value
148	&& !pSym->aSyms[1].szName[1]
149	&& pszSymbol
150	&& *pszSymbol
151	)
152	)
153	)
154	{
155	pSym->aSyms[1].Value = Value;
156	pSym->aSyms[1].uSymbol = uSymbol;
157	pSym->aSyms[1].szName[0] = '\0';
158	if (pszSymbol)
159	strncat(pSym->aSyms[1].szName, pszSymbol, sizeof(pSym->aSyms[1].szName));
160	}
161
162	return VINF_SUCCESS;
163	}
164
165
166	int main(int argc, char **argv)
167	{
168	RTR3Init();
169
170	int rcRet = 0;
171	if (argc <= 2)
172	{
173	RTPrintf("usage: %s <load-addr> <module> [addr1 []]\n", argv[0]);
174	return 1;
175	}
176
177	/*
178	* Load the module.
179	*/
180	RTUINTPTR LoadAddr = (RTUINTPTR)RTStrToUInt64(argv[1]);
181	RTLDRMOD hLdrMod;
182	int rc = RTLdrOpen(argv[2], &hLdrMod);
183	if (RT_FAILURE(rc))
184	{
185	RTPrintf("tstLdr-3: Failed to open '%s': %Rra\n", argv[2], rc);
186	return 1;
187	}
188
189	void *pvBits = RTMemAlloc(RTLdrSize(hLdrMod));
190	rc = RTLdrGetBits(hLdrMod, pvBits, LoadAddr, testGetImport, NULL);
191	if (RT_SUCCESS(rc))
192	{
193	if (argc > 3)
194	{
195	for (int i = 3; i < argc; i++)
196	{
197	TESTNEARSYM NearSym = {0};
198	NearSym.Addr = (RTUINTPTR)RTStrToUInt64(argv[i]);
199	NearSym.aSyms[1].Value = ~(RTUINTPTR)0;
200	rc = RTLdrEnumSymbols(hLdrMod, RTLDR_ENUM_SYMBOL_FLAGS_ALL, pvBits, LoadAddr, testEnumSymbol2, &NearSym);
201	if (RT_SUCCESS(rc))
202	{
203	RTPrintf("tstLdr-3: Addr=%RTptr\n"
204	"%RTptr %s (%d) - %RTptr %s (%d)\n",
205	NearSym.Addr,
206	NearSym.aSyms[0].Value, NearSym.aSyms[0].szName, NearSym.aSyms[0].uSymbol,
207	NearSym.aSyms[1].Value, NearSym.aSyms[1].szName, NearSym.aSyms[1].uSymbol);
208	if (NearSym.Addr - NearSym.aSyms[0].Value < 0x10000)
209	{
210	DISCPUSTATE Cpu;
211	memset(&Cpu, 0, sizeof(Cpu));
212	Cpu.mode = CPUMODE_32BIT;
213	uint8_t pbCode = (uint8_t )pvBits + (NearSym.aSyms[0].Value - LoadAddr);
214	MyDisBlock(&Cpu, (uintptr_t)pbCode,
215	RT_MAX(NearSym.aSyms[1].Value - NearSym.aSyms[0].Value, 0x20000),
216	NearSym.aSyms[0].Value - (RTUINTPTR)pbCode,
217	NearSym.Addr);
218	}
219	}
220	else
221	{
222	RTPrintf("tstLdr-3: Failed to enumerate symbols: %Rra\n", rc);
223	rcRet++;
224	}
225	}
226	}
227	else
228	{
229	/*
230	* Enumerate symbols.
231	*/
232	rc = RTLdrEnumSymbols(hLdrMod, RTLDR_ENUM_SYMBOL_FLAGS_ALL, pvBits, LoadAddr, testEnumSymbol1, NULL);
233	if (RT_FAILURE(rc))
234	{
235	RTPrintf("tstLdr-3: Failed to enumerate symbols: %Rra\n", rc);
236	rcRet++;
237	}
238	}
239	}
240	else
241	{
242	RTPrintf("tstLdr-3: Failed to get bits for '%s' at %RTptr: %Rra\n", argv[2], LoadAddr, rc);
243	rcRet++;
244	}
245	RTMemFree(pvBits);
246	RTLdrClose(hLdrMod);
247
248	/*
249	* Test result summary.
250	*/
251	if (!rcRet)
252	RTPrintf("tstLdr-3: SUCCESS\n");
253	else
254	RTPrintf("tstLdr-3: FAILURE - %d errors\n", rcRet);
255	return !!rcRet;
256	}

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

source: vbox/trunk/src/VBox/Runtime/testcase/tstLdr-3.cpp@ 8061

Download in other formats: