sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-09 08:00:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

minor enhancements - #311

Browse Source
This commit is contained in:
Bernardo Damele 2013-01-18 15:29:21 +00:00
parent 0953ce5b08
commit a73aa422fc
1 changed files with 83 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

142
extra/shutils/regressiontest.py
View File

@ -8,88 +8,41 @@ import os
import re import re
import smtplib import smtplib
import subprocess import subprocess
import sys
import time import time
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
sys.path.append("../../")
from lib.core.revision import getRevisionNumber
TIME = time.strftime("%H:%M:%S %d-%m-%Y", time.gmtime()) TIME = time.strftime("%H:%M:%S %d-%m-%Y", time.gmtime())
SQLMAP_HOME = "/opt/sqlmap" SQLMAP_HOME = "/opt/sqlmap"
REVISION = getRevisionNumber()
SMTP_SERVER = "127.0.0.1" SMTP_SERVER = "127.0.0.1"
SMTP_PORT = 25 SMTP_PORT = 25
SMTP_TIMEOUT = 30 SMTP_TIMEOUT = 30
FROM = "regressiontest@sqlmap.org" FROM = "regressiontest@sqlmap.org"
TO = "dev@sqlmap.org" TO = "dev@sqlmap.org"
SUBJECT = "Regression test results on %s" % TIME SUBJECT = "Regression test results on %s using revision %s" % (TIME, REVISION)
CONTENT = "" CONTENT = ""
TEST_COUNTS = [] TEST_COUNTS = []
ATTACHMENTS = {} ATTACHMENTS = {}
command_line = "cd %s && python sqlmap.py --live-test" % SQLMAP_HOME def prepare_email(content):
proc = subprocess.Popen(command_line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
proc.wait()
stdout, stderr = proc.communicate()
failed_tests = re.findall("running live test case: (.+?) \((\d+)\/\d+\)[\r]*\n.+test failed (at parsing item \"(.+)\" )?\- scan folder: (\/.+) \- traceback: (.*?)( - SQL injection not detected)?[\r]*\n", stdout, re.M)
for failed_test in failed_tests:
title = failed_test[0]
test_count = int(failed_test[1])
parse = failed_test[3] if failed_test[3] else None
output_folder = failed_test[4]
traceback = False if failed_test[5] == "False" else bool(failed_test[5])
detected = False if failed_test[6] else True
TEST_COUNTS.append(test_count)
console_output_fd = codecs.open(os.path.join(output_folder, "console_output"), "rb", "utf8")
console_output = console_output_fd.read()
console_output_fd.close()
ATTACHMENTS[test_count] = str(console_output)
log_fd = codecs.open(os.path.join(output_folder, "debiandev", "log"), "rb", "utf8")
log = log_fd.read()
log_fd.close()
if traceback:
traceback_fd = codecs.open(os.path.join(output_folder, "traceback"), "rb", "utf8")
traceback = traceback_fd.read()
traceback_fd.close()
CONTENT += "Failed test case '%s'" % title
if parse:
CONTENT += " at parsing: %s:\n\n" % parse
CONTENT += "### Log file:\n\n"
CONTENT += "%s\n" % log
elif not detected:
CONTENT += " - SQL injection not detected\n\n"
if traceback:
CONTENT += "### Traceback:\n\n"
CONTENT += "%s\n" % str(traceback)
CONTENT += "#######################################################################\n\n"
if CONTENT:
SUBJECT += " (%s)" % ", ".join("#%d" % count for count in TEST_COUNTS)
CONTENT += "\n\nRegression test finished at %s" % time.strftime("%H:%M:%S %d-%m-%Y", time.gmtime())
msg = MIMEMultipart() msg = MIMEMultipart()
msg["Subject"] = SUBJECT msg["Subject"] = SUBJECT
msg["From"] = FROM msg["From"] = FROM
msg["To"] = TO msg["To"] = TO
msg.attach(MIMEText(CONTENT)) msg.attach(MIMEText(content))
for test_count, attachment in ATTACHMENTS.items(): return msg
attachment = MIMEText(attachment)
attachment.add_header("Content-Disposition", "attachment", filename="test_case_%d_console_output.txt" % test_count)
msg.attach(attachment)
def send_email(msg):
try: try:
s = smtplib.SMTP(host=SMTP_SERVER, port=SMTP_PORT, timeout=SMTP_TIMEOUT) s = smtplib.SMTP(host=SMTP_SERVER, port=SMTP_PORT, timeout=SMTP_TIMEOUT)
#s.set_debuglevel(1) #s.set_debuglevel(1)
@ -98,3 +51,74 @@ if CONTENT:
# Catch all for SMTP exceptions # Catch all for SMTP exceptions
except smtplib.SMTPException, e: except smtplib.SMTPException, e:
print "Failure to send email: %s" % str(e) print "Failure to send email: %s" % str(e)
def main():
command_line = "cd %s && python sqlmap.py --live-test" % SQLMAP_HOME
proc = subprocess.Popen(command_line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
proc.wait()
stdout, stderr = proc.communicate()
if stderr:
msg = prepare_email("Execution of regression test failed with error: %s" % stderr)
send_email(msg)
sys.exit(1)
failed_tests = re.findall("running live test case: (.+?) \((\d+)\/\d+\)[\r]*\n.+test failed (at parsing item \"(.+)\" )?\- scan folder: (\/.+) \- traceback: (.*?)( - SQL injection not detected)?[\r]*\n", stdout, re.M)
for failed_test in failed_tests:
title = failed_test[0]
test_count = int(failed_test[1])
parse = failed_test[3] if failed_test[3] else None
output_folder = failed_test[4]
traceback = False if failed_test[5] == "False" else bool(failed_test[5])
detected = False if failed_test[6] else True
TEST_COUNTS.append(test_count)
console_output_fd = codecs.open(os.path.join(output_folder, "console_output"), "rb", "utf8")
console_output = console_output_fd.read()
console_output_fd.close()
ATTACHMENTS[test_count] = str(console_output)
log_fd = codecs.open(os.path.join(output_folder, "debiandev", "log"), "rb", "utf8")
log = log_fd.read()
log_fd.close()
if traceback:
traceback_fd = codecs.open(os.path.join(output_folder, "traceback"), "rb", "utf8")
traceback = traceback_fd.read()
traceback_fd.close()
CONTENT += "Failed test case '%s'" % title
if parse:
CONTENT += " at parsing: %s:\n\n" % parse
CONTENT += "### Log file:\n\n"
CONTENT += "%s\n" % log
elif not detected:
CONTENT += " - SQL injection not detected\n\n"
if traceback:
CONTENT += "### Traceback:\n\n"
CONTENT += "%s\n" % str(traceback)
CONTENT += "#######################################################################\n\n"
if CONTENT:
SUBJECT += " (%s)" % ", ".join("#%d" % count for count in TEST_COUNTS)
CONTENT += "\n\nRegression test finished at %s" % time.strftime("%H:%M:%S %d-%m-%Y", time.gmtime())
msg = prepare_email(CONTENT)
for test_count, attachment in ATTACHMENTS.items():
attachment = MIMEText(attachment)
attachment.add_header("Content-Disposition", "attachment", filename="test_case_%d_console_output.txt" % test_count)
msg.attach(attachment)
send_email(msg)
if __name__ == "__main__":
main()