Fix for an Issue #588

stamparm · stamparm · commit 854a55166c3a · 2014-01-02T10:29:10.000+01:00
diff --git a/lib/utils/api.py b/lib/utils/api.py
@@ -104,6 +104,7 @@ def init(self):
 class Task(object):
     def __init__(self, taskid):
         self.process = None
+        self.temporary_directory = False
         self.output_directory = None
         self.options = None
         self._original_options = None
@@ -144,12 +145,23 @@ def reset_options(self):
         self.options = AttribDict(self._original_options)
 
     def set_output_directory(self):
+        if self.get_option("oDir"):
+            if os.path.isdir(self.get_option("oDir")):
+                self.output_directory = self.get_option("oDir")
+            else:
+                try:
+                    os.makedirs(self.get_option("oDir"))
+                    self.output_directory = self.get_option("oDir")
+                except OSError:
+                    pass
+
         if not self.output_directory or not os.path.isdir(self.output_directory):
             self.output_directory = tempfile.mkdtemp(prefix="sqlmapoutput-")
+            self.temporary_directory = True
             self.set_option("oDir", self.output_directory)
 
     def clean_filesystem(self):
-        if self.output_directory:
+        if self.output_directory and self.temporary_directory:
             shutil.rmtree(self.output_directory)
 
     def engine_start(self):
