4 * Copyright 2005 Eric Pouech
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * 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
28 #include "wine/test.h"
31 static char selfname
[MAX_PATH
];
33 /* Some functions are only in later versions of kernel32.dll */
34 static HANDLE (WINAPI
*pCreateToolhelp32Snapshot
)(DWORD
, DWORD
);
35 static BOOL (WINAPI
*pModule32First
)(HANDLE
, LPMODULEENTRY32
);
36 static BOOL (WINAPI
*pModule32Next
)(HANDLE
, LPMODULEENTRY32
);
37 static BOOL (WINAPI
*pProcess32First
)(HANDLE
, LPPROCESSENTRY32
);
38 static BOOL (WINAPI
*pProcess32Next
)(HANDLE
, LPPROCESSENTRY32
);
39 static BOOL (WINAPI
*pThread32First
)(HANDLE
, LPTHREADENTRY32
);
40 static BOOL (WINAPI
*pThread32Next
)(HANDLE
, LPTHREADENTRY32
);
42 /* 1 minute should be more than enough */
43 #define WAIT_TIME (60 * 1000)
45 static DWORD WINAPI
sub_thread(void* pmt
)
47 DWORD w
= WaitForSingleObject((HANDLE
)pmt
, WAIT_TIME
);
51 /******************************************************************
54 * generates basic information like:
55 * selfname: the way to reinvoke ourselves
59 * doesn't return if child
65 HANDLE ev1
, ev2
, ev3
, hThread
;
68 argc
= winetest_get_mainargs( &argv
);
69 strcpy(selfname
, argv
[0]);
73 case 2: /* the test program */
75 case 4: /* the sub-process */
76 ev1
= (HANDLE
)(INT_PTR
)atoi(argv
[2]);
77 ev2
= (HANDLE
)(INT_PTR
)atoi(argv
[3]);
78 ev3
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
80 if (ev3
== NULL
) ExitProcess(WAIT_ABANDONED
);
81 hThread
= CreateThread(NULL
, 0, sub_thread
, ev3
, 0, &tid
);
82 if (hThread
== NULL
) ExitProcess(WAIT_ABANDONED
);
83 if (!LoadLibraryA("shell32.dll")) ExitProcess(WAIT_ABANDONED
);
85 /* signal init of sub-process is done */
87 /* wait for parent to have done all its queries */
88 w
= WaitForSingleObject(ev2
, WAIT_TIME
);
89 if (w
!= WAIT_OBJECT_0
) ExitProcess(w
);
90 /* signal sub-thread to terminate */
92 w
= WaitForSingleObject(hThread
, WAIT_TIME
);
93 if (w
!= WAIT_OBJECT_0
) ExitProcess(w
);
94 GetExitCodeThread(hThread
, &w
);
101 static void test_process(DWORD curr_pid
, DWORD sub_pcs_pid
)
110 hSnapshot
= pCreateToolhelp32Snapshot( TH32CS_SNAPPROCESS
, 0 );
111 ok(hSnapshot
!= NULL
, "Cannot create snapshot\n");
113 /* check that this current process is enumerated */
114 pe
.dwSize
= sizeof(pe
);
115 if (pProcess32First( hSnapshot
, &pe
))
119 if (pe
.th32ProcessID
== curr_pid
) found
++;
120 if (pe
.th32ProcessID
== sub_pcs_pid
) { childpos
= num
; found
++; }
121 trace("PID=%x %s\n", pe
.th32ProcessID
, pe
.szExeFile
);
123 } while (pProcess32Next( hSnapshot
, &pe
));
125 ok(found
== 2, "couldn't find self and/or sub-process in process list\n");
127 /* check that first really resets the enumeration */
129 if (pProcess32First( hSnapshot
, &pe
))
133 if (pe
.th32ProcessID
== curr_pid
) found
++;
134 if (pe
.th32ProcessID
== sub_pcs_pid
) found
++;
135 trace("PID=%x %s\n", pe
.th32ProcessID
, pe
.szExeFile
);
137 } while (pProcess32Next( hSnapshot
, &pe
));
139 ok(found
== 2, "couldn't find self and/or sub-process in process list\n");
140 ok(!num
, "mismatch in counting\n");
142 /* one broken program does Process32First() and does not expect anything
143 * interesting to be there, especially not the just forked off child */
144 ok (childpos
!=0, "child is not expected to be at position 0.\n");
146 me
.dwSize
= sizeof(me
);
147 ok(!pModule32First( hSnapshot
, &me
), "shouldn't return a module\n");
149 CloseHandle(hSnapshot
);
150 ok(!pProcess32First( hSnapshot
, &pe
), "shouldn't return a process\n");
153 static void test_thread(DWORD curr_pid
, DWORD sub_pcs_pid
)
159 unsigned curr_found
= 0;
160 unsigned sub_found
= 0;
162 hSnapshot
= pCreateToolhelp32Snapshot( TH32CS_SNAPTHREAD
, 0 );
163 ok(hSnapshot
!= NULL
, "Cannot create snapshot\n");
165 /* check that this current process is enumerated */
166 te
.dwSize
= sizeof(te
);
167 if (pThread32First( hSnapshot
, &te
))
171 if (te
.th32OwnerProcessID
== curr_pid
) curr_found
++;
172 if (te
.th32OwnerProcessID
== sub_pcs_pid
) sub_found
++;
173 if (winetest_debug
> 1)
174 trace("PID=%x TID=%x %d\n", te
.th32OwnerProcessID
, te
.th32ThreadID
, te
.tpBasePri
);
176 } while (pThread32Next( hSnapshot
, &te
));
178 ok(curr_found
== 1, "couldn't find self in thread list\n");
179 ok(sub_found
== 2, "couldn't find sub-process thread's in thread list\n");
181 /* check that first really resets enumeration */
184 if (pThread32First( hSnapshot
, &te
))
188 if (te
.th32OwnerProcessID
== curr_pid
) curr_found
++;
189 if (te
.th32OwnerProcessID
== sub_pcs_pid
) sub_found
++;
190 if (winetest_debug
> 1)
191 trace("PID=%x TID=%x %d\n", te
.th32OwnerProcessID
, te
.th32ThreadID
, te
.tpBasePri
);
193 } while (pThread32Next( hSnapshot
, &te
));
195 ok(curr_found
== 1, "couldn't find self in thread list\n");
196 ok(sub_found
== 2, "couldn't find sub-process thread's in thread list\n");
198 me
.dwSize
= sizeof(me
);
199 ok(!pModule32First( hSnapshot
, &me
), "shouldn't return a module\n");
201 CloseHandle(hSnapshot
);
202 ok(!pThread32First( hSnapshot
, &te
), "shouldn't return a thread\n");
205 static const char* curr_expected_modules
[] =
209 /* FIXME: could test for ntdll on NT and Wine */
211 static const char* sub_expected_modules
[] =
216 /* FIXME: could test for ntdll on NT and Wine */
218 #define NUM_OF(x) (sizeof(x) / sizeof(x[0]))
220 static void test_module(DWORD pid
, const char* expected
[], unsigned num_expected
)
230 ok(NUM_OF(found
) >= num_expected
, "Internal: bump found[] size\n");
232 hSnapshot
= pCreateToolhelp32Snapshot( TH32CS_SNAPMODULE
, pid
);
233 ok(hSnapshot
!= NULL
, "Cannot create snapshot\n");
235 for (i
= 0; i
< num_expected
; i
++) found
[i
] = 0;
236 me
.dwSize
= sizeof(me
);
237 if (pModule32First( hSnapshot
, &me
))
241 trace("PID=%x base=%p size=%x %s %s\n",
242 me
.th32ProcessID
, me
.modBaseAddr
, me
.modBaseSize
, me
.szExePath
, me
.szModule
);
243 ok(me
.th32ProcessID
== pid
, "wrong returned process id\n");
244 for (i
= 0; i
< num_expected
; i
++)
245 if (!lstrcmpi(expected
[i
], me
.szModule
)) found
[i
]++;
247 } while (pModule32Next( hSnapshot
, &me
));
249 for (i
= 0; i
< num_expected
; i
++)
250 ok(found
[i
] == 1, "Module %s is %s\n",
251 expected
[i
], found
[i
] ? "listed more than once" : "not listed");
253 /* check that first really resets the enumeration */
254 for (i
= 0; i
< num_expected
; i
++) found
[i
] = 0;
255 me
.dwSize
= sizeof(me
);
256 if (pModule32First( hSnapshot
, &me
))
260 trace("PID=%x base=%p size=%x %s %s\n",
261 me
.th32ProcessID
, me
.modBaseAddr
, me
.modBaseSize
, me
.szExePath
, me
.szModule
);
262 for (i
= 0; i
< num_expected
; i
++)
263 if (!lstrcmpi(expected
[i
], me
.szModule
)) found
[i
]++;
265 } while (pModule32Next( hSnapshot
, &me
));
267 for (i
= 0; i
< num_expected
; i
++)
268 ok(found
[i
] == 1, "Module %s is %s\n",
269 expected
[i
], found
[i
] ? "listed more than once" : "not listed");
270 ok(!num
, "mismatch in counting\n");
272 pe
.dwSize
= sizeof(pe
);
273 ok(!pProcess32First( hSnapshot
, &pe
), "shouldn't return a process\n");
275 te
.dwSize
= sizeof(te
);
276 ok(!pThread32First( hSnapshot
, &te
), "shouldn't return a thread\n");
278 CloseHandle(hSnapshot
);
279 ok(!pModule32First( hSnapshot
, &me
), "shouldn't return a module\n");
284 DWORD pid
= GetCurrentProcessId();
286 char *p
, module
[MAX_PATH
];
287 char buffer
[MAX_PATH
];
288 SECURITY_ATTRIBUTES sa
;
289 PROCESS_INFORMATION info
;
290 STARTUPINFOA startup
;
293 HANDLE hkernel32
= GetModuleHandleA("kernel32");
295 pCreateToolhelp32Snapshot
= (VOID
*) GetProcAddress(hkernel32
, "CreateToolhelp32Snapshot");
296 pModule32First
= (VOID
*) GetProcAddress(hkernel32
, "Module32First");
297 pModule32Next
= (VOID
*) GetProcAddress(hkernel32
, "Module32Next");
298 pProcess32First
= (VOID
*) GetProcAddress(hkernel32
, "Process32First");
299 pProcess32Next
= (VOID
*) GetProcAddress(hkernel32
, "Process32Next");
300 pThread32First
= (VOID
*) GetProcAddress(hkernel32
, "Thread32First");
301 pThread32Next
= (VOID
*) GetProcAddress(hkernel32
, "Thread32Next");
303 if (!pCreateToolhelp32Snapshot
||
304 !pModule32First
|| !pModule32Next
||
305 !pProcess32First
|| !pProcess32Next
||
306 !pThread32First
|| !pThread32Next
)
308 skip("Needed functions are not available, most likely running on Windows NT\n");
313 ok(r
== 0, "Basic init of sub-process test\n");
316 sa
.nLength
= sizeof(sa
);
317 sa
.lpSecurityDescriptor
= NULL
;
318 sa
.bInheritHandle
= TRUE
;
320 ev1
= CreateEvent(&sa
, FALSE
, FALSE
, NULL
);
321 ev2
= CreateEvent(&sa
, FALSE
, FALSE
, NULL
);
322 ok (ev1
!= NULL
&& ev2
!= NULL
, "Couldn't create events\n");
323 memset(&startup
, 0, sizeof(startup
));
324 startup
.cb
= sizeof(startup
);
325 startup
.dwFlags
= STARTF_USESHOWWINDOW
;
326 startup
.wShowWindow
= SW_SHOWNORMAL
;
328 sprintf(buffer
, "%s tests/toolhelp.c %lu %lu", selfname
, (DWORD_PTR
)ev1
, (DWORD_PTR
)ev2
);
329 ok(CreateProcessA(NULL
, buffer
, NULL
, NULL
, TRUE
, 0, NULL
, NULL
, &startup
, &info
), "CreateProcess\n");
330 /* wait for child to be initialized */
331 w
= WaitForSingleObject(ev1
, WAIT_TIME
);
332 ok(w
== WAIT_OBJECT_0
, "Failed to wait on sub-process startup\n");
334 GetModuleFileNameA( 0, module
, sizeof(module
) );
335 if (!(p
= strrchr( module
, '\\' ))) p
= module
;
337 curr_expected_modules
[0] = p
;
338 sub_expected_modules
[0] = p
;
340 test_process(pid
, info
.dwProcessId
);
341 test_thread(pid
, info
.dwProcessId
);
342 test_module(pid
, curr_expected_modules
, NUM_OF(curr_expected_modules
));
343 test_module(info
.dwProcessId
, sub_expected_modules
, NUM_OF(sub_expected_modules
));
346 winetest_wait_child_process( info
.hProcess
);