接收带有套接字的大型腌制对象时的无限循环答案

【问题标题】：infinite loop when receiving a large pickled object with sockets接收带有套接字的大型腌制对象时的无限循环
【发布时间】：2024-04-12 00:30:10
【问题描述】：

我正在使用远程 linux 服务器，并且我想使用 python 从客户端通过套接字发送一个数组，所以我使用了这段代码：

message = pickle.dumps(faceBlob)
message_header = bytes(f"{len(message):<{HEADER_LENGTH}}", "utf-8")
client_socket.send(message_header + message)

为了在服务器中接收它，我使用了一个 while 循环来捕获所有消息，因为它是 > 4096 ：

def receive_blop(client_socket):
    # Receive our "header" containing message length, it's size is defined and constant
    message_header = client_socket.recv(HEADER_LENGTH)

    # If we received no data, client gracefully closed a connection, for example using socket.close() or socket.shutdown(socket.SHUT_RDWR)
    if not len(message_header):
        return False

    # Convert header to int value
    message_length = int(message_header.decode('utf-8').strip())
    fragments = []
    print(message_length)

    while True:
        # this loop is infinite

        print("I arrived her")
        chunk = client_socket.recv(4096)
        if not chunk:
            break
        fragments.append(chunk)

    data_arr = b"".join(fragments)


    # Return an object of message header and message data
    return {'header': message_header, 'data': data_arr}

服务器仍在打印“我到达这里”，但会收到消息，直到来自客户端的连接结束

【问题讨论】：

标签： python sockets server client client-server

【解决方案1】：

您的循环将继续，直到客户端关闭连接。如果他们在收到您的回复后才关闭，那么您就陷入了僵局。

由于您知道消息长度，因此您可以在收到这么多字节时停止循环。

    received_length = 0
    while received_len < message_length:

        print("I arrived her")
        chunk = client_socket.recv(message_length - received_length)
        if not chunk:
            break
        fragments.append(chunk)
        received_length += len(chunk)

【讨论】：