8 from framework import VppTestCase, VppTestRunner, running_extended_tests, \
10 from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath, DpoProto
13 class VCLAppWorker(Worker):
14 """ VCL Test Application Worker """
16 def __init__(self, build_dir, appname, args, logger, env={}):
17 vcl_lib_dir = "%s/vpp/lib" % build_dir
18 if "iperf" in appname:
20 env.update({'LD_PRELOAD':
21 "%s/libvcl_ldpreload.so" % vcl_lib_dir})
22 elif "sock" in appname:
23 app = "%s/vpp/bin/%s" % (build_dir, appname)
24 env.update({'LD_PRELOAD':
25 "%s/libvcl_ldpreload.so" % vcl_lib_dir})
27 app = "%s/vpp/bin/%s" % (build_dir, appname)
28 self.args = [app] + args
29 super(VCLAppWorker, self).__init__(self.args, logger, env)
32 class VCLTestCase(VppTestCase):
33 """ VCL Test Class """
35 def __init__(self, methodName):
37 self.build_dir = os.getenv(var, None)
38 if self.build_dir is None:
39 raise Exception("Environment variable `%s' not set" % var)
40 self.vppDebug = 'vpp_debug' in self.build_dir
41 self.server_addr = "127.0.0.1"
42 self.server_port = "22000"
43 self.server_args = [self.server_port]
44 self.server_ipv6_addr = "::1"
45 self.server_ipv6_args = ["-6", self.server_port]
47 self.echo_phrase = "Hello, world! Jenny is a friend of mine."
48 self.pre_test_sleep = 0.3
49 self.post_test_sleep = 0.2
51 if os.path.isfile("/tmp/ldp_server_af_unix_socket"):
52 os.remove("/tmp/ldp_server_af_unix_socket")
54 super(VCLTestCase, self).__init__(methodName)
56 def cut_thru_setup(self):
57 self.vapi.session_enable_disable(is_enabled=1)
59 def cut_thru_tear_down(self):
60 self.vapi.session_enable_disable(is_enabled=0)
62 def cut_thru_test(self, server_app, server_args, client_app, client_args):
63 self.env = {'VCL_API_PREFIX': self.shm_prefix,
64 'VCL_APP_SCOPE_LOCAL': "true"}
66 worker_server = VCLAppWorker(self.build_dir, server_app, server_args,
67 self.logger, self.env)
69 self.sleep(self.pre_test_sleep)
70 worker_client = VCLAppWorker(self.build_dir, client_app, client_args,
71 self.logger, self.env)
73 worker_client.join(self.timeout)
75 self.validateResults(worker_client, worker_server, self.timeout)
76 except Exception as error:
77 self.fail("Failed with %s" % error)
78 self.sleep(self.post_test_sleep)
80 def thru_host_stack_setup(self):
81 self.vapi.session_enable_disable(is_enabled=1)
82 self.create_loopback_interfaces(2)
86 for i in self.lo_interfaces:
90 tbl = VppIpTable(self, table_id)
93 i.set_table_ip4(table_id)
97 # Configure namespaces
98 self.vapi.app_namespace_add(namespace_id="1", secret=1234,
99 sw_if_index=self.loop0.sw_if_index)
100 self.vapi.app_namespace_add(namespace_id="2", secret=5678,
101 sw_if_index=self.loop1.sw_if_index)
103 # Add inter-table routes
104 ip_t01 = VppIpRoute(self, self.loop1.local_ip4, 32,
105 [VppRoutePath("0.0.0.0",
107 nh_table_id=2)], table_id=1)
108 ip_t10 = VppIpRoute(self, self.loop0.local_ip4, 32,
109 [VppRoutePath("0.0.0.0",
111 nh_table_id=1)], table_id=2)
112 ip_t01.add_vpp_config()
113 ip_t10.add_vpp_config()
114 self.logger.debug(self.vapi.cli("show ip fib"))
116 def thru_host_stack_tear_down(self):
117 for i in self.lo_interfaces:
122 self.vapi.session_enable_disable(is_enabled=0)
124 def thru_host_stack_ipv6_setup(self):
125 self.vapi.session_enable_disable(is_enabled=1)
126 self.create_loopback_interfaces(2)
130 for i in self.lo_interfaces:
133 tbl = VppIpTable(self, table_id, is_ip6=1)
136 i.set_table_ip6(table_id)
140 # Configure namespaces
141 self.vapi.app_namespace_add(namespace_id="1", secret=1234,
142 sw_if_index=self.loop0.sw_if_index)
143 self.vapi.app_namespace_add(namespace_id="2", secret=5678,
144 sw_if_index=self.loop1.sw_if_index)
146 # Add inter-table routes
147 ip_t01 = VppIpRoute(self, self.loop1.local_ip6, 128,
148 [VppRoutePath("::0", 0xffffffff,
150 proto=DpoProto.DPO_PROTO_IP6)],
151 table_id=1, is_ip6=1)
152 ip_t10 = VppIpRoute(self, self.loop0.local_ip6, 128,
153 [VppRoutePath("::0", 0xffffffff,
155 proto=DpoProto.DPO_PROTO_IP6)],
156 table_id=2, is_ip6=1)
157 ip_t01.add_vpp_config()
158 ip_t10.add_vpp_config()
159 self.logger.debug(self.vapi.cli("show interface addr"))
160 self.logger.debug(self.vapi.cli("show ip6 fib"))
162 def thru_host_stack_ipv6_tear_down(self):
163 for i in self.lo_interfaces:
168 self.vapi.session_enable_disable(is_enabled=0)
170 def thru_host_stack_test(self, server_app, server_args,
171 client_app, client_args):
172 self.env = {'VCL_API_PREFIX': self.shm_prefix,
173 'VCL_APP_SCOPE_GLOBAL': "true",
174 'VCL_APP_NAMESPACE_ID': "1",
175 'VCL_APP_NAMESPACE_SECRET': "1234"}
177 worker_server = VCLAppWorker(self.build_dir, server_app, server_args,
178 self.logger, self.env)
179 worker_server.start()
180 self.sleep(self.pre_test_sleep)
182 self.env.update({'VCL_APP_NAMESPACE_ID': "2",
183 'VCL_APP_NAMESPACE_SECRET': "5678"})
184 worker_client = VCLAppWorker(self.build_dir, client_app, client_args,
185 self.logger, self.env)
186 worker_client.start()
187 worker_client.join(self.timeout)
190 self.validateResults(worker_client, worker_server, self.timeout)
191 except Exception as error:
192 self.fail("Failed with %s" % error)
193 self.sleep(self.post_test_sleep)
195 def validateResults(self, worker_client, worker_server, timeout):
196 if os.path.isdir('/proc/{}'.format(worker_server.process.pid)):
197 self.logger.info("Killing server worker process (pid %d)" %
198 worker_server.process.pid)
199 os.killpg(os.getpgid(worker_server.process.pid), signal.SIGKILL)
201 self.logger.info("Client worker result is `%s'" % worker_client.result)
203 if worker_client.result is None:
207 "Timeout: %ss! Killing client worker process (pid %d)" %
208 (timeout, worker_client.process.pid))
209 os.killpg(os.getpgid(worker_client.process.pid),
214 "Couldn't kill client worker process")
218 "Timeout! Client worker did not finish in %ss" % timeout)
219 self.assert_equal(worker_client.result, 0, "Binary test return code")
222 class LDPCutThruTestCase(VCLTestCase):
223 """ LDP Cut Thru Tests """
226 super(LDPCutThruTestCase, self).setUp()
228 self.cut_thru_setup()
229 self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
230 self.server_addr, self.server_port]
231 self.client_iperf3_timeout = 20
232 self.client_iperf3_args = ["-V4d", "-t 2", "-c", self.server_addr]
233 self.server_iperf3_args = ["-V4d", "-s"]
234 self.client_uni_dir_nsock_timeout = 20
235 self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
239 self.client_bi_dir_nsock_timeout = 20
240 self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
246 self.cut_thru_tear_down()
248 super(LDPCutThruTestCase, self).tearDown()
250 @unittest.skipUnless(running_extended_tests, "part of extended tests")
251 def test_ldp_cut_thru_echo(self):
252 """ run LDP cut thru echo test """
254 self.cut_thru_test("sock_test_server", self.server_args,
255 "sock_test_client", self.client_echo_test_args)
257 def test_ldp_cut_thru_iperf3(self):
258 """ run LDP cut thru iperf3 test """
261 subprocess.check_output(['iperf3', '-v'])
262 except subprocess.CalledProcessError:
263 self.logger.error("WARNING: 'iperf3' is not installed,")
264 self.logger.error(" 'test_ldp_cut_thru_iperf3' not run!")
267 self.timeout = self.client_iperf3_timeout
268 self.cut_thru_test("iperf3", self.server_iperf3_args,
269 "iperf3", self.client_iperf3_args)
271 @unittest.skipUnless(running_extended_tests, "part of extended tests")
272 def test_ldp_cut_thru_uni_dir_nsock(self):
273 """ run LDP cut thru uni-directional (multiple sockets) test """
275 self.timeout = self.client_uni_dir_nsock_timeout
276 self.cut_thru_test("sock_test_server", self.server_args,
278 self.client_uni_dir_nsock_test_args)
280 @unittest.skipUnless(running_extended_tests, "part of extended tests")
281 def test_ldp_cut_thru_bi_dir_nsock(self):
282 """ run LDP cut thru bi-directional (multiple sockets) test """
284 self.timeout = self.client_bi_dir_nsock_timeout
285 self.cut_thru_test("sock_test_server", self.server_args,
287 self.client_bi_dir_nsock_test_args)
290 class VCLCutThruTestCase(VCLTestCase):
291 """ VCL Cut Thru Tests """
294 super(VCLCutThruTestCase, self).setUp()
296 self.cut_thru_setup()
297 self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
298 self.server_addr, self.server_port]
300 self.client_uni_dir_nsock_timeout = 20
301 self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
305 self.client_bi_dir_nsock_timeout = 20
306 self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
312 self.cut_thru_tear_down()
314 super(VCLCutThruTestCase, self).tearDown()
316 def test_vcl_cut_thru_echo(self):
317 """ run VCL cut thru echo test """
319 self.cut_thru_test("vcl_test_server", self.server_args,
320 "vcl_test_client", self.client_echo_test_args)
322 def test_vcl_cut_thru_uni_dir_nsock(self):
323 """ run VCL cut thru uni-directional (multiple sockets) test """
325 self.timeout = self.client_uni_dir_nsock_timeout
326 self.cut_thru_test("vcl_test_server", self.server_args,
328 self.client_uni_dir_nsock_test_args)
330 def test_vcl_cut_thru_bi_dir_nsock(self):
331 """ run VCL cut thru bi-directional (multiple sockets) test """
333 self.timeout = self.client_bi_dir_nsock_timeout
334 self.cut_thru_test("vcl_test_server", self.server_args,
336 self.client_bi_dir_nsock_test_args)
339 class LDPThruHostStackEcho(VCLTestCase):
340 """ LDP Thru Host Stack Echo """
343 super(LDPThruHostStackEcho, self).setUp()
345 self.thru_host_stack_setup()
346 self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
347 self.loop0.local_ip4,
351 self.thru_host_stack_tear_down()
352 super(LDPThruHostStackEcho, self).tearDown()
354 def test_ldp_thru_host_stack_echo(self):
355 """ run LDP thru host stack echo test """
357 self.thru_host_stack_test("sock_test_server", self.server_args,
359 self.client_echo_test_args)
362 class VCLThruHostStackEcho(VCLTestCase):
363 """ VCL Thru Host Stack Echo """
366 super(VCLThruHostStackEcho, self).setUp()
368 self.thru_host_stack_setup()
369 self.client_bi_dir_nsock_timeout = 20
370 self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
372 self.loop0.local_ip4,
374 self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
375 self.loop0.local_ip4,
379 self.logger.debug(self.vapi.cli("show app server"))
380 self.logger.debug(self.vapi.cli("show session verbose"))
381 self.thru_host_stack_tear_down()
382 super(VCLThruHostStackEcho, self).tearDown()
385 class VCLThruHostStackBidirNsock(VCLTestCase):
386 """ VCL Thru Host Stack Bidir Nsock """
389 super(VCLThruHostStackBidirNsock, self).setUp()
391 self.thru_host_stack_setup()
392 self.client_bi_dir_nsock_timeout = 20
393 self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
395 self.loop0.local_ip4,
397 self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
398 self.loop0.local_ip4,
402 self.thru_host_stack_tear_down()
403 super(VCLThruHostStackBidirNsock, self).tearDown()
405 def test_vcl_thru_host_stack_bi_dir_nsock(self):
406 """ run VCL thru host stack bi-directional (multiple sockets) test """
408 self.timeout = self.client_bi_dir_nsock_timeout
409 self.thru_host_stack_test("vcl_test_server", self.server_args,
411 self.client_bi_dir_nsock_test_args)
414 class LDPThruHostStackBidirNsock(VCLTestCase):
415 """ LDP Thru Host Stack Bidir Nsock """
418 super(LDPThruHostStackBidirNsock, self).setUp()
420 self.thru_host_stack_setup()
422 self.client_bi_dir_nsock_timeout = 20
423 self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
424 # OUCH! Host Stack Bug?
426 self.loop0.local_ip4,
429 self.client_bi_dir_nsock_timeout = 20
430 self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
431 # OUCH! Host Stack Bug?
433 self.loop0.local_ip4,
437 self.thru_host_stack_tear_down()
438 super(LDPThruHostStackBidirNsock, self).tearDown()
440 def test_ldp_thru_host_stack_bi_dir_nsock(self):
441 """ run LDP thru host stack bi-directional (multiple sockets) test """
443 self.timeout = self.client_bi_dir_nsock_timeout
444 self.thru_host_stack_test("sock_test_server", self.server_args,
446 self.client_bi_dir_nsock_test_args)
449 class LDPThruHostStackNsock(VCLTestCase):
450 """ LDP Thru Host Stack Nsock """
453 super(LDPThruHostStackNsock, self).setUp()
455 self.thru_host_stack_setup()
457 self.client_uni_dir_nsock_timeout = 20
458 self.numSockets = "2"
460 self.client_uni_dir_nsock_timeout = 20
461 self.numSockets = "5"
463 self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
464 "-I", self.numSockets,
465 self.loop0.local_ip4,
469 self.thru_host_stack_tear_down()
470 super(LDPThruHostStackNsock, self).tearDown()
472 def test_ldp_thru_host_stack_uni_dir_nsock(self):
473 """ run LDP thru host stack uni-directional (multiple sockets) test """
475 self.timeout = self.client_uni_dir_nsock_timeout
476 self.thru_host_stack_test("sock_test_server", self.server_args,
478 self.client_uni_dir_nsock_test_args)
481 class VCLThruHostStackNsock(VCLTestCase):
482 """ VCL Thru Host Stack Nsock """
485 super(VCLThruHostStackNsock, self).setUp()
487 self.thru_host_stack_setup()
489 self.client_uni_dir_nsock_timeout = 20
490 self.numSockets = "2"
492 self.client_uni_dir_nsock_timeout = 20
493 self.numSockets = "5"
495 self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
496 "-I", self.numSockets,
497 self.loop0.local_ip4,
501 self.thru_host_stack_tear_down()
502 super(VCLThruHostStackNsock, self).tearDown()
504 def test_vcl_thru_host_stack_uni_dir_nsock(self):
505 """ run VCL thru host stack uni-directional (multiple sockets) test """
507 self.timeout = self.client_uni_dir_nsock_timeout
508 self.thru_host_stack_test("vcl_test_server", self.server_args,
510 self.client_uni_dir_nsock_test_args)
513 class LDPThruHostStackIperf(VCLTestCase):
514 """ LDP Thru Host Stack Iperf """
517 super(LDPThruHostStackIperf, self).setUp()
519 self.thru_host_stack_setup()
520 self.client_iperf3_timeout = 20
521 self.client_iperf3_args = ["-V4d", "-t 2", "-c", self.loop0.local_ip4]
522 self.server_iperf3_args = ["-V4d", "-s"]
525 self.thru_host_stack_tear_down()
526 super(LDPThruHostStackIperf, self).tearDown()
528 def test_ldp_thru_host_stack_iperf3(self):
529 """ run LDP thru host stack iperf3 test """
532 subprocess.check_output(['iperf3', '-v'])
533 except subprocess.CalledProcessError:
534 self.logger.error("WARNING: 'iperf3' is not installed,")
536 " 'test_ldp_thru_host_stack_iperf3' not run!")
539 self.timeout = self.client_iperf3_timeout
540 self.thru_host_stack_test("iperf3", self.server_iperf3_args,
541 "iperf3", self.client_iperf3_args)
544 class LDPIpv6CutThruTestCase(VCLTestCase):
545 """ LDP IPv6 Cut Thru Tests """
548 super(LDPIpv6CutThruTestCase, self).setUp()
550 self.cut_thru_setup()
551 self.client_iperf3_timeout = 20
552 self.client_uni_dir_nsock_timeout = 20
553 self.client_bi_dir_nsock_timeout = 20
554 self.client_ipv6_echo_test_args = ["-6", "-E", self.echo_phrase, "-X",
555 self.server_ipv6_addr,
557 self.client_ipv6_iperf3_args = ["-V6d", "-t 2", "-c",
558 self.server_ipv6_addr]
559 self.server_ipv6_iperf3_args = ["-V6d", "-s"]
560 self.client_ipv6_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
563 self.server_ipv6_addr,
565 self.client_ipv6_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
568 self.server_ipv6_addr,
572 self.cut_thru_tear_down()
574 super(LDPIpv6CutThruTestCase, self).tearDown()
576 def test_ldp_ipv6_cut_thru_echo(self):
577 """ run LDP IPv6 cut thru echo test """
579 self.cut_thru_test("sock_test_server",
580 self.server_ipv6_args,
582 self.client_ipv6_echo_test_args)
584 @unittest.skipUnless(running_extended_tests, "part of extended tests")
585 def test_ldp_ipv6_cut_thru_iperf3(self):
586 """ run LDP IPv6 cut thru iperf3 test """
589 subprocess.check_output(['iperf3', '-v'])
591 self.logger.error("WARNING: 'iperf3' is not installed,")
593 " 'test_ldp_ipv6_cut_thru_iperf3' not run!")
596 self.timeout = self.client_iperf3_timeout
597 self.cut_thru_test("iperf3", self.server_ipv6_iperf3_args,
598 "iperf3", self.client_ipv6_iperf3_args)
600 @unittest.skipUnless(running_extended_tests, "part of extended tests")
601 def test_ldp_ipv6_cut_thru_uni_dir_nsock(self):
602 """ run LDP IPv6 cut thru uni-directional (multiple sockets) test """
604 self.timeout = self.client_uni_dir_nsock_timeout
605 self.cut_thru_test("sock_test_server", self.server_ipv6_args,
607 self.client_ipv6_uni_dir_nsock_test_args)
609 @unittest.skipUnless(running_extended_tests, "part of extended tests")
610 def test_ldp_ipv6_cut_thru_bi_dir_nsock(self):
611 """ run LDP IPv6 cut thru bi-directional (multiple sockets) test """
613 self.timeout = self.client_bi_dir_nsock_timeout
614 self.cut_thru_test("sock_test_server", self.server_ipv6_args,
616 self.client_ipv6_bi_dir_nsock_test_args)
619 class VCLIpv6CutThruTestCase(VCLTestCase):
620 """ VCL IPv6 Cut Thru Tests """
623 super(VCLIpv6CutThruTestCase, self).setUp()
625 self.cut_thru_setup()
626 self.client_uni_dir_nsock_timeout = 20
627 self.client_bi_dir_nsock_timeout = 20
628 self.client_ipv6_echo_test_args = ["-6", "-E", self.echo_phrase, "-X",
629 self.server_ipv6_addr,
631 self.client_ipv6_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
634 self.server_ipv6_addr,
636 self.client_ipv6_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
639 self.server_ipv6_addr,
643 self.cut_thru_tear_down()
645 super(VCLIpv6CutThruTestCase, self).tearDown()
647 def test_vcl_ipv6_cut_thru_echo(self):
648 """ run VCL IPv6 cut thru echo test """
650 self.cut_thru_test("vcl_test_server",
651 self.server_ipv6_args,
653 self.client_ipv6_echo_test_args)
655 @unittest.skipUnless(running_extended_tests, "part of extended tests")
656 def test_vcl_ipv6_cut_thru_uni_dir_nsock(self):
657 """ run VCL IPv6 cut thru uni-directional (multiple sockets) test """
659 self.timeout = self.client_uni_dir_nsock_timeout
660 self.cut_thru_test("vcl_test_server", self.server_ipv6_args,
662 self.client_ipv6_uni_dir_nsock_test_args)
664 @unittest.skipUnless(running_extended_tests, "part of extended tests")
665 def test_vcl_ipv6_cut_thru_bi_dir_nsock(self):
666 """ run VCL IPv6 cut thru bi-directional (multiple sockets) test """
668 self.timeout = self.client_bi_dir_nsock_timeout
669 self.cut_thru_test("vcl_test_server", self.server_ipv6_args,
671 self.client_ipv6_bi_dir_nsock_test_args)
674 class VCLIpv6ThruHostStackEcho(VCLTestCase):
675 """ VCL IPv6 Thru Host Stack Echo """
678 super(VCLIpv6ThruHostStackEcho, self).setUp()
680 self.thru_host_stack_ipv6_setup()
681 self.client_ipv6_echo_test_args = ["-6", "-E", self.echo_phrase, "-X",
682 self.loop0.local_ip6,
686 self.thru_host_stack_ipv6_tear_down()
687 super(VCLIpv6ThruHostStackEcho, self).tearDown()
689 def test_vcl_ipv6_thru_host_stack_echo(self):
690 """ run VCL IPv6 thru host stack echo test """
692 self.thru_host_stack_test("vcl_test_server",
693 self.server_ipv6_args,
695 self.client_ipv6_echo_test_args)
698 if __name__ == '__main__':
699 unittest.main(testRunner=VppTestRunner)