mirror of
https://github.com/freebsd/freebsd-src
synced 2024-10-15 04:43:53 +00:00
5d3e7166f6
Sponsored by: The FreeBSD Foundation
60 lines
2.1 KiB
C
60 lines
2.1 KiB
C
/*
|
|
* Copyright (c) 2014-2020 Pavel Kalvoda <me@pavelkalvoda.com>
|
|
*
|
|
* libcbor is free software; you can redistribute it and/or modify
|
|
* it under the terms of the MIT license. See LICENSE for details.
|
|
*/
|
|
|
|
#include "cbor/internal/memory_utils.h"
|
|
#include <string.h>
|
|
#include "assertions.h"
|
|
|
|
static void test_safe_multiply(void **_CBOR_UNUSED(_state)) {
|
|
assert_true(_cbor_safe_to_multiply(1, 1));
|
|
assert_true(_cbor_safe_to_multiply(SIZE_MAX, 0));
|
|
assert_true(_cbor_safe_to_multiply(SIZE_MAX, 1));
|
|
assert_false(_cbor_safe_to_multiply(SIZE_MAX, 2));
|
|
assert_false(_cbor_safe_to_multiply(SIZE_MAX, SIZE_MAX));
|
|
}
|
|
|
|
static void test_safe_add(void **_CBOR_UNUSED(_state)) {
|
|
assert_true(_cbor_safe_to_add(1, 1));
|
|
assert_true(_cbor_safe_to_add(SIZE_MAX - 1, 1));
|
|
assert_true(_cbor_safe_to_add(SIZE_MAX, 0));
|
|
assert_false(_cbor_safe_to_add(SIZE_MAX, 1));
|
|
assert_false(_cbor_safe_to_add(SIZE_MAX, 2));
|
|
assert_false(_cbor_safe_to_add(SIZE_MAX, SIZE_MAX));
|
|
assert_false(_cbor_safe_to_add(SIZE_MAX - 1, SIZE_MAX));
|
|
assert_false(_cbor_safe_to_add(SIZE_MAX - 1, SIZE_MAX - 1));
|
|
}
|
|
|
|
static void test_safe_signalling_add(void **_CBOR_UNUSED(_state)) {
|
|
assert_size_equal(_cbor_safe_signaling_add(1, 2), 3);
|
|
assert_size_equal(_cbor_safe_signaling_add(0, 1), 0);
|
|
assert_size_equal(_cbor_safe_signaling_add(0, SIZE_MAX), 0);
|
|
assert_size_equal(_cbor_safe_signaling_add(1, SIZE_MAX), 0);
|
|
assert_size_equal(_cbor_safe_signaling_add(1, SIZE_MAX - 1), SIZE_MAX);
|
|
}
|
|
|
|
static void test_realloc_multiple(void **_CBOR_UNUSED(_state)) {
|
|
unsigned char *data = malloc(1);
|
|
data[0] = 0x2a;
|
|
|
|
data = _cbor_realloc_multiple(data, /*item_size=*/1, /*item_count=*/10);
|
|
assert_size_equal(data[0], 0x2a);
|
|
data[9] = 0x2b; // Sanitizer will stop us if not ok
|
|
free(data);
|
|
|
|
assert_null(_cbor_realloc_multiple(NULL, SIZE_MAX, 2));
|
|
}
|
|
|
|
int main(void) {
|
|
const struct CMUnitTest tests[] = {
|
|
cmocka_unit_test(test_safe_multiply),
|
|
cmocka_unit_test(test_safe_add),
|
|
cmocka_unit_test(test_safe_signalling_add),
|
|
cmocka_unit_test(test_realloc_multiple),
|
|
};
|
|
return cmocka_run_group_tests(tests, NULL, NULL);
|
|
}
|