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

strip additional records before truncating the message

Browse Source
This commit is contained in:
Michael Kuron
2021-02-28 22:46:29 +01:00
parent be42b16652
commit 9cd06d86cf
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy.py
+3 -1
View File
@@ -141,8 +141,10 @@ class DynamicResolver(object):
class TruncatingDNSDatagramProtocol(dns.DNSDatagramProtocol): class TruncatingDNSDatagramProtocol(dns.DNSDatagramProtocol):
def writeMessage(self, message, address): def writeMessage(self, message, address):
if type(message) is dns.Message and len(message.toStr()) > 512: if type(message) is dns.Message and len(message.toStr()) > 512:
message.additional = []
if len(message.toStr()) > 512:
message.trunc = 1 message.trunc = 1
message.answers = message.additional = [] message.answers = []
dns.DNSDatagramProtocol.writeMessage(self, message, address) dns.DNSDatagramProtocol.writeMessage(self, message, address)
def main(): def main():