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

minor fix

Browse Source
This commit is contained in:
Miroslav Stampar 2012-02-07 11:16:03 +00:00
parent f7bf1fbe94
commit 11af0b1bbc
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/controller/checks.py
View File

@ -981,6 +981,9 @@ def checkConnection(suppressOutput=False):
logger.critical(errMsg) logger.critical(errMsg)
if any(code in kb.httpErrorCodes for code in (httplib.NOT_FOUND, )): if any(code in kb.httpErrorCodes for code in (httplib.NOT_FOUND, )):
if conf.multipleTargets:
return False
msg = "it is not recommended to continue in this kind of cases. Do you want to quit and make sure that everything is set up properly? [Y/n] " msg = "it is not recommended to continue in this kind of cases. Do you want to quit and make sure that everything is set up properly? [Y/n] "
if readInput(msg, default="Y") not in ("n", "N"): if readInput(msg, default="Y") not in ("n", "N"):
raise sqlmapSilentQuitException raise sqlmapSilentQuitException

4
lib/core/common.py
View File

@ -960,8 +960,6 @@ def readInput(message, default=None, checkBatch=True):
elif message[-1] == ']': elif message[-1] == ']':
message += " " message += " "
message = "\r%s" % message
if checkBatch and conf.batch: if checkBatch and conf.batch:
if isinstance(default, (list, tuple, set)): if isinstance(default, (list, tuple, set)):
options = ",".join(getUnicode(opt, UNICODE_ENCODING) for opt in default) options = ",".join(getUnicode(opt, UNICODE_ENCODING) for opt in default)
@ -979,7 +977,7 @@ def readInput(message, default=None, checkBatch=True):
data = default data = default
else: else:
logging._acquireLock() logging._acquireLock()
dataToStdout(message, True) dataToStdout("\r%s" % message, True)
data = raw_input() data = raw_input()
#data = raw_input(message.encode(sys.stdout.encoding or UNICODE_ENCODING)) #data = raw_input(message.encode(sys.stdout.encoding or UNICODE_ENCODING))
logging._releaseLock() logging._releaseLock()