[PATCH] [Kernel32]: added some tests for toolhelp functions

Eric Pouech eric.pouech at wanadoo.fr
Sat Apr 8 04:22:38 CDT 2006




A+
---

 dlls/kernel/tests/Makefile.in |    1 
 dlls/kernel/tests/toolhelp.c  |  318 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 319 insertions(+), 0 deletions(-)

diff --git a/dlls/kernel/tests/Makefile.in b/dlls/kernel/tests/Makefile.in
index 9543047..80eb401 100644
--- a/dlls/kernel/tests/Makefile.in
+++ b/dlls/kernel/tests/Makefile.in
@@ -30,6 +30,7 @@ CTESTS = \
 	thread.c \
 	time.c \
 	timer.c \
+	toolhelp.c \
 	virtual.c \
 	volume.c
 
diff --git a/dlls/kernel/tests/toolhelp.c b/dlls/kernel/tests/toolhelp.c
new file mode 100644
index 0000000..16fa906
--- /dev/null
+++ b/dlls/kernel/tests/toolhelp.c
@@ -0,0 +1,318 @@
+/*
+ * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include <stdarg.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <windef.h>
+#include <winbase.h>
+
+#include "tlhelp32.h"
+#include "wine/test.h"
+#include "windef.h"
+#include "winbase.h"
+#include "winuser.h"
+
+static char     selfname[MAX_PATH];
+
+/* 1 minute should be more than enough */
+#define WAIT_TIME       (60 * 1000)
+
+static DWORD WINAPI sub_thread(void* pmt)
+{
+    DWORD w = WaitForSingleObject((HANDLE)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;
+
+    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)atoi(argv[2]);
+        ev2 = (HANDLE)atoi(argv[3]);
+        ev3 = CreateEvent(NULL, FALSE, FALSE, NULL);
+
+        if (ev3 == NULL) ExitProcess(WAIT_ABANDONED);
+        hThread = CreateThread(NULL, 0, sub_thread, ev3, 0, NULL);
+        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;
+    THREADENTRY32       te;
+    MODULEENTRY32       me;
+    unsigned            found = 0;
+    int                 num = 0;
+
+    hSnapshot = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
+    ok(hSnapshot != NULL, "Cannot create snapshot\n");
+
+    /* check that this current process is enumerated */
+    pe.dwSize = sizeof(pe);
+    if (Process32First( hSnapshot, &pe ))
+    {
+        do
+        {
+            if (pe.th32ProcessID == curr_pid) found++;
+            if (pe.th32ProcessID == sub_pcs_pid) found++;
+            trace("PID=%lx %s\n", pe.th32ProcessID, pe.szExeFile);
+            num++;
+        } while (Process32Next( 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 (Process32First( hSnapshot, &pe ))
+    {
+        do
+        {
+            if (pe.th32ProcessID == curr_pid) found++;
+            if (pe.th32ProcessID == sub_pcs_pid) found++;
+            trace("PID=%lx %s\n", pe.th32ProcessID, pe.szExeFile);
+            num--;
+        } while (Process32Next( hSnapshot, &pe ));
+    }
+    ok(found == 2, "couldn't find self and/or sub-process in process list\n");
+    ok(!num, "mismatch in counting\n");
+
+    te.dwSize = sizeof(te);
+    ok(!Thread32First( hSnapshot, &te ), "shouldn't return a thread\n");
+
+    me.dwSize = sizeof(me);
+    ok(!Module32First( hSnapshot, &me ), "shouldn't return a module\n");
+
+    CloseHandle(hSnapshot);
+    ok(!Process32First( hSnapshot, &pe ), "shouldn't return a process\n");
+}
+
+static void test_thread(DWORD curr_pid, DWORD sub_pcs_pid)
+{
+    HANDLE              hSnapshot;
+    PROCESSENTRY32      pe;
+    THREADENTRY32       te;
+    MODULEENTRY32       me;
+    int                 num = 0;
+    unsigned            curr_found = 0;
+    unsigned            sub_found = 0;
+    
+    hSnapshot = CreateToolhelp32Snapshot( TH32CS_SNAPTHREAD, 0 );
+    ok(hSnapshot != NULL, "Cannot create snapshot\n");
+
+    /* check that this current process is enumerated */
+    te.dwSize = sizeof(te);
+    if (Thread32First( hSnapshot, &te ))
+    {
+        do
+        {
+            if (te.th32OwnerProcessID == curr_pid) curr_found++;
+            if (te.th32OwnerProcessID == sub_pcs_pid) sub_found++;
+            trace("PID=%lx TID=%lx %ld\n", te.th32OwnerProcessID, te.th32ThreadID, te.tpBasePri);
+            num++;
+        } while (Thread32Next( hSnapshot, &te ));
+    }
+    ok(curr_found == 1, "couldn't find self in thread list\n");
+    ok(sub_found == 2, "couldn't find sub-process thread's in thread list\n");
+
+    /* check that first really resets enumeration */
+    curr_found = 0;
+    sub_found = 0;
+    if (Thread32First( hSnapshot, &te ))
+    {
+        do
+        {
+            if (te.th32OwnerProcessID == curr_pid) curr_found++;
+            if (te.th32OwnerProcessID == sub_pcs_pid) sub_found++;
+            trace("PID=%lx TID=%lx %ld\n", te.th32OwnerProcessID, te.th32ThreadID, te.tpBasePri);
+            num--;
+        } while (Thread32Next( hSnapshot, &te ));
+    }
+    ok(curr_found == 1, "couldn't find self in thread list\n");
+    ok(sub_found == 2, "couldn't find sub-process thread's in thread list\n");
+
+    pe.dwSize = sizeof(pe);
+    ok(!Process32First( hSnapshot, &pe ), "shouldn't return a process\n");
+
+    me.dwSize = sizeof(me);
+    ok(!Module32First( hSnapshot, &me ), "shouldn't return a module\n");
+
+    CloseHandle(hSnapshot);
+    ok(!Thread32First( hSnapshot, &te ), "shouldn't return a thread\n");
+}
+
+static const char* curr_expected_modules[] =
+{
+    "kernel32.dll",
+    "kernel32_test.exe"
+    /* FIXME: could test for ntdll on NT and Wine */
+};
+static const char* sub_expected_modules[] =
+{
+    "kernel32.dll",
+    "kernel32_test.exe",
+    "shell32.dll"
+    /* FIXME: could test for ntdll on NT and Wine */
+};
+#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];
+    int                 i, num = 0;
+
+    ok(NUM_OF(found) >= num_expected, "Internal: bump found[] size\n");
+
+    hSnapshot = CreateToolhelp32Snapshot( 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 (Module32First( hSnapshot, &me ))
+    {
+        do
+        {
+            trace("PID=%lx base=%p size=%lx %s %s\n",
+                  me.th32ProcessID, me.modBaseAddr, me.modBaseSize, me.szExePath, me.szModule);
+            ok(me.th32ProcessID == pid, "wrong returned process id");
+            for (i = 0; i < num_expected; i++)
+                if (!strcmp(expected[i], me.szModule)) found[i]++;
+            num++;
+        } while (Module32Next( 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 (Module32First( hSnapshot, &me ))
+    {
+        do
+        {
+            trace("PID=%lx base=%p size=%lx %s %s\n",
+                  me.th32ProcessID, me.modBaseAddr, me.modBaseSize, me.szExePath, me.szModule);
+            for (i = 0; i < num_expected; i++)
+                if (!strcmp(expected[i], me.szModule)) found[i]++;
+            num--;
+        } while (Module32Next( 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(!Process32First( hSnapshot, &pe ), "shouldn't return a process\n");
+
+    me.dwSize = sizeof(me);
+    ok(!Thread32First( hSnapshot, &te ), "shouldn't return a thread\n");
+
+    CloseHandle(hSnapshot);
+    ok(!Module32First( hSnapshot, &me ), "shouldn't return a module\n");
+}
+
+START_TEST(toolhelp)
+{
+    DWORD               pid = GetCurrentProcessId();
+    int                 r;
+    char                buffer[MAX_PATH];
+    SECURITY_ATTRIBUTES sa;
+    PROCESS_INFORMATION	info;
+    STARTUPINFOA	startup;
+    HANDLE              ev1, ev2;
+    DWORD               w;
+
+    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 = CreateEvent(&sa, FALSE, FALSE, NULL);
+    ev2 = CreateEvent(&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)ev1, (DWORD)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");
+
+    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);
+    w = WaitForSingleObject(info.hProcess, WAIT_TIME);
+    ok(w == WAIT_OBJECT_0, "Failed to wait on sub-process termination\n");
+    ok(GetExitCodeProcess(info.hProcess, &w), "couldn't get process exit code\n");
+    ok(w == WAIT_OBJECT_0, "Sub-Process failed to terminate properly\n");
+}





More information about the wine-patches mailing list