initial import (Boeing r1752, NRL r878)

daemon/examples/netns/iperf-performance-chain.py

@@ -0,0 +1,109 @@
+#!/usr/bin/python 
+
+# Copyright (c)2013 the Boeing Company.
+# See the LICENSE file included in this distribution.
+
+# This script creates a CORE session, that will connect n nodes together
+# in a chain, with static routes between nodes
+#    number of nodes / number of hops
+#            2            0
+#            3            1
+#            4            2
+#            n          n - 2
+#
+# Use core-cleanup to clean up after this script as the session is left running.
+#
+
+import sys, datetime, optparse
+
+from core import pycore
+from core.misc import ipaddr
+from core.constants import *
+
+# node list (count from 1)
+n = [None]
+
+def add_to_server(session):
+    ''' Add this session to the server's list if this script is executed from
+    the core-daemon server.
+    '''
+    global server
+    try:
+        server.addsession(session)
+        return True
+    except NameError:
+        return False
+
+def main():
+    usagestr = "usage: %prog [-h] [options] [args]"
+    parser = optparse.OptionParser(usage = usagestr)
+    parser.set_defaults(numnodes = 5)
+
+    parser.add_option("-n", "--numnodes", dest = "numnodes", type = int,
+                      help = "number of nodes")
+
+    def usage(msg = None, err = 0):
+        sys.stdout.write("\n")
+        if msg:
+            sys.stdout.write(msg + "\n\n")
+        parser.print_help()
+        sys.exit(err)
+
+    # parse command line options
+    (options, args) = parser.parse_args()
+
+    if options.numnodes < 1:
+        usage("invalid number of nodes: %s" % options.numnodes)
+
+    if options.numnodes >= 255:
+        usage("invalid number of nodes: %s" % options.numnodes)
+
+    for a in args:
+        sys.stderr.write("ignoring command line argument: '%s'\n" % a)
+
+    start = datetime.datetime.now()
+
+    session = pycore.Session(persistent=True)
+    add_to_server(session)
+    print "creating %d nodes"  % options.numnodes
+    left = None
+    prefix = None
+    for i in xrange(1, options.numnodes + 1):
+        tmp = session.addobj(cls = pycore.nodes.CoreNode, name = "n%d" % i,
+                             objid=i)
+        if left:
+            tmp.newnetif(left, ["%s/%s" % (prefix.addr(2), prefix.prefixlen)])
+
+        prefix = ipaddr.IPv4Prefix("10.83.%d.0/24" % i) # limit: i < 255
+        right = session.addobj(cls = pycore.nodes.PtpNet)
+        tmp.newnetif(right, ["%s/%s" % (prefix.addr(1), prefix.prefixlen)])
+        tmp.cmd([SYSCTL_BIN, "net.ipv4.icmp_echo_ignore_broadcasts=0"])
+        tmp.cmd([SYSCTL_BIN, "net.ipv4.conf.all.forwarding=1"])
+        tmp.cmd([SYSCTL_BIN, "net.ipv4.conf.default.rp_filter=0"])
+        tmp.setposition(x=100*i,y=150)
+        n.append(tmp)
+        left = right
+    
+    prefixes = map(lambda(x): ipaddr.IPv4Prefix("10.83.%d.0/24" % x),
+                   xrange(1, options.numnodes + 1))
+                   
+    # set up static routing in the chain
+    for i in xrange(1, options.numnodes + 1):
+        for j in xrange(1, options.numnodes + 1):
+            if j < i - 1:
+                gw = prefixes[i-2].addr(1)
+            elif j > i:
+                if i > len(prefixes) - 1:
+                    continue
+                gw = prefixes[i-1].addr(2)
+            else:
+                continue
+            net = prefixes[j-1]
+            n[i].cmd([IP_BIN, "route", "add", str(net), "via", str(gw)])
+
+
+    print "elapsed time: %s" % (datetime.datetime.now() - start)
+
+if __name__ == "__main__" or __name__ == "__builtin__":
+    main()
+