0975e2b779c3a0fd7ac077ed54429f891755f10d
[openssl.git] / test / testutil.h
1 /*
2  * Copyright 2014-2016 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #ifndef HEADER_TESTUTIL_H
11 # define HEADER_TESTUTIL_H
12
13 #include <stdarg.h>
14
15 #include <openssl/err.h>
16 #include <openssl/e_os2.h>
17 #include <openssl/bn.h>
18
19 /*-
20  * Simple unit tests should implement register_tests().
21  * To register tests, call ADD_TEST or ADD_ALL_TESTS:
22  *
23  * void register_tests(void)
24  * {
25  *     ADD_TEST(test_foo);
26  *     ADD_ALL_TESTS(test_bar, num_test_bar);
27  * }
28  *
29  * Tests that need to perform custom setup or read command-line arguments should
30  * implement test_main():
31  *
32  * int test_main(int argc, char *argv[])
33  * {
34  *     int ret;
35  *
36  *     // Custom setup ...
37  *
38  *     ADD_TEST(test_foo);
39  *     ADD_ALL_TESTS(test_bar, num_test_bar);
40  *     // Add more tests ...
41  *
42  *     ret = run_tests(argv[0]);
43  *
44  *     // Custom teardown ...
45  *
46  *     return ret;
47  * }
48  */
49
50 /* Adds a simple test case. */
51 # define ADD_TEST(test_function) add_test(#test_function, test_function)
52
53 /*
54  * Simple parameterized tests. Calls test_function(idx) for each 0 <= idx < num.
55  */
56 # define ADD_ALL_TESTS(test_function, num) \
57     add_all_tests(#test_function, test_function, num, 1)
58 /*
59  * A variant of the same without TAP output.
60  */
61 # define ADD_ALL_TESTS_NOSUBTEST(test_function, num) \
62     add_all_tests(#test_function, test_function, num, 0)
63
64 /*-
65  * Test cases that share common setup should use the helper
66  * SETUP_TEST_FIXTURE and EXECUTE_TEST macros for test case functions.
67  *
68  * SETUP_TEST_FIXTURE will call set_up() to create a new TEST_FIXTURE_TYPE
69  * object called "fixture". It will also allocate the "result" variable used
70  * by EXECUTE_TEST. set_up() should take a const char* specifying the test
71  * case name and return a TEST_FIXTURE_TYPE by value.
72  *
73  * EXECUTE_TEST will pass fixture to execute_func() by value, call
74  * tear_down(), and return the result of execute_func(). execute_func() should
75  * take a TEST_FIXTURE_TYPE by value and return 1 on success and 0 on
76  * failure.
77  *
78  * Unit tests can define their own SETUP_TEST_FIXTURE and EXECUTE_TEST
79  * variations like so:
80  *
81  * #define SETUP_FOOBAR_TEST_FIXTURE()\
82  *   SETUP_TEST_FIXTURE(FOOBAR_TEST_FIXTURE, set_up_foobar)
83  *
84  * #define EXECUTE_FOOBAR_TEST()\
85  *   EXECUTE_TEST(execute_foobar, tear_down_foobar)
86  *
87  * Then test case functions can take the form:
88  *
89  * static int test_foobar_feature()
90  *      {
91  *      SETUP_FOOBAR_TEST_FIXTURE();
92  *      [...set individual members of fixture...]
93  *      EXECUTE_FOOBAR_TEST();
94  *      }
95  */
96 # define SETUP_TEST_FIXTURE(TEST_FIXTURE_TYPE, set_up)\
97     TEST_FIXTURE_TYPE fixture = set_up(TEST_CASE_NAME); \
98     int result = 0
99
100 # define EXECUTE_TEST(execute_func, tear_down)\
101         result = execute_func(fixture);\
102         tear_down(fixture);\
103         return result
104
105 /* Shorthand if tear_down does nothing. */
106 # define EXECUTE_TEST_NO_TEARDOWN(execute_func)\
107         result = execute_func(fixture);\
108         return result
109
110 /*
111  * TEST_CASE_NAME is defined as the name of the test case function where
112  * possible; otherwise we get by with the file name and line number.
113  */
114 # if !defined(__STDC_VERSION__) || __STDC_VERSION__ < 199901L
115 #  if defined(_MSC_VER)
116 #   define TEST_CASE_NAME __FUNCTION__
117 #  else
118 #   define testutil_stringify_helper(s) #s
119 #   define testutil_stringify(s) testutil_stringify_helper(s)
120 #   define TEST_CASE_NAME __FILE__ ":" testutil_stringify(__LINE__)
121 #  endif                        /* _MSC_VER */
122 # else
123 #  define TEST_CASE_NAME __func__
124 # endif                         /* __STDC_VERSION__ */
125
126 /*
127  * Internal helpers. Test programs shouldn't use these directly, but should
128  * rather link to one of the helper main() methods.
129  */
130
131 /* setup_test() should be called as the first thing in a test main(). */
132 void setup_test(void);
133 /*
134  * finish_test() should be called as the last thing in a test main().
135  * The result of run_tests() should be the input to finish_test().
136  */
137 __owur int finish_test(int ret);
138
139 void add_test(const char *test_case_name, int (*test_fn) ());
140 void add_all_tests(const char *test_case_name, int (*test_fn)(int idx), int num,
141                    int subtest);
142 __owur int run_tests(const char *test_prog_name);
143 void set_test_title(const char *title);
144
145 /*
146  * Declarations for user defined functions
147  */
148 void register_tests(void);
149 int test_main(int argc, char *argv[]);
150
151
152 /*
153  *  Test assumption verification helpers.
154  */
155
156 #define PRINTF_FORMAT(a, b)
157 #if defined(__GNUC__) && defined(__STDC_VERSION__)
158   /*
159    * Because we support the 'z' modifier, which made its appearance in C99,
160    * we can't use __attribute__ with pre C99 dialects.
161    */
162 # if __STDC_VERSION__ >= 199901L
163 #  undef PRINTF_FORMAT
164 #  define PRINTF_FORMAT(a, b)   __attribute__ ((format(printf, a, b)))
165 # endif
166 #endif
167
168 #  define DECLARE_COMPARISON(type, name, opname)                        \
169     int test_ ## name ## _ ## opname(const char *, int,                 \
170                                      const char *, const char *,        \
171                                      const type, const type);
172
173 # define DECLARE_COMPARISONS(type, name)                                \
174     DECLARE_COMPARISON(type, name, eq)                                  \
175     DECLARE_COMPARISON(type, name, ne)                                  \
176     DECLARE_COMPARISON(type, name, lt)                                  \
177     DECLARE_COMPARISON(type, name, le)                                  \
178     DECLARE_COMPARISON(type, name, gt)                                  \
179     DECLARE_COMPARISON(type, name, ge)
180
181 DECLARE_COMPARISONS(int, int)
182 DECLARE_COMPARISONS(unsigned int, uint)
183 DECLARE_COMPARISONS(char, char)
184 DECLARE_COMPARISONS(unsigned char, uchar)
185 DECLARE_COMPARISONS(long, long)
186 DECLARE_COMPARISONS(unsigned long, ulong)
187 /*
188  * Because this comparison uses a printf format specifier that's not
189  * universally known (yet), we provide an option to not have it declared.
190  */
191 # ifndef TESTUTIL_NO_size_t_COMPARISON
192 DECLARE_COMPARISONS(size_t, size_t)
193 # endif
194
195 /*
196  * Pointer comparisons against other pointers and null.
197  * These functions return 1 if the test is true.
198  * Otherwise, they return 0 and pretty-print diagnostics.
199  * These should not be called directly, use the TEST_xxx macros below instead.
200  */
201 DECLARE_COMPARISON(void *, ptr, eq)
202 DECLARE_COMPARISON(void *, ptr, ne)
203 int test_ptr(const char *file, int line, const char *s, const void *p);
204 int test_ptr_null(const char *file, int line, const char *s, const void *p);
205
206 /*
207  * Equality tests for strings where NULL is a legitimate value.
208  * These calls return 1 if the two passed strings compare true.
209  * Otherwise, they return 0 and pretty-print diagnostics.
210  * These should not be called directly, use the TEST_xxx macros below instead.
211  */
212 DECLARE_COMPARISON(char *, str, eq)
213 DECLARE_COMPARISON(char *, str, ne)
214
215 /*
216  * Same as above, but for strncmp.
217  */
218 int test_strn_eq(const char *file, int line, const char *, const char *,
219                  const char *a, const char *b, size_t s);
220 int test_strn_ne(const char *file, int line, const char *, const char *,
221                  const char *a, const char *b, size_t s);
222
223 /*
224  * Equality test for memory blocks where NULL is a legitimate value.
225  * These calls return 1 if the two memory blocks compare true.
226  * Otherwise, they return 0 and pretty-print diagnostics.
227  * These should not be called directly, use the TEST_xxx macros below instead.
228  */
229 int test_mem_eq(const char *, int, const char *, const char *,
230                 const void *, size_t, const void *, size_t);
231 int test_mem_ne(const char *, int, const char *, const char *,
232                 const void *, size_t, const void *, size_t);
233
234 /*
235  * Check a boolean result for being true or false.
236  * They return 1 if the condition is true (i.e. the value is non-zro).
237  * Otherwise, they return 0 and pretty-prints diagnostics using |desc|.
238  * These should not be called directly, use the TEST_xxx macros below instead.
239  */
240 int test_true(const char *file, int line, const char *s, int b);
241 int test_false(const char *file, int line, const char *s, int b);
242
243 /*
244  * Comparisons between BIGNUMs.
245  * BIGNUMS can be compared against other BIGNUMs or zero.
246  * Some additional equality tests against 1 & specific values are provided.
247  * Tests for parity are included as well.
248  */
249 DECLARE_COMPARISONS(BIGNUM *, BN)
250 int test_BN_eq_zero(const char *file, int line, const char *s, const BIGNUM *a);
251 int test_BN_ne_zero(const char *file, int line, const char *s, const BIGNUM *a);
252 int test_BN_lt_zero(const char *file, int line, const char *s, const BIGNUM *a);
253 int test_BN_le_zero(const char *file, int line, const char *s, const BIGNUM *a);
254 int test_BN_gt_zero(const char *file, int line, const char *s, const BIGNUM *a);
255 int test_BN_ge_zero(const char *file, int line, const char *s, const BIGNUM *a);
256 int test_BN_eq_one(const char *file, int line, const char *s, const BIGNUM *a);
257 int test_BN_odd(const char *file, int line, const char *s, const BIGNUM *a);
258 int test_BN_even(const char *file, int line, const char *s, const BIGNUM *a);
259 int test_BN_eq_word(const char *file, int line, const char *bns, const char *ws,
260                     const BIGNUM *a, BN_ULONG w);
261 int test_BN_abs_eq_word(const char *file, int line, const char *bns,
262                         const char *ws, const BIGNUM *a, BN_ULONG w);
263
264 /*
265  * Pretty print a failure message.
266  * These should not be called directly, use the TEST_xxx macros below instead.
267  */
268 void test_error(const char *file, int line, const char *desc, ...)
269     PRINTF_FORMAT(3, 4);
270 void test_error_c90(const char *desc, ...) PRINTF_FORMAT(1, 2);
271 void test_info(const char *file, int line, const char *desc, ...)
272     PRINTF_FORMAT(3, 4);
273 void test_info_c90(const char *desc, ...) PRINTF_FORMAT(1, 2);
274 void test_openssl_errors(void);
275
276 /*
277  * The following macros provide wrapper calls to the test functions with
278  * a default description that indicates the file and line number of the error.
279  *
280  * The following macros guarantee to evaluate each argument exactly once.
281  * This allows constructs such as: if(!TEST_ptr(ptr = OPENSSL_malloc(..)))
282  * to produce better contextual output than:
283  *      ptr = OPENSSL_malloc(..);
284  *      if (!TEST_ptr(ptr))
285  */
286 # define TEST_int_eq(a, b)    test_int_eq(__FILE__, __LINE__, #a, #b, a, b)
287 # define TEST_int_ne(a, b)    test_int_ne(__FILE__, __LINE__, #a, #b, a, b)
288 # define TEST_int_lt(a, b)    test_int_lt(__FILE__, __LINE__, #a, #b, a, b)
289 # define TEST_int_le(a, b)    test_int_le(__FILE__, __LINE__, #a, #b, a, b)
290 # define TEST_int_gt(a, b)    test_int_gt(__FILE__, __LINE__, #a, #b, a, b)
291 # define TEST_int_ge(a, b)    test_int_ge(__FILE__, __LINE__, #a, #b, a, b)
292
293 # define TEST_int_eq(a, b)    test_int_eq(__FILE__, __LINE__, #a, #b, a, b)
294 # define TEST_int_ne(a, b)    test_int_ne(__FILE__, __LINE__, #a, #b, a, b)
295 # define TEST_int_lt(a, b)    test_int_lt(__FILE__, __LINE__, #a, #b, a, b)
296 # define TEST_int_le(a, b)    test_int_le(__FILE__, __LINE__, #a, #b, a, b)
297 # define TEST_int_gt(a, b)    test_int_gt(__FILE__, __LINE__, #a, #b, a, b)
298 # define TEST_int_ge(a, b)    test_int_ge(__FILE__, __LINE__, #a, #b, a, b)
299
300 # define TEST_uint_eq(a, b)   test_uint_eq(__FILE__, __LINE__, #a, #b, a, b)
301 # define TEST_uint_ne(a, b)   test_uint_ne(__FILE__, __LINE__, #a, #b, a, b)
302 # define TEST_uint_lt(a, b)   test_uint_lt(__FILE__, __LINE__, #a, #b, a, b)
303 # define TEST_uint_le(a, b)   test_uint_le(__FILE__, __LINE__, #a, #b, a, b)
304 # define TEST_uint_gt(a, b)   test_uint_gt(__FILE__, __LINE__, #a, #b, a, b)
305 # define TEST_uint_ge(a, b)   test_uint_ge(__FILE__, __LINE__, #a, #b, a, b)
306
307 # define TEST_char_eq(a, b)   test_char_eq(__FILE__, __LINE__, #a, #b, a, b)
308 # define TEST_char_ne(a, b)   test_char_ne(__FILE__, __LINE__, #a, #b, a, b)
309 # define TEST_char_lt(a, b)   test_char_lt(__FILE__, __LINE__, #a, #b, a, b)
310 # define TEST_char_le(a, b)   test_char_le(__FILE__, __LINE__, #a, #b, a, b)
311 # define TEST_char_gt(a, b)   test_char_gt(__FILE__, __LINE__, #a, #b, a, b)
312 # define TEST_char_ge(a, b)   test_char_ge(__FILE__, __LINE__, #a, #b, a, b)
313
314 # define TEST_uchar_eq(a, b)  test_uchar_eq(__FILE__, __LINE__, #a, #b, a, b)
315 # define TEST_uchar_ne(a, b)  test_uchar_ne(__FILE__, __LINE__, #a, #b, a, b)
316 # define TEST_uchar_lt(a, b)  test_uchar_lt(__FILE__, __LINE__, #a, #b, a, b)
317 # define TEST_uchar_le(a, b)  test_uchar_le(__FILE__, __LINE__, #a, #b, a, b)
318 # define TEST_uchar_gt(a, b)  test_uchar_gt(__FILE__, __LINE__, #a, #b, a, b)
319 # define TEST_uchar_ge(a, b)  test_uchar_ge(__FILE__, __LINE__, #a, #b, a, b)
320
321 # define TEST_long_eq(a, b)   test_long_eq(__FILE__, __LINE__, #a, #b, a, b)
322 # define TEST_long_ne(a, b)   test_long_ne(__FILE__, __LINE__, #a, #b, a, b)
323 # define TEST_long_lt(a, b)   test_long_lt(__FILE__, __LINE__, #a, #b, a, b)
324 # define TEST_long_le(a, b)   test_long_le(__FILE__, __LINE__, #a, #b, a, b)
325 # define TEST_long_gt(a, b)   test_long_gt(__FILE__, __LINE__, #a, #b, a, b)
326 # define TEST_long_ge(a, b)   test_long_ge(__FILE__, __LINE__, #a, #b, a, b)
327
328 # define TEST_ulong_eq(a, b)  test_ulong_eq(__FILE__, __LINE__, #a, #b, a, b)
329 # define TEST_ulong_ne(a, b)  test_ulong_ne(__FILE__, __LINE__, #a, #b, a, b)
330 # define TEST_ulong_lt(a, b)  test_ulong_lt(__FILE__, __LINE__, #a, #b, a, b)
331 # define TEST_ulong_le(a, b)  test_ulong_le(__FILE__, __LINE__, #a, #b, a, b)
332 # define TEST_ulong_gt(a, b)  test_ulong_gt(__FILE__, __LINE__, #a, #b, a, b)
333 # define TEST_ulong_ge(a, b)  test_ulong_ge(__FILE__, __LINE__, #a, #b, a, b)
334
335 # define TEST_size_t_eq(a, b) test_size_t_eq(__FILE__, __LINE__, #a, #b, a, b)
336 # define TEST_size_t_ne(a, b) test_size_t_ne(__FILE__, __LINE__, #a, #b, a, b)
337 # define TEST_size_t_lt(a, b) test_size_t_lt(__FILE__, __LINE__, #a, #b, a, b)
338 # define TEST_size_t_le(a, b) test_size_t_le(__FILE__, __LINE__, #a, #b, a, b)
339 # define TEST_size_t_gt(a, b) test_size_t_gt(__FILE__, __LINE__, #a, #b, a, b)
340 # define TEST_size_t_ge(a, b) test_size_t_ge(__FILE__, __LINE__, #a, #b, a, b)
341
342 # define TEST_ptr_eq(a, b)    test_ptr_eq(__FILE__, __LINE__, #a, #b, a, b)
343 # define TEST_ptr_ne(a, b)    test_ptr_ne(__FILE__, __LINE__, #a, #b, a, b)
344 # define TEST_ptr(a)          test_ptr(__FILE__, __LINE__, #a, a)
345 # define TEST_ptr_null(a)     test_ptr_null(__FILE__, __LINE__, #a, a)
346
347 # define TEST_str_eq(a, b)    test_str_eq(__FILE__, __LINE__, #a, #b, a, b)
348 # define TEST_str_ne(a, b)    test_str_ne(__FILE__, __LINE__, #a, #b, a, b)
349 # define TEST_strn_eq(a, b, n) test_strn_eq(__FILE__, __LINE__, #a, #b, a, b, n)
350 # define TEST_strn_ne(a, b, n) test_strn_ne(__FILE__, __LINE__, #a, #b, a, b, n)
351
352 # define TEST_mem_eq(a, m, b, n) test_mem_eq(__FILE__, __LINE__, #a, #b, a, m, b, n)
353 # define TEST_mem_ne(a, m, b, n) test_mem_ne(__FILE__, __LINE__, #a, #b, a, m, b, n)
354
355 # define TEST_true(a)         test_true(__FILE__, __LINE__, #a, (a) != 0)
356 # define TEST_false(a)        test_false(__FILE__, __LINE__, #a, (a) != 0)
357
358 # define TEST_BN_eq(a, b)     test_BN_eq(__FILE__, __LINE__, #a, #b, a, b)
359 # define TEST_BN_ne(a, b)     test_BN_ne(__FILE__, __LINE__, #a, #b, a, b)
360 # define TEST_BN_lt(a, b)     test_BN_lt(__FILE__, __LINE__, #a, #b, a, b)
361 # define TEST_BN_gt(a, b)     test_BN_gt(__FILE__, __LINE__, #a, #b, a, b)
362 # define TEST_BN_le(a, b)     test_BN_le(__FILE__, __LINE__, #a, #b, a, b)
363 # define TEST_BN_ge(a, b)     test_BN_ge(__FILE__, __LINE__, #a, #b, a, b)
364 # define TEST_BN_eq_zero(a)   test_BN_eq_zero(__FILE__, __LINE__, #a, a)
365 # define TEST_BN_ne_zero(a)   test_BN_ne_zero(__FILE__, __LINE__, #a, a)
366 # define TEST_BN_lt_zero(a)   test_BN_lt_zero(__FILE__, __LINE__, #a, a)
367 # define TEST_BN_gt_zero(a)   test_BN_gt_zero(__FILE__, __LINE__, #a, a)
368 # define TEST_BN_le_zero(a)   test_BN_le_zero(__FILE__, __LINE__, #a, a)
369 # define TEST_BN_ge_zero(a)   test_BN_ge_zero(__FILE__, __LINE__, #a, a)
370 # define TEST_BN_eq_one(a)    test_BN_eq_one(__FILE__, __LINE__, #a, a)
371 # define TEST_BN_eq_word(a, w) test_BN_eq_word(__FILE__, __LINE__, #a, #w, a, w)
372 # define TEST_BN_abs_eq_word(a, w) test_BN_abs_eq_word(__FILE__, __LINE__, #a, #w, a, w)
373 # define TEST_BN_odd(a)       test_BN_odd(__FILE__, __LINE__, #a, a)
374 # define TEST_BN_even(a)      test_BN_even(__FILE__, __LINE__, #a, a)
375
376 /*
377  * TEST_error(desc, ...) prints an informative error message in the standard
378  * format.  |desc| is a printf format string.
379  */
380 # if !defined(__STDC_VERSION__) || __STDC_VERSION__ < 199901L
381 #  define TEST_error         test_error_c90
382 #  define TEST_info          test_info_c90
383 # else
384 #  define TEST_error(...)    test_error(__FILE__, __LINE__, __VA_ARGS__)
385 #  define TEST_info(...)     test_info(__FILE__, __LINE__, __VA_ARGS__)
386 # endif
387 # define TEST_openssl_errors test_openssl_errors
388
389 /*
390  * For "impossible" conditions such as malloc failures or bugs in test code,
391  * where continuing the test would be meaningless. Note that OPENSSL_assert
392  * is fatal, and is never compiled out.
393  */
394 # define TEST_check(condition)                  \
395     do {                                        \
396         if (!(condition)) {                     \
397             TEST_openssl_errors();              \
398             OPENSSL_assert(!#condition);        \
399         }                                       \
400     } while (0)
401
402 extern BIO *bio_out;
403 extern BIO *bio_err;
404
405 /*
406  * Utilities to parse a test file.
407  */
408 #define TESTMAXPAIRS        20
409
410 typedef struct pair_st {
411     char *key;
412     char *value;
413 } PAIR;
414
415 typedef struct stanza_st {
416     const char *test_file;      /* Input file name */
417     BIO *fp;                    /* Input file */
418     int curr;                   /* Current line in file */
419     int start;                  /* Line where test starts */
420     int errors;                 /* Error count */
421     int numtests;               /* Number of tests */
422     int numskip;                /* Number of skipped tests */
423     int numpairs;
424     PAIR pairs[TESTMAXPAIRS];
425     BIO *key;                   /* temp memory BIO for reading in keys */
426     char buff[4096];            /* Input buffer for a single key/value */
427 } STANZA;
428
429 /*
430  * Prepare to start reading the file |testfile| as input.
431  */
432 int test_start_file(STANZA *s, const char *testfile);
433 int test_end_file(STANZA *s);
434
435 /*
436  * Read a stanza from the test file.  A stanza consists of a block
437  * of lines of the form
438  *      key = value
439  * The block is terminated by EOF or a blank line.
440  * Return 1 if found, 0 on EOF or error.
441  */
442 int test_readstanza(STANZA *s);
443
444 /*
445  * Clear a stanza, release all allocated memory.
446  */
447 void test_clearstanza(STANZA *s);
448
449 #endif                          /* HEADER_TESTUTIL_H */