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

Update for an Issue #225

Browse Source
This commit is contained in:
Miroslav Stampar 2012-10-30 01:26:19 +01:00
parent 969259607c
commit 7c7aff12c6
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib/core/common.py
View File

@ -786,6 +786,7 @@ def readInput(message, default=None, checkBatch=True):
kb.prependFlag = False
try:
data = raw_input() or default
data = getUnicode(data, system=True) if data else data
except:
time.sleep(0.05) # Reference: http://www.gossamer-threads.com/lists/python/python/781893
kb.prependFlag = True

3
lib/takeover/web.py
View File

@ -33,6 +33,7 @@ from lib.core.common import randomStr
from lib.core.common import readInput
from lib.core.common import singleTimeWarnMessage
from lib.core.convert import hexencode
from lib.core.convert import utf8encode
from lib.core.data import conf
from lib.core.data import kb
from lib.core.data import logger
@ -263,7 +264,7 @@ class Web:
with open(filename, "w+") as f:
_ = decloak(os.path.join(paths.SQLMAP_SHELL_PATH, "stager.%s_" % self.webApi))
_ = _.replace("WRITABLE_DIR", localPath.replace('/', '\\\\') if Backend.isOs(OS.WINDOWS) else localPath)
f.write(_)
f.write(utf8encode(_))
self.unionWriteFile(filename, self.webStagerFilePath, "text")