fix for user-agent injections

This commit is contained in:
Miroslav Stampar 2011-01-23 23:23:30 +00:00
parent 818c9787b2
commit c1145c244e

View File

@ -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)))