实现简单socket的ftp功能

1、服务器端

import socket,os,hashlib
server=socket.socket()
server.bind(("localhost",9999))  # 绑定套接字

server.listen()  # 服务器监听
"""
FTP 流程
1.读取文件名
2.检测文件是否存在
3.打开文件
4.检测文件大小
5.发送文件大小给客户端
6.等客户端确认
7.开始边读边发数据
8.发送md5，双方做对比
"""
while True:
    conn,addr=server.accept()   # 接收一个套接口的连接

    while True:
        data=conn.recv(1024) #接收
        if not data:
            print("客户端已断开")
            break
        cmd,filename= data.decode().split()
        print(filename)
        if os.path.isfile(filename):
            with open(filename,"rb") as f:
                m=hashlib.md5()  # 创建一个md5
                file_size=os.stat(filename).st_size  # 利用os.stat().st_size 获取文件的大小
                conn.send(     str(file_size).encode()     ) # send file size
                client_ack=conn.recv(1024)  # wait for ack 防止连续发送导致的粘包
                for line in f:
                    m.update(line)  # 每次接收一行，就更新md5
                    conn.send(line)  # send line_data
                # print("file md5:",m.hexdigest())
                conn.send(m.hexdigest().encode())  # send md5_data
        print("send done...")


server.close()

2、客户端

import socket,hashlib

# 调用进度条功能
def progress(percent):
    if percent>=1:
        percent=1
    res=int(percent*50)*"█"
    print("
[%-50s] %d%%" %(res,int(percent*100)) ,end="")

# 判断是否下载文件的单位
def k_m_g(size):
    size=int(size)
    if size<1024**2:
        return size/1024,"K"
    elif size<1024**3:
        return size/(1024**2),"M"
    else:
        return size/(1024**3),"G"


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

done_size=1024   # 设置每次下载长度

while True:
    cmd = input(">>:").strip()
    if len(cmd)==0:continue
    if cmd.startswith("get"):
        client.send(cmd.encode('utf-8'))
        server_response = client.recv(1024) # 接收文件大小
        client.send("ready to recv file".encode('utf-8'))  # 响应服务器请求，防止粘包
        file_total_size = int(server_response.decode())   # 文件的总长度
        num,msg=k_m_g(file_total_size)
        print(f'server response:{round(num,2)}{msg}' )
        reversed_size=0
        filename=cmd.split()[1]   # 获得get 后的第一个命令
        with open(filename+'.new',"wb") as f:
            m=hashlib.md5()
            recv=0
            while reversed_size<file_total_size:
                if file_total_size-reversed_size>done_size:   # 判断是否是最后一次传输（防止粘包）
                    size=done_size
                else:
                    size=file_total_size-reversed_size
                data=client.recv(size)
                m.update(data) # 更新client端的md5 用于后续对比完整性
                reversed_size+=len(data)  # 传了多少长度的数据
                f.write(data)
                if recv < file_total_size:  # 小于文件长度就继续传，再调用进度条函数
                    recv += len(data)
                    percent = recv / file_total_size
                    progress(percent)
            else:
                client_file_md5=m.hexdigest()  # 传完后获取接收到数据的md5值
                print("
file recv done...")
            server_file_md5=client.recv(1024)   # 获取服务器端传的md5
            if server_file_md5.decode() != client_file_md5:
                print('MD5准确性校验未通过，数据不准确')

client.close()

 执行效果：