TFTP客户端

96
一只写程序的猿
2017.06.14 21:46* 字数 671

1.TFTP协议介绍

TFTP(Trivial File Transfer Protocol,简单文件传输协议)

是TCP/IP协议族中的一个用来在客户端与服务器之间进行简单文件传输的协议。 TFTP是一个传输文件的简单协议,通常使用UDP协议而实现,但tftp并没有要求实现的具体协议,在特殊需求的场合可以同tcp实现。此协议设计的时候是进行小文件传输的。因此它不具备通常的FTP的许多功能,它只能从文件服务器上获得或写入文件,不能列出目录,不进行认证,它传输8位数据。传输中有三种模式:netascii,这是8位的ASCII码形式,另一种是octet,这是8位源数据类型;最后一种mail已经不再支持,它将返回的数据直接返回给用户而不是保存为文件。

特点:

  • 简单
  • 占用资源小
  • 适合传递小文件
  • 适合在局域网进行传递
  • 端口号为69
  • 基于UDP实现

2.TFTP下载过程

TFTP服务器默认监听69号端口
当客户端发送“下载”请求(即读请求)时,需要向服务器的69端口发送
服务器若批准此请求,则使用一个新的、临时的 端口进行数据传输

传输过程

当服务器找到需要现在的文件后,会立刻打开文件,把文件中的数据通过TFTP协议发送给客户端
如果文件的总大小较大(比如3M),那么服务器分多次发送,每次会从文件中读取512个字节的数据发送过来
因为发送的次数有可能会很多,所以为了让客户端对接收到的数据进行排序,所以在服务器发送那512个字节数据的时候,会多发2个字节的数据,用来存放序号,并且放在512个字节数据的前面,序号是从1开始的
因为需要从服务器上下载文件时,文件可能不存在,那么此时服务器就会发送一个错误的信息过来,为了区分服务发送的是文件内容还是错误的提示信息,所以又用了2个字节 来表示这个数据包的功能(称为操作码),并且在序号的前面

功能
TFTP数据包的格式

相关代码

pack 和unpack

import struct

data = struct.pack('!H8sb5sb',1,b'girl.jpg',0,b'octet',0)
print(type(data))
ret1 = struct.unpack('!H',data[:2])
print(ret1[0])

ret1 = struct.unpack('!8s',data[2:10])
print(ret1)
print(ret1[0].decode('utf-8'))

ret1 = struct.unpack('!H8sb5sb',data)
print(ret1)

download-01

import socket
import struct
import os
import time

def main():
    #创建文件
    myFile = open('girl.jpg','wb')
    #socket对象,发送下载的请求信息,接收信息和发送确认信息
    udpSocket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    #服务器地址
    destAddr = ('192.168.11.74',69)
    #pack信息
    data = struct.pack('!H8sb5sb', 1, b'girl.jpg', 0, b'octet', 0)
    #发送
    udpSocket.sendto(data,destAddr)
    #循环
    while True:
        time.sleep(0.2)
        #接收
        recvData,redvAddr = udpSocket.recvfrom(1024)
        print(redvAddr)
        #unpack信息,获取操作码
        operNum = struct.unpack('!H',recvData[:2])[0]
        #判断
        if operNum==3:
            # unpack信息,获取块编号
            blockNum = struct.unpack('!H', recvData[2:4])[0]
            print(blockNum)
            #获取数据,写到文件中
            myFile.write(recvData[4:])
            #准备ack数据
            ackData = struct.pack('!HH', 4,blockNum)
            #发送ack确认到服务器
            udpSocket.sendto(ackData,redvAddr)

            if len(recvData)<516:
                break

        if operNum==5:
            print('发生异常啦......')
            break
    #关闭文件
    myFile.close()

if __name__ == '__main__':
    main()

运行结果:

服务器的文件
>>
显然下载过后的文件是没问题的

download-02

import socket
import struct
import os
import time

def main():
    #创建文件
    myFile = open('Timor.avi','wb')
    #socket对象,发送下载的请求信息,接收信息和发送确认信息
    udpSocket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    #服务器地址
    destAddr = ('192.168.11.74',69)
    #pack信息
    data = struct.pack('!H23sb5sb',1,'冯提莫 - 刚好遇见你.m4r'.encode('gbk'),0,b'octet',0)
    #发送
    udpSocket.sendto(data,destAddr)
    #定义一个变量,记录写的次数
    num = 0
    #循环
    while True:
        #接收
        recvData,redvAddr = udpSocket.recvfrom(1024)

        #unpack信息,获取操作码
        operNum = struct.unpack('!H',recvData[:2])[0]
        #判断
        if operNum==3:
                # unpack信息,获取块编号
                blockNum = struct.unpack('!H', recvData[2:4])[0]
                print(blockNum)

                #判断
                num = num+1
                if num==65536:
                    num = 0

                if num == blockNum:
                    #获取数据,写到文件中
                    myFile.write(recvData[4:])
                    #准备ack数据
                    ackData = struct.pack('!HH', 4,blockNum)
                    #发送ack确认到服务器
                    udpSocket.sendto(ackData,redvAddr)
                num = blockNum

            if len(recvData)<516:
                break

        if operNum==5:
            print('发生异常啦......')
            break
    #关闭文件
    myFile.close()

if __name__ == '__main__':
    main()

运行结果:

原服务器中的文件
>>
下载后的文件

上传

import socket
import struct
import os


def main():
    #以读字节的方式打开文件
    myFile=open('xx.avi','rb')
    #创建socket UDP对象
    udpSocket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    destAddr = ('192.168.11.66',69)
    #组成上传文件请求包
    date = struct.pack('!H5sb5sb',2,b'x.avi',0,b'octet',0)
    #发送上传文件请求包
    udpSocket.sendto(date,destAddr)

    #数据包编号
    blockNum = 0
    while True:
        recvDate, recvAddr = udpSocket.recvfrom(1024)
        recvNum = struct.unpack('!H', recvDate[2:4])[0]
        print('recvNum:%s' % recvNum)
        date1 = myFile.read(512)
        # 判断是否有发送失败的情况
        if blockNum == recvNum:
            #发送数据包
            blockNum += 1
            if blockNum == 65536:
                blockNum = 0
            blockDate1 = struct.pack('!HH%ss'%len(date1),3,blockNum,date1)
            print('blockNum:%s' % blockNum)
            udpSocket.sendto(blockDate1, recvAddr)
            if len(date1)<512:
                break
    myFile.close()

if __name__ == '__main__':
    main()

日记本