依據(jù)TFTP協(xié)議的服務(wù)端和客戶端

<br />
個人技術(shù)博客地址：http://songmingyao.com/

<br />

今天寫了下依據(jù)TFTP協(xié)議的服務(wù)端和客戶端，端口號設(shè)置為2048。

實現(xiàn)功能：

  • 可讓服務(wù)端客戶端搭配使用實現(xiàn)上傳下載功能
  • 可在服務(wù)端記錄log日志
  • 客戶端可單獨(dú)與Windows上的TFTP程序完成文件傳輸

待完善：

  • 服務(wù)端無退出功能，不退出的話端口不能釋放
  • 代碼均尚未捕獲異常
  • 服務(wù)端文件列表未實時更新
  • 服務(wù)端log日志未設(shè)保護(hù)
  • 未按照MD5校驗值來判斷文件

服務(wù)端代碼

from socket import *
import struct
import os
import time

def send_file():
    global log
    '發(fā)送文件'
    if file_name in file_list: # 檢測服務(wù)端是否存在客戶端要下載的文件
        f = open('./%s'%file_name, 'rb')
        i = 1
        times = 0
        while True:
            content = f.read(512)
            con_len = len(content)
            pack_content = struct.pack('!HH%ds'%con_len, 3, i, content)
            tftp_socket.sendto(pack_content, addr)
            echo_msg = tftp_socket.recvfrom(1024) # 接收客戶端返回值
            echo_op = struct.unpack('!HH', echo_msg[0][:4]) # 讀取客戶端ACK
            if echo_op == (4, i):
                times = 0 # 重置客戶端無響應(yīng)次數(shù)
                i += 1
                if i == 65536:
                    i = 0 # 重置塊編號
            elif echo_op == (4, i-1):
                times += 1 # 客戶端無響應(yīng)次數(shù)統(tǒng)計
                f.seek(1, -512) # 調(diào)整文件讀取位置
                if times > 6:
                    log = open('log.txt', 'a')
                    log.write('<time : %s>\t<ip : %s>\t<op : 請求下載文件%s,中途斷開鏈接,下載失敗!>\n'%(time.ctime(), addr[0], file_name))
                    log.close()
                    break
            if con_len < 512: # 數(shù)據(jù)長度判斷
                log = open('log.txt', 'a')
                log.write('<time : %s>\t<ip : %s>\t<op : 請求下載文件%s,下載成功!>\n'%(time.ctime(), addr[0], file_name))
                log.close()
                break
    else:
        error_info = struct.pack('!HH21sb', 5, 1, 'cannot find this file'.encode('utf-8'), 0) # 返回文件未找到的錯誤信息
        tftp_socket.sendto(error_info, addr)
        log = open('log.txt', 'a')
        log.write('<time : %s>\t<ip : %s>\t<op : 請求下載文件%s,服務(wù)端無此文件,下載失敗!>\n'%(time.ctime(), addr[0], file_name))
        log.close()

def recv_file():
    '接收文件'
    if file_name in file_list:
        error_info = struct.pack('!HH19sb', 5, 2, 'file already exists'.encode('utf-8'), 0) # 返回文件未找到的錯誤信息
        tftp_socket.sendto(error_info, addr)
        log = open('log.txt', 'a')
        log.write('<time : %s>\t<ip : %s>\t<op : 請求上傳文件%s,服務(wù)端已有此文件,上傳失敗!>\n'%(time.ctime(), addr[0], file_name))
        log.close()
    else:
        ack_info = struct.pack('!HH', 4, 0)
        tftp_socket.sendto(ack_info, addr)
        recv_data = tftp_socket.recvfrom(1024) # 接收服務(wù)端信息
        f = open('./%s'%file_name, 'wb')
        i = 0
        while True:
            recv_msg = recv_data[0][4:] # 讀取接收信息
            recv_addr = recv_data[1] # 讀取地址
            recv_id = struct.unpack('!H', recv_data[0][2:4])[0] #讀取塊編號

            tftp_socket.sendto(struct.pack('!HH', 4, recv_id), recv_addr) #發(fā)送ACK

            i += 1
            if i == 65536:
                i = 0

            if i == recv_id: # 防止丟包的時候也寫入了文件
                f.write(recv_msg)
                
            if len(recv_data[0]) < 516:
                log = open('log.txt', 'a')
                log.write('<time : %s>\t<ip : %s>\t<op : 請求上傳文件%s,上傳成功!>\n'%(time.ctime(), addr[0], file_name))
                log.close()
                break
            recv_data = tftp_socket.recvfrom(1024) # 接收服務(wù)端信息

def main():
    global tftp_socket
    global file_name
    global file_list
    global addr
    global log
   
    tftp_socket = socket(AF_INET, SOCK_DGRAM)
    tftp_socket.bind(('', 2048))
    os.chdir('./server_files')

    while True:
        file_list = os.listdir('.') # 文件列表需要不斷更新
        recv_msg = tftp_socket.recvfrom(1024) # 從客戶端獲取信息
        msg_len = len(recv_msg[0])-9 # 獲取文件名長度
        file_name = struct.unpack('%ds'%msg_len, recv_msg[0][2:-7])[0].decode('utf-8') # 解碼出文件名
        addr = recv_msg[1] # 獲取客戶端地址信息
        op = struct.unpack('!H', recv_msg[0][:2])[0] # 獲取客戶端請求操作碼

        if op == 1: # 客戶端請求下載
            send_file()

        elif op == 2: # 客戶端請求上傳
            recv_file()

    log.close()

if __name__ == '__main__':
    tftp_socket = None
    file_name = ''
    file_list = []
    addr = ()
    log = None
    main()

客戶端代碼

from socket import *
import struct
import os
import time

def send_requ(io):
    '發(fā)送請求'
    name_len = len(file_name)
    send_msg = struct.pack('!H%dsb5sb'%name_len, io, file_name.encode('utf-8'), 0, 'octet'.encode('utf-8'), 0 )
    tftp_socket.sendto(send_msg, (ip_addr, port_addr)) 

def recv_file():
    '接收文件' 
    recv_data = tftp_socket.recvfrom(1024) # 接收服務(wù)端信息
    if struct.unpack('!H', recv_data[0][:2])[0] == 5: # 讀取操作碼
        print('-----服務(wù)端無文件 %s!-----'%file_name) 

    elif struct.unpack('!H', recv_data[0][:2])[0] == 3: # 讀取操作碼
        f = open('./%s'%file_name, 'wb')
        i = 0
        while True:
            recv_msg = recv_data[0][4:] # 讀取接收信息
            recv_addr = recv_data[1] # 讀取地址
            recv_id = struct.unpack('!H', recv_data[0][2:4])[0] #讀取塊編號
            print(recv_id, end = ' ')

            tftp_socket.sendto(struct.pack('!HH', 4, recv_id), recv_addr) #發(fā)送ACK

            i += 1
            if i == 65536:
                i = 0

            if i == recv_id: # 防止丟包的時候也寫入了文件
                f.write(recv_msg)
                
            if len(recv_data[0]) < 516:
                print('\n-----文件 %s下載完成!-----'%file_name)
                break
            recv_data = tftp_socket.recvfrom(1024) # 接收服務(wù)端信息

        f.close()

def send_file():
    '發(fā)送文件'
    echo_data = tftp_socket.recvfrom(1024) # 接收服務(wù)端ACK
    if struct.unpack('!H', echo_data[0][:2])[0] == 5: # 讀取操作碼
        print('服務(wù)端已有文件 %s,請勿重復(fù)上傳!'%file_name)

    elif struct.unpack('!HH', echo_data[0]) == (4, 0): # 讀取操作碼
        f = open('./%s'%file_name, 'rb')
        i = 1
        times = 0
        while True:
            send_msg = f.read(512) # 每次發(fā)送512字節(jié)
            msg_len = len(send_msg) 
            send_addr = echo_data[1] # 讀取地址
            pack_data = struct.pack('!HH%ds'%msg_len, 3, i, send_msg) # 打包發(fā)送信息
            tftp_socket.sendto(pack_data, send_addr)
            echo_data = tftp_socket.recvfrom(1024) # 接收客戶端返回值
            echo_op = struct.unpack('!HH', echo_data[0][:4]) # 讀取客戶端ACK
            if echo_op == (4, i):
                times = 0 # 重置服務(wù)端無響應(yīng)次數(shù)
                print(i,end = ' ')
                i += 1
                if i == 65536:
                    i = 0 # 重置塊編號
            elif echo_op == (4, i-1):
                times += 1 # 客戶端無響應(yīng)次數(shù)統(tǒng)計
                f.seek(1, -512) # 調(diào)整文件讀取位置
                if times > 6:
                    print('-----服務(wù)端無響應(yīng),傳輸失敗!-----')
                    break
            if msg_len < 512: # 數(shù)據(jù)長度判斷
                print('\n-----文件 %s上傳完成-----'%file_name)
                break
        f.close()

def main():
    '主函數(shù)'
    global tftp_socket
    global file_name
    global ip_addr
    global port_addr

    print('-'*50)
    print('歡迎使用下載上傳工具 by Shelming.Song')
    ip_addr = input('請輸入服務(wù)器IP地址:')
    port_addr = int(input('請輸入服務(wù)器端口:'))
    print('-'*50)

    tftp_socket = socket(AF_INET, SOCK_DGRAM)
    os.chdir('./client_files')
    

    while True:
        file_list = os.listdir('.') # 文件列表需要不斷更新
        op = input('請選擇您要進(jìn)行的操作:\n1.下載\n2.上傳\n3.退出\n')

        if op == '1': # 下載
            file_name = input('請輸入您要下載的文件名(含后綴名):')
            if file_name in file_list:
                confirm = input('本地已有文件 %s,是否重新下載? y/n:'%file_name)
                if confirm.lower() == 'y':
                    send_requ(1)
                    recv_file()
                else:
                    continue
            else:
                send_requ(1)
                recv_file()

        elif op == '2': # 上傳
            file_name = input('請輸入您要上傳的文件名(含后綴名):')
            if file_name not in file_list:
                print('本地?zé)o文件 %s!:'%file_name)
            else:
                send_requ(2)
                send_file()

        elif op == '3': # 退出
            print('程序即將退出,歡迎再次使用!')
            break

        else:
            print('您的輸入有誤,請重新輸入!')
            break

if __name__ == '__main__':
    tftp_socket = None
    file_name = ''
    ip_addr = ''
    port_addr = 0
    main()

<br />

個人技術(shù)博客地址：http://songmingyao.com/
<br />

最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。

推薦閱讀更多精彩內(nèi)容