def update_trex(package_path='http://trex-tgn.cisco.com/trex/release/latest'):
if not args.allow_update:
raise Exception('Updating server not allowed')
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])
# 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()))
try:
# bu current dir
shutil.move(cur_dir, bu_dir)
shutil.move(unpacked_dirs[0], cur_dir)
# no errors, remove BU dir
if os.path.exists(bu_dir):
shutil.rmtree(bu_dir)
return True
except: # something went wrong, return backup dir
if os.path.exists(cur_dir):
shutil.rmtree(cur_dir)
shutil.move(bu_dir, cur_dir)
raise
def update_trex(package_path='http://trex-tgn.cisco.com/trex/release/latest'):
if not args.allow_update:
raise Exception('Updating server not allowed')
# getting new package
if package_path.startswith('http'):
file_name = package_path.split('/')[-1]
ret_code, stdout, stderr = run_command(
'wget %s -O %s' % (package_path, os.path.join(tmp_dir, file_name)),
timeout=600)
else:
file_name = os.path.basename(package_path)
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])
# clean old unpacked dirs
unpacked_dirs = glob(os.path.join(tmp_dir, 'v[0-9].[0-9][0-9]'))
for unpacked_dir in unpacked_dirs:
shutil.rmtree(unpacked_dir)
# unpacking
ret_code, stdout, stderr = run_command('tar -xzf %s' %
os.path.join(tmp_dir, file_name),
timeout=60,
cwd=tmp_dir)
if ret_code:
raise Exception('Could not untar the package. %s' %
[ret_code, stdout, stderr])
unpacked_dirs = glob(os.path.join(tmp_dir, 'v[0-9].[0-9][0-9]'))
if not len(unpacked_dirs) or len(unpacked_dirs) > 1:
raise Exception('Should be exactly one unpacked directory, got: %s' %
unpacked_dirs)
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()))
try:
# bu current dir
shutil.move(cur_dir, bu_dir)
shutil.move(unpacked_dirs[0], cur_dir)
# no errors, remove BU dir
if os.path.exists(bu_dir):
shutil.rmtree(bu_dir)
return True
except: # something went wrong, return backup dir
if os.path.exists(cur_dir):
shutil.rmtree(cur_dir)
shutil.move(bu_dir, cur_dir)
raise
def update_trex(package_path = 'http://trex-tgn.cisco.com/trex/release/latest'):
if not args.allow_update:
raise Exception('Updating server not allowed')
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])
# 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)
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()))
try:
# bu current dir
shutil.move(cur_dir, bu_dir)
shutil.move(unpacked_dirs[0], cur_dir)
# no errors, remove BU dir
if os.path.exists(bu_dir):
shutil.rmtree(bu_dir)
return True
except: # something went wrong, return backup dir
if os.path.exists(cur_dir):
shutil.rmtree(cur_dir)
shutil.move(bu_dir, cur_dir)
raise
def update_trex(package_path="http://trex-tgn.cisco.com/trex/release/latest"):
if not args.allow_update:
raise Exception("Updading server not allowed")
# getting new package
if package_path.startswith("http://"):
file_name = package_path.split("/")[-1]
ret_code, stdout, stderr = run_command(
"wget %s -O %s" % (package_path, os.path.join(tmp_dir, file_name)), timeout=600
)
else:
file_name = os.path.basename(package_path)
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])
# clean old unpacked dirs
unpacked_dirs = glob(os.path.join(tmp_dir, "v[0-9].[0-9][0-9]"))
for unpacked_dir in unpacked_dirs:
shutil.rmtree(unpacked_dir)
# unpacking
ret_code, stdout, stderr = run_command("tar -xzf %s" % os.path.join(tmp_dir, file_name), timeout=60, cwd=tmp_dir)
if ret_code:
raise Exception("Could not untar the package. %s" % [ret_code, stdout, stderr])
unpacked_dirs = glob(os.path.join(tmp_dir, "v[0-9].[0-9][0-9]"))
if not len(unpacked_dirs) or len(unpacked_dirs) > 1:
raise Exception("Should be exactly one unpacked directory, got: %s" % unpacked_dirs)
try:
# bu current dir
cur_dir = args.trex_dir
bu_dir = "%s_BU%i" % (cur_dir, int(time()))
shutil.move(cur_dir, bu_dir)
shutil.move(unpacked_dirs[0], cur_dir)
# no errors, remove BU dir
if os.path.exists(bu_dir):
shutil.rmtree(bu_dir)
return True
except: # something went wrong, return backup dir
if os.path.exists(cur_dir):
shutil.rmtree(cur_dir)
shutil.move(bu_dir, cur_dir)
raise
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_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))
if base64:
return self.trex_version
else:
return binascii.a2b_base64(self.trex_version)
except Exception as e:
err_str = "Can't get trex version, error: %s" % e
logger.error(err_str)
return Fault(-33, err_str)
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'])
# show -p --master_port METAVAR instead of -p METAVAR --master_port METAVAR
class MyFormatter(RawTextHelpFormatter):
def _update_trex_process(package_path):
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)
updated_package_info['path'] = package_path
updated_package_info['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)
