详解Python3的TFTP文件传输

Python  /  管理员 发布于 7年前   231

TFTP文件传输

功能：

1、获取文件列表

2、上传文件

3、下载文件

4、退出

第一部分，TftpServer部分。

①导入相关模块

from socket import *import osimport signalimport sysimport 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() 

第三部分，展示

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