blob: 973385ad9a37da64bdd63cab671be1a97d759af3 [file] [log] [blame]
#!/usr/bin/python
"""
Copyright 2015 Open Networking Foundation (ONF)
Please refer questions to either the onos test mailing list at <onos-test@onosproject.org>,
the System Testing Plans and Results wiki page at <https://wiki.onosproject.org/x/voMg>,
or the System Testing Guide page at <https://wiki.onosproject.org/x/WYQg>
TestON is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
TestON is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with TestON. If not, see <http://www.gnu.org/licenses/>.
"""
"""
Custom topology for Mininet
Author: kelvin@onlab.us
"""
from mininet.topo import Topo
from mininet.net import Mininet
from mininet.node import Host, RemoteController
from mininet.node import Node
from mininet.node import CPULimitedHost
from mininet.link import TCLink
from mininet.cli import CLI
from mininet.log import setLogLevel
from mininet.util import dumpNodeConnections
from mininet.node import ( UserSwitch, OVSSwitch, IVSSwitch )
import sys
coreSwitches = {}
spineSwitches = {}
leafSwitches = {}
endSwitches = {}
allSwitches = {}
# Counters for nodes
totalSwitches = 0
totalEndSwitches = 0
totalHosts = 0
endSwitchCount = 0 # total count of end switch in each row in gui
class spineTopo( Topo ):
def __init__( self, **opts ):
"Create a topology."
Topo.__init__( self, **opts )
def build( self, s, l, c, e, h, **opts ):
"""
s = number of spine switches
l = number of leaf switches
c = number of core
e = number of end switch
h = number of end host
"""
global totalSwitches
global coreSwitches
global spineSwitches
global leafSwitches
global endSwitches
global totalEndSwitches
global totalHosts
global allSwitches
global endSwitchCount
endSwitchCount = e
print "Creating topology with", s,"spine", l,"leaf", c,"core",\
e,"end switches and",h,"host for each end switches"
self.addCore( c )
self.addSpine( s )
self.addLeaf( l )
self.linkLayer( coreSwitches, spineSwitches )
self.linkLayer( spineSwitches, leafSwitches )
self.linkEndSwitch( e, leafSwitches )
self.linkHosts( h )
allSwitches = coreSwitches
allSwitches.update( spineSwitches )
allSwitches.update( leafSwitches )
allSwitches.update( endSwitches )
deviceData = self.createSwitchDict()
self.genCfgJson( deviceData )
def addCore( self, numSwitch ):
global totalSwitches
global coreSwitches
for i in range( numSwitch ):
coreSwitches[ 'core' + str( i + 1 ) ] = self.addSwitch(
's' + str( totalSwitches + 1 ) )
totalSwitches += 1
def addSpine( self, numSwitch ):
global totalSwitches
global spineSwitches
for i in range( numSwitch ):
spineSwitches[ 'spine' + str( i + 1 ) ] = self.addSwitch(
's' + str( totalSwitches + 1 ) )
totalSwitches += 1
def addLeaf( self, numSwitch ):
global totalSwitches
global leafSwitches
for i in range( numSwitch ):
leafSwitches[ 'leaf' + str( i + 1 ) ] = self.addSwitch(
's' + str( totalSwitches + 1 ) )
totalSwitches += 1
def addEnd( self ):
global totalSwitches
global totalEndSwitches
global endSwitches
endSwitches[ 'end' + str( totalEndSwitches + 1 ) ] = self.addSwitch(
's' + str( totalSwitches + 1 ) )
totalSwitches += 1
totalEndSwitches += 1
return endSwitches[ 'end' + str( totalEndSwitches ) ]
def addEndHosts( self ):
global totalHosts
totalHosts += 1
host = self.addHost( 'h' + str( totalHosts ) )
return host
def linkHosts( self, numHosts ):
global endSwitches
switches = sorted( endSwitches.values() )
for sw in switches:
for i in xrange( numHosts ):
self.addLink( sw, self.addEndHosts() )
def linkLayer( self, topLayer, botLayer ):
"""
Description:
The top layer is the upper layer in the spine topology eg. top layer
can be the spine and the bottom layer is the leaf, another is the
core layer is the top layer and the spine is the bottom layer and
so on.
Required:
topLayer - Upper layer in the spine topology to be linked in the
layer below
botLater - Layer that is below the upper layer to be linked at
"""
topSwitches = sorted( topLayer.keys() )
botSwitches = sorted( botLayer.keys() )
for topSw in topSwitches:
for botSw in botSwitches:
self.addLink( topLayer.get( topSw ), botLayer.get( botSw ) )
def linkEndSwitch( self, numSwitch, leafLayer ):
global totalSwitches
global totalEndSwitches
leaf = sorted( leafLayer.values() )
for i in xrange( len( leafSwitches ) ):
if len( leafSwitches ) == 1:
for j in xrange( numSwitch ):
self.addLink( leaf[ 0 ], self.addEnd() )
break
if len( leafSwitches ) == 2:
for j in xrange( numSwitch ):
endSw = self.addEnd()
self.addLink( leaf[ i ], endSw )
self.addLink( leaf[ i + 1 ], endSw )
break
if i == ( len( leafSwitches ) - 1 ) and len( leafSwitches )%2:
for j in xrange( numSwitch ):
self.addLink( leaf[ i ], self.addEnd() )
break
if i == 0:
for j in xrange( numSwitch ):
endSw = self.addEnd()
self.addLink( leaf[ i ], endSw )
self.addLink( leaf[ i + 1 ], endSw )
continue
if i == 1:
continue
if i%2 == 0:
for j in xrange( numSwitch ):
endSw = self.addEnd()
self.addLink( leaf[ i ], endSw )
self.addLink( leaf[ i + 1 ], endSw )
def genCfgJson( self, deviceData ):
import json
configJson = {}
configJson[ "devices" ] = deviceData
with open( 'spine.json', 'w+' ) as outfile:
json.dump( configJson, outfile )
#cfgFile = open( "spine.json" , 'w+' )
#cfgFile.write( configJson )
#cfgFile.close()
def createSwitchDict( self ):
global allSwitches
global endSwitchCount
latitude = 0
longitude = 0
coreLong = -70
spineLong = -80
leafLong = -90
endLat = 30
rowCount = 0 # count of end switches or rows
colOffSet = 0 # off set for end switches; longitude
#for i in xrange( len( allSwitches ) ):
deviceList = []
deviceDict = {}
for sw in allSwitches:
tempSw = allSwitches.get( sw )
uri = str( "{0:0>16}".format( str( hex( int( tempSw[ 1: ] ) )\
).split( "x" )[ 1 ] ) )
mac = str( "{0:0>12}".format( str( hex( int( tempSw[ 1: ] ) )\
).split( "x" )[ 1 ] ) )
if "core" in sw:
latitude = 45
longitude = coreLong
coreLong += 2.5
elif "spine" in sw:
latitude = 40
longitude = spineLong
spineLong += 1.5
elif "leaf" in sw:
latitude = 35
longitude = leafLong
leafLong += 1.5
elif "end" in sw:
# Reset position and move to the right once every
# number of end switches
if rowCount == endSwitchCount:
colOffSet += 2.5
rowCount = 0
endLat = 30
longitude = -80 + colOffSet
latitude = endLat
endLat -= 1
rowCount += 1
tempItem = { "alias": allSwitches.get( sw ) ,
"uri": "of:" + uri,
"mac": mac,
"annotations": { "name": sw,
"latitude": latitude,
"longitude": longitude },
"type": "SWITCH" }
deviceList.append( tempItem )
return deviceList
#def createHostsJson( hostDict ):
topos = { 'spine': ( lambda s=2, l=3, c=1, e=5, h=1: spineTopo( s=s,
l=l,
c=c,
e=e,
h=h) ) }
# HERE THE CODE DEFINITION OF THE TOPOLOGY ENDS
def setupNetwork():
"Create network"
topo = spineTopo()
#if controller_ip == '':
#controller_ip = '10.0.2.2';
# controller_ip = '127.0.0.1';
network = Mininet( topo=topo,
autoSetMacs=True,
controller=None )
network.start()
CLI( network )
network.stop()
if __name__ == '__main__':
setLogLevel( 'info' )
#setLogLevel('debug')
setupNetwork()