fix scan hanging indefinitely after detectors complete

The asyncio.wait loop waited on cancel_event.wait() even after all
detector tasks finished, causing scans to hang forever. Now breaks
out when only the cancel waiter remains. Also adds port pre-filtering
to endpoint prober (skip closed ports), DNS resolution timeouts, and
per-detector completion logging. Scan time: 26+ min → 0.2s.

Author: symbibot

agentsniff/cli.py

@@ -45,7 +45,7 @@ class Colors:
  / ___ / /_/ /  __/ / / / /_ ___/ / / / / / __/ __/
 /_/  |_\\__, /\\___/_/ /_/\\__//____/_/ /_/_/_/ /_/
       /____/
-{Colors.RESET}{Colors.DIM}  AI Agent Network Scanner v1.2.0
+{Colors.RESET}{Colors.DIM}  AI Agent Network Scanner v1.2.1
   Detect AI agents on your network{Colors.RESET}
 """
 

agentsniff/detectors/dns_monitor.py

@@ -208,12 +208,15 @@ async def _active_dns_check(self, targets: list[str]) -> list[DetectionSignal]:
         self.logger.info("Resolving known LLM API domains for cross-reference...")
         for domain in self.config.all_llm_domains[:20]:  # Top 20 most common
             try:
-                result = await asyncio.get_event_loop().run_in_executor(
-                    None, lambda d=domain: socket.getaddrinfo(d, 443, socket.AF_INET)
+                result = await asyncio.wait_for(
+                    asyncio.get_event_loop().run_in_executor(
+                        None, lambda d=domain: socket.getaddrinfo(d, 443, socket.AF_INET)
+                    ),
+                    timeout=3.0,
                 )
                 for entry in result:
                     llm_ips.add(entry[4][0])
-            except (socket.gaierror, OSError):
+            except (socket.gaierror, OSError, asyncio.TimeoutError):
                 continue
 
         if llm_ips:

agentsniff/detectors/traffic_analyzer.py

@@ -172,13 +172,16 @@ async def setup(self):
             # Strip port if present
             host = domain.split(":")[0]
             try:
-                result = await loop.run_in_executor(
-                    None,
-                    lambda h=host: socket.getaddrinfo(h, 443, socket.AF_INET),
+                result = await asyncio.wait_for(
+                    loop.run_in_executor(
+                        None,
+                        lambda h=host: socket.getaddrinfo(h, 443, socket.AF_INET),
+                    ),
+                    timeout=3.0,
                 )
                 for entry in result:
                     self._llm_ips.add(entry[4][0])
-            except (socket.gaierror, OSError):
+            except (socket.gaierror, OSError, asyncio.TimeoutError):
                 continue
 
         self.logger.info(f"Resolved {len(self._llm_ips)} LLM API IP addresses")

agentsniff/scanner.py

@@ -302,7 +302,9 @@ async def _correlate_and_notify(signals: list[DetectionSignal]):
 
     async def run_detector(detector):
         try:
-            return await detector.scan(targets)
+            signals = await detector.scan(targets)
+            logger.info(f"Detector {detector.name} finished ({len(signals)} signals)")
+            return signals
         except asyncio.CancelledError:
             logger.info(f"Detector {detector.name} cancelled")
             return []
@@ -357,6 +359,15 @@ async def run_detector(detector):
             if cancelled:
                 break
 
+            # All detector tasks done — only cancel_waiter remains
+            if pending == {cancel_waiter}:
+                cancel_waiter.cancel()
+                try:
+                    await cancel_waiter
+                except asyncio.CancelledError:
+                    pass
+                break
+
         if not cancelled:
             cancel_waiter.cancel()
     else:

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentsniff"
-version = "1.2.0"
+version = "1.2.1"
 description = "AI Agent Network Scanner - Detect AI agents on your network"
 readme = "README.md"
 license = {text = "Apache-2.0"}