Add support for specifying DNSTimeout (bumps required dkimpy version to 1.0)

This commit is contained in:
Scott Kitterman
2020-01-03 16:14:08 -05:00
parent 042005b38d
commit 8a2760531b
8 changed files with 65 additions and 8 deletions
+3 -3
View File
@@ -88,7 +88,7 @@ class Session(object):
result = self.dns(cname, qtype, cnames=cnames)
return result
def DNSLookup_pydns(name, qtype, tcpfallback=True, timeout=30):
def DNSLookup_pydns(name, qtype, tcpfallback=True, timeout=5):
try:
# FIXME: To be thread safe, we create a fresh DnsRequest with
# each call. It would be more efficient to reuse
@@ -114,11 +114,11 @@ def DNSLookup_pydns(name, qtype, tcpfallback=True, timeout=30):
except IOError as x:
raise DNS.DNSError('DNS: ' + str(x))
def DNSLookup_dnspython(name,qtype,tcpfallback=True,timeout=30):
def DNSLookup_dnspython(name,qtype,tcpfallback=True,timeout=5):
retVal = []
try:
# FIXME: how to disable TCP fallback in dnspython if not tcpfallback?
answers = dns.resolver.query(name, qtype)
answers = dns.resolver.query(name, qtype, raise_on_no_answer=False, lifetime=timeout)
for rdata in answers:
if qtype == 'A' or qtype == 'AAAA':
retVal.append(((name, qtype), rdata.address))