mirror of
				https://github.com/sqlmapproject/sqlmap.git
				synced 2025-10-31 16:07:55 +03:00 
			
		
		
		
	more enhancements - issue #311
This commit is contained in:
		
							parent
							
								
									a7cab63796
								
							
						
					
					
						commit
						de26a31493
					
				|  | @ -13,36 +13,41 @@ import time | ||||||
| 
 | 
 | ||||||
| from email.mime.text import MIMEText | from email.mime.text import MIMEText | ||||||
| 
 | 
 | ||||||
| TIME = time.strftime("%H:%M:%S", time.gmtime()) | TIME = time.strftime("%H:%M:%S %d-%m-%Y", time.gmtime()) | ||||||
| DATE = time.strftime("%d-%m-%Y", time.gmtime()) |  | ||||||
| 
 | 
 | ||||||
| SMTP_SERVER = "10.129.121.194" | 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 = "bernardo.damele@gmail.com" | TO = "dev@sqlmap.org" | ||||||
| SUBJECT = "Regression test results - started at %s on %s" % (TIME, DATE) | SUBJECT = "Regression test results on on %s" % TIME | ||||||
| CONTENT = "" | CONTENT = "" | ||||||
|  | TEST_COUNTS = [] | ||||||
| 
 | 
 | ||||||
| command_line = "cd ../../ ; rm -f $REGRESSION_FILE ; python sqlmap.py --live-test --run-case 'Invalid logic'" | command_line = "cd ../../ ; rm -f $REGRESSION_FILE ; python sqlmap.py --live-test --run-case 'Invalid logic'" | ||||||
| proc = subprocess.Popen(command_line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) | proc = subprocess.Popen(command_line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) | ||||||
| proc.wait() | proc.wait() | ||||||
| stdout, stderr = proc.communicate() | stdout, stderr = proc.communicate() | ||||||
| failed_tests = re.findall("running live test case: (.+?) \(\d+\/\d+\)[\r]*\n.+test failed (.*?)at parsing item: (.+) \- scan folder is (\/.+)[\r]*\n", stdout, re.I | re.M) | failed_tests = re.findall("running live test case: (.+?) \((\d+)\/\d+\)[\r]*\n(.+?)test failed (.*?)at parsing item: (.+) \- scan folder is (\/.+)[\r]*\n", stdout, re.I | re.M) | ||||||
| 
 | 
 | ||||||
| for failed_test in failed_tests: | for failed_test in failed_tests: | ||||||
|     title = failed_test[0] |     title = failed_test[0] | ||||||
|     traceback = failed_test[1] or None |     test_count = failed_test[1] | ||||||
|     parse = failed_test[2] |     error = True if "the test did not identify the SQL injection" in failed_test[2] else False | ||||||
|     output_folder = failed_test[3] |     traceback = failed_test[2] or None | ||||||
|  |     parse = failed_test[3] | ||||||
|  |     output_folder = failed_test[4] | ||||||
|  | 
 | ||||||
|  |     TEST_COUNTS.append(test_count) | ||||||
| 
 | 
 | ||||||
|     console_output_fd = codecs.open(os.path.join(output_folder, "console_output"), "rb", "utf8") |     console_output_fd = codecs.open(os.path.join(output_folder, "console_output"), "rb", "utf8") | ||||||
|     console_output = console_output_fd.read() |     console_output = console_output_fd.read() | ||||||
|     console_output_fd.close() |     console_output_fd.close() | ||||||
| 
 | 
 | ||||||
|     log_fd = codecs.open(os.path.join(output_folder, "debiandev", "log"), "rb", "utf8") |     if error is False: | ||||||
|     log = log_fd.read() |         log_fd = codecs.open(os.path.join(output_folder, "debiandev", "log"), "rb", "utf8") | ||||||
|     log_fd.close() |         log = log_fd.read() | ||||||
|  |         log_fd.close() | ||||||
| 
 | 
 | ||||||
|     if traceback: |     if traceback: | ||||||
|         traceback_fd = codecs.open(os.path.join(output_folder, "traceback"), "rb", "utf8") |         traceback_fd = codecs.open(os.path.join(output_folder, "traceback"), "rb", "utf8") | ||||||
|  | @ -50,10 +55,13 @@ for failed_test in failed_tests: | ||||||
|         traceback_fd.close() |         traceback_fd.close() | ||||||
| 
 | 
 | ||||||
|     CONTENT += "Failed test case '%s' at parsing: %s:\n\n" % (title, parse) |     CONTENT += "Failed test case '%s' at parsing: %s:\n\n" % (title, parse) | ||||||
|     CONTENT += "### LOG FILE:\n\n" | 
 | ||||||
|     CONTENT += "%s\n" % log |     if error is False: | ||||||
|     CONTENT += "### CONSOLE OUTPUT\n\n" |         CONTENT += "### LOG FILE:\n\n" | ||||||
|     CONTENT += "%s\n" % str(console_output) |         CONTENT += "%s\n" % log | ||||||
|  |     else: | ||||||
|  |         CONTENT += "### CONSOLE OUTPUT\n\n" | ||||||
|  |         CONTENT += "%s\n" % str(console_output) | ||||||
| 
 | 
 | ||||||
|     if traceback: |     if traceback: | ||||||
|         CONTENT += "### TRACEBACK:\n" |         CONTENT += "### TRACEBACK:\n" | ||||||
|  | @ -62,6 +70,7 @@ for failed_test in failed_tests: | ||||||
|     CONTENT += "\n#######################################\n\n" |     CONTENT += "\n#######################################\n\n" | ||||||
| 
 | 
 | ||||||
| if CONTENT: | if CONTENT: | ||||||
|  |     SUBJECT += " (%s)" % ", ".join("#%d" % count for count in TEST_COUNTS) | ||||||
|     msg = MIMEText(CONTENT) |     msg = MIMEText(CONTENT) | ||||||
|     msg["Subject"] = SUBJECT |     msg["Subject"] = SUBJECT | ||||||
|     msg["From"] = FROM |     msg["From"] = FROM | ||||||
|  | @ -69,5 +78,5 @@ if CONTENT: | ||||||
| 
 | 
 | ||||||
|     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) | ||||||
|     s.sendmail(FROM, TO, msg) |     s.sendmail(FROM, TO, msg.as_string()) | ||||||
|     s.quit() |     s.quit() | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user