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
|
#!/usr/bin/env python3
"""Simple program for exposing statistics from Knot DNS over HTTP/HTTPS."""
import http.server
import libknot.control
import json
import ssl
import time
# Configuration.
#libknot.control.load_lib("../src/.libs/libknot.so")
ctl_socket = "/tmp/knot.sock"
ctl_timeout = 2
ctl_flags = "" # set "F" for all supported counters.
http_host = "127.0.0.1"
http_port = 8080
ssl_enable = False
ssl_keyfile = "./mykey.key"
ssl_certfile = "./mycert.crt"
class StatsServer(http.server.BaseHTTPRequestHandler):
def do_GET(self):
self.send_response(200)
self.send_header("Content-type", "text/html")
self.end_headers()
# Connect to Knot server.
ctl = libknot.control.KnotCtl()
ctl.connect(ctl_socket)
ctl.set_timeout(ctl_timeout)
# Get global metrics.
global_stats = dict()
try:
ctl.send_block(cmd="stats", flags=ctl_flags)
global_stats = ctl.receive_stats()
except:
pass
# Get zone metrics.
zone_stats = dict()
try:
ctl.send_block(cmd="zone-stats", flags=ctl_flags)
zone_stats = ctl.receive_stats()
except:
pass
# Disconnect from the server.
ctl.send(libknot.control.KnotCtlType.END)
ctl.close()
# Publish the stats.
stats = {**global_stats, **zone_stats}
self.wfile.write(bytes(json.dumps(stats, indent=4, sort_keys=True), "utf-8"))
httpd = http.server.HTTPServer((http_host, http_port), StatsServer)
if ssl_enable:
httpd.socket = ssl.wrap_socket(httpd.socket, keyfile=ssl_keyfile,
certfile=ssl_certfile, server_side=True)
print("%s: HTTP%s Server Start - %s:%s" %
(time.asctime(), "S" if ssl_enable else "", http_host, http_port))
try:
httpd.serve_forever()
except KeyboardInterrupt:
pass
httpd.server_close()
|
