#!/usr/bin/env python """ Vpp QUIC tests """ import unittest import os import subprocess import signal from framework import VppTestCase, VppTestRunner, running_extended_tests, \ Worker from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath class QUICAppWorker(Worker): """ QUIC Test Application Worker """ def __init__(self, build_dir, appname, args, logger, env={}): app = "%s/vpp/bin/%s" % (build_dir, appname) self.args = [app] + args super(QUICAppWorker, self).__init__(self.args, logger, env) class QUICTestCase(VppTestCase): """ QUIC Test Case """ @classmethod def setUpClass(cls): super(QUICTestCase, cls).setUpClass() @classmethod def tearDownClass(cls): super(QUICTestCase, cls).tearDownClass() def setUp(self): var = "VPP_BUILD_DIR" self.build_dir = os.getenv(var, None) if self.build_dir is None: raise Exception("Environment variable `%s' not set" % var) self.vppDebug = 'vpp_debug' in self.build_dir self.timeout = 20 self.pre_test_sleep = 0.3 self.post_test_sleep = 0.3 self.vapi.session_enable_disable(is_enabled=1) def tearDown(self): self.vapi.session_enable_disable(is_enabled=0) def thru_host_stack_ipv4_setup(self): super(QUICTestCase, self).setUp() self.create_loopback_interfaces(2) self.uri = "quic://%s/1234" % self.loop0.local_ip4 common_args = ["uri", self.uri, "fifo-size", "64"] self.server_echo_test_args = common_args + ["appns", "server"] self.client_echo_test_args = common_args + ["appns", "client", "test-bytes"] table_id = 1 for i in self.lo_interfaces: i.admin_up() if table_id != 0: tbl = VppIpTable(self, table_id) tbl.add_vpp_config() i.set_table_ip4(table_id) i.config_ip4() table_id += 1 # Configure namespaces self.vapi.app_namespace_add_del(namespace_id=b"server", sw_if_index=self.loop0.sw_if_index) self.vapi.app_namespace_add_del(namespace_id=b"client", sw_if_index=self.loop1.sw_if_index) # Add inter-table routes self.ip_t01 = VppIpRoute(self, self.loop1.local_ip4, 32, [VppRoutePath("0.0.0.0", 0xffffffff, nh_table_id=2)], table_id=1) self.ip_t10 = VppIpRoute(self, self.loop0.local_ip4, 32, [VppRoutePath("0.0.0.0", 0xffffffff, nh_table_id=1)], table_id=2) self.ip_t01.add_vpp_config() self.ip_t10.add_vpp_config() self.logger.debug(self.vapi.cli("show ip fib")) def thru_host_stack_ipv4_tear_down(self): # Delete inter-table routes self.ip_t01.remove_vpp_config() self.ip_t10.remove_vpp_config() for i in self.lo_interfaces: i.unconfig_ip4() i.set_table_ip4(0) i.admin_down() def start_internal_echo_server(self, args): error = self.vapi.cli("test echo server %s" % ' '.join(args)) if error: self.logger.critical(error) self.assertNotIn("failed", error) def start_internal_echo_client(self, args): error = self.vapi.cli("test echo client %s" % ' '.join(args)) if error: self.logger.critical(error) self.assertNotIn("failed", error) def internal_ipv4_transfer_test(self, server_args, client_args): self.start_internal_echo_server(server_args) self.start_internal_echo_client(client_args) def start_external_echo_server(self, args): self.worker_server = QUICAppWorker(self.build_dir, "quic_echo", args, self.logger) self.worker_server.start() def start_external_echo_client(self, args): self.client_echo_test_args += "use-svm-api" self.worker_client = QUICAppWorker(self.build_dir, "quic_echo", args, self.logger) self.worker_client.start() self.worker_client.join(self.timeout) try: self.validateExternalTestResults() except Exception as error: self.fail("Failed with %s" % error) def external_ipv4_transfer_test(self, server_args, client_args): self.start_external_echo_server(server_args) self.sleep(self.pre_test_sleep) self.start_external_echo_client(client_args) self.sleep(self.post_test_sleep) def validateExternalTestResults(self): if os.path.isdir('/proc/{}'.format(self.worker_server.process.pid)): self.logger.info("Killing server worker process (pid %d)" % self.worker_server.process.pid) os.killpg(os.getpgid(self.worker_server.process.pid), signal.SIGTERM) self.worker_server.join() self.logger.info("Client worker result is `%s'" % self.worker_client.result) error = False if self.worker_client.result is None: try: error = True self.logger.error( "Timeout: %ss! Killing client worker process (pid %d)" % (self.timeout, self.worker_client.process.pid)) os.killpg(os.getpgid(self.worker_client.process.pid), signal.SIGKILL) self.worker_client.join() except OSError: self.logger.debug( "Couldn't kill client worker process") raise if error: raise Exception( "Timeout! Client worker did not finish in %ss" % timeout) self.assert_equal(self.worker_client.result, 0, "Binary test return code") class QUICInternalEchoIPv4TestCase(QUICTestCase): """ QUIC Internal Echo IPv4 Transfer Test Cases """ @classmethod def setUpClass(cls): super(QUICInternalEchoIPv4TestCase, cls).setUpClass() @classmethod def tearDownClass(cls): super(QUICInternalEchoIPv4TestCase, cls).tearDownClass() def setUp(self): super(QUICInternalEchoIPv4TestCase, self).setUp() self.thru_host_stack_ipv4_setup() def tearDown(self): self.thru_host_stack_ipv4_tear_down() super(QUICInternalEchoIPv4TestCase, self).tearDown() def show_commands_at_teardown(self): self.logger.debug(self.vapi.cli("show session verbose 2")) @unittest.skipUnless(running_extended_tests, "part of extended tests") def test_quic_internal_transfer(self): """ QUIC internal echo client/server transfer """ self.internal_ipv4_transfer_test(self.server_echo_test_args, self.client_echo_test_args + ["no-output", "mbytes", "10"]) class QUICInternalEchoIPv4MultiStreamTestCase(QUICTestCase): """ QUIC Internal Echo IPv4 Transfer Test Cases """ @classmethod def setUpClass(cls): super(QUICInternalEchoIPv4MultiStreamTestCase, cls).setUpClass() @classmethod def tearDownClass(cls): super(QUICInternalEchoIPv4MultiStreamTestCase, cls).tearDownClass() def setUp(self): super(QUICInternalEchoIPv4MultiStreamTestCase, self).setUp() self.thru_host_stack_ipv4_setup() def tearDown(self): self.thru_host_stack_ipv4_tear_down() super(QUICInternalEchoIPv4MultiStreamTestCase, self).tearDown() def show_commands_at_teardown(self): self.logger.debug(self.vapi.cli("show session verbose 2")) @unittest.skipUnless(running_extended_tests, "part of extended tests") def test_quic_internal_multistream_transfer(self): """ QUIC internal echo client/server multi-stream transfer """ self.internal_ipv4_transfer_test(self.server_echo_test_args, self.client_echo_test_args + ["quic-streams", "10", "mbytes", "1", "no-output"]) class QUICExternalEchoIPv4TestCase(QUICTestCase): """ QUIC External Echo IPv4 Transfer Test Cases """ @classmethod def setUpClass(cls): super(QUICExternalEchoIPv4TestCase, cls).setUpClass() @classmethod def tearDownClass(cls): super(QUICExternalEchoIPv4TestCase, cls).tearDownClass() def setUp(self): super(QUICExternalEchoIPv4TestCase, self).setUp() self.thru_host_stack_ipv4_setup() def tearDown(self): super(QUICExternalEchoIPv4TestCase, self).tearDown() self.thru_host_stack_ipv4_tear_down() def show_commands_at_teardown(self): self.logger.debug(self.vapi.cli("show session verbose 2")) @unittest.skipUnless(running_extended_tests, "part of extended tests") def test_quic_external_transfer(self): """ QUIC external echo client/server transfer """ self.external_ipv4_transfer_test(self.server_echo_test_args + ["socket-name", self.api_sock], self.client_echo_test_args + ["socket-name", self.api_sock, "mbytes", "10"]) if __name__ == '__main__': unittest.main(testRunner=VppTestRunner)