依据TFTP协议的服务端和客户端

最新推荐文章于 2024-05-24 21:19:36 发布

流月0

最新推荐文章于 2024-05-24 21:19:36 发布

阅读量1.5k

点赞数

分类专栏：技术博客文章标签： tftp 文件传输 python

本文链接：https://blog.csdn.net/m0_37887449/article/details/73086871

版权

本文介绍了使用Python实现的TFTP协议服务端和客户端，端口号设定为2048，支持文件上传下载及日志记录。服务端存在一些待完善的功能，如无法正常退出，文件列表未实时更新等。完整代码和个人技术博客链接见正文。

摘要由CSDN通过智能技术生成

个人技术博客地址：http://songmingyao.com/

今天写了下依据TFTP协议的服务端和客户端，端口号设置为2048。

实现功能：
* 可让服务端客户端搭配使用实现上传下载功能
* 可在服务端记录log日志
* 客户端可单独与Windows上的TFTP程序完成文件传输

待完善：
* 服务端无退出功能，不退出的话端口不能释放
* 代码均尚未捕获异常
* 服务端文件列表未实时更新
* 服务端log日志未设保护
* 未按照MD5校验值来判断文件

服务端代码

from socket import *
import struct
import os
import time

def send_file():
    global log
    '发送文件'
    if file_name in file_list: # 检测服务端是否存在客户端要下载的文件
        f = open('./%s'%file_name, 'rb')
        i = 1
        times = 0
        while True:
            content = f.read(512)
            con_len = len(content)
            pack_content = struct.pack('!HH%ds'%con_len, 3, i, content)
            tftp_socket.sendto(pack_content, addr)
            echo_msg = tftp_socket.recvfrom(1024) # 接收客户端返回值
            echo_op = struct.unpack('!HH', echo_msg[0][:4]) # 读取客户端ACK
            if echo_op == (4, i):
                times = 0 # 重置客户端无响应次数
                i += 1
                if i == 65536:
                    i = 0 # 重置块编号
            elif echo_op == (4, i-1):
                times += 1 # 客户端无响应次数统计
                f.seek(1, -512) # 调整文件读取位置
                if times > 6:
                    log = open('log.txt', 'a')
                    log.write('<time : %s>\t<ip : %s>\t<op : 请求下载文件%s,中途断开链接,下载失败!>\n'%(time.ctime(), addr[0], file_name))
                    log.close()
                    break
            if con_len < 512: # 数据长度判断
                log = open('log.txt', 'a')
                log.write('<time : %s>\t<ip : %s>\t<op : 请求下载文件%s,下载成功!>\n'%(time.ctime(), addr[0], file_name))
                log.close()
                break
    else:
        error_info = struct.pack('!HH21sb', 5, 1, 'cannot find this file'.encode(&