sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-01-24 08:14:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

minor update to make counting variable totally independent of the urllib2's self.retried

Browse Source
This commit is contained in:
Miroslav Stampar 2011-03-02 10:42:17 +00:00
parent a010386a23
commit 3a1f5744be
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lib/request/basicauthhandler.py
View File

@ -17,6 +17,7 @@ class SmartHTTPBasicAuthHandler(urllib2.HTTPBasicAuthHandler):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
urllib2.HTTPBasicAuthHandler.__init__(self, *args, **kwargs) urllib2.HTTPBasicAuthHandler.__init__(self, *args, **kwargs)
self.retried_req = set() self.retried_req = set()
self.retried_count = 0
def reset_retry_count(self): def reset_retry_count(self):
# Python 2.6.5 will call this on 401 or 407 errors and thus loop # Python 2.6.5 will call this on 401 or 407 errors and thus loop
@ -28,13 +29,13 @@ class SmartHTTPBasicAuthHandler(urllib2.HTTPBasicAuthHandler):
# Reset the retry counter once for each request. # Reset the retry counter once for each request.
if hash(req) not in self.retried_req: if hash(req) not in self.retried_req:
self.retried_req.add(hash(req)) self.retried_req.add(hash(req))
self.retried = 0 self.retried_count = 0
else: else:
if self.retried > 5: if self.retried_count > 5:
raise urllib2.HTTPError(req.get_full_url(), 401, "basic auth failed", raise urllib2.HTTPError(req.get_full_url(), 401, "basic auth failed",
headers, None) headers, None)
else: else:
self.retried += 1 self.retried_count += 1
return urllib2.HTTPBasicAuthHandler.http_error_auth_reqed( return urllib2.HTTPBasicAuthHandler.http_error_auth_reqed(
self, auth_header, host, req, headers) self, auth_header, host, req, headers)