Changeset 10054 in genesis for tools/gformat.py

Timestamp:

Mar 7, 2012, 11:03:33 PM (13 years ago)

Author:

rick

Message:

Include proxy configuration in gformat generation

Related-To: ticket:117

File:

• tools/gformat.py (modified) (9 diffs)

tools/gformat.py

@@ -56 +56 @@
     'rc.conf.local',
     'resolv.conf',
-    'wleiden.yaml'
+    'wleiden.yaml',
     ]
 
@@ -262 +262 @@
   output = generate_header("#");
   output += """\
-hostname='%(nodetype)s%(nodename)s.%(domain)s'
+hostname='%(autogen_fqdn)s.%(domain)s'
 location='%(location)s'
 """ % datadump
@@ -278 +278 @@
 
   # Extra Proxy configuration
+  datadump['ileiden_enable'] = 'yes' if datadump['ileiden'] else 'no'
   if datadump['nodetype'] == 'Proxy':
-    output += '\n# Disable routing\ngateway_enable="no"\n'
-
-  output += '\ninternalif="%s"\n' % datadump['internalif']
-
-  output += '\n'
+    output += """
+#
+# PROXY Configuration
+#
+sshtun_enable="YES"
+sshtun_flags="-R 22%(proxyid)02i:localhost:22"
+
+static_routes="wleiden"
+route_wleiden="-net 172.16.0.0/12 %(internalroute)s"
+
+internalif="%(internalif)s"
+
+# PROXY iLeiden Configuration
+ileiden_enable="%(ileiden_enable)s"
+gateway_enable="%(ileiden_enable)s"
+firewall_enable="%(ileiden_enable)s"
+firewall_script="/etc/ipfw.sh"
+firewall_nat_enable="%(ileiden_enable)s"
+""" % datadump
+
+  if datadump['nodetype'] == 'CNode' and datadump['ileiden']:
+    output += """
+# NODE iLeiden Configuration
+ileiden_enable="%(ileiden_enable)s"
+
+captive_portal_whitelist=""
+captive_portal_interfaces="%(autogen_dhcp_interfaces)s"
+""" % datadump
+
   # lo0 configuration:
   # - 172.32.255.1/32 is the proxy.wleiden.net deflector
@@ -299 +324 @@
     addrs_list['lo0'].append(datadump['masterip'] + "/32")
 
-  wlan_count = 0
   for iface_key in datadump['iface_keys']:
     ifacedump = datadump[iface_key]
@@ -320 +344 @@
     # XXX: Might want to deduct type directly from interface name
     if ifacedump['type'] in ['11a', '11b', '11g', 'wireless']:
-      # Create wlanX interface
-      ifacedump['wlanif'] ="wlan%i" % wlan_count
-      iface_map[interface] = ifacedump['wlanif']
-      wlan_count += 1
-
       # Default to station (client) mode
       ifacedump['wlanmode'] = "sta"
@@ -344 +363 @@
         ifacedump['extra'] = 'regdomain ETSI country NL'
 
-      output += "wlans_%(interface)s='%(wlanif)s'\n" % ifacedump
-      output += ("create_args_%(wlanif)s='wlanmode %(wlanmode)s mode " +\
+      output += "wlans_%(interface)s='%(autogen_ifname)s'\n" % ifacedump
+      output += ("create_args_%(autogen_ifname)s='wlanmode %(wlanmode)s mode " +\
         "%(mode)s ssid %(ssid)s %(extra)s channel %(channel)s'\n") % ifacedump
 
@@ -372 +391 @@
   datadump = yaml.load(f,Loader=Loader)
   datadump['autogen_iface_keys'] = get_interface_keys(datadump)
+
+  wlan_count=0
+  for key in datadump['autogen_iface_keys']:
+    if datadump[key]['type'] in ['11a', '11b', '11g', 'wireless']:
+      datadump[key]['autogen_ifname'] = 'wlan%i' % wlan_count
+      wlan_count += 1
+    else:
+      datadump[key]['autogen_ifname'] = datadump[key]['interface']
+
+
+  dhcp_interfaces = [datadump[key]['autogen_ifname'] for key in datadump['autogen_iface_keys'] if datadump[key]['dhcp'] != 'no']
+  datadump['autogen_dhcp_interfaces'] = ' '.join(dhcp_interfaces)
   datadump['autogen_item'] = item
+  datadump['autogen_fqdn'] = get_fqdn(datadump)
   f.close()
 
@@ -399 +431 @@
 def get_interface_keys(config):
   """ Quick hack to get all interface keys, later stage convert this to a iterator """
-  return [elem for elem in config.keys() if (elem.startswith('iface_') and not "lo0" in elem)]
+  return sorted([elem for elem in config.keys() if (elem.startswith('iface_') and not "lo0" in elem)])
 
 
@@ -485 +517 @@
     if key.startswith('autogen_'):
       del datadump[key]
+    # Interface autogen cleanups
+    elif type(datadump[key]) == dict:
+      for key2 in datadump[key].keys():
+        if key2.startswith('autogen_'):
+          del datadump[key][key2]
+      
 
   output = generate_header("#")