sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-22 09:36:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Patch for an Issue #1106

Browse Source
This commit is contained in:
Miroslav Stampar 2015-01-14 05:16:32 +01:00
parent 06ff8b3a16
commit f9a9ededb1
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/core/option.py
View File

@ -1445,7 +1445,7 @@ def _createTemporaryDirectory():
if not os.path.isdir(tempfile.gettempdir()): if not os.path.isdir(tempfile.gettempdir()):
os.makedirs(tempfile.gettempdir()) os.makedirs(tempfile.gettempdir())
tempfile.tempdir = tempfile.mkdtemp(prefix="sqlmap", suffix=str(os.getpid())) kb.tempDir = tempfile.tempdir = tempfile.mkdtemp(prefix="sqlmap", suffix=str(os.getpid()))
if not os.path.isdir(tempfile.tempdir): if not os.path.isdir(tempfile.tempdir):
os.makedirs(tempfile.tempdir) os.makedirs(tempfile.tempdir)
@ -1772,6 +1772,7 @@ def _setKnowledgeBaseAttributes(flushAll=True):
kb.storeHashesChoice = None kb.storeHashesChoice = None
kb.suppressResumeInfo = False kb.suppressResumeInfo = False
kb.technique = None kb.technique = None
kb.tempDir = None
kb.testMode = False kb.testMode = False
kb.testQueryCount = 0 kb.testQueryCount = 0
kb.testType = None kb.testType = None

3
sqlmap.py
View File

@ -153,7 +153,8 @@ def main():
if conf.get("showTime"): if conf.get("showTime"):
dataToStdout("\n[*] shutting down at %s\n\n" % time.strftime("%X"), forceOutput=True) dataToStdout("\n[*] shutting down at %s\n\n" % time.strftime("%X"), forceOutput=True)
shutil.rmtree(tempfile.tempdir, ignore_errors=True) if kb.get("tempDir"):
shutil.rmtree(kb.tempDir, ignore_errors=True)
kb.threadContinue = False kb.threadContinue = False
kb.threadException = True kb.threadException = True