From 3a48bee9b02540846a7317860bc4d2e2055401f3 Mon Sep 17 00:00:00 2001
From: Bernardo Damele <bernardo.damele@gmail.com>
Date: Sun, 31 Oct 2010 11:03:59 +0000
Subject: [PATCH] Minor code refactoring

---
 lib/request/connect.py | 17 +++--------------
 1 file changed, 3 insertions(+), 14 deletions(-)

diff --git a/lib/request/connect.py b/lib/request/connect.py
index 3b910c47f..812f032b6 100644
--- a/lib/request/connect.py
+++ b/lib/request/connect.py
@@ -302,6 +302,7 @@ class Connect:
         pageLength  = None
         uri         = None
         raise404    = place != "URI"
+        toUrlencode = { "GET": True, "POST": True, "Cookie": conf.cookieUrlencode, "User-Agent": True, "URI": False }
 
         if not place:
             place = kb.injPlace
@@ -315,20 +316,8 @@ class Connect:
 
                 value = agent.replacePayload(value, payload)
 
-        if place == "GET":
-            value = agent.removePayloadDelimiters(value, True)
-
-        elif place == "POST":
-            value = agent.removePayloadDelimiters(value, True)
-
-        elif place == "Cookie":
-            value = agent.removePayloadDelimiters(value, conf.cookieUrlencode)
-
-        elif place == "User-Agent":
-            value = agent.removePayloadDelimiters(value, True)
-
-        elif place == "URI":
-            value = agent.removePayloadDelimiters(value, False)
+        if place:
+            value = agent.removePayloadDelimiters(value, toUrlencode[place])
 
         if conf.checkPayload:
             checkPayload(value)