mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2025-01-24 00:04:23 +03:00
Fix for an Issue #85
This commit is contained in:
parent
4de83daf03
commit
5437f8bf36
|
@ -15,6 +15,7 @@ import zlib
|
||||||
|
|
||||||
from lib.core.common import extractErrorMessage
|
from lib.core.common import extractErrorMessage
|
||||||
from lib.core.common import extractRegexResult
|
from lib.core.common import extractRegexResult
|
||||||
|
from lib.core.common import getPublicTypeMembers
|
||||||
from lib.core.common import getUnicode
|
from lib.core.common import getUnicode
|
||||||
from lib.core.common import readInput
|
from lib.core.common import readInput
|
||||||
from lib.core.common import resetCookieJar
|
from lib.core.common import resetCookieJar
|
||||||
|
@ -53,7 +54,27 @@ def forgeHeaders(items=None):
|
||||||
headers = dict(conf.httpHeaders)
|
headers = dict(conf.httpHeaders)
|
||||||
headers.update(items or {})
|
headers.update(items or {})
|
||||||
|
|
||||||
headers = dict(("-".join(_.capitalize() for _ in key.split('-')), value) for (key, value) in headers.items())
|
class _str(str):
|
||||||
|
def capitalize(self):
|
||||||
|
return _str(self)
|
||||||
|
|
||||||
|
def title(self):
|
||||||
|
return _str(self)
|
||||||
|
|
||||||
|
_ = headers
|
||||||
|
headers = {}
|
||||||
|
for key, value in _.items():
|
||||||
|
success = False
|
||||||
|
if key.upper() not in (_.upper() for _ in getPublicTypeMembers(HTTP_HEADER, True)):
|
||||||
|
try:
|
||||||
|
headers[_str(key)] = value # dirty hack for http://bugs.python.org/issue12455
|
||||||
|
except UnicodeEncodeError: # don't do the hack on non-ASCII header names (they have to be properly encoded later on)
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
success = True
|
||||||
|
if not success:
|
||||||
|
key = '-'.join(_.capitalize() for _ in key.split('-'))
|
||||||
|
headers[key] = value
|
||||||
|
|
||||||
if conf.cj:
|
if conf.cj:
|
||||||
if HTTP_HEADER.COOKIE in headers:
|
if HTTP_HEADER.COOKIE in headers:
|
||||||
|
|
Loading…
Reference in New Issue
Block a user