Update firewall.py

midoks · Dec 24, 2024 · 0a1251a · 0a1251a
1 parent 26147ac
commit 0a1251a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/web/utils/firewall.py b/web/utils/firewall.py
@@ -132,6 +132,17 @@ def reload(self):
         else:
             pass
 
+    def reloadSshd(self):
+        if self.__isUfw:
+            mw.execShell("service ssh restart")
+        elif self.__isIptables:
+            mw.execShell("/etc/init.d/sshd restart")
+        elif self.__isFirewalld:
+            mw.execShell("systemctl restart sshd.service")
+        else:
+            return False
+        return True
+
     def getFwStatus(self):
         if self.__isUfw:
             cmd = "/usr/sbin/ufw status| grep Status | awk -F ':' '{print $2}'"
@@ -251,6 +262,9 @@ def setSshPort(self, port):
         conf = re.sub(rep, "Port " + port + "\n", conf)
         mw.writeFile(file, conf)
 
+        if not self.reloadSshd():
+            return mw.returnData(False, '重启sshd失败,尝试手动重启:service ssh restart!')
+
         self.addAcceptPort(port, 'SSH端口修改', 'port')
         self.reload()
         return mw.returnData(True, '修改成功!')