DNS timeout improvements

initstring · Apr 6, 2020 · 1116b38 · 1116b38
1 parent 8e2381b
commit 1116b38
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/enum_tools/utils.py b/enum_tools/utils.py
@@ -107,14 +107,18 @@ def dns_lookup(nameserver, name):
     by the fast_dns_lookup function.
     """
     res = dns.resolver.Resolver()
+    res.timeout = 10
     res.nameservers = [nameserver]
 
     try:
         res.query(name)
         # If no exception is thrown, return the valid name
         return name
-    except (dns.resolver.NXDOMAIN, dns.exception.Timeout):
+    except dns.resolver.NXDOMAIN:
         return ''
+    except dns.exception.Timeout:
+        print("    [!] DNS Timeut on {}. Investigate if there are many"
+              " of these.".format(name))
 
 def fast_dns_lookup(names, nameserver, callback='', threads=5):
     """