TCP Socket传输图像完整接收及客户端响应异常排查

问题核心原因 #

当前实现的卡滞点出在TCP连接半关闭的时序逻辑上：

• 客户端接收响应的退出条件是recv返回空字节，该场景只有在服务端主动关闭连接/关闭自身写方向时才会触发
• 现有服务端逻辑中，发送完响应后没有立刻关闭写方向，反而卡在cv2.waitKey的等待逻辑中，连接一直处于占用状态，客户端无法感知响应传输完成，会一直阻塞在recv循环
• 同时现有逻辑将图像显示、按键判断放在连接上下文块内，会无意义占用连接生命周期，进一步加剧阻塞

修复要点 #

• 服务端完成响应发送后，立刻主动关闭写方向，通知客户端响应已传输完成
• 将图像解码、显示、按键判断这类和网络传输无关的逻辑，放到连接关闭之后执行，避免不必要的连接占用
• 调整服务端退出逻辑，确保按q键时可以正常退出整个服务循环，而不是仅断开当前连接

修正后代码 #

Client.py（原有逻辑仅补充连接关闭操作，核心逻辑可直接复用） #

import numpy as np
import cv2
from PIL import Image
import base64
import socket
def main(data):
    s = socket.socket()
    s.connect(("127.0.0.1", 999))

    decoded_data = base64.b64decode(data)

    print("Sending...")
    s.sendall(decoded_data)
    # 关闭客户端写方向，通知服务端图像发送完成
    s.shutdown(s.SHUT_WR)

    b_data = b''
    while True:
        txt_data = s.recv(2048)
        if not txt_data: 
            break
        b_data += txt_data
    print('response received from the server: ' + b_data.decode())
    s.close()
    return b_data.decode()

Server.py（调整连接生命周期逻辑） #

import socket
from PIL import Image
import io
import numpy as np
import cv2
import uuid

IP = '127.0.0.1'
PORT = 999
exit_flag = False

with socket.socket() as s:
    s.bind((IP,PORT))
    s.listen(1)
    count = 0
    print ('The server is ready')
    while not exit_flag:
        con, addr = s.accept()
        count = count + 1
        img_dir = 'C:/Users/my_user/stream_images/'
        img_format = '.png'
        img_bytes = b''
        # 连接上下文内仅处理网络收发和必要的图像解码
        with con:
            while True:
                data = con.recv(2048)
                if not data:
                    break
                img_bytes += data
            # 提前完成图像解码，减少连接占用时间
            image_name = img_dir+'frame'+str(count)+img_format
            pil_image = io.BytesIO(img_bytes)
            img_np = np.array(Image.open(pil_image))
            img_np = cv2.rotate(img_np, cv2.ROTATE_90_CLOCKWISE)
            # 发送响应
            cCode = 'Thank you for connecting'
            con.sendall(cCode.encode())
            print("called con.sendall")
            # 主动关闭服务端写方向，通知客户端响应传输完成
            con.shutdown(socket.SHUT_WR)
        
        # 连接关闭后再做UI显示和按键判断，不占用网络连接资源
        cv2.imshow('frame', img_np)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            exit_flag = True
            break
cv2.destroyAllWindows()

多图持续传输优化建议 #

如果后续要持续传输多张图像，不建议每张图新建一次TCP连接（三次握手、断连开销大），可以自定义轻量应用层协议提升效率：

• 每次发图像前，先发固定4字节的图像长度标识（可用struct.pack将长度转为4字节无符号整数）
• 服务端先读4字节拿到本次图像的总长度，再循环recv直到收够指定长度的字节，即可判定单张图接收完成，不需要靠关闭连接判断传输边界
• 响应也用同样的方式加长度头，双方收发都不需要靠断连判断结束，单连接可持续传输多张图，稳定性和传输效率都会明显提升

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