diff options
author | 2016-05-13 20:11:51 +0300 | |
---|---|---|
committer | 2016-05-13 20:11:51 +0300 | |
commit | 89b608ae766705950efc5f4914b01b9a32b6a0e7 (patch) | |
tree | e5b044865eb02fedce4d4e3f9b735cadeaf5f997 /scripts/automation/trex_control_plane | |
parent | fa7a837fb19af1c1fe4365de3f71467b79c2c865 (diff) |
add master daemon
Diffstat (limited to 'scripts/automation/trex_control_plane')
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() |