python网络编程-socket上传下载文件(包括md5验证，大数据发送，粘包处理)

ftp server

1) 读取文件名

2）检查文件是否存在

3）打开文件

4）检查文件大小

5）发送文件大小给客户端

6）等客户端确认

7）开始边读边（md5计算）发数据

8）给客户端发md5

ftp下载文件

服务器
# -*- coding:utf-8 -*-
__author__ = 'shisanjun'
import socket
import os
import hashlib
server=socket.socket()

server.bind(("127.0.0.1",9999))

server.listen(5)

while True:
    print("等待新指令")
    conn,addr=server.accept()
    print("连接信息",addr)

    while True:
        try:
            data=conn.recv(1024)
        except ConnectionResetError as e:
            print(e)
        if not data:            #客户端断开会传个空值过来，会导致进入死循环
            print("客户端断开连接")
            break

        cmd,filename=data.decode("utf-8").split()
        print(filename)

        if os.path.isfile(filename):
            f=open(filename,'rb')
            m=hashlib.md5() #MD5验证
            file_size=os.stat(filename).st_size #获取文件大小
            conn.send(str(file_size).encode()) #给客户端发送文件大小

            conn.recv(1024) #阻塞 因为有两次send，防止出现“粘包”

            for line in f:
                m.update(line)
                conn.send(line)
            else:
                f.close()
                print("file md5 ",m.hexdigest())
        print("send done")

        conn.send(m.hexdigest().encode()) #给客户端发送md5验证值

server.close()

客户端
# -*- coding:utf-8 -*-
__author__ = 'shisanjun'

import os
import socket
import hashlib

client=socket.socket()
client.connect(("127.0.0.1",9999))

while True:
    cmd=input(">>:").strip()
    if len(cmd)==0:continue
    if cmd.startswith("get"):
        client.send(cmd.encode("utf-8")) #给服务器端发送命令

        response=client.recv(1024) #接受服务器相应：文件大小

        total_file_size=int(response.decode("utf-8"))
        client.send(b"read to recv file") #防止粘包，阻塞
        f=open(cmd.split()[1],'wb')
        m=hashlib.md5() #md5验证
        recv_size=0
        while recv_size<total_file_size:
            if total_file_size - recv_size>1024: #判断最后一次接受数据，长度是多少，就收多少，如果还是1024有可能会出现粘包
                size=1024
            else:
                size=total_file_size - recv_size

            recv_data=client.recv(size)
            f.write(recv_data)
            m.update(recv_data)
            recv_size+=len(recv_data) #要以len来计算，最后一次的数据不一定是1024
        else:
            f.close()

        recv_md5=m.hexdigest()

        recv_server_md5=client.recv(1024).decode()

        print("md5 clinet and server", recv_md5,recv_server_md5)

上传文件
客户端
# -*- coding:utf-8 -*-
__author__ = 'shisanjun'

import socket
import json
import os
host=("127.0.0.1",9000)
client=socket.socket()

client.connect(host)

while True:
    cmd_str=input("请输入命令>>").strip()
    if len(cmd_str)==0:continue
    cmd_list=cmd_str.split()

    if cmd_list[0]=="put":
        if len(cmd_list)==1:
            print("not find put filename")
        else:
            filename=cmd_list[1]
            if os.path.isfile(filename):
                file_obj=open(filename,'rb')
                filename2=filename.split("/")[-1]
                print(filename2)
                data_stat="%s %s" %(filename2,os.path.getsize(filename))
                data={
                    "filename":filename2,
                    "filesize":os.path.getsize(filename)
                }
                client.send(json.dumps(data).encode("utf-8"))
                for line in file_obj:
                    client.send(line)
                file_obj.close()
            else:
                print("can not find file")
    elif cmd_list[0]=="get":
        pass

服务器端
# -*- coding:utf-8 -*-
__author__ = 'shisanjun'

import socket
import json
host=("127.0.0.1",9000)
server=socket.socket()

server.bind(host)
server.listen(5)

while True:
    conn,addr=server.accept()

    data=conn.recv(4096)
    print(data.decode("utf-8"))
    data_dict=json.loads(data.decode("utf-8"))
    filename=data_dict.get("filename")
    f=open(filename,"wb")
    recesize=0
    while recesize<data_dict.get("filesize"):
        recefile=conn.recv(4096)
        f.write(recefile)
        recesize+=len(recefile)
    f.close()