ReactOS 0.4.15-dev-7958-gcd0bb1a
KeGuardedMutex.c
Go to the documentation of this file.
1/*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: Kernel-Mode Test Suite Guarded Mutex test
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8#include <kmt_test.h>
9
10#define NDEBUG
11#include <debug.h>
12
13static
14_IRQL_requires_min_(PASSIVE_LEVEL)
15_IRQL_requires_max_(DISPATCH_LEVEL)
16BOOLEAN
17(NTAPI
18*pKeAreAllApcsDisabled)(VOID);
19
20static
21_Acquires_lock_(_Global_critical_region_)
22_Requires_lock_not_held_(*Mutex)
23_Acquires_lock_(*Mutex)
24_IRQL_requires_max_(APC_LEVEL)
25_IRQL_requires_min_(PASSIVE_LEVEL)
26VOID
27(FASTCALL
28*pKeAcquireGuardedMutex)(
29 _Inout_ PKGUARDED_MUTEX GuardedMutex);
30
31static
32_Requires_lock_not_held_(*FastMutex)
33_Acquires_lock_(*FastMutex)
34_IRQL_requires_max_(APC_LEVEL)
35_IRQL_requires_min_(PASSIVE_LEVEL)
36VOID
37(FASTCALL
38*pKeAcquireGuardedMutexUnsafe)(
39 _Inout_ PKGUARDED_MUTEX GuardedMutex);
40
41static
42_Acquires_lock_(_Global_critical_region_)
43_IRQL_requires_max_(APC_LEVEL)
44VOID
45(NTAPI
46*pKeEnterGuardedRegion)(VOID);
47
48static
49_Releases_lock_(_Global_critical_region_)
50_IRQL_requires_max_(APC_LEVEL)
51VOID
52(NTAPI
53*pKeLeaveGuardedRegion)(VOID);
54
55static
56_IRQL_requires_max_(APC_LEVEL)
57_IRQL_requires_min_(PASSIVE_LEVEL)
58VOID
59(FASTCALL
60*pKeInitializeGuardedMutex)(
61 _Out_ PKGUARDED_MUTEX GuardedMutex);
62
63static
64_Requires_lock_held_(*FastMutex)
65_Releases_lock_(*FastMutex)
66_IRQL_requires_max_(APC_LEVEL)
67VOID
68(FASTCALL
69*pKeReleaseGuardedMutexUnsafe)(
70 _Inout_ PKGUARDED_MUTEX GuardedMutex);
71
72static
73_Releases_lock_(_Global_critical_region_)
74_Requires_lock_held_(*Mutex)
75_Releases_lock_(*Mutex)
76_IRQL_requires_max_(APC_LEVEL)
77VOID
78(FASTCALL
79*pKeReleaseGuardedMutex)(
80 _Inout_ PKGUARDED_MUTEX GuardedMutex);
81
82static
83_Must_inspect_result_
84_Success_(return != FALSE)
85_IRQL_requires_max_(APC_LEVEL)
86_Post_satisfies_(return == 1 || return == 0)
87BOOLEAN
88(FASTCALL
89*pKeTryToAcquireGuardedMutex)(
90 _When_ (return, _Requires_lock_not_held_(*_Curr_) _Acquires_exclusive_lock_(*_Curr_)) _Acquires_lock_(_Global_critical_region_)
91 _Inout_ PKGUARDED_MUTEX GuardedMutex);
92
93#define CheckMutex(Mutex, ExpectedCount, ExpectedOwner, ExpectedContention, \
94 ExpectedKernelApcDisable, ExpectedSpecialApcDisable, \
95 KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, \
96 ExpectedIrql) do \
97{ \
98 ok_eq_long((Mutex)->Count, ExpectedCount); \
99 ok_eq_pointer((Mutex)->Owner, ExpectedOwner); \
100 ok_eq_ulong((Mutex)->Contention, ExpectedContention); \
101 ok_eq_int((Mutex)->KernelApcDisable, ExpectedKernelApcDisable); \
102 if (KmtIsCheckedBuild) \
103 ok_eq_int((Mutex)->SpecialApcDisable, ExpectedSpecialApcDisable); \
104 else \
105 ok_eq_int((Mutex)->SpecialApcDisable, 0x5555); \
106 ok_eq_bool(KeAreApcsDisabled(), KernelApcsDisabled || SpecialApcsDisabled); \
107 ok_eq_int(Thread->KernelApcDisable, KernelApcsDisabled); \
108 ok_eq_bool(pKeAreAllApcsDisabled(), AllApcsDisabled); \
109 ok_eq_int(Thread->SpecialApcDisable, SpecialApcsDisabled); \
110 ok_irql(ExpectedIrql); \
111} while (0)
112
113static
114VOID
115TestGuardedMutex(
116 PKGUARDED_MUTEX Mutex,
117 SHORT KernelApcsDisabled,
118 SHORT SpecialApcsDisabled,
119 SHORT AllApcsDisabled,
120 KIRQL OriginalIrql)
121{
122 PKTHREAD Thread = KeGetCurrentThread();
123
124 ok_irql(OriginalIrql);
125 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, 0x5555, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
126
127 /* these ASSERT */
128 if (!KmtIsCheckedBuild || OriginalIrql <= APC_LEVEL)
129 {
130 /* acquire/release normally */
131 pKeAcquireGuardedMutex(Mutex);
132 CheckMutex(Mutex, 0L, Thread, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled - 1, TRUE, OriginalIrql);
133 ok_bool_false(pKeTryToAcquireGuardedMutex(Mutex), "KeTryToAcquireGuardedMutex returned");
134 CheckMutex(Mutex, 0L, Thread, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled - 1, TRUE, OriginalIrql);
135 pKeReleaseGuardedMutex(Mutex);
136 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
137
138 /* try to acquire */
139 ok_bool_true(pKeTryToAcquireGuardedMutex(Mutex), "KeTryToAcquireGuardedMutex returned");
140 CheckMutex(Mutex, 0L, Thread, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled - 1, TRUE, OriginalIrql);
141 pKeReleaseGuardedMutex(Mutex);
142 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
143 }
144 else
145 /* Make the following test happy */
146 Mutex->SpecialApcDisable = SpecialApcsDisabled - 1;
147
148 /* ASSERT */
149 if (!KmtIsCheckedBuild || OriginalIrql == APC_LEVEL || SpecialApcsDisabled < 0)
150 {
151 /* acquire/release unsafe */
152 pKeAcquireGuardedMutexUnsafe(Mutex);
153 CheckMutex(Mutex, 0L, Thread, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
154 pKeReleaseGuardedMutexUnsafe(Mutex);
155 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
156 }
157
158 /* Bugchecks >= DISPATCH_LEVEL */
159 if (!KmtIsCheckedBuild)
160 {
161 /* mismatched acquire/release */
162 pKeAcquireGuardedMutex(Mutex);
163 CheckMutex(Mutex, 0L, Thread, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled - 1, TRUE, OriginalIrql);
164 pKeReleaseGuardedMutexUnsafe(Mutex);
165 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled - 1, TRUE, OriginalIrql);
166 pKeLeaveGuardedRegion();
167 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
168
169 pKeAcquireGuardedMutexUnsafe(Mutex);
170 CheckMutex(Mutex, 0L, Thread, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
171 pKeReleaseGuardedMutex(Mutex);
172 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled + 1, OriginalIrql >= APC_LEVEL || SpecialApcsDisabled != -1, OriginalIrql);
173 pKeEnterGuardedRegion();
174 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
175
176 /* release without acquire */
177 pKeReleaseGuardedMutexUnsafe(Mutex);
178 CheckMutex(Mutex, 0L, NULL, 0LU, 0x5555, SpecialApcsDisabled - 1, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, OriginalIrql);
179 pKeReleaseGuardedMutex(Mutex);
180 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled, KernelApcsDisabled, SpecialApcsDisabled + 1, OriginalIrql >= APC_LEVEL || SpecialApcsDisabled != -1, OriginalIrql);
181 pKeReleaseGuardedMutex(Mutex);
182 /* TODO: here we see that Mutex->Count isn't actually just a count. Test the bits correctly! */
183 CheckMutex(Mutex, 0L, NULL, 0LU, 0x5555, SpecialApcsDisabled, KernelApcsDisabled, SpecialApcsDisabled + 2, OriginalIrql >= APC_LEVEL || SpecialApcsDisabled != -2, OriginalIrql);
184 pKeReleaseGuardedMutex(Mutex);
185 CheckMutex(Mutex, 1L, NULL, 0LU, 0x5555, SpecialApcsDisabled, KernelApcsDisabled, SpecialApcsDisabled + 3, OriginalIrql >= APC_LEVEL || SpecialApcsDisabled != -3, OriginalIrql);
186 Thread->SpecialApcDisable -= 3;
187 }
188
189 /* make sure we survive this in case of error */
190 ok_eq_long(Mutex->Count, 1L);
191 Mutex->Count = 1;
192 ok_eq_int(Thread->KernelApcDisable, KernelApcsDisabled);
193 Thread->KernelApcDisable = KernelApcsDisabled;
194 ok_eq_int(Thread->SpecialApcDisable, SpecialApcsDisabled);
195 Thread->SpecialApcDisable = SpecialApcsDisabled;
196 ok_irql(OriginalIrql);
197}
198
199typedef VOID (FASTCALL *PMUTEX_FUNCTION)(PKGUARDED_MUTEX);
200typedef BOOLEAN (FASTCALL *PMUTEX_TRY_FUNCTION)(PKGUARDED_MUTEX);
201
202typedef struct
203{
204 HANDLE Handle;
205 PKTHREAD Thread;
206 KIRQL Irql;
207 PKGUARDED_MUTEX Mutex;
208 PMUTEX_FUNCTION Acquire;
209 PMUTEX_TRY_FUNCTION TryAcquire;
210 PMUTEX_FUNCTION Release;
211 BOOLEAN Try;
212 BOOLEAN RetExpected;
213 KEVENT InEvent;
214 KEVENT OutEvent;
215} THREAD_DATA, *PTHREAD_DATA;
216
217static
218VOID
219NTAPI
220AcquireMutexThread(
221 PVOID Parameter)
222{
223 PTHREAD_DATA ThreadData = Parameter;
224 KIRQL Irql;
225 BOOLEAN Ret = FALSE;
226 NTSTATUS Status;
227
228 DPRINT("Thread starting\n");
229 KeRaiseIrql(ThreadData->Irql, &Irql);
230
231 if (ThreadData->Try)
232 {
233 Ret = ThreadData->TryAcquire(ThreadData->Mutex);
234 ok_eq_bool(Ret, ThreadData->RetExpected);
235 }
236 else
237 ThreadData->Acquire(ThreadData->Mutex);
238
239 ok_bool_false(KeSetEvent(&ThreadData->OutEvent, 0, TRUE), "KeSetEvent returned");
240 DPRINT("Thread now waiting\n");
241 Status = KeWaitForSingleObject(&ThreadData->InEvent, Executive, KernelMode, FALSE, NULL);
242 DPRINT("Thread done waiting\n");
243 ok_eq_hex(Status, STATUS_SUCCESS);
244
245 if (!ThreadData->Try || Ret)
246 ThreadData->Release(ThreadData->Mutex);
247
248 KeLowerIrql(Irql);
249 DPRINT("Thread exiting\n");
250}
251
252static
253VOID
254InitThreadData(
255 PTHREAD_DATA ThreadData,
256 PKGUARDED_MUTEX Mutex,
257 PMUTEX_FUNCTION Acquire,
258 PMUTEX_TRY_FUNCTION TryAcquire,
259 PMUTEX_FUNCTION Release)
260{
261 ThreadData->Mutex = Mutex;
262 KeInitializeEvent(&ThreadData->InEvent, NotificationEvent, FALSE);
263 KeInitializeEvent(&ThreadData->OutEvent, NotificationEvent, FALSE);
264 ThreadData->Acquire = Acquire;
265 ThreadData->TryAcquire = TryAcquire;
266 ThreadData->Release = Release;
267}
268
269static
270NTSTATUS
271StartThread(
272 PTHREAD_DATA ThreadData,
273 PLARGE_INTEGER Timeout,
274 KIRQL Irql,
275 BOOLEAN Try,
276 BOOLEAN RetExpected)
277{
278 NTSTATUS Status = STATUS_SUCCESS;
279 OBJECT_ATTRIBUTES Attributes;
280
281 ThreadData->Try = Try;
282 ThreadData->Irql = Irql;
283 ThreadData->RetExpected = RetExpected;
284 InitializeObjectAttributes(&Attributes, NULL, OBJ_KERNEL_HANDLE, NULL, NULL);
285 Status = PsCreateSystemThread(&ThreadData->Handle, GENERIC_ALL, &Attributes, NULL, NULL, AcquireMutexThread, ThreadData);
286 ok_eq_hex(Status, STATUS_SUCCESS);
287 Status = ObReferenceObjectByHandle(ThreadData->Handle, SYNCHRONIZE, *PsThreadType, KernelMode, (PVOID *)&ThreadData->Thread, NULL);
288 ok_eq_hex(Status, STATUS_SUCCESS);
289
290 return KeWaitForSingleObject(&ThreadData->OutEvent, Executive, KernelMode, FALSE, Timeout);
291}
292
293static
294VOID
295FinishThread(
296 PTHREAD_DATA ThreadData)
297{
298 NTSTATUS Status = STATUS_SUCCESS;
299
300 KeSetEvent(&ThreadData->InEvent, 0, TRUE);
301 Status = KeWaitForSingleObject(ThreadData->Thread, Executive, KernelMode, FALSE, NULL);
302 ok_eq_hex(Status, STATUS_SUCCESS);
303
304 ObDereferenceObject(ThreadData->Thread);
305 Status = ZwClose(ThreadData->Handle);
306 ok_eq_hex(Status, STATUS_SUCCESS);
307 KeClearEvent(&ThreadData->InEvent);
308 KeClearEvent(&ThreadData->OutEvent);
309}
310
311static
312VOID
313TestGuardedMutexConcurrent(
314 PKGUARDED_MUTEX Mutex)
315{
316 NTSTATUS Status;
317 THREAD_DATA ThreadData;
318 THREAD_DATA ThreadData2;
319 THREAD_DATA ThreadDataUnsafe;
320 THREAD_DATA ThreadDataTry;
321 PKTHREAD Thread = KeGetCurrentThread();
322 LARGE_INTEGER Timeout;
323 Timeout.QuadPart = -50 * 1000 * 10; /* 50 ms */
324
325 InitThreadData(&ThreadData, Mutex, pKeAcquireGuardedMutex, NULL, pKeReleaseGuardedMutex);
326 InitThreadData(&ThreadData2, Mutex, pKeAcquireGuardedMutex, NULL, pKeReleaseGuardedMutex);
327 InitThreadData(&ThreadDataUnsafe, Mutex, pKeAcquireGuardedMutexUnsafe, NULL, pKeReleaseGuardedMutexUnsafe);
328 InitThreadData(&ThreadDataTry, Mutex, NULL, pKeTryToAcquireGuardedMutex, pKeReleaseGuardedMutex);
329
330 /* have a thread acquire the mutex */
331 Status = StartThread(&ThreadData, NULL, PASSIVE_LEVEL, FALSE, FALSE);
332 ok_eq_hex(Status, STATUS_SUCCESS);
333 CheckMutex(Mutex, 0L, ThreadData.Thread, 0LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
334 /* have a second thread try to acquire it -- should fail */
335 Status = StartThread(&ThreadDataTry, NULL, PASSIVE_LEVEL, TRUE, FALSE);
336 ok_eq_hex(Status, STATUS_SUCCESS);
337 CheckMutex(Mutex, 0L, ThreadData.Thread, 0LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
338 FinishThread(&ThreadDataTry);
339
340 /* have another thread acquire it -- should block */
341 Status = StartThread(&ThreadData2, &Timeout, APC_LEVEL, FALSE, FALSE);
342 ok_eq_hex(Status, STATUS_TIMEOUT);
343 CheckMutex(Mutex, 4L, ThreadData.Thread, 1LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
344
345 /* finish the first thread -- now the second should become available */
346 FinishThread(&ThreadData);
347 Status = KeWaitForSingleObject(&ThreadData2.OutEvent, Executive, KernelMode, FALSE, NULL);
348 ok_eq_hex(Status, STATUS_SUCCESS);
349 CheckMutex(Mutex, 0L, ThreadData2.Thread, 1LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
350
351 /* block two more threads */
352 Status = StartThread(&ThreadDataUnsafe, &Timeout, APC_LEVEL, FALSE, FALSE);
353 ok_eq_hex(Status, STATUS_TIMEOUT);
354 CheckMutex(Mutex, 4L, ThreadData2.Thread, 2LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
355
356 Status = StartThread(&ThreadData, &Timeout, PASSIVE_LEVEL, FALSE, FALSE);
357 ok_eq_hex(Status, STATUS_TIMEOUT);
358 CheckMutex(Mutex, 8L, ThreadData2.Thread, 3LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
359
360 /* finish 1 */
361 FinishThread(&ThreadData2);
362 Status = KeWaitForSingleObject(&ThreadDataUnsafe.OutEvent, Executive, KernelMode, FALSE, NULL);
363 ok_eq_hex(Status, STATUS_SUCCESS);
364 CheckMutex(Mutex, 4L, ThreadDataUnsafe.Thread, 3LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
365
366 /* finish 2 */
367 FinishThread(&ThreadDataUnsafe);
368 Status = KeWaitForSingleObject(&ThreadData.OutEvent, Executive, KernelMode, FALSE, NULL);
369 ok_eq_hex(Status, STATUS_SUCCESS);
370 CheckMutex(Mutex, 0L, ThreadData.Thread, 3LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
371
372 /* finish 3 */
373 FinishThread(&ThreadData);
374
375 CheckMutex(Mutex, 1L, NULL, 3LU, 0x5555, -1, 0, 0, FALSE, PASSIVE_LEVEL);
376}
377
378START_TEST(KeGuardedMutex)
379{
380 KGUARDED_MUTEX Mutex;
381 KIRQL OldIrql;
382 PKTHREAD Thread = KeGetCurrentThread();
383 struct {
384 KIRQL Irql;
385 SHORT KernelApcsDisabled;
386 SHORT SpecialApcsDisabled;
387 BOOLEAN AllApcsDisabled;
388 } TestIterations[] =
389 {
390 { PASSIVE_LEVEL, 0, 0, FALSE },
391 { PASSIVE_LEVEL, -1, 0, FALSE },
392 { PASSIVE_LEVEL, -3, 0, FALSE },
393 { PASSIVE_LEVEL, 0, -1, TRUE },
394 { PASSIVE_LEVEL, -1, -1, TRUE },
395 { PASSIVE_LEVEL, -3, -2, TRUE },
396 // 6
397 { APC_LEVEL, 0, 0, TRUE },
398 { APC_LEVEL, -1, 0, TRUE },
399 { APC_LEVEL, -3, 0, TRUE },
400 { APC_LEVEL, 0, -1, TRUE },
401 { APC_LEVEL, -1, -1, TRUE },
402 { APC_LEVEL, -3, -2, TRUE },
403 // 12
404 { DISPATCH_LEVEL, 0, 0, TRUE },
405 { DISPATCH_LEVEL, -1, 0, TRUE },
406 { DISPATCH_LEVEL, -3, 0, TRUE },
407 { DISPATCH_LEVEL, 0, -1, TRUE },
408 { DISPATCH_LEVEL, -1, -1, TRUE },
409 { DISPATCH_LEVEL, -3, -2, TRUE },
410 // 18
411 { HIGH_LEVEL, 0, 0, TRUE },
412 { HIGH_LEVEL, -1, 0, TRUE },
413 { HIGH_LEVEL, -3, 0, TRUE },
414 { HIGH_LEVEL, 0, -1, TRUE },
415 { HIGH_LEVEL, -1, -1, TRUE },
416 { HIGH_LEVEL, -3, -2, TRUE },
417 };
418 int i;
419
420 pKeAreAllApcsDisabled = KmtGetSystemRoutineAddress(L"KeAreAllApcsDisabled");
421 pKeInitializeGuardedMutex = KmtGetSystemRoutineAddress(L"KeInitializeGuardedMutex");
422 pKeAcquireGuardedMutex = KmtGetSystemRoutineAddress(L"KeAcquireGuardedMutex");
423 pKeAcquireGuardedMutexUnsafe = KmtGetSystemRoutineAddress(L"KeAcquireGuardedMutexUnsafe");
424 pKeEnterGuardedRegion = KmtGetSystemRoutineAddress(L"KeEnterGuardedRegion");
425 pKeLeaveGuardedRegion = KmtGetSystemRoutineAddress(L"KeLeaveGuardedRegion");
426 pKeReleaseGuardedMutex = KmtGetSystemRoutineAddress(L"KeReleaseGuardedMutex");
427 pKeReleaseGuardedMutexUnsafe = KmtGetSystemRoutineAddress(L"KeReleaseGuardedMutexUnsafe");
428 pKeTryToAcquireGuardedMutex = KmtGetSystemRoutineAddress(L"KeTryToAcquireGuardedMutex");
429
430 if (skip(pKeAreAllApcsDisabled &&
431 pKeInitializeGuardedMutex &&
432 pKeAcquireGuardedMutex &&
433 pKeAcquireGuardedMutexUnsafe &&
434 pKeEnterGuardedRegion &&
435 pKeLeaveGuardedRegion &&
436 pKeReleaseGuardedMutex &&
437 pKeReleaseGuardedMutexUnsafe &&
438 pKeTryToAcquireGuardedMutex, "No guarded mutexes\n"))
439 {
440 return;
441 }
442
443 for (i = 0; i < sizeof TestIterations / sizeof TestIterations[0]; ++i)
444 {
445 trace("Run %d\n", i);
446 KeRaiseIrql(TestIterations[i].Irql, &OldIrql);
447 Thread->KernelApcDisable = TestIterations[i].KernelApcsDisabled;
448 Thread->SpecialApcDisable = TestIterations[i].SpecialApcsDisabled;
449
450 RtlFillMemory(&Mutex, sizeof Mutex, 0x55);
451 pKeInitializeGuardedMutex(&Mutex);
452 CheckMutex(&Mutex, 1L, NULL, 0LU, 0x5555, 0x5555, TestIterations[i].KernelApcsDisabled, TestIterations[i].SpecialApcsDisabled, TestIterations[i].AllApcsDisabled, TestIterations[i].Irql);
453 TestGuardedMutex(&Mutex, TestIterations[i].KernelApcsDisabled, TestIterations[i].SpecialApcsDisabled, TestIterations[i].AllApcsDisabled, TestIterations[i].Irql);
454
455 Thread->SpecialApcDisable = 0;
456 Thread->KernelApcDisable = 0;
457 KeLowerIrql(OldIrql);
458 }
459
460 trace("Concurrent test\n");
461 RtlFillMemory(&Mutex, sizeof Mutex, 0x55);
462 pKeInitializeGuardedMutex(&Mutex);
463 TestGuardedMutexConcurrent(&Mutex);
464}
StartThread
static NTSTATUS StartThread(PTHREAD_DATA ThreadData, PLARGE_INTEGER Timeout, KIRQL Irql, BOOLEAN Try, BOOLEAN RetExpected)
Definition: KeGuardedMutex.c:271
CheckMutex
#define CheckMutex(Mutex, ExpectedCount, ExpectedOwner, ExpectedContention, ExpectedKernelApcDisable, ExpectedSpecialApcDisable, KernelApcsDisabled, SpecialApcsDisabled, AllApcsDisabled, ExpectedIrql)
AcquireMutexThread
static VOID NTAPI AcquireMutexThread(PVOID Parameter)
Definition: KeGuardedMutex.c:220
InitThreadData
static VOID InitThreadData(PTHREAD_DATA ThreadData, PKGUARDED_MUTEX Mutex, PMUTEX_FUNCTION Acquire, PMUTEX_TRY_FUNCTION TryAcquire, PMUTEX_FUNCTION Release)
Definition: KeGuardedMutex.c:254
FinishThread
static VOID FinishThread(PTHREAD_DATA ThreadData)
Definition: KeGuardedMutex.c:295
TestGuardedMutexConcurrent
static VOID TestGuardedMutexConcurrent(PKGUARDED_MUTEX Mutex)
Definition: KeGuardedMutex.c:313
PTHREAD_DATA
struct THREAD_DATA * PTHREAD_DATA
PMUTEX_FUNCTION
VOID(FASTCALL * PMUTEX_FUNCTION)(PKGUARDED_MUTEX)
Definition: KeGuardedMutex.c:199
PMUTEX_TRY_FUNCTION
BOOLEAN(FASTCALL * PMUTEX_TRY_FUNCTION)(PKGUARDED_MUTEX)
Definition: KeGuardedMutex.c:200
BOOLEAN
unsigned char BOOLEAN
Definition: ProcessorBind.h:185
VOID
#define VOID
Definition: acefi.h:82
ok_eq_hex
#define ok_eq_hex(value, expected)
Definition: apitest.h:77
ok_eq_long
#define ok_eq_long(value, expected)
Definition: apitest.h:62
ok_bool_false
#define ok_bool_false(value, desc)
Definition: apitest.h:79
ok_bool_true
#define ok_bool_true(value, desc)
Definition: apitest.h:78
ok_eq_int
#define ok_eq_int(value, expected)
Definition: apitest.h:60
ok_eq_bool
#define ok_eq_bool(value, expected)
Definition: apitest.h:80
trace
#define trace
Definition: atltest.h:70
skip
#define skip(...)
Definition: atltest.h:64
START_TEST
#define START_TEST(x)
Definition: atltest.h:75
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
U
#define U(x)
Definition: wordpad.c:45
return
return
Definition: dirsup.c:529
Release
_In_ BOOLEAN Release
Definition: cdrom.h:920
Mutex
Definition: Mutex.h:16
_Acquires_lock_
#define _Acquires_lock_(lock)
Definition: concurrencysal.h:78
_Requires_lock_held_
#define _Requires_lock_held_(lock)
Definition: concurrencysal.h:105
_Acquires_exclusive_lock_
#define _Acquires_exclusive_lock_(lock)
Definition: concurrencysal.h:77
_Requires_lock_not_held_
#define _Requires_lock_not_held_(lock)
Definition: concurrencysal.h:106
_Releases_lock_
#define _Releases_lock_(lock)
Definition: concurrencysal.h:100
Irql
_Out_ PKIRQL Irql
Definition: csq.h:179
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
FALSE
#define FALSE
Definition: types.h:117
_IRQL_requires_min_
#define _IRQL_requires_min_(irql)
Definition: driverspecs.h:231
_IRQL_requires_max_
#define _IRQL_requires_max_(irql)
Definition: driverspecs.h:230
Try
int Try(int arg)
Definition: ehframes.cpp:53
PASSIVE_LEVEL
#define PASSIVE_LEVEL
Definition: env_spec_w32.h:693
KIRQL
UCHAR KIRQL
Definition: env_spec_w32.h:591
KeRaiseIrql
#define KeRaiseIrql(irql, oldIrql)
Definition: env_spec_w32.h:597
KeWaitForSingleObject
#define KeWaitForSingleObject(pEvt, foo, a, b, c)
Definition: env_spec_w32.h:478
HIGH_LEVEL
#define HIGH_LEVEL
Definition: env_spec_w32.h:703
KeInitializeEvent
#define KeInitializeEvent(pEvt, foo, foo2)
Definition: env_spec_w32.h:477
APC_LEVEL
#define APC_LEVEL
Definition: env_spec_w32.h:695
KeLowerIrql
#define KeLowerIrql(oldIrql)
Definition: env_spec_w32.h:602
KeSetEvent
#define KeSetEvent(pEvt, foo, foo2)
Definition: env_spec_w32.h:476
DISPATCH_LEVEL
#define DISPATCH_LEVEL
Definition: env_spec_w32.h:696
KeClearEvent
VOID NTAPI KeClearEvent(IN PKEVENT Event)
Definition: eventobj.c:22
Thread
_In_opt_ PFILE_OBJECT _In_opt_ PETHREAD Thread
Definition: fltkernel.h:2653
Handle
ULONG Handle
Definition: gdb_input.c:15
Status
Status
Definition: gdiplustypes.h:25
i
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble const GLfloat const GLdouble const GLfloat GLint i
Definition: glfuncs.h:248
KeGetCurrentThread
#define KeGetCurrentThread
Definition: hal.h:55
OBJ_KERNEL_HANDLE
#define OBJ_KERNEL_HANDLE
Definition: winternl.h:231
RtlFillMemory
#define RtlFillMemory(Dest, Length, Fill)
Definition: winternl.h:599
kmt_test.h
KmtGetSystemRoutineAddress
PVOID KmtGetSystemRoutineAddress(IN PCWSTR RoutineName)
Definition: kmt_test_kernel.h:129
KmtIsCheckedBuild
BOOLEAN KmtIsCheckedBuild
Definition: kmt_test_kernel.h:15
InitializeObjectAttributes
#define InitializeObjectAttributes(p, n, a, r, s)
Definition: reg.c:106
_Success_
#define _Success_(expr)
Definition: ms_sal.h:259
_Inout_
#define _Inout_
Definition: ms_sal.h:378
_Must_inspect_result_
#define _Must_inspect_result_
Definition: ms_sal.h:558
_Out_
#define _Out_
Definition: ms_sal.h:345
_When_
#define _When_(expr, annos)
Definition: ms_sal.h:254
_Post_satisfies_
#define _Post_satisfies_(cond)
Definition: ms_sal.h:588
KernelMode
#define KernelMode
Definition: asm.h:34
ZwClose
NTSYSAPI NTSTATUS NTAPI ZwClose(_In_ HANDLE Handle)
SYNCHRONIZE
#define SYNCHRONIZE
Definition: nt_native.h:61
FASTCALL
#define FASTCALL
Definition: nt_native.h:50
GENERIC_ALL
#define GENERIC_ALL
Definition: nt_native.h:92
NotificationEvent
@ NotificationEvent
Definition: ntdef.template.h:356
PsThreadType
POBJECT_TYPE PsThreadType
Definition: thread.c:20
PsCreateSystemThread
NTSTATUS NTAPI PsCreateSystemThread(OUT PHANDLE ThreadHandle, IN ACCESS_MASK DesiredAccess, IN POBJECT_ATTRIBUTES ObjectAttributes, IN HANDLE ProcessHandle, IN PCLIENT_ID ClientId, IN PKSTART_ROUTINE StartRoutine, IN PVOID StartContext)
Definition: thread.c:602
STATUS_TIMEOUT
#define STATUS_TIMEOUT
Definition: ntstatus.h:81
L
#define L(x)
Definition: ntvdm.h:50
ObReferenceObjectByHandle
NTSTATUS NTAPI ObReferenceObjectByHandle(IN HANDLE Handle, IN ACCESS_MASK DesiredAccess, IN POBJECT_TYPE ObjectType, IN KPROCESSOR_MODE AccessMode, OUT PVOID *Object, OUT POBJECT_HANDLE_INFORMATION HandleInformation OPTIONAL)
Definition: obref.c:494
BOOLEAN
#define BOOLEAN
Definition: pedump.c:73
SHORT
short SHORT
Definition: pedump.c:59
Timeout
static ULONG Timeout
Definition: ping.c:61
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
DPRINT
#define DPRINT
Definition: sndvol32.h:71
THREAD_DATA
Definition: AttachThreadInput.c:10
THREAD_DATA::Mutex
PKGUARDED_MUTEX Mutex
Definition: KeGuardedMutex.c:207
THREAD_DATA::Thread
PKTHREAD Thread
Definition: ExFastMutex.c:143
THREAD_DATA::OutEvent
KEVENT OutEvent
Definition: ExFastMutex.c:152
ThreadData::Handle
HANDLE Handle
Definition: drwtsn32.h:25
_KEVENT
Definition: ketypes.h:811
_KGUARDED_MUTEX
Definition: ketypes.h:826
_KTHREAD
Definition: ketypes.h:1660
_OBJECT_ATTRIBUTES
Definition: umtypes.h:182
NTAPI
#define NTAPI
Definition: typedefs.h:36
_LARGE_INTEGER
Definition: typedefs.h:103
Attributes
_Must_inspect_result_ _In_ WDFDMAENABLER _In_ _In_opt_ PWDF_OBJECT_ATTRIBUTES Attributes
Definition: wdfcommonbuffer.h:97
OldIrql
_Requires_lock_held_ Interrupt _Releases_lock_ Interrupt _In_ _IRQL_restores_ KIRQL OldIrql
Definition: kefuncs.h:778
Executive
@ Executive
Definition: ketypes.h:415
PKGUARDED_MUTEX
struct _KGUARDED_MUTEX * PKGUARDED_MUTEX
ObDereferenceObject
#define ObDereferenceObject
Definition: obfuncs.h:203
Parameter
_Inout_opt_ PVOID Parameter
Definition: rtltypes.h:323