Python实现ftp文件传输

主要步骤可以分为以下几步:

1.读取文件名
2.检测文件是否存在
3.打开文件
4.检测文件大小
5.发送文件大小和 md5值给客户端
6.等客户端确认
7.开始边读边发数据

服务器端代码:

import socket,os,time
import hashlib

server =socket.socket()
server.bind(('0.0.0.0',6666))
server.listen()
print("等待....")
while True:
    conn,addr = server.accept()
    print("new conn:",conn)
    while True:
         data = conn.recv(1024)
         if not data:
             print("client is disconnection")
             break
         cmd,filename = data.decode().split()   #记录指令和文件名
         print(filename)
        #判断当前目录是否存在该文件,而且必须是文件,而不是目录
         if os.path.isfile(filename):
            f = open(filename,'rb')
            #m = hashlib.md5()        # 创建md5
            file_size = os.stat(filename).st_size  #stat() 可以返回文件的大小值
            conn.send((str(file_size)).encode())                # 发送文件大小
            conn.recv(1024)     #等待返回信息
            for line in f:
               # m.updata(line)       
                conn.send(line)
            #print("file md5",m.hexdigest())            #打印md5值
            f.close()


客户端代码:

# Author: zjt
import socket

client = socket.socket()

client.connect(("0.0.0.0",6666))


while True:
    cmd = input(">>>:").strip()

    if len(cmd)==0 :continue

    if cmd.startswith("get"):
        client.send(cmd.encode())
        server_response = client.recv(1024)
        print("server response: ",server_response)
        client.send(b"ready to recv file")

        # 开始接收文件
        file_total_size = int(server_response.decode())
        received_size = 0 # 记录接收文件的大小
        filename = cmd.split()[1]
        # 因为两个目录一致,接收的文件名不能与原文件相同
        f = open(filename+".new","wb")
        while received_size < file_total_size:
            data = client.recv(1024)
            received_size += len(data)
            f.write(data)
            print("total:",file_total_size,"    present: ",received_size)
        else:
            print("file has received done!")
            f.close()

client.close()

        用80M的文件传输测试,效果如下:



程序升级:

前面的代码还没添加md5进行验证,现在对代码进行升级

服务器端代码:

import socket,os,time
import hashlib

server =socket.socket()
server.bind(('0.0.0.0',8888))
server.listen()
print("等待....")
while True:
    conn,addr = server.accept()
    print("new conn:",conn)
    while True:
         data = conn.recv(1024)
         if not data:
             print("client is disconnection")
             break
         cmd,filename = data.decode().split()   #记录指令和文件名
         print(filename)
        #判断当前目录是否存在该文件,而且必须是文件,而不是目录
         if os.path.isfile(filename):
            f = open(filename,'rb')
            m = hashlib.md5()        # 创建md5
            file_size = os.stat(filename).st_size  #stat() 可以返回文件的大小值
            conn.send((str(file_size)).encode())                # 发送文件大小
            conn.recv(1024)     #等待返回信息
            for line in f:
                m.update(line)
                conn.send(line)
            print("file md5",m.hexdigest())             #打印md5值

            f.close()
            conn.send(m.hexdigest().encode())  # 发送md5

            print("我真的已经发过去了",m.hexdigest().encode())

         print("send done")

server.close()

客户端代码:

import socket
import hashlib
client = socket.socket()
client.connect(("0.0.0.0",8888))
while True:
    cmd = input(">>>:").strip()
    if len(cmd)==0 :continue
    if cmd.startswith("get"):
        client.send(cmd.encode())
        server_response = client.recv(1024)
        print("server response: ",server_response)
        client.send(b"ready to recv file")
        # 开始接收文件
        file_total_size = int(server_response.decode())
        received_size = 0 # 记录接收文件的大小
        filename = cmd.split()[1]
        # 因为两个目录一致,接收的文件名不能与原文件相同
        f = open(filename+".new","wb")
        m = hashlib.md5()
        while received_size < file_total_size:
            data = client.recv(1024)
            received_size += len(data)
            m.update(data)
            f.write(data)
            #print("total:",file_total_size,"    present: ",received_size)
        else:
            new_file_md5 = m.hexdigest()
            print("client file md5:",new_file_md5)
            print("file has received done!")
            print("total:",file_total_size,"    present: ",received_size)
            f.close()
            sever_file_md5 = client.recv(1024)
            print("client file md5:",new_file_md5)
            print("server file md5:",sever_file_md5)
client.close()

两个程序在linux 环境下运行,结果如下:


可以看到传输后文件大小变大了一点点,而且md5前后值也不同,说明文件传输发生了改变。

现在讲程序在windows环境下运行,结果如下:


      此时可以看到windows上没有问题,文件大小相同,且md5值也一致。

原因分析:

     之所以会发生这种情况,是因为在linux上运行时,最后一次传输文件与发送md5值的时候,发生可粘包,导致最后一次接收文件的时候,连同md5的数据一并发送了。而客户端也当作一条接收信息,全部接收了。所以客户端出现没有收到来自服务器端的md5值,多出来的那一点点,就是md5值的大小。

解决方法:

      在接收文件的时候,判断当前剩余多少文件需要接收,如果大于1024,就接收1024大小的文件,否则就只接收剩下全部的文件,防止最后一次接收多余的数据。

只需要对客户端代码进行修改,修改后代码如下:

import socket
import hashlib
client = socket.socket()
client.connect(("0.0.0.0",8888))
while True:
    cmd = input(">>>:").strip()
    if len(cmd)==0 :continue
    if cmd.startswith("get"):
        client.send(cmd.encode())
        server_response = client.recv(1024)
        print("server response: ",server_response)
        client.send(b"ready to recv file")
        # 开始接收文件
        file_total_size = int(server_response.decode())
        received_size = 0 # 记录接收文件的大小
        filename = cmd.split()[1]
        f = open(filename+".new","wb")
        m = hashlib.md5()
        while received_size < file_total_size:
            #添加一次判断,使最后一次剩多少就接收多少,避免发生粘包
            if file_total_size - received_size > 1024:
                 size = 1024
            else:   # 最后一次,剩多少收多少
                 size = file_total_size - received_size
            data = client.recv(size)
            received_size += len(data)
            m.update(data)
            f.write(data)
        else:
            new_file_md5 = m.hexdigest() 
            print("client file md5:",new_file_md5)
            print("file has received done!")
            print("total:",file_total_size,"    present: ",received_size)
            print("下一句关闭文件")
            f.close()
            print("开始接收md5 ")   
        sever_file_md5 = client.recv(1024)
        print("client file md5:",new_file_md5)
        print("server file md5:",sever_file_md5)
client.close()

猜你喜欢

转载自blog.csdn.net/zjt980452483/article/details/79813631
今日推荐