关于TCP协议容易出现的黏包问题说明可以参考:https://www.cnblogs.com/Eva-J/articles/8244551.html#_label6
本篇内容大多数也是参考上面博客学习而来,记录一下。
# struct模块:-->模块可以把一个类型,如数字,转成固定长度的bytes import struct ret = struct.pack('i',4096) # 'i'代表int, 就是即将要把一个数字转换成固定长度(4个字节)的bytes类型 print(ret) num = struct.unpack('i',ret) # 输出一个元组 print(num[0])
使用pack可以把一个数字转化成一个固定长度(4个字节)的bytes类型,在recv的时候就可以只接收这4个字节,然后unpack解析出来就能知道接下来需要传输文件的大小。
下面展示基于TCP协议,利用struct进行的视频文件的传输过程:
server.py
# 实现一个大文件的上传或者下载 # 配置文件 ip地址 端口号 import socket import struct import json sk = socket.socket() sk.bind(('127.0.0.1',8080)) sk.listen() buffer = 4096 conn,addr = sk.accept() # 接收 head_len = conn.recv(4) head_len = struct.unpack('i',head_len)[0] json_head = conn.recv(head_len).decode('utf-8') head = json.loads(json_head) filesize = head['filesize'] with open(head['filename'],'wb') as f: while filesize: if filesize >= buffer: content = conn.recv(buffer) f.write(content) filesize -= buffer else: content = conn.recv(filesize) f.write(content) break conn.close() sk.close()
client.py
# 发送端 import socket import os import json import struct sk = socket.socket() sk.connect(('127.0.0.1',8080)) buffer = 4096 # 发送文件 head = {'filepath':r'/Users/zxx/Downloads', 'filename':r'test.mp4', 'filesize':None} file_path = os.path.join(head['filepath'],head['filename']) filesize = os.path.getsize(file_path) head['filesize'] = filesize json_head = json.dumps(head) # 字典转成了字符串 bytes_head = json_head.encode('utf-8') # 字符串转bytes # print(json_head) # print(bytes_head) # 计算head的长度 head_len = len(bytes_head) pack_len = struct.pack('i',head_len) # print(head_len,pack_len) sk.send(pack_len) # 先发报头的长度 sk.send(bytes_head) # 在发送bytes类型报头 with open(file_path,'rb') as f: while filesize: if filesize >= buffer: content = f.read(buffer) # 每次读出来的内容 sk.send(content) filesize -= buffer else: content = f.read(filesize) sk.send(content) break sk.close()
先后执行完server.py 和client.py文件后,即可把本地在下载目录下面的test.mp4文件传输到python文件工作目录中。
结果如下: