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
|
// -*- mode: c++; c-basic-offset: 2; -*-
#include "common.hh"
#include "test.hh"
#include "test_package.hh"
namespace {
bool test_sanity() {
Package pkg1;
Package pkg2;
setup(&pkg1);
uint8_t tmp[4096];
auto size = write_package(pkg1, tmp, sizeof(tmp));
ASSERT_TRUE(size < sizeof(tmp));
ASSERT_EQ(size, read_package(&pkg2, tmp, sizeof(tmp)));
ASSERT_TRUE(pkg_eq(pkg1, pkg2));
return true;
}
bool test_overflow() {
Package pkg1;
setup(&pkg1);
uint8_t tmp[4096];
auto size = write_package(pkg1, tmp, sizeof(tmp));
for (size_t i = 0; i < size; ++i) {
memset(tmp + size, 0, sizeof(tmp) - size);
auto s2 = write_package(pkg1, tmp + size, i);
ASSERT_EQ(size, s2);
ASSERT_EQ(0, tmp[size + i]);
}
auto s2 = write_package(pkg1, tmp + size, size);
ASSERT_EQ(size, s2);
ASSERT_EQ(0, memcmp(tmp, tmp + size, size));
return true;
}
bool test_null() {
Package pkg1;
setup(&pkg1);
uint8_t tmp[4096];
auto size = write_package(pkg1, tmp, sizeof(tmp));
auto s2 = write_package(pkg1, NULL, 0);
ASSERT_EQ(size, s2);
s2 = write_package(pkg1, NULL, sizeof(tmp));
ASSERT_EQ(size, s2);
return true;
}
bool test_empty() {
Package pkg1;
Package pkg2;
setup(&pkg1);
pkg1.source_host.clear();
pkg1.target_host.clear();
uint8_t tmp[4096];
auto size = write_package(pkg1, tmp, sizeof(tmp));
ASSERT_EQ(size, read_package(&pkg2, tmp, sizeof(tmp)));
ASSERT_TRUE(pkg_eq(pkg1, pkg2));
return true;
}
} // namespace
int main(void) {
BEFORE;
RUN(test_sanity());
RUN(test_overflow());
RUN(test_null());
RUN(test_empty());
AFTER;
}
|
