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
|
#!/usr/bin/env python3
''' Check 'stats' query module functionality. '''
import os
import random
from dnstest.libknot import libknot
from dnstest.module import ModStats
from dnstest.test import Test
from dnstest.utils import *
def check_item(server, section, item, value, idx=None, zone=None):
try:
ctl = libknot.control.KnotCtl()
ctl.connect(os.path.join(server.dir, "knot.sock"))
if zone:
ctl.send_block(cmd="zone-stats", section=section, item=item, zone=zone.name)
else:
ctl.send_block(cmd="stats", section=section, item=item)
stats = ctl.receive_stats()
finally:
ctl.send(libknot.control.KnotCtlType.END)
ctl.close()
if not stats and value == -1:
return
if zone:
stats = stats.get("zone").get(zone.name.lower())
if idx:
if value == -1:
isset(idx not in stats.get(section).get(item), idx)
return
else:
data = int(stats.get(section).get(item).get(idx))
else:
data = int(stats.get(section).get(item))
compare(data, value, "%s.%s" % (section, item))
ModStats.check()
proto = random.choice([4, 6])
t = Test(stress=False, tsig=False, address=proto)
knot = t.server("knot")
zones = t.zone("example.") + t.zone(".")
t.link(zones, knot)
knot.add_module(None, ModStats())
knot.add_module(zones[0], ModStats())
knot.add_module(zones[1], ModStats())
t.start()
knot.zones_wait(zones)
knot.ctl("-f reload") # Reset module statistics after wait!
check_item(knot, "server", "zone-count", 2)
check_item(knot, "zone", "max-ttl", 3600, zone=knot.zones["example."])
check_item(knot, "zone", "max-ttl", 518400, zone=knot.zones["."])
xdp = knot.xdp_port is not None and knot.xdp_port > 0
xdp_str = "-xdp" if xdp else ""
resp = knot.dig(zones[0].name, "SOA", tries=1, udp=True, xdp=xdp)
query_size1 = resp.query_size()
reply_size1 = resp.response_size()
resp = knot.dig(zones[0].name, "NS", tries=1, udp=False, xdp=xdp)
query_size2 = resp.query_size()
reply_size2 = resp.response_size()
resp = knot.dig(zones[1].name, "TYPE11", tries=1, udp=True, xdp=xdp)
query_size3 = resp.query_size()
reply_size3 = resp.response_size()
# Successful transfer.
resp = knot.dig(zones[0].name, "AXFR", tries=1, xdp=xdp)
resp.check_xfr(rcode="NOERROR")
xfr_query_size = resp.query_size()
# Cannot get xfr_reply_size :-/
# Check request protocol metrics.
check_item(knot, "mod-stats", "request-protocol", 2, "udp%s" % proto + xdp_str)
check_item(knot, "mod-stats", "request-protocol", 1, "udp%s" % proto + xdp_str, zone=zones[0])
check_item(knot, "mod-stats", "request-protocol", 1, "udp%s" % proto + xdp_str, zone=zones[1])
check_item(knot, "mod-stats", "request-protocol", 2, "tcp%s" % proto + xdp_str)
check_item(knot, "mod-stats", "request-protocol", 2, "tcp%s" % proto + xdp_str, zone=zones[0])
# Successful update.
up = knot.update(zones[1], allow_knsupdate=False)
up.add(zones[1].name, "3600", "AAAA", "::1")
up.send("NOERROR")
ddns_query_size = up.query_size()
# Due to DDNS bulk processing, failed RCODE and response-bytes are not incremented!
# Check request/response bytes metrics.
check_item(knot, "mod-stats", "request-bytes", query_size1 + query_size2 + query_size3,
"query")
check_item(knot, "mod-stats", "request-bytes", ddns_query_size, "update")
check_item(knot, "mod-stats", "request-bytes", xfr_query_size, "other")
check_item(knot, "mod-stats", "response-bytes", reply_size1 + reply_size2 + reply_size3,
"reply")
check_item(knot, "mod-stats", "request-bytes", query_size1 + query_size2, "query",
zone=zones[0])
check_item(knot, "mod-stats", "response-bytes", reply_size1 + reply_size2, "reply",
zone=zones[0])
check_item(knot, "mod-stats", "request-bytes", query_size3, "query", zone=zones[1])
check_item(knot, "mod-stats", "response-bytes", reply_size3, "reply", zone=zones[1])
# Check query size metrics (just for global module).
indices = dict()
for size in [query_size1, query_size2, query_size3]:
idx = "%i-%i" % (int(size / 16) * 16, int(size / 16) * 16 + 15)
if idx not in indices:
indices[idx] = 1
else:
indices[idx] += 1
for size in indices:
check_item(knot, "mod-stats", "query-size", indices[size], idx=size)
# Check reply size metrics (just for global module).
indices = dict()
for size in [reply_size1, reply_size2, reply_size3]:
idx = "%i-%i" % (int(size / 16) * 16, int(size / 16) * 16 + 15)
if idx not in indices:
indices[idx] = 1
else:
indices[idx] += 1
for size in indices:
check_item(knot, "mod-stats", "reply-size", indices[size], idx=size)
# Check query type metrics.
check_item(knot, "mod-stats", "query-type", 1, idx="SOA")
check_item(knot, "mod-stats", "query-type", 1, idx="NS")
check_item(knot, "mod-stats", "query-type", 1, idx="TYPE11")
check_item(knot, "mod-stats", "query-type", 1, idx="SOA", zone=zones[0])
check_item(knot, "mod-stats", "query-type", 1, idx="NS", zone=zones[0])
check_item(knot, "mod-stats", "query-type", -1, idx="TYPE11", zone=zones[0])
check_item(knot, "mod-stats", "query-type", -1, idx="SOA", zone=zones[1])
check_item(knot, "mod-stats", "query-type", -1, idx="NS", zone=zones[1])
check_item(knot, "mod-stats", "query-type", 1, idx="TYPE11", zone=zones[1])
# Check server operation metrics.
check_item(knot, "mod-stats", "server-operation", 3, idx="query")
check_item(knot, "mod-stats", "server-operation", 1, idx="axfr")
check_item(knot, "mod-stats", "server-operation", 1, idx="update")
# Check response code metrics.
check_item(knot, "mod-stats", "response-code", 4, idx="NOERROR")
check_item(knot, "mod-stats", "response-code", 3, idx="NOERROR", zone=zones[0])
check_item(knot, "mod-stats", "response-code", 1, idx="NOERROR", zone=zones[1])
# Check nodata metrics.
check_item(knot, "mod-stats", "reply-nodata", 1, idx="other")
check_item(knot, "mod-stats", "reply-nodata", -1, idx="other", zone=zones[0])
check_item(knot, "mod-stats", "reply-nodata", 1, idx="other", zone=zones[1])
t.end()
|
