基于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)