TFTP#
此项目搭建主要使用的技术是Python的sockt(UDP协议)+TFTP协议
TFTP介绍
TFTP服务器默认监听69号端口
当客户端发送“下载”请求(即读请求)时,需要向服务器的69端口发送服务器若批准此请求,则使使用一个个新的、临时的端口进口数据传输。
为了标记数据已经发送完毕,所以规定,当客户端接收到的数据口于516(2字节操作码+2个字节的序号+512字节数据)时,就意味着服务器发送完毕了
TFTP数据包的格式如下:
操作码 | 功能 |
---|---|
1 | 读请求,即下载 |
2 | 写请求,即上传 |
3 | 表示数据包,即DATA |
4 | 确认码,即ACK |
5 | 错误 |
其中下载代码如下:
#coding=utf-8
from socket import *
import struct,sys,time
#组包
sendcontent = struct.pack("!H%dsb5sb"%len(sys.argv[1]),1,sys.argv[1],0,"octet",0)
#创建socket
tftpSocket = socket(AF_INET, SOCK_DGRAM)
#发送 下载文件请求
tftpSocket.sendto(sendcontent,("192.168.17.26", 69))
downFile = open(sys.argv[1],'w')
packNum = 1
#接收数据
while True:
recvData = tftpSocket.recvfrom(516)
#解析数据
result = struct.unpack("!HH",recvData[0][:4])
#判断服务器返回是否是错误
if result[0] == 5:
print('没有这个文件')
break
#处理返回数据并保存
elif result[0] == 3:
#判断服务器丢包情况,防止数据重复保存
if packNum == result[1]:
downFile.write(recvData[0][4:])
ip = recvData[1]
sendback = struct.pack("!HH",4,result[1])
tftpSocket.sendto(sendback,ip)
packNum += 1
print '\r第%s部分拷贝成功'%result[1],
if len(recvData[0]) < 516:
break
downFile.close()
上传代码如下:
#coding=utf-8
from socket import *
import struct,sys,time
ip = "192.168.17.26"
#组包
sendcontent = struct.pack("!H%dsb5sb"%len(sys.argv[1]),2,sys.argv[1],0,"octet",0)
#创建socket
tftpSocket = socket(AF_INET, SOCK_DGRAM)
#发送 下载文件请求
tftpSocket.sendto(sendcontent,(ip, 69))
i = 0
packNum = 0
while True:
recvData = tftpSocket.recvfrom(1024)
result = struct.unpack("!HH",recvData[0][:4])
try:
#读取待上传文件
upFile = open(sys.argv[1],'r')
upFile.seek(i*512,0)
conent = upFile.read(512)
i += 1
upFile.close()
#判断是否有发送失败的情况
if packNum == result[1]:
#发送上传数据包
upcontent = struct.pack("!HH%ss"%len(conent),3,result[1]+1,conent)
tftpSocket.sendto(upcontent,(ip, recvData[1][1]))
packNum += 1
if len(conent) < 512:
break
except IOError:
print('本地没有这个文件')
break
网友评论