如何在 python 的 socket recv 方法上设置超时？

Posted 2023-02-25

【中文标题】如何在 python 的 socket recv 方法上设置超时？

【英文标题】：How to set timeout on python's socket recv method?

【发布时间】：2011-02-12 17:23:15

【问题描述】：

我需要在 python 的 socket recv 方法上设置超时。怎么办？

【问题讨论】：

仅供参考，如果您确实选择使用超时...您需要知道如何处理超时。这个 SO 问题讨论了超时发生时的处理：***.com/questions/16745409

【参考方案1】：

典型的方法是使用select() 等待数据可用或超时。只有在数据实际可用时才调用recv()。为了安全起见，我们还将套接字设置为非阻塞模式，以保证recv() 永远不会无限期阻塞。 select() 也可用于一次等待多个套接字。

import select

mysocket.setblocking(0)

ready = select.select([mysocket], [], [], timeout_in_seconds)
if ready[0]:
    data = mysocket.recv(4096)

如果您有很多打开的文件描述符，poll() 是select() 更有效的替代方案。

另一种选择是使用socket.settimeout() 为套接字上的所有操作设置超时，但我看到您在另一个答案中明确拒绝了该解决方案。

【讨论】：

使用select 很好，但是你说“你不能”的部分是误导性的，因为有socket.settimeout()。

现在好多了，但我看不出答案是“明确拒绝”的。

使用select 的一个注意事项——如果你在Windows 机器上运行，select 依赖于WinSock 库，它有一个习惯，即some 数据已到达，但不一定是全部。因此，您需要合并一个循环来继续调用select.select()，直到收到所有数据。你如何知道你已经得到了所有的数据（不幸的是）取决于你自己去弄清楚——这可能意味着寻找一个终止符字符串、一定数量的字节，或者只是等待一个定义的超时。

为什么要设置socket非阻塞？我认为这对 select 调用无关紧要（它会阻塞直到可以读取描述符或在这种情况下超时到期）并且如果满足 select 则 recv() 不会阻塞。我使用 recvfrom() 进行了尝试，它似乎可以在没有 setblocking(0) 的情况下正常工作。

ready[0] 只有在服务器响应中没有正文的情况下才会为假？

【参考方案2】：

有socket.settimeout()

【讨论】：

recv 不会超时（至少在我尝试过时）。只有accept()超时。

在设置 socket.settimeout() 后，socket.recv() 对我来说似乎超时了，完全符合预期。这是我编造的吗？其他人可以证实这一点吗？

@Aeonaut 我认为这在大多数情况下都会超时 recv() ，但存在竞争条件。在 socket.recv() Python (2.6) 在内部调用 select/poll 并超时，然后立即调用 recv()。因此，如果您使用阻塞套接字并且在这两个调用之间另一个端点崩溃，您最终可能会无限期地挂在 recv() 上。如果你使用非阻塞套接字，python不会在内部调用select.select，所以我认为Daniel Stutzbach的回答是正确的方法。

其实select()返回的时候我可能理解错了，所以请把之前的评论划掉。 Beej's Guide 说以上是在 recv 上实现超时的有效方法：beej.us/guide/bgnet/output/html/singlepage/… 所以我相信这是一个权威来源。

我不知道为什么使用 select 的解决方案是一个单一的解决方案（更易于维护，实施错误的风险更小）并在后台使用 select（实施是与@DanielStuzbach 的答案相同）。

【参考方案3】：

如前所述，select.select() 和 socket.settimeout() 都可以使用。

请注意，您可能需要致电 settimeout 两次以满足您的需要，例如

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(("",0))
sock.listen(1)
# accept can throw socket.timeout
sock.settimeout(5.0)
conn, addr = sock.accept()

# recv can throw socket.timeout
conn.settimeout(5.0)
conn.recv(1024)

【讨论】：

我认为无论你如何戳和戳这个功能，它都会挂起。我现在尝试了 2 或 4 次超时，但它仍然挂起。 settimeout 也挂起。

当您多次调用.settimeout() 时，您可以首先调用setdefaulttimeout() 方法。

【参考方案4】：

您可以在收到响应之前设置超时，在收到响应后将其设置回无：

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

sock.settimeout(5.0)
data = sock.recv(1024)
sock.settimeout(None)

【参考方案5】：

如果您实现服务器端，您正在寻找的超时是连接套接字的超时而不是主套接字的超时。也就是说，连接socket对象还有一个超时时间，就是socket.accept()方法的输出。因此：

sock.listen(1)
connection, client_address = sock.accept()
connection.settimeout(5)    # This is the one that affects recv() method.
connection.gettimeout()     # This should result 5
sock.gettimeout()           # This outputs None when not set previously, if I remember correctly.

如果你实现客户端，那就很简单了。

sock.connect(server_address)
sock.settimeout(3)

【参考方案6】：

试试这个，它使用底层 C。

timeval = struct.pack('ll', 2, 100)
s.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO, timeval)

【讨论】：

这很棒，因为它允许使用SO_RCVTIMEO 和SO_SNDTIMEO 为发送和接收超时设置不同的值。

为什么是2 以及为什么是100？哪个是超时值？在哪个单位？

timeval = struct.pack('ll', sec, usec) s.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO, timeval) usec = 10000 表示 10 毫秒

这在 Windows 上不起作用，因为参数是 DWORD（64 位）docs.microsoft.com/en-us/windows/win32/winsock/…

【参考方案7】：

您可以使用socket.settimeout()，它接受一个表示秒数的整数参数。例如，socket.settimeout(1) 将超时设置为 1 秒

【参考方案8】：

对最重要的答案有点困惑，所以I've wrote a small gist 提供示例以便更好地理解。

---

选项 #1 - socket.settimeout()

如果sock.recv() 等待超过定义的超时时间，将引发异常。

import socket

sock = socket.create_connection(('neverssl.com', 80))
timeout_seconds = 2
sock.settimeout(timeout_seconds)
sock.send(b'GET / HTTP/1.1\r\nHost: neverssl.com\r\n\r\n')
data = sock.recv(4096)
data = sock.recv(4096) # <- will raise a socket.timeout exception here

---

选项 #2 - select.select()

等待数据发送，直到达到超时。我已经调整了Daniel's answer，所以它会引发异常

import select
import socket

def recv_timeout(sock, bytes_to_read, timeout_seconds):
    sock.setblocking(0)
    ready = select.select([sock], [], [], timeout_seconds)
    if ready[0]:
        return sock.recv(bytes_to_read)

    raise socket.timeout()

sock = socket.create_connection(('neverssl.com', 80))
timeout_seconds = 2
sock.send(b'GET / HTTP/1.1\r\nHost: neverssl.com\r\n\r\n')
data = recv_timeout(sock, 4096, timeout_seconds)
data = recv_timeout(sock, 4096, timeout_seconds) # <- will raise a socket.timeout exception here

【参考方案9】：

正如之前的回复中提到的，您可以使用类似：.settimeout() 例如：

import socket

s = socket.socket()

s.settimeout(1) # Sets the socket to timeout after 1 second of no activity

host, port = "somehost", 4444
s.connect((host, port))

s.send("Hello World!\r\n")

try:
    rec = s.recv(100) # try to receive 100 bytes
except socket.timeout: # fail after 1 second of no activity
    print("Didn't receive data! [Timeout]")
finally:
    s.close()

希望对你有帮助！！

【参考方案10】：

#! /usr/bin/python3.6

# -*- coding: utf-8 -*-
import socket
import time
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
s.settimeout(5)
PORT = 10801

s.bind(('', PORT))
print('Listening for broadcast at ', s.getsockname())
BUFFER_SIZE = 4096
while True:
    try:
        data, address = s.recvfrom(BUFFER_SIZE)
    except socket.timeout:
        print("Didn't receive data! [Timeout 5s]")
        continue

【参考方案11】：

大声疾呼：https://boltons.readthedocs.io/en/latest/socketutils.html

它提供了一个缓冲套接字，这提供了很多非常有用的功能，例如：

.recv_until()    #recv until occurrence of bytes
.recv_closed()   #recv until close
.peek()          #peek at buffer but don't pop values
.settimeout()    #configure timeout (including recv timeout)