236 lines
9.0 KiB
Python
236 lines
9.0 KiB
Python
|
#coding: utf-8
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
# YakPanel
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
# Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
# Author: zhwen <zhw@yakpanel.com>
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
|
||
|
|
#------------------------------
|
||
|
|
# 禁止某个目录运行PHP
|
||
|
|
#------------------------------
|
||
|
|
import public,re,os,json,shutil
|
||
|
|
|
||
|
|
class FileExecuteDeny:
|
||
|
|
|
||
|
|
def _init_conf(self,website):
|
||
|
|
self.ng_website_conf = '/www/server/panel/vhost/nginx/{}.conf'.format(website)
|
||
|
|
self.ap_website_conf = '/www/server/panel/vhost/apache/{}.conf'.format(website)
|
||
|
|
self.ols_website_conf = '/www/server/panel/vhost/openlitespeed/detail/{}.conf'.format(website)
|
||
|
|
self.webserver = public.get_webserver()
|
||
|
|
|
||
|
|
# 获取某个网站禁止运行的目录规则
|
||
|
|
def get_file_deny(self,args):
|
||
|
|
'''
|
||
|
|
# 添加某个网站禁止运行PHP
|
||
|
|
author: zhwen<zhw@yakpanel.com>
|
||
|
|
:param args: website 网站名 str
|
||
|
|
:return:
|
||
|
|
'''
|
||
|
|
self._init_conf(args.website)
|
||
|
|
if self.webserver == 'nginx':
|
||
|
|
data=self._get_nginx_file_deny()
|
||
|
|
elif self.webserver == 'apache':
|
||
|
|
data = self._get_apache_file_deny()
|
||
|
|
else:
|
||
|
|
data = self._get_ols_file_deny()
|
||
|
|
return data
|
||
|
|
|
||
|
|
def _get_nginx_file_deny(self):
|
||
|
|
conf = public.readFile(self.ng_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_DENY_.*',conf)
|
||
|
|
deny_name = []
|
||
|
|
for i in data:
|
||
|
|
tmp = i.split('_')
|
||
|
|
if len(tmp) > 2:
|
||
|
|
deny_name.append('_'.join(tmp[2:]))
|
||
|
|
else:
|
||
|
|
deny_name.append(tmp[-1])
|
||
|
|
result = []
|
||
|
|
for i in deny_name:
|
||
|
|
reg = '#BEGIN_DENY_{}\n\\s*location\\s*\\~\\*\\s*\\^(.*)\\.\\*.*\\((.*)\\)\\$'.format(i.replace("|",r"\|"))
|
||
|
|
re_tmp = re.search(reg,conf)
|
||
|
|
if re_tmp:
|
||
|
|
deny_directory = re_tmp.groups()[0]
|
||
|
|
deny_suffix = re_tmp.groups()[1]
|
||
|
|
result.append({'name':i,'dir':deny_directory,'suffix':deny_suffix})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def _get_apache_file_deny(self):
|
||
|
|
conf = public.readFile(self.ap_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_DENY_.*',conf)
|
||
|
|
deny_name = []
|
||
|
|
for i in data:
|
||
|
|
tmp = i.split('_')
|
||
|
|
if len(tmp) > 2:
|
||
|
|
deny_name.append('_'.join(tmp[2:]))
|
||
|
|
else:
|
||
|
|
deny_name.append(tmp[-1])
|
||
|
|
result = []
|
||
|
|
for i in deny_name:
|
||
|
|
reg = '#BEGIN_DENY_{}\n\\s*<Directory\\s*\\~\\s*"(.*)\\.\\*.*\\((.*)\\)\\$'.format(i.replace("|",r"\|"))
|
||
|
|
deny_directory = re.search(reg,conf).groups()[0]
|
||
|
|
deny_suffix = re.search(reg,conf).groups()[1]
|
||
|
|
result.append({'name':i,'dir':deny_directory,'suffix':deny_suffix})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def _get_ols_file_deny(self):
|
||
|
|
conf = public.readFile(self.ols_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_DENY_.*',conf)
|
||
|
|
deny_name = []
|
||
|
|
for i in data:
|
||
|
|
tmp = i.split('_')
|
||
|
|
if len(tmp) > 2:
|
||
|
|
deny_name.append('_'.join(tmp[2:]))
|
||
|
|
else:
|
||
|
|
deny_name.append(tmp[-1])
|
||
|
|
result = []
|
||
|
|
for i in deny_name:
|
||
|
|
reg = '#BEGIN_DENY_{}\n\\s*rules\\s*RewriteRule\\s*\\^(.*)\\.\\*.*\\((.*)\\)\\$'.format(i.replace("|",r"\|"))
|
||
|
|
deny_directory = re.search(reg, conf).groups()[0]
|
||
|
|
deny_suffix = re.search(reg,conf).groups()[1]
|
||
|
|
result.append({'name':i,'dir':deny_directory,'suffix':deny_suffix})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def set_file_deny(self,args):
|
||
|
|
'''
|
||
|
|
# 添加某个网站禁止运行PHP
|
||
|
|
author: zhwen<zhw@yakpanel.com>
|
||
|
|
:param args: website 网站名 str
|
||
|
|
:param args: deny_name 规则名称 str
|
||
|
|
:param args: suffix 禁止访问的后续名 str
|
||
|
|
:param args: dir 禁止访问的目录 str
|
||
|
|
:param args: deny_name 规则名称
|
||
|
|
:param args: act 操作方法
|
||
|
|
:return:
|
||
|
|
'''
|
||
|
|
tmp = self._check_args(args)
|
||
|
|
if tmp:
|
||
|
|
return tmp
|
||
|
|
deny_name = args.deny_name
|
||
|
|
if not re.match(r"^\w+$",deny_name): return public.return_msg_gettext(False, public.lang("The rule name can only be composed of letters, numbers, and underscores!"))
|
||
|
|
dir = args.dir
|
||
|
|
suffix = args.suffix
|
||
|
|
website = args.website
|
||
|
|
if suffix[-1] == "|":
|
||
|
|
suffix = suffix[:-1]
|
||
|
|
self._init_conf(website)
|
||
|
|
conf = public.readFile(self.ng_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_DENY_.*',conf)
|
||
|
|
exist_deny_name = [i.split('_')[-1] for i in data]
|
||
|
|
if args.act == 'edit':
|
||
|
|
if deny_name not in exist_deny_name:
|
||
|
|
return public.return_msg_gettext(False, public.lang("The specify rule name is not exists! [ {} ]", deny_name))
|
||
|
|
self.del_file_deny(args)
|
||
|
|
else:
|
||
|
|
if deny_name in exist_deny_name:
|
||
|
|
return public.return_msg_gettext(False, public.lang("The specify rule name is already exists! [ {} ]", deny_name))
|
||
|
|
self._set_nginx_file_deny(deny_name,dir,suffix)
|
||
|
|
self._set_apache_file_deny(deny_name,dir,suffix)
|
||
|
|
self._set_ols_file_deny(deny_name,dir,suffix)
|
||
|
|
public.serviceReload()
|
||
|
|
return public.returnMsg(True, public.lang("Setup successfully!"))
|
||
|
|
|
||
|
|
def _set_nginx_file_deny(self,name,dir=None,suffix=None):
|
||
|
|
conf = public.readFile(self.ng_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
if not dir and not suffix:
|
||
|
|
reg = '\\s*#BEGIN_DENY_{n}\n(.|\n)*#END_DENY_{n}\n'.format(n=name)
|
||
|
|
conf = re.sub(reg,'',conf)
|
||
|
|
else:
|
||
|
|
if dir[0] != '/':dir = '/'+dir
|
||
|
|
if dir[-1] != '/':dir = dir+'/'
|
||
|
|
new = '''
|
||
|
|
#BEGIN_DENY_%s
|
||
|
|
location ~* ^%s.*.(%s)$ {
|
||
|
|
deny all;
|
||
|
|
}
|
||
|
|
#END_DENY_%s
|
||
|
|
''' % (name,dir,suffix,name)
|
||
|
|
if '#BEGIN_DENY_{}\n'.format(name) in conf:
|
||
|
|
return True
|
||
|
|
conf = re.sub('#ERROR-PAGE-END','#ERROR-PAGE-END'+new,conf)
|
||
|
|
public.writeFile(self.ng_website_conf,conf)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def _set_apache_file_deny(self,name,dir=None,suffix=None):
|
||
|
|
conf = public.readFile(self.ap_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
if not dir and not suffix:
|
||
|
|
reg = '\\s*#BEGIN_DENY_{n}\n(.|\n)*#END_DENY_{n}'.format(n=name)
|
||
|
|
conf = re.sub(reg,'',conf)
|
||
|
|
else:
|
||
|
|
if dir[0] != '/':dir = '/'+dir
|
||
|
|
if dir[-1] != '/':dir = dir+'/'
|
||
|
|
new = r'''
|
||
|
|
#BEGIN_DENY_{n}
|
||
|
|
<Directory ~ "{d}.*\.({s})$">
|
||
|
|
Order allow,deny
|
||
|
|
Deny from all
|
||
|
|
</Directory>
|
||
|
|
#END_DENY_{n}
|
||
|
|
'''.format(n=name,d=dir,s=suffix)
|
||
|
|
if '#BEGIN_DENY_{}'.format(name) in conf:
|
||
|
|
return True
|
||
|
|
conf = re.sub(r'#DENY\s*FILES',new+'\n #DENY FILES',conf)
|
||
|
|
public.writeFile(self.ap_website_conf,conf)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def _set_ols_file_deny(self,name,dir=None,suffix=None):
|
||
|
|
conf = public.readFile(self.ols_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
if not dir and not suffix:
|
||
|
|
reg = '#BEGIN_DENY_{n}\n(.|\n)*#END_DENY_{n}\\s*'.format(n=name)
|
||
|
|
conf = re.sub(reg,'',conf)
|
||
|
|
else:
|
||
|
|
new = r'''
|
||
|
|
#BEGIN_DENY_{n}
|
||
|
|
rules RewriteRule ^{d}.*\.({s})$ - [F,L]
|
||
|
|
#END_DENY_{n}
|
||
|
|
'''.format(n=name,d=dir,s=suffix)
|
||
|
|
if '#BEGIN_DENY_{}'.format(name) in conf:
|
||
|
|
return True
|
||
|
|
conf = re.sub(r'autoLoadHtaccess\s*1','autoLoadHtaccess 1'+new,conf)
|
||
|
|
public.writeFile(self.ols_website_conf,conf)
|
||
|
|
return True
|
||
|
|
|
||
|
|
# 删除某个网站禁止运行PHP
|
||
|
|
def del_file_deny(self,args):
|
||
|
|
'''
|
||
|
|
# 添加某个网站禁止运行PHP
|
||
|
|
author: zhwen<zhw@yakpanel.com>
|
||
|
|
:param args: website 网站名 str
|
||
|
|
:param args: deny_name 规则名称 str
|
||
|
|
:return:
|
||
|
|
'''
|
||
|
|
self._init_conf(args.website)
|
||
|
|
deny_name = args.deny_name
|
||
|
|
self._set_nginx_file_deny(deny_name)
|
||
|
|
self._set_apache_file_deny(deny_name)
|
||
|
|
self._set_ols_file_deny(deny_name)
|
||
|
|
public.serviceReload()
|
||
|
|
return public.returnMsg(True, public.lang("Successfully deleted!"))
|
||
|
|
|
||
|
|
# 检查传入参数
|
||
|
|
def _check_args(self,args):
|
||
|
|
if hasattr(args,'deny_name'):
|
||
|
|
if len(args.deny_name) < 3:
|
||
|
|
return public.return_msg_gettext(False, public.lang("Rule name needs to be greater than 3 bytes"))
|
||
|
|
if hasattr(args,'suffix'):
|
||
|
|
if not args.suffix:
|
||
|
|
return public.return_msg_gettext(False, public.lang("File suffix cannot be empty"))
|
||
|
|
if hasattr(args,'dir'):
|
||
|
|
if not args.dir:
|
||
|
|
return public.return_msg_gettext(False, public.lang("Directory cannot be empty"))
|