1 /** 2 * \file macros.h 3 * 4 * \brief This file contains generic macros for the purpose of testing. 5 */ 6 7 /* 8 * Copyright The Mbed TLS Contributors 9 * SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later 10 */ 11 12 #ifndef TEST_MACROS_H 13 #define TEST_MACROS_H 14 15 #include "mbedtls/build_info.h" 16 17 #include <stdlib.h> 18 19 #include "mbedtls/platform.h" 20 21 #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C) 22 #include "mbedtls/memory_buffer_alloc.h" 23 #endif 24 #include "common.h" 25 26 /** 27 * \brief This macro tests the expression passed to it as a test step or 28 * individual test in a test case. 29 * 30 * It allows a library function to return a value and return an error 31 * code that can be tested. 32 * 33 * Failing the test means: 34 * - Mark this test case as failed. 35 * - Print a message identifying the failure. 36 * - Jump to the \c exit label. 37 * 38 * This macro expands to an instruction, not an expression. 39 * It may jump to the \c exit label. 40 * 41 * \param TEST The test expression to be tested. 42 */ 43 #define TEST_ASSERT(TEST) \ 44 do { \ 45 if (!(TEST)) \ 46 { \ 47 mbedtls_test_fail( #TEST, __LINE__, __FILE__); \ 48 goto exit; \ 49 } \ 50 } while (0) 51 52 /** This macro asserts fails the test with given output message. 53 * 54 * \param MESSAGE The message to be outputed on assertion 55 */ 56 #define TEST_FAIL(MESSAGE) \ 57 do { \ 58 mbedtls_test_fail(MESSAGE, __LINE__, __FILE__); \ 59 goto exit; \ 60 } while (0) 61 62 /** Evaluate two integer expressions and fail the test case if they have 63 * different values. 64 * 65 * The two expressions should have the same signedness, otherwise the 66 * comparison is not meaningful if the signed value is negative. 67 * 68 * \param expr1 An integral-typed expression to evaluate. 69 * \param expr2 Another integral-typed expression to evaluate. 70 */ 71 #define TEST_EQUAL(expr1, expr2) \ 72 do { \ 73 if (!mbedtls_test_equal( #expr1 " == " #expr2, __LINE__, __FILE__, \ 74 (unsigned long long) (expr1), (unsigned long long) (expr2))) \ 75 goto exit; \ 76 } while (0) 77 78 /** Evaluate two unsigned integer expressions and fail the test case 79 * if they are not in increasing order (left <= right). 80 * 81 * \param expr1 An integral-typed expression to evaluate. 82 * \param expr2 Another integral-typed expression to evaluate. 83 */ 84 #define TEST_LE_U(expr1, expr2) \ 85 do { \ 86 if (!mbedtls_test_le_u( #expr1 " <= " #expr2, __LINE__, __FILE__, \ 87 expr1, expr2)) \ 88 goto exit; \ 89 } while (0) 90 91 /** Evaluate two signed integer expressions and fail the test case 92 * if they are not in increasing order (left <= right). 93 * 94 * \param expr1 An integral-typed expression to evaluate. 95 * \param expr2 Another integral-typed expression to evaluate. 96 */ 97 #define TEST_LE_S(expr1, expr2) \ 98 do { \ 99 if (!mbedtls_test_le_s( #expr1 " <= " #expr2, __LINE__, __FILE__, \ 100 expr1, expr2)) \ 101 goto exit; \ 102 } while (0) 103 104 /** Allocate memory dynamically and fail the test case if this fails. 105 * The allocated memory will be filled with zeros. 106 * 107 * You must set \p pointer to \c NULL before calling this macro and 108 * put `mbedtls_free(pointer)` in the test's cleanup code. 109 * 110 * If \p item_count is zero, the resulting \p pointer will be \c NULL. 111 * This is usually what we want in tests since API functions are 112 * supposed to accept null pointers when a buffer size is zero. 113 * 114 * This macro expands to an instruction, not an expression. 115 * It may jump to the \c exit label. 116 * 117 * \param pointer An lvalue where the address of the allocated buffer 118 * will be stored. 119 * This expression may be evaluated multiple times. 120 * \param item_count Number of elements to allocate. 121 * This expression may be evaluated multiple times. 122 * 123 */ 124 #define TEST_CALLOC(pointer, item_count) \ 125 do { \ 126 TEST_ASSERT((pointer) == NULL); \ 127 if ((item_count) != 0) { \ 128 (pointer) = mbedtls_calloc((item_count), \ 129 sizeof(*(pointer))); \ 130 TEST_ASSERT((pointer) != NULL); \ 131 } \ 132 } while (0) 133 134 /** Allocate memory dynamically and fail the test case if this fails. 135 * The allocated memory will be filled with zeros. 136 * 137 * You must set \p pointer to \c NULL before calling this macro and 138 * put `mbedtls_free(pointer)` in the test's cleanup code. 139 * 140 * If \p item_count is zero, the resulting \p pointer will not be \c NULL. 141 * 142 * This macro expands to an instruction, not an expression. 143 * It may jump to the \c exit label. 144 * 145 * \param pointer An lvalue where the address of the allocated buffer 146 * will be stored. 147 * This expression may be evaluated multiple times. 148 * \param item_count Number of elements to allocate. 149 * This expression may be evaluated multiple times. 150 * 151 * Note: if passing size 0, mbedtls_calloc may return NULL. In this case, 152 * we reattempt to allocate with the smallest possible buffer to assure a 153 * non-NULL pointer. 154 */ 155 #define TEST_CALLOC_NONNULL(pointer, item_count) \ 156 do { \ 157 TEST_ASSERT((pointer) == NULL); \ 158 (pointer) = mbedtls_calloc((item_count), \ 159 sizeof(*(pointer))); \ 160 if (((pointer) == NULL) && ((item_count) == 0)) { \ 161 (pointer) = mbedtls_calloc(1, 1); \ 162 } \ 163 TEST_ASSERT((pointer) != NULL); \ 164 } while (0) 165 166 /* For backwards compatibility */ 167 #define ASSERT_ALLOC(pointer, item_count) TEST_CALLOC(pointer, item_count) 168 169 /** Allocate memory dynamically. If the allocation fails, skip the test case. 170 * 171 * This macro behaves like #TEST_CALLOC, except that if the allocation 172 * fails, it marks the test as skipped rather than failed. 173 */ 174 #define TEST_CALLOC_OR_SKIP(pointer, item_count) \ 175 do { \ 176 TEST_ASSERT((pointer) == NULL); \ 177 if ((item_count) != 0) { \ 178 (pointer) = mbedtls_calloc((item_count), \ 179 sizeof(*(pointer))); \ 180 TEST_ASSUME((pointer) != NULL); \ 181 } \ 182 } while (0) 183 184 /* For backwards compatibility */ 185 #define ASSERT_ALLOC_WEAK(pointer, item_count) TEST_CALLOC_OR_SKIP(pointer, item_count) 186 187 /** Compare two buffers and fail the test case if they differ. 188 * 189 * This macro expands to an instruction, not an expression. 190 * It may jump to the \c exit label. 191 * 192 * \param p1 Pointer to the start of the first buffer. 193 * \param size1 Size of the first buffer in bytes. 194 * This expression may be evaluated multiple times. 195 * \param p2 Pointer to the start of the second buffer. 196 * \param size2 Size of the second buffer in bytes. 197 * This expression may be evaluated multiple times. 198 */ 199 #define TEST_MEMORY_COMPARE(p1, size1, p2, size2) \ 200 do { \ 201 TEST_EQUAL((size1), (size2)); \ 202 if ((size1) != 0) { \ 203 TEST_ASSERT(memcmp((p1), (p2), (size1)) == 0); \ 204 } \ 205 } while (0) 206 207 /* For backwards compatibility */ 208 #define ASSERT_COMPARE(p1, size1, p2, size2) TEST_MEMORY_COMPARE(p1, size1, p2, size2) 209 210 /** 211 * \brief This macro tests the expression passed to it and skips the 212 * running test if it doesn't evaluate to 'true'. 213 * 214 * \param TEST The test expression to be tested. 215 */ 216 #define TEST_ASSUME(TEST) \ 217 do { \ 218 if (!(TEST)) \ 219 { \ 220 mbedtls_test_skip( #TEST, __LINE__, __FILE__); \ 221 goto exit; \ 222 } \ 223 } while (0) 224 225 #define TEST_HELPER_ASSERT(a) if (!(a)) \ 226 { \ 227 mbedtls_fprintf(stderr, "Assertion Failed at %s:%d - %s\n", \ 228 __FILE__, __LINE__, #a); \ 229 mbedtls_exit(1); \ 230 } 231 232 /** Return the smaller of two values. 233 * 234 * \param x An integer-valued expression without side effects. 235 * \param y An integer-valued expression without side effects. 236 * 237 * \return The smaller of \p x and \p y. 238 */ 239 #define MIN(x, y) ((x) < (y) ? (x) : (y)) 240 241 /** Return the larger of two values. 242 * 243 * \param x An integer-valued expression without side effects. 244 * \param y An integer-valued expression without side effects. 245 * 246 * \return The larger of \p x and \p y. 247 */ 248 #define MAX(x, y) ((x) > (y) ? (x) : (y)) 249 250 #endif /* TEST_MACROS_H */ 251
