source: genesis/tools/gformat.py@ 10937

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

Only the known up please.

  • Property svn:executable set to *
  • Property svn:keywords set to Id
File size: 54.2 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 10937 2012-05-20 01:21:02Z 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=dict()):
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
826nameservers_cache = []
827def get_nameservers():
828 if nameservers_cache:
829 return nameservers_cache
830
831 for host in get_hybridlist():
832 hostdump = get_yaml(host)
833 if hostdump['status'] == 'up' and (hostdump['service_proxy_ileiden'] or hostdump['service_proxy_normal']):
834 nameservers_cache.append((hostdump['masterip'], hostdump['autogen_realname']))
835 for host in get_proxylist():
836 hostdump = get_yaml(host)
837 if hostdump['status'] == 'up':
838 nameservers_cache.append((hostdump['masterip'], hostdump['autogen_realname']))
839
840 return nameservers_cache
841
842
843def generate_resolv_conf(datadump):
844 """ Generate configuration file '/etc/resolv.conf' """
845 # XXX: This should properly going to be an datastructure soon
846 datadump['autogen_header'] = generate_header(datadump, "#")
847 datadump['autogen_edge_nameservers'] = ''
848
849
850 for masterip,realname in get_nameservers():
851 datadump['autogen_edge_nameservers'] += "nameserver %-15s # %s\n" % (masterip, realname)
852
853 return Template("""\
854{{ autogen_header }}
855search wleiden.net
856
857# Try local (cache) first
858nameserver 127.0.0.1
859
860{% if service_proxy_normal or service_proxy_ileiden or nodetype == 'Proxy' -%}
861nameserver 8.8.8.8 # Google Public NameServer
862nameserver 8.8.4.4 # Google Public NameServer
863{% else -%}
864# START DYNAMIC LIST - updated by /tools/nameserver-shuffle
865{{ autogen_edge_nameservers }}
866{% endif -%}
867""").render(datadump)
868
869
870
871def generate_ntp_conf(datadump):
872 """ Generate configuration file '/etc/ntp.conf' """
873 # XXX: This should properly going to be an datastructure soon
874
875 datadump['autogen_header'] = generate_header(datadump, "#")
876 datadump['autogen_ntp_servers'] = ''
877 for host in get_proxylist():
878 hostdump = get_yaml(host)
879 datadump['autogen_ntp_servers'] += "server %(masterip)-15s iburst maxpoll 9 # %(autogen_realname)s\n" % hostdump
880 for host in get_hybridlist():
881 hostdump = get_yaml(host)
882 if hostdump['service_proxy_ileiden'] or hostdump['service_proxy_normal']:
883 datadump['autogen_ntp_servers'] += "server %(masterip)-15s iburst maxpoll 9 # %(autogen_realname)s\n" % hostdump
884
885 return Template("""\
886{{ autogen_header }}
887
888{% if service_proxy_normal or service_proxy_ileiden or nodetype == 'Proxy' -%}
889# Machine hooked to internet.
890server 0.nl.pool.ntp.org iburst maxpoll 9
891server 1.nl.pool.ntp.org iburst maxpoll 9
892server 2.nl.pool.ntp.org iburst maxpoll 9
893server 3.nl.pool.ntp.org iburst maxpoll 9
894{% else -%}
895# Local Wireless Leiden NTP Servers.
896server 0.pool.ntp.wleiden.net iburst maxpoll 9
897server 1.pool.ntp.wleiden.net iburst maxpoll 9
898server 2.pool.ntp.wleiden.net iburst maxpoll 9
899server 3.pool.ntp.wleiden.net iburst maxpoll 9
900
901# All the configured NTP servers
902{{ autogen_ntp_servers }}
903{% endif %}
904
905# If a server loses sync with all upstream servers, NTP clients
906# no longer follow that server. The local clock can be configured
907# to provide a time source when this happens, but it should usually
908# be configured on just one server on a network. For more details see
909# http://support.ntp.org/bin/view/Support/UndisciplinedLocalClock
910# The use of Orphan Mode may be preferable.
911#
912server 127.127.1.0
913fudge 127.127.1.0 stratum 10
914""").render(datadump)
915
916
917def generate_pf_hybrid_conf_local(datadump):
918 """ Generate configuration file '/etc/pf.hybrid.conf.local' """
919 datadump['autogen_header'] = generate_header(datadump, "#")
920 return Template("""\
921{{ autogen_header }}
922
923# Redirect some internal facing services outside (7)
924# INFO: {{ rdr_rules|count }} rdr_rules (outside to internal redirect rules) defined.
925{% for protocol, src_port,dest_ip,dest_port in rdr_rules -%}
926rdr on $ext_if inet proto {{ protocol }} from any to $ext_if port {{ src_port }} tag SRV -> {{ dest_ip }} port {{ dest_port }}
927{% endfor -%}
928""").render(datadump)
929
930def generate_motd(datadump):
931 """ Generate configuration file '/etc/motd' """
932 output = Template("""\
933FreeBSD run ``service motd onestart'' to make me look normal
934
935 WWW: {{ autogen_fqdn }} - http://www.wirelessleiden.nl
936 Loc: {{ location }}
937
938Services:
939{% if board == "ALIX2" -%}
940{{" -"}} Core Node ({{ board }})
941{% else -%}
942{{" -"}} Hulp Node ({{ board }})
943{% endif -%}
944{% if service_proxy_normal -%}
945{{" -"}} Normal Proxy
946{% endif -%}
947{% if service_proxy_ileiden -%}
948{{" -"}} iLeiden Proxy
949{% endif -%}
950{% if service_incoming_rdr -%}
951{{" -"}} Incoming port redirects
952{% endif %}
953Interlinks:\n
954""").render(datadump)
955
956 (addrs_list, dhclient_if, extra_ouput) = make_interface_list(datadump)
957 # Just nasty hack to make the formatting looks nice
958 iface_len = max(map(len,addrs_list.keys()))
959 addr_len = max(map(len,[x[0] for x in [x[0] for x in addrs_list.values()]]))
960 for iface,addrs in sorted(addrs_list.iteritems()):
961 if iface in ['lo0']:
962 continue
963 for addr, comment in sorted(addrs,key=lambda x: parseaddr(x[0].split('/')[0])):
964 output += " - %s || %s || %s\n" % (iface.ljust(iface_len), addr.ljust(addr_len), comment)
965
966 output += '\n'
967 output += """\
968Attached bridges:
969"""
970 has_item = False
971 for iface_key in datadump['autogen_iface_keys']:
972 ifacedump = datadump[iface_key]
973 if ifacedump.has_key('ns_ip'):
974 has_item = True
975 output += " - %(autogen_ifname)s || %(mode)s || %(ns_ip)s\n" % ifacedump
976 if not has_item:
977 output += " - none\n"
978
979 return output
980
981
982def format_yaml_value(value):
983 """ Get yaml value in right syntax for outputting """
984 if isinstance(value,str):
985 output = '"%s"' % value
986 else:
987 output = value
988 return output
989
990
991
992def format_wleiden_yaml(datadump):
993 """ Special formatting to ensure it is editable"""
994 output = "# Genesis config yaml style\n"
995 output += "# vim:ts=2:et:sw=2:ai\n"
996 output += "#\n"
997 iface_keys = [elem for elem in datadump.keys() if elem.startswith('iface_')]
998 for key in sorted(set(datadump.keys()) - set(iface_keys)):
999 if key == 'rdr_rules':
1000 output += '%-10s:\n' % 'rdr_rules'
1001 for rdr_rule in datadump[key]:
1002 output += '- %s\n' % rdr_rule
1003 else:
1004 output += "%-10s: %s\n" % (key, format_yaml_value(datadump[key]))
1005
1006 output += "\n\n"
1007
1008 # Format (key, required)
1009 key_order = (
1010 ('comment', True),
1011 ('ip', True),
1012 ('desc', True),
1013 ('sdesc', True),
1014 ('mode', True),
1015 ('type', True),
1016 ('extra_type', False),
1017 ('channel', False),
1018 ('ssid', False),
1019 ('dhcp', True),
1020 ('compass', False),
1021 ('distance', False),
1022 ('ns_ip', False),
1023 ('bullet2_ip', False),
1024 ('ns_mac', False),
1025 ('bullet2_mac', False),
1026 ('ns_type', False),
1027 ('bridge_type', False),
1028 ('status', True),
1029 )
1030
1031 for iface_key in sorted(iface_keys):
1032 try:
1033 remainder = set(datadump[iface_key].keys()) - set([x[0] for x in key_order])
1034 if remainder:
1035 raise KeyError("invalid keys: %s" % remainder)
1036
1037 output += "%s:\n" % iface_key
1038 for key,required in key_order:
1039 if datadump[iface_key].has_key(key):
1040 output += " %-11s: %s\n" % (key, format_yaml_value(datadump[iface_key][key]))
1041 output += "\n\n"
1042 except Exception as e:
1043 print "# Error while processing interface %s" % iface_key
1044 raise
1045
1046 return output
1047
1048
1049
1050def generate_wleiden_yaml(datadump, header=True):
1051 """ Generate (petty) version of wleiden.yaml"""
1052 output = generate_header(datadump, "#") if header else ''
1053
1054 for key in datadump.keys():
1055 if key.startswith('autogen_'):
1056 del datadump[key]
1057 # Interface autogen cleanups
1058 elif type(datadump[key]) == dict:
1059 for key2 in datadump[key].keys():
1060 if key2.startswith('autogen_'):
1061 del datadump[key][key2]
1062
1063 output += format_wleiden_yaml(datadump)
1064 return output
1065
1066
1067def generate_yaml(datadump):
1068 return generate_config(datadump['nodename'], "wleiden.yaml", datadump)
1069
1070
1071
1072def generate_config(node, config, datadump=None):
1073 """ Print configuration file 'config' of 'node' """
1074 output = ""
1075 try:
1076 # Load config file
1077 if datadump == None:
1078 datadump = get_yaml(node)
1079
1080 if config == 'wleiden.yaml':
1081 output += generate_wleiden_yaml(datadump)
1082 elif config == 'authorized_keys':
1083 f = open(os.path.join(NODE_DIR,"global_keys"), 'r')
1084 output += f.read()
1085 f.close()
1086 elif config == 'dnsmasq.conf':
1087 output += generate_dnsmasq_conf(datadump)
1088 elif config == 'dhcpd.conf':
1089 output += generate_dhcpd_conf(datadump)
1090 elif config == 'rc.conf.local':
1091 output += generate_rc_conf_local(datadump)
1092 elif config == 'resolv.conf':
1093 output += generate_resolv_conf(datadump)
1094 elif config == 'ntp.conf':
1095 output += generate_ntp_conf(datadump)
1096 elif config == 'motd':
1097 output += generate_motd(datadump)
1098 elif config == 'pf.hybrid.conf.local':
1099 output += generate_pf_hybrid_conf_local(datadump)
1100 else:
1101 assert False, "Config not found!"
1102 except IOError, e:
1103 output += "[ERROR] Config file not found"
1104 return output
1105
1106
1107
1108def process_cgi_request():
1109 """ When calling from CGI """
1110 # Update repository if requested
1111 form = cgi.FieldStorage()
1112 if form.getvalue("action") == "update":
1113 print "Refresh: 5; url=."
1114 print "Content-type:text/plain\r\n\r\n",
1115 print "[INFO] Updating subverion, please wait..."
1116 print subprocess.Popen(['svn', 'cleanup', "%s/.." % NODE_DIR], stderr=subprocess.STDOUT, stdout=subprocess.PIPE).communicate()[0],
1117 print subprocess.Popen(['svn', 'up', "%s/.." % NODE_DIR], stderr=subprocess.STDOUT, stdout=subprocess.PIPE).communicate()[0],
1118 print "[INFO] All done, redirecting in 5 seconds"
1119 sys.exit(0)
1120
1121
1122 base_uri = os.environ['PATH_INFO']
1123 uri = base_uri.strip('/').split('/')
1124
1125 output = "Template Holder"
1126 content_type='text/plain'
1127 if base_uri.endswith('/create/network.kml'):
1128 content_type='application/vnd.google-earth.kml+xml'
1129 output = make_network_kml.make_graph()
1130 elif not uri[0]:
1131 if is_text_request():
1132 content_type = 'text/plain'
1133 output = '\n'.join(get_hostlist())
1134 else:
1135 content_type = 'text/html'
1136 output = generate_title(get_hostlist())
1137 elif len(uri) == 1:
1138 if is_text_request():
1139 content_type = 'text/plain'
1140 output = generate_node(uri[0])
1141 else:
1142 content_type = 'text/html'
1143 output = generate_node_overview(uri[0])
1144 elif len(uri) == 2:
1145 content_type = 'text/plain'
1146 output = generate_config(uri[0], uri[1])
1147 else:
1148 assert False, "Invalid option"
1149
1150 print "Content-Type: %s" % content_type
1151 print "Content-Length: %s" % len(output)
1152 print ""
1153 print output
1154
1155def get_realname(datadump):
1156 # Proxy naming convention is special, as the proxy name is also included in
1157 # the nodename, when it comes to the numbered proxies.
1158 if datadump['nodetype'] == 'Proxy':
1159 realname = datadump['nodetype'] + datadump['nodename'].replace('proxy','')
1160 else:
1161 # By default the full name is listed and also a shortname CNAME for easy use.
1162 realname = datadump['nodetype'] + datadump['nodename']
1163 return(realname)
1164
1165
1166
1167def make_dns(output_dir = 'dns', external = False):
1168 items = dict()
1169
1170 # hostname is key, IP is value
1171 wleiden_zone = defaultdict(list)
1172 wleiden_cname = dict()
1173
1174 pool = dict()
1175 for node in get_hostlist():
1176 datadump = get_yaml(node)
1177
1178 # Proxy naming convention is special
1179 fqdn = datadump['autogen_realname']
1180 if datadump['nodetype'] in ['CNode', 'Hybrid']:
1181 wleiden_cname[datadump['nodename']] = fqdn
1182
1183 if datadump.has_key('rdr_host'):
1184 remote_target = datadump['rdr_host']
1185 elif datadump.has_key('remote_access') and datadump['remote_access']:
1186 remote_target = datadump['remote_access'].split(':')[0]
1187 else:
1188 remote_target = None
1189
1190 if remote_target:
1191 try:
1192 parseaddr(remote_target)
1193 wleiden_zone[datadump['nodename'] + '.gw'].append((remote_target, False))
1194 except (IndexError, ValueError):
1195 wleiden_cname[datadump['nodename'] + '.gw'] = remote_target + '.'
1196
1197
1198 wleiden_zone[fqdn].append((datadump['masterip'], True))
1199
1200 # Hacking to get proper DHCP IPs and hostnames
1201 for iface_key in get_interface_keys(datadump):
1202 iface_name = iface_key.replace('_','-')
1203 (ip, cidr) = datadump[iface_key]['ip'].split('/')
1204 try:
1205 (dhcp_start, dhcp_stop) = datadump[iface_key]['dhcp'].split('-')
1206 datadump[iface_key]['autogen_netmask'] = cidr2netmask(cidr)
1207 dhcp_part = ".".join(ip.split('.')[0:3])
1208 if ip != datadump['masterip']:
1209 wleiden_zone["dhcp-gateway-%s.%s" % (iface_name, fqdn)].append((ip, False))
1210 for i in range(int(dhcp_start), int(dhcp_stop) + 1):
1211 wleiden_zone["dhcp-%s-%s.%s" % (i, iface_name, fqdn)].append(("%s.%s" % (dhcp_part, i), True))
1212 except (AttributeError, ValueError, KeyError):
1213 # First push it into a pool, to indentify the counter-part later on
1214 addr = parseaddr(ip)
1215 cidr = int(cidr)
1216 addr = addr & ~((1 << (32 - cidr)) - 1)
1217 if pool.has_key(addr):
1218 pool[addr] += [(iface_name, fqdn, ip)]
1219 else:
1220 pool[addr] = [(iface_name, fqdn, ip)]
1221 continue
1222
1223
1224
1225 # WL uses an /29 to configure an interface. IP's are ordered like this:
1226 # MasterA (.1) -- DeviceA (.2) <<>> DeviceB (.3) --- SlaveB (.4)
1227
1228 sn = lambda x: re.sub(r'(?i)^cnode','',x)
1229
1230 # Automatic naming convention of interlinks namely 2 + remote.lower()
1231 for (key,value) in pool.iteritems():
1232 # Make sure they are sorted from low-ip to high-ip
1233 value = sorted(value, key=lambda x: parseaddr(x[2]))
1234
1235 if len(value) == 1:
1236 (iface_name, fqdn, ip) = value[0]
1237 wleiden_zone["2unused-%s.%s" % (iface_name, fqdn)].append((ip, True))
1238
1239 # Device DNS names
1240 if 'cnode' in fqdn.lower():
1241 wleiden_zone["d-at-%s.%s" % (iface_name, fqdn)].append((showaddr(parseaddr(ip) + 1), False))
1242 wleiden_cname["d-at-%s.%s" % (iface_name,sn(fqdn))] = "d-at-%s.%s" % ((iface_name, fqdn))
1243
1244 elif len(value) == 2:
1245 (a_iface_name, a_fqdn, a_ip) = value[0]
1246 (b_iface_name, b_fqdn, b_ip) = value[1]
1247 wleiden_zone["2%s.%s" % (b_fqdn,a_fqdn)].append((a_ip, True))
1248 wleiden_zone["2%s.%s" % (a_fqdn,b_fqdn)].append((b_ip, True))
1249
1250 # Device DNS names
1251 if 'cnode' in a_fqdn.lower() and 'cnode' in b_fqdn.lower():
1252 wleiden_zone["d-at-%s.%s" % (a_iface_name, a_fqdn)].append((showaddr(parseaddr(a_ip) + 1), False))
1253 wleiden_zone["d-at-%s.%s" % (b_iface_name, b_fqdn)].append((showaddr(parseaddr(b_ip) - 1), False))
1254 wleiden_cname["d-at-%s.%s" % (a_iface_name,sn(a_fqdn))] = "d-at-%s.%s" % (a_iface_name, a_fqdn)
1255 wleiden_cname["d-at-%s.%s" % (b_iface_name,sn(b_fqdn))] = "d-at-%s.%s" % (b_iface_name, b_fqdn)
1256 wleiden_cname["d2%s.%s" % (sn(b_fqdn),sn(a_fqdn))] = "d-at-%s.%s" % (a_iface_name, a_fqdn)
1257 wleiden_cname["d2%s.%s" % (sn(a_fqdn),sn(b_fqdn))] = "d-at-%s.%s" % (b_iface_name, b_fqdn)
1258
1259 else:
1260 pool_members = [k[1] for k in value]
1261 for item in value:
1262 (iface_name, fqdn, ip) = item
1263 wleiden_zone["2ring.%s" % (fqdn)].append((ip, True))
1264
1265 # Include static DNS entries
1266 # XXX: Should they override the autogenerated results?
1267 # XXX: Convert input to yaml more useable.
1268 # Format:
1269 ##; this is a comment
1270 ## roomburgh=CNodeRoomburgh1
1271 ## apkerk1.CNodeVosko=172.17.176.8 ;this as well
1272 dns_list = yaml.load(open(os.path.join(NODE_DIR,'../dns/staticDNS.yaml'),'r'))
1273
1274 # Hack to allow special entries, for development
1275 wleiden_raw = {}
1276
1277 for line in dns_list:
1278 reverse = False
1279 k, items = line.items()[0]
1280 if type(items) == dict:
1281 if items.has_key('reverse'):
1282 reverse = items['reverse']
1283 items = items['a']
1284 else:
1285 items = items['cname']
1286 items = [items] if type(items) != list else items
1287 for item in items:
1288 if item.startswith('IN '):
1289 wleiden_raw[k] = item
1290 elif valid_addr(item):
1291 wleiden_zone[k].append((item, reverse))
1292 else:
1293 wleiden_cname[k] = item
1294
1295 details = dict()
1296 # 24 updates a day allowed
1297 details['serial'] = time.strftime('%Y%m%d%H')
1298
1299 if external:
1300 dns_masters = ['siteview.wirelessleiden.nl', 'ns1.vanderzwet.net']
1301 else:
1302 dns_masters = ['sunny.wleiden.net']
1303
1304 details['master'] = dns_masters[0]
1305 details['ns_servers'] = '\n'.join(['\tNS\t%s.' % x for x in dns_masters])
1306
1307 dns_header = '''
1308$TTL 3h
1309%(zone)s. SOA %(master)s. beheer.lijst.wirelessleiden.nl. ( %(serial)s 1d 12h 1w 60s )
1310 ; Serial, Refresh, Retry, Expire, Neg. cache TTL
1311
1312%(ns_servers)s
1313 \n'''
1314
1315
1316 if not os.path.isdir(output_dir):
1317 os.makedirs(output_dir)
1318 details['zone'] = 'wleiden.net'
1319 f = open(os.path.join(output_dir,"db." + details['zone']), "w")
1320 f.write(dns_header % details)
1321
1322 for host,items in wleiden_zone.iteritems():
1323 for ip,reverse in items:
1324 if ip not in ['0.0.0.0']:
1325 f.write("%s.wleiden.net. IN A %s \n" % (host.lower(), ip))
1326 for source,dest in wleiden_cname.iteritems():
1327 dest = dest if dest.endswith('.') else dest + ".wleiden.net."
1328 f.write("%s.wleiden.net. IN CNAME %s\n" % (source.lower(), dest.lower()))
1329 for source, dest in wleiden_raw.iteritems():
1330 f.write("%s.wleiden.net. %s\n" % (source, dest))
1331 f.close()
1332
1333 # Create whole bunch of specific sub arpa zones. To keep it compliant
1334 for s in range(16,32):
1335 details['zone'] = '%i.172.in-addr.arpa' % s
1336 f = open(os.path.join(output_dir,"db." + details['zone']), "w")
1337 f.write(dns_header % details)
1338
1339 #XXX: Not effient, fix to proper data structure and do checks at other
1340 # stages
1341 for host,items in wleiden_zone.iteritems():
1342 for ip,reverse in items:
1343 if not reverse:
1344 continue
1345 if valid_addr(ip):
1346 if valid_addr(ip):
1347 if int(ip.split('.')[1]) == s:
1348 rev_ip = '.'.join(reversed(ip.split('.')))
1349 f.write("%s.in-addr.arpa. IN PTR %s.wleiden.net.\n" % (rev_ip.lower(), host.lower()))
1350 f.close()
1351
1352
1353def usage():
1354 print """Usage: %(prog)s <argument>
1355Argument:
1356\tstandalone [port] = Run configurator webserver [8000]
1357\tdns [outputdir] = Generate BIND compliant zone files in dns [./dns]
1358\tfull-export = Generate yaml export script for heatmap.
1359\tstatic [outputdir] = Generate all config files and store on disk
1360\t with format ./<outputdir>/%%NODE%%/%%FILE%% [./static]
1361\ttest <node> [<file>] = Receive output for certain node [all files].
1362\ttest-cgi <node> <file> = Receive output of CGI script [all files].
1363\tlist <status> <items> = List systems which have certain status
1364
1365Arguments:
1366\t<node> = NodeName (example: HybridRick)
1367\t<file> = %(files)s
1368\t<status> = all|up|down|planned
1369\t<items> = systems|nodes|proxies
1370
1371NOTE FOR DEVELOPERS; you can test your changes like this:
1372 BEFORE any changes in this code:
1373 $ ./gformat.py static /tmp/pre
1374 AFTER the changes:
1375 $ ./gformat.py static /tmp/post
1376 VIEW differences and VERIFY all are OK:
1377 $ diff -urI 'Generated' -r /tmp/pre /tmp/post
1378""" % { 'prog' : sys.argv[0], 'files' : '|'.join(files) }
1379 exit(0)
1380
1381
1382def is_text_request():
1383 """ Find out whether we are calling from the CLI or any text based CLI utility """
1384 try:
1385 return os.environ['HTTP_USER_AGENT'].split()[0] in ['curl', 'fetch', 'wget']
1386 except KeyError:
1387 return True
1388
1389def switchFormat(setting):
1390 if setting:
1391 return "YES"
1392 else:
1393 return "NO"
1394
1395def rlinput(prompt, prefill=''):
1396 import readline
1397 readline.set_startup_hook(lambda: readline.insert_text(prefill))
1398 try:
1399 return raw_input(prompt)
1400 finally:
1401 readline.set_startup_hook()
1402
1403def fix_conflict(left, right, default='i'):
1404 while True:
1405 print "## %-30s | %-30s" % (left, right)
1406 c = raw_input("## Solve Conflict (h for help) <l|r|e|i|> [%s]: " % default)
1407 if not c:
1408 c = default
1409
1410 if c in ['l','1']:
1411 return left
1412 elif c in ['r','2']:
1413 return right
1414 elif c in ['e', '3']:
1415 return rlinput("Edit: ", "%30s | %30s" % (left, right))
1416 elif c in ['i', '4']:
1417 return None
1418 else:
1419 print "#ERROR: '%s' is invalid input (left, right, edit or ignore)!" % c
1420
1421def main():
1422 """Hard working sub"""
1423 # Allow easy hacking using the CLI
1424 if not os.environ.has_key('PATH_INFO'):
1425 if len(sys.argv) < 2:
1426 usage()
1427
1428 if sys.argv[1] == "standalone":
1429 import SocketServer
1430 import CGIHTTPServer
1431 # Hop to the right working directory.
1432 os.chdir(os.path.dirname(__file__))
1433 try:
1434 PORT = int(sys.argv[2])
1435 except (IndexError,ValueError):
1436 PORT = 8000
1437
1438 class MyCGIHTTPRequestHandler(CGIHTTPServer.CGIHTTPRequestHandler):
1439 """ Serve this CGI from the root of the webserver """
1440 def is_cgi(self):
1441 if "favicon" in self.path:
1442 return False
1443
1444 self.cgi_info = (os.path.basename(__file__), self.path)
1445 self.path = ''
1446 return True
1447 handler = MyCGIHTTPRequestHandler
1448 SocketServer.TCPServer.allow_reuse_address = True
1449 httpd = SocketServer.TCPServer(("", PORT), handler)
1450 httpd.server_name = 'localhost'
1451 httpd.server_port = PORT
1452
1453 logger.info("serving at port %s", PORT)
1454 try:
1455 httpd.serve_forever()
1456 except KeyboardInterrupt:
1457 httpd.shutdown()
1458 logger.info("All done goodbye")
1459 elif sys.argv[1] == "test":
1460 # Basic argument validation
1461 try:
1462 node = sys.argv[2]
1463 datadump = get_yaml(node)
1464 except IndexError:
1465 print "Invalid argument"
1466 exit(1)
1467 except IOError as e:
1468 print e
1469 exit(1)
1470
1471
1472 # Get files to generate
1473 gen_files = sys.argv[3:] if len(sys.argv) > 3 else files
1474
1475 # Actual config generation
1476 for config in gen_files:
1477 logger.info("## Generating %s %s", node, config)
1478 print generate_config(node, config, datadump)
1479 elif sys.argv[1] == "test-cgi":
1480 os.environ['PATH_INFO'] = "/".join(sys.argv[2:])
1481 os.environ['SCRIPT_NAME'] = __file__
1482 process_cgi_request()
1483 elif sys.argv[1] == "static":
1484 items = dict()
1485 items['output_dir'] = sys.argv[2] if len(sys.argv) > 2 else "./static"
1486 for node in get_hostlist():
1487 items['node'] = node
1488 items['wdir'] = "%(output_dir)s/%(node)s" % items
1489 if not os.path.isdir(items['wdir']):
1490 os.makedirs(items['wdir'])
1491 datadump = get_yaml(node)
1492 for config in files:
1493 items['config'] = config
1494 logger.info("## Generating %(node)s %(config)s" % items)
1495 f = open("%(wdir)s/%(config)s" % items, "w")
1496 f.write(generate_config(node, config, datadump))
1497 f.close()
1498 elif sys.argv[1] == "wind-export":
1499 items = dict()
1500 for node in get_hostlist():
1501 datadump = get_yaml(node)
1502 sql = """INSERT IGNORE INTO nodes (name, name_ns, longitude, latitude)
1503 VALUES ('%(nodename)s', '%(nodename)s', %(latitude)s, %(longitude)s);""" % datadump;
1504 sql = """INSERT IGNORE INTO users_nodes (user_id, node_id, owner)
1505 VALUES (
1506 (SELECT id FROM users WHERE username = 'rvdzwet'),
1507 (SELECT id FROM nodes WHERE name = '%(nodename)s'),
1508 'Y');""" % datadump
1509 #for config in files:
1510 # items['config'] = config
1511 # print "## Generating %(node)s %(config)s" % items
1512 # f = open("%(wdir)s/%(config)s" % items, "w")
1513 # f.write(generate_config(node, config, datadump))
1514 # f.close()
1515 for node in get_hostlist():
1516 datadump = get_yaml(node)
1517 for iface_key in sorted([elem for elem in datadump.keys() if elem.startswith('iface_')]):
1518 ifacedump = datadump[iface_key]
1519 if ifacedump.has_key('mode') and ifacedump['mode'] == 'ap-wds':
1520 ifacedump['nodename'] = datadump['nodename']
1521 if not ifacedump.has_key('channel') or not ifacedump['channel']:
1522 ifacedump['channel'] = 0
1523 sql = """INSERT INTO links (node_id, type, ssid, protocol, channel, status)
1524 VALUES ((SELECT id FROM nodes WHERE name = '%(nodename)s'), 'ap',
1525 '%(ssid)s', 'IEEE 802.11b', %(channel)s, 'active');""" % ifacedump
1526 elif sys.argv[1] == "full-export":
1527 hosts = {}
1528 for node in get_hostlist():
1529 datadump = get_yaml(node)
1530 hosts[datadump['nodename']] = datadump
1531 print yaml.dump(hosts)
1532
1533 elif sys.argv[1] == "dns":
1534 make_dns(sys.argv[2] if len(sys.argv) > 2 else 'dns', 'external' in sys.argv)
1535 elif sys.argv[1] == "cleanup":
1536 # First generate all datadumps
1537 datadumps = dict()
1538 ssid_to_node = dict()
1539 for host in get_hostlist():
1540 logger.info("# Processing: %s", host)
1541 # Set some boring default values
1542 datadump = { 'board' : 'UNKNOWN' }
1543 datadump.update(get_yaml(host))
1544 datadumps[datadump['autogen_realname']] = datadump
1545
1546 (poel, errors) = make_relations(datadumps)
1547 print "\n".join(["# WARNING: %s" % x for x in errors])
1548
1549 for host,datadump in datadumps.iteritems():
1550 try:
1551 # Convert all yes and no to boolean values
1552 def fix_boolean(dump):
1553 for key in dump.keys():
1554 if type(dump[key]) == dict:
1555 dump[key] = fix_boolean(dump[key])
1556 elif str(dump[key]).lower() in ["yes", "true"]:
1557 dump[key] = True
1558 elif str(dump[key]).lower() in ["no", "false"]:
1559 # Compass richting no (Noord Oost) is valid input
1560 if key != "compass": dump[key] = False
1561 return dump
1562 datadump = fix_boolean(datadump)
1563
1564 if datadump['rdnap_x'] and datadump['rdnap_y']:
1565 datadump['latitude'], datadump['longitude'] = rdnap.rd2etrs(datadump['rdnap_x'], datadump['rdnap_y'])
1566 elif datadump['latitude'] and datadump['longitude']:
1567 datadump['rdnap_x'], datadump['rdnap_y'] = rdnap.etrs2rd(datadump['latitude'], datadump['longitude'])
1568
1569 if datadump['nodename'].startswith('Proxy'):
1570 datadump['nodename'] = datadump['nodename'].lower()
1571
1572 for iface_key in datadump['autogen_iface_keys']:
1573 try:
1574 # All our normal wireless cards are normal APs now
1575 if datadump[iface_key]['type'] in ['11a', '11b', '11g', 'wireless']:
1576 datadump[iface_key]['mode'] = 'ap'
1577 # Wireless Leiden SSID have an consistent lowercase/uppercase
1578 if datadump[iface_key].has_key('ssid'):
1579 ssid = datadump[iface_key]['ssid']
1580 prefix = 'ap-WirelessLeiden-'
1581 if ssid.lower().startswith(prefix.lower()):
1582 datadump[iface_key]['ssid'] = prefix + ssid[len(prefix)].upper() + ssid[len(prefix) + 1:]
1583 if datadump[iface_key].has_key('ns_ip') and not datadump[iface_key].has_key('mode'):
1584 datadump[iface_key]['mode'] = 'autogen-FIXME'
1585 if not datadump[iface_key].has_key('comment'):
1586 datadump[iface_key]['comment'] = 'autogen-FIXME'
1587
1588 if datadump[iface_key]['comment'].startswith('autogen-') and datadump[iface_key].has_key('comment'):
1589 datadump[iface_key] = datadump[iface_key]['desc']
1590
1591 x = datadump[iface_key]['comment']
1592 datadump[iface_key]['comment'] = x[0].upper() + x[1:]
1593
1594
1595 if datadump[iface_key].has_key('desc'):
1596 if datadump[iface_key]['comment'].lower() == datadump[iface_key]['desc'].lower():
1597 del datadump[iface_key]['desc']
1598 else:
1599 print "# ERROR: At %s - %s" % (datadump['nodename'], iface_key)
1600 response = fix_conflict(datadump[iface_key]['comment'], datadump[iface_key]['desc'])
1601 if response:
1602 datadump[iface_key]['comment'] = response
1603 del datadump[iface_key]['desc']
1604
1605 # Check DHCP configuration
1606 dhcp_type(datadump[iface_key])
1607
1608 # Set the compass value based on the angle between the poels
1609 if datadump[iface_key].has_key('ns_ip'):
1610 my_pool = poel[network(datadump[iface_key]['ip'])]
1611 remote_hosts = list(set([x[0] for x in my_pool]) - set([host]))
1612 if remote_hosts:
1613 compass_target = remote_hosts[0]
1614 datadump[iface_key]['compass'] = cd_between_hosts(host, compass_target, datadumps)
1615 except Exception as e:
1616 print "# Error while processing interface %s" % iface_key
1617 raise
1618 store_yaml(datadump)
1619 except Exception as e:
1620 print "# Error while processing %s" % host
1621 raise
1622 elif sys.argv[1] == "list":
1623 use_fqdn = False
1624 if len(sys.argv) < 4 or not sys.argv[2] in ["up", "down", "planned", "all"]:
1625 usage()
1626 if sys.argv[3] == "nodes":
1627 systems = get_nodelist()
1628 elif sys.argv[3] == "proxies":
1629 systems = get_proxylist()
1630 elif sys.argv[3] == "systems":
1631 systems = get_hostlist()
1632 else:
1633 usage()
1634 if len(sys.argv) > 4:
1635 if sys.argv[4] == "fqdn":
1636 use_fqdn = True
1637 else:
1638 usage()
1639
1640 for system in systems:
1641 datadump = get_yaml(system)
1642
1643 output = datadump['autogen_fqdn'] if use_fqdn else system
1644 if sys.argv[2] == "all":
1645 print output
1646 elif datadump['status'] == sys.argv[2]:
1647 print output
1648 elif sys.argv[1] == "create":
1649 if sys.argv[2] == "network.kml":
1650 print make_network_kml.make_graph()
1651 else:
1652 usage()
1653 usage()
1654 else:
1655 # Do not enable debugging for config requests as it highly clutters the output
1656 if not is_text_request():
1657 cgitb.enable()
1658 process_cgi_request()
1659
1660
1661if __name__ == "__main__":
1662 main()
Note: See TracBrowser for help on using the repository browser.