Fix a crash with protected Tor control port (#785)

This commit is contained in:
jan Anja 2022-06-22 21:23:58 +05:00 committed by GitHub
parent 966644baa0
commit 194b2eae74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -8,6 +8,7 @@ from requests import Response, ConnectionError
import urllib.parse as urlparse
import os
from stem import Signal, SocketError
from stem.connection import AuthenticationFailure
from stem.control import Controller
from stem.connection import authenticate_cookie, authenticate_password
@ -63,7 +64,9 @@ def send_tor_signal(signal: Signal) -> bool:
c.signal(signal)
os.environ['TOR_AVAILABLE'] = '1'
return True
except (SocketError, ConnectionRefusedError, ConnectionError):
except (SocketError, AuthenticationFailure,
ConnectionRefusedError, ConnectionError):
# TODO: Handle Tor authentification (password and cookie)
os.environ['TOR_AVAILABLE'] = '0'
return False