Merge pull request #203 from ch1nhpd/master

Update host for burp proxy

arjun/__main__.py

@@ -23,7 +23,7 @@ parser = argparse.ArgumentParser() # defines the parser
 parser.add_argument('-u', help='Target URL', dest='url')
 parser.add_argument('-o', '-oJ', help='Path for json output file.', dest='json_file')
 parser.add_argument('-oT', help='Path for text output file.', dest='text_file')
-parser.add_argument('-oB', help='Port for output to Burp Suite Proxy. Default port is 8080.', dest='burp_port', nargs='?', const=8080)
+parser.add_argument('-oB', help='Output to Burp Suite Proxy. Default is 127.0.0.1:8080.', dest='burp_proxy', nargs='?', const='127.0.0.1:8080')
 parser.add_argument('-d', help='Delay between requests in seconds. (default: 0)', dest='delay', type=float, default=0)
 parser.add_argument('-t', help='Number of concurrent threads. (default: 5)', dest='threads', type=int, default=5)
 parser.add_argument('-w', help='Wordlist file path. (default: {arjundir}/db/large.txt)', dest='wordlist', default=arjun_dir+'/db/large.txt')

arjun/core/exporter.py

@@ -17,9 +17,10 @@ def burp_export(result):
     """
     exports results to Burp Suite by sending request to Burp proxy
     """
+    proxy = ('' if ':' in mem.var['burp_proxy'] else '127.0.0.1:') + mem.var['burp_proxy']
     proxies = {
-        'http': 'http://127.0.0.1:' + mem.var['burp_port'],
-        'https': 'https://127.0.0.1:' + mem.var['burp_port']
+        'http': 'http://' + proxy,
+        'https': 'https://' + proxy
     }
     for url, data in result.items():
         if data['method'] == 'GET':
@@ -55,5 +56,5 @@ def exporter(result):
         json_export(result)
     if mem.var['text_file']:
         text_export(result)
-    if mem.var['burp_port']:
+    if mem.var['burp_proxy']:
         burp_export(result)