FENG宝塔FRP映射插件

#!/usr/bin/python # coding: utf-8 # Author: Xetch import 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 Service After=network.target [Service] Type=simple User=root Restart=on-failure RestartSec=5s ExecStart=%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/