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 49pft_set_rules() 50{ 51 jname=$1 52 shift 53 54 if [ $jname == "noflush" ]; 55 then 56 jname=$1 57 shift 58 else 59 # Flush all states, rules, fragments, ... 60 jexec ${jname} pfctl -F all 61 fi 62 63 while [ $# -gt 0 ]; do 64 printf "$1\n" 65 shift 66 done | jexec ${jname} pfctl -f - 67 if [ $? -ne 0 ]; 68 then 69 atf_fail "Failed to set PF rules in ${jname}" 70 fi 71} 72 73pft_cleanup() 74{ 75 vnet_cleanup 76} 77 78pfsynct_cleanup() 79{ 80 pft_cleanup 81} 82