1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
from socket import *
import struct
import os
import time
def send_file():
global log
'发送文件'
if file_name in file_list: # 检测服务端是否存在客户端要下载的文件
f = open('./%s'%file_name, 'rb')
i = 1
times = 0
while True:
content = f.read(512)
con_len = len(content)
pack_content = struct.pack('!HH%ds'%con_len, 3, i, content)
tftp_socket.sendto(pack_content, addr)
echo_msg = tftp_socket.recvfrom(1024) # 接收客户端返回值
echo_op = struct.unpack('!HH', echo_msg[0][:4]) # 读取客户端ACK
if echo_op == (4, i):
times = 0 # 重置客户端无响应次数
i += 1
if i == 65536:
i = 0 # 重置块编号
elif echo_op == (4, i-1):
times += 1 # 客户端无响应次数统计
f.seek(1, -512) # 调整文件读取位置
if times > 6:
log = open('log.txt', 'a')
log.write('<time : %s>\t<ip : %s>\t<op : 请求下载文件%s,中途断开链接,下载失败!>\n'%(time.ctime(), addr[0], file_name))
log.close()
break
if con_len < 512: # 数据长度判断
log = open('log.txt', 'a')
log.write('<time : %s>\t<ip : %s>\t<op : 请求下载文件%s,下载成功!>\n'%(time.ctime(), addr[0], file_name))
log.close()
break
else:
error_info = struct.pack('!HH21sb', 5, 1, 'cannot find this file'.encode('utf-8'), 0) # 返回文件未找到的错误信息
tftp_socket.sendto(error_info, addr)
log = open('log.txt', 'a')
log.write('<time : %s>\t<ip : %s>\t<op : 请求下载文件%s,服务端无此文件,下载失败!>\n'%(time.ctime(), addr[0], file_name))
log.close()
def recv_file():
'接收文件'
if file_name in file_list:
error_info = struct.pack('!HH19sb', 5, 2, 'file already exists'.encode('utf-8'), 0) # 返回文件未找到的错误信息
tftp_socket.sendto(error_info, addr)
log = open('log.txt', 'a')
log.write('<time : %s>\t<ip : %s>\t<op : 请求上传文件%s,服务端已有此文件,上传失败!>\n'%(time.ctime(), addr[0], file_name))
log.close()
else:
ack_info = struct.pack('!HH', 4, 0)
tftp_socket.sendto(ack_info, addr)
recv_data = tftp_socket.recvfrom(1024) # 接收服务端信息
f = open('./%s'%file_name, 'wb')
i = 0
while True:
recv_msg = recv_data[0][4:] # 读取接收信息
recv_addr = recv_data[1] # 读取地址
recv_id = struct.unpack('!H', recv_data[0][2:4])[0] #读取块编号
tftp_socket.sendto(struct.pack('!HH', 4, recv_id), recv_addr) #发送ACK
i += 1
if i == 65536:
i = 0
if i == recv_id: # 防止丢包的时候也写入了文件
f.write(recv_msg)
if len(recv_data[0]) < 516:
log = open('log.txt', 'a')
log.write('<time : %s>\t<ip : %s>\t<op : 请求上传文件%s,上传成功!>\n'%(time.ctime(), addr[0], file_name))
log.close()
break
recv_data = tftp_socket.recvfrom(1024) # 接收服务端信息
def main():
global tftp_socket
global file_name
global file_list
global addr
global log
tftp_socket = socket(AF_INET, SOCK_DGRAM)
tftp_socket.bind(('', 2048))
os.chdir('./server_files')
while True:
file_list = os.listdir('.') # 文件列表需要不断更新
recv_msg = tftp_socket.recvfrom(1024) # 从客户端获取信息
msg_len = len(recv_msg[0])-9 # 获取文件名长度
file_name = struct.unpack('%ds'%msg_len, recv_msg[0][2:-7])[0].decode('utf-8') # 解码出文件名
addr = recv_msg[1] # 获取客户端地址信息
op = struct.unpack('!H', recv_msg[0][:2])[0] # 获取客户端请求操作码
if op == 1: # 客户端请求下载
send_file()
elif op == 2: # 客户端请求上传
recv_file()
log.close()
if __name__ == '__main__':
tftp_socket = None
file_name = ''
file_list = []
addr = ()
log = None
main()
|