41303b7e9aec15776efc2617e8604b25690b8dfa
[reactos.git] / modules / rostests / winetests / ntdll / process.c
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
28 static NTSTATUS (WINAPI *pNtResumeProcess)(HANDLE);
29 static NTSTATUS (WINAPI *pNtSuspendProcess)(HANDLE);
30 static NTSTATUS (WINAPI *pNtSuspendThread)(HANDLE,PULONG);
31 static NTSTATUS (WINAPI *pNtResumeThread)(HANDLE);
32
33 static 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 for (;;)
127 {
128 ret = WaitForDebugEvent(&ev, INFINITE);
129 ok(ret, "WaitForDebugEvent failed, last error %#x.\n", GetLastError());
130 if (!ret) break;
131
132 if (ev.dwDebugEventCode == OUTPUT_DEBUG_STRING_EVENT) break;
133
134 ret = ContinueDebugEvent(ev.dwProcessId, ev.dwThreadId, DBG_CONTINUE);
135 ok(ret, "ContinueDebugEvent failed, last error %#x.\n", GetLastError());
136 if (!ret) break;
137 }
138
139 ResetEvent(event);
140
141 ret = WaitForSingleObject(event, 200);
142 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
143
144 status = pNtResumeProcess(info.hProcess);
145 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
146
147 ret = WaitForSingleObject(event, 200);
148 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
149
150 status = NtResumeThread(info.hThread, &count);
151 ok(status == STATUS_SUCCESS, "NtResumeProcess failed: %x\n", status);
152 ok(count == 0, "Expected count 0, got %d\n", count);
153
154 ret = WaitForSingleObject(event, 200);
155 ok(ret == WAIT_TIMEOUT, "Expected timeout, got: %d\n", ret);
156
157 ret = ContinueDebugEvent(ev.dwProcessId, ev.dwThreadId, DBG_CONTINUE);
158 ok(ret, "ContinueDebugEvent failed, last error %#x.\n", GetLastError());
159
160 ret = WaitForSingleObject(event, 200);
161 ok(ret == WAIT_OBJECT_0, "Event was not signaled: %d\n", ret);
162
163 TerminateProcess(info.hProcess, 0);
164
165 CloseHandle(info.hProcess);
166 CloseHandle(info.hThread);
167 }
168
169 static void dummy_process(char *event_name)
170 {
171 HANDLE event = OpenEventA(EVENT_ALL_ACCESS, FALSE, event_name);
172
173 while (TRUE)
174 {
175 SetEvent(event);
176 OutputDebugStringA("test");
177 Sleep(5);
178 }
179 }
180
181 START_TEST(process)
182 {
183 HMODULE mod;
184 char **argv;
185 int argc;
186
187 argc = winetest_get_mainargs(&argv);
188 if (argc >= 4 && strcmp(argv[2], "dummy_process") == 0)
189 {
190 dummy_process(argv[3]);
191 return;
192 }
193
194 mod = GetModuleHandleA("ntdll.dll");
195 if (!mod)
196 {
197 win_skip("Not running on NT, skipping tests\n");
198 return;
199 }
200
201 pNtResumeProcess = (void*)GetProcAddress(mod, "NtResumeProcess");
202 pNtSuspendProcess = (void*)GetProcAddress(mod, "NtSuspendProcess");
203 pNtResumeThread = (void*)GetProcAddress(mod, "NtResumeThread");
204 pNtSuspendThread = (void*)GetProcAddress(mod, "NtSuspendThread");
205
206 test_NtSuspendProcess(argv[0]);
207 }