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
|
#include "sntptest.h"
extern "C" {
#include "kod_management.h"
};
class kodDatabaseTest : public sntptest {
protected:
virtual void SetUp() {
kod_init_kod_db("/dev/null", TRUE);
}
};
TEST_F(kodDatabaseTest, SingleEntryHandling) {
char HOST[] = "192.0.2.5";
char REASON[] = "DENY";
add_entry(HOST, REASON);
kod_entry* result;
EXPECT_EQ(1, search_entry(HOST, &result));
EXPECT_STREQ(HOST, result->hostname);
EXPECT_STREQ(REASON, result->type);
}
TEST_F(kodDatabaseTest, MultipleEntryHandling) {
char HOST1[] = "192.0.2.3";
char REASON1[] = "DENY";
char HOST2[] = "192.0.5.5";
char REASON2[] = "RATE";
char HOST3[] = "192.0.10.1";
char REASON3[] = "DENY";
add_entry(HOST1, REASON1);
add_entry(HOST2, REASON2);
add_entry(HOST3, REASON3);
kod_entry* result;
EXPECT_EQ(1, search_entry(HOST1, &result));
EXPECT_STREQ(HOST1, result->hostname);
EXPECT_STREQ(REASON1, result->type);
EXPECT_EQ(1, search_entry(HOST2, &result));
EXPECT_STREQ(HOST2, result->hostname);
EXPECT_STREQ(REASON2, result->type);
EXPECT_EQ(1, search_entry(HOST3, &result));
EXPECT_STREQ(HOST3, result->hostname);
EXPECT_STREQ(REASON3, result->type);
free(result);
}
TEST_F(kodDatabaseTest, NoMatchInSearch) {
char HOST_ADD[] = "192.0.2.6";
char HOST_NOTADD[] = "192.0.6.1";
char REASON[] = "DENY";
add_entry(HOST_ADD, REASON);
kod_entry* result;
EXPECT_EQ(0, search_entry(HOST_NOTADD, &result));
EXPECT_TRUE(result == NULL);
}
TEST_F(kodDatabaseTest, AddDuplicate) {
char HOST[] = "192.0.2.3";
char REASON1[] = "RATE";
char REASON2[] = "DENY";
add_entry(HOST, REASON1);
kod_entry* result1;
ASSERT_EQ(1, search_entry(HOST, &result1));
/*
* Sleeps for two seconds since we want to ensure that
* the timestamp is updated to a new value.
*/
sleep(2);
add_entry(HOST, REASON2);
kod_entry* result2;
ASSERT_EQ(1, search_entry(HOST, &result2));
EXPECT_NE(result1->timestamp, result2->timestamp);
free(result1);
free(result2);
}
TEST_F(kodDatabaseTest, DeleteEntry) {
char HOST1[] = "192.0.2.1";
char HOST2[] = "192.0.2.2";
char HOST3[] = "192.0.2.3";
char REASON[] = "DENY";
add_entry(HOST1, REASON);
add_entry(HOST2, REASON);
add_entry(HOST3, REASON);
kod_entry* result;
ASSERT_EQ(1, search_entry(HOST2, &result));
free(result);
delete_entry(HOST2, REASON);
EXPECT_EQ(0, search_entry(HOST2, &result));
// Ensure that the other entry is still there.
EXPECT_EQ(1, search_entry(HOST1, &result));
free(result);
}
|
