X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=exec%2Fvif_up;h=0f905c6402ea6ecb9530d62c94ee7ff9b9de84d8;hb=153c335cbebbd3ba11fa7fd56fabfcb9eb895b95;hp=b40fef7da6320c202fc0caf63d993cf59eb3b4a3;hpb=8bf17cec36c3d77ce9c2f01330ced5d9f652783e;p=vsys-scripts.git

diff --git a/exec/vif_up b/exec/vif_up
index b40fef7..0f905c6 100755
--- a/exec/vif_up
+++ b/exec/vif_up
@@ -7,6 +7,7 @@
 #   - Interface name (eg [tun|tap]<sliceid>-<n>)
 #   - IP address (eg 1.2.3.4)
 #   - Netmask (as int, e.g. 24)
+#   - Followed by options as name=value pairs
 
 import sys
 import pwd
@@ -14,6 +15,7 @@ import re
 import socket
 import struct
 import os
+import string
 
 vsys_config_dir = "/etc/planetlab/vsys-attributes"
 
@@ -37,13 +39,21 @@ if base is None:
 
 
 ### Read args from stdin
-vif = sys.stdin.readline().strip() # interface name
-vip = sys.stdin.readline().strip() # IP
-vmask = int(sys.stdin.readline().strip()) # netmask as int
+arglines = map(string.strip, sys.stdin.readlines())
 
-# TODO further config args? txqueue, nat? Can add these later...
+if len(arglines)<3:
+    print >>sys.stderr, "Insufficient argument lines."
+    sys.exit(1)
 
+vif = arglines[0] # interface name
+vip = arglines[1] # IP
+vmask = int(arglines[2]) # netmask as int
 
+# Create options list
+if len(arglines)>3:
+    options = arglines[3:]
+else:
+    options = []
 
 # Convert network base addr to int format by unpacking as 32bit net-ordered long (!L)
 base_int = struct.unpack('!L',socket.inet_aton(base))[0]
@@ -87,18 +97,80 @@ if vmask<mask:
 
 
 
+### Process options
+
+opt_txqueuelen = None
+opt_rp_filter = None
+opt_snat = None
+
+
+for optionline in options:
+    if len(optionline)==0: continue
+    try:
+        opt, val = optionline.split('=')
+    except:
+        print >>sys.stderr, "Bad option line: \"%s\"" % (optionline)
+        sys.exit(1)
+
+    if opt=="rp_filter":
+        if val=="0":
+            opt_rp_filter="0"
+        elif val=="1":
+            opt_rp_filter="1"
+        else:
+            print >>sys.stderr, "rp_filter value invalid: \"%s\"" % (val)
+            sys.exit(1)
+
+    elif opt=="txqueuelen":
+        intval = int(val)
+        if intval<1 or intval>10000:
+            print >>sys.stderr, "txqueuelen value %s out of range 1-10000" % (val)
+            sys.exit(1)
+        opt_txqueuelen = intval
+
+    elif opt=="snat":
+	intval = int(val)
+	if val=="1":
+	  opt_snat = True
+    else:
+        print >>sys.stderr, "Unknown option: \"%s\"" % (opt)
+        sys.exit(1)
+
+
+        
 ### Configure interface
 
-cmd_ifconfig = "/sbin/ifconfig %s %s/%d" % (vif, vip, vmask)
+if opt_txqueuelen is None:
+    cmd_ifconfig = "/sbin/ifconfig %s %s/%d" % (vif, vip, vmask)
+else:
+    cmd_ifconfig = "/sbin/ifconfig %s %s/%d txqueuelen %d" % (vif, vip, vmask, opt_txqueuelen)
+
 os.system(cmd_ifconfig)
 
 # Add iptables rules (Clearing old ones first, if they exist)
-cmd_iptables_in = "/sbin/iptables -A INPUT -i %s -m mark ! --mark %d -j DROP" % (vif, sliceid)
-cmd_iptables_del_in = "/sbin/iptables -D INPUT -i %s -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
-cmd_iptables_out = "/sbin/iptables -A OUTPUT -o %s -m mark ! --mark %d -j DROP" % (vif, sliceid)
-cmd_iptables_del_out = "/sbin/iptables -D OUTPUT -o %s -m mark ! --mark %d -j DROP 2>/dev/null" % (vif, sliceid)
+cmd_iptables_in = "/sbin/iptables -A INPUT -i %s -m mark -m state --state NEW ! --mark %d -j DROP" % (vif, sliceid)
+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)
+cmd_iptables_out = "/sbin/iptables -A OUTPUT -o %s -m state --state NEW -m mark ! --mark %d -j DROP" % (vif, sliceid)
+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)
+
+public_src = os.popen("ip route get 1.1.1.1 | head -1 | awk '{print $7;}'").read().rstrip();
+cmd_iptables_pr = "/sbin/iptables -t nat -A POSTROUTING -s %s/%d -j SNAT --to-source %s --random" % (vip, vmask, public_src)
+cmd_iptables_del_pr = "/sbin/iptables -t nat -D POSTROUTING -s %s/%d -j SNAT --to-source %s --random" % (vip, vmask, public_src)
 
 os.system(cmd_iptables_del_in)
 os.system(cmd_iptables_in)
 os.system(cmd_iptables_del_out)
 os.system(cmd_iptables_out)
+
+if (opt_snat):
+    os.system(cmd_iptables_del_pr)
+    os.system(cmd_iptables_pr)
+    #print cmd_iptables_del_pr
+    #print cmd_iptables_pr
+
+# Process additional options
+if opt_rp_filter is not None:
+    rp_cmd = "/sbin/sysctl net.ipv4.conf.%s.rp_filter=%s" % (vif, opt_rp_filter)
+    os.system(rp_cmd)
+
+