sqlmap/tamper/randomcomments.py

43 lines
940 B
Python
Raw Normal View History

#!/usr/bin/env python
"""
$Id$
2012-01-11 18:59:46 +04:00
Copyright (c) 2006-2012 sqlmap developers (http://www.sqlmap.org/)
See the file 'doc/COPYING' for copying permission
"""
import re
from lib.core.common import randomRange
from lib.core.data import kb
from lib.core.enums import PRIORITY
__priority__ = PRIORITY.LOW
2011-04-04 12:18:26 +04:00
def tamper(payload):
"""
2011-04-04 12:18:26 +04:00
Add random comments to SQL keywords
Example: 'INSERT' becomes 'IN/**/S/**/ERT'
"""
2011-04-04 12:18:26 +04:00
retVal = payload
2011-04-04 12:18:26 +04:00
if payload:
for match in re.finditer(r"[A-Za-z_]+", payload):
word = match.group()
if len(word) < 2:
continue
if word.upper() in kb.keywords:
newWord = word[0]
for i in xrange(1, len(word) - 1):
newWord += "%s%s" % ("/**/" if randomRange(0, 1) else "", word[i])
newWord += word[-1]
retVal = retVal.replace(word, newWord)
return retVal