Patch for an Issue #1113

This commit is contained in:
Miroslav Stampar 2015-01-14 13:53:38 +01:00
parent c8787e0404
commit 570d30789b
2 changed files with 7 additions and 3 deletions

View File

@ -97,9 +97,12 @@ class BigArray(list):
with open(filename, "w+b") as fp:
pickle.dump(chunk, fp, pickle.HIGHEST_PROTOCOL)
return filename
except IOError, ex:
except (OSError, IOError), ex:
errMsg = "exception occurred while storing data "
errMsg += "to a temporary file ('%s')" % ex
errMsg += "to a temporary file ('%s'). Please " % ex
errMsg += "make sure that there is enough disk space left. If problem persists, "
errMsg += "try to set environment variable 'TEMP' to a location "
errMsg += "writeable by the current user"
raise SqlmapSystemException, errMsg
def _checkcache(self, index):

View File

@ -1450,9 +1450,10 @@ def _createTemporaryDirectory():
errMsg = "there has been a problem while accessing "
errMsg += "system's temporary directory location(s) ('%s'). Please " % ex
errMsg += "make sure that there is enough disk space left. If problem persists, "
errMsg += "try to set environment variable 'TMP' to a location "
errMsg += "try to set environment variable 'TEMP' to a location "
errMsg += "writeable by the current user"
raise SqlmapSystemException, errMsg
kb.tempDir = tempfile.tempdir = tempfile.mkdtemp(prefix="sqlmap", suffix=str(os.getpid()))
if not os.path.isdir(tempfile.tempdir):
os.makedirs(tempfile.tempdir)