zoc-me/mdns-discovery-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

TXT

Browse Source
This commit is contained in:
Michael Kuron
2020-08-02 22:20:51 +02:00
parent 2a260313de
commit d244c7caec
1 changed files with 19 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy.py
+15 -30
View File
@@ -9,10 +9,13 @@ from twisted.python import log
domain = sys.argv[1]
port = int(sys.argv[2])
ttl = 120
ttl = 0 # TODO 120
timeout = 2
class DynamicResolver(object):
def __init__(self):
self.zeroconf = Zeroconf(ip_version=4)
def _dynamicResponseRequired(self, query):
if str(query.name).endswith(domain):
return True
@@ -31,40 +34,28 @@ class DynamicResolver(object):
if state_change is ServiceStateChange.Added:
services.append(name)
zeroconf = Zeroconf(ip_version=4) # TODO: move out?
sb = ServiceBrowser(zeroconf, localname, [handler])
print("waiting for", localname) # TODO: remove
sb = ServiceBrowser(self.zeroconf, localname, [handler])
time.sleep(timeout)
sb.cancel()
print("done waiting for", localname) # TODO: remove
answers = []
for service in services:
answers.append(dns.RRHeader(name=query.name.name, ttl=ttl, type=dns.PTR, payload=dns.Record_PTR(
name=service[:-6] + domain
)))
return answers
return answers, [], []
def txt(localname):
zeroconf = Zeroconf(ip_version=4) # TODO: move out?
print("waiting for txt", localname) # TODO: remove
info = zeroconf.get_service_info(localname, localname)
print("done waiting for txt", localname) # TODO: remove
info = self.zeroconf.get_service_info(localname, localname)
if info.text == b"\0":
answers = []
else:
data = [b"%s=%s" % (p, info.properties[p]) for p in info.properties]
answers = [dns.RRHeader(name=query.name.name, ttl=ttl, type=dns.TXT, payload=dns.Record_TXT(
info.text # TODO: something wrong here?
*data
))]
print(answers, type(info.text), info.text)
return answers
return answers, [], []
def srv(localname):
zeroconf = Zeroconf(ip_version=4) # TODO: move out?
print("waiting for srv", localname)
info = zeroconf.get_service_info(localname, localname)
print("done waiting for srv", localname)
info = self.zeroconf.get_service_info(localname, localname)
answers = [dns.RRHeader(name=query.name.name, ttl=ttl, type=dns.SRV, payload=dns.Record_SRV(
info.priority, info.weight, info.port, info.server[:-6] + domain
@@ -72,7 +63,7 @@ class DynamicResolver(object):
print(answers)
return answers
return answers, [], []
def host(localname):
answers = [dns.RRHeader(name=query.name.name, ttl=ttl, type=dns.A, payload=dns.Record_A(
@@ -81,31 +72,25 @@ class DynamicResolver(object):
print(answers)
return answers
return answers, [], []
d = defer.Deferred()
if query.type == dns.PTR:
d = threads.deferToThread(browse, localname)
d.addCallback(lambda a: (a, [], []))
return d
elif query.type == dns.TXT:
d = threads.deferToThread(txt, localname)
d.addCallback(lambda a: (a, [], []))
return d
elif query.type == dns.SRV:
d = threads.deferToThread(srv, localname)
d.addCallback(lambda a: (a, [], []))
return d
elif query.type == dns.A:
d = threads.deferToThread(host, localname)
d.addCallback(lambda a: (a, [], []))
return d
else:
answers = []
d.callback((answers, [], []))
if query.type != dns.AAAA:
elif query.type != dns.AAAA:
print("Unsupported request", query)
d.callback(([], [], []))
return d
def query(self, query, timeout=None):