Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | """ |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 4 | onos.py: A basic (?) ONOS Controller() subclass for Mininet |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 5 | |
| 6 | We implement the following classes: |
| 7 | |
| 8 | ONOSController: a custom Controller() subclass to start ONOS |
| 9 | OVSSwitchONOS: a custom OVSSwitch() switch that connects to multiple controllers. |
| 10 | |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 11 | We use single Zookeeper and Ramcloud instances for now. |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 12 | |
| 13 | As a custom file, exports: |
| 14 | |
| 15 | --controller onos |
| 16 | --switch ovso |
| 17 | |
| 18 | Usage: |
| 19 | |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 20 | $ sudo -E ./onos.py |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 21 | |
| 22 | This will start up a simple 2-host, 2 ONOS network |
| 23 | |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 24 | $ sudo -E mn --custom onos.py --controller onos,2 --switch ovso |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 25 | """ |
| 26 | |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 27 | from mininet.node import Controller, OVSSwitch, CPULimitedHost, RemoteController |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 28 | from mininet.net import Mininet |
| 29 | from mininet.cli import CLI |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 30 | from mininet.topo import LinearTopo |
| 31 | from mininet.log import setLogLevel, info, warn |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 32 | from mininet.util import quietRun, numCores |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 33 | |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 34 | # This should be cleaned up to avoid interfering with mn |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 35 | from shutil import copyfile |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 36 | from os import environ, path |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 37 | from functools import partial |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 38 | import time |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 39 | from sys import argv |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 40 | from time import sleep |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 41 | |
| 42 | class ONOS( Controller ): |
| 43 | "Custom controller class for ONOS" |
| 44 | |
| 45 | # Directories and configuration templates |
| 46 | home = environ[ 'HOME' ] |
| 47 | onosDir = home + "/ONOS" |
| 48 | zookeeperDir = home + "/zookeeper-3.4.5" |
| 49 | dirBase = '/tmp' |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 50 | logDir = dirBase + '/onos-logs' |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 51 | configFile = dirBase + '/onos-%s.properties' |
Bob Lantz | 4ed6cff | 2013-12-17 21:43:36 -0800 | [diff] [blame] | 52 | logbackFile = dirBase + '/onos-%s.logback.xml' |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 53 | |
| 54 | # Base ONOS modules |
| 55 | baseModules = ( |
| 56 | 'net.floodlightcontroller.core.FloodlightProvider', |
| 57 | 'net.floodlightcontroller.threadpool.ThreadPool', |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 58 | 'net.onrc.onos.ofcontroller.floodlightlistener.RCNetworkGraphPublisher', |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 59 | 'net.floodlightcontroller.ui.web.StaticWebRoutable', |
| 60 | 'net.onrc.onos.datagrid.HazelcastDatagrid', |
| 61 | 'net.onrc.onos.ofcontroller.flowmanager.FlowManager', |
| 62 | 'net.onrc.onos.ofcontroller.flowprogrammer.FlowProgrammer', |
| 63 | 'net.onrc.onos.ofcontroller.topology.TopologyManager', |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 64 | 'net.onrc.onos.intent.runtime.PathCalcRuntimeModule', |
| 65 | 'net.onrc.onos.intent.runtime.PlanInstallModule', |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 66 | 'net.onrc.onos.registry.controller.ZookeeperRegistry' |
| 67 | ) |
| 68 | |
| 69 | # Additions for reactive forwarding |
| 70 | reactiveModules = ( |
| 71 | 'net.onrc.onos.ofcontroller.proxyarp.ProxyArpManager', |
| 72 | 'net.onrc.onos.ofcontroller.core.config.DefaultConfiguration', |
| 73 | 'net.onrc.onos.ofcontroller.forwarding.Forwarding' |
| 74 | ) |
| 75 | |
| 76 | # Module parameters |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 77 | ofbase = 6633 |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 78 | restbase = 8080 |
| 79 | jmxbase = 7189 |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 80 | |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 81 | fc = 'net.floodlightcontroller.' |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 82 | |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 83 | # Things that vary per ONOS id |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 84 | perNodeConfigBase = { |
| 85 | fc + 'core.FloodlightProvider.openflowport': ofbase, |
| 86 | fc + 'restserver.RestApiServer.port': restbase, |
| 87 | fc + 'core.FloodlightProvider.controllerid': 0 |
| 88 | } |
| 89 | |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 90 | # Things that are static |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 91 | staticConfig = { |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 92 | 'net.onrc.onos.ofcontroller.floodlightlistener.NetworkGraphPublisher.graph_db_store': |
| 93 | 'ramcloud', |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 94 | 'net.floodlightcontroller.core.FloodlightProvider.workerthreads': 16, |
| 95 | 'net.floodlightcontroller.forwarding.Forwarding.idletimeout': 5, |
| 96 | 'net.floodlightcontroller.forwarding.Forwarding.hardtimeout': 0 |
| 97 | } |
| 98 | |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 99 | # Things that are based on onosDir |
| 100 | dirConfig = { |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 101 | 'net.onrc.onos.ofcontroller.floodlightlistener.NetworkGraphPublisher.dbconf': |
| 102 | '%s/conf/ramcloud.conf', |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 103 | 'net.onrc.onos.datagrid.HazelcastDatagrid.datagridConfig': |
| 104 | '%s/conf/hazelcast.xml', |
| 105 | } |
| 106 | |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 107 | proctag = 'mn-onos-id' |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 108 | |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 109 | # List of scripts that we need/use |
| 110 | scripts = ( 'start-zk.sh', 'start-ramcloud-coordinator.sh', |
| 111 | 'start-ramcloud-server.sh', 'start-onos.sh', 'start-rest.sh' ) |
| 112 | |
Bob Lantz | 7a4d610 | 2013-12-17 00:34:55 -0800 | [diff] [blame] | 113 | # For maven debugging |
| 114 | # mvn = 'mvn -o -e -X' |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 115 | |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 116 | def __init__( self, name, n=1, reactive=True, runAsRoot=False, **params): |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 117 | """n: number of ONOS instances to run (1) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 118 | reactive: run in reactive mode (True) |
| 119 | runAsRoot: run ONOS as root (False)""" |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 120 | self.check() |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 121 | self.count = n |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 122 | self.reactive = reactive |
| 123 | self.runAsRoot = runAsRoot |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 124 | self.ids = range( 0, self.count ) |
| 125 | Controller.__init__( self, name, **params ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 126 | self.proxies = [] |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 127 | # We don't need to run as root, and it can interfere |
| 128 | # with starting Zookeeper manually |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 129 | self.user = None |
| 130 | if not self.runAsRoot: |
| 131 | try: |
| 132 | self.user = quietRun( 'who am i' ).split()[ 0 ] |
| 133 | self.sendCmd( 'su', self.user ) |
| 134 | self.waiting = False |
| 135 | except: |
| 136 | warn( '__init__: failed to drop privileges\n' ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 137 | self.cmd( 'mkdir -p', self.logDir ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 138 | # Need to run commands from ONOS dir |
| 139 | self.cmd( 'cd', self.onosDir ) |
| 140 | self.cmd( 'export PATH=$PATH:%s' % self.onosDir ) |
Bob Lantz | 7a4d610 | 2013-12-17 00:34:55 -0800 | [diff] [blame] | 141 | if hasattr( self, 'mvn' ): |
| 142 | self.cmd( 'export MVN="%s"' % self.mvn ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 143 | |
| 144 | def check( self ): |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 145 | "Set onosDir and check for ONOS prerequisites" |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 146 | if not quietRun( 'which java' ): |
| 147 | raise Exception( 'java not found -' |
| 148 | ' make sure it is installed and in $PATH' ) |
| 149 | if not quietRun( 'which mvn' ): |
| 150 | raise Exception( 'Maven (mvn) not found -' |
| 151 | ' make sure it is installed and in $PATH' ) |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 152 | if 'ONOS_HOME' in environ: |
| 153 | self.onosDir = environ[ 'ONOS_HOME' ] |
| 154 | else: |
| 155 | warn( '* $ONOS_HOME is not set - assuming %s\n' % self.onosDir ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 156 | for script in self.scripts: |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 157 | script = path.join( self.onosDir, script ) |
| 158 | if not path.exists( script ): |
| 159 | msg = '%s not found' % script |
| 160 | if 'ONOS_HOME' not in environ: |
| 161 | msg += ' (try setting $ONOS_HOME and/or sudo -E)' |
| 162 | raise Exception( msg ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 163 | |
| 164 | def waitNetstat( self, pid ): |
| 165 | """Wait for pid to show up in netstat |
| 166 | We assume that once a process is listening on some |
| 167 | port, it is ready to go!""" |
| 168 | while True: |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 169 | output = self.cmd( 'sudo netstat -natup | grep %s/' % pid ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 170 | if output: |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 171 | break |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 172 | info( '.' ) |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 173 | time.sleep( 1 ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 174 | info( '\n* Process %d is listening\n' % pid ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 175 | |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 176 | def waitStart( self, procname, pattern, maxWait=10 ): |
| 177 | "Wait for proces to start up and be visible to pgrep" |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 178 | # Check script exit code |
| 179 | exitCode = int( self.cmd( 'echo $?' ) ) |
| 180 | if exitCode != 0: |
| 181 | raise Exception( '%s startup failed with code %d' % |
| 182 | ( procname, exitCode ) ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 183 | info( '* Waiting for %s startup' % procname ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 184 | while True: |
| 185 | result = self.cmd( 'pgrep -f %s' % pattern ) |
| 186 | if result: |
| 187 | break |
| 188 | info( '.' ) |
| 189 | sleep( 1 ) |
| 190 | pid = int( result.split()[ 0 ] ) |
| 191 | return pid |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 192 | |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 193 | def startRamcloud( self, cpu=.6 ): |
| 194 | """Start Ramcloud |
| 195 | cpu: CPU usage limit (in seconds/s)""" |
| 196 | # Edit configuration file |
| 197 | self.cmd( "sed -ibak -e 's/host=.*/host=127.0.0.1/' %s/conf/ramcloud.conf" % |
| 198 | self.onosDir) |
| 199 | # Create a cgroup so Ramcloud doesn't eat all of our CPU |
| 200 | ramcloud = CPULimitedHost( 'ramcloud', inNamespace=False, period_us=5000 ) |
| 201 | ramcloud.setCPUFrac( cpu / numCores() ) |
| 202 | ramcloud.cmd( 'export PATH=%s:$PATH' % self.onosDir ) |
| 203 | ramcloud.cmd( 'export ONOS_LOGDIR=%s' % self.logDir ) |
| 204 | for daemon in 'coordinator', 'server': |
| 205 | ramcloud.cmd( 'start-ramcloud-%s.sh start' % daemon ) |
| 206 | pid = self.waitStart( 'Ramcloud %s' % daemon, 'obj.master/' + daemon ) |
| 207 | self.waitNetstat( pid ) |
| 208 | status = self.cmd( 'start-ramcloud-%s.sh status' % daemon ) |
| 209 | if 'running' not in status: |
| 210 | raise Exception( 'Ramcloud %s startup failed: ' % daemon + status ) |
| 211 | self.ramcloud = ramcloud |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 212 | |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 213 | def stopRamcloud( self ): |
| 214 | "Stop Ramcloud" |
| 215 | for daemon in 'coordinator', 'server': |
| 216 | self.ramcloud.cmd( './start-ramcloud-%s.sh stop' % daemon ) |
| 217 | self.ramcloud.terminate() |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 218 | |
| 219 | def startZookeeper( self, initcfg=True ): |
| 220 | "Start Zookeeper" |
| 221 | # Reinitialize configuration file |
| 222 | if initcfg: |
| 223 | cfg = self.zookeeperDir + '/conf/zoo.cfg' |
| 224 | template = self.zookeeperDir + '/conf/zoo_sample.cfg' |
| 225 | copyfile( template, cfg ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 226 | self.cmd( 'start-zk.sh stop' ) |
| 227 | self.cmd( 'start-zk.sh start' ) |
| 228 | pid = self.waitStart( 'Zookeeper', 'zookeeper' ) |
| 229 | self.waitNetstat( pid ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 230 | status = self.cmd( 'start-zk.sh status' ) |
| 231 | if 'Error' in status: |
| 232 | raise Exception( 'Zookeeper startup failed: ' + status ) |
| 233 | |
| 234 | def stopZookeeper( self ): |
| 235 | "Stop Zookeeper" |
| 236 | self.cmd( 'start-zk.sh stop' ) |
| 237 | |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 238 | def genProperties( self, id, path='/tmp' ): |
| 239 | "Generate ONOS properties file" |
| 240 | filename = path + '/onos-%s.properties' % id |
| 241 | with open( filename, 'w' ) as f: |
| 242 | # Write modules list |
| 243 | modules = list( self.baseModules ) |
| 244 | if self.reactive: |
| 245 | modules += list( self.reactiveModules ) |
| 246 | f.write( 'floodlight.modules = %s\n' % |
| 247 | ',\\\n'.join( modules ) ) |
| 248 | # Write other parameters |
| 249 | for var, val in self.perNodeConfigBase.iteritems(): |
| 250 | if type( val ) is int: |
| 251 | val += id |
| 252 | f.write( '%s = %s\n' % ( var, val ) ) |
| 253 | for var, val in self.staticConfig.iteritems(): |
| 254 | f.write( '%s = %s\n' % ( var, val ) ) |
Bob Lantz | 21b41f6 | 2013-12-18 18:16:38 -0800 | [diff] [blame] | 255 | for var, val in self.dirConfig.iteritems(): |
| 256 | f.write( '%s = %s\n' % ( var, val % self.onosDir) ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 257 | return filename |
| 258 | |
| 259 | def setVars( self, id, propsFile ): |
| 260 | """Set and return environment vars |
| 261 | id: ONOS instance number |
| 262 | propsFile: properties file name""" |
Bob Lantz | 4ed6cff | 2013-12-17 21:43:36 -0800 | [diff] [blame] | 263 | # cassdir = self.cassDir % id |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 264 | logback = self.logbackFile % id |
| 265 | jmxport = self.jmxbase + id |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 266 | logdir = self.logDir |
| 267 | self.cmd( 'export ONOS_LOGDIR=%s' % logdir ) |
| 268 | self.cmd( 'export ONOS_LOGBASE=onos-%d.`hostname`' % id) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 269 | self.cmd( 'export ZOO_LOG_DIR="%s"' % logdir ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 270 | self.cmd( 'export ONOS_LOGBACK="%s"' % logback ) |
| 271 | self.cmd( 'export JMX_PORT=%s' % jmxport ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 272 | self.cmd( 'export JVM_OPTS="-D%s=%s"' % ( |
| 273 | self.proctag, id ) ) |
| 274 | self.cmd( 'export ONOS_PROPS="%s"' % propsFile ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 275 | |
| 276 | def startONOS( self, id ): |
| 277 | """Start ONOS |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 278 | id: new instance number""" |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 279 | start = time.time() |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 280 | self.stopONOS( id ) |
| 281 | propsFile = self.genProperties( id ) |
| 282 | self.setVars( id, propsFile ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 283 | self.cmd( 'start-onos.sh startnokill' ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 284 | # start-onos.sh waits for ONOS startup |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 285 | elapsed = time.time() - start |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 286 | info( '* ONOS %s started in %.2f seconds\n' % ( id, elapsed ) ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 287 | |
| 288 | def stopONOS( self, id ): |
| 289 | """Shut down ONOS |
| 290 | id: identifier for instance""" |
| 291 | pid = self.cmd( "jps -v | grep %s=%s | awk '{print $1}'" % |
| 292 | ( self.proctag, id ) ).strip() |
| 293 | if pid: |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 294 | self.cmd( 'kill', pid ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 295 | |
| 296 | def start( self, *args ): |
| 297 | "Start ONOS instances" |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 298 | info( '* Starting Zookeeper\n' ) |
| 299 | self.startZookeeper() |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 300 | info( '* Starting Ramcloud\n' ) |
| 301 | self.startRamcloud() |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 302 | for id in self.ids: |
| 303 | info( '* Starting ONOS %s\n' % id ) |
| 304 | self.startONOS( id ) |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 305 | self.cmd( 'start-rest.sh start' ) |
| 306 | # Initialize proxies for clist() |
| 307 | self.proxies = [ RemoteController( 'onos-%d' % id, port=(self.ofbase + id ) ) |
| 308 | for id in range( 0, self.count ) ] |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 309 | |
| 310 | def stop( self, *args ): |
| 311 | "Stop ONOS instances" |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 312 | self.cmd( 'start-rest.sh stop' ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 313 | for id in self.ids: |
| 314 | info( '* Stopping ONOS %s\n' % id ) |
| 315 | self.stopONOS( id ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 316 | info( '* Stopping Zookeeper\n' ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 317 | self.stopZookeeper() |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 318 | info( '* Stopping Ramcloud\n' ) |
| 319 | self.stopRamcloud() |
| 320 | for p in self.proxies: |
| 321 | p.stop() |
| 322 | p.proxies = [] |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 323 | |
| 324 | def clist( self ): |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 325 | "Return list of Controller proxies for this ONOS cluster" |
| 326 | return self.proxies |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 327 | |
| 328 | |
| 329 | class OVSSwitchONOS( OVSSwitch ): |
| 330 | "OVS switch which connects to multiple controllers" |
| 331 | def start( self, controllers ): |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 332 | assert len( controllers ) == 1 |
| 333 | c0 = controllers[ 0 ] |
| 334 | assert type( c0 ) == ONOS |
Bob Lantz | 63bbe4c | 2014-02-06 19:29:55 -0800 | [diff] [blame^] | 335 | controllers = c0.clist() |
| 336 | OVSSwitch.start( self, controllers ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 337 | |
| 338 | |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 339 | def waitConnected( switches ): |
| 340 | "Wait until all switches connect to controllers" |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 341 | start = time.time() |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 342 | info( '* Waiting for switches to connect...\n' ) |
| 343 | for s in switches: |
| 344 | info( s ) |
| 345 | while not s.connected(): |
| 346 | info( '.' ) |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 347 | time.sleep( 1 ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 348 | info( ' ' ) |
Bob Lantz | a093082 | 2013-12-17 21:00:41 -0800 | [diff] [blame] | 349 | elapsed = time.time() - start |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 350 | info( '\n* Connected in %.2f seconds\n' % elapsed ) |
| 351 | |
| 352 | |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 353 | controllers = { 'onos': ONOS } |
| 354 | switches = { 'ovso': OVSSwitchONOS } |
| 355 | |
| 356 | |
| 357 | if __name__ == '__main__': |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 358 | # Simple test for ONOS() controller class |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 359 | setLogLevel( 'info' ) |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 360 | size = 2 if len( argv ) != 2 else int( argv[ 1 ] ) |
| 361 | net = Mininet( topo=LinearTopo( size ), |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 362 | controller=partial( ONOS, n=2 ), |
| 363 | switch=OVSSwitchONOS ) |
| 364 | net.start() |
Bob Lantz | 7ead053 | 2013-12-17 20:41:20 -0800 | [diff] [blame] | 365 | waitConnected( net.switches ) |
Bob Lantz | c7c0540 | 2013-12-16 21:22:12 -0800 | [diff] [blame] | 366 | CLI( net ) |
| 367 | net.stop() |