From fe07ea26f41be1eda6f8cccc591a5165dd3fb135 Mon Sep 17 00:00:00 2001 From: Evgeny Zinoviev Date: Sun, 6 Jun 2021 01:19:08 +0300 Subject: error handling fix --- scanner.py | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/scanner.py b/scanner.py index 0a6fa91..f78e4f6 100644 --- a/scanner.py +++ b/scanner.py @@ -38,19 +38,21 @@ class TCPScanner: t.join() def run(self): - try: - while True: - if self.failed: - break - - try: - self._scan(self.q.get(block=False)) - except Exception as e: - logger.exception(e) - self.failed = True - break - except queue.Empty: - return + while True: + if self.failed: + break + + try: + port = self.q.get(block=False) + except queue.Empty: + break + + try: + self._scan(port) + except Exception as e: + logger.exception(e) + self.failed = True + break def _scan(self, port): try: -- cgit v1.2.3