make the dns authoritative check code stronger

This commit is contained in:
Arthur Vuillard
2024-07-16 18:18:53 +02:00
parent cc67f56d66
commit 813b63db7a
+8 -1
View File
@@ -199,9 +199,16 @@ def get_authoritative_server(domain, resolver=None):
while True:
try:
answers = dns.resolver.resolve(dnsname, 'NS')
return str(answers[0].target)
if answers:
first_answer = answers[0]
if hasattr(first_answer, "target"):
return str(first_answer.target)
if hasattr(first_answer, "address"):
return str(first_answer.address)
except dns.resolver.NoAnswer as e:
dnsname = dnsname.parent()
else:
dnsname = dnsname.parent()
def get_authoritative_ip(domain, resolver=None):