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