From 105cd225b3f4d312d1a7e02960b3ad2623739b60 Mon Sep 17 00:00:00 2001 From: aNiceGuy Date: Sun, 8 Jul 2012 13:15:10 +0100 Subject: [PATCH] Improvement for Issue #71: exception when in fileReadMode --- lib/techniques/error/use.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lib/techniques/error/use.py b/lib/techniques/error/use.py index 0beb7d68d..7f7f3b59b 100644 --- a/lib/techniques/error/use.py +++ b/lib/techniques/error/use.py @@ -177,7 +177,10 @@ def __errorFields(expression, expressionFields, expressionFieldsList, expected=N if kb.fileReadMode and output and output.strip(): print elif output is not None and not (threadData.resumed and kb.suppressResumeInfo): - dataToStdout("[%s] [INFO] %s #%d: %s\r\n" % (time.strftime("%X"), "resumed" if threadData.resumed else "retrieved", num+1, safecharencode(output))) + if kb.fileReadMode == False: + dataToStdout("[%s] [INFO] %s: %s\r\n" % (time.strftime("%X"), "resumed" if threadData.resumed else "retrieved", safecharencode(output))) + else: + dataToStdout("[%s] [INFO] %s #%d: %s\r\n" % (time.strftime("%X"), "resumed" if threadData.resumed else "retrieved", num+1, safecharencode(output))) if isinstance(num, int): expression = origExpr