sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-10-26 05:31:04 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for an Issue #588

Browse Source
This commit is contained in:
Miroslav Stampar 2014-01-02 10:29:10 +01:00
parent 9b4b070ecf
commit 854a55166c
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lib/utils/api.py
View File

@ -104,6 +104,7 @@ class Database(object):
class Task(object): class Task(object):
def __init__(self, taskid): def __init__(self, taskid):
self.process = None self.process = None
self.temporary_directory = False
self.output_directory = None self.output_directory = None
self.options = None self.options = None
self._original_options = None self._original_options = None
@ -144,12 +145,23 @@ class Task(object):
self.options = AttribDict(self._original_options) self.options = AttribDict(self._original_options)
def set_output_directory(self): 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): if not self.output_directory or not os.path.isdir(self.output_directory):
self.output_directory = tempfile.mkdtemp(prefix="sqlmapoutput-") self.output_directory = tempfile.mkdtemp(prefix="sqlmapoutput-")
self.temporary_directory = True
self.set_option("oDir", self.output_directory) self.set_option("oDir", self.output_directory)
def clean_filesystem(self): def clean_filesystem(self):
if self.output_directory: if self.output_directory and self.temporary_directory:
shutil.rmtree(self.output_directory) shutil.rmtree(self.output_directory)
def engine_start(self): def engine_start(self):