+++ /dev/null
-/*
- * Toolhelp
- *
- * Copyright 2005 Eric Pouech
- *
- * 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
- * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
- */
-
-#include <stdarg.h>
-#include <stdlib.h>
-#include <stdio.h>
-
-#include "windef.h"
-#include "winbase.h"
-#include "tlhelp32.h"
-#include "wine/test.h"
-#include "winuser.h"
-
-static char selfname[MAX_PATH];
-
-/* Some functions are only in later versions of kernel32.dll */
-static HANDLE (WINAPI *pCreateToolhelp32Snapshot)(DWORD, DWORD);
-static BOOL (WINAPI *pModule32First)(HANDLE, LPMODULEENTRY32);
-static BOOL (WINAPI *pModule32Next)(HANDLE, LPMODULEENTRY32);
-static BOOL (WINAPI *pProcess32First)(HANDLE, LPPROCESSENTRY32);
-static BOOL (WINAPI *pProcess32Next)(HANDLE, LPPROCESSENTRY32);
-static BOOL (WINAPI *pThread32First)(HANDLE, LPTHREADENTRY32);
-static BOOL (WINAPI *pThread32Next)(HANDLE, LPTHREADENTRY32);
-
-/* 1 minute should be more than enough */
-#define WAIT_TIME (60 * 1000)
-
-static DWORD WINAPI sub_thread(void* pmt)
-{
- DWORD w = WaitForSingleObject(pmt, WAIT_TIME);
- return w;
-}
-
-/******************************************************************
- * init
- *
- * generates basic information like:
- * selfname: the way to reinvoke ourselves
- * returns:
- * -1 on error
- * 0 if parent
- * doesn't return if child
- */
-static int init(void)
-{
- int argc;
- char** argv;
- HANDLE ev1, ev2, ev3, hThread;
- DWORD w, tid;
-
- argc = winetest_get_mainargs( &argv );
- strcpy(selfname, argv[0]);
-
- switch (argc)
- {
- case 2: /* the test program */
- return 0;
- case 4: /* the sub-process */
- ev1 = (HANDLE)(INT_PTR)atoi(argv[2]);
- ev2 = (HANDLE)(INT_PTR)atoi(argv[3]);
- ev3 = CreateEventW(NULL, FALSE, FALSE, NULL);
-
- if (ev3 == NULL) ExitProcess(WAIT_ABANDONED);
- hThread = CreateThread(NULL, 0, sub_thread, ev3, 0, &tid);
- if (hThread == NULL) ExitProcess(WAIT_ABANDONED);
- if (!LoadLibraryA("shell32.dll")) ExitProcess(WAIT_ABANDONED);
-
- /* signal init of sub-process is done */
- SetEvent(ev1);
- /* wait for parent to have done all its queries */
- w = WaitForSingleObject(ev2, WAIT_TIME);
- if (w != WAIT_OBJECT_0) ExitProcess(w);
- /* signal sub-thread to terminate */
- SetEvent(ev3);
- w = WaitForSingleObject(hThread, WAIT_TIME);
- if (w != WAIT_OBJECT_0) ExitProcess(w);
- GetExitCodeThread(hThread, &w);
- ExitProcess(w);
- default:
- return -1;
- }
-}
-
-static void test_process(DWORD curr_pid, DWORD sub_pcs_pid)
-{
- HANDLE hSnapshot;
- PROCESSENTRY32 pe;
- MODULEENTRY32 me;
- unsigned found = 0;
- int num = 0;
- int childpos = -1;
-
- hSnapshot = pCreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
- ok(hSnapshot != NULL, "Cannot create snapshot\n");
-
- /* check that this current process is enumerated */
- pe.dwSize = sizeof(pe);
- if (pProcess32First( hSnapshot, &pe ))
- {
- do
- {
- if (pe.th32ProcessID == curr_pid) found++;
- if (pe.th32ProcessID == sub_pcs_pid) { childpos = num; found++; }
- trace("PID=%x %s\n", pe.th32ProcessID, pe.szExeFile);
- num++;
- } while (pProcess32Next( hSnapshot, &pe ));
- }
- ok(found == 2, "couldn't find self and/or sub-process in process list\n");
-
- /* check that first really resets the enumeration */
- found = 0;
- if (pProcess32First( hSnapshot, &pe ))
- {
- do
- {
- if (pe.th32ProcessID == curr_pid) found++;
- if (pe.th32ProcessID == sub_pcs_pid) found++;
- trace("PID=%x %s\n", pe.th32ProcessID, pe.szExeFile);
- num--;
- } while (pProcess32Next( hSnapshot, &pe ));
- }
- ok(found == 2, "couldn't find self and/or sub-process in process list\n");
- ok(!num, "mismatch in counting\n");
-
- /* one broken program does Process32First() and does not expect anything
- * interesting to be there, especially not the just forked off child */
- ok (childpos !=0, "child is not expected to be at position 0.\n");
-
- me.dwSize = sizeof(me);
- ok(!pModule32First( hSnapshot, &me ), "shouldn't return a module\n");
-
- CloseHandle(hSnapshot);
- ok(!pProcess32First( hSnapshot, &pe ), "shouldn't return a process\n");
-}
-
-static void test_thread(DWORD curr_pid, DWORD sub_pcs_pid)
-{
- HANDLE hSnapshot;
- THREADENTRY32 te;
- MODULEENTRY32 me;
- int num = 0;
- unsigned curr_found = 0;
- unsigned sub_found = 0;
-
- hSnapshot = pCreateToolhelp32Snapshot( TH32CS_SNAPTHREAD, 0 );
- ok(hSnapshot != NULL, "Cannot create snapshot\n");
-
- /* check that this current process is enumerated */
- te.dwSize = sizeof(te);
- if (pThread32First( hSnapshot, &te ))
- {
- do
- {
- if (te.th32OwnerProcessID == curr_pid) curr_found++;
- if (te.th32OwnerProcessID == sub_pcs_pid) sub_found++;
- if (winetest_debug > 1)
- trace("PID=%x TID=%x %d\n", te.th32OwnerProcessID, te.th32ThreadID, te.tpBasePri);
- num++;
- } while (pThread32Next( hSnapshot, &te ));
- }
- ok(curr_found, "couldn't find self in thread list\n");
- ok(sub_found >= 2, "couldn't find sub-process threads in thread list\n");
-
- /* check that first really resets enumeration */
- curr_found = 0;
- sub_found = 0;
- if (pThread32First( hSnapshot, &te ))
- {
- do
- {
- if (te.th32OwnerProcessID == curr_pid) curr_found++;
- if (te.th32OwnerProcessID == sub_pcs_pid) sub_found++;
- if (winetest_debug > 1)
- trace("PID=%x TID=%x %d\n", te.th32OwnerProcessID, te.th32ThreadID, te.tpBasePri);
- num--;
- } while (pThread32Next( hSnapshot, &te ));
- }
- ok(curr_found, "couldn't find self in thread list\n");
- ok(sub_found >= 2, "couldn't find sub-process threads in thread list\n");
-
- me.dwSize = sizeof(me);
- ok(!pModule32First( hSnapshot, &me ), "shouldn't return a module\n");
-
- CloseHandle(hSnapshot);
- ok(!pThread32First( hSnapshot, &te ), "shouldn't return a thread\n");
-}
-
-static const char* curr_expected_modules[] =
-{
- "kernel32_test.exe",
- "kernel32.dll",
- "ntdll.dll"
-};
-
-static const char* sub_expected_modules[] =
-{
- "kernel32_test.exe",
- "kernel32.dll",
- "shell32.dll",
- "ntdll.dll"
-};
-
-#define NUM_OF(x) (sizeof(x) / sizeof(x[0]))
-
-static void test_module(DWORD pid, const char* expected[], unsigned num_expected)
-{
- HANDLE hSnapshot;
- PROCESSENTRY32 pe;
- THREADENTRY32 te;
- MODULEENTRY32 me;
- unsigned found[32];
- unsigned i;
- int num = 0;
-
- ok(NUM_OF(found) >= num_expected, "Internal: bump found[] size\n");
-
- hSnapshot = pCreateToolhelp32Snapshot( TH32CS_SNAPMODULE, pid );
- ok(hSnapshot != NULL, "Cannot create snapshot\n");
-
- for (i = 0; i < num_expected; i++) found[i] = 0;
- me.dwSize = sizeof(me);
- if (pModule32First( hSnapshot, &me ))
- {
- do
- {
- trace("PID=%x base=%p size=%x %s %s\n",
- me.th32ProcessID, me.modBaseAddr, me.modBaseSize, me.szExePath, me.szModule);
- ok(me.th32ProcessID == pid, "wrong returned process id\n");
- for (i = 0; i < num_expected; i++)
- if (!lstrcmpiA(expected[i], me.szModule)) found[i]++;
- num++;
- } while (pModule32Next( hSnapshot, &me ));
- }
- for (i = 0; i < num_expected; i++)
- ok(found[i] == 1, "Module %s is %s\n",
- expected[i], found[i] ? "listed more than once" : "not listed");
-
- /* check that first really resets the enumeration */
- for (i = 0; i < num_expected; i++) found[i] = 0;
- me.dwSize = sizeof(me);
- if (pModule32First( hSnapshot, &me ))
- {
- do
- {
- trace("PID=%x base=%p size=%x %s %s\n",
- me.th32ProcessID, me.modBaseAddr, me.modBaseSize, me.szExePath, me.szModule);
- for (i = 0; i < num_expected; i++)
- if (!lstrcmpiA(expected[i], me.szModule)) found[i]++;
- num--;
- } while (pModule32Next( hSnapshot, &me ));
- }
- for (i = 0; i < num_expected; i++)
- ok(found[i] == 1, "Module %s is %s\n",
- expected[i], found[i] ? "listed more than once" : "not listed");
- ok(!num, "mismatch in counting\n");
-
- pe.dwSize = sizeof(pe);
- ok(!pProcess32First( hSnapshot, &pe ), "shouldn't return a process\n");
-
- te.dwSize = sizeof(te);
- ok(!pThread32First( hSnapshot, &te ), "shouldn't return a thread\n");
-
- CloseHandle(hSnapshot);
- ok(!pModule32First( hSnapshot, &me ), "shouldn't return a module\n");
-}
-
-START_TEST(toolhelp)
-{
- DWORD pid = GetCurrentProcessId();
- int r;
- char *p, module[MAX_PATH];
- char buffer[MAX_PATH];
- SECURITY_ATTRIBUTES sa;
- PROCESS_INFORMATION info;
- STARTUPINFOA startup;
- HANDLE ev1, ev2;
- DWORD w;
- HANDLE hkernel32 = GetModuleHandleA("kernel32");
-
- pCreateToolhelp32Snapshot = (VOID *) GetProcAddress(hkernel32, "CreateToolhelp32Snapshot");
- pModule32First = (VOID *) GetProcAddress(hkernel32, "Module32First");
- pModule32Next = (VOID *) GetProcAddress(hkernel32, "Module32Next");
- pProcess32First = (VOID *) GetProcAddress(hkernel32, "Process32First");
- pProcess32Next = (VOID *) GetProcAddress(hkernel32, "Process32Next");
- pThread32First = (VOID *) GetProcAddress(hkernel32, "Thread32First");
- pThread32Next = (VOID *) GetProcAddress(hkernel32, "Thread32Next");
-
- if (!pCreateToolhelp32Snapshot ||
- !pModule32First || !pModule32Next ||
- !pProcess32First || !pProcess32Next ||
- !pThread32First || !pThread32Next)
- {
- win_skip("Needed functions are not available, most likely running on Windows NT\n");
- return;
- }
-
- r = init();
- ok(r == 0, "Basic init of sub-process test\n");
- if (r != 0) return;
-
- sa.nLength = sizeof(sa);
- sa.lpSecurityDescriptor = NULL;
- sa.bInheritHandle = TRUE;
-
- ev1 = CreateEventW(&sa, FALSE, FALSE, NULL);
- ev2 = CreateEventW(&sa, FALSE, FALSE, NULL);
- ok (ev1 != NULL && ev2 != NULL, "Couldn't create events\n");
- memset(&startup, 0, sizeof(startup));
- startup.cb = sizeof(startup);
- startup.dwFlags = STARTF_USESHOWWINDOW;
- startup.wShowWindow = SW_SHOWNORMAL;
-
- sprintf(buffer, "%s tests/toolhelp.c %lu %lu", selfname, (DWORD_PTR)ev1, (DWORD_PTR)ev2);
- ok(CreateProcessA(NULL, buffer, NULL, NULL, TRUE, 0, NULL, NULL, &startup, &info), "CreateProcess\n");
- /* wait for child to be initialized */
- w = WaitForSingleObject(ev1, WAIT_TIME);
- ok(w == WAIT_OBJECT_0, "Failed to wait on sub-process startup\n");
-
- GetModuleFileNameA( 0, module, sizeof(module) );
- if (!(p = strrchr( module, '\\' ))) p = module;
- else p++;
- curr_expected_modules[0] = p;
- sub_expected_modules[0] = p;
-
- test_process(pid, info.dwProcessId);
- test_thread(pid, info.dwProcessId);
- test_module(pid, curr_expected_modules, NUM_OF(curr_expected_modules));
- test_module(info.dwProcessId, sub_expected_modules, NUM_OF(sub_expected_modules));
-
- SetEvent(ev2);
- winetest_wait_child_process( info.hProcess );
-}
projects / reactos.git / blobdiff