详解Python3的TFTP文件传输
作者:Le-Bill 发布时间:2023-06-01 22:29:17
标签: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()
第三部分,展示
一下就不做逐一显示,如有问题,烦请之处修正,共同进步!
来源:https://blog.csdn.net/weixin_42122355/article/details/80791857
0
投稿
猜你喜欢
- <%MaxPerPage=8 ’定义页面最大的记录数为8<br>
- 字典与json字符串区别# python 中的字典格式,是dict类型{'a': 'sd'}如果声明a =
- 引用计数在Python源码中,每一个对象都是一个结构体表示,都有一个计数字段。typedef struct_object { i
- ClickHouse是近年来备受关注的开源列式数据库(DBMS),主要用于数据联机分析(OLAP)领域,于2016年开源。目前国内社区火热,
- 为了更好的说明问题,首先引出下面的题目//请说明下面变量 a-d 的值 var a = [[1][1]]; var b = [['a
- python是解释型语言,本文介绍了Python下利用turtle实现绘图功能的示例,本例所示为Python绘制一个树枝,具体实现代码如下:
- 1、fastcgi ,通过flup模块来支持,在nginx里对应的配置指令是 fastcgi_pass2、http,nginx使用proxy
- asp三天学好ADO对象之第一天 今天说一下Recordset 对象的属性1、CursorType 属性AdOpenForwardOnly:
- 代码如下:ALTER function [dbo].[GetOrderNum]( @ebaystockflag varchar(20)//规
- 从XML中读取数据到内存的实例: public clsSi
- 本文实例为大家分享了python使用Matplotlib绘制分段函数的具体代码,供大家参考,具体内容如下环境Python3Mac OS代码#
- 本讲的内容是使用ASP的ActiveX Server Components(组件),说实话下面的内置组件我们用的很少。一、 Browser
- 文件名全小写,可使用下划线包应该是简短的、小写的名字。如果下划线可以改善可读性可以加入。如mypackage。模块与包的规范同。如mymod
- 前言:macOS自带的Apache可以提供通过http://localhost:8081访问本地文件服务,那么python有没有类似功能的库
- 我们知道现实中的数据通常是杂乱无章的,需要大量的预处理才能使用。Pandas 是应用最广泛的数据分析和处理库之一,它提供了多种对原始数据进行
- (1)序列化即js中的Object转化为字符串1.使用obj.toJSONString()var str=obj.toJSONString(
- messageboxtkinter.messagebox中封装了多种消息框,其输入参数统一为title, message以及其他参数。其中t
- 中国,美国,英国3国时间js同步动态显示,对于做企业网站的朋友相信用的到,特别是做英文网站的朋友,加上这一段代码会给你的网站增色不少!本文j
- 1. 英雄的简单动画实现需求:在游戏初始化定义一个pygame.Rect的变量记录英雄的初始位置在游戏循环中每次让英雄的y-1--向上移动(
- 启动mysql server 失败,查看/var/log/mysqld.err080329 16:01:29 [ERROR] Can'