[WINE]
[reactos.git] / reactos / include / reactos / wine / test.h
1 /*
2 * Definitions for Wine C unit tests.
3 *
4 * Copyright (C) 2002 Alexandre Julliard
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 #ifndef __WINE_WINE_TEST_H
22 #define __WINE_WINE_TEST_H
23
24 #include <stdarg.h>
25 #include <stdlib.h>
26 #include <windef.h>
27 #include <winbase.h>
28
29 #ifdef __WINE_CONFIG_H
30 #error config.h should not be used in Wine tests
31 #endif
32 #ifdef __WINE_WINE_LIBRARY_H
33 #error wine/library.h should not be used in Wine tests
34 #endif
35 #ifdef __WINE_WINE_UNICODE_H
36 #error wine/unicode.h should not be used in Wine tests
37 #endif
38 #ifdef __WINE_WINE_DEBUG_H
39 #error wine/debug.h should not be used in Wine tests
40 #endif
41
42 #ifdef __cplusplus
43 extern "C" {
44 #endif
45
46 #ifndef INVALID_FILE_ATTRIBUTES
47 #define INVALID_FILE_ATTRIBUTES (~0u)
48 #endif
49 #ifndef INVALID_SET_FILE_POINTER
50 #define INVALID_SET_FILE_POINTER (~0u)
51 #endif
52
53 /* debug level */
54 extern int winetest_debug;
55
56 /* running in interactive mode? */
57 extern int winetest_interactive;
58
59 /* current platform */
60 extern const char *winetest_platform;
61
62 extern void winetest_set_location( const char* file, int line );
63 extern void winetest_start_todo( const char* platform );
64 extern int winetest_loop_todo(void);
65 extern void winetest_end_todo( const char* platform );
66 extern int winetest_get_mainargs( char*** pargv );
67 extern LONG winetest_get_failures(void);
68 extern void winetest_add_failures( LONG new_failures );
69 extern void winetest_wait_child_process( HANDLE process );
70
71 extern const char *wine_dbgstr_wn( const WCHAR *str, intptr_t n );
72 extern const char *wine_dbgstr_guid( const GUID *guid );
73 static inline const char *wine_dbgstr_w( const WCHAR *s ) { return wine_dbgstr_wn( s, -1 ); }
74
75 /* strcmpW is available for tests compiled under Wine, but not in standalone
76 * builds under Windows, so we reimplement it under a different name. */
77 static inline int winetest_strcmpW( const WCHAR *str1, const WCHAR *str2 )
78 {
79 while (*str1 && (*str1 == *str2)) { str1++; str2++; }
80 return *str1 - *str2;
81 }
82
83 #ifdef STANDALONE
84
85 #define START_TEST(name) \
86 static void func_##name(void); \
87 const struct test winetest_testlist[] = { { #name, func_##name }, { 0, 0 } }; \
88 static void func_##name(void)
89
90 #else /* STANDALONE */
91
92 #ifdef __cplusplus
93 #define START_TEST(name) extern "C" void func_##name(void)
94 #else
95 #define START_TEST(name) void func_##name(void)
96 #endif
97
98 #endif /* STANDALONE */
99
100 #if defined(__x86_64__) && defined(__GNUC__) && defined(__WINE_USE_MSVCRT)
101 #define __winetest_cdecl __cdecl
102 #define __winetest_va_list __builtin_ms_va_list
103 #else
104 #define __winetest_cdecl
105 #define __winetest_va_list va_list
106 #endif
107
108 extern int broken( int condition );
109 extern int winetest_vok( int condition, const char *msg, __winetest_va_list ap );
110 extern void winetest_vskip( const char *msg, __winetest_va_list ap );
111
112 #ifdef __GNUC__
113
114 extern void __winetest_cdecl winetest_ok( int condition, const char *msg, ... ) __attribute__((format (printf,2,3) ));
115 extern void __winetest_cdecl winetest_skip( const char *msg, ... ) __attribute__((format (printf,1,2)));
116 extern void __winetest_cdecl winetest_win_skip( const char *msg, ... ) __attribute__((format (printf,1,2)));
117 extern void __winetest_cdecl winetest_trace( const char *msg, ... ) __attribute__((format (printf,1,2)));
118
119 #else /* __GNUC__ */
120
121 extern void __winetest_cdecl winetest_ok( int condition, const char *msg, ... );
122 extern void __winetest_cdecl winetest_skip( const char *msg, ... );
123 extern void __winetest_cdecl winetest_win_skip( const char *msg, ... );
124 extern void __winetest_cdecl winetest_trace( const char *msg, ... );
125
126 #endif /* __GNUC__ */
127
128 #define ok_(file, line) (winetest_set_location(file, line), 0) ? (void)0 : winetest_ok
129 #define skip_(file, line) (winetest_set_location(file, line), 0) ? (void)0 : winetest_skip
130 #define win_skip_(file, line) (winetest_set_location(file, line), 0) ? (void)0 : winetest_win_skip
131 #define trace_(file, line) (winetest_set_location(file, line), 0) ? (void)0 : winetest_trace
132
133 #define ok ok_(__FILE__, __LINE__)
134 #define skip skip_(__FILE__, __LINE__)
135 #define win_skip win_skip_(__FILE__, __LINE__)
136 #define trace trace_(__FILE__, __LINE__)
137
138 #define todo(platform) for (winetest_start_todo(platform); \
139 winetest_loop_todo(); \
140 winetest_end_todo(platform))
141
142 #define todo_ros todo("reactos")
143 #ifdef USE_WINE_TODOS
144 #define todo_wine todo_ros
145 #else
146 #define todo_wine todo("wine")
147 #endif
148
149
150 #ifdef NONAMELESSUNION
151 # define U(x) (x).u
152 # define U1(x) (x).u1
153 # define U2(x) (x).u2
154 # define U3(x) (x).u3
155 # define U4(x) (x).u4
156 # define U5(x) (x).u5
157 # define U6(x) (x).u6
158 # define U7(x) (x).u7
159 # define U8(x) (x).u8
160 #else
161 # define U(x) (x)
162 # define U1(x) (x)
163 # define U2(x) (x)
164 # define U3(x) (x)
165 # define U4(x) (x)
166 # define U5(x) (x)
167 # define U6(x) (x)
168 # define U7(x) (x)
169 # define U8(x) (x)
170 #endif
171
172 #ifdef NONAMELESSSTRUCT
173 # define S(x) (x).s
174 # define S1(x) (x).s1
175 # define S2(x) (x).s2
176 # define S3(x) (x).s3
177 # define S4(x) (x).s4
178 # define S5(x) (x).s5
179 #else
180 # define S(x) (x)
181 # define S1(x) (x)
182 # define S2(x) (x)
183 # define S3(x) (x)
184 # define S4(x) (x)
185 # define S5(x) (x)
186 #endif
187
188
189 /************************************************************************/
190 /* Below is the implementation of the various functions, to be included
191 * directly into the generated testlist.c file.
192 * It is done that way so that the dlls can build the test routines with
193 * different includes or flags if needed.
194 */
195
196 #ifdef STANDALONE
197
198 #include <stdio.h>
199
200 #if defined(__x86_64__) && defined(__GNUC__) && defined(__WINE_USE_MSVCRT)
201 # define __winetest_va_start(list,arg) __builtin_ms_va_start(list,arg)
202 # define __winetest_va_end(list) __builtin_ms_va_end(list)
203 #else
204 # define __winetest_va_start(list,arg) va_start(list,arg)
205 # define __winetest_va_end(list) va_end(list)
206 #endif
207
208 struct test
209 {
210 const char *name;
211 void (*func)(void);
212 };
213
214 extern const struct test winetest_testlist[];
215
216 /* debug level */
217 int winetest_debug = 1;
218
219 /* interactive mode? */
220 int winetest_interactive = 0;
221
222 /* current platform */
223 const char *winetest_platform = "windows";
224
225 /* report successful tests (BOOL) */
226 static int report_success = 0;
227
228 /* passing arguments around */
229 static int winetest_argc;
230 static char** winetest_argv;
231
232 static const struct test *current_test; /* test currently being run */
233
234 static LONG successes; /* number of successful tests */
235 static LONG failures; /* number of failures */
236 static LONG skipped; /* number of skipped test chunks */
237 static LONG todo_successes; /* number of successful tests inside todo block */
238 static LONG todo_failures; /* number of failures inside todo block */
239
240 /* The following data must be kept track of on a per-thread basis */
241 typedef struct
242 {
243 const char* current_file; /* file of current check */
244 int current_line; /* line of current check */
245 int todo_level; /* current todo nesting level */
246 int todo_do_loop;
247 char *str_pos; /* position in debug buffer */
248 char strings[2000]; /* buffer for debug strings */
249 } tls_data;
250 static DWORD tls_index;
251
252 static tls_data* get_tls_data(void)
253 {
254 tls_data* data;
255 DWORD last_error;
256
257 last_error=GetLastError();
258 data=(tls_data*)TlsGetValue(tls_index);
259 if (!data)
260 {
261 data=(tls_data*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(tls_data));
262 data->str_pos = data->strings;
263 TlsSetValue(tls_index,data);
264 }
265 SetLastError(last_error);
266 return data;
267 }
268
269 /* allocate some tmp space for a string */
270 static char *get_temp_buffer( size_t n )
271 {
272 tls_data *data = get_tls_data();
273 char *res = data->str_pos;
274
275 if (res + n >= &data->strings[sizeof(data->strings)]) res = data->strings;
276 data->str_pos = res + n;
277 return res;
278 }
279
280 /* release extra space that we requested in gimme1() */
281 static void release_temp_buffer( char *ptr, size_t size )
282 {
283 tls_data *data = get_tls_data();
284 data->str_pos = ptr + size;
285 }
286
287 static void exit_process( int code )
288 {
289 fflush( stdout );
290 ExitProcess( code );
291 }
292
293
294 void winetest_set_location( const char* file, int line )
295 {
296 tls_data* data=get_tls_data();
297 data->current_file=strrchr(file,'/');
298 if (data->current_file==NULL)
299 data->current_file=strrchr(file,'\\');
300 if (data->current_file==NULL)
301 data->current_file=file;
302 else
303 data->current_file++;
304 data->current_line=line;
305 }
306
307 int broken( int condition )
308 {
309 return (strcmp(winetest_platform, "windows") == 0) && condition;
310 }
311
312 /*
313 * Checks condition.
314 * Parameters:
315 * - condition - condition to check;
316 * - msg test description;
317 * - file - test application source code file name of the check
318 * - line - test application source code file line number of the check
319 * Return:
320 * 0 if condition does not have the expected value, 1 otherwise
321 */
322 int winetest_vok( int condition, const char *msg, __winetest_va_list args )
323 {
324 tls_data* data=get_tls_data();
325
326 if (data->todo_level)
327 {
328 if (condition)
329 {
330 fprintf( stdout, "%s:%d: Test succeeded inside todo block: ",
331 data->current_file, data->current_line );
332 vfprintf(stdout, msg, args);
333 InterlockedIncrement(&todo_failures);
334 return 0;
335 }
336 else
337 {
338 /* show todos even if traces are disabled*/
339 /*if (winetest_debug > 0)*/
340 {
341 fprintf( stdout, "%s:%d: Test marked todo: ",
342 data->current_file, data->current_line );
343 vfprintf(stdout, msg, args);
344 }
345 InterlockedIncrement(&todo_successes);
346 return 1;
347 }
348 }
349 else
350 {
351 if (!condition)
352 {
353 fprintf( stdout, "%s:%d: Test failed: ",
354 data->current_file, data->current_line );
355 vfprintf(stdout, msg, args);
356 InterlockedIncrement(&failures);
357 return 0;
358 }
359 else
360 {
361 if (report_success)
362 fprintf( stdout, "%s:%d: Test succeeded\n",
363 data->current_file, data->current_line);
364 InterlockedIncrement(&successes);
365 return 1;
366 }
367 }
368 }
369
370 void __winetest_cdecl winetest_ok( int condition, const char *msg, ... )
371 {
372 __winetest_va_list valist;
373
374 __winetest_va_start(valist, msg);
375 winetest_vok(condition, msg, valist);
376 __winetest_va_end(valist);
377 }
378
379 void __winetest_cdecl winetest_trace( const char *msg, ... )
380 {
381 __winetest_va_list valist;
382 tls_data* data=get_tls_data();
383
384 if (winetest_debug > 0)
385 {
386 fprintf( stdout, "%s:%d: ", data->current_file, data->current_line );
387 __winetest_va_start(valist, msg);
388 vfprintf(stdout, msg, valist);
389 __winetest_va_end(valist);
390 }
391 }
392
393 void winetest_vskip( const char *msg, __winetest_va_list args )
394 {
395 tls_data* data=get_tls_data();
396
397 fprintf( stdout, "%s:%d: Tests skipped: ", data->current_file, data->current_line );
398 vfprintf(stdout, msg, args);
399 skipped++;
400 }
401
402 void __winetest_cdecl winetest_skip( const char *msg, ... )
403 {
404 __winetest_va_list valist;
405 __winetest_va_start(valist, msg);
406 winetest_vskip(msg, valist);
407 __winetest_va_end(valist);
408 }
409
410 void __winetest_cdecl winetest_win_skip( const char *msg, ... )
411 {
412 __winetest_va_list valist;
413 __winetest_va_start(valist, msg);
414 if ((strcmp(winetest_platform, "windows") == 0) || (strcmp(winetest_platform, "reactos") == 0))
415 winetest_vskip(msg, valist);
416 else
417 winetest_vok(0, msg, valist);
418 __winetest_va_end(valist);
419 }
420
421 void winetest_start_todo( const char* platform )
422 {
423 tls_data* data=get_tls_data();
424 if (strcmp(winetest_platform,platform)==0)
425 data->todo_level++;
426 data->todo_do_loop=1;
427 }
428
429 int winetest_loop_todo(void)
430 {
431 tls_data* data=get_tls_data();
432 int do_loop=data->todo_do_loop;
433 data->todo_do_loop=0;
434 return do_loop;
435 }
436
437 void winetest_end_todo( const char* platform )
438 {
439 if (strcmp(winetest_platform,platform)==0)
440 {
441 tls_data* data=get_tls_data();
442 data->todo_level--;
443 }
444 }
445
446 int winetest_get_mainargs( char*** pargv )
447 {
448 *pargv = winetest_argv;
449 return winetest_argc;
450 }
451
452 LONG winetest_get_failures(void)
453 {
454 return failures;
455 }
456
457 void winetest_add_failures( LONG new_failures )
458 {
459 while (new_failures-- > 0)
460 InterlockedIncrement( &failures );
461 }
462
463 void winetest_wait_child_process( HANDLE process )
464 {
465 DWORD exit_code = 1;
466
467 if (WaitForSingleObject( process, 30000 ))
468 fprintf( stdout, "%s: child process wait failed\n", current_test->name );
469 else
470 GetExitCodeProcess( process, &exit_code );
471
472 if (exit_code)
473 {
474 if (exit_code > 255)
475 {
476 fprintf( stdout, "%s: exception 0x%08x in child process\n", current_test->name, exit_code );
477 InterlockedIncrement( &failures );
478 }
479 else
480 {
481 fprintf( stdout, "%s: %u failures in child process\n",
482 current_test->name, exit_code );
483 while (exit_code-- > 0)
484 InterlockedIncrement(&failures);
485 }
486 }
487 }
488
489 const char *wine_dbgstr_wn( const WCHAR *str, intptr_t n )
490 {
491 char *dst, *res;
492 size_t size;
493
494 if (!((ULONG_PTR)str >> 16))
495 {
496 if (!str) return "(null)";
497 res = get_temp_buffer( 6 );
498 sprintf( res, "#%04x", LOWORD(str) );
499 return res;
500 }
501 if (n == -1)
502 {
503 const WCHAR *end = str;
504 while (*end) end++;
505 n = end - str;
506 }
507 if (n < 0) n = 0;
508 size = 12 + min( 300, n * 5 );
509 dst = res = get_temp_buffer( size );
510 *dst++ = 'L';
511 *dst++ = '"';
512 while (n-- > 0 && dst <= res + size - 10)
513 {
514 WCHAR c = *str++;
515 switch (c)
516 {
517 case '\n': *dst++ = '\\'; *dst++ = 'n'; break;
518 case '\r': *dst++ = '\\'; *dst++ = 'r'; break;
519 case '\t': *dst++ = '\\'; *dst++ = 't'; break;
520 case '"': *dst++ = '\\'; *dst++ = '"'; break;
521 case '\\': *dst++ = '\\'; *dst++ = '\\'; break;
522 default:
523 if (c >= ' ' && c <= 126)
524 *dst++ = (char)c;
525 else
526 {
527 *dst++ = '\\';
528 sprintf(dst,"%04x",c);
529 dst+=4;
530 }
531 }
532 }
533 *dst++ = '"';
534 if (n > 0)
535 {
536 *dst++ = '.';
537 *dst++ = '.';
538 *dst++ = '.';
539 }
540 *dst++ = 0;
541 release_temp_buffer( res, dst - res );
542 return res;
543 }
544
545 const char *wine_dbgstr_guid( const GUID *guid )
546 {
547 char *res;
548
549 if (!guid) return "(null)";
550 res = get_temp_buffer( 39 ); /* CHARS_IN_GUID */
551 sprintf( res, "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
552 guid->Data1, guid->Data2, guid->Data3, guid->Data4[0],
553 guid->Data4[1], guid->Data4[2], guid->Data4[3], guid->Data4[4],
554 guid->Data4[5], guid->Data4[6], guid->Data4[7] );
555 return res;
556 }
557
558 /* Find a test by name */
559 static const struct test *find_test( const char *name )
560 {
561 const struct test *test;
562 const char *p;
563 size_t len;
564
565 if ((p = strrchr( name, '/' ))) name = p + 1;
566 if ((p = strrchr( name, '\\' ))) name = p + 1;
567 len = strlen(name);
568 if (len > 2 && !strcmp( name + len - 2, ".c" )) len -= 2;
569
570 for (test = winetest_testlist; test->name; test++)
571 {
572 if (!strncmp( test->name, name, len ) && !test->name[len]) break;
573 }
574 return test->name ? test : NULL;
575 }
576
577
578 /* Display list of valid tests */
579 static void list_tests(void)
580 {
581 const struct test *test;
582
583 fprintf( stdout, "Valid test names:\n" );
584 for (test = winetest_testlist; test->name; test++) fprintf( stdout, " %s\n", test->name );
585 }
586
587 /* Disable false-positive claiming "test" would be NULL-dereferenced */
588 #if defined(_MSC_VER)
589 #pragma warning(push)
590 #pragma warning(disable:28182)
591 #endif
592
593 /* Run a named test, and return exit status */
594 static int run_test( const char *name )
595 {
596 const struct test *test;
597 int status;
598
599 if (!(test = find_test( name )))
600 {
601 fprintf( stdout, "Fatal: test '%s' does not exist.\n", name );
602 exit_process(1);
603 }
604 successes = failures = todo_successes = todo_failures = 0;
605 tls_index=TlsAlloc();
606 current_test = test;
607 test->func();
608
609 /* show test results even if traces are disabled */
610 /*if (winetest_debug)*/
611 {
612 fprintf( stdout, "%s: %d tests executed (%d marked as todo, %d %s), %d skipped.\n",
613 test->name, successes + failures + todo_successes + todo_failures,
614 todo_successes, failures + todo_failures,
615 (failures + todo_failures != 1) ? "failures" : "failure",
616 skipped );
617 }
618 status = (failures + todo_failures < 255) ? failures + todo_failures : 255;
619 return status;
620 }
621
622 #if defined(_MSC_VER)
623 #pragma warning(pop)
624 #endif
625
626 /* Display usage and exit */
627 static void usage( const char *argv0 )
628 {
629 fprintf( stdout, "Usage: %s test_name\n\n", argv0 );
630 list_tests();
631 exit_process(1);
632 }
633
634
635 /* main function */
636 int main( int argc, char **argv )
637 {
638 char p[128];
639
640 setvbuf (stdout, NULL, _IONBF, 0);
641
642 winetest_argc = argc;
643 winetest_argv = argv;
644
645 if (GetEnvironmentVariableA( "WINETEST_PLATFORM", p, sizeof(p) )) winetest_platform = _strdup(p);
646 if (GetEnvironmentVariableA( "WINETEST_DEBUG", p, sizeof(p) )) winetest_debug = atoi(p);
647 if (GetEnvironmentVariableA( "WINETEST_INTERACTIVE", p, sizeof(p) )) winetest_interactive = atoi(p);
648 if (GetEnvironmentVariableA( "WINETEST_REPORT_SUCCESS", p, sizeof(p) )) report_success = atoi(p);
649
650 if (!winetest_interactive) SetErrorMode( SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX );
651
652 if (!argv[1])
653 {
654 if (winetest_testlist[0].name && !winetest_testlist[1].name) /* only one test */
655 return run_test( winetest_testlist[0].name );
656 usage( argv[0] );
657 }
658 if (!strcmp( argv[1], "--list" ))
659 {
660 list_tests();
661 return 0;
662 }
663 return run_test(argv[1]);
664 }
665
666 #endif /* STANDALONE */
667
668 // hack for ntdll winetest (this is defined in excpt.h)
669 #undef exception_info
670
671 // Some helpful definitions
672
673 #define ok_hex(expression, result) \
674 do { \
675 int _value = (expression); \
676 ok(_value == (result), "Wrong value for '%s', expected: " #result " (0x%x), got: 0x%x\n", \
677 #expression, (int)(result), _value); \
678 } while (0)
679
680 #define ok_dec(expression, result) \
681 do { \
682 int _value = (expression); \
683 ok(_value == (result), "Wrong value for '%s', expected: " #result " (%d), got: %d\n", \
684 #expression, (int)(result), _value); \
685 } while (0)
686
687 #define ok_ptr(expression, result) \
688 do { \
689 void *_value = (expression); \
690 ok(_value == (result), "Wrong value for '%s', expected: " #result " (%p), got: %p\n", \
691 #expression, (void*)(result), _value); \
692 } while (0)
693
694 #define ok_size_t(expression, result) \
695 do { \
696 size_t _value = (expression); \
697 ok(_value == (result), "Wrong value for '%s', expected: " #result " (%Ix), got: %Ix\n", \
698 #expression, (size_t)(result), _value); \
699 } while (0)
700
701 #define ok_char(expression, result) ok_hex(expression, result)
702
703 #define ok_err(error) \
704 ok(GetLastError() == (error), "Wrong last error. Expected " #error ", got 0x%lx\n", GetLastError())
705
706 #define ok_str(x, y) \
707 ok(strcmp(x, y) == 0, "Wrong string. Expected '%s', got '%s'\n", y, x)
708
709 #define ok_wstr(x, y) \
710 ok(wcscmp(x, y) == 0, "Wrong string. Expected '%S', got '%S'\n", y, x)
711
712 #define ok_long(expression, result) ok_hex(expression, result)
713 #define ok_int(expression, result) ok_dec(expression, result)
714 #define ok_ntstatus(status, expected) ok_hex(status, expected)
715 #define ok_hdl ok_ptr
716
717 #ifdef __cplusplus
718 } /* extern "C" */
719 #endif
720
721 #endif /* __WINE_WINE_TEST_H */