blob: 13a75e2007899b1ffcc456151f839290f6cdb0fe [file] [log] [blame]
#!/usr/bin/python
"""
Libraries for creating L3 topologies with routing protocols.
"""
from mininet.node import Host, OVSBridge
from mininet.nodelib import NAT
from mininet.log import info, debug, error
from mininet.cli import CLI
from ipaddress import ip_network, ip_address, ip_interface
import os
class RoutedHost(Host):
"""Host that can be configured with multiple IP addresses."""
def __init__(self, name, ips, gateway, *args, **kwargs):
super(RoutedHost, self).__init__(name, *args, **kwargs)
self.ips = ips
self.gateway = gateway
def config(self, **kwargs):
Host.config(self, **kwargs)
self.cmd('ip -4 addr flush dev %s' % self.defaultIntf())
for ip in self.ips:
self.cmd('ip addr add %s dev %s' % (ip, self.defaultIntf()))
self.cmd('ip route add default via %s' % self.gateway)
class RoutedHost6(Host):
"""Host that can be configured with multiple IP addresses."""
def __init__(self, name, ips, gateway, *args, **kwargs):
super(RoutedHost6, self).__init__(name, *args, **kwargs)
self.ips = ips
self.gateway = gateway
def config(self, **kwargs):
Host.config(self, **kwargs)
self.cmd('ip -6 addr flush dev %s' % self.defaultIntf())
for ip in self.ips:
self.cmd('ip -6 addr add %s dev %s' % (ip, self.defaultIntf()))
self.cmd('ip -6 route add default via %s' % self.gateway)
class Router(Host):
"""An L3 router.
Configures the Linux kernel for L3 forwarding and supports rich interface
configuration of IP addresses, MAC addresses and VLANs."""
def __init__(self, name, interfaces, *args, **kwargs):
super(Router, self).__init__(name, **kwargs)
self.interfaces = interfaces
def config(self, **kwargs):
super(Host, self).config(**kwargs)
self.cmd('sysctl net.ipv4.ip_forward=1')
self.cmd('sysctl net.ipv4.conf.all.rp_filter=0')
self.cmd('sysctl net.ipv6.conf.all.forwarding=1')
for intf, configs in self.interfaces.items():
self.cmd('ip -4 addr flush dev %s' % intf)
self.cmd( 'sysctl net.ipv4.conf.%s.rp_filter=0' % intf )
if not isinstance(configs, list):
configs = [configs]
for attrs in configs:
# Configure the vlan if there is one
if 'vlan' in attrs:
vlanName = '%s.%s' % (intf, attrs['vlan'])
self.cmd('ip link add link %s name %s type vlan id %s' %
(intf, vlanName, attrs['vlan']))
self.cmd('ip link set %s up' % vlanName)
addrIntf = vlanName
else:
addrIntf = intf
# Now configure the addresses on the vlan/native interface
if 'mac' in attrs:
self.cmd('ip link set %s down' % addrIntf)
self.cmd('ip link set %s address %s' % (addrIntf, attrs['mac']))
self.cmd('ip link set %s up' % addrIntf)
for addr in attrs['ipAddrs']:
self.cmd('ip addr add %s dev %s' % (addr, addrIntf))
class QuaggaRouter(Router):
"""Runs Quagga to create a router that can speak routing protocols."""
binDir = '/usr/lib/quagga'
logDir = '/var/log/quagga'
def __init__(self, name, interfaces,
defaultRoute=None,
zebraConfFile=None,
protocols= [],
fpm=None,
runDir='/var/run/quagga', *args, **kwargs):
super(QuaggaRouter, self).__init__(name, interfaces, **kwargs)
self.protocols = protocols
self.fpm = fpm
for p in self.protocols:
p.setQuaggaRouter(self)
self.runDir = runDir
self.defaultRoute = defaultRoute
# Ensure required directories exist
try:
original_umask = os.umask(0)
if (not os.path.isdir(QuaggaRouter.logDir)):
os.makedirs(QuaggaRouter.logDir, 0777)
if (not os.path.isdir(self.runDir)):
os.makedirs(self.runDir, 0777)
finally:
os.umask(original_umask)
self.zebraConfFile = zebraConfFile
if (self.zebraConfFile is None):
self.zebraConfFile = '%s/zebrad%s.conf' % (self.runDir, self.name)
self.generateZebra()
self.socket = '%s/zebra%s.api' % (self.runDir, self.name)
self.zebraPidFile = '%s/zebra%s.pid' % (self.runDir, self.name)
def generateZebra(self):
configFile = open(self.zebraConfFile, 'w+')
configFile.write('log file %s/zebrad%s.log\n' % (QuaggaRouter.logDir, self.name))
configFile.write('hostname zebra-%s\n' % self.name)
configFile.write('password %s\n' % 'quagga')
if (self.fpm is not None):
configFile.write('fpm connection ip %s port 2620' % self.fpm)
configFile.close()
def config(self, **kwargs):
super(QuaggaRouter, self).config(**kwargs)
self.cmd('%s/zebra -d -f %s -z %s -i %s'
% (QuaggaRouter.binDir, self.zebraConfFile, self.socket, self.zebraPidFile))
print("\n")
print('%s/zebra -d -f %s -z %s -i %s'
% (QuaggaRouter.binDir, self.zebraConfFile, self.socket, self.zebraPidFile))
for p in self.protocols:
p.config(**kwargs)
if self.defaultRoute:
self.cmd('ip route add default via %s' % self.defaultRoute)
def stopProtocols(self, **kwargs):
for p in self.protocols:
p.stop(**kwargs)
def startProtocols(self, **kwargs):
for p in self.protocols:
p.start(**kwargs)
def terminate(self, **kwargs):
self.cmd("ps ax | grep '%s' | awk '{print $1}' | xargs kill"
% (self.socket))
for p in self.protocols:
p.terminate(**kwargs)
super(QuaggaRouter, self).terminate()
class Protocol(object):
"""Base abstraction of a protocol that the QuaggaRouter can run."""
def setQuaggaRouter(self, qr):
self.qr = qr
def config(self, **kwargs):
pass
def stop(self, **kwargs):
pass
def start(self, **kwargs):
pass
def terminate(self, **kwargs):
pass
class BgpProtocol(Protocol):
"""Configures and runs the BGP protocol in Quagga."""
def __init__(self, configFile=None, asNum=None, neighbors=[], routes=[], *args, **kwargs):
self.configFile = configFile
self.asNum = asNum
self.neighbors = neighbors
self.routes = routes
def config(self, **kwargs):
if self.configFile is None:
self.configFile = '%s/bgpd%s.conf' % (self.qr.runDir, self.qr.name)
self.generateConfig()
bgpdPidFile = '%s/bgpd%s.pid' % (self.qr.runDir, self.qr.name)
self.qr.cmd('%s/bgpd -d -f %s -z %s -i %s'
% (QuaggaRouter.binDir, self.configFile, self.qr.socket, bgpdPidFile))
def stop(self, **kwargs):
self.qr.cmd('pkill -f %s' % self.configFile)
def start(self, **kwargs):
bgpdPidFile = '%s/bgpd%s.pid' % (self.qr.runDir, self.qr.name)
self.qr.cmd('%s/bgpd -d -f %s -z %s -i %s'
% (QuaggaRouter.binDir, self.configFile, self.qr.socket, bgpdPidFile))
def generateConfig(self):
conf = ConfigurationWriter(self.configFile)
def getRouterId(interfaces):
intfAttributes = interfaces.itervalues().next()
print intfAttributes
if isinstance(intfAttributes, list):
# Try use the first set of attributes, but if using vlans they might not have addresses
intfAttributes = intfAttributes[1] if not intfAttributes[0]['ipAddrs'] else intfAttributes[0]
return intfAttributes['ipAddrs'][0].split('/')[0]
conf.writeLine('log file %s/bgpd%s.log' % (QuaggaRouter.logDir, self.qr.name))
conf.writeLine('hostname bgp-%s' % self.qr.name)
conf.writeLine('password %s' % 'quagga')
conf.writeLine('!')
conf.writeLine('router bgp %s' % self.asNum)
conf.indent()
conf.writeLine('bgp router-id %s' % getRouterId(self.qr.interfaces))
conf.writeLine('timers bgp %s' % '3 9')
conf.writeLine('!')
for neighbor in self.neighbors:
conf.writeLine('neighbor %s remote-as %s' % (neighbor['address'], neighbor['as']))
conf.writeLine('neighbor %s ebgp-multihop' % neighbor['address'])
conf.writeLine('neighbor %s timers connect %s' % (neighbor['address'], '5'))
conf.writeLine('neighbor %s advertisement-interval %s' % (neighbor['address'], '5'))
if 'port' in neighbor:
conf.writeLine('neighbor %s port %s' % (neighbor['address'], neighbor['port']))
conf.writeLine('!')
for route in self.routes:
conf.writeLine('network %s' % route)
conf.close()
class OspfProtocol(Protocol):
"""Configures and runs the OSPF protocol in Quagga."""
def __init__(self, configFile=None, *args, **kwargs):
self.configFile = configFile
def config(self, **kwargs):
if self.configFile is None:
self.configFile = '%s/ospfd%s.conf' % (self.qr.runDir, self.qr.name)
self.generateConfig()
ospfPidFile = '%s/ospf%s.pid' % (self.qr.runDir, self.qr.name)
self.qr.cmd('%s/ospfd -d -f %s -z %s -i %s'
% (QuaggaRouter.binDir, self.configFile, self.qr.socket, ospfPidFile))
def generateConfig(self):
conf = ConfigurationWriter(self.configFile)
def getRouterId(interfaces):
intfAttributes = interfaces.itervalues().next()
print intfAttributes
if isinstance(intfAttributes, list):
# Try use the first set of attributes, but if using vlans they might not have addresses
intfAttributes = intfAttributes[1] if not intfAttributes[0]['ipAddrs'] else intfAttributes[0]
return intfAttributes['ipAddrs'][0].split('/')[0]
conf.writeLine('hostname ospf-%s' % self.qr.name)
conf.writeLine('password %s' % 'hello')
conf.writeLine('!')
conf.writeLine('router ospf')
conf.indent()
conf.writeLine('ospf router-id %s' % getRouterId(self.qr.interfaces))
conf.writeLine('!')
for name, intf in self.qr.interfaces.items():
for ip in intf['ipAddrs']:
conf.writeLine('network %s area 0' % ip)
# if intf['ipAddrs'][0].startswith('192.168'):
# writeLine(1, 'passive-interface %s' % name)
conf.close()
class PimProtocol(Protocol):
"""Configures and runs the PIM protcol in Quagga."""
def __init__(self, configFile=None, *args, **kwargs):
self.configFile = configFile
def config(self, **kwargs):
pimPidFile = '%s/pim%s.pid' % (self.qr.runDir, self.qr.name)
self.qr.cmd('%s/pimd -Z -d -f %s -z %s -i %s'
% (QuaggaRouter.binDir, self.configFile, self.qr.socket, pimPidFile))
class ConfigurationWriter(object):
"""Utility class for writing a configuration file."""
def __init__(self, filename):
self.filename = filename
self.indentValue = 0
self.configFile = open(self.filename, 'w+')
def indent(self):
self.indentValue += 1
def unindent(self):
if (self.indentValue > 0):
self.indentValue -= 1
def write(self, string):
self.configFile.write(string)
def writeLine(self, string):
intentStr = ''
for _ in range(0, self.indentValue):
intentStr += ' '
self.write('%s%s\n' % (intentStr, string))
def close(self):
self.configFile.close()
# Backward compatibility for BGP-only use case
class BgpRouter(QuaggaRouter):
"""Quagga router running the BGP protocol."""
def __init__(self, name, interfaces,
asNum=0, neighbors=[], routes=[],
defaultRoute=None,
quaggaConfFile=None,
zebraConfFile=None,
*args, **kwargs):
bgp = BgpProtocol(configFile=quaggaConfFile, asNum=asNum, neighbors=neighbors, routes=routes)
super(BgpRouter, self).__init__(name, interfaces,
zebraConfFile=zebraConfFile,
defaultRoute=defaultRoute,
protocols=[bgp],
*args, **kwargs)
class RouterData(object):
"""Internal data structure storing information about a router."""
def __init__(self, index):
self.index = index
self.neighbors = []
self.interfaces = {}
self.switches = []
def addNeighbor(self, theirAddress, theirAsNum):
self.neighbors.append({'address': theirAddress.ip, 'as': theirAsNum})
def addInterface(self, intf, vlan, address):
if intf not in self.interfaces:
self.interfaces[intf] = InterfaceData(intf)
self.interfaces[intf].addAddress(vlan, address)
def setSwitch(self, switch):
self.switches.append(switch)
class InterfaceData(object):
"""Internal data structure storing information about an interface."""
def __init__(self, number):
self.number = number
self.addressesByVlan = {}
def addAddress(self, vlan, address):
if vlan not in self.addressesByVlan:
self.addressesByVlan[vlan] = []
self.addressesByVlan[vlan].append(address.with_prefixlen)
class RoutedNetwork(object):
"""Creates a host behind a router. This is common boilerplate topology
segment in routed networks."""
@staticmethod
def build(topology, router, hostName, networks):
# There's a convention that the router's addresses are already set up,
# and it has the last address in the network.
def getFirstAddress(network):
return '%s/%s' % (network[1], network.prefixlen)
defaultRoute = AutonomousSystem.getLastAddress(networks[0]).ip
host = topology.addHost(hostName, cls=RoutedHost,
ips=[getFirstAddress(network) for network in networks],
gateway=defaultRoute)
topology.addLink(router, host)
class AutonomousSystem(object):
"""Base abstraction of an autonomous system, which implies some internal
topology and connections to other topology elements (switches/other ASes)."""
psIdx = 1
def __init__(self, asNum, numRouters):
self.asNum = asNum
self.numRouters = numRouters
self.routers = {}
for i in range(1, numRouters + 1):
self.routers[i] = RouterData(i)
self.routerNodes = {}
self.neighbors = []
self.vlanAddresses = {}
def peerWith(self, myRouter, myAddress, theirAddress, theirAsNum, intf=1, vlan=None):
router = self.routers[myRouter]
router.addInterface(intf, vlan, myAddress)
router.addNeighbor(theirAddress, theirAsNum)
def getRouter(self, i):
return self.routerNodes[i]
@staticmethod
def generatePeeringAddresses():
network = ip_network(u'10.0.%s.0/24' % AutonomousSystem.psIdx)
AutonomousSystem.psIdx += 1
return ip_interface('%s/%s' % (network[1], network.prefixlen)), \
ip_interface('%s/%s' % (network[2], network.prefixlen))
@staticmethod
def addPeering(as1, as2, router1=1, router2=1, intf1=1, intf2=1, address1=None, address2=None, useVlans=False):
vlan = AutonomousSystem.psIdx if useVlans else None
if address1 is None or address2 is None:
(address1, address2) = AutonomousSystem.generatePeeringAddresses()
as1.peerWith(router1, address1, address2, as2.asNum, intf=intf1, vlan=vlan)
as2.peerWith(router2, address2, address1, as1.asNum, intf=intf2, vlan=vlan)
@staticmethod
def getLastAddress(network):
return ip_interface(network.network_address + network.num_addresses - 2)
@staticmethod
def getIthAddress(network, i):
return ip_interface('%s/%s' % (network[i], network.prefixlen))
class BasicAutonomousSystem(AutonomousSystem):
"""Basic autonomous system containing one host and one or more routers
which peer with other ASes."""
def __init__(self, num, routes, numRouters=1):
super(BasicAutonomousSystem, self).__init__(65000+num, numRouters)
self.num = num
self.routes = routes
def addLink(self, switch, router=1):
self.routers[router].setSwitch(switch)
def build(self, topology):
self.addRouterAndHost(topology)
def addRouterAndHost(self, topology):
# TODO implementation is messy and needs to be cleaned up
intfs = {}
router = self.routers[1]
for i, router in self.routers.items():
# routerName = 'r%i%i' % (self.num, i)
routerName = 'r%i' % self.num
if not i == 1:
routerName += ('%i' % i)
hostName = 'h%i' % self.num
for j, interface in router.interfaces.items():
nativeAddresses = interface.addressesByVlan.pop(None, [])
peeringIntf = [{'mac' : '00:00:%02x:00:%02x:%02x' % (self.num, i, j),
'ipAddrs' : nativeAddresses}]
for vlan, addresses in interface.addressesByVlan.items():
peeringIntf.append({'vlan': vlan,
'mac': '00:00:%02x:%02x:%02x:%02x' % (self.num, vlan, i, j),
'ipAddrs': addresses})
intfs.update({'%s-eth%s' % (routerName, j-1) : peeringIntf})
# Only add the host to the first router for now
if i == 1:
internalAddresses = []
for route in self.routes:
internalAddresses.append('%s/%s' % (AutonomousSystem.getLastAddress(route).ip, route.prefixlen))
internalIntf = {'ipAddrs' : internalAddresses}
# This is the configuration of the next interface after all the peering interfaces
intfs.update({'%s-eth%s' % (routerName, len(router.interfaces.keys())) : internalIntf})
routerNode = topology.addHost(routerName,
asNum=self.asNum, neighbors=router.neighbors,
routes=self.routes,
cls=BgpRouter, interfaces=intfs)
self.routerNodes[i] = routerNode
for switch in router.switches:
topology.addLink(switch, routerNode)
# Only add the host to the first router for now
if i == 1:
defaultRoute = internalAddresses[0].split('/')[0]
host = topology.addHost(hostName, cls=RoutedHost,
ips=[self.getFirstAddress(route) for route in self.routes],
gateway=defaultRoute)
topology.addLink(routerNode, host)
# def getLastAddress(self, network):
# return ip_address(network.network_address + network.num_addresses - 2)
def getFirstAddress(self, network):
return '%s/%s' % (network[1], network.prefixlen)
# TODO fix this AS - doesn't currently work
class RouteServerAutonomousSystem(BasicAutonomousSystem):
def __init__(self, routerAddress, *args, **kwargs):
BasicAutonomousSystem.__init__(self, *args, **kwargs)
self.routerAddress = routerAddress
def build(self, topology, connectAtSwitch):
switch = topology.addSwitch('as%isw' % self.num, cls=OVSBridge)
self.addRouterAndHost(topology, self.routerAddress, switch)
rsName = 'rs%i' % self.num
routeServer = topology.addHost(rsName,
self.asnum, self.neighbors,
cls=BgpRouter,
interfaces={'%s-eth0' % rsName : {'ipAddrs': [self.peeringAddress]}})
topology.addLink(routeServer, switch)
topology.addLink(switch, connectAtSwitch)
class SdnAutonomousSystem(AutonomousSystem):
"""Runs the internal BGP speakers needed for ONOS routing apps like
SDN-IP."""
routerIdx = 1
def __init__(self, onosIps, num=1, numBgpSpeakers=1, asNum=65000, externalOnos=True,
peerIntfConfig=None, withFpm=False):
super(SdnAutonomousSystem, self).__init__(asNum, numBgpSpeakers)
self.onosIps = onosIps
self.num = num
self.numBgpSpeakers = numBgpSpeakers
self.peerIntfConfig = peerIntfConfig
self.withFpm = withFpm
self.externalOnos = externalOnos
self.internalPeeringSubnet = ip_network(u'1.1.1.0/24')
for router in self.routers.values():
# Add iBGP sessions to ONOS nodes
for onosIp in onosIps:
router.neighbors.append({'address': onosIp, 'as': asNum, 'port': 2000})
# Add iBGP sessions to other BGP speakers
for i, router2 in self.routers.items():
if router == router2:
continue
cpIpBase = self.num*10
ip = AutonomousSystem.getIthAddress(self.internalPeeringSubnet, cpIpBase+i)
router.neighbors.append({'address': ip.ip, 'as': asNum})
def build(self, topology, connectAtSwitch, controlSwitch):
natIp = AutonomousSystem.getLastAddress(self.internalPeeringSubnet)
for i, router in self.routers.items():
num = SdnAutonomousSystem.routerIdx
SdnAutonomousSystem.routerIdx += 1
name = 'bgp%s' % num
cpIpBase = self.num*10
ip = AutonomousSystem.getIthAddress(self.internalPeeringSubnet, cpIpBase+i)
eth0 = { 'ipAddrs' : [ str(ip) ] }
if self.peerIntfConfig is not None:
eth1 = self.peerIntfConfig
else:
nativeAddresses = router.interfaces[1].addressesByVlan.pop(None, [])
eth1 = [{ 'mac': '00:00:00:00:00:%02x' % num,
'ipAddrs' : nativeAddresses }]
for vlan, addresses in router.interfaces[1].addressesByVlan.items():
eth1.append({'vlan': vlan,
'mac': '00:00:00:%02x:%02x:00' % (num, vlan),
'ipAddrs': addresses})
intfs = { '%s-eth0' % name : eth0,
'%s-eth1' % name : eth1 }
bgp = topology.addHost( name, cls=BgpRouter, asNum=self.asNum,
neighbors=router.neighbors,
interfaces=intfs,
defaultRoute=str(natIp.ip),
fpm=self.onosIps[0] if self.withFpm else None )
topology.addLink( bgp, controlSwitch )
topology.addLink( bgp, connectAtSwitch )
if self.externalOnos:
nat = topology.addHost('nat', cls=NAT,
ip='%s/%s' % (natIp.ip, self.internalPeeringSubnet.prefixlen),
subnet=str(self.internalPeeringSubnet), inNamespace=False)
topology.addLink(controlSwitch, nat)
def generateRoutes(baseRange, numRoutes, subnetSize=None):
baseNetwork = ip_network(baseRange)
# We need to get at least 2 addresses out of each subnet, so the biggest
# prefix length we can have is /30
maxPrefixLength = baseNetwork.max_prefixlen - 2
if subnetSize is not None:
return list(baseNetwork.subnets(new_prefix=subnetSize))
trySubnetSize = baseNetwork.prefixlen + 1
while trySubnetSize <= maxPrefixLength and \
len(list(baseNetwork.subnets(new_prefix=trySubnetSize))) < numRoutes:
trySubnetSize += 1
if trySubnetSize > maxPrefixLength:
raise Exception("Can't get enough routes from input parameters")
return list(baseNetwork.subnets(new_prefix=trySubnetSize))[:numRoutes]
class RoutingCli( CLI ):
"""CLI command that can bring a host up or down. Useful for simulating router failure."""
def do_host( self, line ):
args = line.split()
if len(args) != 2:
error( 'invalid number of args: host <host name> {up, down}\n' )
return
host = args[ 0 ]
command = args[ 1 ]
if host not in self.mn or self.mn.get( host ) not in self.mn.hosts:
error( 'invalid host: %s\n' % args[ 1 ] )
else:
if command == 'up':
op = 'up'
elif command == 'down':
op = 'down'
else:
error( 'invalid command: host <host name> {up, down}\n' )
return
for intf in self.mn.get( host ).intfList( ):
intf.link.intf1.ifconfig( op )
intf.link.intf2.ifconfig( op )