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