source: genesis/tools/gformat.py@ 10712

Last change on this file since 10712 was 10705, checked in by rick, 13 years ago

Make template, details later.

Related-To: nodefactory#152

  • Property svn:executable set to *
  • Property svn:keywords set to Id
File size: 45.2 KB
RevLine 
[8242]1#!/usr/bin/env python
2#
3# vim:ts=2:et:sw=2:ai
4# Wireless Leiden configuration generator, based on yaml files'
[9957]5#
6# XXX: This should be rewritten to make use of the ipaddr.py library.
7#
[10058]8# Sample apache configuration (mind the AcceptPathInfo!)
9# ScriptAlias /wleiden/config /usr/local/www/genesis/tools/gformat.py
10# <Directory /usr/local/www/genesis>
11# Allow from all
12# AcceptPathInfo On
13# </Directory>
14#
[8242]15# Rick van der Zwet <info@rickvanderzwet.nl>
[9957]16#
[8622]17
18# Hack to make the script directory is also threated as a module search path.
19import sys
20import os
[9286]21import re
[8622]22sys.path.append(os.path.dirname(__file__))
23
[8242]24import cgi
[8267]25import cgitb
26import copy
[8242]27import glob
28import socket
29import string
30import subprocess
31import time
[8622]32import rdnap
[10378]33import make_network_kml
[8584]34from pprint import pprint
[10281]35from collections import defaultdict
[8575]36try:
37 import yaml
38except ImportError, e:
39 print e
40 print "[ERROR] Please install the python-yaml or devel/py-yaml package"
41 exit(1)
[8588]42
43try:
44 from yaml import CLoader as Loader
45 from yaml import CDumper as Dumper
46except ImportError:
47 from yaml import Loader, Dumper
48
[10584]49from jinja2 import Environment, Template
50def yesorno(value):
51 return "YES" if bool(value) else "NO"
52env = Environment()
53env.filters['yesorno'] = yesorno
54def render_template(datadump, template):
55 result = env.from_string(template).render(datadump)
56 # Make it look pretty to the naked eye, as jinja templates are not so
57 # friendly when it comes to whitespace formatting
58 ## Remove extra whitespace at end of line lstrip() style.
59 result = re.sub(r'\n[\ ]+','\n', result)
60 ## Include only a single newline between an definition and a comment
61 result = re.sub(r'(["\'])\n+([a-z]|\n#\n)',r'\1\n\2', result)
62 ## Remove extra newlines after single comment
63 result = re.sub(r'(#\n)\n+([a-z])',r'\1\2', result)
64 return result
[10110]65
[9697]66import logging
67logging.basicConfig(format='# %(levelname)s: %(message)s' )
68logger = logging.getLogger()
69logger.setLevel(logging.DEBUG)
[8242]70
[9283]71
[8948]72if os.environ.has_key('CONFIGROOT'):
73 NODE_DIR = os.environ['CONFIGROOT']
74else:
[9283]75 NODE_DIR = os.path.abspath(os.path.dirname(__file__)) + '/../nodes'
[8242]76__version__ = '$Id: gformat.py 10705 2012-05-07 14:52:28Z rick $'
77
[8267]78
[9283]79files = [
[8242]80 'authorized_keys',
81 'dnsmasq.conf',
[10410]82 'dhcpd.conf',
[8242]83 'rc.conf.local',
84 'resolv.conf',
[10069]85 'motd',
[10654]86 'ntp.conf',
[10705]87 'pf.hybrid.conf.local',
[10054]88 'wleiden.yaml',
[8242]89 ]
90
[8319]91# Global variables uses
[8323]92OK = 10
93DOWN = 20
94UNKNOWN = 90
[8257]95
[10391]96def get_yaml(item):
97 """ Get configuration yaml for 'item'"""
98 gfile = os.path.join(NODE_DIR,item,'wleiden.yaml')
[8257]99
[10461]100 # Use some boring defaults
[10702]101 datadump = {}
[10391]102 f = open(gfile, 'r')
[10461]103 datadump.update(yaml.load(f,Loader=Loader))
[10702]104 if datadump['nodetype'] == 'Hybrid':
105 defaults = { 'service_proxy_normal' : False, 'service_proxy_ileiden' : False, 'service_accesspoint' : True }
106 for (key,value) in defaults.iteritems():
107 if not datadump.has_key(key):
108 datadump[key] = value
[10391]109 f.close()
110
111 # Preformat certain needed variables for formatting and push those into special object
112 datadump['autogen_iface_keys'] = get_interface_keys(datadump)
113
114 wlan_count=0
115 for key in datadump['autogen_iface_keys']:
116 if datadump[key]['type'] in ['11a', '11b', '11g', 'wireless']:
117 datadump[key]['autogen_ifname'] = 'wlan%i' % wlan_count
118 wlan_count += 1
119 else:
120 datadump[key]['autogen_ifname'] = datadump[key]['interface'].split(':')[0]
121
[10459]122 dhcp_interfaces = [datadump[key]['autogen_ifname'] for key in datadump['autogen_iface_keys'] if datadump[key]['dhcp']]
123 datadump['autogen_dhcp_interfaces'] = ','.join(dhcp_interfaces)
[10391]124 datadump['autogen_item'] = item
125
126 datadump['autogen_realname'] = get_realname(datadump)
127 datadump['autogen_domain'] = datadump['domain'] if datadump.has_key('domain') else 'wleiden.net.'
128 datadump['autogen_fqdn'] = datadump['autogen_realname'] + '.' + datadump['autogen_domain']
129 return datadump
130
131
132def store_yaml(datadump, header=False):
133 """ Store configuration yaml for 'item'"""
134 item = datadump['autogen_item']
135 gfile = os.path.join(NODE_DIR,item,'wleiden.yaml')
136
137 f = open(gfile, 'w')
138 f.write(generate_wleiden_yaml(datadump, header))
139 f.close()
140
141
142
[10281]143def make_relations():
[10270]144 """ Process _ALL_ yaml files to get connection relations """
145 errors = ""
[10281]146 poel = defaultdict(list)
[10270]147 for host in get_hostlist():
148 try:
149 datadump = get_yaml(host)
150 for iface_key in datadump['autogen_iface_keys']:
151 l = datadump[iface_key]['ip']
152 addr, mask = l.split('/')
153
154 # Not parsing of these folks please
155 if not valid_addr(addr):
156 continue
157
158 addr = parseaddr(addr)
159 mask = int(mask)
[10281]160 network = addr & ~((1 << (32 - mask)) - 1)
161 poel[network] += [(host,datadump[iface_key])]
[10270]162 except (KeyError, ValueError), e:
163 errors += "[FOUT] in '%s' interface '%s'" % (host,iface_key)
164 errors += e
165 continue
166 return (poel, errors)
167
168
[8267]169
[8321]170def valid_addr(addr):
171 """ Show which address is valid in which are not """
172 return str(addr).startswith('172.')
173
[10692]174def get_system_list(prefix):
175 return sorted([os.path.basename(os.path.dirname(x)) for x in glob.glob("%s/%s*/wleiden.yaml" % (NODE_DIR, prefix))])
[8321]176
[10692]177get_hybridlist = lambda: get_system_list("Hybrid")
178get_nodelist = lambda: get_system_list("CNode")
179get_proxylist = lambda: get_system_list("Proxy")
[8267]180
[8296]181def get_hostlist():
182 """ Combined hosts and proxy list"""
[10192]183 return get_nodelist() + get_proxylist() + get_hybridlist()
[8267]184
[8588]185def angle_between_points(lat1,lat2,long1,long2):
[9283]186 """
[8588]187 Return Angle in radians between two GPS coordinates
188 See: http://stackoverflow.com/questions/3809179/angle-between-2-gps-coordinates
189 """
190 dy = lat2 - lat1
191 dx = math.cos(math.pi/180*lat1)*(long2 - long1)
192 angle = math.atan2(dy,dx)
193 return angle
[8267]194
[8588]195def angle_to_cd(angle):
196 """ Return Dutch Cardinal Direction estimation in 'one digit' of radian angle """
197
198 # For easy conversion get positive degree
199 degrees = math.degrees(angle)
200 if degrees < 0:
201 360 - abs(degrees)
202
203 # Numbers can be confusing calculate from the 4 main directions
204 p = 22.5
205 if degrees < p:
206 return "n"
[9283]207 elif degrees < (90 - p):
[8588]208 return "no"
[9283]209 elif degrees < (90 + p):
[8588]210 return "o"
[9283]211 elif degrees < (180 - p):
[8588]212 return "zo"
[9283]213 elif degrees < (180 + p):
[8588]214 return "z"
[9283]215 elif degrees < (270 - p):
[8588]216 return "zw"
[9283]217 elif degrees < (270 + p):
[8588]218 return "w"
[9283]219 elif degrees < (360 - p):
[8588]220 return "nw"
221 else:
222 return "n"
223
224
[8267]225def generate_title(nodelist):
[8257]226 """ Main overview page """
[9283]227 items = {'root' : "." }
[10682]228 def fl(spaces, line):
229 return (' ' * spaces) + line + '\n'
230
[8267]231 output = """
[8257]232<html>
233 <head>
234 <title>Wireless leiden Configurator - GFormat</title>
235 <style type="text/css">
236 th {background-color: #999999}
237 tr:nth-child(odd) {background-color: #cccccc}
238 tr:nth-child(even) {background-color: #ffffff}
239 th, td {padding: 0.1em 1em}
240 </style>
241 </head>
242 <body>
243 <center>
[8259]244 <form type="GET" action="%(root)s">
[8257]245 <input type="hidden" name="action" value="update">
246 <input type="submit" value="Update Configuration Database (SVN)">
247 </form>
248 <table>
[10682]249 <caption><h3>Wireless Leiden Configurator</h3></caption>
[8257]250 """ % items
[8242]251
[8296]252 for node in nodelist:
[8257]253 items['node'] = node
[10682]254 output += fl(5, '<tr>') + fl(7,'<td><a href="%(root)s/%(node)s">%(node)s</a></td>' % items)
[8257]255 for config in files:
256 items['config'] = config
[10682]257 output += fl(7,'<td><a href="%(root)s/%(node)s/%(config)s">%(config)s</a></td>' % items)
258 output += fl(5, "</tr>")
[8267]259 output += """
[8257]260 </table>
261 <hr />
262 <em>%s</em>
263 </center>
264 </body>
265</html>
266 """ % __version__
[8242]267
[8267]268 return output
[8257]269
270
[8267]271
272def generate_node(node):
[8257]273 """ Print overview of all files available for node """
[8267]274 return "\n".join(files)
[8242]275
[10270]276def generate_node_overview(host):
277 """ Print overview of all files available for node """
278 datadump = get_yaml(host)
279 params = { 'host' : host }
280 output = "<em><a href='..'>Back to overview</a></em><hr />"
281 output += "<h2>Available files:</h2><ul>"
282 for cf in files:
283 params['cf'] = cf
284 output += '<li><a href="%(host)s/%(cf)s">%(cf)s</a></li>\n' % params
285 output += "</ul>"
[8257]286
[10270]287 # Generate and connection listing
288 output += "<h2>Connected To:</h2><ul>"
[10281]289 (poel, errors) = make_relations()
290 for network, hosts in poel.iteritems():
291 if host in [x[0] for x in hosts]:
292 if len(hosts) == 1:
293 # Single not connected interface
294 continue
295 for remote,ifacedump in hosts:
296 if remote == host:
297 # This side of the interface
298 continue
299 params = { 'remote': remote, 'remote_ip' : ifacedump['ip'] }
300 output += '<li><a href="%(remote)s">%(remote)s</a> -- %(remote_ip)s</li>\n' % params
[10270]301 output += "</ul>"
[10281]302 output += "<h2>MOTD details:</h2><pre>" + generate_motd(datadump) + "</pre>"
[8257]303
[10270]304 output += "<hr /><em><a href='..'>Back to overview</a></em>"
305 return output
306
307
[8242]308def generate_header(ctag="#"):
309 return """\
[9283]310%(ctag)s
[8242]311%(ctag)s DO NOT EDIT - Automatically generated by 'gformat'
312%(ctag)s Generated at %(date)s by %(host)s
[9283]313%(ctag)s
[8242]314""" % { 'ctag' : ctag, 'date' : time.ctime(), 'host' : socket.gethostname() }
315
[8257]316
317
[8242]318def parseaddr(s):
[8257]319 """ Process IPv4 CIDR notation addr to a (binary) number """
[8242]320 f = s.split('.')
321 return (long(f[0]) << 24L) + \
322 (long(f[1]) << 16L) + \
323 (long(f[2]) << 8L) + \
324 long(f[3])
325
[8257]326
327
[8242]328def showaddr(a):
[8257]329 """ Display IPv4 addr in (dotted) CIDR notation """
[8242]330 return "%d.%d.%d.%d" % ((a >> 24) & 0xff, (a >> 16) & 0xff, (a >> 8) & 0xff, a & 0xff)
331
[8257]332
[8584]333def is_member(ip, mask, canidate):
334 """ Return True if canidate is part of ip/mask block"""
335 ip_addr = gformat.parseaddr(ip)
336 ip_canidate = gformat.parseaddr(canidate)
337 mask = int(mask)
338 ip_addr = ip_addr & ~((1 << (32 - mask)) - 1)
339 ip_canidate = ip_canidate & ~((1 << (32 - mask)) - 1)
340 return ip_addr == ip_canidate
[8257]341
[8584]342
343
[10410]344def cidr2netmask(netmask):
[8257]345 """ Given a 'netmask' return corresponding CIDR """
[8242]346 return showaddr(0xffffffff & (0xffffffff << (32 - int(netmask))))
347
[10410]348def get_network(addr, mask):
349 return showaddr(parseaddr(addr) & ~((1 << (32 - int(mask))) - 1))
[8257]350
351
[10410]352def generate_dhcpd_conf(datadump):
353 """ Generate config file '/usr/local/etc/dhcpd.conf """
354 output = generate_header()
355 output += Template("""\
356# option definitions common to all supported networks...
357option domain-name "dhcp.{{ autogen_fqdn }}";
358
359default-lease-time 600;
360max-lease-time 7200;
361
362# Use this to enble / disable dynamic dns updates globally.
363#ddns-update-style none;
364
365# If this DHCP server is the official DHCP server for the local
366# network, the authoritative directive should be uncommented.
367authoritative;
368
369# Use this to send dhcp log messages to a different log file (you also
370# have to hack syslog.conf to complete the redirection).
371log-facility local7;
372
373#
374# Interface definitions
375#
376\n""").render(datadump)
377
378 for iface_key in datadump['autogen_iface_keys']:
379 if not datadump[iface_key].has_key('comment'):
[10455]380 datadump[iface_key]['comment'] = None
[10410]381 output += "## %(interface)s - %(desc)s - %(comment)s\n" % datadump[iface_key]
382
383 (addr, mask) = datadump[iface_key]['ip'].split('/')
384 datadump[iface_key]['addr'] = addr
385 datadump[iface_key]['netmask'] = cidr2netmask(mask)
386 datadump[iface_key]['subnet'] = get_network(addr, mask)
387 try:
388 (dhcp_start, dhcp_stop) = datadump[iface_key]['dhcp'].split('-')
389 except (AttributeError, ValueError):
390 output += "subnet %(subnet)s netmask %(netmask)s {\n ### not autoritive\n}\n\n" % datadump[iface_key]
391 continue
392
393 dhcp_part = ".".join(addr.split('.')[0:3])
394 datadump[iface_key]['dhcp_start'] = dhcp_part + "." + dhcp_start
395 datadump[iface_key]['dhcp_stop'] = dhcp_part + "." + dhcp_stop
396 output += """\
397subnet %(subnet)s netmask %(netmask)s {
398 range %(dhcp_start)s %(dhcp_stop)s;
399 option routers %(addr)s;
400 option domain-name-servers %(addr)s;
401}
402\n""" % datadump[iface_key]
403
404 return output
405
406
407
[8242]408def generate_dnsmasq_conf(datadump):
[8257]409 """ Generate configuration file '/usr/local/etc/dnsmasq.conf' """
[8242]410 output = generate_header()
[10368]411 output += Template("""\
[9283]412# DHCP server options
[8242]413dhcp-authoritative
414dhcp-fqdn
[10391]415domain=dhcp.{{ autogen_fqdn }}
[8242]416domain-needed
417expand-hosts
[10120]418log-async=100
[8242]419
420# Low memory footprint
421cache-size=10000
422
[10368]423\n""").render(datadump)
424
[10281]425 for iface_key in datadump['autogen_iface_keys']:
[8262]426 if not datadump[iface_key].has_key('comment'):
[10455]427 datadump[iface_key]['comment'] = None
[8262]428 output += "## %(interface)s - %(desc)s - %(comment)s\n" % datadump[iface_key]
[8242]429
430 try:
[8257]431 (dhcp_start, dhcp_stop) = datadump[iface_key]['dhcp'].split('-')
[10410]432 (ip, cidr) = datadump[iface_key]['ip'].split('/')
433 datadump[iface_key]['netmask'] = cidr2netmask(cidr)
[8262]434 except (AttributeError, ValueError):
[8242]435 output += "# not autoritive\n\n"
436 continue
437
438 dhcp_part = ".".join(ip.split('.')[0:3])
439 datadump[iface_key]['dhcp_start'] = dhcp_part + "." + dhcp_start
440 datadump[iface_key]['dhcp_stop'] = dhcp_part + "." + dhcp_stop
[10410]441 output += "dhcp-range=%(interface)s,%(dhcp_start)s,%(dhcp_stop)s,%(netmask)s,24h\n\n" % datadump[iface_key]
[9283]442
[8242]443 return output
444
[8257]445
446
[8242]447def generate_rc_conf_local(datadump):
[8257]448 """ Generate configuration file '/etc/rc.conf.local' """
[10455]449 if not datadump.has_key('ileiden'):
450 datadump['autogen_ileiden_enable'] = False
451 else:
452 datadump['autogen_ileiden_enable'] = datadump['ileiden']
[10110]453
[10547]454 datadump['autogen_ileiden_enable'] = switchFormat(datadump['autogen_ileiden_enable'])
455
[10112]456 ileiden_proxies = []
[10367]457 normal_proxies = []
[10112]458 for proxy in get_proxylist():
459 proxydump = get_yaml(proxy)
460 if proxydump['ileiden']:
461 ileiden_proxies.append(proxydump)
[10367]462 else:
463 normal_proxies.append(proxydump)
[10461]464 for host in get_hybridlist():
465 hostdump = get_yaml(host)
[10584]466 if hostdump['service_proxy_ileiden']:
[10461]467 ileiden_proxies.append(hostdump)
[10584]468 if hostdump['service_proxy_normal']:
[10461]469 normal_proxies.append(hostdump)
470
[10585]471 datadump['autogen_ileiden_proxies'] = ileiden_proxies
472 datadump['autogen_normal_proxies'] = normal_proxies
473 datadump['autogen_ileiden_proxies_ips'] = ','.join([x['masterip'] for x in ileiden_proxies])
[10112]474 datadump['autogen_ileiden_proxies_names'] = ','.join([x['autogen_item'] for x in ileiden_proxies])
[10585]475 datadump['autogen_normal_proxies_ips'] = ','.join([x['masterip'] for x in normal_proxies])
[10367]476 datadump['autogen_normal_proxies_names'] = ','.join([x['autogen_item'] for x in normal_proxies])
[10112]477
[8242]478 output = generate_header("#");
[10584]479 output += render_template(datadump, """\
[10391]480hostname='{{ autogen_fqdn }}'
[10110]481location='{{ location }}'
482nodetype="{{ nodetype }}"
[9283]483
[10459]484#
485# Configured listings
486#
487captive_portal_whitelist=""
488{% if nodetype == "Proxy" %}
[10054]489#
[10459]490# Proxy Configuration
[10054]491#
[10110]492{% if gateway -%}
493defaultrouter="{{ gateway }}"
494{% else -%}
495#defaultrouter="NOTSET"
496{% endif -%}
497internalif="{{ internalif }}"
[10112]498ileiden_enable="{{ autogen_ileiden_enable }}"
499gateway_enable="{{ autogen_ileiden_enable }}"
[10238]500pf_enable="yes"
[10302]501pf_rules="/etc/pf.conf"
[10455]502{% if autogen_ileiden_enable -%}
[10234]503pf_flags="-D ext_if={{ externalif }} -D int_if={{ internalif }} -D publicnat={80,443}"
[10238]504lvrouted_enable="{{ autogen_ileiden_enable }}"
505lvrouted_flags="-u -s s00p3rs3kr3t -m 28"
506{% else -%}
507pf_flags="-D ext_if={{ externalif }} -D int_if={{ internalif }} -D publicnat={0}"
[10310]508{% endif -%}
[10238]509{% if internalroute -%}
510static_routes="wleiden"
511route_wleiden="-net 172.16.0.0/12 {{ internalroute }}"
[10110]512{% endif -%}
[10054]513
[10584]514{% elif nodetype == "Hybrid" %}
515 #
516 # Hybrid Configuration
517 #
[10599]518 list_ileiden_proxies="
[10585]519 {% for item in autogen_ileiden_proxies -%}
520 {{ "%-16s"|format(item.masterip) }} # {{ item.autogen_realname }}
521 {% endfor -%}
[10599]522 "
523 list_normal_proxies="
[10585]524 {% for item in autogen_normal_proxies -%}
525 {{ "%-16s"|format(item.masterip) }} # {{ item.autogen_realname }}
526 {% endfor -%}
[10599]527 "
[10585]528
[10584]529 captive_portal_interfaces="{{ autogen_dhcp_interfaces|default('none', true) }}"
530 externalif="{{ externalif|default('vr0', true) }}"
531 masterip="{{ masterip }}"
532
533 # Defined services
534 service_proxy_ileiden="{{ service_proxy_ileiden|yesorno }}"
535 service_proxy_normal="{{ service_proxy_normal|yesorno }}"
536 service_accesspoint="{{ service_accesspoint|yesorno }}"
537 #
[10459]538
[10587]539 {% if service_proxy_ileiden %}
[10584]540 pf_rules="/etc/pf.hybrid.conf"
541 pf_flags="-D ext_if=$externalif -D ext_if_net=$externalif:network -D masterip=$masterip"
[10587]542 pf_flags="$pf_flags -D publicnat=80,443"
543 {% elif service_proxy_normal %}
[10649]544 pf_rules="/etc/pf.hybrid.conf"
[10587]545 pf_flags="-D ext_if=$externalif -D ext_if_net=$externalif:network -D masterip=$masterip"
[10649]546 pf_flags="$pf_flags -D publicnat=0"
[10599]547 lvrouted_flags="$lvrouted_flags -z `make_list "$list_ileiden_proxies" ","`"
[10649]548 named_setfib="1"
549 tinyproxy_setfib="1"
550 dnsmasq_setfib="1"
[10698]551 sshd_setfib="1"
[10584]552 {% else %}
553 pf_rules="/etc/pf.node.conf"
[10587]554 pf_flags=""
[10584]555 {% endif %}
[10459]556
[10584]557 {% if service_proxy_normal %}
558 tinyproxy_enable="yes"
559 {% else %}
560 pen_wrapper_enable="yes"
561 {% endif %}
[10460]562
[10584]563 {% if service_accesspoint %}
564 pf_flags="$pf_flags -D captive_portal_interfaces=$captive_portal_interfaces"
565 {% endif %}
[10459]566
[10584]567 {% if board == "ALIX2" %}
568 #
569 # ''Fat'' configuration, board has 256MB RAM
570 #
571 dnsmasq_enable="NO"
572 named_enable="YES"
573 dhcpd_enable="YES"
574 {% endif -%}
[10459]575
[10584]576 {% if service_proxy_ileiden and gateway %}
577 defaultrouter="{{ gateway }}"
578 {% endif %}
579{% elif nodetype == "CNode" %}
[10459]580#
[10054]581# NODE iLeiden Configuration
[10112]582#
[10585]583
584# iLeiden Proxies {{ autogen_ileiden_proxies_names }}
585list_ileiden_proxies="{{ autogen_ileiden_proxies_ips }}"
586# normal Proxies {{ autogen_normal_proxies_names }}
587list_normal_proxies="{{ autogen_normal_proxies_ips }}"
588
[10564]589captive_portal_interfaces="{{ autogen_dhcp_interfaces }}"
[10367]590
591lvrouted_flags="-u -s s00p3rs3kr3t -m 28 -z $list_ileiden_proxies"
[10110]592{% endif %}
593
[10584]594#
595# Interface definitions
596#\n
597""")
598
[8242]599 # lo0 configuration:
600 # - 172.32.255.1/32 is the proxy.wleiden.net deflector
[9283]601 # - masterip is special as it needs to be assigned to at
[8242]602 # least one interface, so if not used assign to lo0
[9808]603 addrs_list = { 'lo0' : [("127.0.0.1/8", "LocalHost"), ("172.31.255.1/32","Proxy IP")] }
[9283]604 iface_map = {'lo0' : 'lo0'}
[10366]605 dhclient_if = {'lo0' : False}
[8242]606
[8297]607 masterip_used = False
[10281]608 for iface_key in datadump['autogen_iface_keys']:
[8297]609 if datadump[iface_key]['ip'].startswith(datadump['masterip']):
610 masterip_used = True
611 break
[9283]612 if not masterip_used:
[10108]613 addrs_list['lo0'].append((datadump['masterip'] + "/32", 'Master IP Not used in interface'))
[8297]614
[10281]615 for iface_key in datadump['autogen_iface_keys']:
[8242]616 ifacedump = datadump[iface_key]
[10162]617 ifname = ifacedump['autogen_ifname']
[8242]618
[10366]619 # Flag dhclient is possible
620 dhclient_if[ifname] = ifacedump.has_key('dhcpclient') and ifacedump['dhcpclient']
[10318]621
[8242]622 # Add interface IP to list
[9808]623 item = (ifacedump['ip'], ifacedump['desc'])
[10162]624 if addrs_list.has_key(ifname):
625 addrs_list[ifname].append(item)
[8242]626 else:
[10162]627 addrs_list[ifname] = [item]
[8242]628
629 # Alias only needs IP assignment for now, this might change if we
630 # are going to use virtual accesspoints
631 if "alias" in iface_key:
632 continue
633
634 # XXX: Might want to deduct type directly from interface name
635 if ifacedump['type'] in ['11a', '11b', '11g', 'wireless']:
636 # Default to station (client) mode
637 ifacedump['wlanmode'] = "sta"
[10166]638 if ifacedump['mode'] in ['master', 'master-wds', 'ap', 'ap-wds']:
[8242]639 ifacedump['wlanmode'] = "ap"
640 # Default to 802.11b mode
641 ifacedump['mode'] = '11b'
642 if ifacedump['type'] in ['11a', '11b' '11g']:
[9283]643 ifacedump['mode'] = ifacedump['type']
[8242]644
645 if not ifacedump.has_key('channel'):
646 if ifacedump['type'] == '11a':
647 ifacedump['channel'] = 36
648 else:
649 ifacedump['channel'] = 1
650
651 # Allow special hacks at the back like wds and stuff
652 if not ifacedump.has_key('extra'):
653 ifacedump['extra'] = 'regdomain ETSI country NL'
654
[10054]655 output += "wlans_%(interface)s='%(autogen_ifname)s'\n" % ifacedump
656 output += ("create_args_%(autogen_ifname)s='wlanmode %(wlanmode)s mode " +\
[8274]657 "%(mode)s ssid %(ssid)s %(extra)s channel %(channel)s'\n") % ifacedump
[9283]658
[8242]659 elif ifacedump['type'] in ['ethernet', 'eth']:
660 # No special config needed besides IP
661 pass
662 else:
663 assert False, "Unknown type " + ifacedump['type']
664
[9283]665 # Print IP address which needs to be assigned over here
[8242]666 output += "\n"
667 for iface,addrs in sorted(addrs_list.iteritems()):
[10079]668 for addr, comment in sorted(addrs,key=lambda x: parseaddr(x[0].split('/')[0])):
[9808]669 output += "# %s || %s || %s\n" % (iface, addr, comment)
[8242]670
[10366]671 # Write DHCLIENT entry
672 if dhclient_if[iface]:
673 output += "ifconfig_%s='SYNCDHCP'\n\n" % (iface)
674 else:
675 output += "ipv4_addrs_%s='%s'\n\n" % (iface, " ".join([x[0] for x in addrs]))
676
[8242]677 return output
678
[8257]679
680
[8242]681
[8317]682def get_all_configs():
683 """ Get dict with key 'host' with all configs present """
684 configs = dict()
685 for host in get_hostlist():
686 datadump = get_yaml(host)
687 configs[host] = datadump
688 return configs
689
690
[8319]691def get_interface_keys(config):
692 """ Quick hack to get all interface keys, later stage convert this to a iterator """
[10054]693 return sorted([elem for elem in config.keys() if (elem.startswith('iface_') and not "lo0" in elem)])
[8317]694
[8319]695
[8317]696def get_used_ips(configs):
697 """ Return array of all IPs used in config files"""
698 ip_list = []
[8319]699 for config in configs:
[8317]700 ip_list.append(config['masterip'])
[8319]701 for iface_key in get_interface_keys(config):
[8317]702 l = config[iface_key]['ip']
703 addr, mask = l.split('/')
704 # Special case do not process
[8332]705 if valid_addr(addr):
706 ip_list.append(addr)
707 else:
[9728]708 logger.error("## IP '%s' in '%s' not valid" % (addr, config['nodename']))
[8317]709 return sorted(ip_list)
710
711
712
[8242]713def generate_resolv_conf(datadump):
[8257]714 """ Generate configuration file '/etc/resolv.conf' """
[10468]715 # XXX: This should properly going to be an datastructure soon
716 datadump['autogen_header'] = generate_header("#")
717 datadump['autogen_edge_nameservers'] = ''
718 for host in get_proxylist():
719 hostdump = get_yaml(host)
720 datadump['autogen_edge_nameservers'] += "nameserver %(masterip)-15s # %(autogen_realname)s\n" % hostdump
721 for host in get_hybridlist():
722 hostdump = get_yaml(host)
[10584]723 if hostdump['service_proxy_ileiden'] or hostdump['service_proxy_normal']:
[10468]724 datadump['autogen_edge_nameservers'] += "nameserver %(masterip)-15s # %(autogen_realname)s\n" % hostdump
725
726 return Template("""\
727{{ autogen_header }}
[8242]728search wleiden.net
[10468]729
730# Try local (cache) first
[10209]731nameserver 127.0.0.1
[10468]732
[10584]733{% if service_proxy_normal or service_proxy_ileiden or nodetype == 'Proxy' -%}
[10053]734nameserver 8.8.8.8 # Google Public NameServer
735nameserver 8.8.4.4 # Google Public NameServer
[10468]736{% else -%}
[10646]737# START DYNAMIC LIST - updated by /tools/nameserver-shuffle
[10468]738{{ autogen_edge_nameservers }}
739{% endif -%}
740""").render(datadump)
[10209]741
[9283]742
[8242]743
[10654]744def generate_ntp_conf(datadump):
745 """ Generate configuration file '/etc/ntp.conf' """
746 # XXX: This should properly going to be an datastructure soon
747
748 datadump['autogen_header'] = generate_header("#")
749 datadump['autogen_ntp_servers'] = ''
750 for host in get_proxylist():
751 hostdump = get_yaml(host)
752 datadump['autogen_ntp_servers'] += "server %(masterip)-15s iburst maxpoll 9 # %(autogen_realname)s\n" % hostdump
753 for host in get_hybridlist():
754 hostdump = get_yaml(host)
755 if hostdump['service_proxy_ileiden'] or hostdump['service_proxy_normal']:
756 datadump['autogen_ntp_servers'] += "server %(masterip)-15s iburst maxpoll 9 # %(autogen_realname)s\n" % hostdump
757
758 return Template("""\
759{{ autogen_header }}
760
761{% if service_proxy_normal or service_proxy_ileiden or nodetype == 'Proxy' -%}
762# Machine hooked to internet.
763server 0.nl.pool.ntp.org iburst maxpoll 9
764server 1.nl.pool.ntp.org iburst maxpoll 9
765server 2.nl.pool.ntp.org iburst maxpoll 9
766server 3.nl.pool.ntp.org iburst maxpoll 9
767{% else -%}
768# Local Wireless Leiden NTP Servers.
769server 0.pool.ntp.wleiden.net iburst maxpoll 9
770server 1.pool.ntp.wleiden.net iburst maxpoll 9
771server 2.pool.ntp.wleiden.net iburst maxpoll 9
772server 3.pool.ntp.wleiden.net iburst maxpoll 9
773
774# All the configured NTP servers
775{{ autogen_ntp_servers }}
776{% endif %}
777
778# If a server loses sync with all upstream servers, NTP clients
779# no longer follow that server. The local clock can be configured
780# to provide a time source when this happens, but it should usually
781# be configured on just one server on a network. For more details see
782# http://support.ntp.org/bin/view/Support/UndisciplinedLocalClock
783# The use of Orphan Mode may be preferable.
784#
785server 127.127.1.0
786fudge 127.127.1.0 stratum 10
787""").render(datadump)
788
789
[10705]790def generate_pf_hybrid_conf_local(datadump):
791 """ Generate configuration file '/etc/pf.hybrid.conf.local' """
792 datadump['autogen_header'] = generate_header("#")
793 datadump['rdr_rules' ] = ('8081', '172.16.4.46', 'http')
794 return Template("""\
795{{ autogen_header }}
[10654]796
[10705]797# Redirect some internal facing services outside (7)
798# rdr on $ext_if inet proto tcp from any to $ext_if port 8081 tag SRV -> 172.16.4.46 port http
799# XXX: Implement
800""").render(datadump)
801
[10069]802def generate_motd(datadump):
803 """ Generate configuration file '/etc/motd' """
[10568]804 output = Template("""\
[10627]805FreeBSD run ``service motd onestart'' to make me look normal
[8242]806
[10568]807 WWW: {{ autogen_fqdn }} - http://www.wirelessleiden.nl
808 Loc: {{ location }}
[8257]809
[10568]810Services:
811{% if board == "ALIX2" -%}
[10665]812 - Core Node ({{ board }})
[10568]813{% else -%}
[10665]814 - Hulp Node ({{ board }})
[10568]815{% endif -%}
[10584]816{% if service_proxy_normal -%}
[10568]817 - Normal Proxy
818{% endif -%}
[10584]819{% if service_proxy_ileiden -%}
[10568]820 - iLeiden Proxy
821{% endif %}
[10626]822Interlinks:\n
[10568]823""").render(datadump)
[10069]824
825 # XXX: This is a hacky way to get the required data
826 for line in generate_rc_conf_local(datadump).split('\n'):
827 if '||' in line and not line[1:].split()[0] in ['lo0', 'ath0'] :
828 output += " - %s \n" % line[1:]
829 output += """\
830Attached bridges:
831"""
832 for iface_key in datadump['autogen_iface_keys']:
833 ifacedump = datadump[iface_key]
834 if ifacedump.has_key('ns_ip'):
835 output += " - %(interface)s || %(mode)s || %(ns_ip)s\n" % ifacedump
836
837 return output
838
839
[8267]840def format_yaml_value(value):
841 """ Get yaml value in right syntax for outputting """
842 if isinstance(value,str):
[10049]843 output = '"%s"' % value
[8267]844 else:
845 output = value
[9283]846 return output
[8267]847
848
849
850def format_wleiden_yaml(datadump):
[8242]851 """ Special formatting to ensure it is editable"""
[9283]852 output = "# Genesis config yaml style\n"
[8262]853 output += "# vim:ts=2:et:sw=2:ai\n"
[8242]854 output += "#\n"
855 iface_keys = [elem for elem in datadump.keys() if elem.startswith('iface_')]
856 for key in sorted(set(datadump.keys()) - set(iface_keys)):
[8267]857 output += "%-10s: %s\n" % (key, format_yaml_value(datadump[key]))
[9283]858
[8242]859 output += "\n\n"
[9283]860
[8272]861 key_order = [ 'comment', 'interface', 'ip', 'desc', 'sdesc', 'mode', 'type',
862 'extra_type', 'channel', 'ssid', 'dhcp' ]
863
[8242]864 for iface_key in sorted(iface_keys):
865 output += "%s:\n" % iface_key
[8272]866 for key in key_order + list(sorted(set(datadump[iface_key].keys()) - set(key_order))):
867 if datadump[iface_key].has_key(key):
[9283]868 output += " %-11s: %s\n" % (key, format_yaml_value(datadump[iface_key][key]))
[8242]869 output += "\n\n"
870
871 return output
872
873
[8257]874
[10067]875def generate_wleiden_yaml(datadump, header=True):
[8267]876 """ Generate (petty) version of wleiden.yaml"""
[10053]877 for key in datadump.keys():
878 if key.startswith('autogen_'):
879 del datadump[key]
[10054]880 # Interface autogen cleanups
881 elif type(datadump[key]) == dict:
882 for key2 in datadump[key].keys():
883 if key2.startswith('autogen_'):
884 del datadump[key][key2]
885
[10067]886 output = generate_header("#") if header else ''
[8267]887 output += format_wleiden_yaml(datadump)
888 return output
889
890
[8588]891def generate_yaml(datadump):
892 return generate_config(datadump['nodename'], "wleiden.yaml", datadump)
[8267]893
[8588]894
[9283]895
[8298]896def generate_config(node, config, datadump=None):
[8257]897 """ Print configuration file 'config' of 'node' """
[8267]898 output = ""
[8242]899 try:
900 # Load config file
[8298]901 if datadump == None:
902 datadump = get_yaml(node)
[9283]903
[8242]904 if config == 'wleiden.yaml':
[8267]905 output += generate_wleiden_yaml(datadump)
906 elif config == 'authorized_keys':
[10051]907 f = open(os.path.join(NODE_DIR,"global_keys"), 'r')
[8267]908 output += f.read()
[8242]909 f.close()
910 elif config == 'dnsmasq.conf':
[10281]911 output += generate_dnsmasq_conf(datadump)
[10410]912 elif config == 'dhcpd.conf':
913 output += generate_dhcpd_conf(datadump)
[8242]914 elif config == 'rc.conf.local':
[10281]915 output += generate_rc_conf_local(datadump)
[8242]916 elif config == 'resolv.conf':
[10281]917 output += generate_resolv_conf(datadump)
[10654]918 elif config == 'ntp.conf':
919 output += generate_ntp_conf(datadump)
[10069]920 elif config == 'motd':
[10281]921 output += generate_motd(datadump)
[10705]922 elif config == 'pf.hybrid.conf.local':
923 output += generate_pf_hybrid_conf_local(datadump)
[8242]924 else:
[9283]925 assert False, "Config not found!"
[8242]926 except IOError, e:
[8267]927 output += "[ERROR] Config file not found"
928 return output
[8242]929
930
[8257]931
[8258]932def process_cgi_request():
933 """ When calling from CGI """
934 # Update repository if requested
935 form = cgi.FieldStorage()
936 if form.getvalue("action") == "update":
[8259]937 print "Refresh: 5; url=."
[8258]938 print "Content-type:text/plain\r\n\r\n",
939 print "[INFO] Updating subverion, please wait..."
[10143]940 print subprocess.Popen(['svn', 'cleanup', "%s/.." % NODE_DIR], stderr=subprocess.STDOUT, stdout=subprocess.PIPE).communicate()[0],
[10071]941 print subprocess.Popen(['svn', 'up', "%s/.." % NODE_DIR], stderr=subprocess.STDOUT, stdout=subprocess.PIPE).communicate()[0],
[8258]942 print "[INFO] All done, redirecting in 5 seconds"
943 sys.exit(0)
[9283]944
945
[10270]946 base_uri = os.environ['PATH_INFO']
947 uri = base_uri.strip('/').split('/')
948
[10681]949 output = "Template Holder"
950 content_type='text/plain'
[10378]951 if base_uri.endswith('/create/network.kml'):
[10681]952 content_type='application/vnd.google-earth.kml+xml'
953 output = make_network_kml.make_graph()
[10378]954 elif not uri[0]:
[10070]955 if is_text_request():
[10681]956 content_type = 'text/plain'
957 output = '\n'.join(get_hostlist())
[10060]958 else:
[10681]959 content_type = 'text/html'
960 output = generate_title(get_hostlist())
[8258]961 elif len(uri) == 1:
[10270]962 if is_text_request():
[10681]963 content_type = 'text/plain'
964 output = generate_node(uri[0])
[10270]965 else:
[10681]966 content_type = 'text/html'
967 output = generate_node_overview(uri[0])
[8258]968 elif len(uri) == 2:
[10681]969 content_type = 'text/plain'
970 output = generate_config(uri[0], uri[1])
[8258]971 else:
972 assert False, "Invalid option"
[10681]973
974 print "Content-Type: %s" % content_type
975 print "Content-Length: %s" % len(output)
976 print ""
[8267]977 print output
[8242]978
[10391]979def get_realname(datadump):
[10365]980 # Proxy naming convention is special, as the proxy name is also included in
981 # the nodename, when it comes to the numbered proxies.
[8588]982 if datadump['nodetype'] == 'Proxy':
[10391]983 realname = datadump['nodetype'] + datadump['nodename'].replace('proxy','')
[8588]984 else:
985 # By default the full name is listed and also a shortname CNAME for easy use.
[10391]986 realname = datadump['nodetype'] + datadump['nodename']
987 return(realname)
[8259]988
[9283]989
990
[10264]991def make_dns(output_dir = 'dns', external = False):
[8588]992 items = dict()
[8598]993
[8588]994 # hostname is key, IP is value
[10642]995 wleiden_zone = defaultdict(list)
[8588]996 wleiden_cname = dict()
[8598]997
[8588]998 pool = dict()
999 for node in get_hostlist():
1000 datadump = get_yaml(node)
[9283]1001
[8588]1002 # Proxy naming convention is special
[10391]1003 fqdn = datadump['autogen_realname']
[10461]1004 if datadump['nodetype'] in ['CNode', 'Hybrid']:
[8588]1005 wleiden_cname[datadump['nodename']] = fqdn
1006
[10655]1007 wleiden_zone[fqdn].append((datadump['masterip'], True))
[8588]1008
[8598]1009 # Hacking to get proper DHCP IPs and hostnames
[8588]1010 for iface_key in get_interface_keys(datadump):
[8598]1011 iface_name = datadump[iface_key]['interface'].replace(':',"-alias-")
[10410]1012 (ip, cidr) = datadump[iface_key]['ip'].split('/')
[8588]1013 try:
1014 (dhcp_start, dhcp_stop) = datadump[iface_key]['dhcp'].split('-')
[10410]1015 datadump[iface_key]['netmask'] = cidr2netmask(cidr)
[8588]1016 dhcp_part = ".".join(ip.split('.')[0:3])
1017 if ip != datadump['masterip']:
[10655]1018 wleiden_zone["dhcp-gateway-%s.%s" % (iface_name, fqdn)].append((ip, False))
[8588]1019 for i in range(int(dhcp_start), int(dhcp_stop) + 1):
[10655]1020 wleiden_zone["dhcp-%s-%s.%s" % (i, iface_name, fqdn)].append(("%s.%s" % (dhcp_part, i), True))
[8588]1021 except (AttributeError, ValueError):
1022 # First push it into a pool, to indentify the counter-part later on
1023 addr = parseaddr(ip)
[10461]1024 cidr = int(cidr)
1025 addr = addr & ~((1 << (32 - cidr)) - 1)
[9283]1026 if pool.has_key(addr):
[8588]1027 pool[addr] += [(iface_name, fqdn, ip)]
[9283]1028 else:
[8588]1029 pool[addr] = [(iface_name, fqdn, ip)]
1030 continue
1031
[9286]1032
[10656]1033 def pool_to_name(fqdn, pool_members):
[9286]1034 """Convert the joined name to a usable pool name"""
1035
[10656]1036 def isplit(item):
1037 (prefix, name, number) = re.match('^(cnode|hybrid|proxy)([a-z]+)([0-9]*)$',item.lower()).group(1,2,3)
1038 return (prefix, name, number)
1039
1040 my_name = isplit(fqdn.split('.')[0])[1]
[9286]1041
[10656]1042 short_names = defaultdict(list)
[9286]1043 for node in sorted(pool_members):
[10656]1044 (prefix, name, number) = isplit(node)
1045 short_names[name].append((prefix,number))
[9286]1046
[10656]1047 return '-'.join(sorted(short_names.keys()))
[9286]1048
1049
[9957]1050 # WL uses an /29 to configure an interface. IP's are ordered like this:
[9958]1051 # MasterA (.1) -- DeviceA (.2) <<>> DeviceB (.3) --- SlaveB (.4)
[9957]1052
1053 sn = lambda x: re.sub(r'(?i)^cnode','',x)
1054
[8598]1055 # Automatic naming convention of interlinks namely 2 + remote.lower()
[8588]1056 for (key,value) in pool.iteritems():
[9958]1057 # Make sure they are sorted from low-ip to high-ip
1058 value = sorted(value, key=lambda x: parseaddr(x[2]))
1059
[8588]1060 if len(value) == 1:
1061 (iface_name, fqdn, ip) = value[0]
[10655]1062 wleiden_zone["2unused-%s.%s" % (iface_name, fqdn)].append((ip, True))
[9957]1063
1064 # Device DNS names
1065 if 'cnode' in fqdn.lower():
[10655]1066 wleiden_zone["d-at-%s.%s" % (iface_name, fqdn)].append((showaddr(parseaddr(ip) + 1), False))
1067 wleiden_cname["d-at-%s.%s" % (iface_name,sn(fqdn))] = "d-at-%s.%s" % ((iface_name, fqdn))
[9957]1068
[8588]1069 elif len(value) == 2:
1070 (a_iface_name, a_fqdn, a_ip) = value[0]
1071 (b_iface_name, b_fqdn, b_ip) = value[1]
[10655]1072 wleiden_zone["2%s.%s" % (b_fqdn,a_fqdn)].append((a_ip, True))
1073 wleiden_zone["2%s.%s" % (a_fqdn,b_fqdn)].append((b_ip, True))
[9957]1074
1075 # Device DNS names
1076 if 'cnode' in a_fqdn.lower() and 'cnode' in b_fqdn.lower():
[10655]1077 wleiden_zone["d-at-%s.%s" % (a_iface_name, a_fqdn)].append((showaddr(parseaddr(a_ip) + 1), False))
1078 wleiden_zone["d-at-%s.%s" % (b_iface_name, b_fqdn)].append((showaddr(parseaddr(b_ip) - 1), False))
[9957]1079 wleiden_cname["d-at-%s.%s" % (a_iface_name,sn(a_fqdn))] = "d-at-%s.%s" % (a_iface_name, a_fqdn)
1080 wleiden_cname["d-at-%s.%s" % (b_iface_name,sn(b_fqdn))] = "d-at-%s.%s" % (b_iface_name, b_fqdn)
1081 wleiden_cname["d2%s.%s" % (sn(b_fqdn),sn(a_fqdn))] = "d-at-%s.%s" % (a_iface_name, a_fqdn)
1082 wleiden_cname["d2%s.%s" % (sn(a_fqdn),sn(b_fqdn))] = "d-at-%s.%s" % (b_iface_name, b_fqdn)
1083
[8588]1084 else:
1085 pool_members = [k[1] for k in value]
1086 for item in value:
[9283]1087 (iface_name, fqdn, ip) = item
[10656]1088 pool_name = "2pool-" + pool_to_name(fqdn,pool_members)
[10655]1089 wleiden_zone["%s.%s" % (pool_name, fqdn)].append((ip, True))
[8598]1090
1091 # Include static DNS entries
1092 # XXX: Should they override the autogenerated results?
1093 # XXX: Convert input to yaml more useable.
1094 # Format:
1095 ##; this is a comment
1096 ## roomburgh=CNodeRoomburgh1
1097 ## apkerk1.CNodeVosko=172.17.176.8 ;this as well
[10642]1098 dns_list = yaml.load(open(os.path.join(NODE_DIR,'../dns/staticDNS.yaml'),'r'))
[9938]1099
1100 # Hack to allow special entries, for development
[10642]1101 wleiden_raw = {}
[9938]1102
[10642]1103 for line in dns_list:
[10660]1104 reverse = False
[10642]1105 k, items = line.items()[0]
[10660]1106 if type(items) == dict:
1107 if items.has_key('reverse'):
1108 reverse = items['reverse']
1109 items = items['a']
1110 else:
1111 items = items['cname']
1112 items = [items] if type(items) != list else items
[10642]1113 for item in items:
1114 if item.startswith('IN '):
1115 wleiden_raw[k] = item
1116 elif valid_addr(item):
[10660]1117 wleiden_zone[k].append((item, reverse))
[8598]1118 else:
[10642]1119 wleiden_cname[k] = item
[9283]1120
[8598]1121 details = dict()
1122 # 24 updates a day allowed
1123 details['serial'] = time.strftime('%Y%m%d%H')
1124
[10264]1125 if external:
1126 dns_masters = ['siteview.wirelessleiden.nl', 'ns1.vanderzwet.net']
1127 else:
1128 dns_masters = ['sunny.wleiden.net']
1129
1130 details['master'] = dns_masters[0]
1131 details['ns_servers'] = '\n'.join(['\tNS\t%s.' % x for x in dns_masters])
1132
[8598]1133 dns_header = '''
1134$TTL 3h
[10659]1135%(zone)s. SOA %(master)s. beheer.lijst.wirelessleiden.nl. ( %(serial)s 1d 12h 1w 60s )
[8598]1136 ; Serial, Refresh, Retry, Expire, Neg. cache TTL
1137
[10264]1138%(ns_servers)s
[8598]1139 \n'''
1140
[9283]1141
[10264]1142 if not os.path.isdir(output_dir):
1143 os.makedirs(output_dir)
[8598]1144 details['zone'] = 'wleiden.net'
[9284]1145 f = open(os.path.join(output_dir,"db." + details['zone']), "w")
[8598]1146 f.write(dns_header % details)
1147
[10655]1148 for host,items in wleiden_zone.iteritems():
1149 for ip,reverse in items:
[10642]1150 if valid_addr(ip):
1151 f.write("%s.wleiden.net. IN A %s \n" % (host.lower(), ip))
[8588]1152 for source,dest in wleiden_cname.iteritems():
[8636]1153 f.write("%s.wleiden.net. IN CNAME %s.wleiden.net.\n" % (source.lower(), dest.lower()))
[9938]1154 for source, dest in wleiden_raw.iteritems():
1155 f.write("%s.wleiden.net. %s\n" % (source, dest))
[8588]1156 f.close()
[9283]1157
[8598]1158 # Create whole bunch of specific sub arpa zones. To keep it compliant
1159 for s in range(16,32):
1160 details['zone'] = '%i.172.in-addr.arpa' % s
[9284]1161 f = open(os.path.join(output_dir,"db." + details['zone']), "w")
[8598]1162 f.write(dns_header % details)
[8588]1163
[8598]1164 #XXX: Not effient, fix to proper data structure and do checks at other
1165 # stages
[10655]1166 for host,items in wleiden_zone.iteritems():
1167 for ip,reverse in items:
1168 if not reverse:
1169 continue
[10642]1170 if valid_addr(ip):
[10655]1171 if valid_addr(ip):
1172 if int(ip.split('.')[1]) == s:
1173 rev_ip = '.'.join(reversed(ip.split('.')))
1174 f.write("%s.in-addr.arpa. IN PTR %s.wleiden.net.\n" % (rev_ip.lower(), host.lower()))
[8598]1175 f.close()
[8588]1176
[8598]1177
[8259]1178def usage():
[10567]1179 print """Usage: %(prog)s <argument>
1180Argument:
1181\tstandalone [port] = Run configurator webserver [8000]
1182\tdns [outputdir] = Generate BIND compliant zone files in dns [./dns]
[9589]1183\tfull-export = Generate yaml export script for heatmap.
[10567]1184\tstatic [outputdir] = Generate all config files and store on disk
1185\t with format ./<outputdir>/%%NODE%%/%%FILE%% [./static]
1186\ttest <node> <file> = Receive output of CGI script.
1187\tlist <status> <items> = List systems which have certain status
[10563]1188
[10567]1189Arguments:
1190\t<node> = NodeName (example: HybridRick)
1191\t<file> = %(files)s
1192\t<status> = all|up|down|planned
1193\t<items> = systems|nodes|proxies
1194
[10563]1195NOTE FOR DEVELOPERS; you can test your changes like this:
1196 BEFORE any changes in this code:
1197 $ ./gformat.py static /tmp/pre
1198 AFTER the changes:
1199 $ ./gformat.py static /tmp/post
1200 VIEW differences and VERIFY all are OK:
[10564]1201 $ diff -urI 'Generated' -r /tmp/pre /tmp/post
[10567]1202""" % { 'prog' : sys.argv[0], 'files' : '|'.join(files) }
[8259]1203 exit(0)
1204
1205
[10070]1206def is_text_request():
[10107]1207 """ Find out whether we are calling from the CLI or any text based CLI utility """
1208 try:
1209 return os.environ['HTTP_USER_AGENT'].split()[0] in ['curl', 'fetch', 'wget']
1210 except KeyError:
1211 return True
[8259]1212
[10547]1213def switchFormat(setting):
1214 if setting:
1215 return "YES"
1216 else:
1217 return "NO"
1218
[8267]1219def main():
1220 """Hard working sub"""
1221 # Allow easy hacking using the CLI
1222 if not os.environ.has_key('PATH_INFO'):
1223 if len(sys.argv) < 2:
1224 usage()
[9283]1225
[8267]1226 if sys.argv[1] == "standalone":
1227 import SocketServer
1228 import CGIHTTPServer
[10105]1229 # Hop to the right working directory.
1230 os.chdir(os.path.dirname(__file__))
[8267]1231 try:
1232 PORT = int(sys.argv[2])
1233 except (IndexError,ValueError):
1234 PORT = 8000
[9283]1235
[8267]1236 class MyCGIHTTPRequestHandler(CGIHTTPServer.CGIHTTPRequestHandler):
1237 """ Serve this CGI from the root of the webserver """
1238 def is_cgi(self):
1239 if "favicon" in self.path:
1240 return False
[9283]1241
[10364]1242 self.cgi_info = (os.path.basename(__file__), self.path)
[8267]1243 self.path = ''
1244 return True
1245 handler = MyCGIHTTPRequestHandler
[9807]1246 SocketServer.TCPServer.allow_reuse_address = True
[8267]1247 httpd = SocketServer.TCPServer(("", PORT), handler)
1248 httpd.server_name = 'localhost'
1249 httpd.server_port = PORT
[9283]1250
[9728]1251 logger.info("serving at port %s", PORT)
[8860]1252 try:
1253 httpd.serve_forever()
1254 except KeyboardInterrupt:
1255 httpd.shutdown()
[9728]1256 logger.info("All done goodbye")
[8267]1257 elif sys.argv[1] == "test":
1258 os.environ['PATH_INFO'] = "/".join(sys.argv[2:])
1259 os.environ['SCRIPT_NAME'] = __file__
1260 process_cgi_request()
[8296]1261 elif sys.argv[1] == "static":
1262 items = dict()
[10563]1263 items['output_dir'] = sys.argv[2] if len(sys.argv) > 2 else "./static"
[8296]1264 for node in get_hostlist():
1265 items['node'] = node
[10563]1266 items['wdir'] = "%(output_dir)s/%(node)s" % items
[8296]1267 if not os.path.isdir(items['wdir']):
1268 os.makedirs(items['wdir'])
[8298]1269 datadump = get_yaml(node)
[8296]1270 for config in files:
1271 items['config'] = config
[9728]1272 logger.info("## Generating %(node)s %(config)s" % items)
[8296]1273 f = open("%(wdir)s/%(config)s" % items, "w")
[8298]1274 f.write(generate_config(node, config, datadump))
[8296]1275 f.close()
[9514]1276 elif sys.argv[1] == "wind-export":
1277 items = dict()
1278 for node in get_hostlist():
1279 datadump = get_yaml(node)
1280 sql = """INSERT IGNORE INTO nodes (name, name_ns, longitude, latitude)
1281 VALUES ('%(nodename)s', '%(nodename)s', %(latitude)s, %(longitude)s);""" % datadump;
1282 sql = """INSERT IGNORE INTO users_nodes (user_id, node_id, owner)
1283 VALUES (
1284 (SELECT id FROM users WHERE username = 'rvdzwet'),
1285 (SELECT id FROM nodes WHERE name = '%(nodename)s'),
1286 'Y');""" % datadump
1287 #for config in files:
1288 # items['config'] = config
1289 # print "## Generating %(node)s %(config)s" % items
1290 # f = open("%(wdir)s/%(config)s" % items, "w")
1291 # f.write(generate_config(node, config, datadump))
1292 # f.close()
1293 for node in get_hostlist():
1294 datadump = get_yaml(node)
1295 for iface_key in sorted([elem for elem in datadump.keys() if elem.startswith('iface_')]):
1296 ifacedump = datadump[iface_key]
1297 if ifacedump.has_key('mode') and ifacedump['mode'] == 'ap-wds':
1298 ifacedump['nodename'] = datadump['nodename']
1299 if not ifacedump.has_key('channel') or not ifacedump['channel']:
1300 ifacedump['channel'] = 0
1301 sql = """INSERT INTO links (node_id, type, ssid, protocol, channel, status)
1302 VALUES ((SELECT id FROM nodes WHERE name = '%(nodename)s'), 'ap',
1303 '%(ssid)s', 'IEEE 802.11b', %(channel)s, 'active');""" % ifacedump
[9589]1304 elif sys.argv[1] == "full-export":
1305 hosts = {}
1306 for node in get_hostlist():
1307 datadump = get_yaml(node)
1308 hosts[datadump['nodename']] = datadump
1309 print yaml.dump(hosts)
1310
[8584]1311 elif sys.argv[1] == "dns":
[10264]1312 make_dns(sys.argv[2] if len(sys.argv) > 2 else 'dns', 'external' in sys.argv)
[9283]1313 elif sys.argv[1] == "cleanup":
[8588]1314 # First generate all datadumps
1315 datadumps = dict()
1316 for host in get_hostlist():
[9728]1317 logger.info("# Processing: %s", host)
[10436]1318 # Set some boring default values
1319 datadump = { 'board' : 'UNKNOWN' }
1320 datadump.update(get_yaml(host))
[10391]1321 datadumps[datadump['autogen_realname']] = datadump
[9283]1322
[10455]1323
[10156]1324 for host,datadump in datadumps.iteritems():
[10455]1325 # Convert all yes and no to boolean values
1326 def fix_boolean(dump):
1327 for key in dump.keys():
1328 if type(dump[key]) == dict:
1329 dump[key] = fix_boolean(dump[key])
[10459]1330 elif str(dump[key]).lower() in ["yes", "true"]:
[10455]1331 dump[key] = True
[10459]1332 elif str(dump[key]).lower() in ["no", "false"]:
[10455]1333 # Compass richting no (Noord Oost) is valid input
[10459]1334 if key != "compass": dump[key] = False
[10455]1335 return dump
1336 datadump = fix_boolean(datadump)
1337
[10400]1338 if datadump['rdnap_x'] and datadump['rdnap_y']:
1339 datadump['latitude'], datadump['longitude'] = rdnap.rd2etrs(datadump['rdnap_x'], datadump['rdnap_y'])
1340 elif datadump['latitude'] and datadump['longitude']:
1341 datadump['rdnap_x'], datadump['rdnap_y'] = rdnap.etrs2rd(datadump['latitude'], datadump['longitude'])
1342
[10319]1343 if datadump['nodename'].startswith('Proxy'):
1344 datadump['nodename'] = datadump['nodename'].lower()
1345
[10156]1346 for iface_key in datadump['autogen_iface_keys']:
[10703]1347 # All our normal wireless cards are normal APs now
1348 if datadump[iface_key]['type'] in ['11a', '11b', '11g', 'wireless']:
1349 datadump[iface_key]['mode'] = 'ap'
[10156]1350 # Wireless Leiden SSID have an consistent lowercase/uppercase
1351 if datadump[iface_key].has_key('ssid'):
1352 ssid = datadump[iface_key]['ssid']
1353 prefix = 'ap-WirelessLeiden-'
1354 if ssid.lower().startswith(prefix.lower()):
1355 datadump[iface_key]['ssid'] = prefix + ssid[len(prefix)].upper() + ssid[len(prefix) + 1:]
[10162]1356 if datadump[iface_key].has_key('ns_ip') and not datadump[iface_key].has_key('mode'):
1357 datadump[iface_key]['mode'] = 'autogen-FIXME'
1358 if not datadump[iface_key].has_key('desc'):
1359 datadump[iface_key]['desc'] = 'autogen-FIXME'
[10074]1360 store_yaml(datadump)
[9971]1361 elif sys.argv[1] == "list":
[10611]1362 use_fqdn = False
[10567]1363 if len(sys.argv) < 4 or not sys.argv[2] in ["up", "down", "planned", "all"]:
1364 usage()
1365 if sys.argv[3] == "nodes":
[9971]1366 systems = get_nodelist()
[10567]1367 elif sys.argv[3] == "proxies":
[9971]1368 systems = get_proxylist()
[10567]1369 elif sys.argv[3] == "systems":
[10270]1370 systems = get_hostlist()
[9971]1371 else:
1372 usage()
[10611]1373 if len(sys.argv) > 4:
1374 if sys.argv[4] == "fqdn":
1375 use_fqdn = True
1376 else:
1377 usage()
1378
[9971]1379 for system in systems:
1380 datadump = get_yaml(system)
[10611]1381
1382 output = datadump['autogen_fqdn'] if use_fqdn else system
[10567]1383 if sys.argv[2] == "all":
[10611]1384 print output
[10567]1385 elif datadump['status'] == sys.argv[2]:
[10611]1386 print output
[10378]1387 elif sys.argv[1] == "create":
1388 if sys.argv[2] == "network.kml":
1389 print make_network_kml.make_graph()
1390 else:
1391 usage()
[9283]1392 usage()
1393 else:
[10070]1394 # Do not enable debugging for config requests as it highly clutters the output
1395 if not is_text_request():
1396 cgitb.enable()
[9283]1397 process_cgi_request()
1398
1399
1400if __name__ == "__main__":
1401 main()
Note: See TracBrowser for help on using the repository browser.