fix for URI based injections

This commit is contained in:
Miroslav Stampar 2011-01-22 16:23:33 +00:00
parent 7bf05bf2cb
commit 30cd877c4a

View File

@ -75,6 +75,10 @@ class Agent:
paramDict = conf.paramDict[place]
origValue = paramDict[parameter]
if place == PLACE.URI:
origValue = origValue.split('*')[0]
origValue = origValue[origValue.rfind('/') + 1:]
if value is None:
if where == 1:
value = origValue
@ -101,7 +105,7 @@ class Agent:
retValue = ET.tostring(root)
elif place in (PLACE.UA, PLACE.URI):
retValue = paramString.replace("*", self.addPayloadDelimiters(newValue))
retValue = paramString.replace("%s*" % origValue, self.addPayloadDelimiters(newValue))
else:
retValue = paramString.replace("%s=%s" % (parameter, origValue),
"%s=%s" % (parameter, self.addPayloadDelimiters(newValue)))