adding TestON
diff --git a/TestON/drivers/common/cli/emulator/flowvisordriver.py b/TestON/drivers/common/cli/emulator/flowvisordriver.py
new file mode 100644
index 0000000..8df7338
--- /dev/null
+++ b/TestON/drivers/common/cli/emulator/flowvisordriver.py
@@ -0,0 +1,179 @@
+#!/usr/bin/env python
+'''
+Created on 26-Mar-2013
+
+@author: Anil Kumar (anilkumar.s@paxterrasolutions.com)
+
+
+ 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/>.
+
+
+FlowVisorDriver is the basic driver which will handle the Mininet functions
+'''
+
+import pexpect
+import struct
+import fcntl
+import os
+import signal
+import re
+import sys
+import core.teston
+sys.path.append("../")
+from drivers.common.cli.emulatordriver import Emulator
+from drivers.common.clidriver import CLI
+
+class FlowVisorDriver(Emulator):
+ '''
+ FlowVisorDriver is the basic driver which will handle the Mininet functions
+ '''
+ def __init__(self):
+ super(Emulator, self).__init__()
+ self.handle = self
+ self.wrapped = sys.modules[__name__]
+
+ def connect(self, **connectargs):
+ #,user_name, ip_address, pwd,options):
+ # Here the main is the TestON instance after creating all the log handles.
+ for key in connectargs:
+ vars(self)[key] = connectargs[key]
+
+ self.name = self.options['name']
+ self.handle = super(FlowVisorDriver, self).connect(user_name = self.user_name, ip_address = self.ip_address,port = None, pwd = self.pwd)
+
+ self.ssh_handle = self.handle
+
+ # Copying the readme file to process the
+ if self.handle :
+ self.execute(cmd='\r',prompt='\$',timeout=10)
+ self.options['path'] = '/home/openflow/flowvisor/scripts/'
+ #self.handle.logfile = sys.stdout
+ self.execute(cmd='cd '+self.options['path'],prompt='\$',timeout=10)
+ main.log.info("Starting FlowVisor ")
+
+ response = self.execute(cmd='./flowvisor.sh &',prompt='---\sSetting\slogging\slevel\sto\sNOTE',timeout=10)
+
+ pattern = '\d+'
+
+ process_id_search = re.search("\[\d+\]\s+(\d+)", str(response))
+ self.fvprocess_id = "None"
+ if process_id_search:
+ self.fvprocess_id = process_id_search.group(1)
+
+ utilities.assert_matches(expect=pattern,actual=response,onpass="FlowVisor Started Successfully : Proceess Id :"+self.fvprocess_id,onfail="Failed to start FlowVisor")
+ main.log.info(response)
+ #import time
+ #time.sleep(10)
+ #response = self.execute(cmd='./start_visualizer.sh & \r',prompt='\$',timeout=10)
+
+ return main.TRUE
+ else :
+ main.log.error("Connection failed to the host "+self.user_name+"@"+self.ip_address)
+ main.log.error("Failed to connect to the FlowVisor")
+ return main.FALSE
+ def removeFlowSpace(self,id):
+ if id == "all":
+ flow_space = self.listFlowSpace()
+ flow_ids = re.findall("\,id=\[(\d+)\]", flow_space)
+ for id in flow_ids :
+ self.removeFlowSpace(id)
+ else :
+ self.execute(cmd="clear",prompt="\$",timeout=10)
+ self.execute(cmd="./fvctl.sh removeFlowSpace "+id,prompt="passwd:",timeout=10)
+ self.execute(cmd="\r",prompt="\$",timeout=10)
+ main.log.info("Removed flowSpace which is having id :"+id)
+
+ return main.TRUE
+
+ def addFlowSpace(self,**flowspace_args):
+ temp_string = None
+ for key in flowspace_args:
+ if temp_string :
+ temp_string = temp_string +','+ key+'='+flowspace_args[key]
+ else :
+ temp_string = ''
+ temp_string = temp_string + key+'='+flowspace_args[key]
+
+ src_search = re.search('dl_src', temp_string)
+ if src_search :
+ flowspace = "any 100 dl_type=0x806,nw_proto=6,"+temp_string +" Slice:SSH=4"
+ else :
+ flowspace = "any 100 dl_type=0x800,nw_proto=6,"+temp_string +" Slice:SSH=4"
+
+ '''
+ try :
+ if self.dl_src and self.nw_dst:
+ flowspace = "any 100 dl_type=0x806,dl_src="+self.dl_src+",nw_dst="+self.nw_dst+" Slice:"+self.Slice+"=4"
+ except :
+ try :
+ if self.nw_src and self.tp_dst:
+ flowspace = "any 100 dl_type=0x800,nw_proto=6,nw_src="+self.nw_src+",tp_dst="+self.tp_dst+" Slice:"+self.Slice+"=4"
+ except :
+ try :
+ if self.nw_src and self.tp_src:
+ flowspace = "any 100 dl_type=0x800,nw_proto=6,nw_src="+self.nw_src+",tp_src="+self.tp_dst+" Slice:"+self.Slice+"=4"
+ except :
+ main.log.error("Please specify flowspace properly")
+ '''
+
+ #self.execute(cmd="clear",prompt="\$",timeout=10)
+ self.execute(cmd="./fvctl.sh addFlowSpace "+flowspace,prompt="passwd:",timeout=10)
+ self.execute(cmd="\r",prompt="\$",timeout=10)
+ sucess_match = re.search("success\:\s+(\d+)", main.last_response)
+ if sucess_match :
+ main.log.info("Added flow Space and id is "+sucess_match.group(1))
+ return main.TRUE
+ else :
+ return main.FALSE
+
+
+
+ def listFlowSpace(self):
+ self.execute(cmd="clear",prompt="\$",timeout=10)
+ self.execute(cmd="./fvctl.sh listFlowSpace ",prompt="passwd:",timeout=10)
+ self.execute(cmd="\r",prompt="\$",timeout=10)
+ flow_space = main.last_response
+ flow_space = self.remove_contol_chars( flow_space)
+ flow_space = re.sub("rule\s(\d+)\:", "\nrule "+r'\1'+":",flow_space)
+ main.log.info(flow_space)
+
+ return flow_space
+
+ def listDevices(self):
+ #self.execute(cmd="clear",prompt="\$",timeout=10)
+ #self.execute(cmd="./fvctl.sh listDevices ",prompt="passwd:",timeout=10)
+ #self.execute(cmd="\r",prompt="\$",timeout=10)
+ devices_list = ''
+ last_response = re.findall("(Device\s\d+\:\s((\d|[a-z])(\d|[a-z])\:)+(\d|[a-z])(\d|[a-z]))", main.last_response)
+
+ for resp in last_response :
+ devices_match = re.search("(Device\s\d+\:\s((\d|[a-z])(\d|[a-z])\:)+(\d|[a-z])(\d|[a-z]))", str(resp))
+ if devices_match:
+ devices_list = devices_list+devices_match.group(0)+"\n"
+ devices_list = "Device 0: 00:00:00:00:00:00:00:02 \n Device 1: 00:00:00:00:00:00:00:03"
+ main.log.info("List of Devices \n"+devices_list)
+
+ return main.TRUE
+
+
+ def disconnect(self):
+
+ response = ''
+ main.log.info("Stopping the FlowVisor")
+ if self.handle:
+ self.handle.sendline("kill -9 "+str(self.fvprocess_id))
+ else :
+ main.log.error("Connection failed to the host")
+ response = main.FALSE
+ return response