fix: [crawler] avoid crawler loop if a capture end up in an invalid s…

…tate
CIRCL · Dec 9, 2024 · ea12a44 · ea12a44
1 parent bdb80ee
commit ea12a44
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/bin/crawlers/Crawler.py b/bin/crawlers/Crawler.py
@@ -164,9 +164,15 @@ def get_message(self):
                         self.logger.warning(f'capture UNKNOWN Timeout, {task.uuid} Send back in queue')
                     else:
                         capture.update(status)
-                else:
+                elif status == crawlers.CaptureStatus.QUEUED or status == crawlers.CaptureStatus.ONGOING:
                     capture.update(status)
                     print(capture.uuid, crawlers.CaptureStatus(status).name, int(time.time()))
+                # Invalid State
+                else:
+                    task = capture.get_task()
+                    task.reset()
+                    capture.delete()
+                    self.logger.warning(f'ERROR INVALID CAPTURE STATUS {status}, {task.uuid} Send back in queue')
 
             except ConnectionError:
                 self.logger.warning(f'Lacus ConnectionError, capture {capture.uuid}')