oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Wait tor connections until tor client startup

Browse source
This commit is contained in:
shortcutme 2018-06-03 12:46:47 +02:00
parent d125551c0d
commit 329e885da7
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/Tor/TorManager.py
View file

@ -36,6 +36,8 @@ class TorManager(object):
self.start_onions = None
self.conn = None
self.lock = RLock()
self.starting = True
self.event_started = gevent.event.AsyncResult()
if config.tor == "disable":
self.enabled = False
@ -57,6 +59,7 @@ class TorManager(object):
self.proxy_port = int(self.proxy_port)
def start(self):
self.starting = True
try:
if not self.connect():
raise Exception("No connection")
@ -106,6 +109,8 @@ class TorManager(object):
except Exception, err:
self.log.error(u"Error starting Tor client: %s" % Debug.formatException(str(err).decode("utf8", "ignore")))
self.enabled = False
self.starting = False
self.event_started.set(False)
return False
def isSubprocessRunning(self):
@ -183,7 +188,7 @@ class TorManager(object):
# Auth cookie file
res_protocol = self.send("PROTOCOLINFO", conn)
cookie_match = re.search('COOKIEFILE="(.*?)"', res_protocol)
if config.tor_password:
res_auth = self.send('AUTHENTICATE "%s"' % config.tor_password, conn)
elif cookie_match:
@ -201,6 +206,8 @@ class TorManager(object):
assert float(version.replace(".", "0", 2)) >= 207.5, "Tor version >=0.2.7.5 required, found: %s" % version
self.setStatus(u"Connected (%s)" % res_auth)
self.event_started.set(True)
self.connecting = False
self.conn = conn
except Exception, err:
self.conn = None
@ -319,6 +326,9 @@ class TorManager(object):
if not self.enabled:
return False
self.log.debug("Creating new Tor socket to %s:%s" % (onion, port))
if self.starting:
self.log.debug("Waiting for startup...")
self.event_started.get()
if config.tor == "always": # Every socket is proxied by default, in this mode
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
else: