Ignore garbage queries

dorp007 · Oct 19, 2021 · ea1ea01 · ea1ea01
1 parent 0b388dc
commit ea1ea01
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/channel_dns.py b/channel_dns.py
@@ -15,6 +15,7 @@
 import math
 import base64
 import re
+from exceptions import UnicodeDecodeError
 
 class DNSServerFactory(server.DNSServerFactory, object):
     def handleQuery(self, message, protocol, address):
@@ -30,7 +31,12 @@ def handleQuery(self, message, protocol, address):
         else:
             src_ip = protocol.transport.socket.getpeername()[0]
 
-        log.info('Query: {} sent {}'.format(src_ip, query))
+        try:
+            log.info('Query: {} sent {}'.format(src_ip, query))
+        except UnicodeDecodeError:
+            # Invalid query
+            return None
+
         return self.resolver.query(query, src_ip).addCallback(
             self.gotResolverResponse, protocol, message, address
         ).addErrback(