source: genesis/tools/gformat.py@ 10912

Last change on this file since 10912 was 10907, checked in by rick, 13 years ago

Make the motd output look more pretty.

While here: Sync up motd processing quite a bit, by removing the heavy duplicate call to
rc_conf_local.

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