strip additional records before truncating the message

2021-02-28 22:46:29 +01:00
parent be42b16652
commit 9cd06d86cf
1 changed files with 4 additions and 2 deletions
@@ -141,8 +141,10 @@ class DynamicResolver(object):
 class TruncatingDNSDatagramProtocol(dns.DNSDatagramProtocol):
    def writeMessage(self, message, address):
        if type(message) is dns.Message and len(message.toStr()) > 512:
+            message.additional = []
+            if len(message.toStr()) > 512:
                message.trunc = 1
-            message.answers = message.additional = []
+                message.answers = []
        dns.DNSDatagramProtocol.writeMessage(self, message, address)

 def main():