Try to reconnect to Tor on send error

2016-09-04 18:00:08 +02:00 · 2016-09-04 18:00:08 +02:00 · 2a7f03af8a
commit 2a7f03af8a
parent f698ec2502
1 changed files with 11 additions and 2 deletions
--- a/src/Tor/TorManager.py
+++ b/src/Tor/TorManager.py
@ -232,8 +232,17 @@ class TorManager:
        if not conn:
            conn = self.conn
        self.log.debug("> %s" % cmd)
-        conn.send("%s\r\n" % cmd)
-        back = conn.recv(1024 * 64).decode("utf8", "ignore")
+        for retry in range(2):
+            try:
+                conn.send("%s\r\n" % cmd)
+                back = conn.recv(1024 * 64).decode("utf8", "ignore")
+                break
+            except Exception, err:
+                self.log.error("Tor send error: %s, reconnecting..." % err)
+                self.disconnect()
+                time.sleep(1)
+                self.connect()
+                back = None
        self.log.debug("< %s" % back.strip())
        return back