Patch for DREI has_header() (causing double Host: header)

stamparm · stamparm · commit 74c7cf95f264 · 2019-05-07T15:49:03.000+02:00
diff --git a/lib/core/settings.py b/lib/core/settings.py
@@ -18,7 +18,7 @@
 from thirdparty import six
 
 # sqlmap version (<major>.<minor>.<month>.<monthly commit>)
-VERSION = "1.3.5.35"
+VERSION = "1.3.5.36"
 TYPE = "dev" if VERSION.count('.') > 2 and VERSION.split('.')[-1] != '0' else "stable"
 TYPE_COLORS = {"dev": 33, "stable": 90, "pip": 34}
 VERSION_STRING = "sqlmap/%s#%s" % ('.'.join(VERSION.split('.')[:-1]) if VERSION.count('.') > 2 and VERSION.split('.')[-1] == '0' else VERSION, TYPE)
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -419,7 +419,7 @@ def getPage(**kwargs):
                 if isinstance(value, six.string_types):
                     for char in (r"\r", r"\n"):
                         value = re.sub(r"(%s)([^ \t])" % char, r"\g<1>\t\g<2>", value)
-                    headers[getBytes(key)] = getBytes(value.strip("\r\n"))
+                    headers[getBytes(key) if six.PY2 else key] = getBytes(value.strip("\r\n"))  # Note: Python3 has_header() expects non-bytes value
 
             if six.PY2:
                 url = getBytes(url)  # Note: Python3 requires text while Python2 has problems when mixing text with binary POST
