如何检测客户端何时断开与UDS的连接(Unix域套接字)

时间:2015-01-14 15:40:26

标签: python sockets unix-socket

当客户端连接到管道并发送数据时,我可以收到这个罚款,我可以继续接收数据。当客户端断开连接并且while循环仍处于活动状态时,遇到麻烦,connection.recv()不会阻塞,因此会疯狂地循环!所以我需要一种方法来检测客户端是否仍然连接。

我有以下代码:

    pipe = './pipes/uds_defzone-lrecv'

    try:
        os.unlink(pipe)
    except OSError:
        if os.path.exists(pipe):
            raise
    self.logger.debug('Created UDS pipe: ' + pipe)

    sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
    sock.bind(pipe)
    sock.listen(1)

    self.logger.debug('Waiting for connection: ' + pipe)
    connection, client_address = sock.accept()
    self.logger.debug('Connection from: ' + client_address)

    while True:

        self.logger.debug('Waiting for data')
        data = connection.recv(4096)
        self.logger.debug('Received: ' + str(data))

作为参考,sender.py代码:

# Create a UDS socket
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)

# Connect the socket to the port where the server is listening
pipe = './pipes/uds_defzone-lrecv'

logger.debug('connecting to: ' + pipe)

try:
    sock.connect(pipe)
except socket.error, msg:
    logger.debug(msg)
    sys.exit(1)

try:
    message = 'THIS IS A TEST'
    logger.debug('sending: ' + message)
    sock.sendall(message)
    time.sleep(2)

finally:
    logger.debug('closing socket')
    sock.close()

TIA!

更新

我可以使用下面的代码减慢速度,但不完全是我想要的。

    while True:

        try:
            self.logger.debug('Waiting for data')
            data_present = select.select([sock], [], [], 30)
            if data_present[0]:
                data = connection.recv(4096)
                self.logger.debug('Received: ' + data)
        except select.timeout:
            pass

更新2

作为参考,这是我提出的代码:

        while True:

            logger.debug('Waiting for data')
            data = connection.recv(4096)
            if not data == '':
                logger.debug('Received: ' + data)
            else:
                logger.debug('Nothing received')
                break

我在这个过程中想出了一个黑客...可能在客户端可能发送空数据的合法地方可用,或许可能发出信号?

        while True:

            try:
                logger.debug('Waiting for data')
                data = connection.recv(4096)
                # *** This throws an exception when client has disconnected
                x = connection.getpeername()
                logger.debug('Received: ' + data)
            except:
                logger.debug('Client disconnected')
                break

1 个答案:

答案 0 :(得分:3)

  

connection.recv()不会阻塞,因此会疯狂地循环!所以我需要一种方法来检测客户端是否仍然连接。

如果对等方断开连接,则recv数据将返回空数据('')。您需要检查并退出循环。