From a6b473227fbb845aeeaab4de39b2aef5f371ce22 Mon Sep 17 00:00:00 2001
From: Brandon Heller <brandonh@stanford.edu>
Date: Sat, 2 Jan 2010 02:42:08 -0800
Subject: [PATCH] Add mn_run.py, a script to run Mininets.
Simplifies launching a Mininet or running tests on one.
---
bin/mn_run.py | 138 ++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 138 insertions(+)
create mode 100755 bin/mn_run.py
diff --git a/bin/mn_run.py b/bin/mn_run.py
new file mode 100755
index 00000000..5980bd21
--- /dev/null
+++ b/bin/mn_run.py
@@ -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()
\ No newline at end of file
--
GitLab