python实现简单tftp(基于udp协议)
作者:Erick-LONG 时间:2021-01-30 15:01:23
本文实例为大家分享了python实现简单tftp的具体代码,供大家参考,具体内容如下
tftp是基于udp的协议
实现简单的tftp,首先要有tftp的协议图。
tftp默认接收端口为69,但每次有连接过来后,tftp会随机分配一个端口来专门为这个连接来服务。
操作码:1.上传 2.下载 3.传数据 4.接收确认 5.错误码
tftp服务器简单实现:
from threading import Thread
from socket import *
import struct
def upload(filename,user_ip,user_port):
num = 0
f = open(filename,'ab')
s_up = socket(AF_INET,SOCK_DGRAM)
send_data_1 = struct.pack("!HH",4,num)
s_up.sendto(send_data_1,(user_ip,user_port)) #第一次用随机端口发送
while True:
recv_data,user_info = s_up.recvfrom(1024) #第二次客户连接我随机端口
caozuohao_up,ack_num = struct.unpack('!HH',recv_data[:4])
print(caozuohao_up,ack_num,num)
if int(caozuohao_up) == 3 and ack_num == num :
f.write(recv_data[4:])
send_data = struct.pack("!HH",4,num)
s_up.sendto(send_data,(user_ip,user_port)) #第二次我用随机端口发
num = num + 1
if len(recv_data) < 516:
print(user_ip+'上传文件'+filename+':完成')
f.close()
exit()
def download(filename,user_ip,user_port):
s_down = socket(AF_INET, SOCK_DGRAM)
num = 0
try:
f = open(filename,'rb')
except:
error_data = struct.pack('!HHHb',5,5,5,num)
s_down.sendto(error_data, (user_ip,user_port)) #文件不存在时发送
exit() #只会退出此线程
while True:
read_data = f.read(512)
send_data = struct.pack('!HH',3,num) + read_data
s_down.sendto(send_data, (user_ip,user_port)) #数据第一次发送
if len(read_data) < 512:
print('传输完成, 对方下载成功')
exit()
recv_ack = s_down.recv(1024) #第二次接收
caozuoma,ack_num = struct.unpack("!HH", recv_ack)
# print(caozuoma,ack_num,len(read_data))
num += 1
if int(caozuoma) != 4 or int(ack_num) != num-1 :
exit()
f.close()
s = socket(AF_INET,SOCK_DGRAM)
s.bind(('',69))
def main():
while 1:
recv_data,(user_ip,user_port) = s.recvfrom(1024) #第一次客户连接69端口
print(recv_data, user_ip, user_port)
if struct.unpack('!b5sb',recv_data[-7:]) == (0, b'octet', 0):
caozuoma = struct.unpack('!H',recv_data[:2])
filename = recv_data[2:-7].decode('gb2312')
if caozuoma[0] == 1:
print('对方想下载数据',filename)
t = Thread(target = download, args = (filename,user_ip,user_port))
t.start()
elif caozuoma[0] == 2:
print('对方想上传数据',filename)
t = Thread(target = upload, args = (filename,user_ip,user_port))
t.start()
if __name__ == '__main__':
main()
上传数据简单实现:
#!/usr/bin/env python3
#coding=utf-8
import struct
from socket import *
server_ip = '192.168.119.157'
send_data_1 = struct.pack('!H8sb5sb',2,'王辉.jpg'.encode('gb2312'),0,b'octet',0)
s = socket(AF_INET,SOCK_DGRAM)
s.sendto(send_data_1,(server_ip,69)) #第一次发给服务器69端口
f = open('王辉.jpg','rb')
recv_data = s.recvfrom(1024) #第一次接收数据
rand_port = recv_data[1][1]
print()
ack_num = struct.unpack("!HH",recv_data[0][:4])
num = 0
while True:
read_data = f.read(512)
send_data = struct.pack('!HH',3,num) + read_data
s.sendto(send_data,(server_ip,rand_port)) #第二次发给服务器的随机端口
recv_data_2,userinfo = s.recvfrom(1024)
print(recv_data_2)
ack_num = struct.unpack('!H',recv_data_2[2:4])
print(len(read_data),num,ack_num[0],rand_port)
if len(read_data) < 512 or ack_num[0] != num :
break
num = num + 1
下载数据简单实现:
#!/usr/bin/env python3
#coding=utf-8
import struct
from socket import *
filename = 'test.jpg'
server_ip = '192.168.1.113'
send_data = struct.pack('!H%dsb5sb'%len(filename),1,filename.encode('gb2312'),0,'octet'.encode('gb2312'),0)
s = socket(AF_INET,SOCK_DGRAM)
s.sendto(send_data,(server_ip,69)) #第一次发送, 连接服务器69端口
f = open(filename,'ab')
while 1:
recv_data = s.recvfrom(1024) #接收数据
caozuoma,ack_num = struct.unpack('!HH',recv_data[0][:4]) #获取数据块编号
rand_port = recv_data[1][1] #获取服务器的随机端口
if int(caozuoma) == 5:
print('服务器返回: 文件不存在...')
break
print(caozuoma,ack_num,rand_port,len(recv_data[0]))
f.write(recv_data[0][4:])
if len(recv_data[0]) < 516:
break
ack_data = struct.pack("!HH",4,ack_num)
s.sendto(ack_data,(server_ip,rand_port)) #回复ACK确认包
来源:http://www.cnblogs.com/Erick-L/p/7144139.html
标签:python,tftp,udp
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Django Channels 实现点对点实时聊天和消息推送功能
2022-04-03 10:37:56
MySQL最新漏洞分析
2012-07-11 15:41:10
go流程控制代码详解
2023-10-15 18:14:19
Golang利用casbin实现权限验证详解
2023-08-06 23:18:45
![](https://img.aspxhome.com/file/2023/0/93660_0s.png)
用Dreamweaver MX设计各种网页鼠标样式
2008-10-04 10:18:00
Django搭建项目实战与避坑细节详解
2021-04-30 06:46:46
![](https://img.aspxhome.com/file/2023/2/78482_0s.png)
php多任务程序实例解析
2023-11-18 00:22:09
Flask request 对象介绍
2022-11-01 23:45:35
![](https://img.aspxhome.com/file/2023/3/78543_0s.png)
jsp页面中获取servlet请求中的参数的办法详解
2023-06-19 10:52:00
Python数字/字符串补零操作实例代码
2021-07-13 03:16:57
![](https://img.aspxhome.com/file/2023/4/80384_0s.png)
Python matplotlib 绘制双Y轴曲线图的示例代码
2023-04-14 11:07:27
![](https://img.aspxhome.com/file/2023/9/81549_0s.jpg)
解读JavaScript代码 var ie = !-[1,] 最短的IE判定代码
2011-06-06 10:29:00
python中map()与zip()操作方法
2023-11-11 08:09:29
解决TensorFlow程序无限制占用GPU的方法
2021-11-22 13:13:44
![](https://img.aspxhome.com/file/2023/0/95560_0s.png)
使用python 获取进程pid号的方法
2023-06-11 21:25:52
php将12小时制转换成24小时制的方法
2023-11-21 15:56:08
Python通过tkinter实现百度搜索的示例代码
2022-06-28 01:28:46
Python实现遗传算法(二进制编码)求函数最优值方式
2023-05-07 15:51:40
![](https://img.aspxhome.com/file/2023/4/90954_0s.jpg)
Python中序列的修改、散列与切片详解
2022-10-27 14:47:58
![](https://img.aspxhome.com/file/2023/0/104820_0s.png)
Python装饰器用法实例总结
2023-11-18 07:37:36