Add ability to read custom Mininets

bin/mn_run.py

@@ -5,6 +5,7 @@
 '''
 
 from optparse import OptionParser
+import os.path
 import time
 
 try:
@@ -100,6 +101,8 @@ def parse_args(self):
         add_dict_option(opts, HOSTS, HOST_DEF, 'host')
         add_dict_option(opts, CONTROLLERS, CONTROLLER_DEF, 'controller')
 
+        opts.add_option('--custom', type = 'string', default = None,
+                        help = 'read custom mininet from current dir')
         opts.add_option('--test', type = 'choice', choices = TESTS,
                         default = TESTS[0],
                         help = '[' + ' '.join(TESTS) + ']')
@@ -137,6 +140,10 @@ def setup(self):
             raise Exception('multipath topos require multipath-capable '
                             'controller.')
 
+        if self.options.custom:
+            if not os.path.isfile(self.options.custom):
+                raise Exception('could not find custom file: %s' % custom)
+
     def begin(self):
         '''Create and run mininet.'''
 
@@ -156,10 +163,20 @@ def begin(self):
         xterms = self.options.xterms
         mac = self.options.mac
         arp = self.options.arp
-        mn = Mininet(topo, switch, host, controller, controller_params,
-                     in_namespace = in_namespace,
-                     xterms = xterms, auto_set_macs = mac,
-                     auto_static_arp = arp)
+        mn = None
+        if not self.options.custom:
+            mn = Mininet(topo, switch, host, controller, controller_params,
+                         in_namespace = in_namespace,
+                         xterms = xterms, auto_set_macs = mac,
+                         auto_static_arp = arp)
+        else:
+            globals_ = {}
+            locals_ = {}
+            execfile(self.options.custom, globals_, locals_)
+            if 'mn' not in locals_:
+                raise Exception('could not find mn var in custom file')
+            else:
+                mn = locals_['mn']
 
         test = self.options.test
         if test != 'build':

custom/README

+This directory should hold configuration files for custom mininets.
+
+See custom_example.py, which loads the default minimal topology.  The advantage of defining a mininet in a separate file is that you then use the --custom option in mn_run.py to run the CLI or specific tests with it.
+
+Each custom file must define a Mininet object for an 'mn' variable.
+
+To start up a mininet with this custom topology, do:
+  sudo mn_run.py --custom custom_example.py
+

custom/custom_example.py

+'''Example of custom topo
+
+@author Brandon Heller (brandonh@stanford.edu)
+
+'''
+
+from mininet.topo import SingleSwitchTopo
+from mininet.net import Mininet
+from mininet.node import KernelSwitch, Host, Controller, ControllerParams
+
+topo = SingleSwitchTopo(k = 2) # build topology object
+switch = KernelSwitch
+host = Host
+controller = Controller
+controller_params = ControllerParams(0x0a000000, 8) # 10.0.0.0/8
+in_namespace = False
+xterms = False
+mac = True
+arp = True
+
+mn = Mininet(topo, switch, host, controller, controller_params,
+             in_namespace = in_namespace,
+             xterms = xterms, auto_set_macs = mac,
+             auto_static_arp = arp)
+