From c1145c244e37d2a27860f1200bf671572d00a443 Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Sun, 23 Jan 2011 23:23:30 +0000 Subject: [PATCH] fix for user-agent injections --- lib/core/agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lib/core/agent.py b/lib/core/agent.py index ac10c05b7..1b2fad4e1 100644 --- a/lib/core/agent.py +++ b/lib/core/agent.py @@ -104,8 +104,10 @@ class Agent: child.text = self.addPayloadDelimiters(newValue) retValue = ET.tostring(root) - elif place in (PLACE.UA, PLACE.URI): + elif place == PLACE.URI: retValue = paramString.replace("%s*" % origValue, self.addPayloadDelimiters(newValue)) + elif place == PLACE.UA: + retValue = paramString.replace(origValue, self.addPayloadDelimiters(newValue)) else: retValue = paramString.replace("%s=%s" % (parameter, origValue), "%s=%s" % (parameter, self.addPayloadDelimiters(newValue)))