snoop.c 9.93 KB
Newer Older
Alexandre Julliard's avatar
Alexandre Julliard committed
1 2 3 4
/*
 * 386-specific Win16 dll<->dll snooping functions
 *
 * Copyright 1998 Marcus Meissner
5 6 7 8 9 10 11 12 13 14 15 16 17
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
18
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
Alexandre Julliard's avatar
Alexandre Julliard committed
19 20
 */

21 22 23
#include "config.h"
#include "wine/port.h"

Alexandre Julliard's avatar
Alexandre Julliard committed
24
#include <assert.h>
25
#include <string.h>
26
#include <stdarg.h>
27 28
#include <stdio.h>
#include "windef.h"
29
#include "winbase.h"
Alexandre Julliard's avatar
Alexandre Julliard committed
30
#include "winnt.h"
31
#include "wine/winbase16.h"
32
#include "winternl.h"
33
#include "wine/library.h"
34
#include "kernel16_private.h"
35
#include "wine/debug.h"
Alexandre Julliard's avatar
Alexandre Julliard committed
36

37
WINE_DEFAULT_DEBUG_CHANNEL(snoop);
38

39
#include "pshpack1.h"
Alexandre Julliard's avatar
Alexandre Julliard committed
40

41 42
static void WINAPI SNOOP16_Entry(FARPROC proc, LPBYTE args, CONTEXT *context);
static void WINAPI SNOOP16_Return(FARPROC proc, LPBYTE args, CONTEXT *context);
Alexandre Julliard's avatar
Alexandre Julliard committed
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58

typedef	struct tagSNOOP16_FUN {
	/* code part */
	BYTE		lcall;		/* 0x9a call absolute with segment */
	DWORD		snr;
	/* unreached */
	int		nrofargs;
	FARPROC16	origfun;
	char		*name;
} SNOOP16_FUN;

typedef struct tagSNOOP16_DLL {
	HMODULE16	hmod;
	HANDLE16	funhandle;
	SNOOP16_FUN	*funs;
	struct tagSNOOP16_DLL	*next;
59
	char name[1];
Alexandre Julliard's avatar
Alexandre Julliard committed
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80
} SNOOP16_DLL;

typedef struct tagSNOOP16_RETURNENTRY {
	/* code part */
	BYTE		lcall;		/* 0x9a call absolute with segment */
	DWORD		snr;
	/* unreached */
	FARPROC16	origreturn;
	SNOOP16_DLL	*dll;
	DWORD		ordinal;
	WORD		origSP;
	WORD		*args;		/* saved args across a stdcall */
} SNOOP16_RETURNENTRY;

typedef struct tagSNOOP16_RETURNENTRIES {
	SNOOP16_RETURNENTRY entry[65500/sizeof(SNOOP16_RETURNENTRY)];
	HANDLE16	rethandle;
	struct tagSNOOP16_RETURNENTRIES	*next;
} SNOOP16_RETURNENTRIES;

typedef struct tagSNOOP16_RELAY {
81 82 83
	WORD		pushbp;		/* 0x5566 */
	BYTE		pusheax;	/* 0x50 */
	WORD		pushax;		/* 0x5066 */
Alexandre Julliard's avatar
Alexandre Julliard committed
84 85 86 87 88
	BYTE		pushl;		/* 0x68 */
	DWORD		realfun;	/* SNOOP16_Return */
	BYTE		lcall;		/* 0x9a call absolute with segment */
	DWORD		callfromregs;
	WORD		seg;
89
        WORD		lret;           /* 0xcb66 */
Alexandre Julliard's avatar
Alexandre Julliard committed
90 91
} SNOOP16_RELAY;

92
#include "poppack.h"
Alexandre Julliard's avatar
Alexandre Julliard committed
93 94 95 96 97 98 99

static	SNOOP16_DLL		*firstdll = NULL;
static	SNOOP16_RETURNENTRIES 	*firstrets = NULL;
static	SNOOP16_RELAY		*snr;
static	HANDLE16		xsnr = 0;

void
100
SNOOP16_RegisterDLL(HMODULE16 hModule,LPCSTR name) {
Alexandre Julliard's avatar
Alexandre Julliard committed
101 102 103 104
	SNOOP16_DLL	**dll = &(firstdll);
	char		*s;

	if (!TRACE_ON(snoop)) return;
105

106
        TRACE("hmod=%x, name=%s\n", hModule, name);
107

Alexandre Julliard's avatar
Alexandre Julliard committed
108
	if (!snr) {
109
		xsnr=GLOBAL_Alloc(GMEM_ZEROINIT,2*sizeof(*snr),0,WINE_LDT_FLAGS_CODE|WINE_LDT_FLAGS_32BIT);
Alexandre Julliard's avatar
Alexandre Julliard committed
110
		snr = GlobalLock16(xsnr);
111 112 113
		snr[0].pushbp	= 0x5566;
		snr[0].pusheax	= 0x50;
		snr[0].pushax	= 0x5066;
Alexandre Julliard's avatar
Alexandre Julliard committed
114 115 116
		snr[0].pushl	= 0x68;
		snr[0].realfun	= (DWORD)SNOOP16_Entry;
		snr[0].lcall 	= 0x9a;
117
		snr[0].callfromregs = (DWORD)__wine_call_from_16_regs;
118
		snr[0].seg      = wine_get_cs();
119 120 121 122 123
		snr[0].lret     = 0xcb66;

		snr[1].pushbp	= 0x5566;
		snr[1].pusheax	= 0x50;
		snr[1].pushax	= 0x5066;
Alexandre Julliard's avatar
Alexandre Julliard committed
124 125 126
		snr[1].pushl	= 0x68;
		snr[1].realfun	= (DWORD)SNOOP16_Return;
		snr[1].lcall 	= 0x9a;
127
		snr[1].callfromregs = (DWORD)__wine_call_from_16_regs;
128
		snr[1].seg      = wine_get_cs();
129
		snr[1].lret     = 0xcb66;
Alexandre Julliard's avatar
Alexandre Julliard committed
130 131
	}
	while (*dll) {
132
		if ((*dll)->hmod == hModule)
133 134 135 136 137 138
                {
                    /* another dll, loaded at the same address */
                    GlobalUnlock16((*dll)->funhandle);
                    GlobalFree16((*dll)->funhandle);
                    break;
                }
Alexandre Julliard's avatar
Alexandre Julliard committed
139 140
		dll = &((*dll)->next);
	}
141 142 143 144 145 146

	if (*dll)
		*dll = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, *dll, sizeof(SNOOP16_DLL)+strlen(name));
	else
		*dll = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(SNOOP16_DLL)+strlen(name));	

Alexandre Julliard's avatar
Alexandre Julliard committed
147
	(*dll)->next	= NULL;
148
	(*dll)->hmod	= hModule;
Alexandre Julliard's avatar
Alexandre Julliard committed
149 150
	if ((s=strrchr(name,'\\')))
		name = s+1;
151
	strcpy( (*dll)->name, name );
Alexandre Julliard's avatar
Alexandre Julliard committed
152 153
	if ((s=strrchr((*dll)->name,'.')))
		*s='\0';
154
	(*dll)->funhandle = GlobalHandleToSel16(GLOBAL_Alloc(GMEM_ZEROINIT,65535,0,WINE_LDT_FLAGS_CODE));
Alexandre Julliard's avatar
Alexandre Julliard committed
155 156
	(*dll)->funs = GlobalLock16((*dll)->funhandle);
	if (!(*dll)->funs) {
157
		HeapFree(GetProcessHeap(),0,*dll);
158
		FIXME("out of memory\n");
Alexandre Julliard's avatar
Alexandre Julliard committed
159 160 161 162 163 164 165 166 167 168 169 170 171 172
		return;
	}
}

FARPROC16
SNOOP16_GetProcAddress16(HMODULE16 hmod,DWORD ordinal,FARPROC16 origfun) {
	SNOOP16_DLL			*dll = firstdll;
	SNOOP16_FUN			*fun;
	NE_MODULE			*pModule = NE_GetPtr(hmod);
	unsigned char			*cpnt;
	char				name[200];

	if (!TRACE_ON(snoop) || !pModule || !HIWORD(origfun))
		return origfun;
Austin English's avatar
Austin English committed
173
	if (!*(LPBYTE)MapSL((SEGPTR)origfun)) /* 0x00 is an impossible opcode, possible dataref. */
Alexandre Julliard's avatar
Alexandre Julliard committed
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
		return origfun;
	while (dll) {
		if (hmod == dll->hmod)
			break;
		dll=dll->next;
	}
	if (!dll)	/* probably internal */
		return origfun;
	if (ordinal>65535/sizeof(SNOOP16_FUN))
		return origfun;
	fun = dll->funs+ordinal;
	/* already done? */
	fun->lcall 	= 0x9a;
	fun->snr	= MAKELONG(0,xsnr);
	fun->origfun	= origfun;
	if (fun->name)
		return (FARPROC16)(SEGPTR)MAKELONG(((char*)fun-(char*)dll->funs),dll->funhandle);
191
	cpnt = (unsigned char *)pModule + pModule->ne_restab;
Alexandre Julliard's avatar
Alexandre Julliard committed
192 193 194 195 196 197 198 199 200 201
	while (*cpnt) {
		cpnt += *cpnt + 1 + sizeof(WORD);
		if (*(WORD*)(cpnt+*cpnt+1) == ordinal) {
			sprintf(name,"%.*s",*cpnt,cpnt+1);
			break;
		}
	}
	/* Now search the non-resident names table */

	if (!*cpnt && pModule->nrname_handle) {
202
		cpnt = GlobalLock16( pModule->nrname_handle );
Alexandre Julliard's avatar
Alexandre Julliard committed
203 204 205 206 207 208 209 210 211
		while (*cpnt) {
			cpnt += *cpnt + 1 + sizeof(WORD);
			if (*(WORD*)(cpnt+*cpnt+1) == ordinal) {
				    sprintf(name,"%.*s",*cpnt,cpnt+1);
				    break;
			}
		}
	}
	if (*cpnt)
212 213 214 215
        {
            fun->name = HeapAlloc(GetProcessHeap(),0,strlen(name)+1);
            strcpy( fun->name, name );
        }
Alexandre Julliard's avatar
Alexandre Julliard committed
216
	else
217 218
            fun->name = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,1); /* empty string */

219
	if (!SNOOP16_ShowDebugmsgSnoop(dll->name, ordinal, fun->name))
220 221
		return origfun;

Alexandre Julliard's avatar
Alexandre Julliard committed
222 223 224 225 226
	/* more magic. do not try to snoop thunk data entries (MMSYSTEM) */
	if (strchr(fun->name,'_')) {
		char *s=strchr(fun->name,'_');

		if (!strncasecmp(s,"_thunkdata",10)) {
227
			HeapFree(GetProcessHeap(),0,fun->name);
Alexandre Julliard's avatar
Alexandre Julliard committed
228 229 230 231
			fun->name = NULL;
			return origfun;
		}
	}
Alexandre Julliard's avatar
Alexandre Julliard committed
232 233 234 235 236 237 238
	fun->lcall 	= 0x9a;
	fun->snr	= MAKELONG(0,xsnr);
	fun->origfun	= origfun;
	fun->nrofargs	= -1;
	return (FARPROC16)(SEGPTR)MAKELONG(((char*)fun-(char*)dll->funs),dll->funhandle);
}

239
#define CALLER1REF (*(DWORD*)(MapSL( MAKESEGPTR(context->SegSs,LOWORD(context->Esp)+4))))
240
static void WINAPI SNOOP16_Entry(FARPROC proc, LPBYTE args, CONTEXT *context) {
Alexandre Julliard's avatar
Alexandre Julliard committed
241
	DWORD		ordinal=0;
242
	DWORD		entry=(DWORD)MapSL( MAKESEGPTR(context->SegCs,LOWORD(context->Eip)) )-5;
243
	WORD		xcs = context->SegCs;
Alexandre Julliard's avatar
Alexandre Julliard committed
244 245 246 247
	SNOOP16_DLL	*dll = firstdll;
	SNOOP16_FUN	*fun = NULL;
	SNOOP16_RETURNENTRIES	**rets = &firstrets;
	SNOOP16_RETURNENTRY	*ret;
248 249
	unsigned	i=0;
	int		max;
Alexandre Julliard's avatar
Alexandre Julliard committed
250 251 252 253 254 255 256 257 258 259

	while (dll) {
		if (xcs == dll->funhandle) {
			fun = (SNOOP16_FUN*)entry;
			ordinal = fun-dll->funs;
			break;
		}
		dll=dll->next;
	}
	if (!dll) {
260
		FIXME("entrypoint 0x%08x not found\n",entry);
Alexandre Julliard's avatar
Alexandre Julliard committed
261 262 263 264 265 266 267 268 269 270 271
		return; /* oops */
	}
	while (*rets) {
		for (i=0;i<sizeof((*rets)->entry)/sizeof((*rets)->entry[0]);i++)
			if (!(*rets)->entry[i].origreturn)
				break;
		if (i!=sizeof((*rets)->entry)/sizeof((*rets)->entry[0]))
			break;
		rets = &((*rets)->next);
	}
	if (!*rets) {
272
		HANDLE16 hand = GlobalHandleToSel16(GLOBAL_Alloc(GMEM_ZEROINIT,65535,0,WINE_LDT_FLAGS_CODE));
Alexandre Julliard's avatar
Alexandre Julliard committed
273 274 275 276 277 278 279 280 281 282 283 284
		*rets = GlobalLock16(hand);
		(*rets)->rethandle = hand;
		i = 0;	/* entry 0 is free */
	}
	ret = &((*rets)->entry[i]);
	ret->lcall 	= 0x9a;
	ret->snr	= MAKELONG(sizeof(SNOOP16_RELAY),xsnr);
	ret->origreturn	= (FARPROC16)CALLER1REF;
	CALLER1REF	= MAKELONG((char*)&(ret->lcall)-(char*)((*rets)->entry),(*rets)->rethandle);
	ret->dll	= dll;
	ret->args	= NULL;
	ret->ordinal	= ordinal;
285
	ret->origSP	= LOWORD(context->Esp);
Alexandre Julliard's avatar
Alexandre Julliard committed
286

287 288
	context->Eip= LOWORD(fun->origfun);
	context->SegCs = HIWORD(fun->origfun);
Alexandre Julliard's avatar
Alexandre Julliard committed
289

290

291
	DPRINTF("%04x:CALL %s.%d: %s(",GetCurrentThreadId(), dll->name,ordinal,fun->name);
Alexandre Julliard's avatar
Alexandre Julliard committed
292
	if (fun->nrofargs>0) {
Alexandre Julliard's avatar
Alexandre Julliard committed
293 294
		max = fun->nrofargs;
		if (max>16) max=16;
Alexandre Julliard's avatar
Alexandre Julliard committed
295
		for (i=max;i--;)
296
			DPRINTF("%04x%s",*(WORD*)((char *) MapSL( MAKESEGPTR(context->SegSs,LOWORD(context->Esp)) )+8+sizeof(WORD)*i),i?",":"");
Alexandre Julliard's avatar
Alexandre Julliard committed
297 298 299 300
		if (max!=fun->nrofargs)
			DPRINTF(" ...");
	} else if (fun->nrofargs<0) {
		DPRINTF("<unknown, check return>");
301
		ret->args = HeapAlloc(GetProcessHeap(),0,16*sizeof(WORD));
302
		memcpy(ret->args,(LPBYTE)((char *) MapSL( MAKESEGPTR(context->SegSs,LOWORD(context->Esp)) )+8),sizeof(WORD)*16);
Alexandre Julliard's avatar
Alexandre Julliard committed
303
	}
Alexandre Julliard's avatar
Alexandre Julliard committed
304
	DPRINTF(") ret=%04x:%04x\n",HIWORD(ret->origreturn),LOWORD(ret->origreturn));
Alexandre Julliard's avatar
Alexandre Julliard committed
305 306
}

307
static void WINAPI SNOOP16_Return(FARPROC proc, LPBYTE args, CONTEXT *context) {
308
	SNOOP16_RETURNENTRY	*ret = (SNOOP16_RETURNENTRY*)((char *) MapSL( MAKESEGPTR(context->SegCs,LOWORD(context->Eip)) )-5);
Alexandre Julliard's avatar
Alexandre Julliard committed
309 310

	/* We haven't found out the nrofargs yet. If we called a cdecl
311
	 * function it is too late anyway and we can just set '0' (which
Alexandre Julliard's avatar
Alexandre Julliard committed
312 313 314
	 * will be the difference between orig and current SP
	 * If pascal -> everything ok.
	 */
Alexandre Julliard's avatar
Alexandre Julliard committed
315
	if (ret->dll->funs[ret->ordinal].nrofargs<0) {
316
		ret->dll->funs[ret->ordinal].nrofargs=(LOWORD(context->Esp)-ret->origSP-4)/2;
Alexandre Julliard's avatar
Alexandre Julliard committed
317
	}
318 319
	context->Eip = LOWORD(ret->origreturn);
	context->SegCs  = HIWORD(ret->origreturn);
320
        DPRINTF("%04x:RET  %s.%d: %s(",
321 322
                GetCurrentThreadId(),ret->dll->name,ret->ordinal,
                ret->dll->funs[ret->ordinal].name);
323
	if (ret->args) {
Alexandre Julliard's avatar
Alexandre Julliard committed
324 325 326
		int	i,max;

		max = ret->dll->funs[ret->ordinal].nrofargs;
Alexandre Julliard's avatar
Alexandre Julliard committed
327 328
		if (max>16)
			max=16;
329
		if (max<0)
Alexandre Julliard's avatar
Alexandre Julliard committed
330
			max=0;
Alexandre Julliard's avatar
Alexandre Julliard committed
331 332 333

		for (i=max;i--;)
			DPRINTF("%04x%s",ret->args[i],i?",":"");
Alexandre Julliard's avatar
Alexandre Julliard committed
334 335
		if (max!=ret->dll->funs[ret->ordinal].nrofargs)
			DPRINTF(" ...");
336
		HeapFree(GetProcessHeap(),0,ret->args);
Alexandre Julliard's avatar
Alexandre Julliard committed
337
		ret->args = NULL;
338 339 340 341
	}
        DPRINTF(") retval = %04x:%04x ret=%04x:%04x\n",
                (WORD)context->Edx,(WORD)context->Eax,
                HIWORD(ret->origreturn),LOWORD(ret->origreturn));
Alexandre Julliard's avatar
Alexandre Julliard committed
342 343
	ret->origreturn = NULL; /* mark as empty */
}