在Python asyncio中检测套接字EOF

Question

I'm writing a simple socket server that receives some messages.

The only challenge left: If the client sends EOF, the connection does not close or the EOF is detected.

#!/usr/bin/env python

import asyncio


class Protocol(asyncio.Protocol):
    def connection_made(self, transport):
        self.peername = transport.get_extra_info("peername")
        print("Connection from %s" % (self.peername,))
        self.transport = transport

    def eof_received(self):
        print("end of stream!")
        self.close()

    def close(self):
        self.transport.close()

    def connection_lost(self, exc):
        print("Connection lost with %s" % (self.peername,))

    def data_received(self, data):
        print("received: %s" % data)


def main():
    loop = asyncio.get_event_loop()
    coro = loop.create_server(Protocol, "localhost", 1337)
    server = loop.run_until_complete(coro)

    print("Listening on %s..." % (server.sockets[0].getsockname(),))

    try:
        loop.run_forever()
    except KeyboardInterrupt:
        print("exiting...")

    server.close()
    loop.run_until_complete(server.wait_closed())
    loop.close()

if __name__ == "__main__":
    main()

I'm connecting there with strace nc localhost 1337 . When I send lines to nc , they are receieved of course. When I type Ctrl-D in nc , strace instantly reveals that socket was closed.

But the python script does not notice the EOF and keeps the connection open. When I kill nc , then the connection closes.

What can I do to close the connection in the asyncio protocol as soon as nc sends the EOF?

Answer 1

When I type Ctrl-D in nc, strace instantly reveals that socket was closed.

On my system w/ gnu-netcat , I have to run netcat with the -c option for the socket to be shutdown on Ctrl-D. Your script works as expected.

nc6 -x does the same thing, close the socket on eof, not just stdin .

Answer 2

It appears you have to raise an exception to end the loop.

Here is what I have come up with:

class Protocol(asyncio.Protocol):
    def connection_made(self, transport):
        self.peername = transport.get_extra_info("peername")
        print("Connection from %s" % (self.peername,))
        self.transport = transport

    def eof_received(self):
        print("end of stream!")

    def close(self):
        raise KeyboardInterrupt

    def connection_lost(self, exc):
        print("Connection lost with %s" % (self.peername,))
        self.close()

    def data_received(self, data):
        print("received: %s" % data)