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