Closes 15. Add JSON output support

Author: xuvez

VHostScan.py

@@ -34,6 +34,7 @@ def main():
     parser.add_argument("--rate-limit", dest="rate_limit", type=int, help='Amount of time in seconds to delay between each scan (default 0).', default=0)
     parser.add_argument("--waf", dest="add_waf_bypass_headers",   action="store_true", help="If set then simple WAF bypass headers will be sent.", default=False)
     parser.add_argument("-oN",   dest="output_normal", help="Normal output printed to a file when the -oN option is specified with a filename argument." )
+    parser.add_argument("-oJ", dest="output_json", help="JSON output printed to a file when the -oJ option is specified with a filename argument." )
     parser.add_argument("-", dest="stdin", action="store_true", help="By passing a blank '-' you tell VHostScan to expect input from stdin (pipe).", default=False)
 
     arguments = parser.parse_args()    
@@ -104,6 +105,10 @@ def main():
         output.write_normal(arguments.output_normal)
         print("\n[+] Writing normal ouptut to %s" % arguments.output_normal)
 
+    if(arguments.output_json):
+        output.output_json(arguments.output_json)
+        print("\n[+] Writing json ouptut to %s" % arguments.output_json)
+
 
 if __name__ == "__main__":
     main()

lib/helpers/output_helper.py

@@ -4,6 +4,7 @@
 from fuzzywuzzy import fuzz
 import itertools
 import numpy as np
+import json
 
 
 class output_helper(object):
@@ -43,6 +44,21 @@ def output_normal_likely(self):
             return "\n[!] No matches with a unique count of {} or less.".format(depth)
 
 
+    def output_json(self, filename):
+        file = file_helper(filename)
+        list = dict()
+        for host in self.scanner.hosts:
+            headers = {}
+            for header in host.keys:
+                headers[header.split(':')[0]] = header.split(':')[1].strip()
+
+            list[host.hostname] = {'Code': host.response_code,
+                                   'Hash': host.hash,
+                                   'Response': host.content,
+                                   'Headers': headers}
+        file.write_file(json.dumps(list))
+
+
     def output_fuzzy(self):
         output = "\n\n[+] Match similarity using fuzzy logic:"
         request_hashes = {}