From 00623f3a1857cc62ffc97b6ee160a55e3d8c4fb6 Mon Sep 17 00:00:00 2001
From: Username <user@mymx.me>
Date: Sun, 21 Dec 2025 23:37:39 +0100
Subject: [PATCH] connection_pool: add health tracking and backoff

---
 connection_pool.py | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/connection_pool.py b/connection_pool.py
index 1af8b21..0c04fc2 100644
--- a/connection_pool.py
+++ b/connection_pool.py
@@ -127,11 +127,28 @@ class TorConnectionPool(object):
             self._warmup_connections(warmup_target)
 
     def _warmup_connections(self, target_host, target_port=80, timeout=10):
-        """Pre-warm Tor circuits by making test connections."""
+        """Pre-warm Tor circuits by making test connections.
+
+        Makes an actual TCP connection through each Tor host to verify
+        end-to-end connectivity before marking the host as ready.
+        """
         _log('warming up %d tor host(s)...' % len(self.tor_hosts), 'pool')
         warmed = 0
 
         for host_addr in self.tor_hosts:
+            # First verify we can reach the Tor SOCKS port
+            try:
+                tor_host, tor_port = host_addr.split(':')
+                test_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                test_sock.settimeout(5)
+                test_sock.connect((tor_host, int(tor_port)))
+                test_sock.close()
+            except Exception as e:
+                self.host_states[host_addr].record_failure()
+                _log('tor host %s unreachable: %s' % (host_addr, str(e)), 'error')
+                continue
+
+            # Then verify we can route through Tor to an external target
             try:
                 start = time.time()
                 proxy = rocksock.RocksockProxyFromURL('socks5://%s' % host_addr)
@@ -145,10 +162,11 @@ class TorConnectionPool(object):
                 sock.disconnect()
                 elapsed = time.time() - start
                 self.host_states[host_addr].record_success(elapsed)
+                _log('tor host %s ready (%.1fs)' % (host_addr, elapsed), 'pool')
                 warmed += 1
             except Exception as e:
                 self.host_states[host_addr].record_failure()
-                _log('warmup failed for %s: %s' % (host_addr, str(e)), 'debug')
+                _log('tor host %s route failed: %s' % (host_addr, str(e)), 'warn')
 
         self.warmup_complete = True
         _log('warmup complete: %d/%d hosts ready' % (warmed, len(self.tor_hosts)), 'pool')