参考网上一个FTP程序，重写了一遍，并稍加扩展
一、要求
1. 支持多用户同时登录
2. 可以注册用户，密码使用md5加密
3. 可以登录已注册用户
4. 支持cd切换目录，ls查看目录子文件
5. 支持上传、下载文件
6. 可以执行命令，如：ipconfig
二、代码
配置文件
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 4 import os 5 6 BASE_DIR = os.path.dirname(os.path.dirname(__file__)) 7 BASE_HOME = os.path.join(BASE_DIR, ‘home‘) 8 NAME_PWD = os.path.join(BASE_DIR, ‘db‘, ‘name_pwd‘) 9 USER_FILE = os.path.join(BASE_DIR, ‘db‘)
服务端：
1 #!/usr/bin/env python
2 # -*- coding:utf-8 -*-
3
4 import os
5 import hashlib
6 import pickle
7 import subprocess
8 import socketserver
9 from config import settings
10
11 baseHome = settings.BASE_HOME
12 class MyServer(socketserver.BaseRequestHandler):
13 def recv_file(self):
14 ‘‘‘
15 文件传输
16 :return:
17 ‘‘‘
18 conn = self.request
19 a = str(conn.recv(1024),encoding=‘utf-8‘)
20 file_size, file_name = a.split(‘,‘)
21 new_file_name = os.path.join(baseHome, file_name)
22 if file_name in baseHome: #检测文件是否已存在，涉及断点续传
23 has_recv = os.stat(baseHome).st_size #计算临时文件大小
24 conn.sendall(bytes(has_recv, encoding=‘utf-8‘))
25 with open(new_file_name,‘ab‘) as f: #追加模式
26 data = conn.recv(1024)
27 f.write(data)
28 has_recv += len(data)
29 else:
30 has_recv = 0
31 conn.sendall(bytes(‘s‘,encoding=‘utf-8‘)) #客户端收到字符串s，从0开始发送
32 with open(new_file_name, ‘wb‘) as f:
33 while has_recv<= int(file_size):
34 data = conn.recv(1024)
35 f.write(data)
36 has_recv += len(data)
37
38 def send_file(self, fileName):
39 ‘‘‘
40 向客户端发送文件
41 :param fileName:
42 :return:
43 ‘‘‘
44 filePath = os.path.join(self.currDir, fileName)
45 conn = self.request
46 if os.path.exists(filePath):
47 size = os.stat(filePath).st_size
48 conn.sendall(bytes(str(size)+‘,‘+fileName,encoding=‘utf-8‘))
49 ret = conn.recv(1024)
50 r = str(ret,encoding=‘utf-8‘)
51 if r==‘s‘:
52 has_send = 0
53 else:
54 has_send = int(r)
55 with open(filePath,‘rb‘) as f:
56 f.seek(has_send)
57 while has_send<size:
58 data = f.read(1024)
59 conn.sendall(data)
60 has_send+=len(data)
61 conn.sendall(bytes(‘0‘, encoding=‘utf-8‘))
62 else:
63 conn.sendall(bytes(‘0‘, encoding=‘utf-8‘))
64
65 def createDir(self, currDir, newName):
66 ‘‘‘
67 创建文件夹
68 :param currDir:当前所在目录
69 :param newName: 新文件夹名称
70 :return: 是否创建成功
71 ‘‘‘
72 mulu = os.path.join(baseHome, currDir)
73 newFilePath = os.path.join(mulu, newName)
74 if os.path.exists(newFilePath):
75 return ‘2‘
76 else:
77 ret = ‘0‘
78 try:
79 os.makedirs(newFilePath)
80 ret = ‘1‘
81 except OSError as e:
82 ret = ‘0‘
83 return ret
84
85 def command(self):
86 ‘‘‘
87 执行命令
88 :return:
89 ‘‘‘
90 conn = self.request
91 a = conn.recv(1024)
92 ret = str(a, encoding=‘utf-8‘)
93 ret2 = subprocess.check_output(ret, shell=True)
94 r = divmod(len(ret2), 1024)
95 s = r[0]+1
96 conn.sendall(bytes(str(s), encoding=‘utf-8‘))
97 conn.recv(1024)
98 conn.sendall(ret2)
99
100 def md5(self, pwd):
101 ‘‘‘
102 判断密码进行加密
103 :param pwd:
104 :return:
105 ‘‘‘
106 hash = hashlib.md5(bytes(‘xx7‘,encoding=‘utf-8‘))
107 hash.update(bytes(pwd, encoding=‘utf-8‘))
108 return hash.hexdigest()
109
110 def login(self, username, pwd):
111 ‘‘‘
112 登录
113 :param username:用户名
114 :param pwd: 密码
115 :return: 是否登录成功
116 ‘‘‘
117 if os.path.exists(settings.NAME_PWD):
118 s = pickle.load(open(settings.NAME_PWD,‘rb‘))
119 if username in s:
120 if s[username]==self.md5(pwd):
121 return True
122 else:
123 return False
124 else:
125 return False
126
127 def regist(self, username, pwd):
128 ‘‘‘
129 注册
130 :param username:用户名
131 :param pwd: 密码
132 :return: 是否注册成功
133 ‘‘‘
134 conn = self.request
135 s = {}
136 if os.path.exists(settings.NAME_PWD):
137 s = pickle.load(open(settings.NAME_PWD, ‘rb‘))
138 if username in s:
139 return False
140 else:
141 s[username] = self.md5(pwd)
142 mulu = os.path.join(settings.USER_FILE, username)
143 os.makedirs(mulu)
144 pickle.dump(s, open(settings.NAME_PWD, ‘wb‘))
145 return True
146
147 def before(self, username, pwd, ret):
148 ‘‘‘
149 判断注册和登录，并展示用户的详细目录信息，支持cd和ls命令
150 :param username: 用户名
151 :param pwd: 密码
152 :param ret:
153 :return:
154 ‘‘‘
155 conn = self.request
156 if ret == ‘1‘:
157 r = self.login(username,pwd)
158 if r:
159 conn.sendall(bytes(‘y‘,encoding=‘utf-8‘))
160 else:
161 conn.sendall(bytes(‘n‘,encoding=‘utf-8‘))
162 elif ret == ‘2‘:
163 r = self.regist(username, pwd)
164 if r:
165 conn.sendall(bytes(‘y‘,encoding=‘utf-8‘))
166 else:
167 conn.sendall(bytes(‘n‘,encoding=‘utf-8‘))
168
169 def user_file(self, username):
170 ‘‘‘
171 展示用户的详细目录信息，支持cd和ls命令
172 :param username: 用户名
173 :return:
174 ‘‘‘
175 conn = self.request
176 mulu = baseHome
177 self.currDir = mulu
178 conn.sendall(bytes(mulu, encoding=‘utf-8‘))
179 while True:
180 if conn:
181 b = conn.recv(1024)
182 ret = str(b, encoding=‘utf-8‘)
183 try:
184 a, b = ret.split(‘ ‘,1)
185 except Exception as e:
186 a = ret
187 if a == ‘cd‘:
188 if b==‘..‘:
189 mulu = os.path.dirname(mulu)
190 else:
191 mulu = os.path.join(mulu, b)
192 self.currDir = mulu
193 conn.sendall(bytes(mulu, encoding=‘utf-8‘))
194 elif a==‘ls‘:
195 ls = os.listdir(mulu)
196 print(ls)
197 a = ‘,‘.join(ls)
198 if a==‘‘:
199 a = ‘.‘
200 conn.sendall(bytes(a, encoding=‘utf-8‘))
201 elif a==‘mkdir‘:
202 m = self.createDir(self.currDir,b)
203 conn.sendall(bytes(m, encoding=‘utf-8‘))
204 elif a==‘q‘:
205 break
206
207 def handle(self):
208 conn = self.request
209 conn.sendall(bytes(‘welcome‘,encoding=‘utf-8‘))
210 b = conn.recv(1024)
211 ret = str(b, encoding=‘utf-8‘)
212 c = conn.recv(1024)
213 r = str(c, encoding=‘utf-8‘)
214 username, pwd = r.split(‘,‘)
215 self.before(username, pwd, ret)
216 self.user_file(username)
217 while True:
218 a=conn.recv(1024)
219 ret = str(a, encoding=‘utf-8‘)
220 if ret == ‘1‘:
221 self.recv_file()
222 elif ret==‘2‘:
223 self.command()
224 elif ret[0:4]==‘get:‘:
225 self.send_file(ret[4:])
226 elif ret==‘q‘:
227 break
228 else:
229 pass
230
231 if __name__ == ‘__main__‘:
232 server = socketserver.ThreadingTCPServer((‘‘,9999), MyServer)
233 server.serve_forever()
客户端：
1 #!/usr/bin/env python 2 # -*-coding:utf-8 -*- 3 4 import os, sys 5 import socket 6 7 def send_file(file_path): 8 ‘‘‘ 9 发送文件 10 :param file_path:文件名 11 :return: 12 ‘‘‘ 13 size = os.stat(file_path).st_size 14 file_name = os.path.basename(file_path) 15 obj.sendall(bytes(str(size)+‘,‘+file_name,encoding=‘utf-8‘)) 16 ret = obj.recv(1024) 17 r = str(ret, encoding=‘utf-8‘) 18 if r==‘s‘: #文件不存在，从头开始传 19 has_send = 0 20 else: #文件存在 21 has_send = int(r) 22 with open(file_path, ‘rb‘) as f: 23 f.seek(has_send) #定位到已经传到的位置 24 while has_send<size: 25 data = f.read(1024) 26 obj.sendall(data) 27 has_send+=len(data) 28 sys.stdout.write(‘\r‘) #情况文件内容 29 sys.stdout.write(‘已发送%s%%|%s‘ % (int(has_send/size*100), (round(has_send/size*40)*‘|‘))) 30 sys.stdout.flush() #强制刷出内存 31 print(‘上传成功！\n‘) 32 33 def recv_file(toPath, getFile): 34 ‘‘‘ 35 接收要下载的文件 36 :param toPath: 本地要保存文件的存放路径 37 :param getFile: 要下载的文件名称 38 :return: 39 ‘‘‘ 40 obj.sendall(bytes(‘get:‘+getFile,encoding=‘utf-8‘)) 41 a = str(obj.recv(1024), encoding=‘utf-8‘) 42 file_size, file_name = a.split(‘,‘) 43 file_size = int(file_size) 44 if file_size == 0: 45 print(‘没有找到此文件‘) 46 else: 47 new_file_name = os.path.join(toPath, file_name) 48 if file_name in toPath: 49 has_recv = os.stat(toPath).st_size 50 obj.sendall(bytes(has_recv, encoding=‘utf-8‘)) 51 with open(new_file_name,‘ab‘) as f: 52 while has_recv<=file_size: 53 data = obj.recv(1024) 54 f.write(data) 55 has_recv+=len(data) 56 sys.stdout.write(‘\r‘) # 情况文件内容 57 sys.stdout.write(‘已接收%s%%|%s‘ % (int(has_recv / file_size * 100), (round(has_recv / file_size * 40) * ‘|‘))) 58 sys.stdout.flush() # 强制刷出内存 59 else: 60 has_recv = 0 61 obj.sendall(bytes(‘s‘, encoding=‘utf-8‘)) 62 with open(new_file_name, ‘wb‘) as f: 63 while has_recv<= file_size: 64 data = obj.recv(1024) 65 f.write(data) 66 has_recv += len(data) 67 sys.stdout.write(‘\r‘) # 情况文件内容 68 sys.stdout.write(‘已接收%s%%|%s‘ % (int(has_recv / file_size * 100), (round(has_recv / file_size * 40) * ‘|‘))) 69 sys.stdout.flush() # 强制刷出内存 70 print(‘接收成功！\n‘) 71 72 73 def command(command_name): 74 ‘‘‘ 75 执行命令 76 :param command_name: 77 :return: 78 ‘‘‘ 79 obj.sendall(bytes(command_name, encoding=‘utf-8‘)) 80 ret = obj.recv(1024) #接受命令需要接受的次数 81 obj.sendall(bytes(‘收到次数‘,encoding=‘utf-8‘)) 82 r = str(ret, encoding=‘utf-8‘) 83 for i in range(int(r)): #共需接收int(r)次 84 ret = obj.recv(1024) #等待客户端发送 85 r = str(ret, encoding=‘GBK‘) 86 print(r) 87 88 def login(username, pwd): 89 ‘‘‘ 90 登录 91 :param username: 用户名 92 :param pwd: 密码 93 :return: 是否登录成功 94 ‘‘‘ 95 obj.sendall(bytes(username+‘,‘+pwd, encoding=‘utf-8‘)) 96 ret = obj.recv(1024) 97 r = str(ret, encoding=‘utf-8‘) 98 if r==‘y‘: 99 return True 100 else: 101 return False 102 103 def regist(username, pwd): 104 ‘‘‘ 105 注册 106 :param username: 用户名 107 :param pwd: 密码 108 :return: 是否注册成功 109 ‘‘‘ 110 obj.sendall(bytes(username+‘,‘+pwd, encoding=‘utf-8‘)) 111 ret = obj.recv(1024) 112 r = str(ret, encoding=‘utf-8‘) 113 if r==‘y‘: 114 return True 115 else: 116 return False 117 118 def before(username, pwd): 119 ‘‘‘ 120 选择注册和登录，并展示用户的详细目录信息，支持cd和ls命令 121 :param username: 用户名 122 :param pwd: 密码 123 :return: 124 ‘‘‘ 125 a = input(‘请选择 1.登录 2.注册：‘) 126 obj.sendall(bytes(a, encoding=‘utf-8‘)) 127 # obj.recv() 128 if a==‘1‘: 129 ret = login(username, pwd) 130 if ret: 131 print(‘登录成功‘) 132 return 1 133 else: 134 print(‘用户名或密码错误‘) 135 return 0 136 elif a==‘2‘: 137 ret = regist(username, pwd) 138 if ret: 139 print(‘注册成功‘) 140 return 1 141 else: 142 print(‘用户名已存在‘) 143 return 0 144 145 def user_file(username): 146 # obj.sendall(bytes(‘打印用户文件路径‘, encoding=‘utf-8‘)) 147 ret = obj.recv(1024) 148 r = str(ret, encoding=‘utf-8‘) 149 print(r) 150 while True: 151 a = input(‘输入 cd切换目录，ls查看目录详细信息，mkdir创建文件夹，q退出：‘) 152 a = a.strip() 153 obj.sendall(bytes(a, encoding=‘utf-8‘)) 154 if a==‘q‘: 155 break 156 elif a[0:5]==‘mkdir‘: 157 ret = obj.recv(1024) 158 r = str(ret, encoding=‘utf-8‘) 159 if r==‘1‘: 160 print(‘文件夹创建成功‘) 161 elif r==‘2‘: 162 print(‘文件夹已存在！‘) 163 else: 164 print(‘创建失败！‘) 165 else: 166 ret = obj.recv(1024) 167 r=str(ret, encoding=‘utf-8‘) 168 if len(r)==1: #判断是cd结果，还是ls的结果（ls只有一个子目录，直接打印） 169 print(r) 170 else: 171 li = r.split(‘,‘) 172 for i in li: 173 print(i) 174 175 def main(username, pwd): 176 ret = obj.recv(1024) 177 r = str(ret, encoding=‘utf-8‘) 178 print(r) 179 result = before(username, pwd) #判断登录/注册 180 if result: 181 user_file(username) 182 while True: 183 a = input(‘请选择 1.传文件 2.执行命令 3.收文件 q 退出：‘) 184 obj.sendall(bytes(str(a),encoding=‘utf-8‘)) 185 if a==‘1‘: 186 b = input(‘请输入文件路径：‘) 187 if os.path.exists(b): 188 send_file(b) 189 obj.sendall(bytes(‘hhe‘, encoding=‘utf-8‘)) 190 elif a==‘2‘: 191 b = input(‘请输入command：‘) 192 command(b) 193 elif a==‘3‘: 194 b = input(‘请输入存放路径：‘) 195 c = input(‘请输入要获取的文件：‘) 196 recv_file(b, c) 197 elif a==‘q‘: 198 break 199 else: 200 print(‘输入错误！‘) 201 202 obj.close() 203 204 if __name__ == ‘__main__‘: 205 obj = socket.socket() 206 obj.connect((‘192.168.1.100‘,9999)) 207 username = input(‘请输入用户名：‘) 208 pwd = input(‘请输入密码：‘) 209 main(username, pwd)
时间： 2024-10-10 16:47:35