Fix for an Issue #227

This commit is contained in:
Miroslav Stampar 2012-10-30 00:20:49 +01:00
parent 1d07b93730
commit a9094a35fe

View File

@ -33,6 +33,7 @@ from lib.core.enums import PLACE
from lib.core.enums import POST_HINT from lib.core.enums import POST_HINT
from lib.core.exception import sqlmapFilePathException from lib.core.exception import sqlmapFilePathException
from lib.core.exception import sqlmapGenericException from lib.core.exception import sqlmapGenericException
from lib.core.exception import sqlmapMissingPrivileges
from lib.core.exception import sqlmapSyntaxException from lib.core.exception import sqlmapSyntaxException
from lib.core.exception import sqlmapUserQuitException from lib.core.exception import sqlmapUserQuitException
from lib.core.option import authHandler from lib.core.option import authHandler
@ -440,9 +441,18 @@ def __createTargetDirs():
conf.outputPath = tempDir conf.outputPath = tempDir
with open(os.path.join(conf.outputPath, "target.txt"), "w+") as f: try:
_ = kb.originalUrls.get(conf.url) or conf.url or conf.hostname with open(os.path.join(conf.outputPath, "target.txt"), "w+") as f:
f.write(_.encode(UNICODE_ENCODING)) _ = kb.originalUrls.get(conf.url) or conf.url or conf.hostname
f.write(_.encode(UNICODE_ENCODING))
except IOError, msg:
if "denied" in str(msg):
errMsg = "you don't have enough permissions "
else:
errMsg = "something went wrong while trying "
errMsg += "to write to the output directory '%s' (%s)" % (paths.SQLMAP_OUTPUT_PATH, msg)
raise sqlmapMissingPrivileges, errMsg
__createDumpDir() __createDumpDir()
__createFilesDir() __createFilesDir()