From a64407d9db5796d48420ab6b199ae26df59b2494 Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Tue, 10 May 2011 12:40:01 +0000 Subject: [PATCH] minor bug fix for multithreading and lots of connection retries --- lib/core/option.py | 1 - lib/request/connect.py | 6 +++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/lib/core/option.py b/lib/core/option.py index fa661dfb2..9119a7dde 100644 --- a/lib/core/option.py +++ b/lib/core/option.py @@ -1355,7 +1355,6 @@ def __setKnowledgeBaseAttributes(flushAll=True): kb.redirectSetCookie = None kb.responseTimes = [] kb.resumedQueries = {} - kb.retriesCount = 0 kb.singleLogFlags = set() kb.skipOthersDbms = None kb.suppressSession = False diff --git a/lib/request/connect.py b/lib/request/connect.py index 41951ba01..3ee9e0d57 100644 --- a/lib/request/connect.py +++ b/lib/request/connect.py @@ -255,7 +255,7 @@ class Connect: conf.redirectHandled = True # Reset the number of connection retries - kb.retriesCount = 0 + threadData.retriesCount = 0 # Return response object if response: @@ -379,8 +379,8 @@ class Connect: return None, None elif silent or (ignoreTimeout and any(map(lambda x: x in tbMsg, ["timed out", "IncompleteRead"]))): return None, None - elif kb.retriesCount < conf.retries and not kb.threadException and not conf.realTest: - kb.retriesCount += 1 + elif threadData.retriesCount < conf.retries and not kb.threadException and not conf.realTest: + threadData.retriesCount += 1 warnMsg += ", sqlmap is going to retry the request" logger.critical(warnMsg)