From 051f992e91e47ce45d34285e60981580c794c974 Mon Sep 17 00:00:00 2001 From: midoks Date: Tue, 26 Sep 2023 20:09:41 +0800 Subject: [PATCH] Update firewall_api.py --- class/core/firewall_api.py | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) diff --git a/class/core/firewall_api.py b/class/core/firewall_api.py index 60c98684d..6eb436a90 100755 --- a/class/core/firewall_api.py +++ b/class/core/firewall_api.py @@ -409,11 +409,26 @@ class firewall_api: def addAcceptPort(self, port, protocol='tcp'): if self.__isUfw: - mw.execShell('ufw allow ' + port + '/tcp') + if protocol == 'tcp': + mw.execShell('ufw allow ' + port + '/tcp') + if protocol == 'udp': + mw.execShell('ufw allow ' + port + '/udp') + if protocol == 'tcp/udp': + mw.execShell('ufw allow ' + port + '/tcp') + mw.execShell('ufw allow ' + port + '/udp') elif self.__isFirewalld: port = port.replace(':', '-') - cmd = 'firewall-cmd --permanent --zone=public --add-port=' + port + '/tcp' - mw.execShell(cmd) + if protocol == 'tcp': + cmd = 'firewall-cmd --permanent --zone=public --add-port=' + port + '/tcp' + mw.execShell(cmd) + if protocol == 'udp': + cmd = 'firewall-cmd --permanent --zone=public --add-port=' + port + '/udp' + mw.execShell(cmd) + if protocol == 'tcp/udp': + cmd = 'firewall-cmd --permanent --zone=public --add-port=' + port + '/tcp' + mw.execShell(cmd) + cmd = 'firewall-cmd --permanent --zone=public --add-port=' + port + '/udp' + mw.execShell(cmd) elif self.__isIptables: cmd = 'iptables -I INPUT -p tcp -m state --state NEW -m tcp --dport ' + port + ' -j ACCEPT' mw.execShell(cmd)