Added recursive_query function, fixes #6

.gitignore

@@ -64,3 +64,5 @@ target/
 
 #Ipython Notebook
 .ipynb_checkpoints
+
+.idea/

fierce.py

@@ -112,6 +112,21 @@ def reverse_query(resolver, ip):
     return query(resolver, dns.reversename.from_address(ip), record_type='PTR')
 
 
+def recursive_query(resolver, domain, record_type='NS'):
+
+    query_domain = str(domain)
+    query_response = None
+    try:
+        while query_response is None:
+            query_response = query(resolver, query_domain, record_type)
+            query_domain = query_domain.split('.' , 1)[1]
+    except IndexError:
+        return None
+
+    return query_response
+
+
+
 def zone_transfer(address, domain):
     try:
         return dns.zone.from_xfr(dns.query.xfr(address, domain))
@@ -185,15 +200,23 @@ def fierce(**kwargs):
     if not domain.is_absolute():
         domain = domain.concatenate(dns.name.root)
 
-    ns = query(resolver, domain, record_type='NS')
-    domain_name_servers = [n.to_text() for n in ns]
-    print("NS: {}".format(" ".join(domain_name_servers)))
 
-    soa = query(resolver, domain, record_type='SOA')
-    soa_mname = soa[0].mname
-    master = query(resolver, soa_mname, record_type='A')
-    master_address = master[0].address
-    print("SOA: {} ({})".format(soa_mname, master_address))
+    ns = recursive_query(resolver, domain, 'NS')
+
+    if ns:
+        domain_name_servers = [n.to_text() for n in ns]
+    print("NS: {}".format(" ".join(domain_name_servers) if ns else "failure"))
+
+    soa = recursive_query(resolver, domain, record_type='SOA')
+    if soa:
+        soa_mname = soa[0].mname
+        master = query(resolver, soa_mname, record_type='A')
+        master_address = master[0].address
+        print("SOA: {} ({})".format(soa_mname, master_address))
+    else:
+        print("SOA: failure")
+        print("Failed to lookup NS/SOA, Domain does not exist")
+        exit(-1)
 
     zone = zone_transfer(master_address, domain)
     print("Zone: {}".format("success" if zone else "failure"))