FENG宝塔FRP映射插件
资源内容介绍
映射插件 #!/usr/bin/python# coding: utf-8# Author: Xetchimport sys, os, json, ssl, time;pluginPath = "/www/server/panel/plugin/btp_frps";frpsPath = pluginPath + '/bin/frps';frpsIniPath = pluginPath + '/conf/frps.ini';os.chdir("/www/server/panel");sys.path.append("class/");import public, re;from BTPanel import cache;class btp_frps_main():def check(self, get):if not os.path.isfile(frpsPath):return public.returnMsg(False, 'frps 未安装');success, failed = public.ExecShell(frpsPath + ' --version');if success.strip() == '':return public.returnMsg(False, 'frps 未安装或无执行权限');return public.returnMsg(True, {'version': success.strip(),'pid': self.__pid()});def install(self, get):taskName = 'frps';if self.__getTaskStatus(taskName) != 1:return public.returnMsg(False, '安装任务已在队列中');release = self.release();if release['result'] != 'success':return public.returnMsg(False, release['result']);cmd = "cd %s && /bin/bash install.sh download \"%s\"" % (pluginPath, release['url']);public.M('tasks').add('id, name, type, status, addtime, execstr', (None, '安装 [' + taskName + '-' + release['version'] + ']', 'execshell', '0', time.strftime('%Y-%m-%d %H:%M:%S'), cmd))cache.delete('install_task')public.writeFile('/tmp/panelTask.pl', 'True')public.WriteLog('TYPE_SETUP', 'PLUGIN_ADD', (taskName, release['version']));return public.returnMsg(True, '已将安装任务添加到队列');def upload(self, get):filename = pluginPath + '/temp/import.tar.gz';try:from flask import request;file = request.files['import'];if file.filename[-7:] == '.tar.gz' or file.filename[-4:] == '.zip':if file.filename[-4:] == '.zip':filename = pluginPath + '/temp/import.zip';os.system('rm -rf ' + pluginPath + '/temp/import');os.system('mkdir -p ' + pluginPath + '/temp/import');os.system('mkdir -p ' + pluginPath + '/bin');os.system('rm -rf ' + filename);file.save(filename);import panelTask;panelTask.bt_task()._unzip(filename, pluginPath + '/temp/import', '', '/dev/null');success, failed = public.ExecShell('find %s/temp/import -name frps' % pluginPath);if success.strip() != '':os.system('mv -f %s %s' % (success.strip(), frpsPath));os.system('chown root:root ' + frpsPath);os.system('chmod +x ' + frpsPath);os.system('rm -rf ' + pluginPath + '/temp/import');os.system('rm -rf ' + filename);if os.path.isfile(frpsPath):return self.check(get);except:pass;return public.returnMsg(False, '无效文件或未找到 frps');def upgrade(self, get):release = self.release();if release['result'] != 'success':return public.returnMsg(False, release['result']);return public.returnMsg(True, release);def save(self, get):try:data = json.loads(get['json']);os.system('mkdir -p ' + pluginPath + '/conf');public.WriteFile(pluginPath + '/conf/config.json', get['json'], mode='w+');keys = {"bindAddr": "bind_addr","bindPort": "bind_port","bindUdpPort": "bind_udp_port","kcpBindPort": "kcp_bind_port","proxyBindAddr": "proxy_bind_addr","vhostHttpPort": "vhost_http_port","vhostHttpsPort": "vhost_https_port","vhostHttpTimeout": "vhost_http_timeout","dashboardAddr": "dashboard_addr","dashboardPort": "dashboard_port","dashboardUser": "dashboard_user","dashboardPwd": "dashboard_pwd",# "assetsDir": "assets_dir",# "logFile": "log_file","logLevel": "log_level","logMaxDays": "log_max_days","token": "token","heartbeatTimeout": "heartbeat_timeout",# "allowPorts": "allow_ports","maxPoolCount": "max_pool_count","maxPortsPerClient": "max_ports_per_client","subdomainHost": "subdomain_host","tcpMux": "tcp_mux","disableLogColor": "disable_log_color",# "custom404Page": "custom_404_page"};config = '[common]\n';for key in keys.keys():if type(data[key]) == bool:if data[key] == True:data[key] = 'true';else:data[key] = 'false';if str(data[key]) != '':config += '%s = %s\n' % (keys[key], str(data[key]));if type(data['allowPorts']) == list and len(data['allowPorts']) > 0:config += 'allow_ports = %s\n' % ','.join(data['allowPorts']);config += 'log_file = %s/temp/frps.log\n' % pluginPath;filename = pluginPath + '/conf/404.html';if not os.path.isfile(filename):public.WriteFile(filename, '', mode='w+');if data['enabledCustom404Page'] == True and os.path.getsize(filename) > 0:config += 'custom_404_page = %s\n' % filename;public.WriteFile(frpsIniPath, config, mode='w+');return public.returnMsg(True, '保存成功');except ValueError:return public.returnMsg(False, '请求错误，请刷新页面重试');def read(self, get):filename = pluginPath + '/conf/config.json';if os.path.isfile(filename):try:config = json.loads(public.ReadFile(filename, mode='r'));config['custom404Page'] = pluginPath + '/conf/404.html';return public.returnMsg(True, config);except ValueError:pass;return public.returnMsg(False, '配置文件损坏或不存在');def release(self):'''不知道为什么在这里用不了 public.httpGet'''#result = public.httpGet('https://api.github.com/repos/fatedier/frp/releases/latest'); # 获取不到？#result = public.ExecShell('curl https://api.github.com/repos/fatedier/frp/releases/latest');#data = json.loads(result);#return resp.read().decode("utf8");filename = pluginPath + '/temp/release.json';result = 'success';version = '';url = '';remark = '';os.system('mkdir -p ' + pluginPath + '/temp');os.system('rm -rf ' + filename);os.system('wget -O ' + filename + ' https://api.github.com/repos/fatedier/frp/releases/latest');if not os.path.isfile(filename):result = '获取版本信息失败';else:try:data = json.loads(public.ReadFile(filename, mode='r'));version = data['tag_name'].strip()[1:]; # 版本号remark = data['body']; # 更新说明for item in data['assets']:if item['name'].find('linux_amd64') != -1:url = item['browser_download_url']; # 下载地址break;except ValueError:result = '文件解析失败，请稍后再试';os.system('rm -rf ' + filename);return {'version': version,'url': url,'remark': remark,'result': result};def start(self, get):pid = self.__pid();if pid != False:return public.returnMsg(True, 'frps 已开启，PID：%s' % pid);service = '''[Unit]Description=Frp Server ServiceAfter=network.target[Service]Type=simpleUser=rootRestart=on-failureRestartSec=5sExecStart=%s -c %s[Install]WantedBy=multi-user.target''' % (frpsPath, frpsIniPath);# 使用 systemd 管理自启动filename = '/etc/systemd/system/btp_frps.service';if os.path.isdir('/etc/systemd/system') and not os.path.isfile(filename):public.WriteFile(filename, service, mode='w+');os.system('chown root:root %s' % filename);os.system('chmod 755 %s' % filename);if os.path.isfile(filename):os.system('systemctl enable btp_frps');os.system('systemctl start btp_frps');else:os.system('nohup %s -c %s &' % (frpsPath, frpsIniPath));time.sleep(1);pid = self.__pid();if pid != False:return public.returnMsg(True, '开启成功，PID：%s' % pid);return public.returnMsg(False, '开启失败');def stop(self, get):pid = self.__pid();if pid == False:return public.returnMsg(True, 'frps 尚未运行');filename = '/etc/systemd/system/btp_frps.service';if os.path.isfile(filename):os.system('systemctl stop btp_frps');os.system('systemctl disable btp_frps');os.system('rm -rf %s' % filename);else:os.system('kill -9 %s' % pid);return public.returnMsg(True, '关闭成功');def restart(self, get):if os.path.isfile('/etc/systemd/