Fixed bug with oN when --fuzzy-logic not true

Author: Michael Skelton

VHostScan.py

@@ -81,7 +81,7 @@ def main():
                                     arguments.unique_depth, arguments.ignore_http_codes, arguments.ignore_content_length, arguments.fuzzy_logic, arguments.add_waf_bypass_headers)
 
     scanner.scan()
-    output = output_helper(scanner)
+    output = output_helper(scanner, arguments)
 
     print(output.output_normal_likely())
 

lib/helpers/output_helper.py

@@ -7,16 +7,26 @@
 
 
 class output_helper(object):
-    def __init__(self, scanner):
+    def __init__(self, scanner, arguments):
         self.scanner = scanner
+        self.arguments = arguments
 
     def write_normal(self, filename):
 
         file = file_helper(filename)
 
         # todo: finish check_directory (needs regex to split out filename)
         # file.check_directory(filename)
-        file.write_file(self.generate_header() + self.output_normal_likely() + self.output_fuzzy() + self.output_normal_detail())
+
+        output = self.generate_header()
+        output += self.output_normal_likely()
+        
+        if(self.arguments.fuzzy_logic):
+            output += self.output_fuzzy()
+        
+        output += self.output_normal_detail()
+        
+        file.write_file(output)
 
     def output_normal_likely(self):
         uniques = False
@@ -34,7 +44,7 @@ def output_normal_likely(self):
 
 
     def output_fuzzy(self):
-        output = "\n[+] Match similarity using fuzzy logic:"
+        output = "\n\n[+] Match similarity using fuzzy logic:"
         request_hashes = {}
 
         for host in self.scanner.hosts: