TFTP数据包格式，TFTP协议过程分析

　　六、Python实现TFTP协议
　　1、客户端下载文件参考程序
　　#coding=utf-8
　　#导包
　　import sys
　　import struct
　　from socket import *
　　#全局变量
　　g_server_ip = ‘’
　　g_downloadFileName = ‘’
　　#运行程序格式不正确
　　def run_test（）：
　　“判断运行程序传入参数是否有错”
　　global g_server_ip
　　global g_downloadFileName
　　if len（sys.argv） ！= 3：
　　print（“运行程序格式不正确”）
　　print（‘-’*30）
　　print（“tips：”）
　　print（“python3 tftp_download.py 192.168.1.1 test.jpg”）
　　print（‘-’*30）
　　exit（）
　　else：
　　g_server_ip = sys.argv［1］
　　g_downloadFileName = sys.argv［2］
　　#print（g_server_ip， g_downloadFileName）
　　#主程序
　　def main（）：
　　run_test（）
　　# 打包
　　sendDataFirst = struct.pack（‘！H%d***5***’%len（g_downloadFileName）， 1， g_downloadFileName.encode（‘gb2312’）， 0， ‘octet’.encode（‘gb2312’）， 0）
　　# 创建UDP套接字
　　s = socket（AF_INET， SOCK_DGRAM）
　　# 发送下载文件请求数据到指定服务器
　　s.sendto（sendDataFirst， （g_server_ip， 69）） #第一次发送， 连接tftp服务器
　　downloadFlag = True #表示能够下载数据，即不擅长，如果是false那么就删除
　　fileNum = 0 #表示接收文件的序号
　　# 以二进制格式创建新文件
　　f = open（g_downloadFileName， ‘wb’）
　　while True：
　　#3. 接收服务发送回来的应答数据
　　responseData = s.recvfrom（1024）
　　#print（responseData）
　　recvData， serverInfo = responseData
　　# 解包
　　packetOpt = struct.unpack（“！H”， recvData［：2］） #操作码
　　packetNum = struct.unpack（“！H”， recvData［2:4］） #块编号
　　#print（packetOpt， packetNum）
　　# 接收到数据包
　　if packetOpt［0］ == 3： #optNum是一个元组（3，）
　　# 计算出这次文件的序号，是上一次接收到的+1。
　　fileNum += 1
　　# 文件超过了65535 那么就又从0开始计数。
　　if fileNum == 65536：
　　fileNum = 0
　　# 包编号是否和上次相等
　　if fileNum == packetNum［0］：
　　f.write（recvData［4：］） #写入文件
　　fileNum = packetNum［0］
　　# 整理ACK的数据包
　　ackData = struct.pack（“！HH”， 4， packetNum［0］）
　　s.sendto（ackData， serverInfo）
　　# 错误应答
　　elif packetOpt［0］ == 5：
　　print（“sorry，没有这个文件！”）
　　downloadFlag = False
　　break
　　else：
　　print（packetOpt［0］）
　　break
　　# 接收完成，退出程序。
　　if len（recvData） downloadFlag = True
　　print（“%s文件下载完毕！”%g_downloadFileName）
　　break
　　if downloadFlag == True：
　　f.close（）
　　else：
　　os.unlink（g_downloadFileName） #没有下载的文件，就删除刚创建的文件。
　　#调用main函数
　　if __name__ == ‘__main__’：
　　main（）
　　2、客户端上传文件程序
　　#coding=utf-8
　　# 导包
　　import sys
　　import struct
　　from socket import *
　　# 全局变量
　　g_server_ip = ‘’
　　g_uploadFileName = ‘’
　　#运行程序格式不正确
　　def run_test（）：
　　“判断运行程序传入参数是否有错”
　　global g_server_ip
　　global g_uploadFileName
　　if len（sys.argv） ！= 3：
　　print（“运行程序格式不正确”）
　　print（‘-’*30）
　　print（“tips：”）
　　print（“python3 tftp_upload.py 192.168.1.1 test.jpg”）
　　print（‘-’*30）
　　exit（）
　　else：
　　g_server_ip = sys.argv［1］
　　g_uploadFileName = sys.argv［2］
　　#print（g_server_ip， g_uploadFileName）
　　#主程序
　　def main（）：
　　run_test（）
　　# 打包
　　sendDataFirst = struct.pack（‘！H%d***5***’%len（g_uploadFileName）， 2， g_uploadFileName.encode（‘gb2312’）， 0， ‘octet’.encode（‘gb2312’）， 0）
　　# 创建UDP套接字
　　s = socket（AF_INET， SOCK_DGRAM）
　　# 发送上传文件请求到指定服务器
　　s.sendto（sendDataFirst， （g_server_ip， 69）） #第一次发送， 连接tftp服务器
　　fileNum = 0 #表示接收文件的序号
　　# 以二进制格式打开文件
　　f = open（g_uploadFileName， ‘rb’）
　　# 第一次接收数据
　　responseData = s.recvfrom（1024）
　　# print（responseData）
　　recvData， serverInfo = responseData
　　#print（recvData）
　　#print（serverInfo）
　　# 解包
　　packetOpt = struct.unpack（“！H”， recvData［：2］） #操作码
　　packetNum = struct.unpack（“！H”， recvData［2:4］） #块编号
　　#print（packetOpt， packetNum）
　　if packetOpt［0］ == 5：
　　print（“tftp服务器发生错误！”）
　　exit（）
　　while True：
　　#　从文件中读取512字节数据
　　readFileData = f.read（512）
　　#　打包
　　sendData = struct.pack（‘！HH’， 3， fileNum） + readFileData
　　# 发送数据到tftp服务器
　　s.sendto（sendData， serverInfo） #第二次发给服务器的随机端口
　　# 接受服务器回传数据
　　recvData， serverInfo = s.recvfrom（1024）
　　#print（recvData）
　　# 解包
　　packetOpt = struct.unpack（“！H”， recvData［：2］） #操作码
　　packetNum = struct.unpack（“！H”， recvData［2:4］） #块编号
　　if packetOpt［0］ == 5：
　　print（“tftp服务器发生错误！”）
　　exit（）
　　if len（sendData） print（“%s文件上传成功！”%g_uploadFileName）
　　break
　　fileNum += 1
　　#　关闭文件
　　f.close（）
　　# 关闭套接字
　　s.close（）
　　#调用main函数
　　if __name__ == ‘__main__’：
　　main（）
　　3、服务器参考程序
　　#coding=utf-8
　　# 导包
　　import sys
　　import struct
　　from socket import *
　　from threading import Thread
　　‘’‘
　　利用多线程的机制，来实现tftp服务器同时进行上传和下载功能。
　　’‘’
　　# 客户端上传线程
　　def upload_thread（fileName， clientInfo）：
　　“负责处理客户端上传文件”
　　fileNum = 0 #表示接收文件的序号
　　# 以二进制方式打开文件
　　f = open（fileName， ‘wb’）
　　# 创建UDP套接字
　　s = socket（AF_INET， SOCK_DGRAM）
　　# 打包
　　sendDataFirst = struct.pack（“！HH”， 4， fileNum）
　　# 回复客户端上传请求
　　s.sendto（sendDataFirst， clientInfo） #第一次用随机端口发送
　　while True：
　　# 接收客户端发送的数据
　　responseData = s.recvfrom（1024） #第二次客户连接我随机端口
　　# print（responseData）
　　recvData， clientInfo = responseData
　　#print（recvData， clientInfo）
　　# 解包
　　packetOpt = struct.unpack（“！H”， recvData［：2］） #操作码
　　packetNum = struct.unpack（“！H”， recvData［2:4］） #块编号
　　#print（packetOpt， packetNum）
　　# 客户端上传数据
　　if packetOpt［0］ == 3 and packetNum［0］ == fileNum：
　　#　保存数据到文件中
　　f.write（recvData［4：］）
　　#　打包
　　sendData = struct.pack（“！HH”， 4， fileNum）
　　# 回复客户端ACK信号
　　s.sendto（sendData， clientInfo） #第二次用随机端口发
　　fileNum += 1
　　if len（recvData） print（“用户”+str（clientInfo）， end=‘’）
　　print（‘：上传’+fileName+‘文件完成！’）
　　break
　　# 关闭文件
　　f.close（）
　　# 关闭UDP套接字
　　s.close（）
　　# 退出上传线程
　　exit（）
　　# 客户端下载线程
　　def download_thread（fileName， clientInfo）：
　　“负责处理客户端下载文件”
　　# 创建UDP套接字
　　s = socket（AF_INET， SOCK_DGRAM）
　　fileNum = 0 #表示接收文件的序号
　　try：
　　f = open（fileName，‘rb’）
　　except：
　　# 打包
　　errorData = struct.pack（‘！HHHb’， 5， 5， 5， fileNum）
　　# 发送错误信息
　　s.sendto（errorData， clientInfo） #文件不存在时发送
　　exit（） #退出下载线程
　　while True：
　　# 从本地服务器中读取文件内容512字节
　　readFileData = f.read（512）
　　fileNum += 1
　　# 打包
　　sendData = struct.pack（‘！HH’， 3， fileNum） + readFileData
　　# 向客户端发送文件数据
　　s.sendto（sendData， clientInfo） #数据第一次发送
　　if len（sendData） print（“用户”+str（clientInfo）， end=‘’）
　　print（‘：下载’+fileName+‘文件完成！’）
　　break
　　# 第二次接收数据
　　responseData = s.recvfrom（1024）
　　# print（responseData）
　　recvData， clientInfo = responseData
　　#print（recvData， clientInfo）
　　#解包
　　packetOpt = struct.unpack（“！H”， recvData［：2］） #操作码
　　packetNum = struct.unpack（“！H”， recvData［2:4］） #块编号
　　#print（packetOpt， packetNum）
　　if packetOpt［0］ ！= 4 or packetNum［0］ ！= fileNum：
　　print（“文件传输错误！”）
　　break
　　# 关闭文件
　　f.close（）
　　# 关闭UDP套接字
　　s.close（）
　　# 退出下载线程
　　exit（）
　　# main函数
　　def main（）：
　　# 创建ＵＤＰ套接字
　　s = socket（AF_INET， SOCK_DGRAM）
　　# 解决重复绑定端口
　　s.setsockopt（SOL_SOCKET， SO_REUSEADDR， 1）
　　# 绑定任意IP，端口号69
　　s.bind（（‘’， 69））
　　print（“tftp服务器成功启动！”）
　　print（“正在运行中。。.”）
　　while True：
　　# 接收客户端发送的消息
　　recvData， clientInfo = s.recvfrom（1024） #　第一次客户连接69端口
　　#print（clientInfo）
　　#　解包
　　if struct.unpack（‘！b5***’， recvData［-7：］） == （0， b‘octet’， 0）：
　　opcode = struct.unpack（‘！H’，recvData［：2］） #　操作码
　　fileName = recvData［2：-7］.decode（‘gb2312’） #　文件名
　　# 请求下载
　　if opcode［0］ == 1：
　　t = Thread（target=download_thread， args=（fileName， clientInfo））
　　t.start（） # 启动下载线程
　　# 请求上传
　　elif opcode［0］ == 2：
　　t = Thread（target=upload_thread， args=（fileName， clientInfo））
　　t.start（） # 启动上传线程
　　# 关闭UDP套接字
　　s.close（）
　　# 调用main函数
　　if __name__ == ‘__main__’：
　　main（）
　　调试可以通过Wireshark 软件进行调试，也可以打印传输数据信息进行调试！