基于多线程多用户的FTP服务器与客户端功能实现
Posted 水·域
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了基于多线程多用户的FTP服务器与客户端功能实现相关的知识，希望对你有一定的参考价值。
项目介绍：
- 用户加密认证
- 允许同时多用户登录
- 每个用户有自己的家目录 ，且只能访问自己的家目录
- 对用户进行磁盘配额，每个用户的可用空间不同
- 允许用户在ftp server上随意切换目录
- 允许用户查看当前目录下文件
- 允许上传和下载文件，保证文件一致性
- 文件传输过程中显示进度条
实现的原理：
服务器端启用端口监听，并对每一连接启用一个线程，对用户登陆密码采用SHA512进行加密并进行匹配，当用户登陆成功后，实例化FTPS，并引导客户端进入主命令模式，
然后实现FTP的上传功能、下载功能、新建目录、删除文件或目录、切换目录等实例化操作，同时对相关上传下载进行进度条显示，服务器端显示下载或上传文件的大小等
客户端与服务器协商建立连接后，进行用户身份登陆，登陆成功接收服务器指令，转入命令输入窗口，同时对put 与 get命令进行判断，实现特定的上传与下载功能
核心代码实现如下：
服务器端
main.py
#!/usr/bin/env python3.5 # -*-coding:utf8-*- import os,sys,socket,pickle BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASEDIR) from conf import setting from core import file_handler from core import db_handler import select,hashlib import threading def login(username,password): """ FTP登陆验证函数 :param username: :param password: :return: # testDict ={"username":"jjb","password":"123456","file_dir":"E:\\python","file_size":500} # file = \'jjb.pkl\' # fp = open(file,\'wb\') # pickle.dump(testDict,fp) # fp.close() f = open("jjb.pkl","rb") data = pickle.loads(f.read()) f.close() print(data) """ #实例化加密函数 hash = hashlib.sha512() db= db_handler.handler(setting.DATABASE,username) if os.path.isfile(db): f = open(db,"rb") data = pickle.loads(f.read()) f.close() if username == data["name"]: hash.update(bytes(data["password"],"utf8")) hash_pwd = hash.hexdigest() if hash_pwd == password: filedir = data["file_dir"] filesize = data["file_size"] return "True|%s|%s"%(filedir,filesize) else: return "False||" else: return "False||" else: return "False||" def process(conn,addr): flage = "False" # 接收客户端连接请求信息 info = conn.recv(1000) if info.decode() == "connect": conn.send(bytes("login","utf8")) # 接收用户及密码信息 while flage =="False": user_check =conn.recv(8000) # 分割用户名及密码 username,password = str(user_check.decode()).split("|") # 调用登陆验证函数 login_ack = login(username,password) flage,home,size = str(login_ack).split("|") # print(flage,home,size) # print("user_input:",username,"user_pass:",password) if flage =="True": # 登陆成功发送登陆确认信息给客户端 conn.send(bytes("login_ack","utf8")) # 实例化FTPserver ftp = file_handler.FTPs(username,conn,home,size) # 登陆用户，数据连接，工作目录,磁盘配额 ftp.run() break else: # 登陆失败，发送给客户端重新验证 conn.send(bytes("登陆失败！","utf8")) def ftp_server(): \'\'\' 启动FTP服务器端，开启线程监听 :return: \'\'\' server = socket.socket() server.bind((setting.IP_PORT["host"],setting.IP_PORT["port"])) server.listen(10) while True: r,w,e = select.select([server,], [], [], 1) for i,server in enumerate(r): conn,addr = server.accept() # 创建线程 t = threading.Thread(target=process, args=(conn, addr)) # 启动线程 t.start() server.close() def run(): ftp_server() if __name__ =="__main__": run()
file_handler.py
#!/usr/bin/env python3.5 # -*-coding:utf8-*- import os,sys BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASEDIR) import re from core import db_handler from conf import setting import pickle class FTPs(object): \'\'\' ftp操作命令方法： \'\'\' def __init__(self,username,conn,home,total_size): \'\'\' 初始化参数 :param username: 操作用户名 :param conn: sock连接 :param home: 用户根目录 :param total_size: 磁盘配额 :return: \'\'\' self.username = username self.conn = conn self.root = home self.home = self.root self.total_size = int(total_size) self.cmd_file = None # 文件指令 self.psize = 4096 # 文件分片 def getdirsize(self,space): \'\'\' 计算磁盘空间大小 :return: \'\'\' self.dirsize = 0 for root,dirs,files in os.walk(space): self.dirsize += (sum([os.path.getsize(os.path.join(root,name))for name in files])/1024) return int(self.dirsize) def put(self): \'\'\' 上传文件 :return: \'\'\' if self.cmd_file: self.user_space = int(self.getdirsize(self.root)/1024) # 组合接收字符串 self.file_root = \'%s\\\\%s\'% (self.home,self.cmd_file) # # 获取文件名 self.f =os.path.basename(self.file_root) if os.path.isdir(self.home): os.chdir(self.home) else: os.makedirs(self.home) os.chdir(self.home) try: self.conn.send(bytes("f_ack","utf8")) self.size = str(self.conn.recv(1024).decode()).split("|") if self.size[0]== "fsize": self.fss = int(self.size[1]) self.f_total_size = int(self.user_space + (self.fss/1024/1024)) if self.f_total_size < self.total_size: # 判断空间是否超额 self.conn.send(bytes("f_ack_ready","utf8")) self.bsize = 0 print("需要上传文件大小：",self.fss) # 打开文件 f=open(self.f,\'wb\') while self.bsize < self.fss: data = self.conn.recv(self.psize) self.bsize += len(data) f.write(data) self.conn.send(bytes("ok","utf8")) print("实际已上传文件大小：",self.bsize) else: self.conn.send(bytes("上传空间不足！无法上传,你当前磁盘配额为%sM"%self.total_size,"utf8")) except Exception as ex: self.conn.send(bytes(ex,"utf8")) else: self.conn.send(bytes("请上传文件，文件不能为空","utf8")) def get(self): \'\'\' 下载文件 :return: \'\'\' if self.cmd_file: os.chdir(self.home) # 进入用户根目录 self.file = os.getcwd()+"\\\\"+ self.cmd_file if os.path.isfile(self.file): f = open(self.file, \'rb\') self.fsize = os.path.getsize(self.file) # 获取要发送文件的大小 self.conn.send(bytes("f_ack_read","utf8")) self.conn.recv(1000) print("需发送文件大小：",self.fsize) self.conn.send(bytes("fsize|%s"%self.fsize,"utf8")) # 发送文件大小及要发送准备完毕指令 if self.conn.recv(1000).decode() == "f_ack": # 接收对方是否准备就绪 self.fsize = int(self.fsize) self.size = 0 ack ="" while self.size < self.fsize and ack !="ok": data = f.read(self.fsize) # 一次读取分片大小4096 self.conn.send(data) self.size += len(data) print("实际发送文件大小：",self.size) ack = self.conn.recv(1000).decode() # 接收客户端是否下载完指令 self.conn.send(bytes("成功","utf8")) else: self.conn.send(bytes("接收失败","utf8")) else: self.conn.send(bytes("文件不存在","utf8")) else: self.conn.send(bytes("请输入文件名","utf8")) def dir(self): \'\'\' 查看文件 :return: \'\'\' self.current_space =int(self.getdirsize(self.home)) # 文件列表 self.li = "" # 目录列表 self.dl = "" try: os.chdir(self.home) except: os.makedirs(self.home) os.chdir(self.home) try: if os.listdir(os.getcwd()): for self.i in os.listdir(os.getcwd()): self.file = os.getcwd()+\'\\\\\'+self.i if os.path.isfile(self.file): # 获取文件大小 self.fsize = int(os.path.getsize(self.file)/1024) if self.fsize < 1: self.fsize = 4 else: self.fsize +=4 self.li += \'%s -rw-rw-rw- 占用大小：%skb\\r\\n\'% (self.i,self.fsize) else: self.dl += \'%s\\r\\n\'%self.i self.conn.send(bytes("目录：\\r\\n\\r\\n%s 文件：\\r\\n%s\\r\\n \\r\\n当前目录空间大小：%skb"%(self.dl,self.li,self.current_space),"utf8")) else: self.conn.send(bytes("当前目录为:%s"%(self.home),"utf8")) except Exception as ex: self.conn.send(bytes(ex,"utf8")) def cd(self): \'\'\' 进入目录 :return: \'\'\' if self.cmd_file: os.chdir(self.home) # 先进入到工作目录 self.dir_change = os.path.abspath(os.path.join(self.home,"%s\\%s"%(self.home,self.cmd_file))) if self.root in self.dir_change: try: os.chdir(self.dir_change) self.home = self.dir_change self.conn.send(bytes("当前工作目录为：%s"%self.home,"utf8")) except: os.makedirs(self.dir_change) os.chdir(self.dir_change) self.home = self.dir_change self.conn.send(bytes("当前工作目录为：%s"%self.home,"utf8")) else: self.conn.send(bytes("当前工作目录为：%s 更改失败！"%self.home,"utf8")) else: os.chdir(self.home) self.conn.send(bytes("当前工作目录为：%s"%self.home,"utf8")) def mkd(self): \'\'\' 创建目录 :return: \'\'\' if self.cmd_file: try: os.makedirs(self.cmd_file) self.conn.send(bytes("创建目录成功!","utf8")) except Exception as ex: self.conn.send(bytes("创建目录失败！原因:%s"%ex,"utf8")) else: self.conn.send(bytes("请输入文件夹名！","utf8")) def delete(self): \'\'\' 删除文件 :return: \'\'\' os.chdir(self.home) # 进入用户根目录 try: self.file = self.home+\'\\\\\'+ self.cmd_file if os.path.isfile(self.file): os.remove(self.cmd_file) self.conn.send(bytes("文件:%s删除成功！"%self.cmd_file,"utf8")) else: os.removedirs(self.cmd_file) self.conn.send(bytes("目录删除成功！","utf8")) os.chdir(self.root) except Exception: if os.path.isdir(self.root): self.conn.send(bytes("删除失败！","utf8")) else: os.makedirs(self.root) self.conn.send(bytes("删除失败！","utf8")) def help(self): \'\'\' FTP帮助信息 :return: \'\'\' self.conn.send(bytes(""" FTP服务器操作方法有： put------>上传文件至服务器 get------>从服务器上下载文件 dir------>查看服务器文件列表 cd------->进入指定文件夹 delete--->删除文件 mkd ----->创建目录 help----->帮助信息 q ------->退出 ""","utf8")) def run(self): while True: # try: # # 接收客户端发来的命令信息 self.cmd = self.conn.recv(1000) self.cmd_action = str(self.cmd.decode()) # 判断命令是否含有空格 self.fg = re.search("\\s","%s"%self.cmd_action) if self.fg: self.cmd_action,self.cmd_file = str(self.cmd_action).split(" ") else: self.cmd_file =None # print("cmd_action:",self.cmd_action,"cmd_file:",self.cmd_file) if hasattr(FTPs,self.cmd_action): func = getattr(self,self.cmd_action) func() continue else: self.conn.send(b\'command is not found!\') continue # except Exception as ex: # print("系统异常:%s"%ex)
客户端
client.py
#!/usr/bin/env python3.5 # -*-coding:utf8-*- import sys,os,re import socket,hashlib BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASEDIR) from core import file_handler from conf import setting def login(): hash = hashlib.sha512() while True: user_input = input("请输入用户名：").strip() pass_input = input("请输入密码：").strip() if len(user_input) !=0 and len(pass_input) != 0: hash.update(bytes(pass_input,"utf8")) sha_pwd = hash.hexdigest() user = "%s|%s"% (user_input,sha_pwd) return user break def ftp_client(): sk = socket.socket() sk.connect((setting.IP_PORT["host"],setting.IP_PORT["port"])) while True: flage = False sk.send(bytes("connect","utf8")) msg = sk.recv(100) print("欢迎访问FTP服务器,请根据提示进行操作") if msg.decode() == "login": while flage == False: login_user =login() username,password = str(login_user).split("|") sk.send(bytes(login_user,"utf8")) user_info = sk.recv(1000) if user_info.decode() == "login_ack": print("登陆成功！") flage = True break print(user_info.decode()) while flage: cmd_action = input("请输入操作命令如：get fy.py or help ：").strip() if len(cmd_action) == 0:continue if cmd_action == "q": sys.exit() # 判断命令是否含有空格 fg = re.search("\\s","%s"%cmd_action) if fg: cmd,cmd_file = str(cmd_action).split(" ") ftp = file_handler.ftpc(sk,username,cmd_action,setting.DATABASE["local"]) if hasattr(ftp,cmd): func = getattr(ftp,cmd) func() continue else: cmd_file =None sk.send(bytes(cmd_action,"utf8")) rec_msg = sk.recv(8000) print(rec_msg.decode()) if flage == "False": sk.send(bytes("connect","utf8")) sk.close() def run(): ftp_client() if __name__ == "__main__": run()
file_handler.py
#!/usr/bin/env python3.5 # -*-coding:utf8-*- import sys,os,re import socket BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASEDIR) class ftpc(object): def __init__(self,sk,username,cmd_action,home): self.sk = sk self.username = username self.cmd_action = cmd_action self.home = home def put(self): \'\'\' 上传文件 :return: \'\'\' try: os.chdir(self.home) except: os.makedirs(self.home) os.chdir(self.home) # 判断命令是否含有空格 fg = re.search("\\s","%s"%self.cmd_action) if fg: self.cmd,self.cmd_file = str(self.cmd_action).split(" ") if os.path.isfile(os.getcwd()+"\\\\"+self.cmd_file): self.sk.send(bytes(self.cmd_action,"utf8")) # 发送动作命令 rec_msg = self.sk.recv(8000) if rec_msg.decode() == "f_ack": f = open(self.cmd_file, \'rb\') self.fsize = os.path.getsize(self.cmd_file) # 获取要发送文件的大小 self.sk.send(bytes("fsize|%s"%self.fsize,"utf8")) # 发送文件大小 self.ack = self.sk.recv(1000) if self.ack.decode() =="f_ack_ready": self.fsize = int(self.fsize) self.size = 0 ack ="" while self.size < self.fsize and ack !="ok":以上是关于基于多线程多用户的FTP服务器与客户端功能实现的主要内容，如果未能解决你的问题，请参考以下文章