From 4f4329d0a9ffb438a2764b33d888c4e957cb663d Mon Sep 17 00:00:00 2001
From: Mr Chen <midoks@163.com>
Date: Thu, 21 Feb 2019 15:26:14 +0800
Subject: [PATCH] update

---
 class/core/firewall_api.py   | 24 +++++++++++++++++++++++-
 route/static/app/firewall.js |  2 +-
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/class/core/firewall_api.py b/class/core/firewall_api.py
index f9e459eb9..a491b672e 100755
--- a/class/core/firewall_api.py
+++ b/class/core/firewall_api.py
@@ -215,9 +215,31 @@ class firewall_api:
         public.writeLog("防火墙管理", msg)
         return public.returnJson(True, '修改成功!')
 
+    def setSshStatusApi(self):
+        if public.isAppleSystem():
+            return public.returnJson(True, '开发机不能操作!')
+
+        version = public.readFile('/etc/redhat-release')
+        if int(get['status']) == 1:
+            msg = 'SSH服务已停用'
+            act = 'stop'
+        else:
+            msg = 'SSH服务已启用'
+            act = 'start'
+
+        if not os.path.exists('/etc/redhat-release'):
+            public.execShell('service ssh ' + act)
+        elif version.find(' 7.') != -1:
+            public.execShell("systemctl " + act + " sshd.service")
+        else:
+            public.execShell("/etc/init.d/sshd " + act)
+
+        public.writeLog("防火墙管理", msg)
+        return public.returnJson(True, '操作成功!')
+
     def setPingApi(self):
         if public.isAppleSystem():
-            return public.returnJson(True, '开发机不能设置!')
+            return public.returnJson(True, '开发机不能操作!')
 
         status = request.form.get('status')
         filename = '/etc/sysctl.conf'
diff --git a/route/static/app/firewall.js b/route/static/app/firewall.js
index 3e25197e8..1d8f4263e 100755
--- a/route/static/app/firewall.js
+++ b/route/static/app/firewall.js
@@ -194,7 +194,7 @@ function setMstscStatus(){
 	}},function(index){
 		if(index > 0){
 			layer.msg('正在处理,请稍候...',{icon:16,time:20000});
-			$.post('/firewall?action=SetSshStatus','status='+status,function(rdata){
+			$.post('/firewall/set_ssh_status','status='+status,function(rdata){
 				layer.closeAll();
 				layer.msg(rdata.msg,{icon:rdata.status?1:2});
 				refresh();