mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-25 02:53:46 +03:00
132 lines
4.0 KiB
Python
132 lines
4.0 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
safe2bin.py - Simple safe(hex) to binary format converter
|
|
|
|
Copyright (c) 2006-2014 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'doc/COPYING' for copying permission
|
|
"""
|
|
|
|
import binascii
|
|
import re
|
|
import string
|
|
import os
|
|
import sys
|
|
|
|
from optparse import OptionError
|
|
from optparse import OptionParser
|
|
|
|
# Regex used for recognition of hex encoded characters
|
|
HEX_ENCODED_CHAR_REGEX = r"(?P<result>\\x[0-9A-Fa-f]{2})"
|
|
|
|
# Regex used for recognition of representation for hex encoded invalid unicode characters
|
|
INVALID_UNICODE_CHAR_REGEX = r"(?P<result>\\\?[0-9A-Fa-f]{2})"
|
|
|
|
# Raw chars that will be safe encoded to their slash (\) representations (e.g. newline to \n)
|
|
SAFE_ENCODE_SLASH_REPLACEMENTS = "\t\n\r\x0b\x0c"
|
|
|
|
# Characters that don't need to be safe encoded
|
|
SAFE_CHARS = "".join(filter(lambda x: x not in SAFE_ENCODE_SLASH_REPLACEMENTS, string.printable.replace('\\', '')))
|
|
|
|
# String used for temporary marking of slash characters
|
|
SLASH_MARKER = "__SLASH__"
|
|
|
|
def safecharencode(value):
|
|
"""
|
|
Returns safe representation of a given basestring value
|
|
|
|
>>> safecharencode(u'test123')
|
|
u'test123'
|
|
>>> safecharencode(u'test\x01\x02\xff')
|
|
u'test\\01\\02\\03\\ff'
|
|
"""
|
|
|
|
retVal = value
|
|
|
|
if isinstance(value, basestring):
|
|
if any(_ not in SAFE_CHARS for _ in value):
|
|
retVal = retVal.replace('\\', SLASH_MARKER)
|
|
|
|
for char in SAFE_ENCODE_SLASH_REPLACEMENTS:
|
|
retVal = retVal.replace(char, repr(char).strip('\''))
|
|
|
|
retVal = reduce(lambda x, y: x + (y if (y in string.printable or isinstance(value, unicode) and ord(y) >= 160) else '\\x%02x' % ord(y)), retVal, (unicode if isinstance(value, unicode) else str)())
|
|
|
|
retVal = retVal.replace(SLASH_MARKER, "\\\\")
|
|
elif isinstance(value, list):
|
|
for i in xrange(len(value)):
|
|
retVal[i] = safecharencode(value[i])
|
|
|
|
return retVal
|
|
|
|
def safechardecode(value, binary=False):
|
|
"""
|
|
Reverse function to safecharencode
|
|
"""
|
|
|
|
retVal = value
|
|
if isinstance(value, basestring):
|
|
retVal = retVal.replace('\\\\', SLASH_MARKER)
|
|
|
|
while True:
|
|
match = re.search(HEX_ENCODED_CHAR_REGEX, retVal)
|
|
if match:
|
|
retVal = retVal.replace(match.group("result"), (unichr if isinstance(value, unicode) else chr)(ord(binascii.unhexlify(match.group("result").lstrip("\\x")))))
|
|
else:
|
|
break
|
|
|
|
for char in SAFE_ENCODE_SLASH_REPLACEMENTS[::-1]:
|
|
retVal = retVal.replace(repr(char).strip('\''), char)
|
|
|
|
retVal = retVal.replace(SLASH_MARKER, '\\')
|
|
|
|
if binary:
|
|
if isinstance(retVal, unicode):
|
|
retVal = retVal.encode("utf8")
|
|
while True:
|
|
match = re.search(INVALID_UNICODE_CHAR_REGEX, retVal)
|
|
if match:
|
|
retVal = retVal.replace(match.group("result"), chr(ord(binascii.unhexlify(match.group("result").lstrip("\\?")))))
|
|
else:
|
|
break
|
|
|
|
elif isinstance(value, (list, tuple)):
|
|
for i in xrange(len(value)):
|
|
retVal[i] = safechardecode(value[i])
|
|
|
|
return retVal
|
|
|
|
def main():
|
|
usage = '%s -i <input file> [-o <output file>]' % sys.argv[0]
|
|
parser = OptionParser(usage=usage, version='0.1')
|
|
|
|
try:
|
|
parser.add_option('-i', dest='inputFile', help='Input file')
|
|
parser.add_option('-o', dest='outputFile', help='Output file')
|
|
|
|
(args, _) = parser.parse_args()
|
|
|
|
if not args.inputFile:
|
|
parser.error('Missing the input file, -h for help')
|
|
|
|
except (OptionError, TypeError), e:
|
|
parser.error(e)
|
|
|
|
if not os.path.isfile(args.inputFile):
|
|
print 'ERROR: the provided input file \'%s\' is not a regular file' % args.inputFile
|
|
sys.exit(1)
|
|
|
|
f = open(args.inputFile, 'r')
|
|
data = f.read()
|
|
f.close()
|
|
|
|
if not args.outputFile:
|
|
args.outputFile = args.inputFile + '.bin'
|
|
|
|
f = open(args.outputFile, 'wb')
|
|
f.write(safechardecode(data))
|
|
f.close()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|