码迷,mamicode.com
首页 > 编程语言 > 详细

python--》客户端与服务端文件的下载

时间:2017-11-15 23:33:12      阅读:248      评论:0      收藏:0      [点我收藏+]

标签:python


在介绍之前,我们需要了解一个加密算法

        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()

技术分享

技术分享

技术分享


python--》客户端与服务端文件的下载

标签:python

原文地址:http://13348847.blog.51cto.com/13338847/1982137

(1)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!