Merge pull request maurosoria#761 from wdahlenburg/reporting

Include cmdline args in reports
wind-feather · Apr 7, 2021 · 4b9e5f1 · 4b9e5f1
2 parents d856319 + e74b986
commit 4b9e5f1
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 4 deletions.
diff --git a/lib/reports/json_report.py b/lib/reports/json_report.py
@@ -18,6 +18,7 @@
 
 import json
 import time
+import sys
 
 from lib.reports import *
 
@@ -35,7 +36,10 @@ def generate(self):
         headerName = "{0}://{1}:{2}/{3}".format(
             self.protocol, self.host, self.port, self.basePath
         )
-        result = {"time": time.ctime(), headerName: []}
+
+        info = {"args": ' '.join(sys.argv), "time": time.ctime()}
+
+        result = {"info": info, headerName: []}
 
         for path, status, contentLength, redirect in self.pathList:
             entry = {

diff --git a/lib/reports/markdown_report.py b/lib/reports/markdown_report.py
@@ -18,6 +18,7 @@
 
 from lib.reports import *
 import time
+import sys
 
 
 class MarkdownReport(FileBaseReport):
@@ -35,7 +36,8 @@ def generate(self):
             self.protocol, self.host, self.port, self.basePath
         )
 
-        result = "### Time: {0}\n".format(time.ctime())
+        result = "### Args: {0}\n".format(' '.join(sys.argv))
+        result += "### Time: {0}\n".format(time.ctime())
         result += "### Target: {0}\n\n".format(headerName)
         result += "Path | Status | Size | Redirection\n"
         result += "-----|--------|------|------------\n"

diff --git a/lib/reports/plain_text_report.py b/lib/reports/plain_text_report.py
@@ -20,6 +20,7 @@
 from lib.utils.file_utils import FileUtils
 
 import time
+import sys
 
 
 class PlainTextReport(FileBaseReport):
@@ -37,7 +38,7 @@ def addPath(self, path, status, response):
         self.storeData((path, status, contentLength, location))
 
     def generate(self):
-        result = "Time: {0}\n\n".format(time.ctime())
+        result = "# Dirsearch started {0} as: {1}\n\n".format(time.ctime(), ' '.join(sys.argv))
 
         for path, status, contentLength, location in self.pathList:
             result += "{0}  ".format(status)

diff --git a/lib/reports/xml_report.py b/lib/reports/xml_report.py
@@ -18,6 +18,7 @@
 
 from lib.reports import *
 import time
+import sys
 
 
 class XMLReport(FileBaseReport):
@@ -37,7 +38,7 @@ def generate(self):
             self.protocol, self.host, self.port, self.basePath
         )
 
-        result += "<time>{0}</time>\n".format(time.ctime())
+        result += "<dirsearchScan args=\"{0}\" time=\"{1}\">\n".format(' '.join(sys.argv), time.ctime())
         result += "<target url=\"{0}\">\n".format(headerName)
 
         for path, status, contentLength, redirect in self.pathList:
@@ -48,5 +49,6 @@ def generate(self):
             result += " </info>\n"
 
         result += "</target>\n"
+        result += "</dirsearchScan>\n"
 
         return result