Author: melifaro Date: Mon Mar 23 12:21:32 2020 New Revision: 359235 URL: https://svnweb.freebsd.org/changeset/base/359235
Log: Add tests verifying IPv4/IPv6 output for TCP/UDP/raw paths. Reviewed by: kp MFC after: 2 weeks Differential Revision: https://reviews.freebsd.org/D24138 Added: head/tests/sys/common/net_receiver.py (contents, props changed) head/tests/sys/netinet/output.sh (contents, props changed) head/tests/sys/netinet6/output6.sh (contents, props changed) Modified: head/tests/sys/common/Makefile head/tests/sys/common/vnet.subr head/tests/sys/netinet/Makefile head/tests/sys/netinet6/Makefile Modified: head/tests/sys/common/Makefile ============================================================================== --- head/tests/sys/common/Makefile Mon Mar 23 12:17:13 2020 (r359234) +++ head/tests/sys/common/Makefile Mon Mar 23 12:21:32 2020 (r359235) @@ -5,8 +5,10 @@ TESTSDIR= ${TESTSBASE}/sys/common ${PACKAGE}FILES+= vnet.subr ${PACKAGE}FILES+= divert.py ${PACKAGE}FILES+= sender.py +${PACKAGE}FILES+= net_receiver.py ${PACKAGE}FILESMODE_divert.py=0555 ${PACKAGE}FILESMODE_sender.py=0555 +${PACKAGE}FILESMODE_net_receiver.py=0555 .include <bsd.test.mk> Added: head/tests/sys/common/net_receiver.py ============================================================================== --- /dev/null 00:00:00 1970 (empty, because file is newly added) +++ head/tests/sys/common/net_receiver.py Mon Mar 23 12:21:32 2020 (r359235) @@ -0,0 +1,116 @@ +#!/usr/bin/env python +# - +# SPDX-License-Identifier: BSD-2-Clause +# +# Copyright (c) 2020 Alexander V. Chernikov +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions +# are met: +# 1. Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# 2. Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in the +# documentation and/or other materials provided with the distribution. +# +# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND +# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE +# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +# SUCH DAMAGE. +# +# $FreeBSD$ +# + + +from functools import partial +import socket +import select +import argparse +import time + + +def parse_args(): + parser = argparse.ArgumentParser(description='divert socket tester') + parser.add_argument('--sip', type=str, default='', help='IP to listen on') + parser.add_argument('--family', type=str, help='inet/inet6') + parser.add_argument('--ports', type=str, help='packet ports 1,2,3') + parser.add_argument('--match_str', type=str, help='match string to use') + parser.add_argument('--count', type=int, default=1, + help='Number of messages to receive') + parser.add_argument('--test_name', type=str, required=True, + help='test name to run') + return parser.parse_args() + + +def test_listen_tcp(args): + if args.family == 'inet6': + fam = socket.AF_INET6 + else: + fam = socket.AF_INET + sockets = [] + ports = [int(port) for port in args.ports.split(',')] + for port in ports: + s = socket.socket(fam, socket.SOCK_STREAM) + s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + s.setblocking(0) + s.bind((args.sip, port)) + print('binding on {}:{}'.format(args.sip, port)) + s.listen(5) + sockets.append(s) + inputs = sockets + count = 0 + while count < args.count: + readable, writable, exceptional = select.select(inputs, [], inputs) + for s in readable: + (c, address) = s.accept() + print('C: {}'.format(address)) + data = c.recv(9000) + if args.match_str and args.match_str.encode('utf-8') != data: + raise Exception('Expected "{}" but got "{}"'.format( + args.match_str, data.decode('utf-8'))) + count += 1 + c.close() + + +def test_listen_udp(args): + if args.family == 'inet6': + fam = socket.AF_INET6 + else: + fam = socket.AF_INET + sockets = [] + ports = [int(port) for port in args.ports.split(',')] + for port in ports: + s = socket.socket(fam, socket.SOCK_DGRAM) + s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + s.setblocking(0) + s.bind((args.sip, port)) + print('binding on {}:{}'.format(args.sip, port)) + sockets.append(s) + inputs = sockets + count = 0 + while count < args.count: + readable, writable, exceptional = select.select(inputs, [], inputs) + for s in readable: + (data, address) = s.recvfrom(9000) + print('C: {}'.format(address)) + if args.match_str and args.match_str.encode('utf-8') != data: + raise Exception('Expected "{}" but got "{}"'.format( + args.match_str, data.decode('utf-8'))) + count += 1 + + +def main(): + args = parse_args() + test_ptr = globals()[args.test_name] + test_ptr(args) + + +if __name__ == '__main__': + main() Modified: head/tests/sys/common/vnet.subr ============================================================================== --- head/tests/sys/common/vnet.subr Mon Mar 23 12:17:13 2020 (r359234) +++ head/tests/sys/common/vnet.subr Mon Mar 23 12:21:32 2020 (r359235) @@ -13,12 +13,20 @@ vnet_mkepair() { ifname=$(ifconfig epair create) echo $ifname >> created_interfaces.lst + echo ${ifname%a}b >> created_interfaces.lst echo ${ifname%a} } vnet_mkbridge() { ifname=$(ifconfig bridge create) + echo $ifname >> created_interfaces.lst + echo ${ifname} +} + +vnet_mkloopback() +{ + ifname=$(ifconfig lo create) echo $ifname >> created_interfaces.lst echo ${ifname} } Modified: head/tests/sys/netinet/Makefile ============================================================================== --- head/tests/sys/netinet/Makefile Mon Mar 23 12:17:13 2020 (r359234) +++ head/tests/sys/netinet/Makefile Mon Mar 23 12:21:32 2020 (r359235) @@ -9,7 +9,7 @@ ATF_TESTS_C= ip_reass_test \ so_reuseport_lb_test \ socket_afinet -ATF_TESTS_SH= fibs_test redirect divert forward +ATF_TESTS_SH= fibs_test redirect divert forward output PROGS= udp_dontroute tcp_user_cookie Added: head/tests/sys/netinet/output.sh ============================================================================== --- /dev/null 00:00:00 1970 (empty, because file is newly added) +++ head/tests/sys/netinet/output.sh Mon Mar 23 12:21:32 2020 (r359235) @@ -0,0 +1,588 @@ +#!/usr/bin/env atf-sh +#- +# SPDX-License-Identifier: BSD-2-Clause +# +# Copyright (c) 2020 Alexander V. Chernikov +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions +# are met: +# 1. Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# 2. Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in the +# documentation and/or other materials provided with the distribution. +# +# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND +# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE +# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +# SUCH DAMAGE. +# +# $FreeBSD$ +# + +. $(atf_get_srcdir)/../common/vnet.subr + +atf_test_case "output_tcp_setup_success" "cleanup" +output_tcp_setup_success_head() +{ + + atf_set descr 'Test valid IPv4 TCP output' + atf_set require.user root +} + +output_tcp_setup_success_body() +{ + + vnet_init + + net_src="192.0.2." + net_dst="192.0.2." + ip_src="${net_src}1" + ip_dst="${net_dst}2" + plen=24 + text="testtesttst" + port=4242 + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v4t-output_tcp_setup_success" + + epair=$(vnet_mkepair) + + vnet_mkjail ${jname}a ${epair}a + jexec ${jname}a ifconfig ${epair}a up + jexec ${jname}a ifconfig ${epair}a inet ${ip_src}/${plen} + + vnet_mkjail ${jname}b ${epair}b + jexec ${jname}b ifconfig ${epair}b up + + jexec ${jname}b ifconfig ${epair}b inet ${ip_dst}/${plen} + + # run listener + args="--family inet --ports ${port} --match_str ${text}" + echo jexec ${jname}b ${script_name} ${args} + jexec ${jname}b ${script_name} --test_name "test_listen_tcp" ${args} & + cmd_pid=$! + + # wait for the script init + counter=0 + while [ `jexec ${jname}b sockstat -4qlp ${port} | wc -l` != "1" ]; do + sleep 0.01 + counter=$((counter+1)) + if [ ${counter} -ge 50 ]; then break; fi + done + if [ `jexec ${jname}b sockstat -4qlp ${port} | wc -l` != "1" ]; then + echo "App setup failed" + exit 1 + fi + + # run sender + echo -n "${text}" | jexec ${jname}a nc -N ${ip_dst} ${port} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "sender exit code $exit_code" ; fi + + wait ${cmd_pid} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "receiver exit code $exit_code" ; fi +} + +output_tcp_setup_success_cleanup() +{ + vnet_cleanup +} + + +atf_test_case "output_udp_setup_success" "cleanup" +output_udp_setup_success_head() +{ + + atf_set descr 'Test valid IPv4 UDP output' + atf_set require.user root +} + +output_udp_setup_success_body() +{ + + vnet_init + + net_src="192.0.2." + net_dst="192.0.2." + ip_src="${net_src}1" + ip_dst="${net_dst}2" + plen=24 + text="testtesttst" + port=4242 + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v4t-output_udp_setup_success" + + epair=$(vnet_mkepair) + + vnet_mkjail ${jname}a ${epair}a + jexec ${jname}a ifconfig ${epair}a up + jexec ${jname}a ifconfig ${epair}a inet ${ip_src}/${plen} + + vnet_mkjail ${jname}b ${epair}b + jexec ${jname}b ifconfig ${epair}b up + jexec ${jname}b ifconfig ${epair}b inet ${ip_dst}/${plen} + + # run listener + args="--family inet --ports ${port} --match_str ${text}" + echo jexec ${jname}b ${script_name} ${args} + jexec ${jname}b ${script_name} --test_name "test_listen_udp" ${args} & + cmd_pid=$! + + # wait for the script init + counter=0 + while [ `jexec ${jname}b sockstat -4qlp ${port} | wc -l` != "1" ]; do + sleep 0.1 + counterc=$((counter+1)) + if [ ${counter} -ge 50 ]; then break; fi + done + if [ `jexec ${jname}b sockstat -4qlp ${port} | wc -l` != "1" ]; then + echo "App setup failed" + exit 1 + fi + + # run sender + # TODO: switch from nc to some alternative to avoid 1-second delay + echo -n "${text}" | jexec ${jname}a nc -uNw1 ${ip_dst} ${port} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "sender exit code $exit_code" ; fi + + wait ${cmd_pid} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "receiver exit code $exit_code" ; fi +} + +output_udp_setup_success_cleanup() +{ + vnet_cleanup +} + +atf_test_case "output_raw_success" "cleanup" +output_raw_success_head() +{ + + atf_set descr 'Test valid IPv4 raw output' + atf_set require.user root +} + +output_raw_success_body() +{ + + vnet_init + + net_src="192.0.2." + net_dst="192.0.2." + ip_src="${net_src}1" + ip_dst="${net_dst}2" + plen=24 + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v4t-output_raw_success" + + epair=$(vnet_mkepair) + + vnet_mkjail ${jname}a ${epair}a + jexec ${jname}a ifconfig ${epair}a up + jexec ${jname}a ifconfig ${epair}a inet ${ip_src}/${plen} + + vnet_mkjail ${jname}b ${epair}b + jexec ${jname}b ifconfig ${epair}b up + + jexec ${jname}b ifconfig ${epair}b inet ${ip_dst}/${plen} + + atf_check -o match:'1 packets transmitted, 1 packets received' jexec ${jname}a ping -nc1 ${ip_dst} +} + +output_raw_success_cleanup() +{ + vnet_cleanup +} + +# Multipath tests are done the following way: +# epair0 +# jailA lo < > lo jailB +# epair1 +# jailA has 2 routes towards /24 prefix on jailB loopback, via 2 epairs +# jailB has 1 route towards /24 prefix on jailA loopback, via epair0 +# +# jailA initiates connections/sends packets towards IPs on jailB loopback. +# Script then compares amount of packets sent via epair0 and epair1 + +mpath_check() +{ + if [ "`sysctl -i -n net.route.multipath`" != 1 ]; then + atf_skip "This test requires ROUTE_MPATH enabled" + fi +} + +atf_test_case "output_tcp_flowid_mpath_success" "cleanup" +output_tcp_flowid_mpath_success_head() +{ + + atf_set descr 'Test valid IPv4 TCP output flowid generation' + atf_set require.user root +} + +output_tcp_flowid_mpath_success_body() +{ + vnet_init + mpath_check + + net_src="192.0.2." + net_dst="198.51.100." + ip_src="${net_src}1" + ip_dst="${net_dst}1" + plen=24 + text="testtesttst" + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v4t-output_tcp_flowid_mpath_success" + + epair0=$(vnet_mkepair) + epair1=$(vnet_mkepair) + lo_src=$(vnet_mkloopback) + lo_dst=$(vnet_mkloopback) + + vnet_mkjail ${jname}a ${epair0}a ${epair1}a ${lo_src} + # Setup transit IPv4 networks + jexec ${jname}a ifconfig ${epair0}a up + jexec ${jname}a ifconfig ${epair0}a inet 203.0.113.1/30 + jexec ${jname}a ifconfig ${epair1}a up + jexec ${jname}a ifconfig ${epair1}a inet 203.0.113.5/30 + jexec ${jname}a ifconfig ${lo_src} up + + vnet_mkjail ${jname}b ${epair0}b ${epair1}b ${lo_dst} + jexec ${jname}b ifconfig ${epair0}b up + jexec ${jname}b ifconfig ${epair0}b inet 203.0.113.2/30 + jexec ${jname}b ifconfig ${epair1}b up + jexec ${jname}b ifconfig ${epair1}b inet 203.0.113.6/30 + jexec ${jname}b ifconfig ${lo_dst} up + + # DST ips/ports to test + ips="4 29 48 53 55 61 71 80 84 87 90 91 119 131 137 153 154 158 162 169 169 171 176 187 197 228 233 235 236 237 245 251" + ports="53540 49743 43067 9131 16734 5150 14379 40292 20634 51302 3387 24387 9282 14275 42103 26881 42461 29520 45714 11096" + + jexec ${jname}a ifconfig ${lo_src} inet ${ip_src}/32 + + jexec ${jname}b ifconfig ${lo_dst} inet ${ip_dst}/32 + for i in ${ips}; do + jexec ${jname}b ifconfig ${lo_dst} alias ${net_dst}${i}/32 + done + + # Add routes + # A -> towards B via epair0a + jexec ${jname}a route add -4 -net ${net_dst}0/${plen} 203.0.113.2 + # A -> towards B via epair1a + jexec ${jname}a route add -4 -net ${net_dst}0/${plen} 203.0.113.6 + + # B towards A via epair0b + jexec ${jname}b route add -4 -net ${net_src}0/${plen} 203.0.113.1 + + # Base setup verification + atf_check -o match:'1 packets transmitted, 1 packets received' jexec ${jname}a ping -nc1 ${ip_dst} + + # run listener + num_ports=`echo ${ports} | wc -w` + num_ips=`echo ${ips} | wc -w` + count_examples=$((num_ports*num_ips)) + listener_ports=`echo ${ports} | tr ' ' '\n' | sort -n | tr '\n' ',' | sed -e 's?,$??'` + args="--family inet --ports ${listener_ports} --count ${count_examples} --match_str ${text}" + echo jexec ${jname}b ${script_name} ${args} + jexec ${jname}b ${script_name} --test_name "test_listen_tcp" ${args} & + cmd_pid=$! + + # wait for the app init + counter=0 + init=0 + while [ ${counter} -le 50 ]; do + _ports=`jexec ${jname}b sockstat -4ql | awk "\\\$3 == ${cmd_pid} {print \\\$6}"|awk -F: "{print \\\$2}" | sort -n | tr '\n' ','` + if [ "${_ports}" = "${listener_ports}," ]; then + init=1 + break; + fi + done + if [ ${init} -eq 0 ]; then + jexec ${jname}b sockstat -6ql | awk "\$3 == ${cmd_pid}" + echo "App setup failed" + exit 1 + fi + echo "App setup done" + + # run sender + for _ip in ${ips}; do + ip="${net_dst}${_ip}" + for port in ${ports}; do + echo -n "${text}" | jexec ${jname}a nc -nN ${ip} ${port} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "sender exit code $exit_code" ; fi + done + done + + wait ${cmd_pid} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "receiver exit code $exit_code" ; fi + + pkt_0=`jexec ${jname}a netstat -Wf link -I ${epair0}a | head | awk '$1!~/^Name/{print$8}'` + pkt_1=`jexec ${jname}a netstat -Wf link -I ${epair1}a | head | awk '$1!~/^Name/{print$8}'` + if [ ${pkt_0} -le 10 ]; then + echo "Balancing failure: 1: ${pkt_0} 2: ${pkt_1}" + exit 1 + fi + if [ ${pkt_1} -le 10 ]; then + echo "Balancing failure: 1: ${pkt_0} 2: ${pkt_1}" + exit 1 + fi + echo "TCP Balancing: 1: ${pkt_0} 2: ${pkt_1}" +} + +output_tcp_flowid_mpath_success_cleanup() +{ + vnet_cleanup +} + +atf_test_case "output_udp_flowid_mpath_success" "cleanup" +output_udp_flowid_mpath_success_head() +{ + + atf_set descr 'Test valid IPv4 UDP output flowid generation' + atf_set require.user root +} + +output_udp_flowid_mpath_success_body() +{ + + vnet_init + mpath_check + + # Note this test will spawn around ~100 nc processes + + net_src="192.0.2." + net_dst="198.51.100." + ip_src="${net_src}1" + ip_dst="${net_dst}1" + plen=24 + text="testtesttst" + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v4t-output_tcp_flowid_mpath_success" + + epair0=$(vnet_mkepair) + epair1=$(vnet_mkepair) + lo_src=$(vnet_mkloopback) + lo_dst=$(vnet_mkloopback) + + vnet_mkjail ${jname}a ${epair0}a ${epair1}a ${lo_src} + # Setup transit IPv4 networks + jexec ${jname}a ifconfig ${epair0}a up + jexec ${jname}a ifconfig ${epair0}a inet 203.0.113.1/30 + jexec ${jname}a ifconfig ${epair1}a up + jexec ${jname}a ifconfig ${epair1}a inet 203.0.113.5/30 + jexec ${jname}a ifconfig ${lo_src} up + + vnet_mkjail ${jname}b ${epair0}b ${epair1}b ${lo_dst} + jexec ${jname}b ifconfig ${epair0}b up + jexec ${jname}b ifconfig ${epair0}b inet 203.0.113.2/30 + jexec ${jname}b ifconfig ${epair1}b up + jexec ${jname}b ifconfig ${epair1}b inet 203.0.113.6/30 + jexec ${jname}b ifconfig ${lo_dst} up + + # DST ips/ports to test + ips="4 29 48 53 55 61 71 80 84 87 90 91 119 131 137 153 154 158 162 169 169 171 176 187 197 228 233 235 236 237 245 251" + ports="53540 49743 43067 9131 16734 5150 14379 40292 20634 51302 3387 24387 9282 14275 42103 26881 42461 29520 45714 11096" + + jexec ${jname}a ifconfig ${lo_src} inet ${ip_src}/32 + + jexec ${jname}b ifconfig ${lo_dst} inet ${ip_dst}/32 + for i in ${ips}; do + jexec ${jname}b ifconfig ${lo_dst} alias ${net_dst}${i}/32 + done + + # Add routes + # A -> towards B via epair0a + jexec ${jname}a route add -4 -net ${net_dst}0/${plen} 203.0.113.2 + # A -> towards B via epair1a + jexec ${jname}a route add -4 -net ${net_dst}0/${plen} 203.0.113.6 + + # B towards A via epair0b + jexec ${jname}b route add -4 -net ${net_src}0/${plen} 203.0.113.1 + + # Base setup verification + atf_check -o match:'1 packets transmitted, 1 packets received' jexec ${jname}a ping -nc1 ${ip_dst} + + # run listener + num_ports=`echo ${ports} | wc -w` + num_ips=`echo ${ips} | wc -w` + count_examples=$((num_ports*num_ips)) + listener_ports=`echo ${ports} | tr ' ' '\n' | sort -n | tr '\n' ',' | sed -e 's?,$??'` + args="--family inet --ports ${listener_ports} --count ${count_examples} --match_str ${text}" + echo jexec ${jname}b ${script_name} ${args} + jexec ${jname}b ${script_name} --test_name "test_listen_udp" ${args} & + cmd_pid=$! + + # wait for the app init + counter=0 + init=0 + while [ ${counter} -le 50 ]; do + _ports=`jexec ${jname}b sockstat -4ql | awk "\\\$3 == ${cmd_pid} {print \\\$6}"|awk -F: "{print \\\$2}" | sort -n | tr '\n' ','` + if [ "${_ports}" = "${listener_ports}," ]; then + init=1 + break; + fi + done + if [ ${init} -eq 0 ]; then + jexec ${jname}b sockstat -4ql | awk "\$3 == ${cmd_pid}" + echo "App setup failed" + exit 1 + fi + echo "App setup done" + + # run sender + for _ip in ${ips}; do + ip="${net_dst}${_ip}" + for port in ${ports}; do + # XXX: switch to something that allows immediate exit + echo -n "${text}" | jexec ${jname}a nc -nuNw1 ${ip} ${port} & + sleep 0.01 + done + done + + wait ${cmd_pid} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "receiver exit code $exit_code" ; fi + + pkt_0=`jexec ${jname}a netstat -Wf link -I ${epair0}a | head | awk '$1!~/^Name/{print$8}'` + pkt_1=`jexec ${jname}a netstat -Wf link -I ${epair1}a | head | awk '$1!~/^Name/{print$8}'` + if [ ${pkt_0} -le 10 ]; then + echo "Balancing failure: 1: ${pkt_0} 2: ${pkt_1}" + exit 1 + fi + if [ ${pkt_1} -le 10 ]; then + echo "Balancing failure: 1: ${pkt_0} 2: ${pkt_1}" + exit 1 + fi + echo "UDP BALANCING: 1: ${pkt_0} 2: ${pkt_1}" +} + +output_udp_flowid_mpath_success_cleanup() +{ + vnet_cleanup +} + +atf_test_case "output_raw_flowid_mpath_success" "cleanup" +output_raw_flowid_mpath_success_head() +{ + + atf_set descr 'Test valid IPv4 raw output flowid generation' + atf_set require.user root +} + +output_raw_flowid_mpath_success_body() +{ + + vnet_init + mpath_check + + net_src="192.0.2." + net_dst="198.51.100." + ip_src="${net_src}1" + ip_dst="${net_dst}1" + plen=24 + text="testtesttst" + + jname="v4t-output_tcp_flowid_mpath_success" + + epair0=$(vnet_mkepair) + epair1=$(vnet_mkepair) + lo_src=$(vnet_mkloopback) + lo_dst=$(vnet_mkloopback) + + vnet_mkjail ${jname}a ${epair0}a ${epair1}a ${lo_src} + # Setup transit IPv4 networks + jexec ${jname}a ifconfig ${epair0}a up + jexec ${jname}a ifconfig ${epair0}a inet 203.0.113.1/30 + jexec ${jname}a ifconfig ${epair1}a up + jexec ${jname}a ifconfig ${epair1}a inet 203.0.113.5/30 + jexec ${jname}a ifconfig ${lo_src} up + + vnet_mkjail ${jname}b ${epair0}b ${epair1}b ${lo_dst} + jexec ${jname}b ifconfig ${epair0}b up + jexec ${jname}b ifconfig ${epair0}b inet 203.0.113.2/30 + jexec ${jname}b ifconfig ${epair1}b up + jexec ${jname}b ifconfig ${epair1}b inet 203.0.113.6/30 + jexec ${jname}b ifconfig ${lo_dst} up + + # DST ips/ports to test + ips="4 29 48 53 55 61 71 80 84 87 90 91 119 131 137 153 154 158 162 169 169 171 176 187 197 228 233 235 236 237 245 251" + + jexec ${jname}a ifconfig ${lo_src} inet ${ip_src}/32 + + jexec ${jname}b ifconfig ${lo_dst} inet ${ip_dst}/32 + for i in ${ips}; do + jexec ${jname}b ifconfig ${lo_dst} alias ${net_dst}${i}/32 + done + + # Add routes + # A -> towards B via epair0a + jexec ${jname}a route add -4 -net ${net_dst}0/${plen} 203.0.113.2 + # A -> towards B via epair1a + jexec ${jname}a route add -4 -net ${net_dst}0/${plen} 203.0.113.6 + + # B towards A via epair0b + jexec ${jname}b route add -4 -net ${net_src}0/${plen} 203.0.113.1 + + # Base setup verification + atf_check -o match:'1 packets transmitted, 1 packets received' jexec ${jname}a ping -nc1 ${ip_dst} + + # run sender + valid_message='1 packets transmitted, 1 packets received' + for _ip in ${ips}; do + ip="${net_dst}${_ip}" + atf_check -o match:"${valid_message}" jexec ${jname}a ping -nc1 ${ip} + done + + pkt_0=`jexec ${jname}a netstat -Wf link -I ${epair0}a | head | awk '$1!~/^Name/{print$8}'` + pkt_1=`jexec ${jname}a netstat -Wf link -I ${epair1}a | head | awk '$1!~/^Name/{print$8}'` + + jexec ${jname}a netstat -bWf link -I ${epair0}a + jexec ${jname}a netstat -bWf link -I ${epair1}a + if [ ${pkt_0} -le 10 ]; then + echo "Balancing failure: 1: ${pkt_0} 2: ${pkt_1}" + exit 1 + fi + if [ ${pkt_1} -le 10 ]; then + echo "Balancing failure: 1: ${pkt_0} 2: ${pkt_1}" + exit 1 + fi + echo "RAW BALANCING: 1: ${pkt_0} 2: ${pkt_1}" +} + +output_raw_flowid_mpath_success_cleanup() +{ + vnet_cleanup +} + +atf_init_test_cases() +{ + atf_add_test_case "output_tcp_setup_success" + atf_add_test_case "output_udp_setup_success" + atf_add_test_case "output_raw_success" + atf_add_test_case "output_tcp_flowid_mpath_success" + atf_add_test_case "output_udp_flowid_mpath_success" + atf_add_test_case "output_raw_flowid_mpath_success" +} + Modified: head/tests/sys/netinet6/Makefile ============================================================================== --- head/tests/sys/netinet6/Makefile Mon Mar 23 12:17:13 2020 (r359234) +++ head/tests/sys/netinet6/Makefile Mon Mar 23 12:21:32 2020 (r359235) @@ -11,7 +11,8 @@ ATF_TESTS_SH= \ scapyi386 \ redirect \ divert \ - forward6 + forward6 \ + output6 ${PACKAGE}FILES+= exthdr.py ${PACKAGE}FILES+= mld.py Added: head/tests/sys/netinet6/output6.sh ============================================================================== --- /dev/null 00:00:00 1970 (empty, because file is newly added) +++ head/tests/sys/netinet6/output6.sh Mon Mar 23 12:21:32 2020 (r359235) @@ -0,0 +1,658 @@ +#!/usr/bin/env atf-sh +#- +# SPDX-License-Identifier: BSD-2-Clause +# +# Copyright (c) 2020 Alexander V. Chernikov +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions +# are met: +# 1. Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# 2. Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in the +# documentation and/or other materials provided with the distribution. +# +# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND +# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE +# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +# SUCH DAMAGE. +# +# $FreeBSD$ +# + +. $(atf_get_srcdir)/../common/vnet.subr + +atf_test_case "output6_tcp_setup_success" "cleanup" +output6_tcp_setup_success_head() +{ + + atf_set descr 'Test valid IPv6 TCP output' + atf_set require.user root +} + +output6_tcp_setup_success_body() +{ + + vnet_init + + net_src="2001:db8:0:0:1::" + net_dst="2001:db8:0:0:1::" + ip_src="${net_src}1" + ip_dst=${net_dst}4242 + plen=64 + text="testtesttst" + port=4242 + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v6t-output6_tcp_setup_success" + + epair=$(vnet_mkepair) + + vnet_mkjail ${jname}a ${epair}a + jexec ${jname}a ifconfig ${epair}a up + jexec ${jname}a ifconfig ${epair}a inet6 ${ip_src}/${plen} + + vnet_mkjail ${jname}b ${epair}b + jexec ${jname}b ifconfig ${epair}b up + + jexec ${jname}b ifconfig ${epair}b inet6 ${ip_dst}/${plen} + + # wait for DAD to complete + while [ `jexec ${jname}b ifconfig ${epair}b inet6 | grep -c tentative` != "0" ]; do + sleep 0.1 + done + while [ `jexec ${jname}a ifconfig ${epair}a inet6 | grep -c tentative` != "0" ]; do + sleep 0.1 + done + + # run listener + args="--family inet6 --ports ${port} --match_str ${text}" + echo jexec ${jname}b ${script_name} ${args} + jexec ${jname}b ${script_name} --test_name "test_listen_tcp" ${args} & + cmd_pid=$! + + # wait for the script init + counter=0 + while [ `jexec ${jname}b sockstat -6qlp ${port} | wc -l` != "1" ]; do + sleep 0.01 + counter=$((counter+1)) + if [ ${counter} -ge 50 ]; then break; fi + done + if [ `jexec ${jname}b sockstat -6qlp ${port} | wc -l` != "1" ]; then + echo "App setup failed" + exit 1 + fi + + # run sender + echo -n "${text}" | jexec ${jname}a nc -N ${ip_dst} ${port} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "sender exit code $exit_code" ; fi + + wait ${cmd_pid} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "receiver exit code $exit_code" ; fi +} + +output6_tcp_setup_success_cleanup() +{ + vnet_cleanup +} + + +atf_test_case "output6_udp_setup_success" "cleanup" +output6_udp_setup_success_head() +{ + + atf_set descr 'Test valid IPv6 UDP output' + atf_set require.user root +} + +output6_udp_setup_success_body() +{ + + vnet_init + + net_src="2001:db8:0:0:1::" + net_dst="2001:db8:0:0:1::" + ip_src="${net_src}1" + ip_dst=${net_dst}4242 + plen=64 + text="testtesttst" + port=4242 + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v6t-output6_udp_setup_success" + + epair=$(vnet_mkepair) + + vnet_mkjail ${jname}a ${epair}a + jexec ${jname}a ifconfig ${epair}a up + jexec ${jname}a ifconfig ${epair}a inet6 ${ip_src}/${plen} + + vnet_mkjail ${jname}b ${epair}b + jexec ${jname}b ifconfig ${epair}b up + jexec ${jname}b ifconfig ${epair}b inet6 ${ip_dst}/${plen} + + # wait for DAD to complete + while [ `jexec ${jname}b ifconfig ${epair}b inet6 | grep -c tentative` != "0" ]; do + sleep 0.1 + done + while [ `jexec ${jname}a ifconfig ${epair}a inet6 | grep -c tentative` != "0" ]; do + sleep 0.1 + done + + # run listener + args="--family inet6 --ports ${port} --match_str ${text}" + echo jexec ${jname}b ${script_name} ${args} + jexec ${jname}b ${script_name} --test_name "test_listen_udp" ${args} & + cmd_pid=$! + + # wait for the script init + counter=0 + while [ `jexec ${jname}b sockstat -6qlp ${port} | wc -l` != "1" ]; do + sleep 0.1 + counterc=$((counter+1)) + if [ ${counter} -ge 50 ]; then break; fi + done + if [ `jexec ${jname}b sockstat -6qlp ${port} | wc -l` != "1" ]; then + echo "App setup failed" + exit 1 + fi + + # run sender + # TODO: switch from nc to some alternative to avoid 1-second delay + echo -n "${text}" | jexec ${jname}a nc -uNw1 ${ip_dst} ${port} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "sender exit code $exit_code" ; fi + + wait ${cmd_pid} + exit_code=$? + if [ $exit_code -ne 0 ]; then atf_fail "receiver exit code $exit_code" ; fi +} + +output6_udp_setup_success_cleanup() +{ + vnet_cleanup +} + +atf_test_case "output6_raw_success" "cleanup" +output6_raw_success_head() +{ + + atf_set descr 'Test valid IPv6 raw output' + atf_set require.user root +} + +output6_raw_success_body() +{ + + vnet_init + + net_src="2001:db8:0:0:1::" + net_dst="2001:db8:0:0:1::" + ip_src="${net_src}1" + ip_dst=${net_dst}4242 + plen=64 + + script_name=`dirname $0`/../common/net_receiver.py + script_name=`realpath ${script_name}` + jname="v6t-output6_raw_success" + + epair=$(vnet_mkepair) + + vnet_mkjail ${jname}a ${epair}a + jexec ${jname}a ifconfig ${epair}a up + jexec ${jname}a ifconfig ${epair}a inet6 ${ip_src}/${plen} + + vnet_mkjail ${jname}b ${epair}b + jexec ${jname}b ifconfig ${epair}b up + + jexec ${jname}b ifconfig ${epair}b inet6 ${ip_dst}/${plen} + + # wait for DAD to complete + while [ `jexec ${jname}b ifconfig ${epair}b inet6 | grep -c tentative` != "0" ]; do + sleep 0.1 + done + while [ `jexec ${jname}a ifconfig ${epair}a inet6 | grep -c tentative` != "0" ]; do + sleep 0.1 + done *** DIFF OUTPUT TRUNCATED AT 1000 LINES *** _______________________________________________ svn-src-all@freebsd.org mailing list https://lists.freebsd.org/mailman/listinfo/svn-src-all To unsubscribe, send any mail to "svn-src-all-unsubscr...@freebsd.org"