diff --git a/arjun/__main__.py b/arjun/__main__.py index 89ba84c..1dd14fd 100644 --- a/arjun/__main__.py +++ b/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') diff --git a/arjun/core/exporter.py b/arjun/core/exporter.py index d45604e..529d736 100644 --- a/arjun/core/exporter.py +++ b/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)