Python3简单的TFTP文件传输

TFTP文件传输

功能：

1、获取文件列表

2、上传文件

3、下载文件

4、退出

第一部分，TftpServer部分。

①导入相关模块

from socket import *
import os
import signal
import sys
import time

②确定文件路径

# 文件库路径
FILE_PATH = "/home/tarena/"

③建立一个类，用来实现服务器功能模块

class TftpServer(object):
    def __init__(self, connfd):
        self.connfd = connfd

    def do_list(self):
        # 获取列表
        file_list = os.listdir(FILE_PATH)
        # 如果对应的路径内没有文件,返回Empty
        if not file_list:
            self.connfd.send('Empty'.encode())
            return
        # 路径存在文件,向客户端发送OK
        else:
            self.connfd.send(b'OK')
            time.sleep(0.1)

        files = ""
        for file in file_list:
            # 排除以'.'开头的隐藏文件
            if file[0] != '.' and \
                    os.path.isfile(FILE_PATH + file):
                files = files + file + '#'
        # 返回文件列表
        self.connfd.send(files.encode())

    # 下载文件功能
    def do_get(self, filename):
        try:
            fd = open(FILE_PATH + filename, 'rb')
        except:
            self.connfd.send("File doesn't exist".encode())
            return
        # 如果能正常打开,发送OK
        self.connfd.send(b"OK")
        time.sleep(0.1)
        # 开始发送文件
        try:
            for line in fd:
                self.connfd.send(line)
            fd.close()
        except Exception as e:
            print(e)
        time.sleep(0.1)
        self.connfd.send(b'##')
        print("File send over")

    # 开始上传文件
    def do_put(self, filename):
        try:
            fd = open(FILE_PATH + filename, 'w')
        except:
            self.connfd.send("Some error")
        # 如果能正常打开文件,则发送OK
        self.connfd.send(b'OK')
        # 开始发送
        while True:
            # data为文件内容
            data = self.connfd.recv(1024).decode()
            if data == "##":
                break
            fd.write(data)
        fd.close()
        print("上传完毕")

④主流程控制

def main():
    # 创建套接字/地址/端口
    HOST = '0.0.0.0'
    PORT = 8888
    ADDR = (HOST, PORT)

    sockfd = socket()
    # 设置端口可重用
    sockfd.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
    # 绑定地址
    sockfd.bind(ADDR)
    # 设置监听队列大小
    sockfd.listen(5)

    signal.signal(signal.SIGCHLD, signal.SIG_IGN)
    print("Listen to port 8888....")

    while True:
        try:
            connfd, addr = sockfd.accept()
        except KeyboardInterrupt:
            sockfd.close()
            sys.exit("Server exit")
        except Exception as e:
            print(e)
            continue
        print("Client login:", addr)
        # 创建父子进程
        pid = os.fork()
        if pid < 0:
            print("Process creation failed")
            continue
        elif pid == 0:
            # 子进程负责请求接收和发送,所以节省资源,关闭连接套字
            sockfd.close()
            tftp = TftpServer(connfd)
            # 接收客户端请求
            while True:
                data = connfd.recv(1024).decode()
                if not data:
                    continue
                # 调用do_list方法获取文件列表
                elif data[0] == 'L':
                    tftp.do_list()
                # data ==> G filename
                # 文件名以G开头,以空格为间隔发送过来
                elif data[0] == 'G':
                    filename = data.split(' ')[-1]
                    tftp.do_get(filename)
                elif data[0] == 'P':
                    filename = data.split(' ')[-1]
                    tftp.do_put(filename)
                elif data[0] == 'Q':
                    print("客户端退出")
                    sys.exit(0)

        else:
            connfd.close()
            continue

⑤运行主控制流程，等待客户端连接

if __name__ == "__main__":
    main()

第二部分，TftpClient

①导入相关模块

from socket import *
import sys
import time

②实现基本的请求功能

class TftpServer(object):
    def __init__(self, sockfd):
        self.sockfd = sockfd

    def do_list(self):
        self.sockfd.send(b"L")  # 发送请求类型
        # 等待接收服务器端确认
        data = self.sockfd.recv(1024).decode()
        if data == 'OK':
            data = self.sockfd.recv(4096).decode()
            files = data.split('#')
            for file in files:
                print(file)
            print("%%%%%There is file list%%%%%\n")
        else:
            # 失败的原因由服务器发送过来
            print(data)

    def do_get(self, filename):
        self.sockfd.send(('G '+filename).encode())
        data = self.sockfd.recv(1024).decode()
        if data == 'OK':
            fd = open(filename, 'w')
            while True:
                data = self.sockfd.recv(1024).decode()
                if data == "##":
                    break
                fd.write(data)
            fd.close()
            print("%s Download over\n" % filename)
        else:
            print(data)

    def do_put(self, filename):
        try:
            fd = open(filename, 'rb')
        except:
            print("There is no such file")
            return
        self.sockfd.send(("P " + filename).encode())
        data = self.sockfd.recv(1024).decode()
        if data == 'OK':
            for line in fd:
                self.sockfd.send(line)
            fd.close()
            time.sleep(0.1)
            self.sockfd.send(b'##')
            print("%s upload over" % filename)
        else:
            print(data)

    def do_quit(self):
        self.sockfd.send(b'Q')

③主流程控制

# 套接字连接
def main():
    if len(sys.argv) < 3:
        print("argv is error")
        return
    HOST = sys.argv[1]
    PORT = int(sys.argv[2])
    ADDR = (HOST, PORT)

    sockfd = socket()
    sockfd.connect(ADDR)

    tftp = TftpServer(sockfd)  # tftp对象调用请求方法

    while True:
        print("=======命令选项========")
        print("******* list *********")
        print("*******get file ******")
        print("*******put file ******")
        print("******* quit *********")
        print("======================")

        cmd = input("请输入命令>>")

        if cmd.strip() == 'list':
            tftp.do_list()
        elif cmd[:3] == "get":
            filename = cmd.split(' ')[-1]
            tftp.do_get(filename)
        elif cmd[:3] == "put":
            filename = cmd.split(' ')[-1]
            tftp.do_put(filename)
        elif cmd.strip() == "quit":
            tftp.do_quit()
            sockfd.close()
            sys.exit("Welcome")
        else:
            print("Enter the right order!!!")
            continue

④运行客户端

if __name__ == "__main__":
    main()

第三部分，展示

 

一下就不做逐一显示，如有问题，烦请之处修正，共同进步！