root-context/exec/vif_down

   1 #!/usr/bin/python2
   2 # VSYS script to ifdown 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
   9 import sys
  10 import pwd
  11 import os
  12 import re
  13 import string
  14
  15 if len(sys.argv) != 2: sys.exit(1)
  16
  17 # VSYS scripts get slicename as $1
  18 slicename=sys.argv[1]
  19 sliceid = pwd.getpwnam(slicename).pw_uid
  20
  21
  22 arglines = map(string.strip, sys.stdin.readlines())
  23
  24 if len(arglines)<1:
  25     print >>sys.stderr, "Insufficient argument lines."
  26     sys.exit(1)
  27
  28 vif = arglines[0] # interface name
  29
  30 # Create options list
  31 if len(arglines)>1:
  32     options = arglines[1:]
  33 else:
  34     options = []
  35
  36 opt_delete = False
  37
  38 for optionline in options:
  39     if len(optionline)==0: continue
  40     try:
  41         opt, val = optionline.split('=')
  42     except:
  43         print >>sys.stderr, "Bad option line: \"%s\"" % (optionline)
  44         sys.exit(1)
  45
  46     if opt=="delete":
  47         if val=="0":
  48             opt_delete=False
  49         elif val=="1":
  50             opt_delete=True
  51         else:
  52             print >>sys.stderr, "rp_filter value invalid: \"%s\"" % (val)
  53             sys.exit(1)
  54     else:
  55         print >>sys.stderr, "Unknown option: \"%s\"" % (opt)
  56         sys.exit(1)
  57
  58 # Validate interface name
  59 if len(vif)>16:
  60     print >>sys.stderr, "Interface name %s invalid"%(vif)
  61     sys.exit(1)
  62
  63 if re.match(r'(tun|tap)%d-\d+' % sliceid, vif ) is None:
  64     print >>sys.stderr, "Interface name %s does not match slice id %d."%(vif, sliceid)
  65     sys.exit(1)
  66
  67
  68
  69 # Remove iptables rules
  70 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()
  71 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)
  72 cmd_iptables_del_out = "/sbin/iptables -D OUTPUT -o %s -m mark -m state --state NEW ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
  73 cmd_iptables_del_pr = "/sbin/iptables -t nat -D POSTROUTING -s $(ip ro | grep 'dev %s' | head -1 | awk '{print $1}') -j SNAT --to-source %s --random 2>/dev/null" % (vif, public_src,)
  74 cmd_gre_del = "ip link del %s 2>/dev/null" % (vif,)
  75 cmd_iptables_del_gre_pr = "/sbin/iptables -t mangle -D INPUT -i %s -m mark --mark 0 -j MARK --set-mark %s 2>/dev/null" % (vif, sliceid)
  76
  77 os.system(cmd_iptables_del_in)
  78 os.system(cmd_iptables_del_out)
  79 os.system(cmd_iptables_del_pr)
  80 os.system(cmd_iptables_del_gre_pr)
  81
  82 # Bring interface down
  83 cmd_ifconfig = "/sbin/ifconfig %s down" % (vif)
  84 os.system(cmd_ifconfig)
  85
  86 # Delete GRE tunnel (if any)
  87 if (opt_delete):
  88     os.system(cmd_gre_del)