mirror of
https://github.com/jlu5/PyLink.git
synced 2024-12-26 04:32:51 +01:00
IRCNetwork: fix broken ping timeout handling
Check for ping outs in the ping scheduler instead of the listener... If the connection is dead, the listener won't ever be called.
This commit is contained in:
parent
b2421f5e15
commit
56c035a1f5
23
classes.py
23
classes.py
@ -1528,6 +1528,10 @@ class PyLinkNetworkCoreWithUtils(PyLinkNetworkCore):
|
|||||||
class TLSValidationError(ConnectionError):
|
class TLSValidationError(ConnectionError):
|
||||||
"""Exception raised when additional TLS verifications fail."""
|
"""Exception raised when additional TLS verifications fail."""
|
||||||
|
|
||||||
|
# When this many pings in a row are missed, the ping timer loop will force a disconnect on the
|
||||||
|
# next cycle. Effectively the ping timeout is: pingfreq * (KEEPALIVE_MAX_MISSED + 1)
|
||||||
|
KEEPALIVE_MAX_MISSED = 2
|
||||||
|
|
||||||
class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
||||||
S2S_BUFSIZE = 510
|
S2S_BUFSIZE = 510
|
||||||
|
|
||||||
@ -1546,9 +1550,8 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
|||||||
super()._init_vars(*args, **kwargs)
|
super()._init_vars(*args, **kwargs)
|
||||||
|
|
||||||
# Set IRC specific variables for ping checking and queuing
|
# Set IRC specific variables for ping checking and queuing
|
||||||
self.lastping = time.time()
|
self.lastping = time.time() # This actually tracks the last message received as of 2.0-alpha4
|
||||||
self.pingfreq = self.serverdata.get('pingfreq') or 90
|
self.pingfreq = self.serverdata.get('pingfreq') or 90
|
||||||
self.pingtimeout = self.pingfreq * 3
|
|
||||||
|
|
||||||
self.maxsendq = self.serverdata.get('maxsendq', 4096)
|
self.maxsendq = self.serverdata.get('maxsendq', 4096)
|
||||||
self._queue = queue.Queue(self.maxsendq)
|
self._queue = queue.Queue(self.maxsendq)
|
||||||
@ -1560,6 +1563,12 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
|||||||
if self._aborted.is_set():
|
if self._aborted.is_set():
|
||||||
return
|
return
|
||||||
|
|
||||||
|
elapsed = time.time() - self.lastping
|
||||||
|
if elapsed > (self.pingfreq * KEEPALIVE_MAX_MISSED):
|
||||||
|
log.error('(%s) Disconnected from IRC: Ping timeout (%d secs)', self.name, elapsed)
|
||||||
|
self.disconnect()
|
||||||
|
return
|
||||||
|
|
||||||
self._ping_timer = threading.Timer(self.pingfreq, self._schedule_ping)
|
self._ping_timer = threading.Timer(self.pingfreq, self._schedule_ping)
|
||||||
self._ping_timer.daemon = True
|
self._ping_timer.daemon = True
|
||||||
self._ping_timer.name = 'Ping timer loop for %s' % self.name
|
self._ping_timer.name = 'Ping timer loop for %s' % self.name
|
||||||
@ -1673,7 +1682,6 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
|||||||
|
|
||||||
log.info("Connecting to network %r on %s:%s", self.name, ip, port)
|
log.info("Connecting to network %r on %s:%s", self.name, ip, port)
|
||||||
|
|
||||||
# Use a lower timeout for the initial connect.
|
|
||||||
self._socket.settimeout(self.pingfreq)
|
self._socket.settimeout(self.pingfreq)
|
||||||
|
|
||||||
# Start the actual connection
|
# Start the actual connection
|
||||||
@ -1688,8 +1696,6 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
|||||||
self._socket.close()
|
self._socket.close()
|
||||||
return
|
return
|
||||||
|
|
||||||
self._socket.settimeout(self.pingtimeout)
|
|
||||||
|
|
||||||
# Make sure future reads never block, since select doesn't always guarantee this.
|
# Make sure future reads never block, since select doesn't always guarantee this.
|
||||||
self._socket.setblocking(False)
|
self._socket.setblocking(False)
|
||||||
|
|
||||||
@ -1847,10 +1853,6 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
|||||||
self._log_connection_error('(%s) Connection lost, disconnecting.', self.name)
|
self._log_connection_error('(%s) Connection lost, disconnecting.', self.name)
|
||||||
self.disconnect()
|
self.disconnect()
|
||||||
return
|
return
|
||||||
elif (time.time() - self.lastping) > self.pingtimeout:
|
|
||||||
self._log_connection_error('(%s) Connection timed out.', self.name)
|
|
||||||
self.disconnect()
|
|
||||||
return
|
|
||||||
|
|
||||||
while b'\n' in self._buffer:
|
while b'\n' in self._buffer:
|
||||||
line, self._buffer = self._buffer.split(b'\n', 1)
|
line, self._buffer = self._buffer.split(b'\n', 1)
|
||||||
@ -1858,6 +1860,9 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
|
|||||||
line = line.decode(self.encoding, "replace")
|
line = line.decode(self.encoding, "replace")
|
||||||
self.parse_irc_command(line)
|
self.parse_irc_command(line)
|
||||||
|
|
||||||
|
# Update the last message received time
|
||||||
|
self.lastping = time.time()
|
||||||
|
|
||||||
def _send(self, data):
|
def _send(self, data):
|
||||||
"""Sends raw text to the uplink server."""
|
"""Sends raw text to the uplink server."""
|
||||||
if self._aborted.is_set():
|
if self._aborted.is_set():
|
||||||
|
Loading…
Reference in New Issue
Block a user