Adding switch --output-dir (Issue #53)

stamparm · stamparm · commit 168aeadf7623 · 2012-07-03T00:50:23.000+02:00
diff --git a/lib/core/dump.py b/lib/core/dump.py
@@ -25,6 +25,7 @@
 from lib.core.data import kb
 from lib.core.data import logger
 from lib.core.enums import DBMS
+from lib.core.exception import sqlmapGenericException
 from lib.core.exception import sqlmapValueException
 from lib.core.replication import Replication
 from lib.core.settings import BLANK
@@ -61,7 +62,11 @@ def _write(self, data, n=True, console=True):
 
     def setOutputFile(self):
         self._outputFile = "%s%slog" % (conf.outputPath, os.sep)
-        self._outputFP = codecs.open(self._outputFile, "ab", UNICODE_ENCODING)
+        try:
+            self._outputFP = codecs.open(self._outputFile, "ab", UNICODE_ENCODING)
+        except IOError, ex:
+            errMsg = "error occurred while opening log file ('%s')" % ex
+            raise sqlmapGenericException, errMsg
 
     def getOutputFile(self):
         return self._outputFile
diff --git a/lib/core/option.py b/lib/core/option.py
@@ -1374,6 +1374,9 @@ class _(unicode): pass
     if conf.torType:
         conf.torType = conf.torType.upper()
 
+    if conf.oDir:
+        paths.SQLMAP_OUTPUT_PATH = conf.oDir
+
     threadData = getCurrentThreadData()
     threadData.reset()
 
diff --git a/lib/core/optiondict.py b/lib/core/optiondict.py
@@ -173,6 +173,7 @@
                                "forms":             "boolean",
                                "freshQueries":      "boolean",
                                "hexConvert":        "boolean",
+                               "oDir":              "string",
                                "parseErrors":       "boolean",
                                "replicate":         "boolean",
                                "updateAll":         "boolean",
diff --git a/lib/core/revision.py b/lib/core/revision.py
@@ -12,6 +12,10 @@
 from subprocess import Popen as execute
 
 def getRevisionNumber():
+    """
+    Returns revision number in a GitHub style
+    """
+
     retVal = None
     filePath = None
 
diff --git a/lib/parse/cmdline.py b/lib/parse/cmdline.py
@@ -540,6 +540,10 @@ def cmdLineParser():
                             action="store_true",
                             help="Uses DBMS hex function(s) for data retrieval")
 
+        general.add_option("--output-dir", dest="oDir",
+                            action="store",
+                            help="Custom output directory path")
+
         general.add_option("--parse-errors", dest="parseErrors",
                                   action="store_true",
                                   help="Parse and display DBMS error messages from responses")
diff --git a/sqlmap.conf b/sqlmap.conf
@@ -596,6 +596,9 @@ freshQueries = False
 # Valid: True or False
 hexConvert = False
 
+# Custom output directory path.
+oDir =
+
 # Parse and display DBMS error messages from responses.
 # Valid: True or False
 parseErrors = False
