38

在我的 python 脚本中，我使用以下命令激活了 TCP Keepalive：

x = s.setsockopt( socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)

我的目标是关闭套接字连接，如果 5 分钟没有传输（*）。我在 Windows 上工作，我的 python 脚本只接收而不向客户端程序传输任何数据。

我所知道的是，默认情况下，如果 2 小时内没有传输，那么只有我可以使用 try 和 except 关闭连接。我知道，对于 Windows，我可以通过转到注册表来手动减少等待时间。但是有没有一种方法可以从我的脚本中修改它？

(*) 这里的“不传输”是指“有东西悄悄地吃掉网络上的数据包”，而不是“我不想发送任何东西”。

4

2 回答 2

71

您可以使用 setsockopt() 在已打开的套接字上设置 TCP keepalive 计时器。

import socket

def set_keepalive_linux(sock, after_idle_sec=1, interval_sec=3, max_fails=5):
    """Set TCP keepalive on an open socket.

    It activates after 1 second (after_idle_sec) of idleness,
    then sends a keepalive ping once every 3 seconds (interval_sec),
    and closes the connection after 5 failed ping (max_fails), or 15 seconds
    """
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
    sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPIDLE, after_idle_sec)
    sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPINTVL, interval_sec)
    sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPCNT, max_fails)

def set_keepalive_osx(sock, after_idle_sec=1, interval_sec=3, max_fails=5):
    """Set TCP keepalive on an open socket.

    sends a keepalive ping once every 3 seconds (interval_sec)
    """
    # scraped from /usr/include, not exported by python's socket module
    TCP_KEEPALIVE = 0x10
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
    sock.setsockopt(socket.IPPROTO_TCP, TCP_KEEPALIVE, interval_sec)

有关 Windows 上的等效选项，请参阅 msdn。查看Python 源代码，您似乎需要在 Unix 中设置SO_KEEPALIVE类似于sock.setsockopt，并且 [可选？] 设置SIO_KEEPALIVE_VALS为sock.ioctl.

于 2013-02-13T14:25:22.337 回答
15

对于 Windows，在 python 中：

这将启用套接字保持活动，具有 10 秒的保持活动时间和 3 秒的保持活动间隔。

sock.ioctl(socket.SIO_KEEPALIVE_VALS, (1, 10000, 3000))
于 2015-08-07T01:13:28.980 回答