python文件传输 python实现ftp文件传输功能
zjtMeng 人气:0主要步骤可以分为以下几步：
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()
加载全部内容