few more fixes for proper redirecting mechanism

This commit is contained in:
Miroslav Stampar 2012-03-15 19:47:59 +00:00
parent 48e8c978fb
commit 7d313ac911
2 changed files with 8 additions and 2 deletions

View File

@ -665,6 +665,9 @@ def checkDynParam(place, parameter, value):
dynamicity might depend on another parameter.
"""
if kb.redirectChoice:
return None
kb.matchRatio = None
dynResult = None
randInt = randomInt()
@ -746,6 +749,9 @@ def checkStability():
like for instance string matching (--string).
"""
if kb.redirectChoice:
return None
infoMsg = "testing if the url is stable, wait a few seconds"
logger.info(infoMsg)

View File

@ -38,7 +38,7 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
return retVal
def _ask_redirect_choice(self, redcode, redurl):
if kb.redirectChoice is None:
if kb.redirectChoice is None and kb.originalPage:
msg = "sqlmap got a %d redirect to " % redcode
msg += "'%s'. What do you want to do? " % redurl
msg += "\n[1] Follow the redirection (default)"
@ -93,7 +93,7 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
dbgMsg += "redirect response content (%s)" % msg
logger.debug(dbgMsg)
if kb.redirectChoice == REDIRECTION.FOLLOW:
if kb.redirectChoice == REDIRECTION.FOLLOW or kb.originalPage is None:
req.headers[HTTPHEADER.HOST] = getHostHeader(redurl)
result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers)
else: