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
|
/* Copyright (C) 2019 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include <assert.h>
#include <tap/basic.h>
#include "knot/zone/node.h"
#include "libknot/libknot.h"
static knot_rrset_t *create_dummy_rrset(const knot_dname_t *owner, uint16_t type)
{
knot_rrset_t *r = knot_rrset_new(owner, type, KNOT_CLASS_IN, 3600, NULL);
assert(r);
uint8_t wire[16] = { 0 };
memcpy(wire, "testtest", strlen("testtest"));
int ret = knot_rrset_add_rdata(r, wire, strlen("testtest"), NULL);
assert(ret == KNOT_EOK);
(void)ret;
return r;
}
static knot_rrset_t *create_dummy_rrsig(const knot_dname_t *owner, uint16_t type)
{
knot_rrset_t *r = knot_rrset_new(owner, KNOT_RRTYPE_RRSIG, KNOT_CLASS_IN,
3600, NULL);
assert(r);
uint8_t wire[sizeof(uint16_t)];
knot_wire_write_u16(wire, type);
int ret = knot_rrset_add_rdata(r, wire, sizeof(uint16_t), NULL);
assert(ret == KNOT_EOK);
(void)ret;
return r;
}
int main(int argc, char *argv[])
{
plan_lazy();
knot_dname_t *dummy_owner = knot_dname_from_str_alloc("test.");
// Test new
zone_node_t *node = node_new(dummy_owner, false, false, NULL);
ok(node != NULL, "Node: new");
assert(node);
ok(knot_dname_is_equal(node->owner, dummy_owner), "Node: new - set fields");
// Test RRSet addition
knot_rrset_t *dummy_rrset = create_dummy_rrset(dummy_owner, KNOT_RRTYPE_TXT);
int ret = node_add_rrset(node, dummy_rrset, NULL);
ok(ret == KNOT_EOK && node->rrset_count == 1 &&
knot_rdataset_eq(&dummy_rrset->rrs, &node->rrs[0].rrs), "Node: add RRSet.");
// Test RRSet getters
knot_rrset_t *n_rrset = node_create_rrset(node, KNOT_RRTYPE_TXT);
ok(n_rrset && knot_rrset_equal(n_rrset, dummy_rrset, true),
"Node: create existing RRSet.");
knot_rrset_free(n_rrset, NULL);
n_rrset = node_create_rrset(node, KNOT_RRTYPE_SOA);
ok(n_rrset == NULL, "Node: create non-existing RRSet.");
knot_rrset_t stack_rrset = node_rrset(node, KNOT_RRTYPE_TXT);
ok(knot_rrset_equal(&stack_rrset, dummy_rrset, true), "Node: get existing RRSet.");
stack_rrset = node_rrset(node, KNOT_RRTYPE_SOA);
ok(knot_rrset_empty(&stack_rrset), "Node: get non-existent RRSet.");
knot_rdataset_t *n_rdataset = node_rdataset(node, KNOT_RRTYPE_TXT);
ok(n_rdataset && knot_rdataset_eq(n_rdataset, &dummy_rrset->rrs),
"Node: get existing rdataset.");
n_rdataset = node_rdataset(node, KNOT_RRTYPE_SOA);
ok(n_rdataset == NULL, "Node: get non-existing rdataset.");
stack_rrset = node_rrset_at(node, 0);
ok(knot_rrset_equal(&stack_rrset, dummy_rrset, true),
"Node: get existing position.");
stack_rrset = node_rrset_at(node, 1);
ok(knot_rrset_empty(&stack_rrset), "Node: get non-existent position.");
// Test TTL mismatch
dummy_rrset->ttl = 1800;
ret = node_add_rrset(node, dummy_rrset, NULL);
ok(ret == KNOT_ETTL && node->rrset_count == 1,
"Node: add RRSet, TTL mismatch.");
knot_rrset_free(dummy_rrset, NULL);
// Test bool functions
ok(node_rrtype_exists(node, KNOT_RRTYPE_TXT), "Node: type exists.");
ok(!node_rrtype_exists(node, KNOT_RRTYPE_AAAA), "Node: type does not exist.");
ok(!node_rrtype_is_signed(node, KNOT_RRTYPE_TXT), "Node: type is not signed.");
dummy_rrset = create_dummy_rrsig(dummy_owner, KNOT_RRTYPE_TXT);
ret = node_add_rrset(node, dummy_rrset, NULL);
assert(ret == KNOT_EOK);
ok(node_rrtype_is_signed(node, KNOT_RRTYPE_TXT), "Node: type is signed.");
knot_rrset_free(dummy_rrset, NULL);
// Test remove RRset
node_remove_rdataset(node, KNOT_RRTYPE_AAAA);
ok(node->rrset_count == 2, "Node: remove non-existent rdataset.");
node_remove_rdataset(node, KNOT_RRTYPE_TXT);
ok(node->rrset_count == 1, "Node: remove existing rdataset.");
// "Test" freeing
node_free_rrsets(node, NULL);
ok(node->rrset_count == 0, "Node: free RRSets.");
node_free(node, NULL);
knot_dname_free(dummy_owner, NULL);
return 0;
}
|