Python-TFTP项目搭建

2017-02-08  本文已影响0人  FangHao

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
上一篇下一篇

猜你喜欢

热点阅读