exec/vif_up

   1 #!/usr/bin/python
   2 # VSYS script to configure per-slice virtual network interfaces from the root slice
   3 #   Thom Haddow - 06/10/09
   4 #
   5 # Gets slice name as argv[1]
   6 # Takes remaining args on stdin:
   7 #   - Interface name (eg [tun|tap]<sliceid>-<n>)
   8 #   - IP address (eg 1.2.3.4)
   9 #   - Netmask (as int, e.g. 24)
  10 #   - Followed by options as name=value pairs
  11
  12 import sys
  13 import pwd
  14 import re
  15 import socket
  16 import struct
  17 import o
  18 vif = arglines[0] # interface name
  19 vip = arglines[1] # IP
  20 vmask = int(arglines[2]) # netmask as int
  21
  22 # Create options list
  23 if len(arglines)>3:
  24     options = arglines[3:]
  25 else:
  26     options = []
  27
  28 # Convert network base addr to int format by unpacking as 32bit net-ordered long (!L)
  29 base_int = struct.unpack('!L',socket.inet_aton(base))[0]
  30 mask = int(mask)
  31
  32
  33 ### Validate args
  34 # Validate interface name
  35 if len(vif)>16:
  36     print >>sys.stderr, "Interface name %s invalid"%(vif)
  37     sys.exit(1)
  38
  39 if re.match(r'(tun|tap)%d-\d+' % sliceid, vif ) is None:
  40     print >>sys.stderr, "Interface name %s does not match slice id %d."%(vif, sliceid)
  41     sys.exit(1)
  42
  43
  44
  45 # Validate requested IP and convert to int format.
  46 try:
  47     vip_int = struct.unpack('!L',socket.inet_aton(vip))[0]
  48 except socket.error:
  49     print >>sys.stderr, "Invalid IP: %s" % vip
  50     sys.exit(1)
  51
  52 # Check IP is in netblock
  53 if (vip_int>>(32-mask)) != (base_int>>(32-mask)):
  54     print >>sys.stderr, "Requested IP %s not in netblock %s/%d" % (vip,base,mask)
  55     sys.exit(1)
  56
  57 # TODO. Check IP is not in use?
  58
  59 # Validate mask: Check requested mask is sane and within our netblock
  60 if vmask>32 or vmask <8:
  61     print >>sys.stderr, "Requested netmask /%d is invalid" %(vmask)
  62     sys.exit(1)
  63
  64 if vmask<mask:
  65     print >>sys.stderr, "Requested netmask /%d larger than allocation /%d" %(vmask, mask)
  66     sys.exit(1)
  67
  68
  69
  70 ### Process options
  71
  72 opt_txqueuelen = None
  73 opt_rp_filter = None
  74 opt_snat = None
  75 opt_ovs_dp = None
  76 opt_pointopoint = None
  77
  78
  79 for optionline in options:
  80     if len(optionline)==0: continue
  81     try:
  82         opt, val = optionline.split('=')
  83     except:
  84         print >>sys.stderr, "Bad option line: \"%s\"" % (optionline)
  85         sys.exit(1)
  86
  87     if opt=="rp_filter":
  88         if val=="0":
  89             opt_rp_filter="0"
  90         elif val=="1":
  91             opt_rp_filter="1"
  92         else:
  93             print >>sys.stderr, "rp_filter value invalid: \"%s\"" % (val)
  94             sys.exit(1)
  95
  96     elif opt=="txqueuelen":
  97         intval = int(val)
  98         if intval<1 or intval>10000:
  99             print >>sys.stderr, "txqueuelen value %s out of range 1-10000" % (val)
 100             sys.exit(1)
 101         opt_txqueuelen = intval
 102     elif opt=="snat":
 103         intval = int(val)
 104         if val=="1":
 105             opt_snat = True
 106     elif opt=="pointopoint":
 107         opt_pointopoint = val.strip()
 108         try:
 109             socket.inet_aton(opt_pointopoint)
 110         except socket.error,e:
 111             print >>sys.stderr, "pointopoint: %s" % (e,)
 112             sys.exit(1)
 113     elif opt=="vswitch":
 114         opt_ovs_dp = val
 115     else:
 116         print >>sys.stderr, "Unknown option: \"%s\"" % (opt)
 117         sys.exit(1)
 118
 119
 120 ### Configure interface
 121
 122 cmd_ifconfig = "/sbin/ifconfig %s %s" % (vif, vip)
 123 if opt_pointopoint is None:
 124     cmd_ifconfig += "/%d" % (vmask,)
 125 else:
 126     # point-to-point mask
 127     cmd_ifconfig += " netmask 255.255.255.255"
 128 if opt_txqueuelen is not None:
 129     cmd_ifconfig += " txqueuelen %d" % (opt_txqueuelen,)
 130 if opt_pointopoint is not None:
 131     cmd_ifconfig += " pointopoint %s" % (opt_pointopoint,)
 132
 133 os.system(cmd_ifconfig)
 134
 135 # Add iptables rules (Clearing old ones first, if they exist)
 136 cmd_iptables_in = "/sbin/iptables -A INPUT -i %s -m mark -m state --state NEW ! --mark %d -j DROP" % (vif, sliceid)
 137 cmd_iptables_del_in = "/sbin/iptables -D INPUT -i %s -m mark -m state --state NEW ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
 138 cmd_iptables_out = "/sbin/iptables -A OUTPUT -o %s -m state --state NEW -m mark ! --mark %d -j DROP" % (vif, sliceid)
 139 cmd_iptables_del_out = "/sbin/iptables -D OUTPUT -o %s -m state --state NEW -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
 140
 141 public_src = os.popen("ifconfig | grep $(ip route | grep default | awk '{print $3}' | awk -F. '{print $1\"[.]\"$2}') | head -1 | awk '{print $2}' | awk -F : '{print $2}'").read().rstrip()
 142 cmd_iptables_pr = "/sbin/iptables -t nat -A POSTROUTING -s %s/%d -j SNAT --to-source %s --random" % (vip, vmask, public_src)
 143 cmd_iptables_del_pr = "/sbin/iptables -t nat -D POSTROUTING -s %s/%d -j SNAT --to-source %s --random > /dev/null 2>&1" % (vip, vmask, public_src)
 144
 145 os.system(cmd_iptables_del_in)
 146 os.system(cmd_iptables_in)
 147 os.system(cmd_iptables_del_out)
 148 os.system(cmd_iptables_out)
 149
 150 # always remove snat rules
 151 # in case there are leftovers from previous calls
 152 os.system(cmd_iptables_del_pr)
 153 if (opt_snat):
 154     os.system(cmd_iptables_pr)
 155     #print cmd_iptables_del_pr
 156     #print cmd_iptables_pr
 157
 158 #if opt_pointopoint and vmask < 32:
 159 #    # Make target vnet masked address
 160 #    p2p_int = struct.unpack('!L', socket.inet_aton(opt_pointopoint))[0]
 161 #    vip_vnet = socket.inet_ntoa(
 162 #        struct.pack('!L', ((p2p_int>>(32-vmask))<<(32-vmask))) )
 163 #
 164 #    cmd_routing_p2p = "/sbin/route add -net %s/%d gw %s dev %s" % (vip_vnet, vmask, opt_pointopoint, vif)
 165 #    print >>sys.stderr, cmd_routing_p2p
 166 #    os.system(cmd_routing_p2p)
 167
 168 # Process additional options
 169 if opt_rp_filter is not None:
 170     rp_cmd = "/sbin/sysctl net.ipv4.conf.%s.rp_filter=%s" % (vif, opt_rp_filter)
 171     os.system(rp_cmd)
 172
 173 # OVS datapath
 174 if opt_ovs_dp is not None:
 175     cmd_ovs_addif = "ovs-dpctl add-if %s %s"%(opt_ovs_dp,vif)
 176     os.system(cmd_ovs_addif)
 177