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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#include "config.h"
#include <cmocka.h>
#include <cmocka_private.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
static void torture_test_calloc(void **state)
{
void *ptr;
(void)state; /* unsused */
ptr = test_calloc(2, SIZE_MAX);
assert_null(ptr);
ptr = test_calloc(SIZE_MAX, 2);
assert_null(ptr);
/* overflows to 0 */
ptr = test_calloc(2, (SIZE_MAX/2)+1);
assert_null(ptr);
ptr = test_calloc(3, (SIZE_MAX/2)+42);
assert_null(ptr);
}
static void torture_test_malloc(void **state)
{
char *str;
size_t str_len;
size_t len;
(void)state; /* unsused */
str_len = 12;
str = (char *)test_malloc(str_len);
assert_non_null(str);
len = snprintf(str, str_len, "test string");
assert_int_equal(len, 11);
len = strlen(str);
assert_int_equal(len, 11);
test_free(str);
}
static void torture_test_realloc(void **state)
{
char *str;
char *tmp;
size_t str_len;
size_t len;
(void)state; /* unsused */
str_len = 16;
str = (char *)test_malloc(str_len);
assert_non_null(str);
len = snprintf(str, str_len, "test string 123");
assert_int_equal(len, 15);
len = strlen(str);
assert_int_equal(len, 15);
str_len = 20;
tmp = test_realloc(str, str_len);
assert_non_null(tmp);
str = tmp;
len = strlen(str);
assert_string_equal(tmp, "test string 123");
snprintf(str + len, str_len - len, "4567");
assert_string_equal(tmp, "test string 1234567");
test_free(str);
}
static void torture_test_realloc_set0(void **state)
{
char *str;
size_t str_len;
(void)state; /* unsused */
str_len = 16;
str = (char *)test_malloc(str_len);
assert_non_null(str);
/* realloc(ptr, 0) is like a free() */
str = (char *)test_realloc(str, 0);
assert_null(str);
}
int main(void) {
const struct CMUnitTest alloc_tests[] = {
cmocka_unit_test(torture_test_calloc),
cmocka_unit_test(torture_test_malloc),
cmocka_unit_test(torture_test_realloc),
cmocka_unit_test(torture_test_realloc_set0),
};
return cmocka_run_group_tests(alloc_tests, NULL, NULL);
}
|
