mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 09:36:35 +03:00
minor update (that default quit in --batch was bothering me - my original idea and it was bad :)
This commit is contained in:
parent
ed4cfbb6d2
commit
8a8b94883b
|
@ -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
|
||||
|
|
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue
Block a user