Set timeout for connection

This commit is contained in:
shortcutme 2018-10-30 04:41:11 +01:00
parent d89d770e73
commit 87503b875a
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE

View file

@ -127,6 +127,8 @@ class Connection(object):
if "TCP_NODELAY" in dir(socket): if "TCP_NODELAY" in dir(socket):
self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1) self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
timeout_before = self.sock.gettimeout()
self.sock.settimeout(30)
self.sock.connect((self.ip, int(self.port))) self.sock.connect((self.ip, int(self.port)))
# Implicit SSL # Implicit SSL
@ -150,6 +152,8 @@ class Connection(object):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.connect((self.ip, int(self.port))) self.sock.connect((self.ip, int(self.port)))
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