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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
/*
* robust_deserialization_unit.c
*
*/
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <roaring/roaring.h>
#include "config.h"
#include "test.h"
long filesize(FILE* fp) {
fseek(fp, 0L, SEEK_END);
return ftell(fp);
}
char* readfile(FILE* fp, size_t * bytes) {
*bytes = filesize(fp);
char* buf = malloc(*bytes);
if(buf == NULL) return NULL;
rewind(fp);
size_t cnt = fread(buf, 1, *bytes, fp);
if(*bytes != cnt){
free(buf);
return NULL;
}
return buf;
}
int compare(char* x, char* y, size_t size) {
for (size_t i = 0; i < size; ++i) {
if (x[i] != y[i]) {
return (int)(i + 1);
}
}
return 0;
}
int test_deserialize(const char * filename) {
FILE* fp = fopen(filename, "rb");
if(fp == NULL) {
printf("Could not open %s, check your configuration. \n",filename);
assert_false(fp == NULL);
}
size_t bytes;
char* input_buffer = readfile(fp, &bytes);
if(input_buffer == NULL) {
printf("Could not read bytes from %s, check your configuration. \n",filename);
assert_false(input_buffer == NULL);
}
printf("Binary content read.\n");
roaring_bitmap_t* bitmap =
roaring_bitmap_portable_deserialize_safe(input_buffer, bytes);
if(bitmap == NULL) {
printf("Null bitmap loaded.\n");
free(input_buffer);
return 1; // this is the expected behavior
}
printf("Non-null bitmap loaded.\n");
size_t expected_size = roaring_bitmap_portable_size_in_bytes(bitmap);
char* output_buffer = malloc(expected_size);
size_t actual_size =
roaring_bitmap_portable_serialize(bitmap, output_buffer);
if(actual_size != expected_size) {
free(input_buffer);
free(output_buffer);
assert_int_equal(actual_size, expected_size);
}
int compare_result = compare(input_buffer, output_buffer, actual_size);
free(output_buffer);
free(input_buffer);
fclose(fp);
roaring_bitmap_free(bitmap);
assert_false(compare_result);
return compare_result;
}
void test_robust_deserialize1() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput1.bin");
test_deserialize(filename);
}
void test_robust_deserialize2() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput2.bin");
test_deserialize(filename);
}
void test_robust_deserialize3() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput3.bin");
test_deserialize(filename);
}
void test_robust_deserialize4() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput4.bin");
test_deserialize(filename);
}
void test_robust_deserialize5() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput5.bin");
test_deserialize(filename);
}
void test_robust_deserialize6() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput6.bin");
test_deserialize(filename);
}
void test_robust_deserialize7() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput7.bin");
test_deserialize(filename);
}
void test_robust_deserialize8() {
char filename[1024];
strcpy(filename, TEST_DATA_DIR);
strcat(filename, "crashproneinput8.bin");
test_deserialize(filename);
}
int main() {
const struct CMUnitTest tests[] = {
cmocka_unit_test(test_robust_deserialize1),
cmocka_unit_test(test_robust_deserialize2),
cmocka_unit_test(test_robust_deserialize3),
cmocka_unit_test(test_robust_deserialize4),
cmocka_unit_test(test_robust_deserialize5),
cmocka_unit_test(test_robust_deserialize6),
cmocka_unit_test(test_robust_deserialize7),
};
return cmocka_run_group_tests(tests, NULL, NULL);
}
|
