python 基于tcp协议的文件传输3_解决粘包问题

Posted 2020-11-25 混子1

server

import json
import struct
import socket
# 接收
sk = socket.socket()
sk.bind((‘127.0.0.1‘,9001))
sk.listen()

conn,_ =sk.accept()
msg_len = conn.recv(4)
dic_len = struct.unpack(‘i‘,msg_len)[0]
msg = conn.recv(dic_len).decode(‘utf-8‘)
msg = json.loads(msg)

with open(msg[‘filename‘],‘wb‘) as f:
    while msg[‘filesize‘] > 0:
        content = conn.recv(1024)
        msg[‘filesize‘] -= len(content)
        f.write(content)
conn.close()
sk.close()

client

import os
import json
import struct
import socket
# 发送
sk = socket.socket()
# sk.connect((‘192.168.14.109‘,9012))
sk.connect((‘127.0.0.1‘,9001))

# 文件名文件大小
abs_path = r‘D:python22期day28 课上视频3.网络基础概念.mp4‘
filename = os.path.basename(abs_path)
filesize = os.path.getsize(abs_path)
dic = {‘filename‘:filename,‘filesize‘:filesize}
str_dic = json.dumps(dic)
b_dic = str_dic.encode(‘utf-8‘)
mlen = struct.pack(‘i‘,len(b_dic))
sk.send(mlen)   # 4个字节 表示字典转成字节之后的长度
sk.send(b_dic)  # 具体的字典数据

with open(abs_path,mode = ‘rb‘) as f:
    while filesize>0:
        content = f.read(1024)
        filesize -= len(content)
        sk.send(content)
sk.close()


# 先登录
# 能上传到固定目录下载(从指定文件夹下选择文件)