Import Urls from Burp Suite - Feature

core/utils.py

@@ -2,7 +2,7 @@
 import tld
 import json
 import tempfile
-
+import csv
 
 def host(string):
     if string and '*' not in string:
@@ -67,3 +67,13 @@ def extractHeaders(headers):
         except IndexError:
             pass
     return sorted_headers
+
+def url_from_logs(inp_file):
+    urls = []
+    with open(inp_file, 'r') as file:
+        log_csv = csv.DictReader(file)
+        for line in log_csv:
+            url = line['Host']+line['Path']
+            if url not in urls:
+                urls.append(url)
+    return urls

corsy.py

@@ -6,7 +6,7 @@
 import argparse
 
 from core.tests import active_tests
-from core.utils import host, prompt, format_result, create_url_list
+from core.utils import host, prompt, format_result, create_url_list, url_from_logs
 from core.colors import bad, end, red, run, good, grey, green, white, yellow
 
 
@@ -29,6 +29,7 @@
 parser.add_argument('-t', help='thread count', dest='threads', type=int, default=2)
 parser.add_argument('-d', help='request delay', dest='delay', type=float, default=0)
 parser.add_argument('-q', help='don\'t print help tips', dest='quiet', action='store_true')
+parser.add_argument('-f', help='import from burp logs', dest='file')
 parser.add_argument('--headers', help='add headers', dest='header_dict', nargs='?', const=True)
 args = parser.parse_args()
 
@@ -39,6 +40,7 @@
 inp_file = args.inp_file
 json_file = args.json_file
 header_dict = args.header_dict
+log_file = args.file
 
 if type(header_dict) == bool:
     header_dict = extractHeaders(prompt())
@@ -54,7 +56,10 @@
         'Connection': 'close',
     }
 
-urls = create_url_list(target, inp_file)
+if inp_file:
+    urls = create_url_list(target, inp_file)
+elif log_file:
+    urls = url_from_logs(log_file)
 
 def cors(target, header_dict, delay):
     url = target