1# $FreeBSD$ 2# 3# SPDX-License-Identifier: BSD-2-Clause-FreeBSD 4# 5# Copyright (c) 2018 Kristof Provost <kp@FreeBSD.org> 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 30atf_test_case "exhaust" "cleanup" 31exhaust_head() 32{ 33 atf_set descr 'Test exhausting the NAT pool' 34 atf_set require.user root 35} 36 37exhaust_body() 38{ 39 pft_init 40 41 epair_nat=$(vnet_mkepair) 42 epair_echo=$(vnet_mkepair) 43 44 vnet_mkjail nat ${epair_nat}b ${epair_echo}a 45 vnet_mkjail echo ${epair_echo}b 46 47 ifconfig ${epair_nat}a 192.0.2.2/24 up 48 route add -net 198.51.100.0/24 192.0.2.1 49 50 jexec nat ifconfig ${epair_nat}b 192.0.2.1/24 up 51 jexec nat ifconfig ${epair_echo}a 198.51.100.1/24 up 52 jexec nat sysctl net.inet.ip.forwarding=1 53 54 jexec echo ifconfig ${epair_echo}b 198.51.100.2/24 up 55 jexec echo /usr/sbin/inetd -p inetd-echo.pid $(atf_get_srcdir)/echo_inetd.conf 56 57 # Enable pf! 58 jexec nat pfctl -e 59 pft_set_rules nat \ 60 "nat pass on ${epair_echo}a inet from 192.0.2.0/24 to any -> (${epair_echo}a) port 30000:30001 sticky-address" 61 62 # Sanity check 63 atf_check -s exit:0 -o ignore ping -c 3 198.51.100.2 64 65 echo "foo" | nc -N 198.51.100.2 7 66 echo "foo" | nc -N 198.51.100.2 7 67 68 # This one will fail, but that's expected 69 echo "foo" | nc -N 198.51.100.2 7 & 70 71 sleep 1 72 73 # If the kernel is stuck in pf_get_sport() this will not succeed either. 74 timeout 2 jexec nat pfctl -sa 75 if [ $? -eq 124 ]; then 76 # Timed out 77 atf_fail "pfctl timeout" 78 fi 79} 80 81exhaust_cleanup() 82{ 83 rm -f inetd-echo.pid 84 pft_cleanup 85} 86 87atf_test_case "nested_anchor" "cleanup" 88nested_anchor_head() 89{ 90 atf_set descr 'Test setting and retrieving nested nat anchors' 91 atf_set require.user root 92} 93 94nested_anchor_body() 95{ 96 pft_init 97 98 epair=$(vnet_mkepair) 99 100 vnet_mkjail nat ${epair}a 101 102 pft_set_rules nat \ 103 "nat-anchor \"foo\"" 104 105 echo "nat-anchor \"bar\"" | jexec nat pfctl -g -a foo -f - 106 echo "nat on ${epair}a from any to any -> (${epair}a)" | jexec nat pfctl -g -a "foo/bar" -f - 107 108 atf_check -s exit:0 -o inline:"nat-anchor \"foo\" all { 109 nat-anchor \"bar\" all { 110 nat on ${epair}a all -> (${epair}a) round-robin 111 } 112} 113" jexec nat pfctl -sn -a "*" 114 115} 116 117nested_anchor_cleanup() 118{ 119 pft_cleanup 120} 121 122atf_init_test_cases() 123{ 124 atf_add_test_case "exhaust" 125 atf_add_test_case "nested_anchor" 126} 127