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 33atf_test_case "valid_redirect" "cleanup" 34valid_redirect_head() { 35 36 atf_set descr 'Test valid IPv4 redirect' 37 atf_set require.user root 38 atf_set require.progs scapy 39} 40 41valid_redirect_body() { 42 43 ids=65533 44 id=`printf "%x" ${ids}` 45 if [ $$ -gt 65535 ]; then 46 xl=`printf "%x" $(($$ - 65535))` 47 yl="1" 48 else 49 xl=`printf "%x" $$` 50 yl="" 51 fi 52 53 vnet_init 54 55 ip4a="192.0.2.1" 56 ip4b="192.0.2.2" 57 58 net4="198.51.100.0/24" 59 dst_addr4="198.51.100.42" 60 61 # remote_rtr 62 remote_rtr_ip="192.0.2.3" 63 remote_rtr_mac="00:00:5E:00:53:42" 64 65 new_rtr_ip="192.0.2.4" 66 67 script_name="redirect.py" 68 69 epair=$(vnet_mkepair) 70 ifconfig ${epair}a up 71 ifconfig ${epair}a inet ${ip4a}/24 72 73 jname="v4t-${id}-${yl}-${xl}" 74 vnet_mkjail ${jname} ${epair}b 75 jexec ${jname} ifconfig ${epair}b up 76 jexec ${jname} ifconfig ${epair}b inet ${ip4b}/24 77 78 # Setup static entry for the remote router 79 jexec ${jname} arp -s ${remote_rtr_ip} ${remote_rtr_mac} 80 # setup prefix reachable via router 81 jexec ${jname} route add -4 -net ${net4} ${remote_rtr_ip} 82 83 local_ip=${ip4b} 84 local_mac=`jexec ${jname} ifconfig ${epair}b ether | awk '$1~/ether/{print$2}'` 85 86 # echo "LOCAL: ${local_ip} ${local_mac}" 87 # echo "REMOTE: ${remote_rtr_ip} ${remote_rtr_mac}" 88 89 atf_check -s exit:0 $(atf_get_srcdir)/${script_name} \ 90 --smac ${remote_rtr_mac} --dmac ${local_mac} \ 91 --sip ${remote_rtr_ip} --dip ${local_ip} \ 92 --route ${dst_addr4} --gw ${new_rtr_ip} \ 93 --iface ${epair}a 94 95 count=`jexec ${jname} route -n get -4 ${dst_addr4} | grep destination | grep -c ${dst_addr4}` 96 # Verify redirect got installed 97 atf_check_equal "1" "${count}" 98} 99 100valid_redirect_cleanup() { 101 102 vnet_cleanup 103} 104 105atf_init_test_cases() 106{ 107 108 atf_add_test_case "valid_redirect" 109} 110 111# end 112 113