在介绍之前,我们需要了解一个加密算法
MD5 校验和(checksum)通过对接收的传输数据执行散列运算来检查数据的正确性。计算出的散列值拿来和随数据传输的散列值比较。如果两个值相同,说明传输的数据完整无误、没有被窜改过(前提是散列值没有被窜改),从而可以放心使用。
如客户往我们数据中心同步一个文件,该文件使用MD5校验,那么客户在发送文件的同时会再发一个存有校验码的文件,我们拿到该文件后做MD5运算,得到的计算结果与客户发送的校验码相比较,如果一致则认为客户发送的文件没有出错,否则认为文件出错需要重新发送。
另外值得注意的是,文件传输过程中,还有一些粘包问题,这个问题,我会单独列出博客来整理
敬请期待
同时由于个人时间限制,上传文件暂时移到明天(需要注意的是,这里的文件指的是,txt,jpg,png等格式,而不是文件夹,关于文件夹,后面我们会将它放到一个web上下载,这里不做处理)
首先老规矩:先建立
server = socket.socket()
server.bind(('localhost',6666))
server.listen()
while True:
conn, addr = server.accept()
print("new addr:",addr)
这个是相当格式化的部分,建立服务端,与客户端进行连接,并打印客户端的地址
client = socket.socket()
client.connect(('localhost', 6666))
同理,这是客户端
while True:
cmd = input(">>>:").strip()
if len(cmd)==0:
continue
if cmd.startswith("get"):
filename = cmd.split()[1] #获取文件名
client.send(cmd.encode("utf-8"))#客户端把要下载的文件信息,交给服务端
这个上节已经分享到过,这里不作解释
这个是客户端要求你所输入的文件来源,输入格式是
get C:\Users\Public\Pictures\938876-20160611152927293-1786781422.png
有个同学,会有疑问cmd.split()[1],为什么会是一哇,不能是二或三吗
C:\Users\Public\Pictures\938876-20160611152927293-1786781422.png
注意,这里他的前面有一个空格,所以,可以提取后面的一位
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开。。。")
break
cmd, filename = data.decode().split()#对信息进行解码,并除去空格,转化为两个字符串
print(filename)
服务端进行接受,其中 filename 为你的文件来源
if os.path.isfile(filename):
此步是为了判断,文件是否存在,如果存在,可以继续进行
f = open(filename,"rb")#打开文件
file_size = os.stat(filename).st_size
conn.send(str(file_size).encode("utf-8")) #发送文件大小
发送文件大小
客户端要开始接受文件大小了:
server_resp_size = client.recv(1024) #接收文件总的大小
file_total_size = int(server_resp_size.decode())
print("file size:", server_resp_size)#将文件大小进行打印
大家学到此处,可还有疑问?如果有疑问,欢迎咨询.
上面第一阶段,已经完成了,接下来就是正式的文件下载阶段了
client.send(b"ready to recv file...")#客户端,发消息,我要开始接收文件了
conn.recv(1024) #服务端,接收消息
服务端 for line in f:
m.update(line)
conn.send(line)
开始遍历文件,更新消息
客户端;f = open(filename+".new","wb"),创建一个以new格式的文件,用来接收服务
端发来的消息
客户端开始接受文件
while recv_size < file_total_size:#目的是为了防止黏包问题出现
if file_total_size - recv_size > 1024: #判断最后一次,之前接收大小设置为1024
size = 1024
else: #最后一次不足1024,则只接收文件剩余的部分,不包含MD5
size = file_total_size - recv_size
print("the last size:",size)
data = client.recv(size)
recv_size += len(data)
f.write(data)
服务端大致思路:
所以总的服务端代码
import os,socket,hashlib
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
conn, addr = server.accept()
print("new addr:",addr)
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开。。。")
break
cmd, filename = data.decode().split()
print(filename)
if os.path.isfile(filename):
f = open(filename,"rb")
m = hashlib.md5()
file_size = os.stat(filename).st_size
conn.send(str(file_size).encode("utf-8")) #发送文件大小
conn.recv(1024) #等待回复
for line in f:
m.update(line)
conn.send(line)
print("file md5:",m.hexdigest())
f.close()
conn.send(m.hexdigest().encode("utf-8")) #发送MD5,与上面的“conn.send(line)”可能出现粘包
print("send done...")
server.close()
客户端代码:
import socket,hashlib
client = socket.socket()
client.connect(('localhost', 9999))
while True:
cmd = input(">>>:").strip()
if len(cmd)==0:
continue
if cmd.startswith("get"):
filename = cmd.split()[1] #获取文件名
client.send(cmd.encode("utf-8"))
server_resp_size = client.recv(1024) #接收文件总的大小
file_total_size = int(server_resp_size.decode())
print("file size:", server_resp_size)
client.send(b"ready to recv file...")
f = open(filename+".new","wb")
recv_size = 0
m = hashlib.md5()
while recv_size < file_total_size:
if file_total_size - recv_size > 1024: #判断最后一次,之前接收大小设置为1024
size = 1024
else: #最后一次不足1024,则只接收文件剩余的部分,不包含MD5
size = file_total_size - recv_size
print("the last size:",size)
data = client.recv(size)
recv_size += len(data)
f.write(data)
m.update(data)
else:
client_md5 = m.hexdigest()
print("recv done...")
print("total size:", file_total_size, "had been received:", recv_size)
server_md5 = client.recv(1024)
print("server md5:", client_md5, "server md5:", server_md5.decode())
f.close()
client.close()
本文转自眉间雪 51CTO博客,原文链接:http://blog.51cto.com/13348847/1982209,如需转载请自行联系原作者