mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-23 01:56:36 +03:00
104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2019 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
import os
|
|
import re
|
|
|
|
from xml.etree import ElementTree as et
|
|
|
|
from lib.core.common import getSafeExString
|
|
from lib.core.data import conf
|
|
from lib.core.data import paths
|
|
from lib.core.datatype import AttribDict
|
|
from lib.core.exception import SqlmapInstallationException
|
|
from lib.core.settings import PAYLOAD_XML_FILES
|
|
|
|
def cleanupVals(text, tag):
|
|
if tag == "clause" and '-' in text:
|
|
text = re.sub(r"(\d+)-(\d+)", lambda match: ','.join(str(_) for _ in xrange(int(match.group(1)), int(match.group(2)) + 1)), text)
|
|
|
|
if tag in ("clause", "where"):
|
|
text = text.split(',')
|
|
|
|
if isinstance(text, basestring):
|
|
text = int(text) if text.isdigit() else text
|
|
|
|
elif isinstance(text, list):
|
|
count = 0
|
|
|
|
for _ in text:
|
|
text[count] = int(_) if _.isdigit() else _
|
|
count += 1
|
|
|
|
if len(text) == 1 and tag not in ("clause", "where"):
|
|
text = text[0]
|
|
|
|
return text
|
|
|
|
def parseXmlNode(node):
|
|
for element in node.getiterator("boundary"):
|
|
boundary = AttribDict()
|
|
|
|
for child in element.getchildren():
|
|
if child.text:
|
|
values = cleanupVals(child.text, child.tag)
|
|
boundary[child.tag] = values
|
|
else:
|
|
boundary[child.tag] = None
|
|
|
|
conf.boundaries.append(boundary)
|
|
|
|
for element in node.getiterator("test"):
|
|
test = AttribDict()
|
|
|
|
for child in element.getchildren():
|
|
if child.text and child.text.strip():
|
|
values = cleanupVals(child.text, child.tag)
|
|
test[child.tag] = values
|
|
else:
|
|
if len(child.getchildren()) == 0:
|
|
test[child.tag] = None
|
|
continue
|
|
else:
|
|
test[child.tag] = AttribDict()
|
|
|
|
for gchild in child.getchildren():
|
|
if gchild.tag in test[child.tag]:
|
|
prevtext = test[child.tag][gchild.tag]
|
|
test[child.tag][gchild.tag] = [prevtext, gchild.text]
|
|
else:
|
|
test[child.tag][gchild.tag] = gchild.text
|
|
|
|
conf.tests.append(test)
|
|
|
|
def loadBoundaries():
|
|
try:
|
|
doc = et.parse(paths.BOUNDARIES_XML)
|
|
except Exception, ex:
|
|
errMsg = "something appears to be wrong with "
|
|
errMsg += "the file '%s' ('%s'). Please make " % (paths.BOUNDARIES_XML, getSafeExString(ex))
|
|
errMsg += "sure that you haven't made any changes to it"
|
|
raise SqlmapInstallationException(errMsg)
|
|
|
|
root = doc.getroot()
|
|
parseXmlNode(root)
|
|
|
|
def loadPayloads():
|
|
for payloadFile in PAYLOAD_XML_FILES:
|
|
payloadFilePath = os.path.join(paths.SQLMAP_XML_PAYLOADS_PATH, payloadFile)
|
|
|
|
try:
|
|
doc = et.parse(payloadFilePath)
|
|
except Exception, ex:
|
|
errMsg = "something appears to be wrong with "
|
|
errMsg += "the file '%s' ('%s'). Please make " % (payloadFilePath, getSafeExString(ex))
|
|
errMsg += "sure that you haven't made any changes to it"
|
|
raise SqlmapInstallationException(errMsg)
|
|
|
|
root = doc.getroot()
|
|
parseXmlNode(root)
|