net.py

   1 #
   2 # $Id$
   3 #
   4
   5 """network configuration"""
   6
   7 # system provided modules
   8 import os, string, time, socket
   9
  10 # PlanetLab system modules
  11 import sioc, plnet
  12
  13 # local modules
  14 import bwlimit, logger, iptables
  15
  16 def start(options, conf):
  17     logger.log("net plugin starting up...")
  18
  19 def GetSlivers(data, config, plc):
  20     logger.verbose("net:GetSlivers called.")
  21     InitInterfaces(plc, data) # writes sysconfig files.
  22     if 'OVERRIDES' in dir(config):
  23         if config.OVERRIDES.get('net_max_rate') == '-1':
  24             logger.log("net: Slice and node BW Limits disabled.")
  25             if len(bwlimit.tc("class show dev eth0")):
  26                 logger.verbose("*** DISABLING NODE BW LIMITS ***")
  27                 bwlimit.stop()
  28         else:
  29             InitNodeLimit(data)
  30             InitI2(plc, data)
  31     else:
  32         InitNodeLimit(data)
  33         InitI2(plc, data)
  34     InitNAT(plc, data)
  35
  36
  37 def InitNodeLimit(data):
  38     if not 'networks' in data: return
  39
  40     # query running network interfaces
  41     devs = sioc.gifconf()
  42     ips = dict(zip(devs.values(), devs.keys()))
  43     macs = {}
  44     for dev in devs:
  45         macs[sioc.gifhwaddr(dev).lower()] = dev
  46
  47     for network in data['networks']:
  48         # Get interface name preferably from MAC address, falling
  49         # back on IP address.
  50         hwaddr=network['mac']
  51         if hwaddr <> None: hwaddr=hwaddr.lower()
  52         if hwaddr in macs:
  53             dev = macs[network['mac']]
  54         elif network['ip'] in ips:
  55             dev = ips[network['ip']]
  56         else:
  57             logger.log('%s: no such interface with address %s/%s' % (network['hostname'], network['ip'], network['mac']))
  58             continue
  59
  60         # Get current node cap
  61         try:
  62             old_bwlimit = bwlimit.get_bwcap(dev)
  63         except:
  64             old_bwlimit = None
  65
  66         # Get desired node cap
  67         if network['bwlimit'] is None or network['bwlimit'] < 0:
  68             new_bwlimit = bwlimit.bwmax
  69         else:
  70             new_bwlimit = network['bwlimit']
  71
  72         if old_bwlimit != new_bwlimit:
  73             # Reinitialize bandwidth limits
  74             bwlimit.init(dev, new_bwlimit)
  75
  76             # XXX This should trigger an rspec refresh in case
  77             # some previously invalid sliver bwlimit is now valid
  78             # again, or vice-versa.
  79
  80 def InitI2(plc, data):
  81     if not 'groups' in data: return
  82
  83     if "Internet2" in data['groups']:
  84         logger.log("This is an Internet2 node.  Setting rules.")
  85         i2nodes = []
  86         i2nodeids = plc.GetNodeGroups(["Internet2"])[0]['node_ids']
  87         for node in plc.GetInterfaces({"node_id": i2nodeids}, ["ip"]):
  88             # Get the IPs
  89             i2nodes.append(node['ip'])
  90         # this will create the set if it doesn't already exist
  91         # and add IPs that don't exist in the set rather than
  92         # just recreateing the set.
  93         bwlimit.exempt_init('Internet2', i2nodes)
  94
  95         # set the iptables classification rule if it doesnt exist.
  96         cmd = '-A POSTROUTING -m set --set Internet2 dst -j CLASSIFY --set-class 0001:2000 --add-mark'
  97         rules = []
  98         ipt = os.popen("/sbin/iptables-save")
  99         for line in ipt.readlines(): rules.append(line.strip(" \n"))
 100         ipt.close()
 101         if cmd not in rules:
 102             logger.verbose("net:  Adding iptables rule for Internet2")
 103             os.popen("/sbin/iptables -t mangle " + cmd)
 104
 105 def InitNAT(plc, data):
 106     if not 'networks' in data: return
 107
 108     # query running network interfaces
 109     devs = sioc.gifconf()
 110     ips = dict(zip(devs.values(), devs.keys()))
 111     macs = {}
 112     for dev in devs:
 113         macs[sioc.gifhwaddr(dev).lower()] = dev
 114
 115     ipt = iptables.IPTables()
 116     for network in data['networks']:
 117         # Get interface name preferably from MAC address, falling
 118         # back on IP address.
 119         hwaddr=network['mac']
 120         if hwaddr <> None: hwaddr=hwaddr.lower()
 121         if hwaddr in macs:
 122             dev = macs[network['mac']]
 123         elif network['ip'] in ips:
 124             dev = ips[network['ip']]
 125         else:
 126             logger.log('%s: no such interface with address %s/%s' % (network['hostname'], network['ip'], network['mac']))
 127             continue
 128
 129         try:
 130             settings = plc.GetInterfaceTags({'interface_tag_id': network['interface_tag_ids']})
 131         except:
 132             continue
 133
 134         for setting in settings:
 135             if setting['category'].upper() != 'FIREWALL':
 136                 continue
 137             if setting['name'].upper() == 'EXTERNAL':
 138                 # Enable NAT for this interface
 139                 ipt.add_ext(dev)
 140             elif setting['name'].upper() == 'INTERNAL':
 141                 ipt.add_int(dev)
 142             elif setting['name'].upper() == 'PF': # XXX Uglier code is hard to find...
 143                 for pf in setting['value'].split("\n"):
 144                     fields = {}
 145                     for field in pf.split(","):
 146                         (key, val) = field.split("=", 2)
 147                         fields[key] = val
 148                     if 'new_dport' not in fields:
 149                         fields['new_dport'] = fields['dport']
 150                     if 'source' not in fields:
 151                         fields['source'] = "0.0.0.0/0"
 152                     ipt.add_pf(fields)
 153     ipt.commit()
 154
 155 def InitInterfaces(plc, data):
 156     if not 'networks' in data: return
 157     plnet.InitInterfaces(logger, plc, data)
 158
 159 def start(options, config):
 160     pass