From a6842f26799e1d3c527ba5d94df9aed033c65ec9 Mon Sep 17 00:00:00 2001
From: midoks <midoks@163.com>
Date: Wed, 18 Jan 2023 17:42:56 +0800
Subject: [PATCH] Update firewall_api.py

---
 class/core/firewall_api.py | 15 ++-------------
 1 file changed, 2 insertions(+), 13 deletions(-)

diff --git a/class/core/firewall_api.py b/class/core/firewall_api.py
index 69dbc2664..ec650207e 100755
--- a/class/core/firewall_api.py
+++ b/class/core/firewall_api.py
@@ -255,26 +255,15 @@ class firewall_api:
         conf = re.sub(rep, "Port " + port + "\n", conf)
         mw.writeFile(file, conf)
 
+        self.addAcceptPortArgs(port, 'SSH端口修改', 'port')
         if self.__isUfw:
-            mw.execShell('ufw allow ' + port + '/tcp')
             mw.execShell("service ssh restart")
         elif self.__isIptables:
-            mw.execShell(
-                'iptables -I INPUT -p tcp -m state --state NEW -m tcp --dport ' + port + ' -j ACCEPT')
             mw.execShell("/etc/init.d/sshd restart")
         elif self.__isFirewalld:
-            mw.execShell('setenforce 0')
-            mw.execShell(
-                'sed -i "s#SELINUX=enforcing#SELINUX=disabled#" /etc/selinux/config')
             mw.execShell("systemctl restart sshd.service")
         else:
-            pass
-
-        self.firewallReload()
-        # mw.M('firewall').where(
-        #     "ps=?", ('SSH远程管理服务',)).setField('port', port)
-        msg = "改SSH端口为[{}]成功!".format(port)
-        mw.writeLog("防火墙管理", msg)
+            return mw.returnJson(False, '修改失败!')
         return mw.returnJson(True, '修改成功!')
 
     def setSshStatusApi(self):