tests/sys/netpfil/pf/pfsync.sh

   1 # $FreeBSD$
   2 #
   3 # SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   4 #
   5 # Copyright (c) 2018 Orange Business Services
   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 . $(atf_get_srcdir)/utils.subr
  29
  30 atf_test_case "basic" "cleanup"
  31 basic_head()
  32 {
  33         atf_set descr 'Basic pfsync test'
  34         atf_set require.user root
  35 }
  36
  37 basic_body()
  38 {
  39         common_body
  40 }
  41
  42 common_body()
  43 {
  44         defer=$1
  45         pfsynct_init
  46
  47         epair_sync=$(vnet_mkepair)
  48         epair_one=$(vnet_mkepair)
  49         epair_two=$(vnet_mkepair)
  50
  51         vnet_mkjail one ${epair_one}a ${epair_sync}a
  52         vnet_mkjail two ${epair_two}a ${epair_sync}b
  53
  54         # pfsync interface
  55         jexec one ifconfig ${epair_sync}a 192.0.2.1/24 up
  56         jexec one ifconfig ${epair_one}a 198.51.100.1/24 up
  57         jexec one ifconfig pfsync0 \
  58                 syncdev ${epair_sync}a \
  59                 maxupd 1 \
  60                 $defer \
  61                 up
  62         jexec two ifconfig ${epair_two}a 198.51.100.2/24 up
  63         jexec two ifconfig ${epair_sync}b 192.0.2.2/24 up
  64         jexec two ifconfig pfsync0 \
  65                 syncdev ${epair_sync}b \
  66                 maxupd 1 \
  67                 $defer \
  68                 up
  69
  70         # Enable pf!
  71         jexec one pfctl -e
  72         pft_set_rules one \
  73                 "set skip on ${epair_sync}a" \
  74                 "pass keep state"
  75         jexec two pfctl -e
  76         pft_set_rules two \
  77                 "set skip on ${epair_sync}b" \
  78                 "pass keep state"
  79
  80         ifconfig ${epair_one}b 198.51.100.254/24 up
  81
  82         ping -c 1 -S 198.51.100.254 198.51.100.1
  83
  84         # Give pfsync time to do its thing
  85         sleep 2
  86
  87         if ! jexec two pfctl -s states | grep icmp | grep 198.51.100.1 | \
  88             grep 198.51.100.2 ; then
  89                 atf_fail "state not found on synced host"
  90         fi
  91 }
  92
  93 basic_cleanup()
  94 {
  95         pfsynct_cleanup
  96 }
  97
  98 atf_test_case "defer" "cleanup"
  99 defer_head()
 100 {
 101         atf_set descr 'Defer mode pfsync test'
 102         atf_set require.user root
 103 }
 104
 105 defer_body()
 106 {
 107         common_body defer
 108 }
 109
 110 defer_cleanup()
 111 {
 112         pfsynct_cleanup
 113 }
 114
 115 atf_test_case "bulk" "cleanup"
 116 bulk_head()
 117 {
 118         atf_set descr 'Test bulk updates'
 119         atf_set require.user root
 120 }
 121
 122 bulk_body()
 123 {
 124         pfsynct_init
 125
 126         epair_sync=$(vnet_mkepair)
 127         epair_one=$(vnet_mkepair)
 128         epair_two=$(vnet_mkepair)
 129
 130         vnet_mkjail one ${epair_one}a ${epair_sync}a
 131         vnet_mkjail two ${epair_two}a ${epair_sync}b
 132
 133         # pfsync interface
 134         jexec one ifconfig ${epair_sync}a 192.0.2.1/24 up
 135         jexec one ifconfig ${epair_one}a 198.51.100.1/24 up
 136         jexec one ifconfig pfsync0 \
 137                 syncdev ${epair_sync}a \
 138                 maxupd 1\
 139                 up
 140         jexec two ifconfig ${epair_two}a 198.51.100.2/24 up
 141         jexec two ifconfig ${epair_sync}b 192.0.2.2/24 up
 142
 143         # Enable pf
 144         jexec one pfctl -e
 145         pft_set_rules one \
 146                 "set skip on ${epair_sync}a" \
 147                 "pass keep state"
 148         jexec two pfctl -e
 149         pft_set_rules two \
 150                 "set skip on ${epair_sync}b" \
 151                 "pass keep state"
 152
 153         ifconfig ${epair_one}b 198.51.100.254/24 up
 154
 155         # Create state prior to setting up pfsync
 156         ping -c 1 -S 198.51.100.254 198.51.100.1
 157
 158         # Wait before setting up pfsync on two, so we don't accidentally catch
 159         # the update anyway.
 160         sleep 1
 161
 162         # Now set up pfsync in jail two
 163         jexec two ifconfig pfsync0 \
 164                 syncdev ${epair_sync}b \
 165                 up
 166
 167         # Give pfsync time to do its thing
 168         sleep 2
 169
 170         jexec two pfctl -s states
 171         if ! jexec two pfctl -s states | grep icmp | grep 198.51.100.1 | \
 172             grep 198.51.100.2 ; then
 173                 atf_fail "state not found on synced host"
 174         fi
 175 }
 176
 177 bulk_cleanup()
 178 {
 179         pfsynct_cleanup
 180 }
 181
 182 atf_init_test_cases()
 183 {
 184         atf_add_test_case "basic"
 185         atf_add_test_case "defer"
 186         atf_add_test_case "bulk"
 187 }