exec/vif_down

   1 #!/usr/bin/python
   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
  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 # Read interface name
  23 vif = sys.stdin.readline().strip() # interface name
  24
  25
  26 # Validate interface name
  27 if len(vif)>16:
  28     print >>sys.stderr, "Interface name %s invalid"%(vif)
  29     sys.exit(1)
  30
  31 if re.match(r'(tun|tap)%d-\d+' % sliceid, vif ) is None:
  32     print >>sys.stderr, "Interface name %s does not match slice id %d."%(vif, sliceid)
  33     sys.exit(1)
  34
  35
  36
  37 # Remove iptables rules
  38 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()
  39 cmd_iptables_del_in = "/sbin/iptables -D INPUT -i %s -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
  40 cmd_iptables_del_out = "/sbin/iptables -D OUTPUT -o %s -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
  41 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" % (vif, public_src,)
  42
  43 os.system(cmd_iptables_del_in)
  44 os.system(cmd_iptables_del_out)
  45 os.system(cmd_iptables_del_pr)
  46
  47 # Bring interface down
  48 cmd_ifconfig = "/sbin/ifconfig %s down" % (vif)
  49 os.system(cmd_ifconfig)
  50