def _update_trex_process(package_path):
if not os.path.exists(tmp_dir):
os.makedirs(tmp_dir)
file_name = 'trex_package.tar.gz'
# getting new package
if package_path.startswith('http'):
ret_code, stdout, stderr = run_command('wget %s -O %s' % (package_path, os.path.join(tmp_dir, file_name)), timeout = 600)
else:
ret_code, stdout, stderr = run_command('rsync -Lc %s %s' % (package_path, os.path.join(tmp_dir, file_name)), timeout = 300)
if ret_code:
raise Exception('Could not get requested package. Result: %s' % [ret_code, stdout, stderr])
# calculating hash
ret_code, stdout, stderr = run_command('sha1sum -b %s' % os.path.join(tmp_dir, file_name), timeout = 30)
if ret_code:
raise Exception('Could not calculate hash of package. Result: %s' % [ret_code, stdout, stderr])
package_sha1 = stdout.strip().split()[0]
# clean old unpacked dirs
tmp_files = glob(os.path.join(tmp_dir, '*'))
for tmp_file in tmp_files:
if os.path.isdir(tmp_file) and not os.path.islink(tmp_file):
shutil.rmtree(tmp_file)
# unpacking
ret_code, stdout, stderr = run_command('tar -xzf %s' % os.path.join(tmp_dir, file_name), timeout = 120, cwd = tmp_dir)
if ret_code:
raise Exception('Could not untar the package. %s' % [ret_code, stdout, stderr])
tmp_files = glob(os.path.join(tmp_dir, '*'))
unpacked_dirs = []
for tmp_file in tmp_files:
if os.path.isdir(tmp_file) and not os.path.islink(tmp_file):
unpacked_dirs.append(tmp_file)
if len(unpacked_dirs) != 1:
raise Exception('Should be exactly one unpacked directory, got: %s' % unpacked_dirs)
os.chmod(unpacked_dirs[0], 0o777) # allow core dumps to be written
cur_dir = args.trex_dir
if os.path.islink(cur_dir) or os.path.isfile(cur_dir):
os.unlink(cur_dir)
if not os.path.exists(cur_dir):
os.makedirs(cur_dir)
os.chmod(cur_dir, 0o777)
bu_dir = '%s_BU%i' % (cur_dir, int(time.time()))
try:
# bu current dir
shutil.move(cur_dir, bu_dir)
shutil.move(unpacked_dirs[0], cur_dir)
CUpdate.info = {'path': package_path, 'sha1': package_sha1}
logging.info('Done updating, success')
except BaseException as e: # something went wrong, return backup dir
logging.error('Error while updating: %s' % e)
if os.path.exists(cur_dir):
shutil.rmtree(cur_dir)
shutil.move(bu_dir, cur_dir)
raise
finally:
if os.path.exists(bu_dir):
shutil.rmtree(bu_dir)
def get_trex_cmds(self):
logger.info('Processing get_trex_cmds() command.')
ret_code, stdout, stderr = 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 get_trex_version (self, base64 = True):
try:
logger.info("Processing get_trex_version() command.")
if not self.trex_version:
ret_code, stdout, stderr = run_command('./t-rex-64 --help', cwd = self.TREX_PATH)
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')
self.trex_version = binascii.b2a_base64(search_result.group(1).encode(errors='replace'))
if base64:
return self.trex_version.decode(errors='replace')
else:
return binascii.a2b_base64(self.trex_version).decode(errors='replace')
except Exception as e:
err_str = "Can't get trex version, error: %s" % e
logger.error(err_str)
return Fault(-33, err_str)
# returns True if given path is under current dir or /tmp
def _check_path_under_current_or_temp(path):
if not os.path.relpath(path, '/tmp').startswith(os.pardir):
return True
if not os.path.relpath(path, os.getcwd()).startswith(os.pardir):
return True
return False
### Main ###
if os.getuid() != 0:
fail('Please run this program as root/with sudo')
pid = os.getpid()
ret, out, err = run_command('taskset -pc 0 %s' % pid)
if ret:
fail('Could not set self affinity to core zero. Result: %s' %
[ret, out, err])
daemon_actions = OrderedDict([
('start', 'start the daemon'), ('stop', 'exit the daemon process'),
('show', 'prompt the status of daemon process (running / not running)'),
('restart', 'stop, then start again the daemon process')
])
actions_help = 'Specify action command to be applied on master daemon.\n' +\
'\n'.join([' (*) %-11s: %s' % (key, val) for key, val in daemon_actions.items()])
daemons = {}.fromkeys(['master_daemon', 'trex_daemon_server', 'stl_rpc_proxy'])
