1 /* 2 * Copyright (c) 2014-2020 Pavel Kalvoda <me@pavelkalvoda.com> 3 * 4 * libcbor is free software; you can redistribute it and/or modify 5 * it under the terms of the MIT license. See LICENSE for details. 6 */ 7 8 #include "cbor/internal/memory_utils.h" 9 #include <string.h> 10 #include "assertions.h" 11 12 static void test_safe_multiply(void **_CBOR_UNUSED(_state)) { 13 assert_true(_cbor_safe_to_multiply(1, 1)); 14 assert_true(_cbor_safe_to_multiply(SIZE_MAX, 0)); 15 assert_true(_cbor_safe_to_multiply(SIZE_MAX, 1)); 16 assert_false(_cbor_safe_to_multiply(SIZE_MAX, 2)); 17 assert_false(_cbor_safe_to_multiply(SIZE_MAX, SIZE_MAX)); 18 } 19 20 static void test_safe_add(void **_CBOR_UNUSED(_state)) { 21 assert_true(_cbor_safe_to_add(1, 1)); 22 assert_true(_cbor_safe_to_add(SIZE_MAX - 1, 1)); 23 assert_true(_cbor_safe_to_add(SIZE_MAX, 0)); 24 assert_false(_cbor_safe_to_add(SIZE_MAX, 1)); 25 assert_false(_cbor_safe_to_add(SIZE_MAX, 2)); 26 assert_false(_cbor_safe_to_add(SIZE_MAX, SIZE_MAX)); 27 assert_false(_cbor_safe_to_add(SIZE_MAX - 1, SIZE_MAX)); 28 assert_false(_cbor_safe_to_add(SIZE_MAX - 1, SIZE_MAX - 1)); 29 } 30 31 static void test_safe_signalling_add(void **_CBOR_UNUSED(_state)) { 32 assert_size_equal(_cbor_safe_signaling_add(1, 2), 3); 33 assert_size_equal(_cbor_safe_signaling_add(0, 1), 0); 34 assert_size_equal(_cbor_safe_signaling_add(0, SIZE_MAX), 0); 35 assert_size_equal(_cbor_safe_signaling_add(1, SIZE_MAX), 0); 36 assert_size_equal(_cbor_safe_signaling_add(1, SIZE_MAX - 1), SIZE_MAX); 37 } 38 39 static void test_realloc_multiple(void **_CBOR_UNUSED(_state)) { 40 unsigned char *data = malloc(1); 41 data[0] = 0x2a; 42 43 data = _cbor_realloc_multiple(data, /*item_size=*/1, /*item_count=*/10); 44 assert_size_equal(data[0], 0x2a); 45 data[9] = 0x2b; // Sanitizer will stop us if not ok 46 free(data); 47 48 assert_null(_cbor_realloc_multiple(NULL, SIZE_MAX, 2)); 49 } 50 51 int main(void) { 52 const struct CMUnitTest tests[] = { 53 cmocka_unit_test(test_safe_multiply), 54 cmocka_unit_test(test_safe_add), 55 cmocka_unit_test(test_safe_signalling_add), 56 cmocka_unit_test(test_realloc_multiple), 57 }; 58 return cmocka_run_group_tests(tests, NULL, NULL); 59 } 60