mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 17:46:37 +03:00
Patch for an Issue #1157
This commit is contained in:
parent
66c2a79397
commit
38011743bb
|
@ -166,7 +166,7 @@ class Connect(object):
|
||||||
|
|
||||||
if not kb.dnsMode and conn:
|
if not kb.dnsMode and conn:
|
||||||
headers = conn.info()
|
headers = conn.info()
|
||||||
if headers and (headers.getheader(HTTP_HEADER.CONTENT_ENCODING, "").lower() in ("gzip", "deflate")\
|
if headers and hasattr(headers, "getheader") and (headers.getheader(HTTP_HEADER.CONTENT_ENCODING, "").lower() in ("gzip", "deflate")\
|
||||||
or "text" not in headers.getheader(HTTP_HEADER.CONTENT_TYPE, "").lower()):
|
or "text" not in headers.getheader(HTTP_HEADER.CONTENT_TYPE, "").lower()):
|
||||||
retVal = conn.read(MAX_CONNECTION_TOTAL_SIZE)
|
retVal = conn.read(MAX_CONNECTION_TOTAL_SIZE)
|
||||||
if len(retVal) == MAX_CONNECTION_TOTAL_SIZE:
|
if len(retVal) == MAX_CONNECTION_TOTAL_SIZE:
|
||||||
|
|
|
@ -5,6 +5,7 @@ Copyright (c) 2006-2015 sqlmap developers (http://sqlmap.org/)
|
||||||
See the file 'doc/COPYING' for copying permission
|
See the file 'doc/COPYING' for copying permission
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
import types
|
||||||
import urllib2
|
import urllib2
|
||||||
import urlparse
|
import urlparse
|
||||||
|
|
||||||
|
@ -124,6 +125,25 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
|
||||||
result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers)
|
result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers)
|
||||||
except urllib2.HTTPError, e:
|
except urllib2.HTTPError, e:
|
||||||
result = e
|
result = e
|
||||||
|
|
||||||
|
# Dirty hack for http://bugs.python.org/issue15701
|
||||||
|
try:
|
||||||
|
result.info()
|
||||||
|
except AttributeError:
|
||||||
|
def _(self):
|
||||||
|
return getattr(self, "hdrs") or {}
|
||||||
|
result.info = types.MethodType(_, result)
|
||||||
|
|
||||||
|
if not hasattr(result, "read"):
|
||||||
|
def _(self, length=None):
|
||||||
|
return e.msg
|
||||||
|
result.read = types.MethodType(_, result)
|
||||||
|
|
||||||
|
if not getattr(result, "url", None):
|
||||||
|
result.url = redurl
|
||||||
|
|
||||||
|
if not getattr(result, "code", None):
|
||||||
|
result.code = 999
|
||||||
except:
|
except:
|
||||||
redurl = None
|
redurl = None
|
||||||
result = fp
|
result = fp
|
||||||
|
|
Loading…
Reference in New Issue
Block a user