Minor refactoring

This commit is contained in:
Miroslav Stampar
2012-07-24 01:21:32 +02:00
parent b820975217
commit 7f4fa7c27d
11 changed files with 23 additions and 23 deletions

View File

@@ -36,27 +36,27 @@ def tamper(payload):
if payload and payload.find("IFNULL") > -1:
while payload.find("IFNULL(") > -1:
index = payload.find("IFNULL(")
deepness = 1
depth = 1
comma, end = None, None
for i in xrange(index + len("IFNULL("), len(payload)):
if deepness == 1 and payload[i] == ',':
if depth == 1 and payload[i] == ',':
comma = i
elif deepness == 1 and payload[i] == ')':
elif depth == 1 and payload[i] == ')':
end = i
break
elif payload[i] == '(':
deepness += 1
depth += 1
elif payload[i] == ')':
deepness -= 1
depth -= 1
if comma and end:
A = payload[index + len("IFNULL("):comma]
B = payload[comma + 1:end]
newVal = "IF(ISNULL(%s),%s,%s)" % (A, B, A)
_ = payload[index + len("IFNULL("):comma]
__ = payload[comma + 1:end]
newVal = "IF(ISNULL(%s),%s,%s)" % (_, __, _)
payload = payload[:index] + newVal + payload[end+1:]
else:
break