VBoxDef2LazyLoad.cpp@ 63562

Last change on this file since 63562 was 63193, checked in by vboxsync, 8 years ago
unused function warning
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 38.9 KB

Line
1	/* $Id: VBoxDef2LazyLoad.cpp 63193 2016-08-09 08:49:28Z vboxsync $ */
2	/** @file
3	* VBoxDef2LazyLoad - Lazy Library Loader Generator.
4	*
5	* @note Only tested on win.amd64 & darwin.amd64.
6	*/
7
8	/*
9	* Copyright (C) 2013-2016 Oracle Corporation
10	*
11	* This file is part of VirtualBox Open Source Edition (OSE), as
12	* available from http://www.virtualbox.org. This file is free software;
13	* you can redistribute it and/or modify it under the terms of the GNU
14	* General Public License (GPL) as published by the Free Software
15	* Foundation, in version 2 as it comes in the "COPYING" file of the
16	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
17	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
18	*/
19
20
21	/*********************************************************************************************************************************
22	* Header Files *
23	*********************************************************************************************************************************/
24	#include <ctype.h>
25	#include <stdio.h>
26	#include <string.h>
27	#include <stdlib.h>
28	#include <iprt/types.h>
29
30
31	/*********************************************************************************************************************************
32	* Structures and Typedefs *
33	*********************************************************************************************************************************/
34	typedef struct MYEXPORT
35	{
36	struct MYEXPORT *pNext;
37	/** Pointer to unmangled name for stdcall (after szName), NULL if not. */
38	char *pszUnstdcallName;
39	/** Pointer to the exported name. */
40	char const *pszExportedNm;
41	unsigned uOrdinal;
42	bool fNoName;
43	char szName[1];
44	} MYEXPORT;
45	typedef MYEXPORT *PMYEXPORT;
46
47
48
49	/*********************************************************************************************************************************
50	* Global Variables *
51	*********************************************************************************************************************************/
52	/** @name Options
53	* @{ */
54	static const char *g_pszOutput = NULL;
55	static const char *g_pszLibrary = NULL;
56	static const char *g_apszInputs[8] = { NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL };
57	static unsigned g_cInputs = 0;
58	static bool g_fIgnoreData = true;
59	static bool g_fWithExplictLoadFunction = false;
60	static bool g_fSystemLibrary = false;
61	/** @} */
62
63	/** Pointer to the export name list head. */
64	static PMYEXPORT g_pExpHead = NULL;
65	/** Pointer to the next pointer for insertion. */
66	static PMYEXPORT *g_ppExpNext = &g_pExpHead;
67
68
69
70	#if 0 /* unused */
71	static const char leftStrip(const char psz)
72	{
73	while (isspace(*psz))
74	psz++;
75	return psz;
76	}
77	#endif
78
79
80	static char leftStrip(char psz)
81	{
82	while (isspace(*psz))
83	psz++;
84	return psz;
85	}
86
87
88	static unsigned wordLength(const char *pszWord)
89	{
90	unsigned off = 0;
91	char ch;
92	while ( (ch = pszWord[off]) != '\0'
93	&& ch != '='
94	&& ch != ','
95	&& ch != ':'
96	&& !isspace(ch) )
97	off++;
98	return off;
99	}
100
101
102	/**
103	* Parses the module definition file, collecting export information.
104	*
105	* @returns RTEXITCODE_SUCCESS or RTEXITCODE_FAILURE, in the latter case full
106	* details has been displayed.
107	* @param pInput The input stream.
108	*/
109	static RTEXITCODE parseInputInner(FILE pInput, const char pszInput)
110	{
111	/*
112	* Process the file line-by-line.
113	*/
114	bool fInExports = false;
115	unsigned iLine = 0;
116	char szLine[16384];
117	while (fgets(szLine, sizeof(szLine), pInput))
118	{
119	iLine++;
120
121	/*
122	* Strip leading and trailing spaces from the line as well as
123	* trailing comments.
124	*/
125	char *psz = leftStrip(szLine);
126	if (*psz == ';')
127	continue; /* comment line. */
128
129	char *pszComment = strchr(psz, ';');
130	if (pszComment)
131	*pszComment = '\0';
132
133	unsigned cch = (unsigned)strlen(psz);
134	while (cch > 0 && (isspace(psz[cch - 1]) \|\| psz[cch - 1] == '\r' \|\| psz[cch - 1] == '\n'))
135	psz[--cch] = '\0';
136
137	if (!cch)
138	continue;
139
140	/*
141	* Check for known directives.
142	*/
143	size_t cchWord0 = wordLength(psz);
144	#define WORD_CMP(pszWord1, cchWord1, szWord2) \
145	( (cchWord1) == sizeof(szWord2) - 1 && memcmp(pszWord1, szWord2, sizeof(szWord2) - 1) == 0 )
146	if (WORD_CMP(psz, cchWord0, "EXPORTS"))
147	{
148	fInExports = true;
149
150	/* In case there is an export on the same line. (Really allowed?) */
151	psz = leftStrip(psz + sizeof("EXPORTS") - 1);
152	if (!*psz)
153	continue;
154	}
155	/* Directives that we don't care about, but need to catch in order to
156	terminate the EXPORTS section in a timely manner. */
157	else if ( WORD_CMP(psz, cchWord0, "NAME")
158	\|\| WORD_CMP(psz, cchWord0, "LIBRARY")
159	\|\| WORD_CMP(psz, cchWord0, "DESCRIPTION")
160	\|\| WORD_CMP(psz, cchWord0, "STACKSIZE")
161	\|\| WORD_CMP(psz, cchWord0, "SECTIONS")
162	\|\| WORD_CMP(psz, cchWord0, "SEGMENTS")
163	\|\| WORD_CMP(psz, cchWord0, "VERSION")
164	)
165	{
166	fInExports = false;
167	}
168
169	/*
170	* Process exports:
171	* entryname[=internalname] [@ordinal[ ][NONAME]] [DATA] [PRIVATE]
172	*/
173	if (fInExports)
174	{
175	const char *pchName = psz;
176	unsigned cchName = wordLength(psz);
177
178	psz = leftStrip(psz + cchName);
179	if (*psz == '=')
180	{
181	psz = leftStrip(psz + 1);
182	psz = leftStrip(psz + wordLength(psz));
183	}
184
185	bool fNoName = false;
186	unsigned uOrdinal = ~0U;
187	if (*psz == '@')
188	{
189	psz++;
190	if (!isdigit(*psz))
191	{
192	fprintf(stderr, "%s:%u: error: Invalid ordinal spec.\n", pszInput, iLine);
193	return RTEXITCODE_FAILURE;
194	}
195	uOrdinal = *psz++ - '0';
196	while (isdigit(*psz))
197	{
198	uOrdinal *= 10;
199	uOrdinal += *psz++ - '0';
200	}
201	psz = leftStrip(psz);
202	cch = wordLength(psz);
203	if (WORD_CMP(psz, cch, "NONAME"))
204	{
205	fNoName = true;
206	psz = leftStrip(psz + cch);
207	}
208	}
209
210	while (*psz)
211	{
212	cch = wordLength(psz);
213	if (WORD_CMP(psz, cch, "DATA"))
214	{
215	if (!g_fIgnoreData)
216	{
217	fprintf(stderr, "%s:%u: error: Cannot wrap up DATA export '%.*s'.\n",
218	pszInput, iLine, cchName, pchName);
219	return RTEXITCODE_SUCCESS;
220	}
221	}
222	else if (!WORD_CMP(psz, cch, "PRIVATE"))
223	{
224	fprintf(stderr, "%s:%u: error: Cannot wrap up DATA export '%.*s'.\n",
225	pszInput, iLine, cchName, pchName);
226	return RTEXITCODE_SUCCESS;
227	}
228	psz = leftStrip(psz + cch);
229	}
230
231	/*
232	* Check for stdcall mangling.
233	*/
234	size_t cbExp = sizeof(MYEXPORT) + cchName;
235	unsigned cchStdcall = 0;
236	if (cchName > 3 && *pchName == '_' && isdigit(pchName[cchName - 1]))
237	{
238	if (cchName > 3 && pchName[cchName - 2] == '@')
239	cchStdcall = 2;
240	else if (cchName > 4 && pchName[cchName - 3] == '@' && isdigit(pchName[cchName - 2]))
241	cchStdcall = 3;
242	if (cchStdcall)
243	cbExp += cchName - 1 - cchStdcall;
244	}
245
246	/*
247	* Add the export.
248	*/
249
250	PMYEXPORT pExp = (PMYEXPORT)malloc(cbExp);
251	if (!pExp)
252	{
253	fprintf(stderr, "%s:%u: error: Out of memory.\n", pszInput, iLine);
254	return RTEXITCODE_SUCCESS;
255	}
256	memcpy(pExp->szName, pchName, cchName);
257	pExp->szName[cchName] = '\0';
258	if (!cchStdcall)
259	{
260	pExp->pszUnstdcallName = NULL;
261	pExp->pszExportedNm = pExp->szName;
262	}
263	else
264	{
265	pExp->pszUnstdcallName = &pExp->szName[cchName + 1];
266	memcpy(pExp->pszUnstdcallName, pchName + 1, cchName - 1 - cchStdcall);
267	pExp->pszUnstdcallName[cchName - 1 - cchStdcall] = '\0';
268	pExp->pszExportedNm = pExp->pszUnstdcallName;
269	}
270	pExp->uOrdinal = uOrdinal;
271	pExp->fNoName = fNoName;
272	pExp->pNext = NULL;
273	*g_ppExpNext = pExp;
274	g_ppExpNext = &pExp->pNext;
275	}
276	}
277
278	/*
279	* Why did we quit the loop, EOF or error?
280	*/
281	if (feof(pInput))
282	return RTEXITCODE_SUCCESS;
283	fprintf(stderr, "error: Read while reading '%s' (iLine=%u).\n", pszInput, iLine);
284	return RTEXITCODE_FAILURE;
285	}
286
287
288	/**
289	* Parses a_apszInputs, populating the list pointed to by g_pExpHead.
290	*
291	* @returns RTEXITCODE_SUCCESS or RTEXITCODE_FAILURE, in the latter case full
292	* details has been displayed.
293	*/
294	static RTEXITCODE parseInputs(void)
295	{
296	RTEXITCODE rcExit = RTEXITCODE_SUCCESS;
297	for (unsigned i = 0; i < g_cInputs; i++)
298	{
299	FILE *pInput = fopen(g_apszInputs[i], "r");
300	if (pInput)
301	{
302	RTEXITCODE rcExit2 = parseInputInner(pInput, g_apszInputs[i]);
303	fclose(pInput);
304	if (rcExit2 == RTEXITCODE_SUCCESS && !g_pExpHead)
305	{
306	fprintf(stderr, "error: Found no exports in '%s'.\n", g_apszInputs[i]);
307	rcExit2 = RTEXITCODE_FAILURE;
308	}
309	if (rcExit2 != RTEXITCODE_SUCCESS)
310	rcExit = rcExit2;
311	}
312	else
313	{
314	fprintf(stderr, "error: Failed to open '%s' for reading.\n", g_apszInputs[i]);
315	rcExit = RTEXITCODE_FAILURE;
316	}
317	}
318	return rcExit;
319	}
320
321
322	/**
323	* Generates the assembly source code, writing it to @a pOutput.
324	*
325	* @returns RTEXITCODE_SUCCESS or RTEXITCODE_FAILURE, in the latter case full
326	* details has been displayed.
327	* @param pOutput The output stream (caller checks it for errors
328	* when closing).
329	*/
330	static RTEXITCODE generateOutputInner(FILE *pOutput)
331	{
332	fprintf(pOutput, ";;\n");
333	for (unsigned i = 0; i < g_cInputs; i++)
334	fprintf(pOutput, ";; Autogenerated from '%s'.\n", g_apszInputs[i]);
335
336	fprintf(pOutput,
337	";; DO NOT EDIT!\n"
338	";;\n"
339	"\n"
340	"\n"
341	"%%include \"iprt/asmdefs.mac\"\n"
342	"\n"
343	"\n");
344
345	/*
346	* Put the thunks first for alignment and other reasons. It's the hot part of the code.
347	*/
348	fprintf(pOutput,
349	";\n"
350	"; Thunks.\n"
351	";\n"
352	"BEGINCODE\n");
353	for (PMYEXPORT pExp = g_pExpHead; pExp; pExp = pExp->pNext)
354	if (!pExp->pszUnstdcallName)
355	fprintf(pOutput,
356	"BEGINPROC %s\n"
357	" jmp RTCCPTR_PRE [g_pfn%s xWrtRIP]\n"
358	"ENDPROC %s\n",
359	pExp->szName, pExp->szName, pExp->szName);
360	else
361	fprintf(pOutput,
362	"%%ifdef RT_ARCH_X86\n"
363	"global %s\n"
364	"%s:\n"
365	" jmp RTCCPTR_PRE [g_pfn%s xWrtRIP]\n"
366	"%%else\n"
367	"BEGINPROC %s\n"
368	" jmp RTCCPTR_PRE [g_pfn%s xWrtRIP]\n"
369	"ENDPROC %s\n"
370	"%%endif\n",
371	pExp->szName, pExp->szName, pExp->pszUnstdcallName,
372	pExp->pszUnstdcallName, pExp->pszUnstdcallName, pExp->pszUnstdcallName);
373
374	fprintf(pOutput,
375	"\n"
376	"\n");
377
378	/*
379	* Import pointers
380	*/
381	fprintf(pOutput,
382	";\n"
383	"; Import pointers. Initialized to point a lazy loading stubs.\n"
384	";\n"
385	"BEGINDATA\n"
386	"g_apfnImports:\n");
387	for (PMYEXPORT pExp = g_pExpHead; pExp; pExp = pExp->pNext)
388	fprintf(pOutput,
389	"%%ifdef ASM_FORMAT_PE\n"
390	"global __imp_%s\n"
391	"__imp_%s:\n"
392	"%%endif\n"
393	"g_pfn%s RTCCPTR_DEF ___LazyLoad___%s\n"
394	"\n",
395	pExp->szName,
396	pExp->szName,
397	pExp->pszExportedNm,
398	pExp->pszExportedNm);
399	fprintf(pOutput,
400	"RTCCPTR_DEF 0 ; Terminator entry for traversal.\n"
401	"\n"
402	"\n");
403
404	/*
405	* Now for the less important stuff, starting with the names.
406	*
407	* We keep the names separate so we can traverse them in parallel to
408	* g_apfnImports in the load-everything routine further down.
409	*/
410	fprintf(pOutput,
411	";\n"
412	"; Imported names.\n"
413	";\n"
414	"BEGINCODE\n"
415	"g_szLibrary: db '%s',0\n"
416	"\n"
417	"g_szzNames:\n",
418	g_pszLibrary);
419	for (PMYEXPORT pExp = g_pExpHead; pExp; pExp = pExp->pNext)
420	if (!pExp->fNoName)
421	fprintf(pOutput, " g_sz%s:\n db '%s',0\n", pExp->pszExportedNm, pExp->pszExportedNm);
422	else
423	fprintf(pOutput, " g_sz%s:\n db '#%u',0\n", pExp->pszExportedNm, pExp->uOrdinal);
424	fprintf(pOutput,
425	"g_EndOfNames: db 0\n"
426	"\n"
427	"g_szFailLoadFmt: db 'Lazy loader failed to load \"%%s\": %%Rrc', 10, 0\n"
428	"g_szFailResolveFmt: db 'Lazy loader failed to resolve symbol \"%%s\" in \"%%s\": %%Rrc', 10, 0\n"
429	"\n"
430	"\n");
431
432	/*
433	* The per import lazy load code.
434	*/
435	fprintf(pOutput,
436	";\n"
437	"; Lazy load+resolve stubs.\n"
438	";\n"
439	"BEGINCODE\n");
440	for (PMYEXPORT pExp = g_pExpHead; pExp; pExp = pExp->pNext)
441	{
442	if (!pExp->fNoName)
443	fprintf(pOutput,
444	"___LazyLoad___%s:\n"
445	/* "int3\n" */
446	"%%ifdef RT_ARCH_AMD64\n"
447	" lea rax, [g_sz%s wrt rip]\n"
448	" lea r10, [g_pfn%s wrt rip]\n"
449	" call LazyLoadResolver\n"
450	"%%elifdef RT_ARCH_X86\n"
451	" push g_sz%s\n"
452	" push g_pfn%s\n"
453	" call LazyLoadResolver\n"
454	" add esp, 8h\n"
455	"%%else\n"
456	" %%error \"Unsupported architecture\"\n"
457	"%%endif\n"
458	,
459	pExp->pszExportedNm,
460	pExp->pszExportedNm,
461	pExp->pszExportedNm,
462	pExp->pszExportedNm,
463	pExp->pszExportedNm);
464	else
465	fprintf(pOutput,
466	"___LazyLoad___%s:\n"
467	/* "int3\n" */
468	"%%ifdef RT_ARCH_AMD64\n"
469	" mov eax, %u\n"
470	" lea r10, [g_pfn%s wrt rip]\n"
471	" call LazyLoadResolver\n"
472	"%%elifdef RT_ARCH_X86\n"
473	" push %u\n"
474	" push g_pfn%s\n"
475	" call LazyLoadResolver\n"
476	" add esp, 8h\n"
477	"%%else\n"
478	" %%error \"Unsupported architecture\"\n"
479	"%%endif\n"
480	,
481	pExp->pszExportedNm,
482	pExp->uOrdinal,
483	pExp->pszExportedNm,
484	pExp->uOrdinal,
485	pExp->pszExportedNm);
486	if (!pExp->pszUnstdcallName)
487	fprintf(pOutput, " jmp NAME(%s)\n", pExp->szName);
488	else
489	fprintf(pOutput,
490	"%%ifdef RT_ARCH_X86\n"
491	" jmp %s\n"
492	"%%else\n"
493	" jmp NAME(%s)\n"
494	"%%endif\n"
495	,
496	pExp->szName, pExp->szName);
497	fprintf(pOutput, "\n");
498	}
499	fprintf(pOutput,
500	"\n"
501	"\n"
502	"\n");
503
504	/*
505	* The code that does the loading and resolving.
506	*/
507	fprintf(pOutput,
508	";\n"
509	"; The module handle.\n"
510	";\n"
511	"BEGINDATA\n"
512	"g_hMod RTCCPTR_DEF 0\n"
513	"\n"
514	"\n"
515	"\n");
516
517	/*
518	* How we load the module needs to be selectable later on.
519	*
520	* The LazyLoading routine returns the module handle in RCX/ECX, caller
521	* saved all necessary registers.
522	*/
523	if (!g_fSystemLibrary)
524	fprintf(pOutput,
525	";\n"
526	";SUPR3DECL(int) SUPR3HardenedLdrLoadAppPriv(const char *pszFilename, PRTLDRMOD phLdrMod,\n"
527	"; uint32_t fFlags, PRTERRINFO pErrInfo);\n"
528	";\n"
529	"EXTERN_IMP2 SUPR3HardenedLdrLoadAppPriv\n"
530	"%%ifdef IN_RT_R3\n"
531	"extern NAME(RTAssertMsg2Weak)\n"
532	"%%else\n"
533	"EXTERN_IMP2 RTAssertMsg2Weak\n"
534	"%%endif\n"
535	"BEGINCODE\n"
536	"\n"
537	"LazyLoading:\n"
538	" mov xCX, [g_hMod xWrtRIP]\n"
539	" or xCX, xCX\n"
540	" jnz .return\n"
541	"\n"
542	"%%ifdef ASM_CALL64_GCC\n"
543	" xor rcx, rcx ; pErrInfo\n"
544	" xor rdx, rdx ; fFlags (local load)\n"
545	" lea rsi, [g_hMod wrt rip] ; phLdrMod\n"
546	" lea rdi, [g_szLibrary wrt rip] ; pszFilename\n"
547	" sub rsp, 08h\n"
548	" call IMP2(SUPR3HardenedLdrLoadAppPriv)\n"
549	" add rsp, 08h\n"
550	"\n"
551	"%%elifdef ASM_CALL64_MSC\n"
552	" xor r9, r9 ; pErrInfo\n"
553	" xor r8, r8 ; fFlags (local load)\n"
554	" lea rdx, [g_hMod wrt rip] ; phLdrMod\n"
555	" lea rcx, [g_szLibrary wrt rip] ; pszFilename\n"
556	" sub rsp, 28h\n"
557	" call IMP2(SUPR3HardenedLdrLoadAppPriv)\n"
558	" add rsp, 28h\n"
559	"\n"
560	"%%elifdef RT_ARCH_X86\n"
561	" sub xSP, 0ch\n"
562	" push 0 ; pErrInfo\n"
563	" push 0 ; fFlags (local load)\n"
564	" push g_hMod ; phLdrMod\n"
565	" push g_szLibrary ; pszFilename\n"
566	" call IMP2(SUPR3HardenedLdrLoadAppPriv)\n"
567	" add esp, 1ch\n"
568	"%%else\n"
569	" %%error \"Unsupported architecture\"\n"
570	"%%endif\n");
571	else
572	fprintf(pOutput,
573	";\n"
574	"; RTDECL(int) RTLdrLoadSystem(const char *pszFilename, bool fNoUnload, PRTLDRMOD phLdrMod);\n"
575	";\n"
576	"%%ifdef IN_RT_R3\n"
577	"extern NAME(RTLdrLoadSystem)\n"
578	"extern NAME(RTAssertMsg2Weak)\n"
579	"%%else\n"
580	"EXTERN_IMP2 RTLdrLoadSystem\n"
581	"EXTERN_IMP2 RTAssertMsg2Weak\n"
582	"%%endif\n"
583	"BEGINCODE\n"
584	"\n"
585	"LazyLoading:\n"
586	" mov xCX, [g_hMod xWrtRIP]\n"
587	" or xCX, xCX\n"
588	" jnz .return\n"
589	"\n"
590	"%%ifdef ASM_CALL64_GCC\n"
591	" lea rdx, [g_hMod wrt rip] ; phLdrMod\n"
592	" mov esi, 1 ; fNoUnload=true\n"
593	" lea rdi, [g_szLibrary wrt rip] ; pszFilename\n"
594	" sub rsp, 08h\n"
595	" %%ifdef IN_RT_R3\n"
596	" call NAME(RTLdrLoadSystem)\n"
597	" %%else\n"
598	" call IMP2(RTLdrLoadSystem)\n"
599	" %%endif\n"
600	" add rsp, 08h\n"
601	"\n"
602	"%%elifdef ASM_CALL64_MSC\n"
603	" lea r8, [g_hMod wrt rip] ; phLdrMod\n"
604	" mov edx, 1 ; fNoUnload=true\n"
605	" lea rcx, [g_szLibrary wrt rip] ; pszFilename\n"
606	" sub rsp, 28h\n"
607	" %%ifdef IN_RT_R3\n"
608	" call NAME(RTLdrLoadSystem)\n"
609	" %%else\n"
610	" call IMP2(RTLdrLoadSystem)\n"
611	" %%endif\n"
612	" add rsp, 28h\n"
613	"\n"
614	"%%elifdef RT_ARCH_X86\n"
615	" push g_hMod ; phLdrMod\n"
616	" push 1 ; fNoUnload=true\n"
617	" push g_szLibrary ; pszFilename\n"
618	" %%ifdef IN_RT_R3\n"
619	" call NAME(RTLdrLoadSystem)\n"
620	" %%else\n"
621	" call IMP2(RTLdrLoadSystem)\n"
622	" %%endif\n"
623	" add esp, 0ch\n"
624	"%%else\n"
625	" %%error \"Unsupported architecture\"\n"
626	"%%endif\n");
627	fprintf(pOutput,
628	" or eax, eax\n"
629	" jnz .badload\n"
630	" mov xCX, [g_hMod xWrtRIP]\n"
631	".return:\n"
632	" ret\n"
633	"\n"
634	".badload:\n"
635	"%%ifdef ASM_CALL64_GCC\n"
636	" mov edx, eax\n"
637	" lea rsi, [g_szLibrary wrt rip]\n"
638	" lea rdi, [g_szFailLoadFmt wrt rip]\n"
639	" sub rsp, 08h\n"
640	"%%elifdef ASM_CALL64_MSC\n"
641	" mov r8d, eax\n"
642	" lea rdx, [g_szLibrary wrt rip]\n"
643	" lea rcx, [g_szFailLoadFmt wrt rip]\n"
644	" sub rsp, 28h\n"
645	"%%elifdef RT_ARCH_X86\n"
646	" push eax\n"
647	" push g_szLibrary\n"
648	" push g_szFailLoadFmt\n"
649	"%%endif\n"
650	"%%ifdef IN_RT_R3\n"
651	" call NAME(RTAssertMsg2Weak)\n"
652	"%%else\n"
653	" call IMP2(RTAssertMsg2Weak)\n"
654	"%%endif\n"
655	".badloadloop:\n"
656	" int3\n"
657	" jmp .badloadloop\n"
658	"LazyLoading_End:\n"
659	"\n"
660	"\n");
661
662
663	fprintf(pOutput,
664	";\n"
665	";RTDECL(int) RTLdrGetSymbol(RTLDRMOD hLdrMod, const char pszSymbol, void *ppvValue);\n"
666	";\n"
667	"%%ifdef IN_RT_R3\n"
668	"extern NAME(RTLdrGetSymbol)\n"
669	"%%else\n"
670	"EXTERN_IMP2 RTLdrGetSymbol\n"
671	"%%endif\n"
672	"BEGINCODE\n"
673	"LazyLoadResolver:\n"
674	"%%ifdef RT_ARCH_AMD64\n"
675	" push rbp\n"
676	" mov rbp, rsp\n"
677	" push r15\n"
678	" push r14\n"
679	" mov r15, rax ; name\n"
680	" mov r14, r10 ; ppfn\n"
681	" push r9\n"
682	" push r8\n"
683	" push rcx\n"
684	" push rdx\n"
685	" push r12\n"
686	" %%ifdef ASM_CALL64_GCC\n"
687	" push rsi\n"
688	" push rdi\n"
689	" mov r12, rsp\n"
690	" %%else\n"
691	" mov r12, rsp\n"
692	" sub rsp, 20h\n"
693	" %%endif\n"
694	" and rsp, 0fffffff0h ; Try make sure the stack is aligned\n"
695	"\n"
696	" call LazyLoading ; returns handle in rcx\n"
697	" %%ifdef ASM_CALL64_GCC\n"
698	" mov rdi, rcx ; hLdrMod\n"
699	" mov rsi, r15 ; pszSymbol\n"
700	" mov rdx, r14 ; ppvValue\n"
701	" %%else\n"
702	" mov rdx, r15 ; pszSymbol\n"
703	" mov r8, r14 ; ppvValue\n"
704	" %%endif\n"
705	" %%ifdef IN_RT_R3\n"
706	" call NAME(RTLdrGetSymbol)\n"
707	" %%else\n"
708	" call IMP2(RTLdrGetSymbol)\n"
709	" %%endif\n"
710	" or eax, eax\n"
711	" jnz .badsym\n"
712	"\n"
713	" mov rsp, r12\n"
714	" %%ifdef ASM_CALL64_GCC\n"
715	" pop rdi\n"
716	" pop rsi\n"
717	" %%endif\n"
718	" pop r12\n"
719	" pop rdx\n"
720	" pop rcx\n"
721	" pop r8\n"
722	" pop r9\n"
723	" pop r14\n"
724	" pop r15\n"
725	" leave\n"
726	"\n"
727	"%%elifdef RT_ARCH_X86\n"
728	" push ebp\n"
729	" mov ebp, esp\n"
730	" push eax\n"
731	" push ecx\n"
732	" push edx\n"
733	" and esp, 0fffffff0h\n"
734	"\n"
735	".loaded:\n"
736	" call LazyLoading ; returns handle in ecx\n"
737	" push dword [ebp + 8] ; value addr\n"
738	" push dword [ebp + 12] ; symbol name\n"
739	" push ecx\n"
740	" %%ifdef IN_RT_R3\n"
741	" call NAME(RTLdrGetSymbol)\n"
742	" %%else\n"
743	" call IMP2(RTLdrGetSymbol)\n"
744	" %%endif\n"
745	" or eax, eax\n"
746	" jnz .badsym\n"
747	" lea esp, [ebp - 0ch]\n"
748	" pop edx\n"
749	" pop ecx\n"
750	" pop eax\n"
751	" leave\n"
752	"%%else\n"
753	" %%error \"Unsupported architecture\"\n"
754	"%%endif\n"
755	" ret\n"
756	"\n"
757	".badsym:\n"
758	"%%ifdef ASM_CALL64_GCC\n"
759	" mov ecx, eax\n"
760	" lea rdx, [g_szLibrary wrt rip]\n"
761	" mov rsi, r15\n"
762	" lea rdi, [g_szFailResolveFmt wrt rip]\n"
763	" sub rsp, 08h\n"
764	"%%elifdef ASM_CALL64_MSC\n"
765	" mov r9d, eax\n"
766	" mov r8, r15\n"
767	" lea rdx, [g_szLibrary wrt rip]\n"
768	" lea rcx, [g_szFailResolveFmt wrt rip]\n"
769	" sub rsp, 28h\n"
770	"%%elifdef RT_ARCH_X86\n"
771	" push eax\n"
772	" push dword [ebp + 12]\n"
773	" push g_szLibrary\n"
774	" push g_szFailResolveFmt\n"
775	"%%endif\n"
776	"%%ifdef IN_RT_R3\n"
777	" call NAME(RTAssertMsg2Weak)\n"
778	"%%else\n"
779	" call IMP2(RTAssertMsg2Weak)\n"
780	"%%endif\n"
781	".badsymloop:\n"
782	" int3\n"
783	" jmp .badsymloop\n"
784	"\n"
785	"LazyLoadResolver_End:\n"
786	"\n"
787	"\n"
788	);
789
790
791
792	/*
793	* C callable method for explicitly loading the library and optionally
794	* resolving all the imports.
795	*/
796	if (g_fWithExplictLoadFunction)
797	{
798	if (g_fSystemLibrary) /* Lazy bird. */
799	{
800	fprintf(stderr, "error: cannot use --system with --explicit-load-function, sorry\n");
801	return RTEXITCODE_FAILURE;
802	}
803
804	int cchLibBaseName = (int)(strchr(g_pszLibrary, '.') ? strchr(g_pszLibrary, '.') - g_pszLibrary : strlen(g_pszLibrary));
805	fprintf(pOutput,
806	";;\n"
807	"; ExplicitlyLoad%.*s(bool fResolveAllImports, pErrInfo);\n"
808	";\n"
809	"EXTERN_IMP2 RTErrInfoSet\n"
810	"BEGINCODE\n"
811	"BEGINPROC ExplicitlyLoad%.*s\n"
812	" push xBP\n"
813	" mov xBP, xSP\n"
814	" push xBX\n"
815	"%%ifdef ASM_CALL64_GCC\n"
816	" %%define pszCurStr r14\n"
817	" push r14\n"
818	"%%else\n"
819	" %%define pszCurStr xDI\n"
820	" push xDI\n"
821	"%%endif\n"
822	" sub xSP, 40h\n"
823	"\n"
824	" ;\n"
825	" ; Save parameters on stack (64-bit only).\n"
826	" ;\n"
827	"%%ifdef ASM_CALL64_GCC\n"
828	" mov [xBP - xCB * 3], rdi ; fResolveAllImports\n"
829	" mov [xBP - xCB * 4], rsi ; pErrInfo\n"
830	"%%elifdef ASM_CALL64_MSC\n"
831	" mov [xBP - xCB * 3], rcx ; fResolveAllImports\n"
832	" mov [xBP - xCB * 4], rdx ; pErrInfo\n"
833	"%%endif\n"
834	"\n"
835	" ;\n"
836	" ; Is the module already loaded?\n"
837	" ;\n"
838	" cmp RTCCPTR_PRE [g_hMod xWrtRIP], 0\n"
839	" jnz .loaded\n"
840	"\n"
841	" ;\n"
842	" ; Load the module.\n"
843	" ;\n"
844	"%%ifdef ASM_CALL64_GCC\n"
845	" mov rcx, [xBP - xCB * 4] ; pErrInfo\n"
846	" xor rdx, rdx ; fFlags (local load)\n"
847	" lea rsi, [g_hMod wrt rip] ; phLdrMod\n"
848	" lea rdi, [g_szLibrary wrt rip] ; pszFilename\n"
849	" call IMP2(SUPR3HardenedLdrLoadAppPriv)\n"
850	"\n"
851	"%%elifdef ASM_CALL64_MSC\n"
852	" mov r9, [xBP - xCB * 4] ; pErrInfo\n"
853	" xor r8, r8 ; fFlags (local load)\n"
854	" lea rdx, [g_hMod wrt rip] ; phLdrMod\n"
855	" lea rcx, [g_szLibrary wrt rip] ; pszFilename\n"
856	" call IMP2(SUPR3HardenedLdrLoadAppPriv)\n"
857	"\n"
858	"%%elifdef RT_ARCH_X86\n"
859	" sub xSP, 0ch\n"
860	" push dword [xBP + 12] ; pErrInfo\n"
861	" push 0 ; fFlags (local load)\n"
862	" push g_hMod ; phLdrMod\n"
863	" push g_szLibrary ; pszFilename\n"
864	" call IMP2(SUPR3HardenedLdrLoadAppPriv)\n"
865	" add esp, 1ch\n"
866	"%%else\n"
867	" %%error \"Unsupported architecture\"\n"
868	"%%endif\n"
869	" or eax, eax\n"
870	" jnz .return\n"
871	"\n"
872	" ;\n"
873	" ; Resolve the imports too if requested to do so.\n"
874	" ;\n"
875	".loaded:\n"
876	"%%ifdef ASM_ARCH_X86\n"
877	" cmp byte [xBP + 8], 0\n"
878	"%%else\n"
879	" cmp byte [xBP - xCB * 3], 0\n"
880	"%%endif\n"
881	" je .return\n"
882	"\n"
883	" lea pszCurStr, [g_szzNames xWrtRIP]\n"
884	" lea xBX, [g_apfnImports xWrtRIP]\n"
885	".next_import:\n"
886	" cmp RTCCPTR_PRE [xBX], 0\n"
887	" je .return\n"
888	"%%ifdef ASM_CALL64_GCC\n"
889	" mov rdx, xBX ; ppvValue\n"
890	" mov rsi, pszCurStr ; pszSymbol\n"
891	" mov rdi, [g_hMod wrt rip] ; hLdrMod\n"
892	" call IMP2(RTLdrGetSymbol)\n"
893	"%%elifdef ASM_CALL64_MSC\n"
894	" mov r8, xBX ; ppvValue\n"
895	" mov rdx, pszCurStr ; pszSymbol\n"
896	" mov rcx, [g_hMod wrt rip] ; pszSymbol\n"
897	" call IMP2(RTLdrGetSymbol)\n"
898	"%%else\n"
899	" push xBX ; ppvValue\n"
900	" push pszCurStr ; pszSymbol\n"
901	" push RTCCPTR_PRE [g_hMod] ; hLdrMod\n"
902	" call IMP2(RTLdrGetSymbol)\n"
903	" add xSP, 0ch\n"
904	"%%endif\n"
905	" or eax, eax\n"
906	" jnz .symbol_error\n"
907	"\n"
908	" ; Advance.\n"
909	" add xBX, RTCCPTR_CB\n"
910	" xor eax, eax\n"
911	" mov xCX, 0ffffffffh\n"
912	"%%ifdef ASM_CALL64_GCC\n"
913	" mov xDI, pszCurStr\n"
914	" repne scasb\n"
915	" mov pszCurStr, xDI\n"
916	"%%else\n"
917	" repne scasb\n"
918	"%%endif\n"
919	" jmp .next_import\n"
920	"\n"
921	" ;\n"
922	" ; Error loading a symbol. Call RTErrInfoSet on pErrInfo (preserves eax).\n"
923	" ;\n"
924	".symbol_error:\n"
925	"%%ifdef ASM_CALL64_GCC\n"
926	" mov rdx, pszCurStr ; pszMsg\n"
927	" mov esi, eax ; rc\n"
928	" mov rdi, [xBP - xCB * 4] ; pErrInfo\n"
929	" call IMP2(RTErrInfoSet)\n"
930	"%%elifdef ASM_CALL64_MSC\n"
931	" mov r8, pszCurStr ; pszMsg\n"
932	" mov edx, eax ; rc\n"
933	" mov rcx, [xBP - xCB * 4] ; pErrInfo\n"
934	" call IMP2(RTErrInfoSet)\n"
935	"%%else\n"
936	" push pszCurStr ; pszMsg\n"
937	" push eax ; pszSymbol\n"
938	" push dword [xBP + 0ch] ; pErrInfo\n"
939	" call IMP2(RTErrInfoSet)\n"
940	" add xSP, 0ch\n"
941	"%%endif\n"
942	" "
943	"\n"
944	".return:\n"
945	" mov pszCurStr, [xBP - xCB * 2]\n"
946	" mov xBX, [xBP - xCB * 1]\n"
947	" leave\n"
948	" ret\n"
949	"ENDPROC ExplicitlyLoad%.*s\n"
950	"\n"
951	"\n"
952	,
953	cchLibBaseName, g_pszLibrary,
954	cchLibBaseName, g_pszLibrary,
955	cchLibBaseName, g_pszLibrary
956	);
957	}
958
959
960	return RTEXITCODE_SUCCESS;
961	}
962
963
964	/**
965	* Generates the assembly source code, writing it to g_pszOutput.
966	*
967	* @returns RTEXITCODE_SUCCESS or RTEXITCODE_FAILURE, in the latter case full
968	* details has been displayed.
969	*/
970	static RTEXITCODE generateOutput(void)
971	{
972	RTEXITCODE rcExit = RTEXITCODE_FAILURE;
973	FILE *pOutput = fopen(g_pszOutput, "w");
974	if (pOutput)
975	{
976	rcExit = generateOutputInner(pOutput);
977	if (fclose(pOutput))
978	{
979	fprintf(stderr, "error: Error closing '%s'.\n", g_pszOutput);
980	rcExit = RTEXITCODE_FAILURE;
981	}
982	}
983	else
984	fprintf(stderr, "error: Failed to open '%s' for writing.\n", g_pszOutput);
985	return rcExit;
986	}
987
988
989	/**
990	* Displays usage information.
991	*
992	* @returns RTEXITCODE_SUCCESS.
993	* @param pszArgv0 The argv[0] string.
994	*/
995	static int usage(const char *pszArgv0)
996	{
997	printf("usage: %s [options] --libary <loadname> --output <lazyload.asm> <input.def>\n"
998	"\n"
999	"Options:\n"
1000	" --explicit-load-function, --no-explicit-load-function\n"
1001	" Whether to include the explicit load function, default is not to.\n"
1002	"\n"
1003	"Copyright (C) 2013-2016 Oracle Corporation\n"
1004	, pszArgv0);
1005
1006	return RTEXITCODE_SUCCESS;
1007	}
1008
1009
1010	int main(int argc, char **argv)
1011	{
1012	/*
1013	* Parse options.
1014	*/
1015	for (int i = 1; i < argc; i++)
1016	{
1017	const char *psz = argv[i];
1018	if (*psz == '-')
1019	{
1020	if (!strcmp(psz, "--output") \|\| !strcmp(psz, "-o"))
1021	{
1022	if (++i >= argc)
1023	{
1024	fprintf(stderr, "syntax error: File name expected after '%s'.\n", psz);
1025	return RTEXITCODE_SYNTAX;
1026	}
1027	g_pszOutput = argv[i];
1028	}
1029	else if (!strcmp(psz, "--library") \|\| !strcmp(psz, "-l"))
1030	{
1031	if (++i >= argc)
1032	{
1033	fprintf(stderr, "syntax error: Library name expected after '%s'.\n", psz);
1034	return RTEXITCODE_SYNTAX;
1035	}
1036	g_pszLibrary = argv[i];
1037	}
1038	else if (!strcmp(psz, "--explicit-load-function"))
1039	g_fWithExplictLoadFunction = true;
1040	else if (!strcmp(psz, "--no-explicit-load-function"))
1041	g_fWithExplictLoadFunction = false;
1042	else if (!strcmp(psz, "--system"))
1043	g_fSystemLibrary = true;
1044	/** @todo Support different load methods so this can be used on system libs and
1045	* such if we like. */
1046	else if ( !strcmp(psz, "--help")
1047	\|\| !strcmp(psz, "-help")
1048	\|\| !strcmp(psz, "-h")
1049	\|\| !strcmp(psz, "-?") )
1050	return usage(argv[0]);
1051	else if ( !strcmp(psz, "--version")
1052	\|\| !strcmp(psz, "-V"))
1053	{
1054	printf("$Revision: 63193 $\n");
1055	return RTEXITCODE_SUCCESS;
1056	}
1057	else
1058	{
1059	fprintf(stderr, "syntax error: Unknown option '%s'.\n", psz);
1060	return RTEXITCODE_SYNTAX;
1061	}
1062	}
1063	else
1064	{
1065	if (g_cInputs >= RT_ELEMENTS(g_apszInputs))
1066	{
1067	fprintf(stderr, "syntax error: Too many input files, max is %d.\n", (int)RT_ELEMENTS(g_apszInputs));
1068	return RTEXITCODE_SYNTAX;
1069	}
1070	g_apszInputs[g_cInputs++] = argv[i];
1071	}
1072	}
1073	if (g_cInputs == 0)
1074	{
1075	fprintf(stderr, "syntax error: No input file specified.\n");
1076	return RTEXITCODE_SYNTAX;
1077	}
1078	if (!g_pszOutput)
1079	{
1080	fprintf(stderr, "syntax error: No output file specified.\n");
1081	return RTEXITCODE_SYNTAX;
1082	}
1083	if (!g_pszLibrary)
1084	{
1085	fprintf(stderr, "syntax error: No library name specified.\n");
1086	return RTEXITCODE_SYNTAX;
1087	}
1088
1089	/*
1090	* Do the job.
1091	*/
1092	RTEXITCODE rcExit = parseInputs();
1093	if (rcExit == RTEXITCODE_SUCCESS)
1094	rcExit = generateOutput();
1095	return rcExit;
1096	}
1097

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

source: vbox/trunk/src/bldprogs/VBoxDef2LazyLoad.cpp@ 63562

Download in other formats: