Merge pull request #14 from MattKeeley/bug-Crash-on-SOA-Resolver

bug fix: Crash on SOA Resolver

libs/dns.py

@@ -5,7 +5,10 @@ def get_soa_record(domain):
     """Returns the SOA record of a given domain."""
     resolver = dns.resolver.Resolver()
     resolver.nameservers = ['1.1.1.1']
-    query = resolver.resolve(domain, 'SOA')
+    try:
+        query = resolver.resolve(domain, 'SOA')
+    except:
+        return None
     if query:
         for data in query:
             dns_server = str(data.mname)

test.py

@@ -42,5 +42,8 @@ class TestSpoofy(unittest.TestCase):
     def test_spoofing_not_possible(self):
         unittest.TestCase().assertEqual(logic.is_spoofable('test_8.com', 'none', 's', 'v=spf1 include:fakest.domain.com', '~all', 1, 'quarantine', 100), 8)
 
+    def test_possible_bug_fix1(self):
+        unittest.TestCase().assertEqual(logic.is_spoofable('sub.test_9.com', None, None, None, None, None, None, None), 0)
+
 if __name__ == '__main__':
     unittest.main()