269 lines
8.9 KiB
Python
269 lines
8.9 KiB
Python
#! /usr/bin/env python
|
|
# -*- coding: iso8859-15 -*-
|
|
|
|
import socket
|
|
import sys, re
|
|
from pyPgSQL import PgSQL
|
|
sys.path.append('/usr/scripts/gestion/')
|
|
from affich_tools import tableau
|
|
|
|
def stats(ip_crans=[], ip_ext=[], show=['ip_crans','ip_ext','port_crans','port_ext'], upload_mini=0, show_limit=10, begin_time=24, end_time=0):
|
|
"""
|
|
Retourne une chaine de caratères formatée avec le tableau de statistiques
|
|
d'upload de l'ip fourni
|
|
|
|
ip_crans : ips des machine
|
|
chaine de caratère si il a qu'une machine
|
|
liste si il y a plusieurs machines
|
|
view : liste des champs à afficher parmi :
|
|
ip_crans, ip_ext, port_crans, port_ext
|
|
upload_mini : n'affiche que les lignes avec plus d'upload que la valeur donnée
|
|
show_limit : nombre max de lignes à afficher
|
|
begin : date de départ d'analyse (en heure avant maintenant)
|
|
end : date de fin d'analyse (en heure avant maintenant)
|
|
"""
|
|
|
|
if type(ip_crans) == str:
|
|
ip_crans = [ip_crans]
|
|
|
|
if type(ip_ext) == str:
|
|
ip_ext = [ip_ext]
|
|
|
|
# variables pour la requete et l'affichage
|
|
##########################################
|
|
|
|
select = []
|
|
largeur = []
|
|
titre = []
|
|
format = []
|
|
alignement = []
|
|
|
|
if 'ip_crans' in show and len(ip_crans)!=1:
|
|
select.append('ip_crans')
|
|
largeur.append(13)
|
|
titre.append('machine crans')
|
|
format.append('s')
|
|
alignement.append('c')
|
|
|
|
if 'ip_ext' in show :
|
|
select.append('ip_ext')
|
|
largeur.append('*')
|
|
titre.append('machine ext')
|
|
format.append('s')
|
|
alignement.append('c')
|
|
|
|
if 'port_ext' in show :
|
|
select.append('port_ext')
|
|
largeur.append(10)
|
|
titre.append('port ext')
|
|
format.append('s')
|
|
alignement.append('d')
|
|
|
|
if 'port_crans' in show :
|
|
select.append('port_crans')
|
|
largeur.append(10)
|
|
titre.append('port crans')
|
|
format.append('s')
|
|
alignement.append('d')
|
|
|
|
select += ['sum(download) as download','sum(upload) as upload']
|
|
largeur += [10,10]
|
|
titre += ['download','upload']
|
|
format += ['o','o']
|
|
alignement += ['d','d']
|
|
|
|
# requete dans la base
|
|
######################
|
|
ip_crans = ' OR '.join([ "ip_crans='%s'"%x for x in ip_crans ])
|
|
if not ip_crans: ip_crans='true'
|
|
ip_ext = ' OR '.join([ "ip_ext='%s'"%x for x in ip_ext ])
|
|
if not ip_ext: ip_ext='true'
|
|
|
|
requete = "SELECT * FROM ( SELECT %s FROM upload WHERE (%s) AND (%s) AND (date > timestamp 'now' - interval '%d hours') AND (date < timestamp 'now' - interval '%d hours') GROUP BY %s) AS resultat_intemediaire WHERE upload>='%d' ORDER BY upload DESC LIMIT %d;" % (','.join(select), ip_crans, ip_ext, begin_time, end_time, ','.join(show), upload_mini*1024*1024, show_limit)
|
|
|
|
pgsql = PgSQL.connect(host='/var/run/postgresql', database='filtrage', user='crans')
|
|
curseur = pgsql.cursor()
|
|
curseur.execute(requete)
|
|
results = curseur.fetchall()
|
|
|
|
# on transforme tout en chaine
|
|
results = [ [ str(x) for x in line ] for line in results ]
|
|
|
|
# on modifie les ip en noms de machine et les ports en noms
|
|
def nom_de_machine (ip) :
|
|
try:
|
|
return socket.gethostbyaddr(ip)[0]
|
|
except:
|
|
return ip
|
|
|
|
port_to_service = {}
|
|
for service,port in [ re.split('[ \t]+',x.strip().replace('/tcp','').replace('/udp',''))[:2] for x in open('/etc/services').readlines() if x[0] not in ['\n','#'] ] :
|
|
port_to_service[port]=service
|
|
|
|
for champ in select:
|
|
if champ == 'ip_ext':
|
|
col = select.index(champ)
|
|
results = [ x[:col] + [nom_de_machine(x[col])] + x[col+1:] for x in results ]
|
|
|
|
elif champ == 'ip_crans':
|
|
col = select.index(champ)
|
|
results = [ x[:col] + [nom_de_machine(x[col]).split('.')[0]] + x[col+1:] for x in results ]
|
|
|
|
elif 'port' in champ:
|
|
col = select.index(champ)
|
|
results = [ x[:col] + [port_to_service.get(x[col],x[col])] + x[col+1:] for x in results ]
|
|
|
|
return tableau(results, titre=titre, largeur=largeur, alignement=alignement, format=format)
|
|
|
|
if __name__ == '__main__' :
|
|
|
|
help = """Statistiques d'upload d'une machine du crans
|
|
usage: analyse.py [options]
|
|
Option fait partie des options suivantes :
|
|
--ip-crans
|
|
ip de la machine crans
|
|
--ip-ext
|
|
ip de la machine extérieure
|
|
--show
|
|
champs à afficher (parmi ip_crans, ip_ext, port_crans, port_ext)
|
|
--upload-mini
|
|
upload mini des lignes à afficher
|
|
--show-limit
|
|
nombre maximum de lignes à afficher
|
|
--begin-time
|
|
heure de départ de l'analyse (en heure depuis maintenant)
|
|
--end-time
|
|
heure de fin de l'analyse (en heure depuis maintenant)
|
|
|
|
Exemples :
|
|
sudo /usr/scripts/surveillance/analyse.py bilou.crans.org"""
|
|
|
|
# import des modules
|
|
import getopt
|
|
|
|
# aide
|
|
######
|
|
if '-h' in sys.argv or '--help' in sys.argv or len(sys.argv)==1:
|
|
print help
|
|
sys.exit(0)
|
|
|
|
# parsage des arguments
|
|
#######################
|
|
try :
|
|
opts, args = getopt.getopt(sys.argv[1:],'',['ip-crans=','ip-ext=','show=','upload-mini=','begin-time=','end-time=','show-limit='])
|
|
except getopt.GetoptError,message :
|
|
print help
|
|
sys.exit(4)
|
|
|
|
print "Statistiques d'upload"
|
|
|
|
# recherche de la machine crans
|
|
###############################
|
|
ip_crans = []
|
|
ip_crans_nom = []
|
|
for key,value in opts :
|
|
if key == '--ip-crans' :
|
|
try :
|
|
ip_crans.append(socket.gethostbyaddr(value)[2][0])
|
|
ip_crans_nom.append(socket.gethostbyaddr(value)[0])
|
|
except socket.gaierror :
|
|
print "Hôte %s inconnu" % value
|
|
sys.exit(5)
|
|
if len(ip_crans_nom)==1:
|
|
print ' depuis la machine %s' % ip_crans_nom[0]
|
|
elif ip_crans_nom:
|
|
print ' depuis les machines %s' % ', '.join(ip_crans_nom)
|
|
|
|
# recherche de la machine ext
|
|
#############################
|
|
ip_ext = []
|
|
ip_ext_nom = []
|
|
for key,value in opts :
|
|
if key == '--ip-ext' :
|
|
# recherche de l'ip de la machine extérieur
|
|
try:
|
|
ip_ext.append(socket.gethostbyaddr(value)[2][0])
|
|
except socket.herror:
|
|
ip_ext.append(value)
|
|
except socket.gaierror :
|
|
print "Hôte %s inconnu" % value
|
|
sys.exit(5)
|
|
|
|
# recherche du nom d'hote
|
|
try :
|
|
ip_ext_nom.append(socket.gethostbyaddr(value)[0])
|
|
except socket.herror:
|
|
ip_ext_nom.append(ip_ext[-1])
|
|
except socket.gaierror :
|
|
ip_ext_nom.append(ip_ext[-1])
|
|
if len(ip_ext_nom)==1:
|
|
print ' vers la machine extérieure %s' % ip_ext_nom[0]
|
|
elif ip_ext_nom:
|
|
print ' vers les machines extérieures %s' % ', '.join(ip_ext_nom)
|
|
|
|
# limite d'affichage
|
|
####################
|
|
show = [ x[1] for x in opts if x[0] == '--show' and x[1] in ['ip_crans','ip_ext','port_crans','port_ext'] ]
|
|
if not show :
|
|
show = ['ip_crans','ip_ext','port_crans','port_ext']
|
|
else :
|
|
print ' affichage de %s' % ', '.join(show)
|
|
|
|
# upload mini à afficher
|
|
########################
|
|
upload_mini = 0
|
|
for key,value in opts :
|
|
if key == '--upload-mini' :
|
|
try :
|
|
upload_mini = int(value)
|
|
except :
|
|
print 'L\'upload mini doit être un entier (en MO)'
|
|
sys.exit(4)
|
|
break
|
|
if upload_mini:
|
|
print ' pour les traffics supérieurs à %d Mo' % upload_mini
|
|
|
|
# nombre limite d'enregristrements
|
|
##################################
|
|
limit = 10
|
|
for key,value in opts :
|
|
if key == '--show-limit' :
|
|
try :
|
|
limit = int(value)
|
|
print ' affichage des %d premiers résultats' % limit
|
|
except :
|
|
print 'Le nombre limite n\'est pas un entier'
|
|
sys.exit(3)
|
|
break
|
|
|
|
# début de l'analyse
|
|
####################
|
|
begin_time = 24
|
|
for key,value in opts :
|
|
if key == '--begin-time' :
|
|
try :
|
|
begin_time = int(value)
|
|
except :
|
|
print 'Le nombre d\'heures doit être un entier'
|
|
sys.exit(4)
|
|
break
|
|
|
|
# fin de l'analyse
|
|
##################
|
|
end_time = 0
|
|
for key,value in opts :
|
|
if key == '--end-time' :
|
|
try :
|
|
end_time = int(value)
|
|
except :
|
|
print 'Le nombre d\'heures doit être un entier'
|
|
sys.exit(4)
|
|
break
|
|
if begin_time != 24 or end_time:
|
|
print ' entre il y a %d heures et il y a %d heures' % (begin_time,end_time)
|
|
|
|
# affichage du résultat
|
|
#######################
|
|
#print stats(ip_crans, group, upload_mini, limit, heures)
|
|
print stats(ip_crans, ip_ext, show, upload_mini, limit, begin_time, end_time)
|
|
|