GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2024-11-01 01:09:22 +01:00
Code Issues Projects Releases Wiki Activity

IRCNetwork: add an experimental implementation of TCP keepalive

Browse Source
This commit is contained in:
James Lu 2018-06-14 00:41:18 -07:00
parent d3125d9a8f
commit 26d72aa97f
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
classes.py
View File

@ -20,6 +20,7 @@ import re
import collections import collections
import collections.abc import collections.abc
import textwrap import textwrap
import sys
try: try:
import ircmatch import ircmatch
@ -1685,6 +1686,28 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
self._socket.settimeout(self.pingfreq) self._socket.settimeout(self.pingfreq)
# Adapted from https://stackoverflow.com/questions/12248132/
# Use TCP keepalive to manage timeouts
if hasattr(socket, 'TCP_KEEPIDLE'):
self._socket.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
self._socket.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPIDLE, self.pingfreq)
self._socket.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPINTVL, self.pingfreq)
self._socket.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPCNT, KEEPALIVE_MAX_MISSED)
log.debug('(%s) enabled Linux socket keepalive', self.name)
elif sys.platform == 'win32':
# SIO_KEEPALIVE_VALS as documented at https://msdn.microsoft.com/en-us/library/dd877220%28v=vs.85%29.aspx
# 1st value: whether to enable keepalive (1 = on, 0 = off)
# 2nd value: idle time before first keepalive packet is sent, in milliseconds
# 3rd value: time between keepalive packets, in milliseconds
# According to https://msdn.microsoft.com/en-us/library/ee470551(v=vs.85).aspx
# "On Windows Vista and later, the number of keep-alive probes (data retransmissions) is set to 10 and cannot be changed."
self._socket.ioctl(socket.SIO_KEEPALIVE_VALS, (1, self.pingfreq * 1000, self.pingfreq * 1000))
log.debug('(%s) enabled Windows socket keepalive', self.name)
else:
log.debug('(%s) No TCP keepalive support yet, (sys.platform=%r)', self.name,
sys.platform)
# Start the actual connection # Start the actual connection
self._socket.connect((ip, port)) self._socket.connect((ip, port))