详解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() 

第三部分，展示

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