1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
#ifndef TEST_ALLOCATOR_H_
#define TEST_ALLOCATOR_H_
#include "assertions.h"
#include "cbor.h"
// Harness for mocking `malloc` and `realloc`
typedef enum call_expectation {
MALLOC,
MALLOC_FAIL,
REALLOC,
REALLOC_FAIL
} call_expectation;
void set_mock_malloc(int calls, ...);
void finalize_mock_malloc(void);
void *instrumented_malloc(size_t size);
void *instrumented_realloc(void *ptr, size_t size);
#define WITH_MOCK_MALLOC(block, malloc_calls, ...) \
do { \
cbor_set_allocs(instrumented_malloc, instrumented_realloc, free); \
set_mock_malloc(malloc_calls, __VA_ARGS__); \
block; \
finalize_mock_malloc(); \
cbor_set_allocs(malloc, realloc, free); \
} while (0)
#define WITH_FAILING_MALLOC(block) WITH_MOCK_MALLOC(block, 1, MALLOC_FAIL)
#endif // TEST_ALLOCATOR_H_
|