2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: Kernel-Mode Test Suite test framework declarations
5 * PROGRAMMER: Thomas Faber <thfabba@gmx.de>
8 /* Inspired by Wine C unit tests, Copyright (C) 2002 Alexandre Julliard
9 * Inspired by ReactOS kernel-mode regression tests,
10 * Copyright (C) Aleksey Bragin, Filip Navara
13 #ifndef _KMTEST_TEST_H_
14 #define _KMTEST_TEST_H_
18 typedef VOID
KMT_TESTFUNC(VOID
);
19 typedef KMT_TESTFUNC
*PKMT_TESTFUNC
;
24 KMT_TESTFUNC
*TestFunction
;
25 } KMT_TEST
, *PKMT_TEST
;
27 typedef const KMT_TEST CKMT_TEST
, *PCKMT_TEST
;
29 extern const KMT_TEST TestList
[];
33 volatile LONG Successes
;
34 volatile LONG Failures
;
35 volatile LONG Skipped
;
36 volatile LONG LogBufferLength
;
37 LONG LogBufferMaxLength
;
38 CHAR LogBuffer
[ANYSIZE_ARRAY
];
39 } KMT_RESULTBUFFER
, *PKMT_RESULTBUFFER
;
41 #ifdef KMT_STANDALONE_DRIVER
42 #define KMT_KERNEL_MODE
44 typedef NTSTATUS (KMT_IRP_HANDLER
)(
45 IN PDEVICE_OBJECT DeviceObject
,
47 IN PIO_STACK_LOCATION IoStackLocation
);
48 typedef KMT_IRP_HANDLER
*PKMT_IRP_HANDLER
;
50 NTSTATUS
KmtRegisterIrpHandler(IN UCHAR MajorFunction
, IN PDEVICE_OBJECT DeviceObject OPTIONAL
, IN PKMT_IRP_HANDLER IrpHandler
);
51 NTSTATUS
KmtUnregisterIrpHandler(IN UCHAR MajorFunction
, IN PDEVICE_OBJECT DeviceObject OPTIONAL
, IN PKMT_IRP_HANDLER IrpHandler
);
53 typedef NTSTATUS (KMT_MESSAGE_HANDLER
)(
54 IN PDEVICE_OBJECT DeviceObject
,
56 IN PVOID Buffer OPTIONAL
,
58 IN OUT PSIZE_T OutLength
);
59 typedef KMT_MESSAGE_HANDLER
*PKMT_MESSAGE_HANDLER
;
61 NTSTATUS
KmtRegisterMessageHandler(IN ULONG ControlCode OPTIONAL
, IN PDEVICE_OBJECT DeviceObject OPTIONAL
, IN PKMT_MESSAGE_HANDLER MessageHandler
);
62 NTSTATUS
KmtUnregisterMessageHandler(IN ULONG ControlCode OPTIONAL
, IN PDEVICE_OBJECT DeviceObject OPTIONAL
, IN PKMT_MESSAGE_HANDLER MessageHandler
);
66 TESTENTRY_NO_CREATE_DEVICE
= 1,
67 TESTENTRY_NO_REGISTER_DISPATCH
= 2,
68 TESTENTRY_NO_REGISTER_UNLOAD
= 4,
69 } KMT_TESTENTRY_FLAGS
;
71 NTSTATUS
TestEntry(IN PDRIVER_OBJECT DriverObject
, IN PCUNICODE_STRING RegistryPath
, OUT PCWSTR
*DeviceName
, OUT INT
*Flags
);
72 VOID
TestUnload(IN PDRIVER_OBJECT DriverObject
);
73 #endif /* defined KMT_STANDALONE_DRIVER */
75 #ifdef KMT_KERNEL_MODE
76 /* Device Extension layout */
79 PKMT_RESULTBUFFER ResultBuffer
;
81 } KMT_DEVICE_EXTENSION
, *PKMT_DEVICE_EXTENSION
;
82 #elif defined KMT_USER_MODE
83 DWORD
KmtRunKernelTest(IN PCSTR TestName
);
85 VOID
KmtLoadDriver(IN PCWSTR ServiceName
, IN BOOLEAN RestartIfRunning
);
86 VOID
KmtUnloadDriver(VOID
);
87 VOID
KmtOpenDriver(VOID
);
88 VOID
KmtCloseDriver(VOID
);
90 DWORD
KmtSendToDriver(IN DWORD ControlCode
);
91 DWORD
KmtSendStringToDriver(IN DWORD ControlCode
, IN PCSTR String
);
92 DWORD
KmtSendBufferToDriver(IN DWORD ControlCode
, IN OUT PVOID Buffer OPTIONAL
, IN DWORD InLength
, IN OUT PDWORD OutLength
);
93 #endif /* defined KMT_USER_MODE */
95 extern PKMT_RESULTBUFFER ResultBuffer
;
98 #define KMT_FORMAT(type, fmt, first) __attribute__((__format__(type, fmt, first)))
99 #elif !defined __GNUC__
100 #define KMT_FORMAT(type, fmt, first)
101 #endif /* !defined __GNUC__ */
103 #define START_TEST(name) VOID Test_##name(VOID)
105 #ifndef KMT_STRINGIZE
106 #define KMT_STRINGIZE(x) #x
107 #endif /* !defined KMT_STRINGIZE */
108 #define ok(test, ...) ok_(test, __FILE__, __LINE__, __VA_ARGS__)
109 #define trace(...) trace_( __FILE__, __LINE__, __VA_ARGS__)
110 #define skip(test, ...) skip_(test, __FILE__, __LINE__, __VA_ARGS__)
112 #define ok_(test, file, line, ...) KmtOk(test, file ":" KMT_STRINGIZE(line), __VA_ARGS__)
113 #define trace_(file, line, ...) KmtTrace( file ":" KMT_STRINGIZE(line), __VA_ARGS__)
114 #define skip_(test, file, line, ...) KmtSkip(test, file ":" KMT_STRINGIZE(line), __VA_ARGS__)
116 VOID
KmtVOk(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, va_list Arguments
) KMT_FORMAT(ms_printf
, 3, 0);
117 VOID
KmtOk(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, ...) KMT_FORMAT(ms_printf
, 3, 4);
118 VOID
KmtVTrace(PCSTR FileAndLine
, PCSTR Format
, va_list Arguments
) KMT_FORMAT(ms_printf
, 2, 0);
119 VOID
KmtTrace(PCSTR FileAndLine
, PCSTR Format
, ...) KMT_FORMAT(ms_printf
, 2, 3);
120 BOOLEAN
KmtVSkip(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, va_list Arguments
) KMT_FORMAT(ms_printf
, 3, 0);
121 BOOLEAN
KmtSkip(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, ...) KMT_FORMAT(ms_printf
, 3, 4);
123 #ifdef KMT_KERNEL_MODE
124 #define ok_irql(irql) ok(KeGetCurrentIrql() == irql, "IRQL is %d, expected %d\n", KeGetCurrentIrql(), irql)
125 #endif /* defined KMT_KERNEL_MODE */
126 #define ok_eq_print(value, expected, spec) ok((value) == (expected), #value " = " spec ", expected " spec "\n", value, expected)
127 #define ok_eq_pointer(value, expected) ok_eq_print(value, expected, "%p")
128 #define ok_eq_int(value, expected) ok_eq_print(value, expected, "%d")
129 #define ok_eq_uint(value, expected) ok_eq_print(value, expected, "%u")
130 #define ok_eq_long(value, expected) ok_eq_print(value, expected, "%ld")
131 #define ok_eq_ulong(value, expected) ok_eq_print(value, expected, "%lu")
132 #define ok_eq_hex(value, expected) ok_eq_print(value, expected, "0x%08lx")
133 #define ok_bool_true(value, desc) ok((value) == TRUE, desc " FALSE, expected TRUE\n")
134 #define ok_bool_false(value, desc) ok((value) == FALSE, desc " TRUE, expected FALSE\n")
135 #define ok_eq_bool(value, expected) ok((value) == (expected), #value " = %s, expected %s\n", \
136 (value) ? "TRUE" : "FALSE", \
137 (expected) ? "TRUE" : "FALSE")
138 #define ok_eq_str(value, expected) ok(!strcmp(value, expected), #value " = \"%s\", expected \"%s\"\n", value, expected)
139 #define ok_eq_wstr(value, expected) ok(!wcscmp(value, expected), #value " = \"%ls\", expected \"%ls\"\n", value, expected)
141 #define KMT_MAKE_CODE(ControlCode) CTL_CODE(FILE_DEVICE_UNKNOWN, \
142 0xC00 + (ControlCode), \
146 #if defined KMT_DEFINE_TEST_FUNCTIONS
147 PKMT_RESULTBUFFER ResultBuffer
= NULL
;
149 #if defined KMT_USER_MODE
150 static PKMT_RESULTBUFFER
KmtAllocateResultBuffer(SIZE_T LogBufferMaxLength
)
152 PKMT_RESULTBUFFER Buffer
= HeapAlloc(GetProcessHeap(), 0, FIELD_OFFSET(KMT_RESULTBUFFER
, LogBuffer
[LogBufferMaxLength
]));
154 Buffer
->Successes
= 0;
155 Buffer
->Failures
= 0;
157 Buffer
->LogBufferLength
= 0;
158 Buffer
->LogBufferMaxLength
= LogBufferMaxLength
;
163 static VOID
KmtFreeResultBuffer(PKMT_RESULTBUFFER Buffer
)
165 HeapFree(GetProcessHeap(), 0, Buffer
);
167 #endif /* defined KMT_USER_MODE */
169 static VOID
KmtAddToLogBuffer(PKMT_RESULTBUFFER Buffer
, PCSTR String
, SIZE_T Length
)
179 OldLength
= Buffer
->LogBufferLength
;
180 NewLength
= OldLength
+ Length
;
181 if (NewLength
> Buffer
->LogBufferMaxLength
)
183 /* TODO: indicate failure somehow */
187 } while (InterlockedCompareExchange(&Buffer
->LogBufferLength
, NewLength
, OldLength
) != OldLength
);
189 memcpy(&Buffer
->LogBuffer
[OldLength
], String
, Length
);
192 #ifdef KMT_KERNEL_MODE
193 INT __cdecl
KmtVSNPrintF(PSTR Buffer
, SIZE_T BufferMaxLength
, PCSTR Format
, va_list Arguments
) KMT_FORMAT(ms_printf
, 3, 0);
194 #elif defined KMT_USER_MODE
195 #define KmtVSNPrintF vsnprintf
196 #endif /* defined KMT_USER_MODE */
198 KMT_FORMAT(ms_printf
, 5, 0)
199 static SIZE_T
KmtXVSNPrintF(PSTR Buffer
, SIZE_T BufferMaxLength
, PCSTR FileAndLine
, PCSTR Prepend
, PCSTR Format
, va_list Arguments
)
201 SIZE_T BufferLength
= 0;
207 Slash
= strrchr(FileAndLine
, '\\');
209 FileAndLine
= Slash
+ 1;
210 Slash
= strrchr(FileAndLine
, '/');
212 FileAndLine
= Slash
+ 1;
214 Length
= min(BufferMaxLength
, strlen(FileAndLine
));
215 memcpy(Buffer
, FileAndLine
, Length
);
217 BufferLength
+= Length
;
218 BufferMaxLength
-= Length
;
222 Length
= min(BufferMaxLength
, strlen(Prepend
));
223 memcpy(Buffer
, Prepend
, Length
);
225 BufferLength
+= Length
;
226 BufferMaxLength
-= Length
;
230 Length
= KmtVSNPrintF(Buffer
, BufferMaxLength
, Format
, Arguments
);
231 /* vsnprintf can return more than maxLength, we don't want to do that */
232 BufferLength
+= min(Length
, BufferMaxLength
);
237 KMT_FORMAT(ms_printf
, 5, 6)
238 static SIZE_T
KmtXSNPrintF(PSTR Buffer
, SIZE_T BufferMaxLength
, PCSTR FileAndLine
, PCSTR Prepend
, PCSTR Format
, ...)
242 va_start(Arguments
, Format
);
243 BufferLength
= KmtXVSNPrintF(Buffer
, BufferMaxLength
, FileAndLine
, Prepend
, Format
, Arguments
);
248 VOID
KmtFinishTest(PCSTR TestName
)
250 CHAR MessageBuffer
[512];
251 SIZE_T MessageLength
;
256 MessageLength
= KmtXSNPrintF(MessageBuffer
, sizeof MessageBuffer
, NULL
, NULL
,
257 "%s: %ld tests executed (0 marked as todo, %ld failures), %ld skipped.\n",
259 ResultBuffer
->Successes
+ ResultBuffer
->Failures
,
260 ResultBuffer
->Failures
,
261 ResultBuffer
->Skipped
);
262 KmtAddToLogBuffer(ResultBuffer
, MessageBuffer
, MessageLength
);
265 VOID
KmtVOk(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, va_list Arguments
)
267 CHAR MessageBuffer
[512];
268 SIZE_T MessageLength
;
275 InterlockedIncrement(&ResultBuffer
->Successes
);
277 if (0/*KmtReportSuccess*/)
279 MessageLength
= KmtXSNPrintF(MessageBuffer
, sizeof MessageBuffer
, FileAndLine
, ": Test succeeded\n", NULL
);
280 KmtAddToLogBuffer(ResultBuffer
, MessageBuffer
, MessageLength
);
285 InterlockedIncrement(&ResultBuffer
->Failures
);
286 MessageLength
= KmtXVSNPrintF(MessageBuffer
, sizeof MessageBuffer
, FileAndLine
, ": Test failed: ", Format
, Arguments
);
287 KmtAddToLogBuffer(ResultBuffer
, MessageBuffer
, MessageLength
);
291 VOID
KmtOk(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, ...)
294 va_start(Arguments
, Format
);
295 KmtVOk(Condition
, FileAndLine
, Format
, Arguments
);
299 VOID
KmtVTrace(PCSTR FileAndLine
, PCSTR Format
, va_list Arguments
)
301 CHAR MessageBuffer
[512];
302 SIZE_T MessageLength
;
304 MessageLength
= KmtXVSNPrintF(MessageBuffer
, sizeof MessageBuffer
, FileAndLine
, ": ", Format
, Arguments
);
305 KmtAddToLogBuffer(ResultBuffer
, MessageBuffer
, MessageLength
);
308 VOID
KmtTrace(PCSTR FileAndLine
, PCSTR Format
, ...)
311 va_start(Arguments
, Format
);
312 KmtVTrace(FileAndLine
, Format
, Arguments
);
316 BOOLEAN
KmtVSkip(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, va_list Arguments
)
318 CHAR MessageBuffer
[512];
319 SIZE_T MessageLength
;
326 InterlockedIncrement(&ResultBuffer
->Skipped
);
327 MessageLength
= KmtXVSNPrintF(MessageBuffer
, sizeof MessageBuffer
, FileAndLine
, ": Tests skipped: ", Format
, Arguments
);
328 KmtAddToLogBuffer(ResultBuffer
, MessageBuffer
, MessageLength
);
334 BOOLEAN
KmtSkip(INT Condition
, PCSTR FileAndLine
, PCSTR Format
, ...)
338 va_start(Arguments
, Format
);
339 Ret
= KmtVSkip(Condition
, FileAndLine
, Format
, Arguments
);
344 #endif /* defined KMT_DEFINE_TEST_FUNCTIONS */
346 #endif /* !defined _KMTEST_TEST_H_ */