个人技术博客地址: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(&