基于TCP协议传输文件

基于TCP协议传输文件

\'\'\'
版本一：
此版本一次写入整个文件大小，
只能传输小文件，大文件可能会传不全
\'\'\'
#server端----------------------------------------------
import socket
import json
sk = socket.socket()
sk.bind((\'127.0.0.1\',9001))
sk.listen()

con,addr = sk.accept()
msg = con.recv(1024).decode(\'utf-8\')
dic = json.loads(msg)

with open(dic[\'filename\'],mode=\'wb\') as f :
    content = con.recv(dic[\'filesize\'])
    print(\'--->\',len(content))
    f.write(content)
con.close()
sk.close()
#client端————————————————————————————————————————————————
import os
import socket
sk = socket.socket()
sk.connect((\'127.0.0.1\',9001))

#获取文件名|文件小
abs_path = r\'D:\Wang.YH\Videos\人民至上.mp4\'
filename = os.path.basename(abs_path)
filesize = os.path.getsize(abs_path)
dic = {\'filename\':filename,
       \'filesize\':filesize}

str_dic = json.dumps(dic)
sk.send(str_dic.encode(\'utf-8\'))

with open(abs_path,mode=\'rb\') as f:
    content = f.read()
    sk.send(content)
sk.close()

\'\'\'
版本二：
分次传输文件
不足：可能会产生粘包现象
\'\'\'
#serfer端——————————————————————————————————————
import socket
import json
sk = socket.socket()
sk.bind((\'127.0.0.1\',9001))
sk.listen()

con,addr = sk.accept()
msg = con.recv(1024).decode(\'utf-8\')
dic = json.loads(msg)
with open(dic[\'filename\'],mode=\'wb\') as f:
    while dic[\'filesize\'] >0:
        content = con.recv(1024)
        dic[\'filesize\'] -=len(content)
        f.write(content)
con.close()
sk.close()
#client端————————————————————————————————————————————————
import os
import socket
import json
sk = socket.socket()
sk.connect((\'127.0.0.1\',9001))
#获取文件名|文件小
abs_path = r\'D:\Wang.YH\Videos\人民至上.mp4\'
filename = os.path.basename(abs_path)
filesize = os.path.getsize(abs_path)
dic = {\'filename\':filename,
       \'filesize\':filesize}

str_dic = json.dumps(dic)
sk.send(str_dic.encode(\'utf-8\'))

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

\'\'\'
版本三：
解决粘包
\'\'\'
#server端————————————————————————————————————————
import socket
import json
import struct
sk = socket.socket()
sk.bind((\'127.0.0.1\',9001))
sk.listen()

con,addr = sk.accept()

msg_len = con.recv(4)
dic_len = struct.unpack(\'i\',msg_len)[0]
msg = con.recv(dic_len).decode(\'utf-8\')
dic = json.loads(msg)


with open(dic[\'filename\'],mode=\'wb\') as f:
    while dic[\'filesize\'] >0:
        content = con.recv(1024)
        dic[\'filesize\'] -=len(content)
        f.write(content)

con.close()
sk.close()

#client端——————————————————————————————————
import os
import socket
import json
import struct

sk = socket.socket()
sk.connect((\'127.0.0.1\',9001))

#获取文件名|文件小
abs_path = r\'D:\Wang.YH\Videos\人民至上.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)
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()