Skript kontrollib OpenSSH logist ebaõnnestunud autentimisi ning lävendi ületamisel blokeerib IP aadressi

#!/usr/bin/python
# -*- coding: utf-8 -*-
#Autor: Risto Siitan
#Ryhm:A22
#Skript kontrollib OpenSSH logist ebaõnnestunud autentimisi ning lävendi ületamisel blokeerib IP aadressi
#smtplib on vajalik e-maili saatmiseks
import smtplib
import sys
#loome sõnastiku
failures = {}
#Kontrollime, kas logi fail on olemas:
if len(sys.argv) < 2:
        print "Kasuta järgnevalt: <skripti nimi> <logifail>"
        sys.exit(0)
else:
#avame hosts.deny ning vastava logi faili
	kir=open("/etc/hosts.deny",'r+')
	fh = open(sys.argv[1])
	for rida in fh.readlines():
#Leiame ebaõnnestunud sisse logimiste IP-aadressid ja lisame sõnastikku:
		if "Failed password" in rida: 
			r = rida.split()[12]
			if failures.has_key(r):
				failures[r] += 1
			else: 
				failures[r] = 1
#Omistame uue nime sõnastikule ning sorteerime suuremast väiksemaks(-arv):
	massiiv = failures.items()
	massiiv.sort(key = lambda (failures, arv):-arv)
	for failures, arv in massiiv:
#Lisame lävendi(antud juhul 9):
		if (arv > 9):
#Kirjutame lävendist üle olevad IP aadressid hosts.deny faili:
			vana=kir.read()
			kir.write(vana +'%s\n' % (failures))
	kir.close()
	fh.close()
 
#Saadame e-mailiga teavituse:
	sender = 'risto@siitan.com'
	receivers = ['risto@siitan.com']
 
	message = """From: Automaat
	To: Admin
	Subject: Blokeeriti IP aadress
	/etc/hosts.deny listi lisati uus IP aadress.
	"""
 
	try:
#elion - mail.neti.ee; starman - mail.starman.ee; uninet - smtp.uninet.ee; stv - mail.stv.ee)
	   smtpObj = smtplib.SMTP('mail.neti.ee')
	   smtpObj.sendmail(sender, receivers, message)         
	   print "Email saadetud"
	except smtplib.SMTPException:
	   print "Error: Ei saanud email-i saata"