2 * 386-specific Win32 dll<->dll snooping functions
4 * Copyright 1998 Marcus Meissner
15 #include "stackframe.h"
16 #include "debugtools.h"
17 #include "wine/exception.h"
19 DEFAULT_DEBUG_CHANNEL(snoop
);
21 static WINE_EXCEPTION_FILTER(page_fault
)
23 if (GetExceptionCode() == EXCEPTION_ACCESS_VIOLATION
||
24 GetExceptionCode() == EXCEPTION_PRIV_INSTRUCTION
)
25 return EXCEPTION_EXECUTE_HANDLER
;
26 return EXCEPTION_CONTINUE_SEARCH
;
29 char **debug_snoop_excludelist
= NULL
, **debug_snoop_includelist
= NULL
;
33 extern void WINAPI
SNOOP_Entry();
34 extern void WINAPI
SNOOP_Return();
36 #ifdef NEED_UNDERSCORE_PREFIX
44 typedef struct tagSNOOP_FUN
{
46 BYTE lcall
; /* 0xe8 call snoopentry (relative) */
47 /* NOTE: If you move snoopentry OR nrofargs fix the relative offset
50 DWORD snoopentry
; /* SNOOP_Entry relative */
57 typedef struct tagSNOOP_DLL
{
62 struct tagSNOOP_DLL
*next
;
66 typedef struct tagSNOOP_RETURNENTRY
{
68 BYTE lcall
; /* 0xe8 call snoopret relative*/
69 /* NOTE: If you move snoopret OR origreturn fix the relative offset
72 DWORD snoopret
; /* SNOOP_Ret relative */
78 DWORD
*args
; /* saved args across a stdcall */
81 typedef struct tagSNOOP_RETURNENTRIES
{
82 SNOOP_RETURNENTRY entry
[4092/sizeof(SNOOP_RETURNENTRY
)];
83 struct tagSNOOP_RETURNENTRIES
*next
;
84 } SNOOP_RETURNENTRIES
;
88 static SNOOP_DLL
*firstdll
= NULL
;
89 static SNOOP_RETURNENTRIES
*firstrets
= NULL
;
91 /***********************************************************************
92 * SNOOP_ShowDebugmsgSnoop
94 * Simple function to decide if a particular debugging message is
97 int SNOOP_ShowDebugmsgSnoop(const char *dll
, int ord
, const char *fname
) {
99 if(debug_snoop_excludelist
|| debug_snoop_includelist
) {
102 int len
, len2
, itemlen
, show
;
104 if(debug_snoop_excludelist
) {
106 listitem
= debug_snoop_excludelist
;
109 listitem
= debug_snoop_includelist
;
113 sprintf(buf
, "%s.%d", dll
, ord
);
115 for(; *listitem
; listitem
++) {
116 itemlen
= strlen(*listitem
);
117 if((itemlen
== len
&& !strncasecmp(*listitem
, buf
, len
)) ||
118 (itemlen
== len2
&& !strncasecmp(*listitem
, buf
, len2
)) ||
119 !strcasecmp(*listitem
, fname
)) {
130 SNOOP_RegisterDLL(HMODULE hmod
,LPCSTR name
,DWORD ordbase
,DWORD nrofordinals
) {
131 SNOOP_DLL
**dll
= &(firstdll
);
134 if (!TRACE_ON(snoop
)) return;
136 if ((*dll
)->hmod
== hmod
)
137 return; /* already registered */
138 dll
= &((*dll
)->next
);
140 *dll
= HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY
,sizeof(SNOOP_DLL
)+strlen(name
));
143 (*dll
)->ordbase
= ordbase
;
144 (*dll
)->nrofordinals
= nrofordinals
;
145 strcpy( (*dll
)->name
, name
);
146 if ((s
=strrchr((*dll
)->name
,'.')))
148 (*dll
)->funs
= VirtualAlloc(NULL
,nrofordinals
*sizeof(SNOOP_FUN
),MEM_COMMIT
|MEM_RESERVE
,PAGE_EXECUTE_READWRITE
);
149 memset((*dll
)->funs
,0,nrofordinals
*sizeof(SNOOP_FUN
));
151 HeapFree(GetProcessHeap(),0,*dll
);
152 FIXME("out of memory\n");
158 SNOOP_GetProcAddress(HMODULE hmod
,LPCSTR name
,DWORD ordinal
,FARPROC origfun
) {
159 SNOOP_DLL
*dll
= firstdll
;
162 IMAGE_SECTION_HEADER
*pe_seg
= PE_SECTIONS(hmod
);
164 if (!TRACE_ON(snoop
)) return origfun
;
165 if (!*(LPBYTE
)origfun
) /* 0x00 is an imposs. opcode, poss. dataref. */
167 for (j
=0;j
<PE_HEADER(hmod
)->FileHeader
.NumberOfSections
;j
++)
168 /* 0x42 is special ELF loader tag */
169 if ((pe_seg
[j
].VirtualAddress
==0x42) ||
170 (((DWORD
)origfun
-hmod
>=pe_seg
[j
].VirtualAddress
)&&
171 ((DWORD
)origfun
-hmod
<pe_seg
[j
].VirtualAddress
+
172 pe_seg
[j
].SizeOfRawData
176 /* If we looked through all sections (and didn't find one)
177 * or if the sectionname contains "data", we return the
178 * original function since it is most likely a datareference.
180 if ( (j
==PE_HEADER(hmod
)->FileHeader
.NumberOfSections
) ||
181 (strstr(pe_seg
[j
].Name
,"data")) ||
182 !(pe_seg
[j
].Characteristics
& IMAGE_SCN_CNT_CODE
)
187 if (hmod
== dll
->hmod
)
191 if (!dll
) /* probably internal */
193 if (!SNOOP_ShowDebugmsgSnoop(dll
->name
,ordinal
,name
))
195 assert(ordinal
< dll
->nrofordinals
);
196 fun
= dll
->funs
+ordinal
;
199 fun
->name
= HeapAlloc(GetProcessHeap(),0,strlen(name
)+1);
200 strcpy( fun
->name
, name
);
202 /* NOTE: origreturn struct member MUST come directly after snoopentry */
203 fun
->snoopentry
= (char*)SNOOP_Entry
-((char*)(&fun
->nrofargs
));
204 fun
->origfun
= origfun
;
207 return (FARPROC
)&(fun
->lcall
);
211 SNOOP_PrintArg(DWORD x
) {
212 static char buf
[200];
214 char * volatile ret
=0;
216 if ( !HIWORD(x
) ) { /* trivial reject to avoid faults */
217 sprintf(buf
,"%08lx",x
);
225 if (s
[i
]<0x20) {nostring
=1;break;}
226 if (s
[i
]>=0x80) {nostring
=1;break;}
231 snprintf(buf
,sizeof(buf
),"%08lx %s",x
,debugstr_an((LPSTR
)x
,sizeof(buf
)-10));
236 __EXCEPT(page_fault
){}
245 if (s
[i
]<0x20) {nostring
=1;break;}
246 if (s
[i
]>0x100) {nostring
=1;break;}
251 snprintf(buf
,sizeof(buf
),"%08lx %s",x
,debugstr_wn((LPWSTR
)x
,sizeof(buf
)-10));
256 __EXCEPT(page_fault
){}
260 sprintf(buf
,"%08lx",x
);
264 #define CALLER1REF (*(DWORD*)context->Esp)
266 void WINAPI
SNOOP_DoEntry( CONTEXT86
*context
)
268 DWORD ordinal
=0,entry
= context
->Eip
- 5;
269 SNOOP_DLL
*dll
= firstdll
;
270 SNOOP_FUN
*fun
= NULL
;
271 SNOOP_RETURNENTRIES
**rets
= &firstrets
;
272 SNOOP_RETURNENTRY
*ret
;
276 if ( ((char*)entry
>=(char*)dll
->funs
) &&
277 ((char*)entry
<=(char*)(dll
->funs
+dll
->nrofordinals
))
279 fun
= (SNOOP_FUN
*)entry
;
280 ordinal
= fun
-dll
->funs
;
286 FIXME("entrypoint 0x%08lx not found\n",entry
);
289 /* guess cdecl ... */
290 if (fun
->nrofargs
<0) {
291 /* Typical cdecl return frame is:
293 * which has (for xxxxxxxx up to 255 the opcode "83 C4 xx".
294 * (after that 81 C2 xx xx xx xx)
296 LPBYTE reteip
= (LPBYTE
)CALLER1REF
;
299 if ((reteip
[0]==0x83)&&(reteip
[1]==0xc4))
300 fun
->nrofargs
=reteip
[2]/4;
306 for (i
=0;i
<sizeof((*rets
)->entry
)/sizeof((*rets
)->entry
[0]);i
++)
307 if (!(*rets
)->entry
[i
].origreturn
)
309 if (i
!=sizeof((*rets
)->entry
)/sizeof((*rets
)->entry
[0]))
311 rets
= &((*rets
)->next
);
314 *rets
= VirtualAlloc(NULL
,4096,MEM_COMMIT
|MEM_RESERVE
,PAGE_EXECUTE_READWRITE
);
315 memset(*rets
,0,4096);
316 i
= 0; /* entry 0 is free */
318 ret
= &((*rets
)->entry
[i
]);
320 /* NOTE: origreturn struct member MUST come directly after snoopret */
321 ret
->snoopret
= ((char*)SNOOP_Return
)-(char*)(&ret
->origreturn
);
322 ret
->origreturn
= (FARPROC
)CALLER1REF
;
323 CALLER1REF
= (DWORD
)&ret
->lcall
;
326 ret
->ordinal
= ordinal
;
327 ret
->origESP
= context
->Esp
;
329 context
->Eip
= (DWORD
)fun
->origfun
;
331 DPRINTF("%08lx:CALL %s.%ld: %s(",GetCurrentThreadId(),dll
->name
,dll
->ordbase
+ordinal
,fun
->name
);
332 if (fun
->nrofargs
>0) {
333 max
= fun
->nrofargs
; if (max
>16) max
=16;
335 DPRINTF("%s%s",SNOOP_PrintArg(*(DWORD
*)(context
->Esp
+ 4 + sizeof(DWORD
)*i
)),(i
<fun
->nrofargs
-1)?",":"");
336 if (max
!=fun
->nrofargs
)
338 } else if (fun
->nrofargs
<0) {
339 DPRINTF("<unknown, check return>");
340 ret
->args
= HeapAlloc(GetProcessHeap(),0,16*sizeof(DWORD
));
341 memcpy(ret
->args
,(LPBYTE
)(context
->Esp
+ 4),sizeof(DWORD
)*16);
343 DPRINTF(") ret=%08lx\n",(DWORD
)ret
->origreturn
);
347 void WINAPI
SNOOP_DoReturn( CONTEXT86
*context
)
349 SNOOP_RETURNENTRY
*ret
= (SNOOP_RETURNENTRY
*)(context
->Eip
- 5);
351 /* We haven't found out the nrofargs yet. If we called a cdecl
352 * function it is too late anyway and we can just set '0' (which
353 * will be the difference between orig and current ESP
354 * If stdcall -> everything ok.
356 if (ret
->dll
->funs
[ret
->ordinal
].nrofargs
<0)
357 ret
->dll
->funs
[ret
->ordinal
].nrofargs
=(context
->Esp
- ret
->origESP
-4)/4;
358 context
->Eip
= (DWORD
)ret
->origreturn
;
362 DPRINTF("%08lx:RET %s.%ld: %s(",
363 GetCurrentThreadId(),
364 ret
->dll
->name
,ret
->dll
->ordbase
+ret
->ordinal
,ret
->dll
->funs
[ret
->ordinal
].name
);
365 max
= ret
->dll
->funs
[ret
->ordinal
].nrofargs
;
369 DPRINTF("%s%s",SNOOP_PrintArg(ret
->args
[i
]),(i
<max
-1)?",":"");
370 DPRINTF(") retval = %08lx ret=%08lx\n",
371 context
->Eax
,(DWORD
)ret
->origreturn
);
372 HeapFree(GetProcessHeap(),0,ret
->args
);
375 DPRINTF("%08lx:RET %s.%ld: %s() retval = %08lx ret=%08lx\n",
376 GetCurrentThreadId(),
377 ret
->dll
->name
,ret
->dll
->ordbase
+ret
->ordinal
,ret
->dll
->funs
[ret
->ordinal
].name
,
378 context
->Eax
, (DWORD
)ret
->origreturn
);
379 ret
->origreturn
= NULL
; /* mark as empty */
382 /* assembly wrappers that save the context */
383 __ASM_GLOBAL_FUNC( SNOOP_Entry
,
384 "call " __ASM_NAME("__wine_call_from_32_regs") "\n\t"
385 ".long " __ASM_NAME("SNOOP_DoEntry") ",0" );
386 __ASM_GLOBAL_FUNC( SNOOP_Return
,
387 "call " __ASM_NAME("__wine_call_from_32_regs") "\n\t"
388 ".long " __ASM_NAME("SNOOP_DoReturn") ",0" );
390 #else /* !__i386__ */
391 void SNOOP_RegisterDLL(HMODULE hmod
,LPCSTR name
,DWORD nrofordinals
) {
392 if (!TRACE_ON(snoop
)) return;
393 FIXME("snooping works only on i386 for now.\n");
396 FARPROC
SNOOP_GetProcAddress(HMODULE hmod
,LPCSTR name
,DWORD ordinal
,FARPROC origfun
) {
399 #endif /* !__i386__ */