shodan to stop on no credits

hack-umbrella · Apr 24, 2019 · 682e77a · 682e77a
1 parent 43553f7
commit 682e77a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/wfuzz/plugin_api/payloadtools.py b/src/wfuzz/plugin_api/payloadtools.py
@@ -184,6 +184,8 @@ def _do_search(self):
                 self.page_queue.task_done()
                 if "Invalid page size" in str(e):
                     self.results_queue.put(None)
+                elif "Insufficient query credits" in str(e):
+                    self.results_queue.put(None)
                 else:
                     self.results_queue.put(e)
                 continue
@@ -217,8 +219,6 @@ def _stop(self):
         self._threads = []
 
         self.results_queue.put(None)
-        self._cancel_job = False
-        self._started = False
 
     def __next__(self):
         if not self._started:
@@ -230,6 +230,8 @@ def __next__(self):
 
         if res is None:
             self._stop()
+            self._cancel_job = False
+            self._started = False
             raise StopIteration
         elif isinstance(res, Exception):
             self._stop()