Minor cleanup

This commit is contained in:
Miroslav Stampar 2012-09-25 14:21:57 +02:00
parent fccdb824bb
commit 560e0fcb25

View File

@ -73,7 +73,7 @@ def dnsUse(payload, expression):
if Backend.getIdentifiedDbms() in (DBMS.MSSQL, DBMS.PGSQL): if Backend.getIdentifiedDbms() in (DBMS.MSSQL, DBMS.PGSQL):
comment = queries[Backend.getIdentifiedDbms()].comment.query comment = queries[Backend.getIdentifiedDbms()].comment.query
query = agent.prefixQuery("; %s" % expressionUnescaped) query = agent.prefixQuery("; %s" % expressionUnescaped)
query = agent.suffixQuery("%s;%s" % (query, comment)) query = agent.suffixQuery(query, comment)
forgedPayload = agent.payload(newValue=query) forgedPayload = agent.payload(newValue=query)
else: else:
forgedPayload = safeStringFormat(payload, (expressionUnescaped, randomInt(1), randomInt(3))) forgedPayload = safeStringFormat(payload, (expressionUnescaped, randomInt(1), randomInt(3)))