Python Socket编程：确保MP4等大文件流完整接收的实践指南

# 客户端（接收方）原始逻辑
import socket

if __name__ == '__main__':
    soc = socket.socket()
    # 假设已连接到服务器，并通过某种方式获取到预期数据长度data_len
    # soc.connect(('6.tcp.eu.ngrok.io', 19717)) # 示例连接
    # data_len = int(soc.recv(16).decode()) # 示例接收长度

    # 假设data_len已获取
    data_len = 102400 # 假设总数据长度为100KB

    with open('new.mp4', 'wb') as f:
        read = 0
        while read < data_len:
            # 错误假设：recv(4096)总是返回4096字节
            f.write(soc.recv(4096))
            read += 4096

# 客户端（接收方）修正后的逻辑
import socket

def receive_all_data(sock, total_data_len):
    """
    从socket接收指定长度的所有数据。
    """
    received_data = b''
    current_read_len = 0
    chunk_size = 4096 # 每次尝试接收的字节数

    while current_read_len < total_data_len:
        # 接收数据，data是实际接收到的字节串
        data = sock.recv(chunk_size)

        # 检查对端是否关闭连接
        if not data:
            print(f"警告：对端在接收到 {current_read_len}/{total_data_len} 字节后关闭了连接。")
            break # 对端关闭，退出循环

        # 将接收到的数据添加到总数据中
        received_data += data
        # 更新已接收的字节数，使用len(data)而不是chunk_size
        current_read_len += len(data)

    return received_data

if __name__ == '__main__':
    soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        # 假设服务器地址和端口
        server_address = ('6.tcp.eu.ngrok.io', 19717) # 替换为你的ngrok地址和端口
        print(f"尝试连接到 {server_address[0]}:{server_address[1]}...")
        soc.connect(server_address)
        print("连接成功！")

        # 第一步：接收数据长度信息
        # 服务器通常会先发送一个固定长度的字符串，表示后续数据的总长度
        len_header = soc.recv(16).decode().strip('0') # 假设长度信息是16位定长，左侧补0
        if not len_header:
            print("错误：未接收到数据长度头部信息或对端关闭。")
            soc.close()
            exit()

        total_data_len = int(len_header)
        print(f"预期接收数据总长度：{total_data_len} 字节")

        # 第二步：接收实际数据
        # 使用封装的函数确保完整接收
        file_data = receive_all_data(soc, total_data_len)

        # 检查是否接收完整
        if len(file_data) == total_data_len:
            print(f"成功接收 {len(file_data)} 字节数据。")
            with open('new_received.mp4', 'wb') as f:
                f.write(file_data)
            print("文件 'new_received.mp4' 已保存。")
        else:
            print(f"接收数据不完整。预期 {total_data_len} 字节，实际接收 {len(file_data)} 字节。")

    except ConnectionRefusedError:
        print(f"错误：连接被拒绝。请检查服务器是否运行，以及地址 {server_address} 是否正确。")
    except Exception as e:
        print(f"发生错误：{e}")
    finally:
        soc.close()
        print("Socket连接已关闭。")