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

Set strict timeout for connection handshake

Browse source
This commit is contained in:
shortcutme 2019-01-20 16:15:32 +01:00
parent 1803e47c02
commit f1a54bd7d0
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/Connection/Connection.py
View file

@ -164,13 +164,14 @@ class Connection(object):
self.sock.settimeout(30) self.sock.settimeout(30)
self.sock.connect(sock_address) self.sock.connect(sock_address)
self.sock.settimeout(timeout_before)
# Detect protocol # Detect protocol
self.send({"cmd": "handshake", "req_id": 0, "params": self.getHandshakeInfo()}) self.send({"cmd": "handshake", "req_id": 0, "params": self.getHandshakeInfo()})
event_connected = self.event_connected event_connected = self.event_connected
gevent.spawn(self.messageLoop) gevent.spawn(self.messageLoop)
return event_connected.get() # Wait for handshake connect_res = event_connected.get() # Wait for handshake
self.sock.settimeout(timeout_before)
return connect_res
# Handle incoming connection # Handle incoming connection
def handleIncomingConnection(self, sock): def handleIncomingConnection(self, sock):