#coding=utf-8
from socket import *
import time,struct
class TFTPServer():
    
    #操作碼
    DOWNLOAD    = 1
    UPLOAD      = 2
    DATA        = 3
    ACK         = 4
    ERROR       = 5
    
    def __init__(self):
        self.serverSocket = socket(AF_INET, SOCK_DGRAM)
        self.serverSocket.bind(("", 69))

    def run(self):
        while True:
            print("#"*30)
            print("等待客戶端連接！")
            self.listen()

            if self.cmdType == self.DOWNLOAD:
                self.download()
            elif self.cmdType == self.UPLOAD:
                self.upload()

    def listen(self):
        # 等待client鏈接
        self.recvData, self.recvAddr = self.serverSocket.recvfrom(1024)
        self.cmdType = struct.unpack("!H", self.recvData[:2])[0]

    # 客戶端請求下載
    def download(self):
        # 新建隨機端口
        udpSocket = socket(AF_INET, SOCK_DGRAM)

        fileReq = None
        try:
            # 讀取客戶端要下載的文件名
            fileReqName = self.recvData[2:-7].decode()
            print(fileReqName)
            print("客戶端請求下載文件：%s" % fileReqName)
            # 打開文件
            try:
                fileReq = open(fileReqName, "rb")
            except:
                print("文件 《%s》 不存在" % fileReqName)
                # 向 client 發送異常信息
                errInfo = struct.pack("!HHHb", 5, 5, 5, 0)
                udpSocket.sendto(errInfo, self.recvAddr)
                return False

            # 初始化塊編號
            frameNum = 1

            while True:
                fileData = fileReq.read(512)
                # 打包
                # frameData = struct.pack(str("!HH%ds" % len(fileData)), 3, frameNum, fileData)
                frameData = struct.pack(str("!HH"), 3, frameNum) + fileData
                # 發送
                for i in range(0, 3):
                    udpSocket.sendto(frameData, self.recvAddr)

                    # 若文件已發送完成
                    if len(fileData) < 512:
                        print("文件發送完成！")
                        fileReq.close()
                        fileReq = None
                        return True

                    # 等待client響應
                    # 接收數據
                    self.recvData, self.recvAddr = udpSocket.recvfrom(1024)
                    cmdType, recvFrameNum = struct.unpack("!HH", self.recvData[:4])
                    if cmdType == self.ACK and recvFrameNum == frameNum:
                        break
                    elif i == 3:
                        print("鏈接異常，取消發送")
                        # 向 client 發送異常信息
                        errInfo = struct.pack("!HHHb", 5, 5, 5, 0)
                        udpSocket.sendto(errInfo, self.recvAddr)
                        exit()

                # 編號+1
                frameNum += 1
        finally:
            if fileReq != None:
                fileReq.close()

    # 客戶端請求上傳
    def upload(self):
        # 新建隨機端口
        udpSocket = socket(AF_INET, SOCK_DGRAM)

        # 發送相應，確認開始接收
        ack = struct.pack("!HH", self.ACK, 0)
        udpSocket.sendto(ack, self.recvAddr)

        # 幀計數，用於數據校驗
        recvFrameNum = 1
        # 等待 client 數據
        while True:
            #接收數據
            recvData, recvAddr = udpSocket.recvfrom(1024)
            #數據幀： 操作碼2 快編號2 數據n
            cmdType, frameNum = struct.unpack("!HH", recvData[:4])
            # 判斷接收的是否是“數據”
            if cmdType == self.DATA and frameNum == recvFrameNum:
                print("接收到第%d幀數據！" % frameNum)  # for test
                # 打開文件
                if frameNum == 1:
                    fileRecv = open("upload.txt", "ab")
                    fileRecv.write(b"#" * 30 + time.strftime("%Y-%m-%d %H:%M:%S").encode() + b"#" * 30 + b"\n")

                fileRecv.write(recvData[4:])

                # 響應 : 操作碼2 快編號2
                ack = struct.pack("!HH", self.ACK, frameNum)
                udpSocket.sendto(ack, recvAddr)

                # 判斷是否發送完
                if len(recvData) < 516:
                    fileRecv.close()
                    fileRecv = None
                    print("接收完畢， 關閉文件。")
                    break

                # 計數+1
                recvFrameNum += 1

            elif cmdType == self.ERROR:
                print("客戶端連接異常，退出！")
                break


server = TFTPServer()
server.run()

TFTP_server python實現（個人備忘）