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