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
|
// -*- mode: c++; c-basic-offset: 2; -*-
#include "common.hh"
#include "test.hh"
#include "package.hh"
namespace {
void setup(Package* pkg) {
pkg->id = 42;
pkg->timestamp.tv_sec = 123;
pkg->timestamp.tv_nsec = 999999999;
pkg->source_port = 0;
pkg->source_host = "source";
pkg->target_port = 65535;
pkg->target_host = "target";
}
bool pkg_eq(Package const& p1, Package const& p2) {
return p1.id == p2.id
&& p1.timestamp.tv_sec == p2.timestamp.tv_sec
&& p1.timestamp.tv_nsec == p2.timestamp.tv_nsec
&& p1.source_port == p2.source_port
&& p1.source_host.compare(p2.source_host) == 0
&& p1.target_port == p2.target_port
&& p1.target_host.compare(p2.target_host) == 0;
}
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;
}
|
