Little cleaning a mess with url encoding and post hint types

stamparm · stamparm · commit e1ffdde5326f · 2013-03-27T13:39:27.000+01:00
diff --git a/lib/core/settings.py b/lib/core/settings.py
@@ -518,7 +518,10 @@
 MULTIPART_RECOGNITION_REGEX = r"(?i)Content-Disposition:[^;]+;\s*name="
 
 # Default POST data content-type
-DEFAULT_CONTENT_TYPE = "application/x-www-form-urlencoded"
+DEFAULT_CONTENT_TYPE = "application/x-www-form-urlencoded; charset=utf-8"
+
+# Raw text POST data content-type
+PLAIN_TEXT_CONTENT_TYPE = "text/plain; charset=utf-8"
 
 # Length used while checking for existence of Suhosin-patch (like) protection mechanism
 SUHOSIN_MAX_VALUE_LENGTH = 512
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -76,6 +76,7 @@
 from lib.core.settings import LARGE_CHUNK_TRIM_MARKER
 from lib.core.settings import PAYLOAD_DELIMITER
 from lib.core.settings import PERMISSION_DENIED_REGEX
+from lib.core.settings import PLAIN_TEXT_CONTENT_TYPE
 from lib.core.settings import UNENCODED_ORIGINAL_VALUE
 from lib.core.settings import URI_HTTP_HEADER
 from lib.core.settings import WARN_TIME_STDEV
@@ -586,7 +587,7 @@ def queryPage(value=None, place=None, content=False, getRatioValue=False, silent
         pageLength = None
         uri = None
         code = None
-        skipUrlEncode = conf.skipUrlEncode
+        urlEncodePost = None
 
         if not place:
             place = kb.injection.place or PLACE.GET
@@ -597,11 +598,16 @@ def queryPage(value=None, place=None, content=False, getRatioValue=False, silent
         payload = agent.extractPayload(value)
         threadData = getCurrentThreadData()
 
-        if skipUrlEncode is None and conf.httpHeaders:
+        if conf.httpHeaders:
             headers = dict(conf.httpHeaders)
-            _ = max(headers[_] if _.upper() == HTTP_HEADER.CONTENT_TYPE.upper() else None for _ in headers.keys())
-            if _ and "urlencoded" not in _:
-                skipUrlEncode = True
+            contentType = max(headers[_] if _.upper() == HTTP_HEADER.CONTENT_TYPE.upper() else None for _ in headers.keys())
+            urlEncodePost = contentType and "urlencoded" in contentType
+
+            if conf.skipUrlEncode and urlEncodePost:
+                urlEncodePost = False
+                conf.httpHeaders = [_ for _ in conf.httpHeaders if _[1] != contentType]
+                contentType = POST_HINT_CONTENT_TYPES.get(kb.postHint, PLAIN_TEXT_CONTENT_TYPE)
+                conf.httpHeaders.append((HTTP_HEADER.CONTENT_TYPE, contentType))
 
         if payload:
             if kb.tamperFunctions:
@@ -628,8 +634,8 @@ def queryPage(value=None, place=None, content=False, getRatioValue=False, silent
                         payload = json.dumps(payload)[1:-1]
                 value = agent.replacePayload(value, payload)
             else:
-                if not skipUrlEncode and place in (PLACE.GET, PLACE.POST, PLACE.COOKIE, PLACE.URI):
-                    # GET, POST, URI and Cookie payload needs to be throughly URL encoded
+                # GET, POST, URI and Cookie payload needs to be throughly URL encoded
+                if place in (PLACE.GET, PLACE.URI, PLACE.COOKIE) and not conf.skipUrlEncode or place in (PLACE.POST,) and urlEncodePost:
                     payload = urlencode(payload, '%', False, place != PLACE.URI)
                     value = agent.replacePayload(value, payload)
 
@@ -745,13 +751,13 @@ def _randomizeParameter(paramString, randomParameter):
                         else:
                             get += "%s%s=%s" % (delimiter, name, value)
 
-        if not skipUrlEncode:
+        if not conf.skipUrlEncode:
             get = urlencode(get, limit=True)
 
         if post is not None:
-            if place not in (PLACE.POST, PLACE.CUSTOM_POST) and '%' in getattr(post, UNENCODED_ORIGINAL_VALUE, ""):
+            if place not in (PLACE.POST, PLACE.CUSTOM_POST) and hasattr(post, UNENCODED_ORIGINAL_VALUE):
                 post = getattr(post, UNENCODED_ORIGINAL_VALUE)
-            elif not skipUrlEncode and kb.postHint not in POST_HINT_CONTENT_TYPES.keys():
+            elif urlEncodePost:
                 post = urlencode(post, spaceplus=kb.postSpaceToPlus)
 
         if timeBasedCompare:
