diff --git a/firewall/firewall.py b/firewall/firewall.py index c65fe57c..38782bde 100644 --- a/firewall/firewall.py +++ b/firewall/firewall.py @@ -33,9 +33,10 @@ class Firewall: banList = dict() - def __init__(self, banTime, logSys): + def __init__(self, banTime, logSys, interface): self.banTime = banTime self.logSys = logSys + self.interface = interface def addBanIP(self, ip, debug): """ Bans an IP. diff --git a/firewall/iptables.py b/firewall/iptables.py index 9983c27a..7b78c69c 100644 --- a/firewall/iptables.py +++ b/firewall/iptables.py @@ -38,11 +38,11 @@ class Iptables(Firewall): def banIP(self, ip): """ Returns query to ban IP. """ - query = "iptables -I INPUT 1 -i eth0 -s "+ip+" -j DROP" + query = "iptables -I INPUT 1 -i "+self.interface+" -s "+ip+" -j DROP" return query def unBanIP(self, ip): """ Returns query to unban IP. """ - query = "iptables -D INPUT -i eth0 -s "+ip+" -j DROP" + query = "iptables -D INPUT -i "+self.interface+" -s "+ip+" -j DROP" return query