2011-02-07 02:25:55 +03:00
|
|
|
#!/usr/bin/env python
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
$Id$
|
|
|
|
|
|
|
|
|
|
Copyright (c) 2006-2010 sqlmap developers (http://sqlmap.sourceforge.net/)
|
|
|
|
|
See the file 'doc/COPYING' for copying permission
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
import urllib2
|
|
|
|
|
|
|
|
|
|
class SmartHTTPBasicAuthHandler(urllib2.HTTPBasicAuthHandler):
|
|
|
|
|
"""
|
|
|
|
|
Reference: http://selenic.com/hg/rev/6c51a5056020
|
|
|
|
|
Fix for a: http://bugs.python.org/issue8797
|
|
|
|
|
"""
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
|
urllib2.HTTPBasicAuthHandler.__init__(self, *args, **kwargs)
|
|
|
|
|
self.retried_req = set()
|
|
|
|
|
|
|
|
|
|
def reset_retry_count(self):
|
|
|
|
|
# Python 2.6.5 will call this on 401 or 407 errors and thus loop
|
|
|
|
|
# forever. We disable reset_retry_count completely and reset in
|
|
|
|
|
# http_error_auth_reqed instead.
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
def http_error_auth_reqed(self, auth_header, host, req, headers):
|
|
|
|
|
# Reset the retry counter once for each request.
|
|
|
|
|
if hash(req) not in self.retried_req:
|
|
|
|
|
self.retried_req.add(hash(req))
|
|
|
|
|
self.retried = 0
|
|
|
|
|
return urllib2.HTTPBasicAuthHandler.http_error_auth_reqed(
|
|
|
|
|
self, auth_header, host, req, headers)
|
