基于TCP协议的大文件传输(粘包问题处理)

基于TCP的大文件上传服务端实现

# 服务端
# -*- coding: utf-8 -*-

from socket import *
import json, struct

server = socket(AF_INET, SOCK_STREAM)
server.bind((\'127.0.0.1\', 8080))  # 本地循环地址
server.listen(5)

while True:
    # 链接循环  accept() -> (socket object, address info)
    conn, addr = server.accept()
    try:
        # 先接收报头
        header = conn.recv(4)
        # 解析报头获取字典长度  unpack(fmt, buffer) -> (v1, v2, ...)
        header_len = struct.unpack(\'i\', header)[0]
        # 获取字典
        dic_bytes = conn.recv(header_len)
        dic_json = dic_bytes.decode(\'utf-8\')
        dic = json.loads(dic_json)
        file_size = dic.get(\'file_size\')
        file_name = dic.get(\'file_name\')
        # 接收文件
        recv_size = 0
        with open(file_name, \'wb\') as f:
            # 循环接收文件数据
            while recv_size < file_size:
                data = conn.recv(1024)
                f.write(data)  # 写进文件
                recv_size += len(data)
        print(dic.get(\'msg\'))
        conn.send(\'文件传输成功\'.encode(\'utf-8\'))
    except ConnectionResetError:
        break

#客户端

# -*- coding: utf-8 -*-
import socket
import os
import json
import struct

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect((\'127.0.0.1\', 8080))

# 文件大小和名字
file_size = os.path.getsize(r\'E:测试视频.mp4\')  # 文件地址自己写入
file_name = \'性感荷官在线发牌.mp4\'
# 定义一个字典
dic = {
    \'file_size\': file_size,
    \'file_name\': file_name,
    \'msg\': \'注意身体哟\'
}
dic_bytes = json.dumps(dic).encode()
# 制作字典报头
dic_header = struct.pack(\'i\', len(dic_bytes))
# 发送报头
client.send(dic_header)
# 发送字典
client.send(dic_bytes)
# 发送文件
with open(r\'E:测试视频.mp4\', \'rb\') as f:
    for line in f:
        client.send(line)

data = client.recv(1024).decode(\'utf-8\')
print(data)