Add mn_run.py, a script to run Mininets.

Simplifies launching a Mininet or running tests on one.
2010-01-02 02:42:08 -08:00
parent 8b5062a3a1
commit a6b473227f
1 changed files with 138 additions and 0 deletions
@@ -0,0 +1,138 @@
+#!/usr/bin/env python
+'''Mininet runner
+
+@author Brandon Heller (brandonh@stanford.edu)
+'''
+
+from optparse import OptionParser
+import time
+
+from ripcord.topo import FatTreeTopo
+
+from mininet.logging_mod import lg, set_loglevel, LEVELS
+from mininet.net import Mininet, init
+from mininet.node import Switch, Host, Controller, ControllerParams
+from mininet.topo import TreeTopo
+
+# built in topologies, created only when run
+TOPO_DEF = 'minimal'
+TOPOS = {'minimal' :  (lambda: TreeTopo(depth = 2, fanout = 2)),
+         'tree16' :   (lambda: TreeTopo(depth = 3, fanout = 4)),
+         'tree64' :   (lambda: TreeTopo(depth = 4, fanout = 4)),
+         'fattree4' : (lambda: FatTreeTopo(k = 4)),
+         'fattree6' : (lambda: FatTreeTopo(k = 6))}
+
+SWITCH_DEF = 'kernel'
+SWITCHES = {'kernel' : Switch}
+
+HOST_DEF = 'process'
+HOSTS = {'process' : Host}
+
+CONTROLLER_DEF = 'ref'
+CONTROLLERS = {'ref' : Controller}
+
+# optional tests to run
+TESTS = ['cli', 'build', 'ping_all', 'ping_pair']
+
+
+def add_dict_option(opts, choices_dict, default, name, help_str = None):
+    '''Convenience function to add choices dicts to OptionParser.
+    
+    @param opts OptionParser instance
+    @param choices_dict dictionary of valid choices, must include default
+    @param default default choice key
+    @param name long option name
+    @param help string
+    '''
+    if default not in choices_dict:
+        raise Exception('Invalid  default %s for choices dict: %s' %
+                        (default, name))
+    if not help_str:
+        help_str = '[' + ' '.join(choices_dict.keys()) + ']'
+    opts.add_option('--' + name,
+                    type = 'choice',
+                    choices = choices_dict.keys(),
+                    default = default,
+                    help = help_str)
+
+
+class MininetRunner(object):
+    '''Build, setup, and run Mininet.'''
+
+    def __init__(self):
+        '''Init.'''
+        self.options = None
+        
+        self.parse_args()
+        self.setup()
+        self.begin()
+
+    def parse_args(self):
+        '''Parse command-line args and return options object.
+        
+        @return opts parse options dict
+        '''
+        opts = OptionParser()
+        add_dict_option(opts, TOPOS, TOPO_DEF, 'topo')
+        add_dict_option(opts, SWITCHES, SWITCH_DEF, 'switch')
+        add_dict_option(opts, HOSTS, HOST_DEF, 'host')
+        add_dict_option(opts, CONTROLLERS, CONTROLLER_DEF, 'controller')
+
+        opts.add_option('--test', type = 'choice', choices = TESTS,
+                        default = TESTS[0],
+                        help = '[' + ' '.join(TESTS) + ']')
+        opts.add_option('--xterm', '-x', action = 'store_true',
+                        default = False, help = 'spawn xterms for each node')
+        opts.add_option('--verbosity', '-v', type = 'choice',
+                        choices = LEVELS.keys(), default = 'info',
+                        help = '[' + ' '.join(LEVELS.keys()) + ']')
+        self.options = opts.parse_args()[0]
+
+    def setup(self):
+        '''Setup and validate environment.'''
+
+        # set logging verbosity
+        set_loglevel(self.options.verbosity)
+
+        # validate environment setup
+        init()
+
+    def begin(self):
+        '''Create and run mininet.'''
+
+        start = time.time()
+
+        topo = TOPOS[self.options.topo]() # build topology object
+        switch = SWITCHES[self.options.switch]
+        host = HOSTS[self.options.host]
+        controller = CONTROLLERS[self.options.controller]
+
+        controller_params = ControllerParams(0x0a000000, 8) # 10.0.0.0/8
+        mn = Mininet(topo, switch, host, controller, controller_params)
+
+        if self.options.test != 'build':
+    
+            if self.options.test == 'cli':
+                mn.interact()
+            elif self.options.test == 'ping_all':
+                mn.start()
+                dropped = mn.ping_test(verbose = True)
+                lg.info("*** Test results: %i%% dropped\n", dropped)
+                mn.stop()
+            elif self.options.test == 'ping_pair':
+                mn.start()
+                hosts_unsorted = sorted(mn.topo.hosts())
+                hosts = [hosts_unsorted[0], hosts_unsorted[1]]
+                dropped = mn.ping_test(hosts = hosts, verbose = True)
+                lg.info("*** Test results: %i%% dropped\n", dropped)
+                mn.stop()
+            else:
+                raise NotImplementedError('unknown test: %s' %
+                                          self.options.test)
+
+        elapsed = float(time.time() - start)
+        print ('completed in %0.3f seconds' % elapsed)
+
+
+if __name__ == "__main__":
+    MininetRunner()