source: genesis/tools/gformat.py@ 10682

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

Prettyfy output to allow better debugging...

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