Python实现基于多线程、多用户的FTP服务器与客户端功能完整实例
发布时间：2020-05-30 10:36:19 所属栏目：Python 来源：互联网
导读：本文实例讲述了Python实现基于多线程、多用户的FTP服务器与客户端功能。分享给大家供大家参考，具体如下：
|
本文实例讲述了Python实现基于多线程、多用户的FTP服务器与客户端功能。分享给大家供大家参考，具体如下： 项目介绍： 1. 用户加密认证 实现的原理： 服务器端启用端口监听，并对每一连接启用一个线程，对用户登陆密码采用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,size)
# print("user_input:",username,"user_pass:",password)
if flage =="True":
# 登陆成功发送登陆确认信息给客户端
conn.send(bytes("login_ack","utf8"))
# 实例化FTPserver
ftp = file_handler.FTPs(username,conn,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,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- 占用大小：%skbrn'% (self.i,self.fsize)
else:
self.dl += '%srn'%self.i
self.conn.send(bytes("目录：rnrn%s 文件：rn%srn rn当前目录空间大小：%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)
客户端 （编辑：安卓应用网） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |