508 lines
18 KiB
Python
Executable file
508 lines
18 KiB
Python
Executable file
#! /usr/bin/env python
|
|
# -*- coding: iso-8859-15 -*-
|
|
|
|
# Copyright (C) Frédéric Pauget
|
|
# Licence : GPLv2
|
|
|
|
"""Ce script permet de lancer la reconfiguration des divers services
|
|
|
|
Usage: %(prog)s options
|
|
Les options possibles sont :
|
|
\t%(options)s
|
|
Les options avec = doivent être suivies d'un argument. Plusieurs
|
|
arguments peuvent être founis pour une même option, les séparer par &
|
|
"""
|
|
|
|
import sys, signal, os, commands, getopt, smtplib, shutil
|
|
|
|
sys.path.append('/usr/scripts/gestion')
|
|
|
|
from ldap_crans import crans_ldap, crans, smtpserv, hostname, invite
|
|
from lock import *
|
|
from affich_tools import anim, cprint, OK, ERREUR, WARNING
|
|
from time import localtime, strftime, time, sleep
|
|
import config
|
|
|
|
args_autorises = ['quiet', 'remove=', 'add=', 'list' ,'help', 'reconnect']
|
|
if hostname == 'zamok' :
|
|
args_autorises += [ 'home=', 'mail_bienvenue=', 'ML-ENS=', 'droits', 'switch=' , 'dhcp', 'dns', 'firewall' , 'del_user=', 'blacklist_upload', 'autostatus']
|
|
elif hostname == 'nectaris' :
|
|
args_autorises += [ 'conf_wifi', 'bornes_wifi=' , 'droits-nectaris', 'dhcp-nectaris']
|
|
elif hostname == 'sila' :
|
|
args_autorises += [ 'bl_squid_upload', 'blacklist_virus' , 'blacklist_warez' , 'bl_chbre_invalide', 'bl_carte_etudiant' ]
|
|
elif hostname == 'komaz' :
|
|
args_autorises += [ 'firewall-komaz=' , 'firewall-komaz-ports=' , 'firewall-komaz-blacklist']
|
|
|
|
# On vérifie que l'on est root
|
|
if os.getuid() != 0:
|
|
sys.stderr.write("Il faut être root\n")
|
|
sys.exit(1)
|
|
|
|
signal.signal(signal.SIGINT,signal.SIG_IGN) # Pas de Ctrl-C
|
|
|
|
db = crans_ldap()
|
|
make_lock('auto_generate','Big lock',nowait=1)
|
|
|
|
##### Options fournies ?
|
|
try :
|
|
if len(sys.argv) > 1 :
|
|
options, arg = getopt.getopt(sys.argv[1:], 'h', args_autorises)
|
|
else :
|
|
options, arg = ( [],'')
|
|
except getopt.error, msg :
|
|
sys.stderr.write('%s\n' % msg)
|
|
sys.exit(255)
|
|
|
|
debug = 1 # défaut
|
|
to_do = {}
|
|
|
|
for opt, val in options :
|
|
if opt == '--quiet' :
|
|
debug = 0
|
|
|
|
elif opt == '--remove' :
|
|
for serv in val.split('&') :
|
|
print 'Supression de %s' % serv
|
|
db.services_to_restart('-%s' % serv)
|
|
sys.exit(0)
|
|
|
|
elif opt == '--list' :
|
|
print 'Services à redémarrer :'
|
|
print db.services_to_restart()
|
|
sys.exit(0)
|
|
|
|
elif opt == '--reconnect' :
|
|
# Personnes à reconnecter
|
|
print 'Recheche personnes en fin de sanction.'
|
|
hier = strftime('%d/%m/%Y %H:%M'.split()[0],localtime(time()-60*60*24))
|
|
c = db.search('blacklist=*,%s*' % hier)
|
|
services = []
|
|
for a_reco in c['adherent'] + c['machine'] + c['club'] :
|
|
for bl in a_reco.blacklist() :
|
|
fin, sanction = bl.split(',')[1:3]
|
|
if fin.split()[0] == hier and sanction not in services :
|
|
services.append(sanction)
|
|
for s in services :
|
|
print "Ajout de blacklist_%s pour reconfiguration" % s
|
|
db.services_to_restart('blacklist_%s' % s.encode())
|
|
sys.exit(0)
|
|
|
|
elif opt=='--add' :
|
|
# Ajout d'un item dans les services à redémarrer
|
|
for serv in val.split('&') :
|
|
if serv.find(',')!=-1 :
|
|
serv, arg = serv.split(',',1)
|
|
arg = arg.split(',')
|
|
else :
|
|
arg = []
|
|
|
|
print 'Ajout de %s (%s)' % ( serv, arg )
|
|
db.services_to_restart(serv, arg)
|
|
sys.exit(0)
|
|
|
|
elif opt =='-h' or opt == '--help' :
|
|
print __doc__ % { 'prog' : sys.argv[0].split('/')[-1] , 'options' : '\n\t'.join(args_autorises) }
|
|
sys.exit(0)
|
|
|
|
elif len(opt)>2 and opt[:2]=='--' :
|
|
to_do[opt[2:]] = val.split('&')
|
|
|
|
##### Lecture de la base LDAP si besion ce qu'il y a a faire et préparation
|
|
if not to_do :
|
|
if debug : print 'Lecture services à redémarrer dans la base LDAP'
|
|
to_do = db.services_to_restart()
|
|
auto = 1
|
|
else :
|
|
auto = 0
|
|
if debug : print 'Services à redémarrer imposés (non lecture de la base LDAP)'
|
|
|
|
inst = []
|
|
|
|
if hostname == 'zamok':
|
|
if 'del_user' in to_do.keys() :
|
|
if auto : db.services_to_restart('-del_user')
|
|
cprint('Archivage fichiers utilisateur','gras')
|
|
for args in to_do['del_user'] :
|
|
anim('\t' + args)
|
|
try :
|
|
login, home = args.split(',')
|
|
if not login or not home :
|
|
raise ValueError('Argument invalide')
|
|
warn = ''
|
|
f = config.cimetiere + '/files/' + strftime('%Y-%m-%d-%Hh%Mm_') + login + '.tar.bz2'
|
|
status, output = commands.getstatusoutput("tar cjf '%s' '%s' /var/spool/mail/%s" % ( f,home,login) )
|
|
if ( status != 512 and status!=0 ) or not os.path.isfile(f) :
|
|
# La 512 est si un des paths n'exite pas.
|
|
raise OSError(output)
|
|
if os.path.isdir(home) and os.stat(home)[4] >= 500 :
|
|
shutil.rmtree(home)
|
|
else :
|
|
warn += '%s incorrect\n' % home
|
|
if os.path.isfile('/var/spool/mail/' + login ) :
|
|
os.unlink('/var/spool/mail/' + login)
|
|
else :
|
|
warn += '/var/spool/mail/%s incorrect\n' % login
|
|
|
|
if warn :
|
|
print WARNING
|
|
if debug :
|
|
sys.stderr.write(warn)
|
|
else :
|
|
print OK
|
|
except :
|
|
print ERREUR
|
|
# On devra le refaire la prochaine fois
|
|
if auto : db.services_to_restart('del_user',[args])
|
|
if debug :
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if 'home' in to_do.keys() :
|
|
if auto : db.services_to_restart('-home')
|
|
cprint('Création home','gras')
|
|
for args in to_do['home'] :
|
|
anim('\t' + args)
|
|
try :
|
|
home, uid, login = args.split(',')
|
|
### Home
|
|
if not os.path.exists(home) :
|
|
# Le home n'existe pas
|
|
os.mkdir(home, 0755)
|
|
os.chown(home, int(uid) ,config.gid)
|
|
elif os.path.isdir(home) :
|
|
# Il y un répertoire existant
|
|
# Bon UID ?
|
|
stat = os.stat(home)
|
|
if stat[4] != int(uid) or stat[5] != config.gid :
|
|
# Le home n'est pas à la bonne personne
|
|
raise OSError('home existant')
|
|
|
|
### Mail
|
|
try :
|
|
os.mkdir(home + '/Mail', 0700)
|
|
os.chown(home + '/Mail', int(uid) ,config.gid)
|
|
except :
|
|
# Pas grave
|
|
pass
|
|
|
|
### Quota
|
|
status, output = commands.getstatusoutput('/usr/sbin/edquota -p pauget %s' % login )
|
|
if status :
|
|
print WARNING
|
|
if debug :
|
|
sys.stderr.write(output+'\n')
|
|
else :
|
|
print OK
|
|
except :
|
|
print ERREUR
|
|
# On devra le refaire la prochaine fois
|
|
if auto : db.services_to_restart('home',[args])
|
|
if debug :
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if 'mail_bienvenue' in to_do.keys() :
|
|
if auto : db.services_to_restart('-mail_bienvenue')
|
|
cprint('Envoi mail de bienvenue','gras')
|
|
for mail in to_do['mail_bienvenue'] :
|
|
anim('\t' + mail)
|
|
try :
|
|
From = "respbats@crans.org"
|
|
To = mail
|
|
if To.find('@') == -1 : To += '@crans.org'
|
|
conn=smtplib.SMTP(smtpserv)
|
|
conn.sendmail(From, To , config.txt_mail_bienvenue % { 'From' : From, 'To' : To })
|
|
conn.quit()
|
|
print OK
|
|
except Exception, c:
|
|
print ERREUR
|
|
if auto : db.services_to_restart('mail_bienvenue',[mail])
|
|
if debug :
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if 'ML-ENS' in to_do.keys() :
|
|
if auto : db.services_to_restart('-ML-ENS')
|
|
cprint('Inscription ML-ENS','gras')
|
|
for mail in to_do['ML-ENS'] :
|
|
anim('\t'+mail)
|
|
status, output = commands.getstatusoutput("echo '%s' | /usr/sbin/add_members -r - com-ens >/dev/null 2>&1" % mail)
|
|
if status :
|
|
# Il y a eu une erreur
|
|
print ERREUR
|
|
if auto : db.services_to_restart('ML-ENS', [mail])
|
|
if debug :
|
|
sys.stderr.write(output+'\n')
|
|
else :
|
|
print OK
|
|
|
|
if 'droits' in to_do.keys() :
|
|
if auto : db.services_to_restart('-droits')
|
|
db.services_to_restart('droits-nectaris')
|
|
try:
|
|
from gen_confs.droits import droits
|
|
a = droits()
|
|
a.uids = to_do['droits']
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto : db.services_to_restart('droits',to_do['droits'])
|
|
|
|
if 'blacklist_upload' in to_do.keys() :
|
|
if auto : db.services_to_restart('-blacklist_upload')
|
|
db.services_to_restart('bl_squid_upload')
|
|
db.services_to_restart('firewall-komaz-blacklist')
|
|
try:
|
|
from gen_confs.firewall import bl_upload_fw
|
|
a = bl_upload_fw()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto: db.services_to_restart('blacklist_upload')
|
|
|
|
if 'switch' in to_do.keys() :
|
|
if auto : db.services_to_restart('-switch')
|
|
try:
|
|
from gen_confs.switchs import switch
|
|
a = switch(to_do['switch'])
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto : db.services_to_restart('switch',to_do['switch'])
|
|
sys.stdout.write('Erreur dans la config des switchs.\n')
|
|
if debug :
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if 'mail_modification_machine' in to_do.keys() :
|
|
|
|
cprint('Modification d\'une machine : envoi d\'un mail à roots@crans.org','gras')
|
|
|
|
try :
|
|
from mail import mail_details
|
|
|
|
# liste des machines
|
|
machines = []
|
|
for ip in to_do['mail_modification_machine'] :
|
|
anim('\t' + ip)
|
|
print OK
|
|
machines += db.search('ipHostNumber=%s' % ip)['machine']
|
|
|
|
# envoi du mail
|
|
mail_details(machines, Subject = 'Modification d\'une machine de l\'association')
|
|
|
|
if auto : db.services_to_restart('-mail_modification_machine')
|
|
print OK
|
|
|
|
except :
|
|
print ERREUR
|
|
|
|
if 'autostatus' in to_do.keys() :
|
|
if auto : db.services_to_restart('-autostatus')
|
|
try:
|
|
from autostatus import autostatus
|
|
a = autostatus()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto : db.services_to_restart('autostatus')
|
|
|
|
# Les services suivants ont besoin de la liste des machines
|
|
# On va donc la lire une seule fois pour leur passer ensuite
|
|
|
|
restartlater = []
|
|
if 'dns' in to_do.keys() :
|
|
from gen_confs.bind import dns
|
|
inst.append([dns(),"dns"])
|
|
|
|
if 'dhcp' in to_do.keys() :
|
|
from gen_confs.dhcpd import dhcp
|
|
inst.append([dhcp(),"dhcp"])
|
|
|
|
if 'firewall' in to_do.keys() :
|
|
# Quand sila et komaz liront la base LDAP
|
|
from gen_confs.firewall import firewall
|
|
inst.append([firewall(),"firewall"])
|
|
|
|
|
|
elif hostname == 'nectaris':
|
|
restart_wifi_update = False
|
|
if 'conf_wifi' in to_do.keys() or 'bornes_wifi' in to_do.keys():
|
|
# On arrete wifi-update
|
|
os.system("kill $(cat /var/run/wifi-update.pid)")
|
|
sleep(1)
|
|
restart_wifi_update = True
|
|
|
|
# On s'occupe de nectaris
|
|
if 'conf_wifi' in to_do.keys():
|
|
if auto : db.services_to_restart('-conf_wifi')
|
|
try:
|
|
from gen_confs.wifi import conf_wifi
|
|
a = conf_wifi()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
# On va aussi demander la reconfiguration de toutes les bornes
|
|
to_do['bornes_wifi'] = []
|
|
except:
|
|
if auto : db.services_to_restart('conf_wifi')
|
|
sys.stdout.write('Erreur dans la config du wifi.\n')
|
|
|
|
if 'dhcp-nectaris' in to_do.keys() :
|
|
from gen_confs.dhcpd import dhcp
|
|
inst.append([dhcp(),"dhcp-nectaris"])
|
|
|
|
if 'bornes_wifi' in to_do.keys():
|
|
if auto : db.services_to_restart('-bornes_wifi')
|
|
try:
|
|
from gen_confs.wifi import bornes_wifi
|
|
if to_do['bornes_wifi']==['all']:
|
|
a = bornes_wifi()
|
|
else:
|
|
a = bornes_wifi(to_do['bornes_wifi'])
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
# Pas terrible : on va redemarrer les bornes qui ont redemarré correctement
|
|
if auto : db.services_to_restart('bornes_wifi', to_do['bornes_wifi'])
|
|
sys.stdout.write('Erreur dans la config des bornes wifi.\n')
|
|
|
|
if 'droits-nectaris' in to_do.keys():
|
|
if auto : db.services_to_restart('-droits-nectaris')
|
|
try:
|
|
from gen_confs.droits import droits
|
|
a = droits()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto : db.services_to_restart('droits-nectaris')
|
|
sys.stdout.write('Erreur dans la config des droits sur nectaris.\n')
|
|
|
|
if restart_wifi_update:
|
|
os.system("cd / ; systrace -d /etc/systrace/wifi-update -A /usr/local/bin/twistd --syslog -o -y /usr/scripts/gestion/wifi-update.py --pidfile=/var/run/wifi-update.pid")
|
|
|
|
elif hostname == 'sila' :
|
|
if 'bl_carte_etudiant' in to_do.keys() :
|
|
if auto : db.services_to_restart('-bl_carte_etudiant')
|
|
try:
|
|
from gen_confs.squid import squid_carte
|
|
a = squid_carte()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto: db.services_to_restart('bl_carte_etudiant')
|
|
|
|
if 'blacklist_virus' in to_do.keys() :
|
|
if auto : db.services_to_restart('-blacklist_virus')
|
|
try:
|
|
from gen_confs.squid import squid_virus
|
|
a = squid_virus()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto: db.services_to_restart('blacklist_virus')
|
|
|
|
if 'blacklist_warez' in to_do.keys() :
|
|
if auto : db.services_to_restart('-blacklist_warez')
|
|
try:
|
|
from gen_confs.squid import squid_warez
|
|
a = squid_warez()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto: db.services_to_restart('blacklist_warez')
|
|
|
|
if 'bl_squid_upload' in to_do.keys() :
|
|
if auto : db.services_to_restart('-bl_squid_upload')
|
|
try:
|
|
from gen_confs.squid import squid_upload
|
|
a = squid_upload()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto: db.services_to_restart('bl_squid_upload')
|
|
|
|
if 'bl_chbre_invalide' in to_do.keys() :
|
|
if auto : db.services_to_restart('-bl_chbre_invalide')
|
|
try:
|
|
from gen_confs.squid import squid_chbre
|
|
a = squid_chbre()
|
|
a.debug = debug
|
|
a.reconfigure()
|
|
except:
|
|
if auto: db.services_to_restart('bl_chbre_invalide')
|
|
|
|
elif hostname == 'komaz' :
|
|
if 'firewall-komaz' in to_do.keys() or \
|
|
'firewall-komaz-ports' in to_do.keys() or \
|
|
'firewall-komaz-blacklist' in to_do.keys() :
|
|
from firewall_komaz import firewall_komaz
|
|
fw = firewall_komaz()
|
|
fw.debug = debug
|
|
|
|
cprint('Reconfiguration firewall','gras')
|
|
if 'firewall-komaz' in to_do.keys() :
|
|
if auto : db.services_to_restart('-firewall-komaz')
|
|
try :
|
|
fw.mac_ip_maj(to_do['firewall-komaz'])
|
|
except :
|
|
if auto : db.services_to_restart('firewall-komaz',to_do['firewall-komaz'])
|
|
|
|
if 'firewall-komaz-ports' in to_do.keys() :
|
|
if auto : db.services_to_restart('-firewall-komaz-ports')
|
|
try :
|
|
fw.port_maj(to_do['firewall-komaz-ports'])
|
|
except :
|
|
if auto : db.services_to_restart('firewall-komaz-ports',to_do['firewall-komaz-ports'])
|
|
|
|
if 'firewall-komaz-blacklist' in to_do.keys() :
|
|
if auto : db.services_to_restart('-firewall-komaz-blacklist')
|
|
try :
|
|
fw.blacklist()
|
|
except :
|
|
if auto : db.services_to_restart('firewall-komaz-blacklist',to_do['firewall-komaz-blacklist'])
|
|
|
|
# On indique que les services seront a priori redemarrés
|
|
if auto :
|
|
for i in inst:
|
|
db.services_to_restart("-%s" % i[1])
|
|
## On fait ce qu'il reste à faire
|
|
|
|
if inst :
|
|
# Récolte des données
|
|
cprint('Lecture base LDAP','gras')
|
|
# Machines de l'assoce
|
|
machines = crans(db.conn).machines()
|
|
# Machines des invités
|
|
machines += invite(db.conn).machines()
|
|
# Machines des adhérents et clubs de l'année en cours
|
|
base = db.search('paiement=ok')
|
|
base = base['adherent'] + base['club']
|
|
a = anim('\ttri machines',len(base))
|
|
for adh in base :
|
|
a.cycle()
|
|
# Adhérent ayant payé l'année en cours
|
|
if 'bloq' in adh.blacklist_actif() :
|
|
# Adhérent ignoré
|
|
continue
|
|
machines += adh.machines()
|
|
a.reinit()
|
|
print OK
|
|
|
|
# Reconfiguration des services
|
|
|
|
for i in inst :
|
|
i[0].debug = debug
|
|
i[0].machines = machines
|
|
try :
|
|
i[0].reconfigure()
|
|
except :
|
|
# Erreur, il faudra relancer le service la prochaine fois
|
|
if auto : db.services_to_restart('%s' % i[1])
|
|
sys.stdout.write('Erreur dans la config de %s.\n' % i[1])
|
|
|
|
if debug :
|
|
print 'Non traité ici mais signalé dans la base LDAP : \n\t', db.services_to_restart()
|
|
|
|
signal.signal(signal.SIGINT,signal.SIG_DFL) # Comportement normal de Ctrl-C
|
|
|
|
remove_lock('auto_generate')
|