minor update (that default quit in --batch was bothering me - my original idea and it was bad :)

This commit is contained in:
Miroslav Stampar 2011-06-27 14:14:49 +00:00
parent ed4cfbb6d2
commit 8a8b94883b
2 changed files with 11 additions and 14 deletions

View File

@ -484,20 +484,17 @@ def checkSqlInjection(place, parameter, value):
warnMsg = "user aborted during detection phase"
logger.warn(warnMsg)
if not conf.batch:
message = "How do you want to proceed? [(S)kip current test/(e)nd detection phase/(n)ext parameter/(q)uit]"
choice = readInput(message, default="S")
message = "How do you want to proceed? [(S)kip current test/(e)nd detection phase/(n)ext parameter/(q)uit]"
choice = readInput(message, default="S", checkBatch=False)
if choice[0] in ("s", "S"):
pass
elif choice[0] in ("n", "N"):
return None
elif choice[0] in ("e", "E"):
kb.endDetection = True
elif choice[0] in ("q", "Q"):
raise sqlmapUserQuitException
else:
if choice[0] in ("s", "S"):
pass
elif choice[0] in ("n", "N"):
return None
elif choice[0] in ("e", "E"):
kb.endDetection = True
elif choice[0] in ("q", "Q"):
raise sqlmapUserQuitException
finally:
# Reset forced back-end DBMS value

View File

@ -742,7 +742,7 @@ def strToHex(inpStr):
return hexStr
def readInput(message, default=None):
def readInput(message, default=None, checkBatch=True):
"""
@param message: message to display on terminal.
@type message: C{str}
@ -758,7 +758,7 @@ def readInput(message, default=None):
message = "\r%s" % message
if conf.batch:
if checkBatch and conf.batch:
if isinstance(default, (list, tuple, set)):
options = ",".join([getUnicode(opt, UNICODE_ENCODING) for opt in default])
elif default: