| #!/usr/bin/env python |
| |
| import time |
| import pexpect |
| import struct |
| import fcntl |
| import os |
| import sys |
| import signal |
| import sys |
| import re |
| import json |
| sys.path.append( "../" ) |
| from drivers.common.clidriver import CLI |
| |
| |
| class QuaggaCliDriver( CLI ): |
| |
| def __init__( self ): |
| super( CLI, self ).__init__() |
| |
| # TODO: simplify this method |
| def connect( self, **connectargs ): |
| for key in connectargs: |
| vars( self )[ key ] = connectargs[ key ] |
| self.name = self.options[ 'name' ] |
| self.handle = super( QuaggaCliDriver, self ).connect( |
| user_name=self.user_name, |
| ip_address="127.0.0.1", |
| port=self.port, |
| pwd=self.pwd ) |
| if self.handle: |
| return self.handle |
| else: |
| main.log.info( "NO HANDLE" ) |
| return main.FALSE |
| |
| |
| def connectQuagga( self ): |
| self.name = self.options[ 'name' ] |
| # self.handle = super( QuaggaCliDriver,self ).connect( |
| # user_name=self.user_name, ip_address=self.ip_address,port=self.port, |
| # pwd=self.pwd ) |
| self.handle = super( QuaggaCliDriver, self ).connect( |
| user_name=self.user_name, |
| ip_address="1.1.1.1", |
| port=self.port, |
| pwd=self.pwd ) |
| main.log.info( "connect parameters:" + str( self.user_name ) + ";" |
| + str( self.ip_address ) + ";" + str( self.port ) |
| + ";" + str(self.pwd ) ) |
| |
| if self.handle: |
| # self.handle.expect( "",timeout=10 ) |
| # self.handle.expect( "\$",timeout=10 ) |
| self.handle.sendline( "telnet localhost 2605" ) |
| # self.handle.expect( "Password:", timeout=5 ) |
| self.handle.expect( "Password:" ) |
| self.handle.sendline( "hello" ) |
| # self.handle.expect( "bgpd", timeout=5 ) |
| self.handle.expect( "bgpd" ) |
| self.handle.sendline( "enable" ) |
| # self.handle.expect( "bgpd#", timeout=5 ) |
| self.handle.expect( "bgpd#" ) |
| return self.handle |
| else: |
| main.log.info( "NO HANDLE" ) |
| return main.FALSE |
| |
| def loginQuagga( self, ip_address ): |
| self.name = self.options[ 'name' ] |
| self.handle = super( QuaggaCliDriver, self ).connect( |
| user_name=self.user_name, ip_address=ip_address, |
| port=self.port, pwd=self.pwd ) |
| main.log.info( "connect parameters:" + str( self.user_name ) + ";" |
| + str( self.ip_address ) + ";" + str( self.port ) |
| + ";" + str( self.pwd ) ) |
| |
| if self.handle: |
| # self.handle.expect( "" ) |
| # self.handle.expect( "\$" ) |
| self.handle.sendline( "telnet localhost 2605" ) |
| # self.handle.expect( "Password:", timeout=5 ) |
| self.handle.expect( "Password:" ) |
| self.handle.sendline( "hello" ) |
| # self.handle.expect( "bgpd", timeout=5 ) |
| self.handle.expect( "bgpd" ) |
| self.handle.sendline( "enable" ) |
| # self.handle.expect( "bgpd#", timeout=5 ) |
| self.handle.expect( "bgpd#" ) |
| main.log.info( "I am in quagga on host " + str( ip_address ) ) |
| |
| return self.handle |
| else: |
| main.log.info( "NO HANDLE" ) |
| return main.FALSE |
| |
| def enterConfig( self, asn ): |
| main.log.info( "I am in enter_config method!" ) |
| try: |
| self.handle.sendline( "" ) |
| self.handle.expect( "bgpd#" ) |
| except: |
| main.log.warn( "Probably not currently in enable mode!" ) |
| self.disconnect() |
| return main.FALSE |
| self.handle.sendline( "configure terminal" ) |
| self.handle.expect( "config", timeout=5 ) |
| routerAS = "router bgp " + str( asn ) |
| try: |
| self.handle.sendline( routerAS ) |
| self.handle.expect( "config-router", timeout=5 ) |
| return main.TRUE |
| except: |
| return main.FALSE |
| |
| def generatePrefixes( self, net, numRoutes ): |
| main.log.info( "I am in generate_prefixes method!" ) |
| |
| # each IP prefix is composed by "net" + "." + m + "." + n + "." + x |
| # the length of each IP prefix is 24 |
| routes = [] |
| routesGen = 0 |
| m = numRoutes / 256 |
| n = numRoutes % 256 |
| |
| for i in range( 0, m ): |
| for j in range( 0, 256 ): |
| network = str( net ) + "." + str( i ) + "." + str( j ) \ |
| + ".0/24" |
| routes.append( network ) |
| routesGen = routesGen + 1 |
| |
| for j in range( 0, n ): |
| network = str( net ) + "." + str( m ) + "." + str( j ) + ".0/24" |
| routes.append( network ) |
| routesGen = routesGen + 1 |
| |
| if routesGen == numRoutes: |
| main.log.info( "Successfully generated " + str( numRoutes ) |
| + " prefixes!" ) |
| return routes |
| return main.FALSE |
| |
| # This method generates a multiple to single point intent( |
| # MultiPointToSinglePointIntent ) for a given route |
| def generateExpectedSingleRouteIntent( self, prefix, nextHop, nextHopMac, |
| sdnipData ): |
| |
| ingresses = [] |
| egress = "" |
| for peer in sdnipData[ 'bgpPeers' ]: |
| if peer[ 'ipAddress' ] == nextHop: |
| egress = "of:" + str( |
| peer[ 'attachmentDpid' ] ).replace( ":", "" ) + ":" \ |
| + str( peer[ 'attachmentPort' ] ) |
| for peer in sdnipData[ 'bgpPeers' ]: |
| if not peer[ 'ipAddress' ] == nextHop: |
| ingress = "of:" + str( |
| peer[ 'attachmentDpid' ] ).replace( ":", "" ) + ":" \ |
| + str( peer[ 'attachmentPort' ] ) |
| if not ingress == egress and ingress not in ingresses: |
| ingresses.append( ingress ) |
| # ingresses.append( "of:" + str( peer[ 'attachmentDpid' ] |
| # ).replace( ":", "" ) + ":" + str( peer[ 'attachmentPort' |
| # ] ) ) |
| |
| selector = "ETH_TYPE{ethType=800},IPV4_DST{ip=" + prefix + "}" |
| treatment = "[ETH_DST{mac=" + str( nextHopMac ) + "}]" |
| |
| intent = egress + "/" + str( sorted( ingresses ) ) + "/" + \ |
| selector + "/" + treatment |
| return intent |
| |
| def generateExpectedOnePeerRouteIntents( self, prefixes, nextHop, |
| nextHopMac, sdnipJsonFilePath ): |
| intents = [] |
| sdnipJsonFile = open( sdnipJsonFilePath ).read() |
| |
| sdnipData = json.loads( sdnipJsonFile ) |
| |
| for prefix in prefixes: |
| intents.append( |
| self.generateExpectedSingleRouteIntent( |
| prefix, nextHop, nextHopMac, sdnipData ) ) |
| return sorted( intents ) |
| |
| # TODO |
| # This method generates all expected route intents for all BGP peers |
| def generateExpectedRouteIntents( self ): |
| intents = [] |
| return intents |
| |
| # This method extracts all actual routes from ONOS CLI |
| def extractActualRoutes( self, getRoutesResult ): |
| routesJsonObj = json.loads( getRoutesResult ) |
| |
| allRoutesActual = [] |
| for route in routesJsonObj['routes4']: |
| if 'prefix' in route: |
| if route[ 'prefix' ] == '172.16.10.0/24': |
| continue |
| allRoutesActual.append( |
| route[ 'prefix' ] + "/" + route[ 'nextHop' ] ) |
| |
| return sorted( allRoutesActual ) |
| |
| # This method extracts all actual route intents from ONOS CLI |
| def extractActualRouteIntents( self, getIntentsResult ): |
| intents = [] |
| # TODO: delete the line below when change to Mininet demo script |
| # getIntentsResult=open( "../tests/SdnIpTest/intents.json" ).read() |
| intentsJsonObj = json.loads( getIntentsResult ) |
| |
| for intent in intentsJsonObj: |
| if intent[ 'appId' ] != "org.onosproject.sdnip": |
| continue |
| if intent[ 'type' ] == "MultiPointToSinglePointIntent" \ |
| and intent[ 'state' ] == 'INSTALLED': |
| egress = str( intent[ 'egress' ][ 'device' ] ) + ":" \ |
| + str( intent[ 'egress' ][ 'port' ] ) |
| ingress = [] |
| for attachmentPoint in intent[ 'ingress' ]: |
| ingress.append( |
| str( attachmentPoint[ 'device' ] ) + ":" |
| + str( attachmentPoint[ 'port' ] ) ) |
| |
| selector = intent[ 'selector' ].replace( |
| "[", "" ).replace( "]", "" ).replace( " ", "" ) |
| if str( selector ).startswith( "IPV4" ): |
| str1, str2 = str( selector ).split( "," ) |
| selector = str2 + "," + str1 |
| |
| intent = egress + "/" + str( sorted( ingress ) ) + "/" + \ |
| selector + "/" + intent[ 'treatment' ] |
| intents.append( intent ) |
| return sorted( intents ) |
| |
| # This method extracts all actual BGP intents from ONOS CLI |
| def extractActualBgpIntents( self, getIntentsResult ): |
| intents = [] |
| # TODO: delete the line below when change to Mininet demo script |
| # getIntentsResult=open( "../tests/SdnIpTest/intents.json" ).read() |
| intentsJsonObj = json.loads( getIntentsResult ) |
| |
| for intent in intentsJsonObj: |
| if intent[ 'appId' ] != "org.onosproject.sdnip": |
| continue |
| if intent[ 'type' ] == "PointToPointIntent" \ |
| and "protocol=6" in str( intent[ 'selector' ] ): |
| ingress = str( intent[ 'ingress' ][ 'device' ] ) + ":" \ |
| + str( intent[ 'ingress' ][ 'port' ] ) |
| egress = str( intent[ 'egress' ][ 'device' ] ) + ":" + \ |
| str( intent[ 'egress' ][ 'port' ] ) |
| selector = str( intent[ 'selector' ] ).replace( " ", "" )\ |
| .replace( "[", "" ).replace( "]", "" ).split( "," ) |
| intent = ingress + "/" + egress + "/" + \ |
| str( sorted( selector ) ) |
| intents.append( intent ) |
| |
| return sorted( intents ) |
| |
| # This method generates a single point to single point intent( |
| # PointToPointIntent ) for BGP path |
| def generateExpectedBgpIntents( self, sdnipJsonFilePath ): |
| from operator import eq |
| |
| sdnipJsonFile = open( sdnipJsonFilePath ).read() |
| sdnipData = json.loads( sdnipJsonFile ) |
| |
| intents = [] |
| bgpPeerAttachmentPoint = "" |
| bgpSpeakerAttachmentPoint = "of:" + str( |
| sdnipData[ 'bgpSpeakers' ][ 0 ][ 'attachmentDpid' ] )\ |
| .replace( ":", "" ) + ":" \ |
| + str( sdnipData[ 'bgpSpeakers' ][ 0 ][ 'attachmentPort' ] ) |
| for peer in sdnipData[ 'bgpPeers' ]: |
| bgpPeerAttachmentPoint = "of:" \ |
| + str( peer[ 'attachmentDpid' ] ).replace( ":", "" ) \ |
| + ":" + str( peer[ 'attachmentPort' ] ) |
| # find out the BGP speaker IP address for this BGP peer |
| bgpSpeakerIpAddress = "" |
| for interfaceAddress in \ |
| sdnipData[ 'bgpSpeakers' ][ 0 ][ 'interfaceAddresses' ]: |
| # if eq( interfaceAddress[ 'interfaceDpid' ],sdnipData[ |
| # 'bgpSpeakers' ][ 0 ][ 'attachmentDpid' ] ) and eq( |
| # interfaceAddress[ 'interfacePort' ], sdnipData[ 'bgpSpeakers' |
| # ][ 0 ][ 'attachmentPort' ] ): |
| if eq( interfaceAddress[ 'interfaceDpid' ], |
| peer[ 'attachmentDpid' ] ) \ |
| and eq( interfaceAddress[ 'interfacePort' ], |
| peer[ 'attachmentPort' ] ): |
| bgpSpeakerIpAddress = interfaceAddress[ 'ipAddress' ] |
| break |
| else: |
| continue |
| |
| # from bgpSpeakerAttachmentPoint to bgpPeerAttachmentPoint |
| # direction |
| selectorStr = "IPV4_SRC{ip=" + bgpSpeakerIpAddress + "/32}," \ |
| + "IPV4_DST{ip=" + peer[ 'ipAddress' ] + "/32}," \ |
| + "IP_PROTO{protocol=6}, ETH_TYPE{ethType=800}, \ |
| TCP_DST{tcpPort=179}" |
| selector = selectorStr.replace( " ", "" ).replace("[", "" )\ |
| .replace( "]", "" ).split( "," ) |
| intent = bgpSpeakerAttachmentPoint + "/" + \ |
| bgpPeerAttachmentPoint + "/" + str( sorted( selector ) ) |
| intents.append( intent ) |
| |
| selectorStr = "IPV4_SRC{ip=" + bgpSpeakerIpAddress + "/32}," \ |
| + "IPV4_DST{ip=" + peer[ 'ipAddress' ] + "/32}," \ |
| + "IP_PROTO{protocol=6}, ETH_TYPE{ethType=800}, \ |
| TCP_SRC{tcpPort=179}" |
| selector = selectorStr.replace( " ", "" ).replace("[", "" )\ |
| .replace( "]", "" ).split( "," ) |
| intent = bgpSpeakerAttachmentPoint + "/" \ |
| + bgpPeerAttachmentPoint + "/" + str( sorted( selector ) ) |
| intents.append( intent ) |
| |
| # from bgpPeerAttachmentPoint to bgpSpeakerAttachmentPoint |
| # direction |
| selectorStr = "IPV4_SRC{ip=" + peer[ 'ipAddress' ] + "/32}," \ |
| + "IPV4_DST{ip=" + bgpSpeakerIpAddress + "/32}," \ |
| + "IP_PROTO{protocol=6}, ETH_TYPE{ethType=800}, \ |
| TCP_DST{tcpPort=179}" |
| selector = selectorStr.replace( " ", "" ).replace("[", "" )\ |
| .replace( "]", "" ).split( "," ) |
| intent = bgpPeerAttachmentPoint + "/" \ |
| + bgpSpeakerAttachmentPoint + "/" + str( sorted( selector ) ) |
| intents.append( intent ) |
| |
| selectorStr = "IPV4_SRC{ip=" + peer[ 'ipAddress' ] + "/32}," \ |
| + "IPV4_DST{ip=" + bgpSpeakerIpAddress + "/32}," \ |
| + "IP_PROTO{protocol=6}, ETH_TYPE{ethType=800}, \ |
| TCP_SRC{tcpPort=179}" |
| selector = selectorStr.replace( " ", "" ).replace( "[", "" )\ |
| .replace( "]", "" ).split( "," ) |
| intent = bgpPeerAttachmentPoint + "/" \ |
| + bgpSpeakerAttachmentPoint + "/" + str( sorted( selector ) ) |
| intents.append( intent ) |
| |
| return sorted( intents ) |
| |
| def addRoutes( self, routes, routeRate ): |
| main.log.info( "I am in add_routes method!" ) |
| |
| routesAdded = 0 |
| try: |
| self.handle.sendline( "" ) |
| # self.handle.expect( "config-router" ) |
| self.handle.expect( "config-router", timeout=5 ) |
| except: |
| main.log.warn( "Probably not in config-router mode!" ) |
| self.disconnect() |
| main.log.info( "Start to add routes" ) |
| |
| for i in range( 0, len( routes ) ): |
| routeCmd = "network " + routes[ i ] |
| try: |
| self.handle.sendline( routeCmd ) |
| self.handle.expect( "bgpd", timeout=5 ) |
| except: |
| main.log.warn( "Failed to add route" ) |
| self.disconnect() |
| # waitTimer = 1.00 / routeRate |
| # time.sleep( waitTimer ) |
| if routesAdded == len( routes ): |
| main.log.info( "Finished adding routes" ) |
| return main.TRUE |
| return main.FALSE |
| |
| def deleteRoutes( self, routes, routeRate ): |
| main.log.info( "I am in delete_routes method!" ) |
| |
| routesAdded = 0 |
| try: |
| self.handle.sendline( "" ) |
| # self.handle.expect( "config-router" ) |
| self.handle.expect( "config-router", timeout=5 ) |
| except: |
| main.log.warn( "Probably not in config-router mode!" ) |
| self.disconnect() |
| main.log.info( "Start to delete routes" ) |
| |
| for i in range( 0, len( routes ) ): |
| routeCmd = "no network " + routes[ i ] |
| try: |
| self.handle.sendline( routeCmd ) |
| self.handle.expect( "bgpd", timeout=5 ) |
| except: |
| main.log.warn( "Failed to delete route" ) |
| self.disconnect() |
| # waitTimer = 1.00 / routeRate |
| # time.sleep( waitTimer ) |
| if routesAdded == len( routes ): |
| main.log.info( "Finished deleting routes" ) |
| return main.TRUE |
| return main.FALSE |
| |
| def pingTest( self, ip_address, pingTestFile, pingTestResultFile ): |
| main.log.info( "Start the ping test on host:" + str( ip_address ) ) |
| |
| self.name = self.options[ 'name' ] |
| self.handle = super( QuaggaCliDriver, self ).connect( |
| user_name=self.user_name, ip_address=ip_address, |
| port=self.port, pwd=self.pwd ) |
| main.log.info( "connect parameters:" + str( self.user_name ) + ";" |
| + str( self.ip_address ) + ";" + str( self.port ) |
| + ";" + str( self.pwd ) ) |
| |
| if self.handle: |
| # self.handle.expect( "" ) |
| # self.handle.expect( "\$" ) |
| main.log.info( "I in host " + str( ip_address ) ) |
| main.log.info( pingTestFile + " > " + pingTestResultFile + " &" ) |
| self.handle.sendline( |
| pingTestFile + |
| " > " + |
| pingTestResultFile + |
| " &" ) |
| self.handle.expect( "\$", timeout=60 ) |
| handle = self.handle.before |
| |
| return handle |
| else: |
| main.log.info( "NO HANDLE" ) |
| return main.FALSE |
| |
| def pingTestAndCheckAllPass( self, ip_address): |
| main.log.info( "Start the ping test on host:" + str( ip_address ) ) |
| |
| self.name = self.options[ 'name' ] |
| self.handle = super( QuaggaCliDriver, self ).connect( |
| user_name=self.user_name, ip_address=ip_address, |
| port=self.port, pwd=self.pwd ) |
| main.log.info( "connect parameters:" + str( self.user_name ) + ";" |
| + str( self.ip_address ) + ";" + str( self.port ) |
| + ";" + str( self.pwd ) ) |
| |
| testPass = main.TRUE |
| |
| if self.handle: |
| # self.handle.expect( "" ) |
| # self.handle.expect( "\$" ) |
| main.log.info( "I in host " + str( ip_address ) ) |
| |
| # Ping to 3.0.x.1 is just for sanity check. It always succeeds. |
| for m in range( 3, 6 ): |
| for n in range( 1, 11 ): |
| hostIp = str( m ) + ".0." + str( n ) + ".1" |
| main.log.info( "Ping to " + hostIp ) |
| try: |
| self.handle.sendline("ping -c 1 " + hostIp) |
| self.handle.expect( "64 bytes from", timeout=1 ) |
| except: |
| main.log.warn("Ping error") |
| testPass = main.FALSE |
| |
| return testPass |
| else: |
| main.log.info( "NO HANDLE" ) |
| return main.FALSE |
| |
| def pingTestAndCheckAllFail( self, ip_address): |
| main.log.info( "Start the ping test on host:" + str( ip_address ) ) |
| |
| self.name = self.options[ 'name' ] |
| self.handle = super( QuaggaCliDriver, self ).connect( |
| user_name=self.user_name, ip_address=ip_address, |
| port=self.port, pwd=self.pwd ) |
| main.log.info( "connect parameters:" + str( self.user_name ) + ";" |
| + str( self.ip_address ) + ";" + str( self.port ) |
| + ";" + str( self.pwd ) ) |
| |
| testPass = main.TRUE |
| |
| if self.handle: |
| # self.handle.expect( "" ) |
| # self.handle.expect( "\$" ) |
| main.log.info( "I in host " + str( ip_address ) ) |
| |
| for m in range( 4, 6 ): |
| for n in range( 1, 11 ): |
| hostIp = str( m ) + ".0." + str( n ) + ".1" |
| main.log.info( "Ping to " + hostIp ) |
| try: |
| self.handle.sendline("ping -c 1 " + hostIp) |
| self.handle.expect( "64 bytes from", timeout=1 ) |
| testPass = main.FALSE |
| except: |
| main.log.warn("Ping error") |
| |
| return testPass |
| else: |
| main.log.info( "NO HANDLE" ) |
| return main.FALSE |
| |
| |
| |
| |
| # Please use the generateRoutes plus addRoutes instead of this one! |
| def addRoute( self, net, numRoutes, routeRate ): |
| try: |
| self.handle.sendline( "" ) |
| self.handle.expect( "config-router" ) |
| except: |
| main.log.warn( "Probably not in config-router mode!" ) |
| self.disconnect() |
| main.log.info( "Adding Routes" ) |
| j = 0 |
| k = 0 |
| while numRoutes > 255: |
| numRoutes = numRoutes - 255 |
| j = j + 1 |
| k = numRoutes % 254 |
| routesAdded = 0 |
| if numRoutes > 255: |
| numRoutes = 255 |
| for m in range( 1, j + 1 ): |
| for n in range( 1, numRoutes + 1 ): |
| network = str( net ) + "." + str( m ) + "." + str( n ) \ |
| + ".0/24" |
| routeCmd = "network " + network |
| try: |
| self.handle.sendline( routeCmd ) |
| self.handle.expect( "bgpd" ) |
| except: |
| main.log.warn( "failed to add route" ) |
| self.disconnect() |
| waitTimer = 1.00 / routeRate |
| time.sleep( waitTimer ) |
| routesAdded = routesAdded + 1 |
| for d in range( j + 1, j + 2 ): |
| for e in range( 1, k + 1 ): |
| network = str( net ) + "." + str( d ) + "." + str( e ) \ |
| + ".0/24" |
| routeCmd = "network " + network |
| try: |
| self.handle.sendline( routeCmd ) |
| self.handle.expect( "bgpd" ) |
| except: |
| main.log.warn( "failed to add route" ) |
| self.disconnect |
| waitTimer = 1.00 / routeRate |
| time.sleep( waitTimer ) |
| routesAdded = routesAdded + 1 |
| if routesAdded == numRoutes: |
| return main.TRUE |
| return main.FALSE |
| |
| # Please use deleteRoutes method instead of this one! |
| def delRoute( self, net, numRoutes, routeRate ): |
| try: |
| self.handle.sendline( "" ) |
| self.handle.expect( "config-router" ) |
| except: |
| main.log.warn( "Probably not in config-router mode!" ) |
| self.disconnect() |
| main.log.info( "Deleting Routes" ) |
| j = 0 |
| k = 0 |
| while numRoutes > 255: |
| numRoutes = numRoutes - 255 |
| j = j + 1 |
| k = numRoutes % 254 |
| routesDeleted = 0 |
| if numRoutes > 255: |
| numRoutes = 255 |
| for m in range( 1, j + 1 ): |
| for n in range( 1, numRoutes + 1 ): |
| network = str( net ) + "." + str( m ) + "." + str( n ) \ |
| + ".0/24" |
| routeCmd = "no network " + network |
| try: |
| self.handle.sendline( routeCmd ) |
| self.handle.expect( "bgpd" ) |
| except: |
| main.log.warn( "Failed to delete route" ) |
| self.disconnect() |
| waitTimer = 1.00 / routeRate |
| time.sleep( waitTimer ) |
| routesDeleted = routesDeleted + 1 |
| for d in range( j + 1, j + 2 ): |
| for e in range( 1, k + 1 ): |
| network = str( net ) + "." + str( d ) + "." + str( e ) \ |
| + ".0/24" |
| routeCmd = "no network " + network |
| try: |
| self.handle.sendline( routeCmd ) |
| self.handle.expect( "bgpd" ) |
| except: |
| main.log.warn( "Failed to delete route" ) |
| self.disconnect() |
| waitTimer = 1.00 / routeRate |
| time.sleep( waitTimer ) |
| routesDeleted = routesDeleted + 1 |
| if routesDeleted == numRoutes: |
| return main.TRUE |
| return main.FALSE |
| |
| def checkRoutes( self, brand, ip, user, pw ): |
| def pronto( ip, user, passwd ): |
| print "Connecting to Pronto switch" |
| child = pexpect.spawn( "telnet " + ip ) |
| i = child.expect( [ "login:", "CLI#", pexpect.TIMEOUT ] ) |
| if i == 0: |
| print "user_name and password required. Passing login info." |
| child.sendline( user ) |
| child.expect( "Password:" ) |
| child.sendline( passwd ) |
| child.expect( "CLI#" ) |
| print "Logged in, getting flowtable." |
| child.sendline( "flowtable brief" ) |
| for t in range( 9 ): |
| t2 = 9 - t |
| print "\r" + str( t2 ) |
| sys.stdout.write( "\033[F" ) |
| time.sleep( 1 ) |
| print "Scanning flowtable" |
| child.expect( "Flow table show" ) |
| count = 0 |
| while True: |
| i = child.expect( [ '17\d\.\d{1,3}\.\d{1,3}\.\d{1,3}', |
| 'CLI#', pexpect.TIMEOUT ] ) |
| if i == 0: |
| count = count + 1 |
| elif i == 1: |
| print "Pronto flows: " + str( count ) + "\nDone\n" |
| break |
| else: |
| break |
| |
| def cisco( ip, user, passwd ): |
| print "Establishing Cisco switch connection" |
| child = pexpect.spawn( "ssh " + user + "@" + ip ) |
| i = child.expect( [ "Password:", "CLI#", pexpect.TIMEOUT ] ) |
| if i == 0: |
| print "Password required. Passing now." |
| child.sendline( passwd ) |
| child.expect( "#" ) |
| print "Logged in. Retrieving flow table then counting flows." |
| child.sendline( "show openflow switch all flows all" ) |
| child.expect( "Logical Openflow Switch" ) |
| print "Flow table retrieved. Counting flows" |
| count = 0 |
| while True: |
| i = child.expect( [ "nw_src=17", "#", pexpect.TIMEOUT ] ) |
| if i == 0: |
| count = count + 1 |
| elif i == 1: |
| print "Cisco flows: " + str( count ) + "\nDone\n" |
| break |
| else: |
| break |
| if brand == "pronto" or brand == "PRONTO": |
| pronto( ip, user, passwd ) |
| # elif brand == "cisco" or brand == "CISCO": |
| # cisco( ip,user,passwd ) |
| |
| def disable_bgp_peer( self, peer, peer_as ): |
| main.log.info( "I am in disconnect_peer_session method!" ) |
| |
| try: |
| self.handle.sendline( "" ) |
| # self.handle.expect( "config-router" ) |
| self.handle.expect( "config-router", timeout=5 ) |
| except: |
| main.log.warn( "Probably not in config-router mode!" ) |
| self.disconnect() |
| main.log.info( "Start to disable peer" ) |
| |
| cmd = "no neighbor " + peer + " remote-as " + peer_as |
| try: |
| self.handle.sendline( cmd ) |
| self.handle.expect( "bgpd", timeout=5 ) |
| except: |
| main.log.warn( "Failed to disable peer" ) |
| self.disconnect() |
| |
| def disconnect( self ): |
| """ |
| Called when Test is complete to disconnect the Quagga handle. |
| """ |
| response = '' |
| try: |
| self.handle.close() |
| except: |
| main.log.error( "Connection failed to the host" ) |
| response = main.FALSE |
| return response |
| |