Python Socket接收图片时陷入循环且尾部缺失，如何修复？

问题原因 #

你的代码核心问题是：客户端发送完所有图片数据后，没有告知服务器“数据已发送完毕”，导致服务器的conn.recv(2048)一直阻塞等待下一段数据，无法跳出循环，最终文件未正常关闭，图片尾部数据缺失。另外客户端代码里还有个小错误：file.close少了括号，应该是file.close()，不过这不是主要问题。

解决方案 #

这里提供两种可靠的修复方式：

方式一：发送完数据后关闭Socket写端 #

客户端发送完所有数据后，调用s.shutdown(socket.SHUT_WR)关闭写操作，这样服务器的recv会收到空字节，从而跳出循环。

修改后的客户端代码：

import socket  # 确保导入socket模块
if dataAlt[0] == 'receiveImage':
    try:
        file = open(f'ClientSaves/{dataAlt[1]}','rb')
        print('sending image...')
        image_data = file.read(2048)
        while image_data:
            s.send(image_data)
            image_data = file.read(2048)
        file.close()  # 修正：添加括号
        # 关闭Socket写端，通知服务器数据已发送完成
        s.shutdown(socket.SHUT_WR)
        print('image sent')
    except Exception as e:
        message = f'file {dataAlt[1]} send fail: {str(e)}'
        print(message)

服务器端代码无需修改，当客户端关闭写端后，conn.recv(2048)会返回空字节，循环结束，文件正常关闭。

方式二：预先发送文件大小（更可靠） #

通过先发送文件的总大小，让服务器明确知道需要接收多少字节，避免依赖连接状态判断结束，这种方式在复杂网络环境下更稳定。

客户端代码：

import os
import socket
if dataAlt[0] == 'receiveImage':
    try:
        file_path = f'ClientSaves/{dataAlt[1]}'
        # 获取文件总大小
        file_size = os.path.getsize(file_path)
        # 将文件大小转为4字节的二进制数据发送给服务器
        s.sendall(file_size.to_bytes(4, byteorder='big'))
        
        file = open(file_path,'rb')
        print('sending image...')
        sent_total = 0
        while sent_total < file_size:
            # 每次读取不超过剩余大小的chunk
            image_data = file.read(min(2048, file_size - sent_total))
            s.sendall(image_data)
            sent_total += len(image_data)
        file.close()
        print('image sent')
    except Exception as e:
        message = f'file {dataAlt[1]} send fail: {str(e)}'
        print(message)

服务器端代码：

if usedata[0] == 'receiveImage':
    file = open(f'ServerSaves/{usedata[1]}','wb')
    print('receiving image...')
    # 先接收4字节的文件大小数据
    size_data = conn.recv(4)
    if not size_data:
        file.close()
        print('接收文件大小失败')
        return
    file_size = int.from_bytes(size_data, byteorder='big')
    
    received_total = 0
    while received_total < file_size:
        # 每次接收不超过剩余需要的字节数
        image_chunk = conn.recv(min(2048, file_size - received_total))
        if not image_chunk:
            print('连接中断，接收未完成')
            break
        file.write(image_chunk)
        received_total += len(image_chunk)

    file.close()
    print('image received')

效果说明 #

两种方式都能解决服务器卡顿的问题，确保图片完整保存。方式二更适合需要保证数据完整性的场景，方式一则更简单直接。

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