| #!/usr/bin/env python |
| |
| import time |
| import pexpect |
| import struct, fcntl, os, sys, 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 = 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("") |
| self.handle.expect("\$") |
| self.handle.sendline("telnet localhost 2605") |
| self.handle.expect("Password:", timeout = 5) |
| self.handle.sendline("hello") |
| self.handle.expect("bgpd",timeout = 5) |
| self.handle.sendline("enable") |
| self.handle.expect("bgpd#", timeout = 5) |
| 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.sendline("hello") |
| self.handle.expect("bgpd", timeout=5) |
| self.handle.sendline("enable") |
| self.handle.expect("bgpd#", timeout=5) |
| main.log.info("I am inside BGP peer Quagga!") |
| |
| return self.handle |
| else: |
| main.log.info("NO HANDLE") |
| return main.FALSE |
| |
| def enter_config(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 generate_routes(self, net, numRoutes): |
| main.log.info("I am in generate_routes method!") |
| |
| # a IP prefix will be composed by "net" + "." + m + "." + n + "." + x |
| # the length of each IP prefix is 24 |
| routes = [] |
| routes_gen = 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) |
| routes_gen = routes_gen + 1 |
| |
| for j in range(0, n): |
| network = str(net) + "." + str(m) + "." + str(j) + ".0/24" |
| routes.append(network) |
| routes_gen = routes_gen + 1 |
| |
| if routes_gen == numRoutes: |
| main.log.info("Successfully generated " + str(numRoutes) |
| + " routes!") |
| return routes |
| return main.FALSE |
| |
| def add_routes(self, routes, routeRate): |
| main.log.info("I am in add_routes method!") |
| |
| routes_added = 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 routes_added == len(routes): |
| main.log.info("Finished adding routes") |
| return main.TRUE |
| return main.FALSE |
| |
| # please use the generate_routes plus add_routes instead of this one |
| def add_route(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.report("Adding Routes") |
| j=0 |
| k=0 |
| while numRoutes > 255: |
| numRoutes = numRoutes - 255 |
| j = j + 1 |
| k = numRoutes % 254 |
| routes_added = 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) |
| routes_added = routes_added + 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) |
| routes_added = routes_added + 1 |
| if routes_added == numRoutes: |
| return main.TRUE |
| return main.FALSE |
| def del_route(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.report("Deleting Routes") |
| j=0 |
| k=0 |
| while numRoutes > 255: |
| numRoutes = numRoutes - 255 |
| j = j + 1 |
| k = numRoutes % 254 |
| routes_deleted = 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) |
| routes_deleted = routes_deleted + 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) |
| routes_deleted = routes_deleted + 1 |
| if routes_deleted == numRoutes: |
| return main.TRUE |
| return main.FALSE |
| def check_routes(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 "Username 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 1: |
| 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 1: |
| 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 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 |