From 324dd4fdc2bdfed32cc0dbb3aa3cc4c8c453c946 Mon Sep 17 00:00:00 2001
From: lostcontrol <lostcontrol@a942ae1a-1317-0410-a47c-b1dcaea8d605>
Date: Mon, 18 Oct 2004 20:50:56 +0000
Subject: [PATCH] - Fix retryAllowed. It is now conf["maxretry"] - Add SIGTERM
 handler in order to exit properly when in daemon mode

git-svn-id: https://fail2ban.svn.sourceforge.net/svnroot/fail2ban/trunk@37 a942ae1a-1317-0410-a47c-b1dcaea8d605
---
 fail2ban.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/fail2ban.py b/fail2ban.py
index 9e640895..f445b584 100755
--- a/fail2ban.py
+++ b/fail2ban.py
@@ -146,6 +146,15 @@ def createDaemon():
 
 	return(0)
 
+def sigTERMhandler(signum, frame):
+	""" Handles the TERM signal when in daemon mode in order to
+		exit properly.
+	"""
+	logSys.debug("Signal handler called with sig "+`signum`)
+	logSys.info("Restoring iptables...")
+	fireWall.flushBanList(conf["debug"])
+	logSys.info("Exiting...")
+	sys.exit(0)
 
 if __name__ == "__main__":
 	
@@ -289,6 +298,7 @@ if __name__ == "__main__":
 			logSys.set_formatstring(log4py.FMT_DEBUG)
 		elif c == "background" and conf[c]:
 			retCode = createDaemon()
+			signal.signal(signal.SIGTERM, sigTERMhandler)
 			logSys.set_target(conf["logfile"])
 			if retCode != 0:
 				logSys.error("Unable to start daemon")
@@ -351,7 +361,7 @@ if __name__ == "__main__":
 			iterFailList = failList.iteritems()
 			for i in range(len(failList)):
 				element = iterFailList.next()
-				if element[1][0] >= retryAllowed:
+				if element[1][0] >= conf["maxretry"]:
 					fireWall.addBanIP(element[0], conf["debug"])
 			
 		except KeyboardInterrupt: