python实现ftp文件传输功能
作者:zjtMeng
这篇文章主要为大家详细介绍了python实现ftp文件传输功能,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
本文实例为大家分享了python实现ftp文件传输的具体代码,供大家参考,具体内容如下
主要步骤可以分为以下几步:
1.读取文件名
2.检测文件是否存在
3.打开文件
4.检测文件大小
5.发送文件大小和 md5值给客户端
6.等客户端确认
7.开始边读边发数据
服务器端代码:
import socket,os,time import hashlib server =socket.socket() server.bind(('0.0.0.0',6666)) server.listen() print("等待....") while True: conn,addr = server.accept() print("new conn:",conn) while True: data = conn.recv(1024) if not data: print("client is disconnection") break cmd,filename = data.decode().split() #记录指令和文件名 print(filename) #判断当前目录是否存在该文件,而且必须是文件,而不是目录 if os.path.isfile(filename): f = open(filename,'rb') #m = hashlib.md5() # 创建md5 file_size = os.stat(filename).st_size #stat() 可以返回文件的大小值 conn.send((str(file_size)).encode()) # 发送文件大小 conn.recv(1024) #等待返回信息 for line in f: # m.updata(line) conn.send(line) #print("file md5",m.hexdigest()) #打印md5值 f.close()
客户端代码:
# Author: zjt import socket client = socket.socket() client.connect(("0.0.0.0",6666)) while True: cmd = input(">>>:").strip() if len(cmd)==0 :continue if cmd.startswith("get"): client.send(cmd.encode()) server_response = client.recv(1024) print("server response: ",server_response) client.send(b"ready to recv file") # 开始接收文件 file_total_size = int(server_response.decode()) received_size = 0 # 记录接收文件的大小 filename = cmd.split()[1] # 因为两个目录一致,接收的文件名不能与原文件相同 f = open(filename+".new","wb") while received_size < file_total_size: data = client.recv(1024) received_size += len(data) f.write(data) print("total:",file_total_size," present: ",received_size) else: print("file has received done!") f.close() client.close()
用80M的文件传输测试,效果如下:
程序升级:
前面的代码还没添加md5进行验证,现在对代码进行升级
服务器端代码:
import socket,os,time import hashlib server =socket.socket() server.bind(('0.0.0.0',8888)) server.listen() print("等待....") while True: conn,addr = server.accept() print("new conn:",conn) while True: data = conn.recv(1024) if not data: print("client is disconnection") break cmd,filename = data.decode().split() #记录指令和文件名 print(filename) #判断当前目录是否存在该文件,而且必须是文件,而不是目录 if os.path.isfile(filename): f = open(filename,'rb') m = hashlib.md5() # 创建md5 file_size = os.stat(filename).st_size #stat() 可以返回文件的大小值 conn.send((str(file_size)).encode()) # 发送文件大小 conn.recv(1024) #等待返回信息 for line in f: m.update(line) conn.send(line) print("file md5",m.hexdigest()) #打印md5值 f.close() conn.send(m.hexdigest().encode()) # 发送md5 print("我真的已经发过去了",m.hexdigest().encode()) print("send done") server.close()
客户端代码:
import socket import hashlib client = socket.socket() client.connect(("0.0.0.0",8888)) while True: cmd = input(">>>:").strip() if len(cmd)==0 :continue if cmd.startswith("get"): client.send(cmd.encode()) server_response = client.recv(1024) print("server response: ",server_response) client.send(b"ready to recv file") # 开始接收文件 file_total_size = int(server_response.decode()) received_size = 0 # 记录接收文件的大小 filename = cmd.split()[1] # 因为两个目录一致,接收的文件名不能与原文件相同 f = open(filename+".new","wb") m = hashlib.md5() while received_size < file_total_size: data = client.recv(1024) received_size += len(data) m.update(data) f.write(data) #print("total:",file_total_size," present: ",received_size) else: new_file_md5 = m.hexdigest() print("client file md5:",new_file_md5) print("file has received done!") print("total:",file_total_size," present: ",received_size) f.close() sever_file_md5 = client.recv(1024) print("client file md5:",new_file_md5) print("server file md5:",sever_file_md5) client.close()
两个程序在linux 环境下运行,结果如下:
可以看到传输后文件大小变大了一点点,而且md5前后值也不同,说明文件传输发生了改变。
现在讲程序在windows环境下运行,结果如下:
此时可以看到windows上没有问题,文件大小相同,且md5值也一致。
原因分析:
之所以会发生这种情况,是因为在linux上运行时,最后一次传输文件与发送md5值的时候,发生可粘包,导致最后一次接收文件的时候,连同md5的数据一并发送了。而客户端也当作一条接收信息,全部接收了。所以客户端出现没有收到来自服务器端的md5值,多出来的那一点点,就是md5值的大小。
解决方法:
在接收文件的时候,判断当前剩余多少文件需要接收,如果大于1024,就接收1024大小的文件,否则就只接收剩下全部的文件,防止最后一次接收多余的数据。
只需要对客户端代码进行修改,修改后代码如下:
import socket import hashlib client = socket.socket() client.connect(("0.0.0.0",8888)) while True: cmd = input(">>>:").strip() if len(cmd)==0 :continue if cmd.startswith("get"): client.send(cmd.encode()) server_response = client.recv(1024) print("server response: ",server_response) client.send(b"ready to recv file") # 开始接收文件 file_total_size = int(server_response.decode()) received_size = 0 # 记录接收文件的大小 filename = cmd.split()[1] f = open(filename+".new","wb") m = hashlib.md5() while received_size < file_total_size: #添加一次判断,使最后一次剩多少就接收多少,避免发生粘包 if file_total_size - received_size > 1024: size = 1024 else: # 最后一次,剩多少收多少 size = file_total_size - received_size data = client.recv(size) received_size += len(data) m.update(data) f.write(data) else: new_file_md5 = m.hexdigest() print("client file md5:",new_file_md5) print("file has received done!") print("total:",file_total_size," present: ",received_size) print("下一句关闭文件") f.close() print("开始接收md5 ") sever_file_md5 = client.recv(1024) print("client file md5:",new_file_md5) print("server file md5:",sever_file_md5) client.close()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。