1#- 2# SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3# 4# Copyright (c) 2019 Ahsan Barkati 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# $FreeBSD$ 28# 29 30. $(atf_get_srcdir)/utils.subr 31. $(atf_get_srcdir)/runner.subr 32 33tos_head() 34{ 35 atf_set descr 'set-tos test' 36 atf_set require.user root 37 atf_set require.progs scapy 38} 39 40tos_body() 41{ 42 firewall=$1 43 firewall_init $firewall 44 45 epair_send=$(vnet_mkepair) 46 ifconfig ${epair_send}a 192.0.2.1/24 up 47 48 epair_recv=$(vnet_mkepair) 49 ifconfig ${epair_recv}a up 50 51 vnet_mkjail iron ${epair_send}b ${epair_recv}b 52 jexec iron ifconfig ${epair_send}b 192.0.2.2/24 up 53 jexec iron ifconfig ${epair_recv}b 198.51.100.2/24 up 54 jexec iron sysctl net.inet.ip.forwarding=1 55 jexec iron arp -s 198.51.100.3 00:01:02:03:04:05 56 route add -net 198.51.100.0/24 192.0.2.2 57 58 # Check if the firewall is able to set the ToS bits 59 firewall_config "iron" ${firewall} \ 60 "pf" \ 61 "scrub out proto icmp set-tos 36" \ 62 "ipfw" \ 63 "ipfw -q add 100 setdscp 9 ip from any to any" 64 # dscp is set to 9 because last two bits are for 65 # EN and hence tos would be 36 66 67 atf_check -s exit:0 $(atf_get_srcdir)/pft_ping.py \ 68 --sendif ${epair_send}a \ 69 --to 198.51.100.3 \ 70 --recvif ${epair_recv}a \ 71 --expect-tos 36 72 73 # Check if the firewall is able to set the ToS bits 74 # and persists the EN bits (if already set) 75 firewall_config "iron" ${firewall} \ 76 "pf" \ 77 "scrub out proto icmp set-tos 36" \ 78 "ipfw" \ 79 "ipfw -q add 100 setdscp 9 ip from any to any" 80 81 atf_check -s exit:0 $(atf_get_srcdir)/pft_ping.py \ 82 --sendif ${epair_send}a \ 83 --to 198.51.100.3 \ 84 --recvif ${epair_recv}a \ 85 --send-tos 3 \ 86 --expect-tos 39 87 88 # Check if the firewall is able to filter the 89 # packets based on the ToS value 90 firewall_config "iron" ${firewall} \ 91 "pf" \ 92 "block all tos 36" \ 93 "ipfw" \ 94 "ipfw -q add 100 deny all from any to any dscp 9" 95 96 atf_check -s exit:1 $(atf_get_srcdir)/pft_ping.py \ 97 --sendif ${epair_send}a \ 98 --to 198.51.100.3 \ 99 --recvif ${epair_recv}a \ 100 --send-tos 36 101 102 atf_check -s exit:0 $(atf_get_srcdir)/pft_ping.py \ 103 --sendif ${epair_send}a \ 104 --to 198.51.100.3 \ 105 --recvif ${epair_recv}a \ 106 --send-tos 32 107} 108 109tos_cleanup() 110{ 111 firewall=$1 112 firewall_cleanup $firewall 113} 114 115setup_tests \ 116 "tos" \ 117 "pf" \ 118 "ipfw" 119