github
/
fail2ban
mirror of https://github.com/fail2ban/fail2ban
1
0
Fork 0
Code Issues Releases Wiki Activity

- Add log4py support 

- Remove old verbose mode
- Add debug feature


git-svn-id: https://fail2ban.svn.sourceforge.net/svnroot/fail2ban/trunk@17 a942ae1a-1317-0410-a47c-b1dcaea8d605
0.6
Cyril Jaquier 2004-10-11 10:22:41 +00:00
parent be1755cac1
commit c286d56855
1 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
firewall/firewall.py
View File

@ -24,36 +24,36 @@ __date__ = "$Date$"
__copyright__ = "Copyright (c) 2004 Cyril Jaquier" __copyright__ = "Copyright (c) 2004 Cyril Jaquier"
__license__ = "GPL" __license__ = "GPL"
import time import time, os
class Firewall: class Firewall:
banList = dict() banList = dict()
def __init__(self, banTime, verbose = False): def __init__(self, banTime, logSys):
self.banTime = banTime self.banTime = banTime
self.verbose = verbose self.logSys = logSys
def addBanIP(self, ip): def addBanIP(self, ip, debug):
if not self.inBanList(ip): if not self.inBanList(ip):
self.logSys.info("Ban "+ip)
self.banList[ip] = time.time() self.banList[ip] = time.time()
self.executeCmd(self.banIP(ip)) self.executeCmd(self.banIP(ip), debug)
else: else:
if self.verbose: self.logSys.info(ip+" already in ban list")
print ip, "already in ban list"
def delBanIP(self, ip): def delBanIP(self, ip, debug):
if self.inBanList(ip): if self.inBanList(ip):
self.logSys.info("Unban "+ip)
del self.banList[ip] del self.banList[ip]
self.executeCmd(self.unBanIP(ip)) self.executeCmd(self.unBanIP(ip), debug)
else: else:
if self.verbose: self.logSys.info(ip+" not in ban list")
print ip, "not in ban list"
def inBanList(self, ip): def inBanList(self, ip):
return self.banList.has_key(ip) return self.banList.has_key(ip)
def checkForUnBan(self): def checkForUnBan(self, debug):
""" Check for user to remove from ban list. """ Check for user to remove from ban list.
""" """
banListTemp = self.banList.copy() banListTemp = self.banList.copy()
@ -63,21 +63,21 @@ class Firewall:
ip = element[0] ip = element[0]
btime = element[1] btime = element[1]
if btime < time.time()-self.banTime: if btime < time.time()-self.banTime:
self.delBanIP(ip) self.delBanIP(ip, debug)
if self.verbose:
print '`->', time.time()
def flushBanList(self): def flushBanList(self, debug):
iterBanList = self.banList.iteritems() iterBanList = self.banList.iteritems()
for i in range(len(self.banList)): for i in range(len(self.banList)):
element = iterBanList.next() element = iterBanList.next()
ip = element[0] ip = element[0]
self.delBanIP(ip) self.delBanIP(ip, debug)
def executeCmd(self, cmd): def executeCmd(self, cmd, debug):
if self.verbose: self.logSys.debug(cmd)
print cmd if not debug:
return #os.system(cmd) return os.system(cmd)
else:
return None
def viewBanList(self): def viewBanList(self):
iterBanList = self.banList.iteritems() iterBanList = self.banList.iteritems()