8f87a897ca59c48c58f84a823eee2545c28c850f
[vsys-scripts.git] / 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 # Bring interface down
38
39 cmd_ifconfig = "/sbin/ifconfig %s down" % (vif)
40 os.system(cmd_ifconfig)
41
42 # Remove iptables rules
43 cmd_iptables_del_in = "/sbin/iptables -D INPUT -i %s -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
44 cmd_iptables_del_out = "/sbin/iptables -D OUTPUT -o %s -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
45
46 os.system(cmd_iptables_del_in)
47 os.system(cmd_iptables_del_out)