sqlmap/lib/core/testing.py

238 lines
7.3 KiB
Python
Raw Normal View History

2010-09-15 17:28:56 +04:00
#!/usr/bin/env python
"""
2012-07-12 21:38:03 +04:00
Copyright (c) 2006-2012 sqlmap developers (http://sqlmap.org/)
2010-10-15 03:18:29 +04:00
See the file 'doc/COPYING' for copying permission
2010-09-15 17:28:56 +04:00
"""
2010-09-27 17:26:46 +04:00
import doctest
2010-09-15 17:28:56 +04:00
import os
2010-09-26 18:02:13 +04:00
import re
import shutil
2010-09-15 17:28:56 +04:00
import sys
2010-09-26 18:02:13 +04:00
import tempfile
2010-09-26 14:47:04 +04:00
import time
2010-09-15 17:28:56 +04:00
2012-12-11 15:02:06 +04:00
from extra.beep.beep import beep
2010-09-26 18:02:13 +04:00
from lib.controller.controller import start
2010-11-24 00:00:42 +03:00
from lib.core.common import clearConsoleLine
2010-09-26 14:47:04 +04:00
from lib.core.common import dataToStdout
2010-10-07 02:43:04 +04:00
from lib.core.common import readXmlFile
2010-09-15 17:28:56 +04:00
from lib.core.data import conf
from lib.core.data import logger
from lib.core.data import paths
2010-09-26 18:02:13 +04:00
from lib.core.option import init
from lib.core.option import setVerbosity
2010-09-27 17:26:46 +04:00
from lib.core.optiondict import optDict
2010-09-26 18:02:13 +04:00
from lib.parse.cmdline import cmdLineParser
2010-09-15 17:28:56 +04:00
failedItem = None
2010-09-15 17:28:56 +04:00
def smokeTest():
"""
This will run the basic smoke testing of a program
"""
retVal = True
2010-09-26 14:47:04 +04:00
count, length = 0, 0
2011-06-16 16:12:30 +04:00
for root, _, files in os.walk(paths.SQLMAP_ROOT_PATH):
2012-12-03 20:20:18 +04:00
if any(_ in root for _ in ("thirdparty", "extra")):
2011-06-16 16:12:30 +04:00
continue
for ifile in files:
2010-09-26 14:47:04 +04:00
length += 1
2010-09-15 17:28:56 +04:00
for root, _, files in os.walk(paths.SQLMAP_ROOT_PATH):
2012-12-03 20:20:18 +04:00
if any(_ in root for _ in ("thirdparty", "extra")):
2011-06-16 16:12:30 +04:00
continue
for ifile in files:
2012-12-06 17:15:44 +04:00
if os.path.splitext(ifile)[1].lower() == ".py" and ifile != "__init__.py":
path = os.path.join(root, os.path.splitext(ifile)[0])
2010-09-15 17:28:56 +04:00
path = path.replace(paths.SQLMAP_ROOT_PATH, '.')
path = path.replace(os.sep, '.').lstrip('.')
try:
__import__(path)
module = sys.modules[path]
except Exception, msg:
retVal = False
2010-09-26 14:47:04 +04:00
dataToStdout("\r")
2011-06-18 15:26:17 +04:00
errMsg = "smoke test failed at importing module '%s' (%s):\n%s" % (path, os.path.join(root, ifile), msg)
2010-09-15 17:28:56 +04:00
logger.error(errMsg)
else:
# Run doc tests
# Reference: http://docs.python.org/library/doctest.html
(failure_count, test_count) = doctest.testmod(module)
if failure_count > 0:
retVal = False
2010-09-26 14:47:04 +04:00
count += 1
status = '%d/%d (%d%s) ' % (count, length, round(100.0*count/length), '%')
2012-12-06 17:15:44 +04:00
dataToStdout("\r[%s] [INFO] complete: %s" % (time.strftime("%X"), status))
2010-09-26 14:47:04 +04:00
2010-11-24 00:00:42 +03:00
clearConsoleLine()
2010-09-15 17:28:56 +04:00
if retVal:
logger.info("smoke test final result: PASSED")
2010-09-15 17:28:56 +04:00
else:
logger.error("smoke test final result: FAILED")
2010-09-15 17:28:56 +04:00
return retVal
2010-09-15 17:32:42 +04:00
2010-09-27 17:26:46 +04:00
def adjustValueType(tagName, value):
for family in optDict.keys():
for name, type_ in optDict[family].items():
if type(type_) == tuple:
type_ = type_[0]
if tagName == name:
if type_ == "boolean":
value = (value == "True")
elif type_ == "integer":
value = int(value)
elif type_ == "float":
value = float(value)
break
return value
2010-09-15 17:32:42 +04:00
def liveTest():
"""
2010-09-15 17:55:28 +04:00
This will run the test of a program against the live testing environment
2010-09-15 17:32:42 +04:00
"""
global failedItem
2010-09-26 18:56:55 +04:00
retVal = True
count = 0
global_ = {}
vars_ = {}
2010-10-07 02:43:04 +04:00
livetests = readXmlFile(paths.LIVE_TESTS_XML)
length = len(livetests.getElementsByTagName("case"))
element = livetests.getElementsByTagName("global")
if element:
for item in element:
2010-09-26 18:02:13 +04:00
for child in item.childNodes:
if child.nodeType == child.ELEMENT_NODE and child.hasAttribute("value"):
2010-09-27 17:26:46 +04:00
global_[child.tagName] = adjustValueType(child.tagName, child.getAttribute("value"))
element = livetests.getElementsByTagName("vars")
if element:
for item in element:
for child in item.childNodes:
if child.nodeType == child.ELEMENT_NODE and child.hasAttribute("value"):
vars_[child.tagName] = child.getAttribute("value")
2010-09-26 18:02:13 +04:00
for case in livetests.getElementsByTagName("case"):
count += 1
name = None
parse = []
switches = dict(global_)
if case.hasAttribute("name"):
name = case.getAttribute("name")
2010-09-26 18:02:13 +04:00
2012-12-19 16:45:46 +04:00
if conf.runCase and ((conf.runCase.isdigit() and conf.runCase != count) or not re.search(conf.runCase, name, re.DOTALL | re.I)):
continue
2010-09-26 18:02:13 +04:00
if case.getElementsByTagName("switches"):
for child in case.getElementsByTagName("switches")[0].childNodes:
if child.nodeType == child.ELEMENT_NODE and child.hasAttribute("value"):
2010-09-27 17:26:46 +04:00
value = replaceVars(child.getAttribute("value"), vars_)
switches[child.tagName] = adjustValueType(child.tagName, value)
2010-09-26 18:02:13 +04:00
if case.getElementsByTagName("parse"):
for item in case.getElementsByTagName("parse")[0].getElementsByTagName("item"):
2010-09-26 18:02:13 +04:00
if item.hasAttribute("value"):
parse.append(replaceVars(item.getAttribute("value"), vars_))
2010-09-26 18:02:13 +04:00
msg = "running live test case '%s' (%d/%d)" % (name, count, length)
logger.info(msg)
result = runCase(switches, parse)
if result:
logger.info("test passed")
else:
2012-12-17 17:29:37 +04:00
logger.error("test failed at parsing item: %s" % failedItem)
beep()
if conf.stopFail is True:
return retVal
retVal &= result
dataToStdout("\n")
2010-09-26 18:56:55 +04:00
if retVal:
logger.info("live test final result: PASSED")
2010-09-26 18:56:55 +04:00
else:
logger.error("live test final result: FAILED")
2010-09-26 18:56:55 +04:00
return retVal
2010-09-26 18:02:13 +04:00
def initCase(switches=None):
global failedItem
failedItem = None
paths.SQLMAP_OUTPUT_PATH = tempfile.mkdtemp(prefix="sqlmaptest-")
2011-04-30 17:20:05 +04:00
paths.SQLMAP_DUMP_PATH = os.path.join(paths.SQLMAP_OUTPUT_PATH, "%s", "dump")
paths.SQLMAP_FILES_PATH = os.path.join(paths.SQLMAP_OUTPUT_PATH, "%s", "files")
logger.debug("using output directory '%s' for this test case" % paths.SQLMAP_OUTPUT_PATH)
2010-09-26 18:02:13 +04:00
cmdLineOptions = cmdLineParser()
cmdLineOptions.liveTest = cmdLineOptions.smokeTest = False
if switches:
for key, value in switches.items():
2010-09-27 17:26:46 +04:00
if key in cmdLineOptions.__dict__:
cmdLineOptions.__dict__[key] = value
init(cmdLineOptions, True)
conf.verbose = 0
setVerbosity()
2010-09-26 18:56:55 +04:00
def cleanCase():
shutil.rmtree(paths.SQLMAP_OUTPUT_PATH, True)
2010-09-26 18:56:55 +04:00
conf.verbose = 1
setVerbosity()
2010-09-26 18:02:13 +04:00
def runCase(switches=None, parse=None):
2010-09-26 18:56:55 +04:00
retVal = True
global failedItem
2010-09-26 18:56:55 +04:00
initCase(switches)
2010-09-26 18:56:55 +04:00
result = start()
if result == False: # if None, ignore
logger.error("the test did not run")
2010-09-26 18:56:55 +04:00
retVal = False
if parse and retVal:
2012-12-19 16:33:37 +04:00
ifile = open(conf.dumper.getOutputFile(), "rb")
content = ifile.read()
ifile.close()
2012-12-19 17:47:17 +04:00
for item in parse:
if item.startswith("r'") and item.endswith("'"):
if not re.search(item[2:-1], content, re.DOTALL):
retVal = False
failedItem = item
2012-12-19 17:47:17 +04:00
break
elif content.find(item) < 0:
2010-09-26 18:56:55 +04:00
retVal = False
failedItem = item
2012-12-19 17:47:17 +04:00
2010-09-26 18:56:55 +04:00
break
cleanCase()
return retVal
2010-09-26 18:02:13 +04:00
def replaceVars(item, vars_):
2010-09-26 18:02:13 +04:00
retVal = item
2012-12-19 18:25:29 +04:00
if item and vars_:
for var in re.findall("\$\{([^}]+)\}", item):
if var in vars_:
retVal = retVal.replace("${%s}" % var, vars_[var])
2012-12-19 18:25:29 +04:00
2012-12-06 17:15:44 +04:00
return retVal