在Python中捕获断开的套接字

2 投票
1 回答
2344 浏览
提问于 2025-04-16 06:04

我在处理一个问题,就是当出现“管道损坏”的错误时,怎么检测到这个坏掉的连接。下面的代码可以作为例子:

服务器代码:

import errno, select, socket, time, SocketServer

class MetaServer(object):
    def __init__(self):
        self.server = Server(None, Handler, bind_and_activate=False)
    def run(self, sock, addr):
        rfile = sock.makefile('rb', 1)
        self.server.process_request(sock, addr)
        while 1:
            r, _, _ = select.select([rfile], [], [], 1.0)
            if r:
                print 'Got %s' % rfile.readline()
            else:
                print 'nothing to read'

class Server(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
    allow_reuse_address = True
    daemon_threads = True

class Handler(SocketServer.StreamRequestHandler):
    def handle(self):
        print 'connected!'
        try:
            while 1:
                self.wfile.write('testing...')
                time.sleep(1)
        except socket.error as e:
            if e.errno == errno.EPIPE:
                print 'Broken pipe!'
        self.finish()
        self.request.close()

if __name__ == '__main__':
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(('127.0.0.1', 8081))
    s.listen(5)
    ms = MetaServer()
    while 1:
        client, address = s.accept()
        ms.run(client, address)

客户端代码:

import select, socket

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1', 8081))
while 1:
    r, _, _ = select.select([s], [], [], 1.0)
    if not r:
        continue
    msg = s.recv(1024)
    print 'Got %s' % (msg,)

现在,如果我同时运行服务器和客户端,一切都很正常,每秒钟我都会收到一个“没有读取到数据”的消息。但是,一旦我按下CTRL-C退出客户端,服务器就会变得很疯狂,开始从一个应该坏掉的连接中“读取”数据,结果出现很多“收到”消息。

有没有什么方法可以在MetaServer.run()这个函数里检测到这个坏掉的连接,以避免上面提到的情况发生呢?

1 个回答

3

是的,这个内容在文档里并没有详细说明,但这是老旧的Unix系统的一个行为:当你收到一个空字符串时,你需要中止操作。

撰写回答