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
|
#!/bin/python3
from sys import argv
import unbound
# our own module used by several scripts in the project
from ztdns_db_connectivity import start_db_connection
class dns_queries:
def __init__(self, dns_IP, dns_id, services):
self.dns_IP = dns_IP
self.dns_id = dns_id
self.services = services
class single_query:
def __init__(self, cursor, vpn_id, dns_id, service_id):
self.cursor = cursor
self.vpn_id = vpn_id
self.dns_id = dns_id
self.service_id = service_id
def query_planned_queries(cursor, hour, vpn_id):
# return [
# # dns server IP | dns server id | service_id | service_name
# dns_queries("195.98.79.117", 23, [[89, "devuan.org"],
# [44, "gry.pl"],
# [112, "invidio.us"]]),
# dns_queries("192.71.245.208", 33, [[77, "debian.org"],
# [22, "nie.ma.takiej.domeny"],
# [100, "onet.pl"]])
# ]
cursor.execute('''
SELECT DISTINCT d.IP, d.dns_id
FROM user_side_queries AS q JOIN user_side_dns AS d
ON dns.id = q.dns_id
WHERE q.vpn_id = %s
''', (vpn_id,))
dnss = cursor.fetchall()
dnss_to_query = []
for dns_IP, dns_id in dnss:
cursor.execute('''
SELECT s.id, s.name
FROM user_side_services AS s JOIN user_side_queries AS q
ON s.id = q.service_id
WHERE q.vpn_id = %s AND q.dns_id = %s
''', (vpn_id, dns_id))
queries = dns_queries(dns_IP, dns_id, cursor.fetchall())
dnss_to_query.append(queries)
return dnss_to_query
def resolve_call_back(mydata, status, result):
query = mydata
# debugging
print("callback called for {}".format(result.qname))
if status==0 and result.havedata:
print("Result:",result.data.address_list)
# write to database
query.cursor.execute('''
INSERT INTO user_side_responses (date, result, dns_id, service_id, vpn_id)
VALUES (current_timestamp, '', %s, %s, %s)
RETURNING id
''', (query.dns_id, query.service_id, query.vpn_id))
responses_id = query.cursor.fetchone()[0]
if status==0 and result.havedata:
for address in data.address_list:
query.cursor.execute('''
INSERT INTO user_side_response (returned_ip, responses_id)
VALUES(%s, %s)
''', (address, responses_id))
# no committing, since auto-commit mode is set on the connection
hour = argv[1]
vpn_id = argv[2]
connection = start_db_connection()
cursor = connection.cursor()
contexts = []
for dns_query in query_planned_queries(cursor, hour, vpn_id):
ctx = unbound.ub_ctx()
ctx.set_fwd(dns_IP)
for service_id, service_name in dns_query.services:
print("starting resolution of {} through {}".format(service_name, dns_IP))
query = single_query(cursor, vpn_id, dns_query.dns_id, service_id)
ctx.resolve_async(service_name, query, resolve_call_back,
unbound.RR_TYPE_A, unbound.RR_CLASS_IN)
contexts.append(ctx)
for ctx in contexts:
ctx.wait()
cursor.close()
connection.close()
|
