xref: /freebsd/tests/sys/netinet/divert.sh (revision 2f513db72b034fd5ef7f080b11be5c711c15186a)
1#!/usr/bin/env atf-sh
2#-
3# SPDX-License-Identifier: BSD-2-Clause
4#
5# Copyright (c) 2020 Alexander V. Chernikov
6#
7# Redistribution and use in source and binary forms, with or without
8# modification, are permitted provided that the following conditions
9# are met:
10# 1. Redistributions of source code must retain the above copyright
11#    notice, this list of conditions and the following disclaimer.
12# 2. Redistributions in binary form must reproduce the above copyright
13#    notice, this list of conditions and the following disclaimer in the
14#    documentation and/or other materials provided with the distribution.
15#
16# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26# SUCH DAMAGE.
27#
28# $FreeBSD$
29#
30
31. $(atf_get_srcdir)/../common/vnet.subr
32
33load_divert_module() {
34	kldstat -q -n ipdivert
35	if [ $? -ne  0 ]; then
36		atf_skip "ipdivert module is not loaded"
37	fi
38}
39
40atf_test_case "ipdivert_ip_output_remote_success" "cleanup"
41ipdivert_ip_output_remote_success_head() {
42
43	atf_set descr 'Test diverting IPv4 packet to remote destination'
44	atf_set require.user root
45	atf_set require.progs scapy
46}
47
48ipdivert_ip_output_remote_success_body() {
49
50	ids=65530
51	id=`printf "%x" ${ids}`
52	if [ $$ -gt 65535 ]; then
53		xl=`printf "%x" $(($$ - 65535))`
54		yl="1"
55	else
56		xl=`printf "%x" $$`
57		yl=""
58	fi
59
60	vnet_init
61	load_divert_module
62
63	ip4a="192.0.2.5"
64	ip4b="192.0.2.6"
65
66	script_name="../common/divert.py"
67
68	epair=$(vnet_mkepair)
69	ifconfig ${epair}a up
70	ifconfig ${epair}a inet ${ip4a}/30
71
72	jname="v4t-${id}-${yl}-${xl}"
73	vnet_mkjail ${jname} ${epair}b
74	jexec ${jname} ifconfig ${epair}b up
75	jexec ${jname} ifconfig ${epair}b inet ${ip4b}/30
76
77	atf_check -s exit:0 $(atf_get_srcdir)/${script_name} \
78		--dip ${ip4b} --test_name ipdivert_ip_output_remote_success
79
80	count=`jexec ${jname} netstat -s -p icmp  | grep 'Input histogram:' -A8 | grep -c 'echo: '`
81	# Verify redirect got installed
82	atf_check_equal "1" "${count}"
83}
84
85ipdivert_ip_output_remote_success_cleanup() {
86
87	vnet_cleanup
88}
89
90atf_test_case "ipdivert_ip_input_local_success" "cleanup"
91ipdivert_ip_input_local_success_head() {
92
93	atf_set descr 'Test diverting IPv4 packet to remote destination'
94	atf_set require.user root
95	atf_set require.progs scapy
96}
97
98ipdivert_ip_input_local_success_body() {
99
100	ids=65529
101	id=`printf "%x" ${ids}`
102	if [ $$ -gt 65535 ]; then
103		xl=`printf "%x" $(($$ - 65535))`
104		yl="1"
105	else
106		xl=`printf "%x" $$`
107		yl=""
108	fi
109
110	vnet_init
111	load_divert_module
112
113	ip4a="192.0.2.5"
114	ip4b="192.0.2.6"
115
116	script_name="../common/divert.py"
117
118	epair=$(vnet_mkepair)
119	ifconfig ${epair}a up
120	ifconfig ${epair}a inet ${ip4a}/30
121
122	jname="v4t-${id}-${yl}-${xl}"
123	vnet_mkjail ${jname} ${epair}b
124	jexec ${jname} ifconfig ${epair}b up
125	jexec ${jname} ifconfig ${epair}b inet ${ip4b}/30
126
127	atf_check -s exit:0 jexec ${jname} $(atf_get_srcdir)/${script_name} \
128		--dip ${ip4b} --test_name ipdivert_ip_input_local_success
129
130	count=`jexec ${jname} netstat -s -p icmp  | grep 'Input histogram:' -A8 | grep -c 'echo: '`
131	# Verify redirect got installed
132	atf_check_equal "1" "${count}"
133}
134
135ipdivert_ip_input_local_success_cleanup() {
136
137	vnet_cleanup
138}
139
140atf_init_test_cases()
141{
142
143	atf_add_test_case "ipdivert_ip_output_remote_success"
144	atf_add_test_case "ipdivert_ip_input_local_success"
145}
146
147# end
148
149