makeroute

   1 #!/usr/bin/env python
   2 # vim: set et sw=4 sts=4 :
   3
   4 import sys, os
   5
   6 def get_campos(line):
   7     r = []
   8     for c in line.split('\t'):
   9         r.append(c.strip('"\n'))
  10     return r
  11
  12 def parse(f):
  13     hosts = {}
  14     line = f.readline()
  15     while line:
  16         if line.startswith('"tabla de ruteo de '):
  17             host = line[19:-7]
  18             hosts[host] = {}
  19             line = f.readline()
  20             hosts[host]['ifaces'] = []
  21             while line and line.strip() <> '"red"\t"ip"\t"mascara"\t"gateway"\t"interfaz"\t"metrica"':
  22                 if line.startswith('"'):
  23                     (iface, ip, mascara, broadcast, peer) = get_campos(line)[:5]
  24                     hosts[host]['ifaces'].append({
  25                         'iface': iface,
  26                         'ip': ip,
  27                         'mascara': mascara,
  28                         'broadcast': broadcast,
  29                         'peer': peer,
  30                     })
  31                 line = f.readline()
  32             line = f.readline()
  33             hosts[host]['rutas'] = []
  34             while line and line.strip() <> '"--"':
  35                 if line.startswith('"'):
  36                     (red, ip, mascara, gateway, iface, metrica) = get_campos(line)
  37                     if not metrica: metrica = "1"
  38                     hosts[host]['rutas'].append({
  39                         'red': red,
  40                         'ip': ip,
  41                         'mascara': mascara,
  42                         'gateway': gateway,
  43                         'iface': iface,
  44                         'metrica': metrica,
  45                         })
  46                 line = f.readline()
  47         line = f.readline()
  48     return hosts
  49
  50 def up(host, fd):
  51     hosts = parse(fd)
  52     ppp = 0
  53     for iface in hosts[host]['ifaces']:
  54         if iface['iface'].startswith('ppp'):
  55             print 'UP: pppd -detach crtscts lock %(ip)s:%(peer)s /dev/ttyS0' % iface
  56             os.system('pppd -detach crtscts lock %(ip)s:%(peer)s /dev/ttyS0 &' % iface)
  57             ppp = 1
  58         else:
  59             print 'UP: ifconfig %(iface)s %(ip)s broadcast %(broadcast)s netmask %(mascara)s' % iface
  60             os.system('ifconfig %(iface)s %(ip)s broadcast %(broadcast)s netmask %(mascara)s' % iface)
  61     if ppp:
  62         print 'while sleep 1; do if ping -c1 %(peer)s 2>&1 > /dev/null; then break; else echo Esperando link ppp...; fi; done' % iface
  63         os.system('while sleep 1; do if ping -c1 %(peer)s 2>&1 > /dev/null; then break; else echo Esperando link ppp...; fi; done' % iface)
  64     for ruta in hosts[host]['rutas']:
  65         if ruta['metrica'] <> "0":
  66             print 'UP: route add -net %(ip)s gw %(gateway)s netmask %(mascara)s dev %(iface)s metric %(metrica)s' % ruta
  67             os.system('route add -net %(ip)s gw %(gateway)s netmask %(mascara)s dev %(iface)s metric %(metrica)s' % ruta)
  68
  69 def down(host, fd):
  70     hosts = parse(fd)
  71     for iface in hosts[host]['ifaces']:
  72         print 'DOWN: ifconfig %(iface)s down' % iface
  73         if iface['iface'].startswith('ppp'):
  74             os.system('killall pppd')
  75         else:
  76             os.system('ifconfig %(iface)s down' % iface)
  77
  78 def list(fd):
  79     hosts = parse(fd)
  80     for host in hosts.keys():
  81         print host
  82
  83 if __name__ == '__main__':
  84     fd = file('routers_labo.csv')
  85     if len(sys.argv) < 2:
  86         print >>sys.stderr, 'Uso: %s [up|down] host' % sys.argv[0]
  87         sys.exit(1)
  88     if sys.argv[1] == 'up':
  89         up(sys.argv[2], fd)
  90     elif sys.argv[1] == 'down':
  91         down(sys.argv[2], fd)
  92     elif sys.argv[1] == 'list':
  93         list(fd)
  94