def main() -> int:
# Initialize the argument parser and subparsers
# First we initialize general arguments.
parser = argparse.ArgumentParser(description="Linerate test control plane")
parser.add_argument(
"--server-addr",
type=str,
help="The server address",
default="127.0.0.1",
required=False,
)
parser.add_argument(
"--trex-config",
type=str,
help="The Trex config to be placed on the server.",
required=True,
)
parser.add_argument(
"--keep-running",
action="store_true",
default=False,
help="Keep Trex running after the test.",
)
parser.add_argument(
"--force-restart",
action="store_true",
default=False,
help="Force restart the Trex process " + "if there is one running.",
)
# Second, we initialize subparsers from all test scripts
subparsers = parser.add_subparsers(
dest="test",
help="The test profile, which is the " +
"filename(without .py) in the test directory",
required=True,
)
test_py_list = glob.glob(join(dirname(__file__), "tests", "*.py"))
test_list = [
basename(f)[:-3] for f in test_py_list
if isfile(f) and not f.endswith("__init__.py")
]
for test in test_list:
test_class = get_test_class(test)
if not test_class:
continue
test_parser = subparsers.add_parser(test)
test_class.setup_subparser(test_parser)
# Finally, we get the arguments
args = parser.parse_args()
# Set up the Trex server
if not os.path.exists(args.trex_config):
logging.error("Can not find Trex config file: %s", args.trex_config)
return
if not os.path.isfile(args.trex_config):
logging.error("%s is not a file", args.trex_config)
return 1
trex_config_file_on_server = TREX_FILES_DIR + os.path.basename(
args.trex_config)
trex_daemon_client = CTRexClient(args.server_addr)
trex_started = False
try:
logging.info("Pushing Trex config %s to the server", args.trex_config)
if not trex_daemon_client.push_files(args.trex_config):
logging.error("Unable to push %s to Trex server", args.trex_config)
return 1
if args.force_restart:
logging.info("Killing all Trexes... with meteorite... Boom!")
trex_daemon_client.kill_all_trexes()
# Wait until Trex enter the Idle state
start_time = time.time()
success = False
while time.time() - start_time < DEFAULT_KILL_TIMEOUT:
if trex_daemon_client.is_idle():
success = True
break
time.sleep(1)
if not success:
logging.error("Unable to kill Trex process, please login " +
"to the server and kill it manually.")
return 1
if not trex_daemon_client.is_idle():
logging.info("The Trex server process is running")
logging.warning("A Trex server process is still running, " +
"use --force-restart to kill it if necessary.")
return 1
test_class = get_test_class(args.test)
if not test_class:
logging.error("Unable to get test class for test %s", args.test)
return 1
test_type = test_class.test_type()
logging.info("Starting Trex with %s mode", test_class.test_type())
try:
start_trex_function = getattr(trex_daemon_client,
"start_{}".format(test_type))
except AttributeError:
logging.error("Unkonwon test type %s", test_type)
return 1
# Not checking the return value from this
# call since it always return True
start_trex_function(cfg=trex_config_file_on_server)
trex_started = True
# Start the test
if test_type == "stateless":
trex_client = STLClient(server=args.server_addr)
elif test_type == "astf":
trex_client = ASTFClient(server=args.server_addr)
else:
logging.error("Unknown test type %s", test_type)
return 1
test = test_class(trex_client)
try:
logging.info("Connecting to Trex server...")
trex_client.connect()
logging.info("Acquaring ports...")
trex_client.acquire()
logging.info("Resetting and clearing port...")
trex_client.reset() # Resets configs from all ports
trex_client.clear_stats() # Clear status from all ports
logging.info("Running the test: %s...", test)
test.start(args)
except STLError as e:
logging.error("Got error from Trex server: %s", e)
return 1
finally:
logging.info("Cleaning up Trex client")
trex_client.stop()
trex_client.release()
trex_client.disconnect()
except ConnectionRefusedError:
logging.error(
"Unable to connect to server %s.\n" +
"Did you start the Trex daemon?",
args.server_addr,
)
return 1
except ProtocolError as pe:
logging.error("%s", pe)
return 1
except TRexError as te:
logging.error("TRex error: %s", te.msg)
return 1
except TRexInUseError as tiue:
logging.error("TRex is already taken: %s", tiue.msg)
return 1
except TRexRequestDenied as trd:
logging.error("Request denied: %s", trd.msg)
return 1
finally:
if trex_started and not args.keep_running:
logging.info("Stopping Trex server")
trex_daemon_client.stop_trex()
def main():
"""
Read in all command line arguments.
"""
parser = argparse.ArgumentParser(
description="Start TRex daemon client and run PTF command")
parser.add_argument(
"--ptf-dir",
help="Directory containing PTF tests",
type=str,
required=True,
)
parser.add_argument(
"--trex-address",
help="Address of the remote TRex daemon server",
type=str,
required=False,
)
parser.add_argument(
"--bess-address",
help="Address of the remote BESS-UPF instance",
type=str,
required=False,
)
parser.add_argument(
"--trex-config",
help=
"Location of the TRex config file to be pushed to the remote server",
type=str,
required=False,
)
parser.add_argument(
"--trex-hw-mode",
help="Enables NIC HW acceleration and disables TRex software mode",
action="store_true",
required=False,
)
args, unknown_args = parser.parse_known_args()
# ensure PTF command is available
if not check_ptf():
error("Cannot find PTF executable")
sys.exit(1)
"""
Run either linerate or unary test case, depending on arguments.
"""
# if line rate test, need to perform set up of TRex traffic generator
if args.trex_address is not None:
if args.trex_hw_mode:
trex_args = None
else:
trex_args = "--software --no-hw-flow-stat"
trex_daemon_client = CTRexClient(args.trex_address,
trex_args=trex_args)
info("Starting TRex daemon client...")
success = set_up_trex_server(
trex_daemon_client,
args.trex_address,
args.trex_config,
)
if not success:
error("Failed to set up TRex daemon client!")
sys.exit(2)
info("Running linerate test(s)...")
success = run_test(
bess_addr=args.bess_address,
ptfdir=args.ptf_dir,
trex_server_addr=args.trex_address,
extra_args=unknown_args,
)
if not success:
error("Failed to run linerate tests!")
trex_daemon_client.stop_trex()
sys.exit(3)
trex_daemon_client.stop_trex()
# if unary test, can skip TRex set up and just run PTF command
else:
info("Running unary test(s)...")
success = run_test(
p4info_path=args.p4info,
device_id=args.device_id,
grpc_addr=args.grpc_addr,
cpu_port=args.cpu_port,
ptfdir=args.ptf_dir,
port_map_path=args.port_map,
platform=args.platform,
generate_tv=args.generate_tv,
loopback=args.loopback,
profile=args.profile,
extra_args=unknown_args,
)
if not success:
error("Failed running unary tests!")
sys.exit(4)