1# $FreeBSD$ 2# Utility functions 3## 4# SPDX-License-Identifier: BSD-2-Clause-FreeBSD 5# 6# Copyright (c) 2017 Kristof Provost <kp@FreeBSD.org> 7# 8# Redistribution and use in source and binary forms, with or without 9# modification, are permitted provided that the following conditions 10# are met: 11# 1. Redistributions of source code must retain the above copyright 12# notice, this list of conditions and the following disclaimer. 13# 2. Redistributions in binary form must reproduce the above copyright 14# notice, this list of conditions and the following disclaimer in the 15# documentation and/or other materials provided with the distribution. 16# 17# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 21# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27# SUCH DAMAGE. 28 29. $(atf_get_srcdir)/../../common/vnet.subr 30 31pft_init() 32{ 33 vnet_init 34 35 if [ ! -c /dev/pf ]; then 36 atf_skip "This test requires pf" 37 fi 38} 39 40pfsynct_init() 41{ 42 pft_init 43 44 if ! kldstat -q -m pfsync; then 45 atf_skip "This test requires pfsync" 46 fi 47} 48 49pflog_init() 50{ 51 if ! kldstat -q -m pflog; then 52 atf_skip "This test requires pflog" 53 fi 54} 55 56pft_set_rules() 57{ 58 jname=$1 59 shift 60 61 if [ $jname == "noflush" ]; 62 then 63 jname=$1 64 shift 65 else 66 # Flush all states, rules, fragments, ... 67 jexec ${jname} pfctl -F all 68 fi 69 70 while [ $# -gt 0 ]; do 71 printf "$1\n" 72 shift 73 done | jexec ${jname} pfctl -f - 74 if [ $? -ne 0 ]; 75 then 76 atf_fail "Failed to set PF rules in ${jname}" 77 fi 78} 79 80pft_cleanup() 81{ 82 vnet_cleanup 83} 84 85pfsynct_cleanup() 86{ 87 pft_cleanup 88} 89 90is_altq_supported() 91{ 92 sysctl -q kern.features.altq >/dev/null || \ 93 atf_skip "Test requires ALTQ" 94 95 while [ -n "$1" ] 96 do 97 sysctl -q kern.features.altq.${1} >/dev/null || \ 98 atf_skip "Test required ALTQ_${1}" 99 shift 100 done 101} 102 103altq_init() 104{ 105 pft_init 106 is_altq_supported 107} 108 109altq_cleanup() 110{ 111 pft_cleanup 112} 113