NePA TesT: Community Network Emulator

"Library of potentially useful topologies for Mininet"

from mininet.topo import Topo

from mininet.net import Mininet

class TreeTopo( Topo ):

    "Topology for a tree network with a given depth and fanout."

    def __init__( self, depth=1, fanout=2 ):

        super( TreeTopo, self ).__init__()

        # Numbering:  h1..N, s1..M

        self.hostNum = 1

        self.switchNum = 1

        # Build topology

        self.addTree( depth, fanout )

    def addTree( self, depth, fanout ):

        """Add a subtree starting with node n.

           returns: last node added"""

        isSwitch = depth > 0

        if isSwitch:

            node = self.addSwitch( 's%s' % self.switchNum )

            self.switchNum += 1

            for _ in range( fanout ):

                child = self.addTree( depth - 1, fanout )

                self.addLink( node, child )

        else:

            node = self.addHost( 'h%s' % self.hostNum )

            self.hostNum += 1

        return node

def TreeNet( depth=1, fanout=2, **kwargs ):

    "Convenience function for creating tree networks."

    topo = TreeTopo( depth, fanout )

    return Mininet( topo, **kwargs )

class TorusTopo( Topo ):

    """2-D Torus topology

       WARNING: this topology has LOOPS and WILL NOT WORK

       with the default controller or any Ethernet bridge

       without STP turned on! It can be used with STP, e.g.:

       # mn --topo torus,3,3 --switch lxbr,stp=1 --test pingall"""

    def __init__( self, x, y, *args, **kwargs ):

        Topo.__init__( self, *args, **kwargs )

        if x < 3 or y < 3:

            raise Exception( 'Please use 3x3 or greater for compatibility '

                            'with Mininet 2.1.0' )

        hosts, switches, dpid = {}, {}, 0

        # Create and wire interior

        for i in range( 0, x ):

            for j in range( 0, y ):

                loc = '%dx%d' % ( i + 1, j + 1 )

                # dpid cannot be zero for OVS

                dpid = ( i + 1 ) * 256 + ( j + 1 )

                switch = switches[ i, j ] = self.addSwitch( 's' + loc, dpid='%016x' % dpid )

                host = hosts[ i, j ] = self.addHost( 'h' + loc )

                self.addLink( host, switch )

        # Connect switches

        for i in range( 0, x ):

            for j in range( 0, y ):

                sw1 = switches[ i, j ]

                sw2 = switches[ i, ( j + 1 ) % y ]

                sw3 = switches[ ( i + 1 ) % x, j ]

                self.addLink( sw1, sw2 )

                self.addLink( sw1, sw3 )