51 lines
1.4 KiB
Python
Executable file
51 lines
1.4 KiB
Python
Executable file
#! /usr/bin/env python
|
|
# -*- coding: iso-8859-15 -*-
|
|
|
|
# Ajout d'un whos aux mails d'arpwatch
|
|
# Auteur : Stéphane Glondu
|
|
# Licence : GPLv2
|
|
|
|
import sys, os, sre, smtplib
|
|
from commands import getstatusoutput
|
|
|
|
find_mac = sre.compile(r'[0-9A-Fa-f]{1,2}(?::[0-9A-Fa-f]{1,2}){5}')
|
|
|
|
def get_machine(mac):
|
|
commande = "/usr/scripts/gestion/whos.py --debug mac=%s" % mac
|
|
r, s = getstatusoutput(commande)
|
|
if r == 3:
|
|
# Aucun résultat trouvé
|
|
return ""
|
|
else:
|
|
# On supprime les couleurs
|
|
s = sre.sub('\x1b\[1;([0-9]|[0-9][0-9])m', '', s)
|
|
# On rajoute éventuellement le code de retour
|
|
if r != 0:
|
|
s = """
|
|
Code de retour anormal : %d
|
|
Commande : %s
|
|
%s
|
|
""" % (r, commande, s)
|
|
else:
|
|
s = "\n" + s + "\n"
|
|
return s
|
|
|
|
if __name__ == "__main__":
|
|
texte = sys.stdin.read()
|
|
# On récupère les destinataires dans les arguments (très ad hoc)
|
|
recipients = sys.argv[2].split(',')
|
|
# On complète le message
|
|
try:
|
|
macs = find_mac.findall(texte)
|
|
for mac in macs:
|
|
texte += get_machine(mac)
|
|
except:
|
|
# En cas d'exception, on envoie le traceback
|
|
import traceback
|
|
texte += '\n'
|
|
texte += '\n'.join(traceback.format_exception(sys.exc_type, sys.exc_value, sys.exc_traceback))
|
|
texte += '\n-- \narpwatch_sendmail.py\n'
|
|
smtp = smtplib.SMTP()
|
|
smtp.connect()
|
|
smtp.sendmail("arpwatch@crans.org", recipients, texte)
|
|
smtp.quit()
|