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

major bug fix (different HTTP content charsets are now properly handled)

Browse Source
This commit is contained in:
Miroslav Stampar 2010-06-09 14:40:36 +00:00
parent 654d707d5d
commit eaef068c90
2 changed files with 18 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/request/basic.py
View File

@ -89,13 +89,13 @@ def parseResponse(page, headers):
kb.absFilePaths.add(absFilePath) kb.absFilePaths.add(absFilePath)
def decodePage(page, encoding): def decodePage(page, contentEncoding, contentType):
""" """
Decode gzip/deflate HTTP response Decode compressed/charset HTTP response
""" """
if isinstance(encoding, basestring) and encoding.lower() in ('gzip', 'x-gzip', 'deflate'): if isinstance(contentEncoding, basestring) and contentEncoding.lower() in ('gzip', 'x-gzip', 'deflate'):
if encoding == 'deflate': if contentEncoding == 'deflate':
# http://stackoverflow.com/questions/1089662/python-inflate-and-deflate-implementations # http://stackoverflow.com/questions/1089662/python-inflate-and-deflate-implementations
data = StringIO.StringIO(zlib.decompress(page, -15)) data = StringIO.StringIO(zlib.decompress(page, -15))
else: else:
@ -103,4 +103,8 @@ def decodePage(page, encoding):
page = data.read() page = data.read()
#http://stackoverflow.com/questions/1020892/python-urllib2-read-to-unicode
if contentType and (contentType.find('charset=') != -1):
page = unicode(page, contentType.split('charset=')[-1])
return page return page

6
lib/request/connect.py
View File

@ -106,8 +106,7 @@ class Connect:
page = conn.read() page = conn.read()
responseHeaders = conn.info() responseHeaders = conn.info()
encoding = responseHeaders.get("Content-Encoding") page = decodePage(page, responseHeaders.get("Content-Encoding"), responseHeaders.get("Content-Type"))
page = decodePage(page, encoding)
return page return page
@ -190,8 +189,7 @@ class Connect:
status = conn.msg status = conn.msg
responseHeaders = conn.info() responseHeaders = conn.info()
encoding = responseHeaders.get("Content-Encoding") page = decodePage(page, responseHeaders.get("Content-Encoding"), responseHeaders.get("Content-Type"))
page = decodePage(page, encoding)
except urllib2.HTTPError, e: except urllib2.HTTPError, e:
if e.code == 401: if e.code == 401: