首先,TFTP協(xié)議遵循以下格式:

tftp協(xié)議

我們只需要根據(jù)這些協(xié)議, 來決定我們每次請求數(shù)據(jù)是什么

構(gòu)造請求數(shù)據(jù)時(shí)的注意點(diǎn):(struct的格式)
例如

fileName = "123456.jpg"
dlreq = struct.pack("!H10sb5sb", 1, fileName.encode("utf-8"), 0, b"octet", 0)

注意: "!H10sb5sb"中, H標(biāo)示兩個(gè)字節(jié), s表示一個(gè)字節(jié), b標(biāo)示1個(gè)字節(jié). 具體含義如下圖:

fmt的格式

• 注1.q和Q只在機(jī)器支持64位操作時(shí)有意思

• 注2.每個(gè)格式前可以有一個(gè)數(shù)字，表示個(gè)數(shù)

• 注3.s格式表示一定長度的字符串，4s表示長度為4的字符串，但是p表示的是pascal字符串

• 注4.P用來轉(zhuǎn)換一個(gè)指針，其長度和機(jī)器字長相關(guān)

• 注5.最后一個(gè)可以用來表示指針類型的，占4個(gè)字節(jié)

符號(hào)fmt

然后上代碼:

from socket import *
import struct
import sys
from threading import Thread

g_ip = ""
g_port = 0



def downLoad():

    g_socket = socket(AF_INET, SOCK_DGRAM)
    g_socket.bind(("", 9090))
    fileName = input("請輸入要下載的文件名: ")
    fileNameLen = len(fileName)
    # 下載請求格式控制字符串
    dlreqStr = "!H" + str(fileNameLen) + "sb5sb"
    # 構(gòu)造下載請求數(shù)據(jù)
    dlreq = struct.pack(dlreqStr, 1, fileName.encode("utf-8"), 0, b"octet", 0)

    sendAdd = (g_ip, g_port)
    g_socket.sendto(dlreq, sendAdd)

    p_Num = 0
    recvfile = ""

    while True:
        recvData, recvAddr = g_socket.recvfrom(1024)
        recvdataLen = len(recvData)
        cmd_tuple = struct.unpack("!HH", recvData[:4])
        cmd = cmd_tuple[0]
        currentPackNum = cmd_tuple[1]
        if cmd == 3:
            if currentPackNum == 1:
                recvfile = open(fileName, "wb")

            if p_Num + 1 == currentPackNum:
                recvfile.write(recvData[4:])
                p_Num += 1
                print("第 %d 次接收到數(shù)據(jù)"%p_Num)

                ackBuf = struct.pack("!HH", 4, p_Num)
                g_socket.sendto(ackBuf, recvAddr)

            if recvdataLen < 516:
                recvfile.close()
                print("已經(jīng)下載成功!")
                break

        elif cmd == 5:
            print("錯(cuò)誤!! 錯(cuò)誤塊: %d"%currentPackNum)
            break

    recvfile.close()
    g_socket.close()



def upLoad():
    upSocket = socket(AF_INET, SOCK_DGRAM)
    num = 0
    
    fileName = input("輸入要上傳的文件:")
    sendAdd = (g_ip, g_port)

    fileNameLen = len(fileName)
    # 上傳請求格式控制字符串
    ulreqStr = "!H" + str(fileNameLen) + "sb5sb"
    # 構(gòu)造下載請求數(shù)據(jù)
    ulreq = struct.pack(ulreqStr, 2, fileName.encode("utf-8"), 0, b"octet", 0)

    upSocket.sendto(ulreq, sendAdd)

    recvData1 = upSocket.recvfrom(1024)
    rand_port = recvData1[1][1]
    ack_num = struct.unpack("!H",recvData1[0][2:4])
    sendAdd = (g_ip, rand_port)

    try:
        fileRead = open(fileName, 'rb')
    except:
        errorData = struct.pack("!HHHb", 5, 5, 5, num)
        upSocket.sendto(errorData, sendAdd)
        exit() # 退出線程

    while True:
        readData = fileRead.read(512)
        sendData = struct.pack("!HH", 3, num) + readData
        upSocket.sendto(sendData, sendAdd)

        if len(sendData) < 516:
            print("傳輸成功!!")
            break

        recv_ack = upSocket.recvfrom(1024)[0]
        recv_cmd, ack_num = struct.unpack("!HH", recv_ack)

        print("接收到了 %d", ack_num)

        num +=1

        if int(recv_cmd) != 4 or int(ack_num != num - 1):
            break
    upSocket.close()
    fileRead.close()





def main():

    global g_ip
    global g_port


    g_ip = input("請輸入ip地址: ")
    g_port = int(input("請輸入端口號(hào): "))

    t1 = Thread(target = upLoad)

    t1.start()
    t1.join()
    
if __name__ == "__main__":
    main()