tests/sys/netinet/divert.sh

   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
  33 load_divert_module() {
  34         kldstat -q -n ipdivert
  35         if [ $? -ne  0 ]; then
  36                 atf_skip "ipdivert module is not loaded"
  37         fi
  38 }
  39
  40 atf_test_case "ipdivert_ip_output_remote_success" "cleanup"
  41 ipdivert_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
  48 ipdivert_ip_output_remote_success_body() {
  49
  50         if [ "$(atf_config_get ci false)" = "true" ] && \
  51                 [ "$(uname -p)" = "i386" ]; then
  52                 atf_skip "https://bugs.freebsd.org/245764"
  53         fi
  54
  55         ids=65530
  56         id=`printf "%x" ${ids}`
  57         if [ $$ -gt 65535 ]; then
  58                 xl=`printf "%x" $(($$ - 65535))`
  59                 yl="1"
  60         else
  61                 xl=`printf "%x" $$`
  62                 yl=""
  63         fi
  64
  65         vnet_init
  66         load_divert_module
  67
  68         ip4a="192.0.2.5"
  69         ip4b="192.0.2.6"
  70
  71         script_name="../common/divert.py"
  72
  73         epair=$(vnet_mkepair)
  74         ifconfig ${epair}a up
  75         ifconfig ${epair}a inet ${ip4a}/30
  76
  77         jname="v4t-${id}-${yl}-${xl}"
  78         vnet_mkjail ${jname} ${epair}b
  79         jexec ${jname} ifconfig ${epair}b up
  80         jexec ${jname} ifconfig ${epair}b inet ${ip4b}/30
  81
  82         atf_check -s exit:0 $(atf_get_srcdir)/${script_name} \
  83                 --dip ${ip4b} --test_name ipdivert_ip_output_remote_success
  84
  85         count=`jexec ${jname} netstat -s -p icmp  | grep 'Input histogram:' -A8 | grep -c 'echo: '`
  86         # Verify redirect got installed
  87         atf_check_equal "1" "${count}"
  88 }
  89
  90 ipdivert_ip_output_remote_success_cleanup() {
  91
  92         vnet_cleanup
  93 }
  94
  95 atf_test_case "ipdivert_ip_input_local_success" "cleanup"
  96 ipdivert_ip_input_local_success_head() {
  97
  98         atf_set descr 'Test diverting IPv4 packet to remote destination'
  99         atf_set require.user root
 100         atf_set require.progs scapy
 101 }
 102
 103 ipdivert_ip_input_local_success_body() {
 104
 105         if [ "$(atf_config_get ci false)" = "true" ] && \
 106                 [ "$(uname -p)" = "i386" ]; then
 107                 atf_skip "https://bugs.freebsd.org/245764"
 108         fi
 109
 110         ids=65529
 111         id=`printf "%x" ${ids}`
 112         if [ $$ -gt 65535 ]; then
 113                 xl=`printf "%x" $(($$ - 65535))`
 114                 yl="1"
 115         else
 116                 xl=`printf "%x" $$`
 117                 yl=""
 118         fi
 119
 120         vnet_init
 121         load_divert_module
 122
 123         ip4a="192.0.2.5"
 124         ip4b="192.0.2.6"
 125
 126         script_name="../common/divert.py"
 127
 128         epair=$(vnet_mkepair)
 129         ifconfig ${epair}a up
 130         ifconfig ${epair}a inet ${ip4a}/30
 131
 132         jname="v4t-${id}-${yl}-${xl}"
 133         vnet_mkjail ${jname} ${epair}b
 134         jexec ${jname} ifconfig ${epair}b up
 135         jexec ${jname} ifconfig ${epair}b inet ${ip4b}/30
 136
 137         atf_check -s exit:0 jexec ${jname} $(atf_get_srcdir)/${script_name} \
 138                 --dip ${ip4b} --test_name ipdivert_ip_input_local_success
 139
 140         count=`jexec ${jname} netstat -s -p icmp  | grep 'Input histogram:' -A8 | grep -c 'echo: '`
 141         # Verify redirect got installed
 142         atf_check_equal "1" "${count}"
 143 }
 144
 145 ipdivert_ip_input_local_success_cleanup() {
 146
 147         vnet_cleanup
 148 }
 149
 150 atf_init_test_cases()
 151 {
 152
 153         atf_add_test_case "ipdivert_ip_output_remote_success"
 154         atf_add_test_case "ipdivert_ip_input_local_success"
 155 }
 156
 157 # end
 158