From df3cc38cd9237dbf3a6f38c1c7ed94dcfe34142a Mon Sep 17 00:00:00 2001 From: Bernardo Damele Date: Sun, 3 Feb 2013 15:39:07 +0000 Subject: [PATCH] minor improvements --- lib/core/testing.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/lib/core/testing.py b/lib/core/testing.py index f113648b3..1d02bbfbe 100644 --- a/lib/core/testing.py +++ b/lib/core/testing.py @@ -169,14 +169,16 @@ def liveTest(): msg = "running live test case: %s (%d/%d)" % (name, count, length) logger.info(msg) - try: - result = runCase(switches, parse) - except SqlmapNotVulnerableException: - vulnerable = False + initCase(switches) test_case_fd = codecs.open(os.path.join(paths.SQLMAP_OUTPUT_PATH, "test_case"), "wb", UNICODE_ENCODING) test_case_fd.write("%s\n" % name) + try: + result = runCase(parse) + except SqlmapNotVulnerableException: + vulnerable = False + if result is True: logger.info("test passed") cleanCase() @@ -250,13 +252,11 @@ def initCase(switches=None): def cleanCase(): shutil.rmtree(paths.SQLMAP_OUTPUT_PATH, True) -def runCase(switches=None, parse=None): +def runCase(parse): global failedItem global failedParseOn global failedTraceBack - initCase(switches) - LOGGER_HANDLER.stream = sys.stdout = tempfile.SpooledTemporaryFile(max_size=0, mode="w+b", prefix="sqlmapstdout-") retVal = True handled_exception = None