1# 2# SPDX-License-Identifier: BSD-2-Clause 3# 4# Copyright (c) 2025 Kristof Provost <kp@FreeBSD.org> 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions 8# are met: 9# 1. Redistributions of source code must retain the above copyright 10# notice, this list of conditions and the following disclaimer. 11# 2. Redistributions in binary form must reproduce the above copyright 12# notice, this list of conditions and the following disclaimer in the 13# documentation and/or other materials provided with the distribution. 14# 15# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25# SUCH DAMAGE. 26 27. $(atf_get_srcdir)/utils.subr 28 29atf_test_case "basic" "cleanup" 30basic_head() 31{ 32 atf_set descr 'Basic pf_snmp test' 33 atf_set require.user root 34} 35 36basic_body() 37{ 38 pft_init 39 40 epair=$(vnet_mkepair) 41 42 ifconfig ${epair}b 192.0.2.2/24 up 43 44 vnet_mkjail alcatraz ${epair}a 45 jexec alcatraz ifconfig ${epair}a 192.0.2.1/24 up 46 47 # Start bsnmpd 48 jexec alcatraz bsnmpd -c $(atf_get_srcdir)/bsnmpd.conf 49 50 jexec alcatraz pfctl -e 51 pft_set_rules alcatraz \ 52 "pass" 53 54 # Sanity check, and create state 55 atf_check -s exit:0 -o ignore \ 56 ping -c 1 192.0.2.1 57 58 # pf should be enabled 59 atf_check -s exit:0 -o match:'pfStatusRunning.0 = true' \ 60 bsnmpwalk -s public@192.0.2.1 -i pf_tree.def begemot 61} 62 63basic_cleanup() 64{ 65 pft_cleanup 66} 67 68atf_test_case "table" "cleanup" 69table_head() 70{ 71 atf_set descr 'Test tables and pf_snmp' 72 atf_set require.user root 73} 74 75table_body() 76{ 77 pft_init 78 79 epair=$(vnet_mkepair) 80 81 ifconfig ${epair}b 192.0.2.2/24 up 82 83 vnet_mkjail alcatraz ${epair}a 84 jexec alcatraz ifconfig ${epair}a 192.0.2.1/24 up 85 86 jexec alcatraz pfctl -e 87 pft_set_rules alcatraz \ 88 "table <foo> counters { 192.0.2.0/24 }" \ 89 "pass in from <foo>" 90 91 # Start bsnmpd after creating the table so we don't have to wait for 92 # a refresh timeout 93 jexec alcatraz bsnmpd -c $(atf_get_srcdir)/bsnmpd.conf 94 95 # Sanity check, and create state 96 atf_check -s exit:0 -o ignore \ 97 ping -c 1 192.0.2.1 98 99 # We should have one table 100 atf_check -s exit:0 -o match:'pfTablesTblNumber.0 = 1' \ 101 bsnmpwalk -s public@192.0.2.1 -i pf_tree.def begemot 102 103 # We have the 'foo' table 104 atf_check -s exit:0 -o match:'pfTablesTblDescr.* = foo' \ 105 bsnmpwalk -s public@192.0.2.1 -i pf_tree.def pfTables 106 107 # Which contains address 192.0.2.0/24 108 atf_check -s exit:0 -o match:'pfTablesAddrNet.* = 192.0.2.0' \ 109 bsnmpwalk -s public@192.0.2.1 -i pf_tree.def pfTables 110 atf_check -s exit:0 -o match:'pfTablesAddrPrefix.* = 24' \ 111 bsnmpwalk -s public@192.0.2.1 -i pf_tree.def pfTables 112} 113 114table_cleanup() 115{ 116 pft_cleanup 117} 118 119atf_init_test_cases() 120{ 121 atf_add_test_case "basic" 122 atf_add_test_case "table" 123} 124