Difrex/surok
Difrex/surok
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Server IPs added

Browse Source
This commit is contained in:
Denis Zheleztsov 2016-11-16 14:51:57 +03:00
parent 5cd3f7c9b8
commit fa7427d993
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
surok/discovery.py
View File

@ -42,8 +42,9 @@ def resolve(app, conf):
return hosts return hosts
# Do SRV queries # Do DNS queries
# Return array: [{"name": "f.q.d.n", "port": 8876}] # Return array:
# [{"name": "f.q.d.n", "port": 8876, "ip": ["10.10.10.1", "10.10.10.2"]}]
def do_query(fqdn, loglevel): def do_query(fqdn, loglevel):
servers = [] servers = []
try: try:
@ -51,14 +52,18 @@ def do_query(fqdn, loglevel):
resolver.lifetime = 1 resolver.lifetime = 1
resolver.timeout = 1 resolver.timeout = 1
query = resolver.query(fqdn, 'SRV') query = resolver.query(fqdn, 'SRV')
for rdata in query: for rdata in query:
info = str(rdata).split() info = str(rdata).split()
server = {'name': info[3][:-1], 'port': info[2]} name = info[3][:-1]
port = info[2]
server = {'name': name, 'port': port, 'ip': []}
a_query = resolver.query(name, 'A')
for a_rdata in a_query:
server['ip'].append(a_rdata)
servers.append(server) servers.append(server)
except DNSException as e: except DNSException as e:
if loglevel != 'info': if loglevel != 'info':
error("Could not resolve " + fqdn + ': ' + str(e)) error("Could not resolve " + fqdn)
return servers return servers