333 lines
16 KiB
Python
333 lines
16 KiB
Python
#coding: utf-8
|
|
# + -------------------------------------------------------------------
|
|
# | YakPanel
|
|
# + -------------------------------------------------------------------
|
|
# | Copyright (c) 2015-2016 YakPanel(www.yakpanel.com) All rights reserved.
|
|
# + -------------------------------------------------------------------
|
|
# | Author: hwliang <hwl@yakpanel.com>
|
|
# + -------------------------------------------------------------------
|
|
import public,db,re,os,firewalls
|
|
import firewalls_v2 as firewalls
|
|
from public.validate import Param
|
|
try:
|
|
from YakPanel import session
|
|
except: pass
|
|
class ftp:
|
|
__runPath = None
|
|
|
|
def __init__(self):
|
|
self.__runPath = '/www/server/pure-ftpd/bin'
|
|
|
|
|
|
#添加FTP
|
|
def AddUser(self,get):
|
|
# 校验参数
|
|
try:
|
|
get.validate([
|
|
Param('ftp_username').String(),
|
|
Param('ftp_password').String(),
|
|
Param('path').String(),
|
|
Param('ps').String(),
|
|
|
|
], [
|
|
public.validate.trim_filter(),
|
|
])
|
|
except Exception as ex:
|
|
public.print_log("error info: {}".format(ex))
|
|
return public.return_message(-1, 0, str(ex))
|
|
try:
|
|
if not os.path.exists('/www/server/pure-ftpd/sbin/pure-ftpd'):
|
|
return_message=public.return_msg_gettext(False,public.lang('Please install the Pure-FTPd service in the software store first.'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
import files_v2,time
|
|
fileObj=files_v2.files()
|
|
if get['ftp_username'].strip().find(' ') != -1:
|
|
return_message=public.returnMsg(False,public.lang('Username cannot contain spaces'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
if re.search(r"\W+",get['ftp_username']):
|
|
return_message={'code':501,'msg':public.get_msg_gettext('Username is illegal, special characters are NOT allowed!')}
|
|
return public.return_message(-1,0, return_message)
|
|
if len(get['ftp_username']) < 3:
|
|
return_message={'code':501,'msg':public.get_msg_gettext('Username is illegal, cannot be less than 3 characters!')}
|
|
return public.return_message(-1,0, return_message)
|
|
if not fileObj.CheckDir(get['path']):
|
|
return_message={'code':501,'msg':public.get_msg_gettext('System critical directory cannot be used as FTP directory!')}
|
|
return public.return_message(-1,0, return_message)
|
|
if public.M('ftps').where('name=?',(get.ftp_username.strip(),)).count():
|
|
return public.return_message(-1, 0, public.lang("User [{}] exists!", get.ftp_username))
|
|
username = get['ftp_username'].strip()
|
|
if re.search("[\\/\\\\:\\*\\?\"\'\\<\\>\\|]+",username):
|
|
return_message=public.return_msg_gettext(False,public.lang("Name cannot contain /\\:*?\"<>| symbol"))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
password = get['ftp_password'].strip()
|
|
if len(password) < 6:
|
|
return_message=public.return_msg_gettext(False, public.lang('Password must be at least [{}] characters',("6",)))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
get.path = get['path'].replace(' ','')
|
|
get.path = get.path.replace("\\", "/")
|
|
fileObj.CreateDir(get)
|
|
public.ExecShell('chown www.www ' + get.path)
|
|
public.ExecShell(self.__runPath + '/pure-pw useradd "' + username + '" -u www -d ' + get.path + '<<EOF \n' + password + '\n' + password + '\nEOF')
|
|
self.FtpReload()
|
|
ps = public.xssencode2(get['ps'])
|
|
if get['ps']=='': ps= public.lang('Edit notes');
|
|
addtime=time.strftime('%Y-%m-%d %X',time.localtime())
|
|
|
|
pid = 0
|
|
if hasattr(get,'pid'): pid = get.pid
|
|
public.M('ftps').add('pid,name,password,path,status,ps,addtime',(pid,username,password,get.path,1,ps,addtime))
|
|
public.write_log_gettext('FTP manager', 'Successfully added FTP user [{}]!',(username,))
|
|
return_message=public.return_msg_gettext(True,public.lang('Setup successfully!'))
|
|
del return_message['status']
|
|
return public.return_message(0,0, return_message['msg'])
|
|
except Exception as ex:
|
|
public.write_log_gettext('FTP manager', 'Failed to add FTP user[{}]! => {}',(username,str(ex)))
|
|
return_message=public.return_msg_gettext(False,public.lang('Failed to add'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
|
|
#删除用户
|
|
def DeleteUser(self,get):
|
|
# 校验参数
|
|
try:
|
|
get.validate([
|
|
Param('username').String(),
|
|
Param('id').Integer(),
|
|
], [
|
|
public.validate.trim_filter(),
|
|
])
|
|
except Exception as ex:
|
|
public.print_log("error info: {}".format(ex))
|
|
return public.return_message(-1, 0, str(ex))
|
|
try:
|
|
username = get['username']
|
|
id = get['id']
|
|
if public.M('ftps').where("id=? and name=?", (id,username, )).count()==0:
|
|
return_message=public.return_msg_gettext(False, public.lang('Delete error'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
public.ExecShell(self.__runPath + '/pure-pw userdel "' + username + '"')
|
|
self.FtpReload()
|
|
public.M('ftps').where("id=?",(id,)).delete()
|
|
public.write_log_gettext('FTP manager', 'Successfully deleted FTP user[{}]!',(username,))
|
|
return_message=public.return_msg_gettext(True, public.lang('Successfully deleted'))
|
|
del return_message['status']
|
|
return public.return_message(0,0, return_message['msg'])
|
|
except Exception as ex:
|
|
public.write_log_gettext('FTP manager', 'Faided to delete FTP user[{}]! => {}',(username,str(ex)))
|
|
return_message=public.return_msg_gettext(False,public.lang('Failed to delete'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
|
|
|
|
#修改用户密码
|
|
def SetUserPassword(self,get):
|
|
# 校验参数
|
|
try:
|
|
get.validate([
|
|
Param('ftp_username').String(),
|
|
Param('new_password').String(),
|
|
Param('id').Integer(),
|
|
|
|
], [
|
|
public.validate.trim_filter(),
|
|
])
|
|
except Exception as ex:
|
|
public.print_log("error info: {}".format(ex))
|
|
return public.return_message(-1, 0, str(ex))
|
|
try:
|
|
id = get['id']
|
|
username = get['ftp_username'].strip()
|
|
password = get['new_password'].strip()
|
|
if public.M('ftps').where("id=? and name=?", (id,username, )).count()==0:
|
|
return_message=public.return_msg_gettext(False, public.lang('DEL_ERROR'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
if len(password) < 6:
|
|
return_message=public.return_msg_gettext(False,public.lang('Password must be at least [{}] characters',"6"))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
public.ExecShell(self.__runPath + '/pure-pw passwd "' + username + '"<<EOF \n' + password + '\n' + password + '\nEOF')
|
|
self.FtpReload()
|
|
public.M('ftps').where("id=?",(id,)).setField('password',password)
|
|
public.write_log_gettext('FTP manager', 'Successfully changed password for FTP user[{}]!',(username,))
|
|
return_message=public.return_msg_gettext(True,public.lang('Setup successfully!'))
|
|
del return_message['status']
|
|
return public.return_message(0,0, return_message['msg'])
|
|
except Exception as ex:
|
|
public.write_log_gettext('FTP manager', 'Failed to change password FTP user[{}]! => {}',(username,str(ex)))
|
|
return_message=public.return_msg_gettext(False,public.lang('Failed to modify'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
|
|
|
|
#设置用户状态
|
|
def SetStatus(self,get):
|
|
msg = public.get_msg_gettext('Turn off');
|
|
if get.status != '0': msg = public.get_msg_gettext('Turn on');
|
|
try:
|
|
id = get['id']
|
|
username = get['username']
|
|
status = get['status']
|
|
if public.M('ftps').where("id=? and name=?", (id,username, )).count()==0:
|
|
return_message=public.return_msg_gettext(False, public.lang('DEL_ERROR'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
if int(status)==0:
|
|
public.ExecShell(self.__runPath + '/pure-pw usermod "' + username + '" -r 1')
|
|
else:
|
|
public.ExecShell(self.__runPath + '/pure-pw usermod "' + username + "\" -r ''")
|
|
self.FtpReload()
|
|
public.M('ftps').where("id=?",(id,)).setField('status',status)
|
|
public.write_log_gettext('FTP manager','Successfully {} FTP user [{}]!', (msg,username))
|
|
return_message=public.return_msg_gettext(True, public.lang('Setup successfully!'))
|
|
del return_message['status']
|
|
return public.return_message(0,0, return_message['msg'])
|
|
except Exception as ex:
|
|
public.write_log_gettext('FTP manager','Failed to {} FTP user [{}]! => {}', (msg,username,str(ex)))
|
|
return_message=public.return_msg_gettext(False,public.lang('{} FTP user failed!',msg))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
|
|
'''
|
|
* 设置FTP端口
|
|
* @param Int _GET['port'] 端口号
|
|
* @return bool
|
|
'''
|
|
def setPort(self,get):
|
|
# 校验参数
|
|
try:
|
|
get.validate([
|
|
Param('port').Integer(),
|
|
], [
|
|
public.validate.trim_filter(),
|
|
])
|
|
except Exception as ex:
|
|
public.print_log("error info: {}".format(ex))
|
|
return public.return_message(-1, 0, str(ex))
|
|
try:
|
|
port = get['port'].strip()
|
|
if not port:
|
|
return public.return_message(-1, 0, public.lang("Please enter an integer for the port"))
|
|
if int(port) < 1 or int(port) > 65535:
|
|
return public.return_message(-1, 0, public.lang("Port range is incorrect!"))
|
|
check_used = public.check_port_stat(int(port),public.GetLocalIp())
|
|
if check_used == 2:
|
|
return public.return_message(-1, 0, public.lang("Port[{}] is used!", str(port)))
|
|
data = public.ExecShell('lsof -i:' + str(port))[0]
|
|
if len(data) !=0:
|
|
return public.return_message(-1, 0, public.lang("Port[{}] is used!", str(port)))
|
|
file = '/www/server/pure-ftpd/etc/pure-ftpd.conf'
|
|
conf = public.readFile(file)
|
|
rep = u"\n#?\\s*Bind\\s+[0-9]+\\.[0-9]+\\.[0-9]+\\.+[0-9]+,([0-9]+)"
|
|
#preg_match(rep,conf,tmp)
|
|
conf = re.sub(rep,"\nBind 0.0.0.0," + port,conf)
|
|
public.writeFile(file,conf)
|
|
public.ExecShell('/etc/init.d/pure-ftpd restart')
|
|
public.write_log_gettext('FTP manager', "Successfully modified FTP port to [{}]!",(port,))
|
|
#添加防火墙
|
|
#data = ftpinfo(port=port,ps = 'FTP端口')
|
|
get.port=port
|
|
get.ps = public.get_msg_gettext('FTP port');
|
|
firewalls.firewalls().AddAcceptPort(get)
|
|
session['port']=port
|
|
return public.return_message(0, 0, public.lang("Setup successfully!"))
|
|
except Exception as ex:
|
|
public.write_log_gettext('FTP manager', 'Failed to modify FTP port! => {}',(str(ex),))
|
|
return public.return_message(-1, 0, public.lang("Failed to modify"))
|
|
|
|
#重载配置
|
|
def FtpReload(self):
|
|
public.ExecShell(self.__runPath + '/pure-pw mkdb /www/server/pure-ftpd/etc/pureftpd.pdb')
|
|
|
|
def get_login_logs(self, get):
|
|
import ftp_log_v2 as ftplog
|
|
ftpobj = ftplog.ftplog()
|
|
return ftpobj.get_login_log(get)
|
|
def get_action_logs(self, get):
|
|
import ftp_log_v2 as ftplog
|
|
ftpobj = ftplog.ftplog()
|
|
return ftpobj.get_action_log(get)
|
|
|
|
def set_ftp_logs(self, get):
|
|
import ftp_log_v2 as ftplog
|
|
ftpobj = ftplog.ftplog()
|
|
result = ftpobj.set_ftp_log(get)
|
|
return result
|
|
|
|
#修改用户密码
|
|
def set_user_home(self,get):
|
|
"""
|
|
change user home
|
|
id: ftp id
|
|
path: the new ftp user home
|
|
ftp_username: ftp username
|
|
migrate: migrate ftp user data to the new home
|
|
|
|
"""
|
|
# 校验参数
|
|
try:
|
|
get.validate([
|
|
Param('ftp_username').String(),
|
|
Param('path').String(),
|
|
Param('id').Integer(),
|
|
Param('migrate').Integer(),
|
|
|
|
], [
|
|
public.validate.trim_filter(),
|
|
])
|
|
except Exception as ex:
|
|
public.print_log("error info: {}".format(ex))
|
|
return public.return_message(-1, 0, str(ex))
|
|
try:
|
|
id = get['id']
|
|
path = get['path']
|
|
username = get['ftp_username']
|
|
# get the old path in the panel sqlite db
|
|
old_path = public.M("ftps").where("id=?",(id,)).getField('path')
|
|
# check the auth ftp user if exists
|
|
auth_conf_file = '/www/server/pure-ftpd/etc/pureftpd.passwd'
|
|
auth_conf = public.readFile(auth_conf_file)
|
|
if not auth_conf:
|
|
return_message=public.returnMsg(False, public.lang('FTP account has not been set up'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
# get the user specified conf
|
|
auth_conf_list = [i for i in auth_conf.split('\n')]
|
|
rep = '^{}:.*'.format(username)
|
|
macth_conf = [i for i in auth_conf_list if re.search(rep,i)]
|
|
if not macth_conf:
|
|
return_message=public.returnMsg(False, public.lang('FTP account has not been set up1'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
if len(macth_conf) > 1:
|
|
return_message=public.returnMsg(False, public.lang('Matching multiple configurations, this operation has been stopped!'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|
|
if not os.path.exists(path):
|
|
os.makedirs(path)
|
|
public.ExecShell('chown www.www ' + path)
|
|
# replace the old path
|
|
result = macth_conf[0]
|
|
specified_user_conf = result.replace(old_path,path)
|
|
auth_conf = auth_conf.replace(result,specified_user_conf)
|
|
public.writeFile(auth_conf_file,auth_conf)
|
|
if get.migrate == '1':
|
|
public.ExecShell('cp -rp {}/* {}'.format(old_path,path))
|
|
self.FtpReload()
|
|
public.M('ftps').where("id=?",(id,)).setField('path',path)
|
|
public.write_log_gettext('FTP manager', 'Successfully changed password for FTP user[{}]!',(path,))
|
|
return_message=public.return_msg_gettext(True,public.lang('Setup successfully!'))
|
|
del return_message['status']
|
|
return public.return_message(0,0, return_message['msg'])
|
|
except Exception as ex:
|
|
return public.get_error_info()
|
|
public.write_log_gettext('FTP manager', 'FTP_PASS_ERR',(path,str(ex)))
|
|
return_message=public.returnMsg(False, public.lang('Editing error'))
|
|
del return_message['status']
|
|
return public.return_message(-1,0, return_message['msg'])
|