multithreading save to session on abort

stamparm · stamparm · commit 9edd468cafd4 · 2010-07-19T08:37:45.000Z
diff --git a/lib/techniques/blind/inference.py b/lib/techniques/blind/inference.py
@@ -354,6 +354,17 @@ def downloadThread():
                             dataToStdout("\r[%s] [INFO] retrieved: %s" % (time.strftime("%X"), replaceNewlineTabs(output, stdout=True)))
                             iolock.release()
 
+                #TODO: more
+                if not conf.threadContinue:
+                    if int(threading.currentThread().getName()) == numThreads - 1:
+                        partialValue = unicode()
+                        for v in value:
+                            if isinstance(v, basestring) and v is not None:
+                                partialValue += v
+
+                        if len(partialValue) > 0:
+                            dataToSessionFile(replaceNewlineTabs(partialValue))
+
             except (sqlmapConnectionException, sqlmapValueException), errMsg:
                 print
                 conf.threadException = True
@@ -381,7 +392,7 @@ def downloadThread():
 
         # Start the threads
         for numThread in range(numThreads):
-            thread = threading.Thread(target=downloadThread)
+            thread = threading.Thread(target=downloadThread, name=str(numThread))
             thread.start()
             threads.append(thread)
 
@@ -497,7 +508,6 @@ def downloadThread():
                 break
 
             finalValue += val
-
             dataToSessionFile(replaceNewlineTabs(val))
 
             if showEta:
