Repository navigation
Expand file tree
/
Copy pathtftp_server.py
More file actions
executable file
·123 lines (104 loc) · 3.03 KB
/
Copy pathtftp_server.py
File metadata and controls
executable file
·123 lines (104 loc) · 3.03 KB
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
114
115
116
117
118
119
120
121
122
123
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from socket import *
import multiprocessing as mp
import os
import signal
import sys
import time
# 文件库
FILE_PATH = "C:/Users/admin/Desktop/"
# 服务器功能
class TftpServer(object):
def __init__(self, connfd):
self.connfd = connfd
def handler(self):
while True:
data = self.connfd.recv(1024).decode()
if not data or data[0] == 'Q':
print(self.connfd.getpeername, "客户端退出")
self.connfd.close()
sys.exit(0)
elif data[0] == "L":
self.do_list()
elif data[0] == "G":
filename = data[2:]
self.do_get(filename)
elif data[0] == "P":
filename = data[2:]
self.do_put(filename)
def do_list(self):
# 获取列表
file_list = os.listdir(FILE_PATH)
if not file_list:
self.connfd.send("文件库为空".encode())
return
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("文件不存在".encode())
return
self.connfd.send(b'OK')
time.sleep(0.1)
# 发送文件
while True:
data = fd.read(1024)
if not data:
time.sleep(0.1)
self.connfd.send(b'##')
break
self.connfd.send(data)
print("文件发送完成")
def do_put(self, filename):
try:
fd = open(FILE_PATH + filename, 'wb')
except:
self.connfd.send("无法上传".encode())
return
self.connfd.send(b'OK')
while True:
data = self.connfd.recv(1024)
if data == b"##":
break
fd.write(data)
fd.close()
print("上传完毕")
# 控制程序流程 创建套接字，接收链接，创建子进程
def main():
HOST = "0.0.0.0"
PORT = 8888
ADDR = (HOST, PORT)
s = socket()
s.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
s.bind(ADDR)
s.listen(5)
print("Listen to port 8888....")
# signal.signal(signal.SIGCHLD,signal.SIG_IGN)
while True:
try:
connfd, addr = s.accept()
except KeyboardInterrupt:
s.close()
sys.exit("退出服务器")
except Exception as e:
print(e)
continue
print("客户端登录：", addr)
tftp = TftpServer(connfd)
p = mp.Process(target=tftp.handler)
p.start()
time.sleep(0.1)
p.join()
if __name__ == "__main__":
main()