huwcampbell.com
Reactos
1/*
2 * Unit test suite for process functions
3 *
4 * Copyright 2017 Michael Müller
5 *
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.
10 *
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.
15 *
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
19 */
20
21#include <stdio.h>
22
23#include "ntdll_test.h"
24
25#include "windef.h"
26#include "winbase.h"
27
28static NTSTATUS (WINAPI *pNtResumeProcess)(HANDLE);
29static NTSTATUS (WINAPI *pNtSuspendProcess)(HANDLE);
30static NTSTATUS (WINAPI *pNtSuspendThread)(HANDLE,PULONG);
31static NTSTATUS (WINAPI *pNtResumeThread)(HANDLE);
32
33static void test_NtSuspendProcess(char *process_name)
34{
35 PROCESS_INFORMATION info;
36 DEBUG_EVENT ev;
37 STARTUPINFOA startup;
38 NTSTATUS status;
39 HANDLE event;
40 char buffer[MAX_PATH];
41 ULONG count;
42 DWORD ret;
43
44 status = pNtResumeProcess(GetCurrentProcess());
45 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
46
47 event = CreateEventA(NULL, TRUE, FALSE, "wine_suspend_event");
48 ok(!!event, "Failed to create event: %u\n", GetLastError());
49
50 memset(&startup, 0, sizeof(startup));
51 startup.cb = sizeof(startup);
52
53 sprintf(buffer, "%s tests/process.c dummy_process wine_suspend_event", process_name);
54 ret = CreateProcessA(NULL, buffer, NULL, NULL, FALSE, 0, NULL, NULL, &startup, &info);
55 ok(ret, "CreateProcess failed with error %u\n", GetLastError());
56
57 ret = WaitForSingleObject(event, 500);
58 ok(ret == WAIT_OBJECT_0, "Event was not signaled: %d\n", ret);
59
60 status = pNtSuspendProcess(info.hProcess);
61 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
62
63 ResetEvent(event);
64
65 ret = WaitForSingleObject(event, 200);
66 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
67
68 status = NtResumeThread(info.hThread, &count);
69 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
70 ok(count == 1, "Expected count 1, got %d\n", count);
71
72 ret = WaitForSingleObject(event, 200);
73 ok(ret == WAIT_OBJECT_0, "Event was not signaled: %d\n", ret);
74
75 status = pNtResumeProcess(info.hProcess);
76 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
77
78 status = pNtSuspendThread(info.hThread, &count);
79 ok(status == STATUS_SUCCESS, "NtSuspendThread failed: %x\n", status);
80 ok(count == 0, "Expected count 0, got %d\n", count);
81
82 ResetEvent(event);
83
84 ret = WaitForSingleObject(event, 200);
85 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
86
87 status = pNtResumeProcess(info.hProcess);
88 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
89
90 ret = WaitForSingleObject(event, 200);
91 ok(ret == WAIT_OBJECT_0, "Event was not signaled: %d\n", ret);
92
93 status = pNtSuspendThread(info.hThread, &count);
94 ok(status == STATUS_SUCCESS, "NtSuspendThread failed: %x\n", status);
95 ok(count == 0, "Expected count 0, got %d\n", count);
96
97 status = pNtSuspendThread(info.hThread, &count);
98 ok(status == STATUS_SUCCESS, "NtSuspendThread failed: %x\n", status);
99 ok(count == 1, "Expected count 1, got %d\n", count);
100
101 ResetEvent(event);
102
103 ret = WaitForSingleObject(event, 200);
104 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
105
106 status = pNtResumeProcess(info.hProcess);
107 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
108
109 ret = WaitForSingleObject(event, 200);
110 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
111
112 status = pNtResumeProcess(info.hProcess);
113 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
114
115 ret = WaitForSingleObject(event, 200);
116 ok(ret == WAIT_OBJECT_0, "Event was not signaled: %d\n", ret);
117
118 ret = DebugActiveProcess(info.dwProcessId);
119 ok(ret, "Failed to debug process: %d\n", GetLastError());
120
121 ResetEvent(event);
122
123 ret = WaitForSingleObject(event, 200);
124 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
125
126 disable_success_count
127 for (;;)
128 {
129 ret = WaitForDebugEvent(&ev, INFINITE);
130 ok(ret, "WaitForDebugEvent failed, last error %#x.\n", GetLastError());
131 if (!ret) break;
132
133 if (ev.dwDebugEventCode == OUTPUT_DEBUG_STRING_EVENT) break;
134
135 ret = ContinueDebugEvent(ev.dwProcessId, ev.dwThreadId, DBG_CONTINUE);
136 ok(ret, "ContinueDebugEvent failed, last error %#x.\n", GetLastError());
137 if (!ret) break;
138 }
139
140 ResetEvent(event);
141
142 ret = WaitForSingleObject(event, 200);
143 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
144
145 status = pNtResumeProcess(info.hProcess);
146 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
147
148 ret = WaitForSingleObject(event, 200);
149 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
150
151 status = NtResumeThread(info.hThread, &count);
152 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
153 ok(count == 0, "Expected count 0, got %d\n", count);
154
155 ret = WaitForSingleObject(event, 200);
156 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
157
158 ret = ContinueDebugEvent(ev.dwProcessId, ev.dwThreadId, DBG_CONTINUE);
159 ok(ret, "ContinueDebugEvent failed, last error %#x.\n", GetLastError());
160
161 ret = WaitForSingleObject(event, 200);
162 ros_skip_flaky
163 ok(ret == WAIT_OBJECT_0, "Event was not signaled: %d\n", ret);
164
165 TerminateProcess(info.hProcess, 0);
166
167 CloseHandle(info.hProcess);
168 CloseHandle(info.hThread);
169}
170
171static void dummy_process(char *event_name)
172{
173 HANDLE event = OpenEventA(EVENT_ALL_ACCESS, FALSE, event_name);
174
175 while (TRUE)
176 {
177 SetEvent(event);
178 OutputDebugStringA("test");
179 Sleep(5);
180 }
181}
182
183START_TEST(process)
184{
185 HMODULE mod;
186 char **argv;
187 int argc;
188
189 argc = winetest_get_mainargs(&argv);
190 if (argc >= 4 && strcmp(argv[2], "dummy_process") == 0)
191 {
192 dummy_process(argv[3]);
193 return;
194 }
195
196 mod = GetModuleHandleA("ntdll.dll");
197 if (!mod)
198 {
199 win_skip("Not running on NT, skipping tests\n");
200 return;
201 }
202
203 pNtResumeProcess = (void*)GetProcAddress(mod, "NtResumeProcess");
204 pNtSuspendProcess = (void*)GetProcAddress(mod, "NtSuspendProcess");
205 pNtResumeThread = (void*)GetProcAddress(mod, "NtResumeThread");
206 pNtSuspendThread = (void*)GetProcAddress(mod, "NtSuspendThread");
207
208 test_NtSuspendProcess(argv[0]);
209}