ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

kernel32: Added some tests for toolhelp functions.

This commit is contained in:
Eric Pouech 2006-04-08 11:22:38 +02:00 committed by Alexandre Julliard
parent 4f3682f428
commit 69804258ca
3 changed files with 320 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dlls/kernel/tests/.gitignore vendored
View File

@ -24,5 +24,6 @@ testlist.c
thread.ok
time.ok
timer.ok
toolhelp.ok
virtual.ok
volume.ok

1
dlls/kernel/tests/Makefile.in
View File

@ -30,6 +30,7 @@ CTESTS = \
thread.c \
time.c \
timer.c \
toolhelp.c \
virtual.c \
volume.c

318
dlls/kernel/tests/toolhelp.c Normal file
View File

@ -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");
}