sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-03 05:04:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

detecting infinite redirect loops (Feature #192)

Browse Source
This commit is contained in:
Miroslav Stampar 2010-07-19 12:38:30 +00:00
parent b37dca1c2c
commit d2f88b6ebe
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lib/request/redirecthandler.py
View File

@ -24,7 +24,17 @@ Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import urllib2 import urllib2
from lib.core.exception import sqlmapConnectionException
class SmartRedirectHandler(urllib2.HTTPRedirectHandler): class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
# maximum number of redirections to any single URL
# this is needed because of the state that cookies introduce
max_repeats = 4
# maximum total number of redirections (regardless of URL) before
# assuming we're in a loop
max_redirections = 10
def common_http_redirect(self, result, headers, code): def common_http_redirect(self, result, headers, code):
if "location" in headers: if "location" in headers:
result.redurl = headers.getheaders("location")[0].split("?")[0] result.redurl = headers.getheaders("location")[0].split("?")[0]
@ -36,9 +46,17 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
return result return result
def http_error_301(self, req, fp, code, msg, headers): def http_error_301(self, req, fp, code, msg, headers):
self.infinite_loop_check(req)
result = urllib2.HTTPRedirectHandler.http_error_301(self, req, fp, code, msg, headers) result = urllib2.HTTPRedirectHandler.http_error_301(self, req, fp, code, msg, headers)
return self.common_http_redirect(result, headers, code) return self.common_http_redirect(result, headers, code)
def http_error_302(self, req, fp, code, msg, headers): def http_error_302(self, req, fp, code, msg, headers):
self.infinite_loop_check(req)
result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers) result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers)
return self.common_http_redirect(result, headers, code) return self.common_http_redirect(result, headers, code)
def infinite_loop_check(self, req):
if hasattr(req, 'redirect_dict') and (req.redirect_dict.get(req.get_full_url(), 0) >= self.max_repeats or len(req.redirect_dict) >= self.max_redirections):
errMsg = "infinite redirect loop detected (%s). " % ", ".join(item for item in req.redirect_dict.keys())
errMsg += "please check all provided parameters and/or provide missing ones."
raise sqlmapConnectionException, errMsg