summaryrefslogtreecommitdiffstats
path: root/scripts/automation/trex_control_plane
diff options
context:
space:
mode:
authorYaroslav Brustinov <ybrustin@cisco.com>2016-05-13 20:11:51 +0300
committerYaroslav Brustinov <ybrustin@cisco.com>2016-05-13 20:11:51 +0300
commit89b608ae766705950efc5f4914b01b9a32b6a0e7 (patch)
treee5b044865eb02fedce4d4e3f9b735cadeaf5f997 /scripts/automation/trex_control_plane
parentfa7a837fb19af1c1fe4365de3f71467b79c2c865 (diff)
add master daemon
Diffstat (limited to 'scripts/automation/trex_control_plane')
-rwxr-xr-xscripts/automation/trex_control_plane/server/CCustomLogger.py4
-rwxr-xr-xscripts/automation/trex_control_plane/server/extended_daemon_runner.py146
-rwxr-xr-xscripts/automation/trex_control_plane/server/trex_daemon_server25
-rwxr-xr-xscripts/automation/trex_control_plane/server/trex_daemon_server.py79
-rwxr-xr-xscripts/automation/trex_control_plane/server/trex_server.py137
-rwxr-xr-xscripts/automation/trex_control_plane/stf/trex_stf_lib/trex_client.py224
6 files changed, 312 insertions, 303 deletions
diff --git a/scripts/automation/trex_control_plane/server/CCustomLogger.py b/scripts/automation/trex_control_plane/server/CCustomLogger.py
index ecf7d519..a8823cea 100755
--- a/scripts/automation/trex_control_plane/server/CCustomLogger.py
+++ b/scripts/automation/trex_control_plane/server/CCustomLogger.py
@@ -31,6 +31,10 @@ def setup_custom_logger(name, log_path = None):
def setup_daemon_logger (name, log_path = None):
# first make sure path availabe
+ try:
+ os.unlink(log_path)
+ except:
+ pass
logging.basicConfig(level = logging.INFO,
format = '%(asctime)s %(name)-10s %(module)-20s %(levelname)-8s %(message)s',
datefmt = '%m-%d %H:%M',
diff --git a/scripts/automation/trex_control_plane/server/extended_daemon_runner.py b/scripts/automation/trex_control_plane/server/extended_daemon_runner.py
deleted file mode 100755
index 7bc25aac..00000000
--- a/scripts/automation/trex_control_plane/server/extended_daemon_runner.py
+++ /dev/null
@@ -1,146 +0,0 @@
-#!/usr/bin/python
-
-import outer_packages
-import lockfile
-from daemon import runner,daemon
-from daemon.runner import *
-import os, sys
-from argparse import ArgumentParser
-from trex_server import trex_parser
-try:
- from termstyle import termstyle
-except ImportError:
- import termstyle
-
-
-def daemonize_parser(parser_obj, action_funcs, help_menu):
- """Update the regular process parser to deal with daemon process options"""
- parser_obj.description += " (as a daemon process)"
- parser_obj.usage = None
- parser_obj.add_argument("action", choices=action_funcs,
- action="store", help=help_menu)
-
-
-class ExtendedDaemonRunner(runner.DaemonRunner):
- """ Controller for a callable running in a separate background process.
-
- The first command-line argument is the action to take:
-
- * 'start': Become a daemon and call `app.run()`.
- * 'stop': Exit the daemon process specified in the PID file.
- * 'restart': Stop, then start.
-
- """
-
- help_menu = """Specify action command to be applied on server.
- (*) start : start the application in as a daemon process.
- (*) show : prompt an updated status of daemon process (running/ not running).
- (*) stop : exit the daemon process.
- (*) restart : stop, then start again the application as daemon process
- (*) start-live : start the application in live mode (no daemon process).
- """
-
- def __init__(self, app, parser_obj):
- """ Set up the parameters of a new runner.
- THIS METHOD INTENTIONALLY DO NOT INVOKE SUPER __init__() METHOD
-
- :param app: The application instance; see below.
- :return: ``None``.
-
- The `app` argument must have the following attributes:
-
- * `stdin_path`, `stdout_path`, `stderr_path`: Filesystem paths
- to open and replace the existing `sys.stdin`, `sys.stdout`,
- `sys.stderr`.
-
- * `pidfile_path`: Absolute filesystem path to a file that will
- be used as the PID file for the daemon. If ``None``, no PID
- file will be used.
-
- * `pidfile_timeout`: Used as the default acquisition timeout
- value supplied to the runner's PID lock file.
-
- * `run`: Callable that will be invoked when the daemon is
- started.
-
- """
- super(runner.DaemonRunner, self).__init__()
- # update action_funcs to support more operations
- self.update_action_funcs()
-
- daemonize_parser(parser_obj, self.action_funcs, ExtendedDaemonRunner.help_menu)
- args = parser_obj.parse_args()
- self.action = unicode(args.action)
-
- self.app = app
- self.daemon_context = daemon.DaemonContext()
- self.daemon_context.stdin = open(app.stdin_path, 'rt')
- try:
- self.daemon_context.stdout = open(app.stdout_path, 'w+t')
- except IOError as err:
- # catch 'tty' error when launching server from remote location
- app.stdout_path = "/dev/null"
- self.daemon_context.stdout = open(app.stdout_path, 'w+t')
- self.daemon_context.stderr = open(app.stderr_path,
- 'a+t', buffering=0)
-
- self.pidfile = None
- if app.pidfile_path is not None:
- self.pidfile = make_pidlockfile(app.pidfile_path, app.pidfile_timeout)
- self.daemon_context.pidfile = self.pidfile
-
- # mask out all arguments that aren't relevant to main app script
-
- def update_action_funcs(self):
- self.action_funcs.update({u'start-live': self._start_live, u'show': self._show}) # add key (=action), value (=desired func)
-
- @staticmethod
- def _start_live(self):
- self.app.run()
-
- @staticmethod
- def _show(self):
- if self.pidfile.is_locked():
- print termstyle.red("TRex server daemon is running")
- else:
- print termstyle.red("TRex server daemon is NOT running")
-
- def do_action(self):
- self.__prevent_duplicate_runs()
- self.__prompt_init_msg()
- try:
- super(ExtendedDaemonRunner, self).do_action()
- if self.action == 'stop':
- self.__verify_termination()
- except runner.DaemonRunnerStopFailureError:
- if self.action == 'restart':
- # error means server wasn't running in the first place- so start it!
- self.action = 'start'
- self.do_action()
-
-
- def __prevent_duplicate_runs(self):
- if self.action == 'start' and self.pidfile.is_locked():
- print termstyle.green("Server daemon is already running")
- exit(1)
- elif self.action == 'stop' and not self.pidfile.is_locked():
- print termstyle.green("Server daemon is not running")
- exit(1)
-
- def __prompt_init_msg(self):
- if self.action == 'start':
- print termstyle.green("Starting daemon server...")
- elif self.action == 'stop':
- print termstyle.green("Stopping daemon server...")
-
- def __verify_termination(self):
- pass
-# import time
-# while self.pidfile.is_locked():
-# time.sleep(2)
-# self._stop()
-#
-
-
-if __name__ == "__main__":
- pass
diff --git a/scripts/automation/trex_control_plane/server/trex_daemon_server b/scripts/automation/trex_control_plane/server/trex_daemon_server
deleted file mode 100755
index 3494e303..00000000
--- a/scripts/automation/trex_control_plane/server/trex_daemon_server
+++ /dev/null
@@ -1,25 +0,0 @@
-#!/usr/bin/python
-
-import os
-import sys
-
-core = 0
-
-if '--core' in sys.argv:
- try:
- idx = sys.argv.index('--core')
- core = int(sys.argv[idx + 1])
- if core > 31 or core < 0:
- print "Error: please provide core argument between 0 to 31"
- exit(-1)
- del sys.argv[idx:idx+2]
- except IndexError:
- print "Error: please make sure core option provided with argument"
- exit(-1)
- except ValueError:
- print "Error: please make sure core option provided with integer argument"
- exit(-1)
-
-str_argv = ' '.join(sys.argv[1:])
-cmd = "taskset -c {core} python automation/trex_control_plane/server/trex_daemon_server.py {argv}".format(core = core, argv = str_argv)
-os.system(cmd)
diff --git a/scripts/automation/trex_control_plane/server/trex_daemon_server.py b/scripts/automation/trex_control_plane/server/trex_daemon_server.py
deleted file mode 100755
index 9784d42a..00000000
--- a/scripts/automation/trex_control_plane/server/trex_daemon_server.py
+++ /dev/null
@@ -1,79 +0,0 @@
-#!/usr/bin/python
-
-import outer_packages
-import daemon
-from trex_server import do_main_program, trex_parser
-import CCustomLogger
-
-import logging
-import time
-import sys
-import os, errno
-import grp
-import signal
-from daemon import runner
-from extended_daemon_runner import ExtendedDaemonRunner
-import lockfile
-import errno
-
-class TRexServerApp(object):
- def __init__(self):
- TRexServerApp.create_working_dirs()
- self.stdin_path = '/dev/null'
- self.stdout_path = '/dev/tty' # All standard prints will come up from this source.
- self.stderr_path = "/var/log/trex/trex_daemon_server.log" # All log messages will come up from this source
- self.pidfile_path = '/var/run/trex/trex_daemon_server.pid'
- self.pidfile_timeout = 5 # timeout in seconds
-
- def run(self):
- do_main_program()
-
-
- @staticmethod
- def create_working_dirs():
- if not os.path.exists('/var/log/trex'):
- os.mkdir('/var/log/trex')
- if not os.path.exists('/var/run/trex'):
- os.mkdir('/var/run/trex')
-
-
-
-def main ():
-
- trex_app = TRexServerApp()
-
- # setup the logger
- default_log_path = '/var/log/trex/trex_daemon_server.log'
-
- try:
- CCustomLogger.setup_daemon_logger('TRexServer', default_log_path)
- logger = logging.getLogger('TRexServer')
- logger.setLevel(logging.INFO)
- formatter = logging.Formatter("%(asctime)s %(name)-10s %(module)-20s %(levelname)-8s %(message)s")
- handler = logging.FileHandler("/var/log/trex/trex_daemon_server.log")
- logger.addHandler(handler)
- except EnvironmentError, e:
- if e.errno == errno.EACCES: # catching permission denied error
- print "Launching user must have sudo privileges in order to run TRex daemon.\nTerminating daemon process."
- exit(-1)
-
- daemon_runner = ExtendedDaemonRunner(trex_app, trex_parser)
-
- #This ensures that the logger file handle does not get closed during daemonization
- daemon_runner.daemon_context.files_preserve=[handler.stream]
-
- try:
- if not set(['start', 'stop']).isdisjoint(set(sys.argv)):
- print "Logs are saved at: {log_path}".format( log_path = default_log_path )
- daemon_runner.do_action()
-
- except lockfile.LockTimeout as inst:
- logger.error(inst)
- print inst
- print """
- Please try again once the timeout has been reached.
- If this error continues, consider killing the process manually and restart the daemon."""
-
-
-if __name__ == "__main__":
- main()
diff --git a/scripts/automation/trex_control_plane/server/trex_server.py b/scripts/automation/trex_control_plane/server/trex_server.py
index e32fc9d1..3dcb3e97 100755
--- a/scripts/automation/trex_control_plane/server/trex_server.py
+++ b/scripts/automation/trex_control_plane/server/trex_server.py
@@ -27,6 +27,8 @@ from zmq_monitor_thread import ZmqMonitorSession
from argparse import ArgumentParser, RawTextHelpFormatter
from json import JSONEncoder
import re
+import shlex
+import tempfile
# setup the logger
@@ -52,6 +54,8 @@ class CTRexServer(object):
trex_zmq_port : int
the port number on which trex's zmq module will interact with daemon server
default value: 4500
+ nice: int
+ priority of the TRex process
Instantiate a TRex client object, and connecting it to listening daemon-server
"""
@@ -76,7 +80,6 @@ class CTRexServer(object):
raise Exception(err)
def add(self, x, y):
- print "server function add ",x,y
logger.info("Processing add function. Parameters are: {0}, {1} ".format( x, y ))
return x + y
# return Fault(-10, "")
@@ -123,38 +126,49 @@ class CTRexServer(object):
raise
# set further functionality and peripherals to server instance
+ self.server.register_function(self.add)
+ self.server.register_function(self.cancel_reservation)
+ self.server.register_function(self.connectivity_check)
+ self.server.register_function(self.force_trex_kill)
+ self.server.register_function(self.get_file)
+ self.server.register_function(self.get_files_list)
+ self.server.register_function(self.get_files_path)
+ self.server.register_function(self.get_running_info)
+ self.server.register_function(self.get_running_status)
+ self.server.register_function(self.get_trex_cmds)
+ self.server.register_function(self.get_trex_daemon_log)
+ self.server.register_function(self.get_trex_log)
+ self.server.register_function(self.get_trex_version)
+ self.server.register_function(self.is_reserved)
+ self.server.register_function(self.is_running)
+ self.server.register_function(self.kill_all_trexes)
+ self.server.register_function(self.push_file)
+ self.server.register_function(self.reserve_trex)
+ self.server.register_function(self.start_trex)
+ self.server.register_function(self.stop_trex)
+ self.server.register_function(self.wait_until_kickoff_finish)
+ signal.signal(signal.SIGTSTP, self.stop_handler)
+ signal.signal(signal.SIGTERM, self.stop_handler)
try:
- self.server.register_function(self.add)
- self.server.register_function(self.cancel_reservation)
- self.server.register_function(self.connectivity_check)
- self.server.register_function(self.force_trex_kill)
- self.server.register_function(self.get_file)
- self.server.register_function(self.get_files_list)
- self.server.register_function(self.get_files_path)
- self.server.register_function(self.get_running_info)
- self.server.register_function(self.get_running_status)
- self.server.register_function(self.get_trex_daemon_log)
- self.server.register_function(self.get_trex_log)
- self.server.register_function(self.get_trex_version)
- self.server.register_function(self.is_reserved)
- self.server.register_function(self.is_running)
- self.server.register_function(self.push_file)
- self.server.register_function(self.reserve_trex)
- self.server.register_function(self.start_trex)
- self.server.register_function(self.stop_trex)
- self.server.register_function(self.wait_until_kickoff_finish)
- signal.signal(signal.SIGTSTP, self.stop_handler)
- signal.signal(signal.SIGTERM, self.stop_handler)
self.zmq_monitor.start()
self.server.serve_forever()
except KeyboardInterrupt:
logger.info("Daemon shutdown request detected." )
- except Exception as e:
- logger.error(e)
finally:
self.zmq_monitor.join() # close ZMQ monitor thread resources
self.server.shutdown()
- pass
+ #self.server.server_close()
+
+ def _run_command(self, command, timeout = 15, cwd = None):
+ if timeout:
+ command = 'timeout %s %s' % (timeout, command)
+ # pipes might stuck, even with timeout
+ with tempfile.TemporaryFile() as stdout_file, tempfile.TemporaryFile() as stderr_file:
+ proc = subprocess.Popen(shlex.split(command), stdout=stdout_file, stderr=stderr_file, cwd = cwd)
+ proc.wait()
+ stdout_file.seek(0)
+ stderr_file.seek(0)
+ return (proc.returncode, stdout_file.read().decode(errors = 'replace'), stderr_file.read().decode(errors = 'replace'))
# get files from Trex server and return their content (mainly for logs)
@staticmethod
@@ -215,8 +229,7 @@ class CTRexServer(object):
try:
logger.info("Processing get_trex_version() command.")
if not self.trex_version:
- help_print = subprocess.Popen(['./t-rex-64', '--help'], cwd = self.TREX_PATH, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
- (stdout, stderr) = help_print.communicate()
+ ret_code, stdout, stderr = self._run_command('./t-rex-64 --help', cwd = self.TREX_PATH, timeout = 0)
search_result = re.search('\n\s*(Version\s*:.+)', stdout, re.DOTALL)
if not search_result:
raise Exception('Could not determine version from ./t-rex-64 --help')
@@ -226,7 +239,7 @@ class CTRexServer(object):
else:
return binascii.a2b_base64(self.trex_version)
except Exception as e:
- err_str = "Can't get trex version, error: {0}".format(e)
+ err_str = "Can't get trex version, error: %s" % e
logger.error(err_str)
return Fault(-33, err_str)
@@ -301,7 +314,7 @@ class CTRexServer(object):
return False
- def start_trex(self, trex_cmd_options, user, block_to_success = True, timeout = 40):
+ def start_trex(self, trex_cmd_options, user, block_to_success = True, timeout = 40, stateless = False):
with self.start_lock:
logger.info("Processing start_trex() command.")
if self.is_reserved():
@@ -313,8 +326,8 @@ class CTRexServer(object):
logger.info("TRex is already taken, cannot create another run until done.")
return Fault(-13, '') # raise at client TRexInUseError
- try:
- server_cmd_data = self.generate_run_cmd(**trex_cmd_options)
+ try:
+ server_cmd_data = self.generate_run_cmd(stateless = stateless, **trex_cmd_options)
self.zmq_monitor.first_dump = True
self.trex.start_trex(self.TREX_PATH, server_cmd_data)
logger.info("TRex session has been successfully initiated.")
@@ -342,7 +355,7 @@ class CTRexServer(object):
except TypeError as e:
logger.error("TRex command generation failed, probably because either -f (traffic generation .yaml file) and -c (num of cores) was not specified correctly.\nReceived params: {params}".format( params = trex_cmd_options) )
- raise TypeError('TRex -f (traffic generation .yaml file) and -c (num of cores) must be specified.')
+ raise TypeError('TRex -f (traffic generation .yaml file) and -c (num of cores) must be specified. %s' % e)
def stop_trex(self, seq):
@@ -362,6 +375,41 @@ class CTRexServer(object):
logger.info("Processing force_trex_kill() command. --> Killing TRex session indiscriminately.")
return self.trex.stop_trex()
+ # returns list of tuples (pid, command line) of running TRex(es)
+ def get_trex_cmds(self):
+ logger.info('Processing get_trex_cmds() command.')
+ ret_code, stdout, stderr = self._run_command('ps -u root --format pid,comm,cmd')
+ if ret_code:
+ raise Exception('Failed to determine running processes, stderr: %s' % stderr)
+ trex_cmds_list = []
+ for line in stdout.splitlines():
+ pid, proc_name, full_cmd = line.strip().split(' ', 2)
+ pid = pid.strip()
+ full_cmd = full_cmd.strip()
+ if proc_name.find('t-rex-64') >= 0:
+ trex_cmds_list.append((pid, full_cmd))
+ return trex_cmds_list
+
+
+ def kill_all_trexes(self):
+ logger.info('Processing kill_all_trexes() command.')
+ trex_cmds_list = self.get_trex_cmds()
+ if not trex_cmds_list:
+ return False
+ for pid, cmd in trex_cmds_list:
+ logger.info('Killing process %s %s' % (pid, cmd))
+ self._run_command('kill %s' % pid)
+ ret_code_ps, _, _ = self._run_command('ps -p %s' % pid)
+ if not ret_code_ps:
+ logger.info('Killing with -9.')
+ self._run_command('kill -9 %s' % pid)
+ ret_code_ps, _, _ = self._run_command('ps -p %s' % pid)
+ if not ret_code_ps:
+ logger.info('Could not kill process.')
+ raise Exception('Could not kill process %s %s' % (pid, cmd))
+ return True
+
+
def wait_until_kickoff_finish (self, timeout = 40):
# block until TRex exits Starting state
logger.info("Processing wait_until_kickoff_finish() command.")
@@ -377,14 +425,19 @@ class CTRexServer(object):
logger.info("Processing get_running_info() command.")
return self.trex.get_running_info()
- def generate_run_cmd (self, f, d, iom = 0, export_path="/tmp/trex.txt", **kwargs):
- """ generate_run_cmd(self, trex_cmd_options, export_path) -> str
+
+ def generate_run_cmd (self, iom = 0, export_path="/tmp/trex.txt", stateless = False, **kwargs):
+ """ generate_run_cmd(self, iom, export_path, kwargs) -> str
Generates a custom running command for the kick-off of the TRex traffic generator.
Returns a tuple of command (string) and export path (string) to be issued on the trex server
Parameters
----------
+ iom: int
+ 0 = don't print stats screen to log, 1 = print stats (can generate huge logs)
+ stateless: boolean
+ True = run as stateless, False = require -f and -d arguments
kwargs: dictionary
Dictionary of parameters for trex. For example: (c=1, nc=True, l_pkt_mode=3).
Notice that when sending command line parameters that has -, you need to replace it with _.
@@ -396,7 +449,8 @@ class CTRexServer(object):
if 'results_file_path' in kwargs:
export_path = kwargs['results_file_path']
del kwargs['results_file_path']
-
+ if stateless:
+ kwargs['i'] = True
# adding additional options to the command
trex_cmd_options = ''
@@ -408,18 +462,23 @@ class CTRexServer(object):
else:
trex_cmd_options += (dash + '{k} {val}'.format( k = tmp_key, val = value ))
- cmd = "{nice}{run_command} -f {gen_file} -d {duration} --iom {io} {cmd_options} --no-key > {export}".format( # -- iom 0 disables the periodic log to the screen (not needed)
+ if not stateless:
+ if 'f' not in kwargs:
+ raise Exception('Argument -f should be specified in stateful command')
+ if 'd' not in kwargs:
+ raise Exception('Argument -d should be specified in stateful command')
+
+ cmd = "{nice}{run_command} --iom {io} {cmd_options} --no-key > {export}".format( # -- iom 0 disables the periodic log to the screen (not needed)
nice = '' if self.trex_nice == 0 else 'nice -n %s ' % self.trex_nice,
run_command = self.TREX_START_CMD,
- gen_file = f,
- duration = d,
cmd_options = trex_cmd_options,
io = iom,
export = export_path )
logger.info("TREX FULL COMMAND: {command}".format(command = cmd) )
- return (cmd, export_path, long(d))
+ return (cmd, export_path, kwargs.get('d', 0))
+
def __check_trex_path_validity(self):
# check for executable existance
diff --git a/scripts/automation/trex_control_plane/stf/trex_stf_lib/trex_client.py b/scripts/automation/trex_control_plane/stf/trex_stf_lib/trex_client.py
index fd409b16..342c75de 100755
--- a/scripts/automation/trex_control_plane/stf/trex_stf_lib/trex_client.py
+++ b/scripts/automation/trex_control_plane/stf/trex_stf_lib/trex_client.py
@@ -39,7 +39,7 @@ class CTRexClient(object):
This class defines the client side of the RESTfull interaction with TRex
"""
- def __init__(self, trex_host, max_history_size = 100, filtered_latency_amount = 0.001, trex_daemon_port = 8090, trex_zmq_port = 4500, verbose = False):
+ def __init__(self, trex_host, max_history_size = 100, filtered_latency_amount = 0.001, trex_daemon_port = 8090, master_daemon_port = 8091, trex_zmq_port = 4500, verbose = False):
"""
Instantiate a TRex client object, and connecting it to listening daemon-server
@@ -60,6 +60,10 @@ class CTRexClient(object):
the port number on which the trex-daemon server can be reached
default value: **8090**
+ master_daemon_port : int
+ the port number on which the master-daemon server can be reached
+
+ default value: **8091**
trex_zmq_port : int
the port number on which trex's zmq module will interact with daemon server
@@ -78,19 +82,50 @@ class CTRexClient(object):
except: # give it another try
self.trex_host = socket.gethostbyname(trex_host)
self.trex_daemon_port = trex_daemon_port
+ self.master_daemon_port = master_daemon_port
self.trex_zmq_port = trex_zmq_port
self.seq = None
+ self._last_sample = time.time()
+ self.__default_user = get_current_user()
self.verbose = verbose
self.result_obj = CTRexResult(max_history_size, filtered_latency_amount)
self.decoder = JSONDecoder()
- self.trex_server_path = "http://{hostname}:{port}/".format( hostname = self.trex_host, port = trex_daemon_port )
- self.__verbose_print("Connecting to TRex @ {trex_path} ...".format( trex_path = self.trex_server_path ) )
self.history = jsonrpclib.history.History()
- self.server = jsonrpclib.Server(self.trex_server_path, history = self.history)
- self.check_server_connectivity()
- self.__verbose_print("Connection established successfully!")
- self._last_sample = time.time()
- self.__default_user = get_current_user()
+ self.master_daemon_path = "http://{hostname}:{port}/".format( hostname = self.trex_host, port = master_daemon_port )
+ self.trex_server_path = "http://{hostname}:{port}/".format( hostname = self.trex_host, port = trex_daemon_port )
+ self.connect_master()
+ self.connect_server()
+
+
+ def connect_master(self):
+ '''
+ Connects to Master daemon via JsonRPC.
+ This daemon controls TRex daemon server.
+ '''
+ try:
+ print('Connecting to Master daemon @ %s ...' % self.master_daemon_path)
+ self.master_daemon = jsonrpclib.Server(self.master_daemon_path, history = self.history)
+ print self.check_master_connectivity()
+ print('Connected to Master daemon.')
+ return True
+ except Exception as e:
+ print(e)
+ return False
+
+ def connect_server(self):
+ '''
+ Connects to TRex daemon server via JsonRPC.
+ This daemon controls TRex. (start/stop
+ '''
+ try:
+ print('Connecting to TRex daemon server @ %s ...' % self.trex_server_path)
+ self.server = jsonrpclib.Server(self.trex_server_path, history = self.history)
+ self.check_server_connectivity()
+ print('Connected TRex server daemon.')
+ return True
+ except Exception as e:
+ print(e)
+ return False
def add (self, x, y):
@@ -105,7 +140,7 @@ class CTRexClient(object):
def start_trex (self, f, d, block_to_success = True, timeout = 40, user = None, trex_development = False, **trex_cmd_options):
"""
- Request to start a TRex run on server.
+ Request to start a TRex run on server in stateful mode.
:parameters:
f : str
@@ -168,6 +203,53 @@ class CTRexClient(object):
else: # TRex is has been started by another user
raise TRexInUseError('TRex is already being used by another user or process. Try again once TRex is back in IDLE state.')
+
+ def start_stateless(self, block_to_success = True, timeout = 40, user = None, **trex_cmd_options):
+ """
+ Request to start a TRex run on server in stateless mode.
+
+ :parameters:
+ block_to_success : bool
+ determine if this method blocks until TRex changes state from 'Starting' to either 'Idle' or 'Running'
+
+ default value : **True**
+ timeout : int
+ maximum time (in seconds) to wait in blocking state until TRex changes state from 'Starting' to either 'Idle' or 'Running'
+
+ default value: **40**
+ user : str
+ the identity of the the run issuer.
+ trex_cmd_options : key, val
+ sets desired TRex options using key=val syntax, separated by comma.
+ for keys with no value, state key=True
+
+ :return:
+ **True** on success
+
+ :raises:
+ + :exc:`trex_exceptions.TRexError`, in case one of the trex_cmd_options raised an exception at server.
+ + :exc:`trex_exceptions.TRexInUseError`, in case TRex is already taken.
+ + :exc:`trex_exceptions.TRexRequestDenied`, in case TRex is reserved for another user than the one trying start TRex.
+ + ProtocolError, in case of error in JSON-RPC protocol.
+
+ """
+ try:
+ user = user or self.__default_user
+ retval = self.server.start_trex(trex_cmd_options, user, block_to_success, timeout, True)
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ except ProtocolError:
+ raise
+ finally:
+ self.prompt_verbose_data()
+
+ if retval!=0:
+ self.seq = retval # update seq num only on successful submission
+ return True
+ else: # TRex is has been started by another user
+ raise TRexInUseError('TRex is already being used by another user or process. Try again once TRex is back in IDLE state.')
+
+
def stop_trex (self):
"""
Request to stop a TRex run on server.
@@ -238,6 +320,40 @@ class CTRexClient(object):
finally:
self.prompt_verbose_data()
+ def kill_all_trexes(self):
+ """
+ Kills running TRex processes (if exists) on the server, not only owned by current daemon.
+ Raises exception upon error killing.
+
+ :return:
+ + **True** if any process killed
+ + **False** otherwise.
+
+ """
+ try:
+ return self.server.kill_all_trexes()
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ finally:
+ self.prompt_verbose_data()
+
+
+ def get_trex_cmds(self):
+ """
+ Gets list of running TRex pids and command lines.
+ Can be used to verify if any TRex is running.
+
+ :return:
+ List of tuples (pid, command) of running TRexes
+ """
+ try:
+ return self.server.get_trex_cmds()
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ finally:
+ self.prompt_verbose_data()
+
+
def wait_until_kickoff_finish(self, timeout = 40):
"""
Block the client application until TRex changes state from 'Starting' to either 'Idle' or 'Running'
@@ -306,8 +422,9 @@ class CTRexClient(object):
raise
except ProtocolError as err:
raise
- finally:
- self.prompt_verbose_data()
+ #is printed by self.get_running_info()
+ #finally:
+ # self.prompt_verbose_data()
def is_idle (self):
"""
@@ -844,11 +961,11 @@ class CTRexClient(object):
return method_to_call()
except socket.error as e:
if e.errno == errno.ECONNREFUSED:
- raise SocketError(errno.ECONNREFUSED, "Connection from TRex server was refused. Please make sure the server is up.")
+ raise SocketError(errno.ECONNREFUSED, "Connection to TRex daemon server was refused. Please make sure the server is up.")
def check_server_connectivity (self):
"""
- Checks for server valid connectivity.
+ Checks TRex daemon server for connectivity.
"""
try:
socket.gethostbyname(self.trex_host)
@@ -857,7 +974,86 @@ class CTRexClient(object):
raise socket.gaierror(e.errno, "Could not resolve server hostname. Please make sure hostname entered correctly.")
except socket.error as e:
if e.errno == errno.ECONNREFUSED:
- raise socket.error(errno.ECONNREFUSED, "Connection from TRex server was refused. Please make sure the server is up.")
+ raise socket.error(errno.ECONNREFUSED, "Connection to TRex daemon server was refused. Please make sure the server is up.")
+ finally:
+ self.prompt_verbose_data()
+
+
+ def master_add(self, x, y):
+ ''' Sanity check for Master daemon '''
+ try:
+ return self.master_daemon.add(x,y)
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ finally:
+ self.prompt_verbose_data()
+
+
+ def check_master_connectivity (self):
+ '''
+ Check Master daemon for connectivity.
+ Return True upon success
+ '''
+ try:
+ socket.gethostbyname(self.trex_host)
+ return self.master_daemon.check_connectivity()
+ except socket.gaierror as e:
+ raise socket.gaierror(e.errno, "Could not resolve server hostname. Please make sure hostname entered correctly.")
+ except socket.error as e:
+ if e.errno == errno.ECONNREFUSED:
+ raise socket.error(errno.ECONNREFUSED, "Connection to Master daemon was refused. Please make sure the server is up.")
+ finally:
+ self.prompt_verbose_data()
+
+ def is_trex_daemon_running(self):
+ '''
+ Check if TRex server daemon is running.
+ Returns True/False
+ '''
+ try:
+ return self.master_daemon.is_trex_daemon_running()
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ finally:
+ self.prompt_verbose_data()
+
+ def restart_trex_daemon(self):
+ '''
+ Restart TRex server daemon. Useful after update.
+ Will not fail if daemon is initially stopped.
+ '''
+ try:
+ return self.master_daemon.restart_trex_daemon()
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ finally:
+ self.prompt_verbose_data()
+
+ def start_trex_daemon(self):
+ '''
+ Start TRex server daemon.
+ :return:
+ + **True** if success.
+ + **False** if TRex server daemon already running.
+ '''
+ try:
+ return self.master_daemon.start_trex_daemon()
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
+ finally:
+ self.prompt_verbose_data()
+
+ def stop_trex_daemon(self):
+ '''
+ Stop TRex server daemon.
+ :return:
+ + **True** if success.
+ + **False** if TRex server daemon already running.
+ '''
+ try:
+ return self.master_daemon.stop_trex_daemon()
+ except AppError as err:
+ self._handle_AppError_exception(err.args[0])
finally:
self.prompt_verbose_data()