套接字正在等待连接超时

Question

我正在尝试实现一个超时，当在定义的时间间隔内没有接收到连接时，该超时会终止python脚本。到目前为止，我使用以下代码实现了超时：

import sys
import socket
# Create a TCP/IP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Bind the socket to the port
server_address = ('192.168.43.112', 5001)
print >>sys.stderr, 'starting up on %s port %s' % server_address
sock.bind(server_address)
# Listen for incoming connections
sock.listen(1)
while True:
    try:
        # Wait for a connection
        print >>sys.stderr, 'waiting for a connection'
        connection, client_address = sock.accept()
        try:
            print >>sys.stderr, 'connection from', client_address
            # Receive the data in small chunks and retransmit it
            while True:
                data = connection.recv(16)
                print >>sys.stderr, 'received "%s"' % data
                if data:
                    print >>sys.stderr, 'Do stuff here'
                else:
                    print >>sys.stderr, 'no more data from', client_address
                    sock.settimeout(5)
                    break
        finally:
            # Clean up the connection
            connection.close()
    except socket.timeout:
        break

在建立连接并结束同一连接后，脚本在5秒钟后终止，从这个意义上说，代码工作正常。然而，如果在超时窗口期间我尝试进行另一个连接，我会出现以下错误：

starting up on 192.168.43.112 port 5001
waiting for a connection
connection from ('192.168.43.1', 47550)
received "Data 0
"
Do stuff here
received ""
no more data from ('192.168.43.1', 47550)
waiting for a connection
connection from ('192.168.43.1', 39010)
---------------------------------------------------------------------------
error                                     Traceback (most recent call last)
/Users/location/Desktop/sandbox/data_fetcher.py in <module>()
     24             # Receive the data in small chunks and retransmit it
     25             while True:
---> 26                 data = connection.recv(16)
     27                 print >>sys.stderr, 'received "%s"' % data
     28                 if data:
error: [Errno 35] Resource temporarily unavailable

Answer 1

我不完全确定你希望这一切如何工作，我发现现在这种情况发生有点令人惊讶（我没想到超时会有这种效果），但基于EAGAIN错误（错误号35），发生的情况是，主套接字上的超时（只有在第一次连接后才会设置）导致第二个接受的套接字也处于非阻塞模式。这意味着，当您调用connection.recv并且没有立即的数据时，您会得到该OSError。

我怀疑其中一些操作系统可能会有所不同，但我能够在FreeBSD上复制它（您可能在Linux上运行）。

一个最小的改变——我不认为这一定是最好的编码方式，但它确实有效——就是显式地将可接受的套接字设置为阻塞：

        # Wait for a connection
        print >>sys.stderr, 'waiting for a connection'
        connection, client_address = sock.accept()
        connection.setblocking(1)

有了这一点，代码的表现要好得多（我添加了一个小的测试框架，它将代码作为一个单独的过程进行分解，然后以不同的延迟进行几个连接）。