fix bug in consolidate_dns

1 files changed, 7 insertions, 1 deletions

diff --git a/repos/consolidate_dns/index b/repos/consolidate_dns/index
index c66e54b..71a51c6 100755
--- a/repos/consolidate_dns/index
+++ b/repos/consolidate_dns/index
@@ -44,6 +44,8 @@ if dnsrecon_enabled:
   p.wait()
   reader = csv.reader(open(dnsrecon_tmp))
   for row in reader:
+    if not row[1] in domains: 
+      domains[row[1]] = []
     domains[row[1]] += row[2:]
   print ("...finished with [%d] domains" %reader.line_num)
   os.unlink(dnsrecon_tmp)
@@ -55,7 +57,9 @@ if gxfr_enabled:
   p.wait()
   reader = csv.reader(open(gxfr_tmp))
   for row in reader:
-      domains[row[0]] += row[1:]
+    if not row[0] in domains: 
+      domains[row[0]] = []
+    domains[row[0]] += row[1:]
   print ("...finished with [%d] domains" %reader.line_num)
   os.unlink(gxfr_tmp)
 if bxfr_enabled:
@@ -65,6 +69,8 @@ if bxfr_enabled:
   p.wait()
   reader = csv.reader(open(bxfr_tmp))
   for row in reader:
+    if not row[0] in domains: 
+      domains[row[0]] = []
     domains[row[0]] += row[1:]
   print ("...finished with [%d] domains" %reader.line_num)
   os.unlink(bxfr_tmp)