Files
CyberPanel/install/firewallUtilities.py
2018-11-14 13:51:32 -05:00

93 lines
2.9 KiB
Python

import sys
import subprocess
import shutil
import installLog as logging
import argparse
import os
import shlex
import socket
class FirewallUtilities:
@staticmethod
def addRule(proto,port):
try:
command = 'sudo firewall-cmd --permanent --zone=public --add-port=' + port + '/' + proto
#if port == "21":
# command = "sudo firewall-cmd --add-service=ftp --permanent"
# cmd = shlex.split(command)
# res = subprocess.call(cmd)
#ipAddress = "0.0.0.0/0"
#ruleFamily = 'rule family="ipv4"'
#sourceAddress = 'source address="' + ipAddress + '"'
#ruleProtocol = 'port protocol="' + proto + '"'
#rulePort = 'port="' + port + '"'
#command = "sudo firewall-cmd --permanent --zone=public --add-rich-rule='" + ruleFamily + " " + sourceAddress + " " + ruleProtocol + " " + rulePort + " " + "accept'"
cmd = shlex.split(command)
res = subprocess.call(cmd)
command = 'sudo firewall-cmd --reload'
cmd = shlex.split(command)
res = subprocess.call(cmd)
import install as inst
if inst.preFlightsChecks.resFailed(inst.get_distro(), res):
inst.preFlightsChecks.stdOut("Failed to install rule: " + command + " Error #" + str(res), 1)
return 0
except OSError, msg:
logging.InstallLog.writeToFile(str(msg) + " [addRule]")
return 0
except ValueError, msg:
logging.InstallLog.writeToFile(str(msg) + " [addRule]")
return 0
return 1
@staticmethod
def deleteRule(proto, port):
try:
command = 'sudo firewall-cmd --permanent --zone-public --remove-port=' + port + '/' + proto
#if port=="21":
# command = "sudo firewall-cmd --remove-service=ftp --permanent"
# cmd = shlex.split(command)
# res = subprocess.call(cmd)
#ipAddress = "0.0.0.0/0"
#ruleFamily = 'rule family="ipv4"'
#sourceAddress = 'source address="' + ipAddress + '"'
#ruleProtocol = 'port protocol="' + proto + '"'
#rulePort = 'port="' + port + '"'
#command = "sudo firewall-cmd --permanent --zone=public --remove-rich-rule='" + ruleFamily + " " + sourceAddress + " " + ruleProtocol + " " + rulePort + " " + "accept'"
cmd = shlex.split(command)
res = subprocess.call(cmd)
command = 'sudo firewall-cmd --reload'
cmd = shlex.split(command)
res = subprocess.call(cmd)
except OSError, msg:
logging.InstallLog.writeToFile(str(msg) + " [deleteRule]")
return 0
except ValueError, msg:
logging.InstallLog.writeToFile(str(msg) + " [deleteRule]")
return 0
return 1