Python使用PIL与BytesIO经Socket传图片随机被截断问题求助

随机报错的核心原因是对TCP socket的recv方法行为认知偏差：

• TCP是面向字节流的协议，无天然消息边界，recv(size)的语义是最多读取size个字节，而非必须读满size字节才返回。只要内核接收缓冲区有数据，无论数据量多少都会立刻返回，极端场景下网络暂未传输数据时，还会返回空字节串。
• 现有代码仅调用一次recv，网络状态好时可一次性收全完整PNG数据则运行正常，若仅收到部分数据，BytesIO中存储的是不完整的PNG文件，解析时就会抛出文件截断错误。
• 报错struct.error: unpack_from requires a buffer of at least 4 bytes是因为单次recv返回了空字节，连解包所需的4字节长度数据都未收全。

核心思路是发送数据前先传输固定长度的内容大小标识，接收端先收全长度标识，再循环读取直至收满对应大小的完整数据，避免单次recv返回数据不全的问题。
修改后的完整实现如下：

import struct
import io
from PIL import Image

# 新增工具方法：循环调用recv直至收满指定大小的字节
def recv_exact(self, size):
    buffer = b""
    while len(buffer) < size:
        chunk = self.conn.recv(size - len(buffer))
        if not chunk:
            raise ConnectionError("Socket连接中断，数据接收失败")
        buffer += chunk
    return buffer

def exchange_image(self, img_send):
    fd = io.BytesIO()
    img_send.save(fd, "png")
    img_bytes = fd.getvalue()
    img_recv = self.exchange(img_bytes)
    return Image.open(io.BytesIO(img_recv))
    
def exchange(self, to_send):
    # 将要发送的内容长度打包为4字节大端序整数，作为消息头
    send_header = struct.pack(">I", len(to_send))
    if self.role == "client":
        # 先发消息头，再发实际内容
        self.conn.sendall(send_header)
        self.conn.sendall(to_send)
        # 先接收响应的4字节消息头，解析出响应内容长度后收全数据
        recv_header = self.recv_exact(4)
        recv_len = struct.unpack(">I", recv_header)[0]
        return self.recv_exact(recv_len)
    elif self.role == "server":
        # 先接收客户端请求的4字节消息头，解析出请求内容长度后收全数据
        recv_header = self.recv_exact(4)
        recv_len = struct.unpack(">I", recv_header)[0]
        received = self.recv_exact(recv_len)
        # 发送响应的消息头和实际内容
        self.conn.sendall(struct.pack(">I", len(to_send)))
        self.conn.sendall(to_send)
        return received

• 无需再硬编码MAX_IMG_SIZE作为接收参数，通过消息头传递长度的方式可自适应任意合法大小的图片，同时避免接收多余字节影响后续消息解析。
• 若传输超大图片，可调整单次recv的读取块大小，减少内存拷贝开销。

内容的提问来源于stack exchange，提问作者WernerDrasche