Starting workers before data producing. Worker threads will do the job while…

Starting workers before data producing. Worker threads will do the job while data is being produced.

routersploit/threads.py

@@ -14,6 +14,7 @@ from .exceptions import StopThreadPoolExecutor
 
 
 data_queue = queue.Queue()
+data_producing = threading.Event()
 
 
 class WorkerThread(threading.Thread):
@@ -22,7 +23,7 @@ class WorkerThread(threading.Thread):
         self.name = name
 
     def run(self):
-        while not data_queue.empty():
+        while data_producing.is_set() or not data_queue.empty():
             record = data_queue.get()
             target = record[0]
             args = record[1:]
@@ -40,24 +41,28 @@ class ThreadPoolExecutor(object):
     def __init__(self, threads):
         self.threads = threads
         self.workers = []
+        self.worker = None
+        self.start_time = None
 
     def __enter__(self):
         self.workers = []
+        data_producing.set()
         for worker_id in xrange(int(self.threads)):
             worker = WorkerThread(
                 name='worker-{}'.format(worker_id),
             )
+            worker.start()
             self.workers.append(worker)
+
+        self.worker = worker
+        self.start_time = time.time()
         return self
 
     def __exit__(self, *args):
-        for worker in self.workers:
-            worker.start()
-
-        start = time.time()
+        data_producing.clear()
         try:
-            while worker.isAlive():
-                worker.join(1)
+            while self.worker.isAlive():
+                self.worker.join(1)
         except KeyboardInterrupt:
             utils.print_info()
             utils.print_status("Waiting for already scheduled jobs to finish...")
@@ -67,7 +72,7 @@ class ThreadPoolExecutor(object):
                 worker.join()
             data_queue.unfinished_tasks = 0
 
-        utils.print_status('Elapsed time: ', time.time() - start, 'seconds')
+        utils.print_status('Elapsed time: ', time.time() - self.start_time, 'seconds')
 
     def submit(self, *args):
         data_queue.put(args)