1 #include <setjmp.h> 2 #include <stdarg.h> 3 #include <stddef.h> 4 5 #include <cmocka.h> 6 7 #include "assertions.h" 8 #include "cbor.h" 9 #include "stream_expectations.h" 10 11 static size_t generate_overflow_data(unsigned char **overflow_data) { 12 int i; 13 *overflow_data = (unsigned char *)malloc(CBOR_MAX_STACK_SIZE + 3); 14 for (i = 0; i < CBOR_MAX_STACK_SIZE + 1; i++) { 15 (*overflow_data)[i] = 0xC2; // tag of positive bignum 16 } 17 (*overflow_data)[CBOR_MAX_STACK_SIZE + 1] = 0x41; // bytestring of length 1 18 (*overflow_data)[CBOR_MAX_STACK_SIZE + 2] = 0x01; // a bignum of value 1 19 return CBOR_MAX_STACK_SIZE + 3; 20 } 21 22 static void test_stack_over_limit(void **state) { 23 unsigned char *overflow_data; 24 size_t overflow_data_len; 25 struct cbor_load_result res; 26 overflow_data_len = generate_overflow_data(&overflow_data); 27 cbor_load(overflow_data, overflow_data_len, &res); 28 free(overflow_data); 29 assert_int_equal(res.error.code, CBOR_ERR_MEMERROR); 30 } 31 32 int main() { 33 const struct CMUnitTest tests[] = {cmocka_unit_test(test_stack_over_limit)}; 34 return cmocka_run_group_tests(tests, NULL, NULL); 35 } 36