wfpayload, fix non fuzzres payload

hack-umbrella · Apr 10, 2019 · ae012c5 · ae012c5
1 parent f46009f
commit ae012c5
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/src/wfuzz/wfuzz.py b/src/wfuzz/wfuzz.py
@@ -9,6 +9,8 @@
 from .ui.console.common import help_banner2
 from .ui.console.clparser import CLParser
 
+from .fuzzobjects import FuzzResult
+
 
 def main():
     kb = None
@@ -105,19 +107,26 @@ def usage():
 
     try:
         session_options = CLParser(sys.argv).parse_cl()
-        printer = View(session_options)
-        printer.header(None)
+        printer = None
 
         for res in payload(**session_options):
             if len(res) > 1:
                 raise FuzzExceptBadOptions("wfpayload can only be used to generate one word dictionaries")
             else:
                 r = res[0]
 
-            r._description = field
+            # TODO: option to not show headers in fuzzres
+            # TODO: all should be same object
+            if isinstance(r, FuzzResult):
+                if printer is None:
+                    printer = View(session_options)
+                    printer.header(None)
 
-            printer.result(r)
-        print("")
+                if field:
+                    r._description = field
+                printer.result(r)
+            else:
+                print(r)
 
     except KeyboardInterrupt:
         pass