Fix tor peers not being removed from connection pool when they fail
This commit is contained in:
parent
69a31d1d83
commit
86615305d7
@ -101,7 +101,7 @@ def start_gossip_client():
|
||||
# transport plugin handles the new peer
|
||||
add_onionr_thread(
|
||||
get_new_peers,
|
||||
60, 'get_new_peers', initial_sleep=120)
|
||||
120, 'get_new_peers', initial_sleep=120)
|
||||
|
||||
# Start a new thread to stream blocks from peers
|
||||
# These blocks are being diffused and are stored in
|
||||
|
@ -24,9 +24,6 @@ def _do_ask_peer(peer):
|
||||
_ask_peer(peer)
|
||||
except TimeoutError:
|
||||
logger.debug("Timed out when asking for new peers")
|
||||
except GeneralProxyError:
|
||||
logger.debug("Proxy error")
|
||||
logger.debug(format_exc(), terminal=True)
|
||||
except Exception:
|
||||
logger.error(format_exc(), terminal=True)
|
||||
|
||||
@ -49,7 +46,7 @@ def _ask_peer(peer):
|
||||
'address': peer,
|
||||
'callback': connectpeer.connect_peer
|
||||
}
|
||||
logger.info("Got new peer from exchange " + peer.decode('utf-8'), terminal=True)
|
||||
#logger.info("Got new peer from exchange " + peer.decode('utf-8'), terminal=True)
|
||||
onionrevents.event('announce_rec', data=connect_data, threaded=True)
|
||||
s.close()
|
||||
|
||||
|
@ -67,7 +67,7 @@ def stream_from_peers():
|
||||
stream_times = 100
|
||||
try:
|
||||
sock = peer.get_socket(CONNECT_TIMEOUT)
|
||||
except ConnectionRefusedError:
|
||||
except (TimeoutError, ConnectionRefusedError) as _:
|
||||
need_socket_lock.release()
|
||||
return
|
||||
except Exception:
|
||||
@ -118,7 +118,7 @@ def stream_from_peers():
|
||||
raise
|
||||
# Tell them to keep streaming
|
||||
sock.sendall(int(1).to_bytes(1, 'big'))
|
||||
except (BrokenPipeError, TimeoutError) as e:
|
||||
except (BrokenPipeError, TimeoutError, ConnectionError) as e:
|
||||
pass
|
||||
#logger.debug(f"{e} when streaming from peers", terminal=True)
|
||||
#logger.debug(traceback.format_exc())
|
||||
|
@ -3,12 +3,24 @@ import socks
|
||||
from gossip.peerset import gossip_peer_set
|
||||
import logger
|
||||
|
||||
|
||||
class HandleRevc:
|
||||
def __init__(self, sock):
|
||||
self.sock_recv = sock.recv
|
||||
self.sock = sock
|
||||
|
||||
def recv(self, *args, **kwargs):
|
||||
return self.sock_recv(*args, **kwargs)
|
||||
try:
|
||||
got_data = self.sock_recv(*args, **kwargs)
|
||||
if not len(got_data):
|
||||
raise ConnectionError("Peer socket returned empty value")
|
||||
return got_data
|
||||
except Exception:
|
||||
try:
|
||||
gossip_peer_set.remove(self.sock)
|
||||
except KeyError:
|
||||
pass
|
||||
raise
|
||||
|
||||
|
||||
class TorPeer:
|
||||
@ -37,7 +49,7 @@ class TorPeer:
|
||||
pass
|
||||
else:
|
||||
logger.debug(f"Could not create socket to peer {self.transport_address}", terminal=True)
|
||||
raise
|
||||
raise TimeoutError
|
||||
mock_recv = HandleRevc(s)
|
||||
s.recv = mock_recv.recv
|
||||
return s
|
||||
|
Loading…
Reference in New Issue
Block a user