minor bug fix for multithreading and lots of connection retries

stamparm · stamparm · commit a64407d9db57 · 2011-05-10T12:40:01.000Z
diff --git 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
@@ -255,7 +255,7 @@ def getPage(**kwargs):
                 conf.redirectHandled = True
 
             # Reset the number of connection retries
-            kb.retriesCount = 0
+            threadData.retriesCount = 0
 
             # Return response object
             if response:
@@ -379,8 +379,8 @@ def getPage(**kwargs):
                 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)
