TestON/tests/SCPF/SCPFscaleTopo/dependencies/spine.py

#!/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()