resistant to error's, DNSSEC done

altf4arnold · Feb 26, 2024 · 06ef8ab · 06ef8ab
1 parent 644cc45
commit 06ef8ab
Showing 1 changed file with 24 additions and 6 deletions.
diff --git a/tldtester/sorter.py b/tldtester/sorter.py
@@ -48,23 +48,41 @@ def grabber(data):
         dnsseckeys = []
         Arecords = 0
         AAAArecords = 0
-        ns = dns.resolver.resolve(tld, 'NS')
-        for server in ns:
-            nsservers.append(server.to_text())
+        try:
+            ns = dns.resolver.resolve(tld, 'NS')
+            for server in ns:
+                nsservers.append(server.to_text())
+        except Exception as e:
+            print(e)
         for Arecord in nsservers:
             try:
-                dns.resolver.resolve(Arecord, 'A')
+                try:
+                    dns.resolver.resolve(Arecord, 'A')
+                except Exception as e:
+                    # retry
+                    print(e)
+                    dns.resolver.resolve(Arecord, 'A')
                 Arecords += 1
             except Exception as e:
                 print(e)
         for AAAArecord in nsservers:
             try:
-                dns.resolver.resolve(AAAArecord, 'AAAA')
+                try:
+                    dns.resolver.resolve(AAAArecord, 'AAAA')
+                except Exception as e:
+                    # retry
+                    print(e)
+                    dns.resolver.resolve(AAAArecord, 'AAAA')
                 AAAArecords += 1
             except Exception as e:
                 print(e)
         try:
-            ds = dns.resolver.resolve(tld, 'DS')
+            try:
+                ds = dns.resolver.resolve(tld, 'DS')
+            except Exception as e:
+                # retry
+                print(e)
+                ds = dns.resolver.resolve(tld, 'DS')
             for dsrecord in ds:
                 algo = dsrecord.to_text()
                 line = algo.split()