#! /usr/bin/atf-sh
#	$NetBSD: net_common.sh,v 1.44 2022/11/02 09:35:12 ozaki-r Exp $
#
# Copyright (c) 2016 Internet Initiative Japan Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

#
# Common utility functions for tests/net
#

export PATH="/sbin:/usr/sbin:/bin:/usr/bin"

HIJACKING="env LD_PRELOAD=/usr/lib/librumphijack.so \
    RUMPHIJACK=path=/rump,socket=all:nolocal,sysctl=yes"
ONEDAYISH="(23h5[0-9]m|1d0h0m)[0-9]+s ?"

extract_new_packets()
{
	local bus=$1
	local old=./.__old

	if [ ! -f $old ]; then
		old=/dev/null
	fi

	shmif_dumpbus -p - $bus 2>/dev/null |
	    tcpdump -n -e -r - 2>/dev/null > ./.__new
	diff -u $old ./.__new | grep '^+' | cut -d '+' -f 2   > ./.__diff
	mv -f ./.__new ./.__old
	cat ./.__diff
}

check_route()
{
	local target=$1
	local gw=$2
	local flags=${3:-\.\+}
	local ifname=${4:-\.\+}

	target=$(echo $target | sed 's/\./\\./g')
	if [ "$gw" = "" ]; then
		gw=".+"
	else
		gw=$(echo $gw | sed 's/\./\\./g')
	fi

	atf_check -s exit:0 -e ignore \
	    -o match:"^$target +$gw +$flags +- +- +.+ +$ifname" \
	    rump.netstat -rn
}

check_route_flags()
{

	check_route "$1" "" "$2" ""
}

check_route_gw()
{

	check_route "$1" "$2" "" ""
}

check_route_no_entry()
{
	local target=$(echo "$1" | sed 's/\./\\./g')

	atf_check -s exit:0 -e ignore -o not-match:"^$target" rump.netstat -rn
}

get_linklocal_addr()
{

	RUMP_SERVER=${1} rump.ifconfig ${2} inet6 |
	    awk "/fe80/ {sub(/%$2/, \"\"); sub(/\\/[0-9]*/, \"\"); print \$2;}"

	return 0
}

get_macaddr()
{

	RUMP_SERVER=${1} rump.ifconfig ${2} | awk '/address/ {print $2;}'
}

HTTPD_PID=./.__httpd.pid
start_httpd()
{
	local sock=$1
	local ip=$2
	local backup=$RUMP_SERVER

	export RUMP_SERVER=$sock

	# start httpd in daemon mode
	atf_check -s exit:0 env LD_PRELOAD=/usr/lib/librumphijack.so \
	    /usr/libexec/httpd -P $HTTPD_PID -i $ip -b -s $(pwd)

	export RUMP_SERVER=$backup

	sleep 3
}

stop_httpd()
{

	if [ -f $HTTPD_PID ]; then
		kill -9 $(cat $HTTPD_PID)
		rm -f $HTTPD_PID
		sleep 1
	fi
}

NC_PID=./.__nc.pid
start_nc_server()
{
	local sock=$1
	local port=$2
	local outfile=$3
	local proto=${4:-ipv4}
	local extra_opts="$5"
	local backup=$RUMP_SERVER
	local opts=

	export RUMP_SERVER=$sock

	if [ $proto = ipv4 ]; then
		opts="-l -4"
	else
		opts="-l -6"
	fi
	opts="$opts $extra_opts"

	env LD_PRELOAD=/usr/lib/librumphijack.so nc $opts $port > $outfile &
	echo $! > $NC_PID

	if [ $proto = ipv4 ]; then
		$DEBUG && rump.netstat -a -f inet
	else
		$DEBUG && rump.netstat -a -f inet6
	fi

	export RUMP_SERVER=$backup

	sleep 1
}

stop_nc_server()
{

	if [ -f $NC_PID ]; then
		kill -9 $(cat $NC_PID)
		rm -f $NC_PID
		sleep 1
	fi
}

BASIC_LIBS="-lrumpnet -lrumpnet_net -lrumpnet_netinet -lrumpnet_shmif"
FS_LIBS="$BASIC_LIBS -lrumpdev -lrumpvfs -lrumpfs_ffs"
CRYPTO_LIBS="$BASIC_LIBS -lrumpdev -lrumpdev_opencrypto \
    -lrumpkern_z -lrumpkern_crypto"
NPF_LIBS="$BASIC_LIBS -lrumpdev -lrumpvfs -lrumpdev_bpf -lrumpnet_npf"
CRYPTO_NPF_LIBS="$CRYPTO_LIBS -lrumpvfs -lrumpdev_bpf -lrumpnet_npf"
BPF_LIBS="$BASIC_LIBS -lrumpdev -lrumpvfs -lrumpdev_bpf"

# We cannot keep variables between test phases, so need to store in files
_rump_server_socks=./.__socks
_rump_server_ifaces=./.__ifaces
_rump_server_buses=./.__buses
_rump_server_macaddrs=./.__macaddrs

DEBUG_SYSCTL_ENTRIES="net.inet.arp.debug net.inet6.icmp6.nd6_debug \
    net.inet.ipsec.debug"

IPSEC_KEY_DEBUG=${IPSEC_KEY_DEBUG:-false}

_rump_server_start_common()
{
	local sock=$1
	local backup=$RUMP_SERVER

	shift 1

	atf_check -s exit:0 rump_server "$@" "$sock"

	if $DEBUG; then
		# Enable debugging features in the kernel
		export RUMP_SERVER=$sock
		for ent in $DEBUG_SYSCTL_ENTRIES; do
			if rump.sysctl -q $ent; then
				atf_check -s exit:0 rump.sysctl -q -w $ent=1
			fi
		done
		export RUMP_SERVER=$backup
	fi
	if $IPSEC_KEY_DEBUG; then
		# Enable debugging features in the kernel
		export RUMP_SERVER=$sock
		if rump.sysctl -q net.key.debug; then
			atf_check -s exit:0 \
			    rump.sysctl -q -w net.key.debug=0xffff
		fi
		export RUMP_SERVER=$backup
	fi

	echo $sock >> $_rump_server_socks
	$DEBUG && cat $_rump_server_socks
}

rump_server_start()
{
	local sock=$1
	local lib=
	local libs="$BASIC_LIBS"

	shift 1

	for lib
	do
		libs="$libs -lrumpnet_$lib"
	done

	_rump_server_start_common $sock $libs

	return 0
}

rump_server_fs_start()
{
	local sock=$1
	local lib=
	local libs="$FS_LIBS"

	shift 1

	for lib
	do
		libs="$libs -lrumpnet_$lib"
	done

	_rump_server_start_common $sock $libs

	return 0
}

rump_server_crypto_start()
{
	local sock=$1
	local lib=
	local libs="$CRYPTO_LIBS"

	shift 1

	for lib
	do
		libs="$libs -lrumpnet_$lib"
	done

	_rump_server_start_common $sock $libs

	return 0
}

rump_server_npf_start()
{
	local sock=$1
	local lib=
	local libs="$NPF_LIBS"

	shift 1

	for lib
	do
		libs="$libs -lrumpnet_$lib"
	done

	_rump_server_start_common $sock $libs

	return 0
}

rump_server_crypto_npf_start()
{
	local sock=$1
	local lib=
	local libs="$CRYPTO_NPF_LIBS"

	shift 1

	for lib
	do
		libs="$libs -lrumpnet_$lib"
	done

	_rump_server_start_common $sock $libs

	return 0
}

rump_server_bpf_start()
{
	local sock=$1
	local lib=
	local libs="$BPF_LIBS"

	shift 1

	for lib
	do
		libs="$libs -lrumpnet_$lib"
	done

	_rump_server_start_common $sock $libs

	return 0
}

rump_server_add_iface()
{
	local sock=$1
	local ifname=$2
	local bus=$3
	local backup=$RUMP_SERVER
	local macaddr=

	export RUMP_SERVER=$sock
	atf_check -s exit:0 rump.ifconfig $ifname create
	if [ -n "$bus" ]; then
		atf_check -s exit:0 rump.ifconfig $ifname linkstr $bus
	fi

	macaddr=$(get_macaddr $sock $ifname)
	if [ -n "$macaddr" ]; then
		if [ -f $_rump_server_macaddrs ]; then
			atf_check -s not-exit:0 \
			    grep -q $macaddr $_rump_server_macaddrs
		fi
		echo $macaddr >> $_rump_server_macaddrs
	fi

	export RUMP_SERVER=$backup

	echo $sock $ifname >> $_rump_server_ifaces
	$DEBUG && cat $_rump_server_ifaces

	echo $bus >> $_rump_server_buses
	cat $_rump_server_buses |sort -u >./.__tmp
	mv -f ./.__tmp $_rump_server_buses
	$DEBUG && cat $_rump_server_buses

	return 0
}

rump_server_check_poolleaks()
{
	local target=$1

	# XXX rumphijack doesn't work with a binary with suid/sgid bits like
	# vmstat.  Use a copied one to drop sgid bit as a workaround until
	# vmstat stops using kvm(3) for /dev/kmem and the sgid bit.
	cp /usr/bin/vmstat ./vmstat
	reqs=$($HIJACKING ./vmstat -mv | awk "/$target/ {print \$3;}")
	rels=$($HIJACKING ./vmstat -mv | awk "/$target/ {print \$5;}")
	rm -f ./vmstat
	atf_check_equal '$target$reqs' '$target$rels'
}

#
# rump_server_check_memleaks detects memory leaks.  It can detect leaks of pool
# objects that are guaranteed to be all deallocated at this point, i.e., all
# created interfaces are destroyed.  Currently only llentpl satisfies this
# constraint.  This mechanism can't be applied to objects allocated through
# pool_cache(9) because it doesn't track released objects explicitly.
#
rump_server_check_memleaks()
{

	rump_server_check_poolleaks llentrypl
	# This doesn't work for objects allocated through pool_cache
	#rump_server_check_poolleaks mbpl
	#rump_server_check_poolleaks mclpl
	#rump_server_check_poolleaks socket
}

rump_server_destroy_ifaces()
{
	local backup=$RUMP_SERVER
	local output=ignore
	local reqs= rels=

	$DEBUG && cat $_rump_server_ifaces

	# Try to dump states before destroying interfaces
	for sock in $(cat $_rump_server_socks); do
		export RUMP_SERVER=$sock
		if $DEBUG; then
			output=save:/dev/stdout
		fi
		atf_check -s exit:0 -o $output rump.ifconfig
		atf_check -s exit:0 -o $output rump.netstat -nr
		# XXX still need hijacking
		atf_check -s exit:0 -o $output $HIJACKING rump.netstat -nai
		atf_check -s exit:0 -o $output rump.arp -na
		atf_check -s exit:0 -o $output rump.ndp -na
		atf_check -s exit:0 -o $output $HIJACKING ifmcstat
	done

	# XXX using pipe doesn't work. See PR bin/51667
	#cat $_rump_server_ifaces | while read sock ifname; do
	# Destroy interfaces in the reverse order
	tac $_rump_server_ifaces > __ifaces
	while read sock ifname; do
		export RUMP_SERVER=$sock
		if rump.ifconfig -l |grep -q $ifname; then
			if $DEBUG; then
				rump.ifconfig -v $ifname
			fi
			atf_check -s exit:0 rump.ifconfig $ifname destroy
		fi
		atf_check -s exit:0 -o ignore rump.ifconfig
	done < __ifaces
	rm -f __ifaces

	for sock in $(cat $_rump_server_socks); do
		export RUMP_SERVER=$sock
		rump_server_check_memleaks
	done

	export RUMP_SERVER=$backup

	return 0
}

rump_server_halt_servers()
{
	local backup=$RUMP_SERVER

	$DEBUG && cat $_rump_server_socks
	for sock in $(cat $_rump_server_socks); do
		env RUMP_SERVER=$sock rump.halt
	done
	export RUMP_SERVER=$backup

	return 0
}

extract_rump_server_core()
{

	if [ -f rump_server.core ]; then
		gdb -ex bt /usr/bin/rump_server rump_server.core
		# Extract kernel logs including a panic message
		strings rump_server.core |grep -E '^\[.+\] '
	fi
}

dump_kernel_stats()
{
	local sock=$1

	echo "### Dumping $sock"
	export RUMP_SERVER=$sock
	rump.ifconfig -av
	rump.netstat -nr
	# XXX still need hijacking
	$HIJACKING rump.netstat -nai
	# XXX workaround for vmstat with the sgid bit
	cp /usr/bin/vmstat ./vmstat
	$HIJACKING ./vmstat -m
	rm -f ./vmstat
	rump.arp -na
	rump.ndp -na
	$HIJACKING ifmcstat
	$HIJACKING dmesg
}

rump_server_dump_servers()
{
	local backup=$RUMP_SERVER

	$DEBUG && cat $_rump_server_socks
	for sock in $(cat $_rump_server_socks); do
		dump_kernel_stats $sock
	done
	export RUMP_SERVER=$backup

	extract_rump_server_core
	return 0
}

rump_server_dump_buses()
{

	if [ ! -f $_rump_server_buses ]; then
		return 0
	fi

	$DEBUG && cat $_rump_server_buses
	for bus in $(cat $_rump_server_buses); do
		echo "### Dumping $bus"
		shmif_dumpbus -p - $bus 2>/dev/null| tcpdump -n -e -r -
	done
	return 0
}

cleanup()
{

	rump_server_halt_servers
}

dump()
{

	rump_server_dump_servers
	rump_server_dump_buses
}

skip_if_qemu()
{
	if drvctl -l qemufwcfg0 >/dev/null 2>&1
	then
	    atf_skip "unreliable under qemu, skip until PR kern/43997 fixed"
	fi
}

test_create_destroy_common()
{
	local sock=$1
	local ifname=$2
	local test_address=${3:-false}
	local ipv4="10.0.0.1/24"
	local ipv6="fc00::1"

	export RUMP_SERVER=$sock

	atf_check -s exit:0 rump.ifconfig $ifname create
	atf_check -s exit:0 rump.ifconfig $ifname destroy

	atf_check -s exit:0 rump.ifconfig $ifname create
	atf_check -s exit:0 rump.ifconfig $ifname up
	atf_check -s exit:0 rump.ifconfig $ifname down
	atf_check -s exit:0 rump.ifconfig $ifname destroy

	# Destroy while UP
	atf_check -s exit:0 rump.ifconfig $ifname create
	atf_check -s exit:0 rump.ifconfig $ifname up
	atf_check -s exit:0 rump.ifconfig $ifname destroy

	if ! $test_address; then
		return
	fi

	# With an IPv4 address
	atf_check -s exit:0 rump.ifconfig $ifname create
	atf_check -s exit:0 rump.ifconfig $ifname inet $ipv4
	atf_check -s exit:0 rump.ifconfig $ifname up
	atf_check -s exit:0 rump.ifconfig $ifname destroy

	# With an IPv6 address
	atf_check -s exit:0 rump.ifconfig $ifname create
	atf_check -s exit:0 rump.ifconfig $ifname inet6 $ipv6
	atf_check -s exit:0 rump.ifconfig $ifname up
	atf_check -s exit:0 rump.ifconfig $ifname destroy

	unset RUMP_SERVER
}
#	$NetBSD: t_cbq.sh,v 1.3 2021/07/16 02:33:32 ozaki-r Exp $
#
# Copyright (c) 2021 Internet Initiative Japan Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

SOCK_LOCAL=unix://altq_local
SOCK_REMOTE=unix://altq_remote
BUS=bus_altq
TIMEOUT=3

# rumphijack can't handle AF_LOCAL socket (/var/run/altq_quip) correctly,
# so use the socket via the host.
HIJACKING_ALTQ="$HIJACKING,blanket=/dev/altq/altq:/dev/altq/cbq:/etc/altq.conf:/var/run/altqd.pid"

DEBUG=${DEBUG:-false}

IP_LOCAL1=10.0.0.1
IP_LOCAL2=10.0.1.1
IP_REMOTE11=10.0.0.2
IP_REMOTE12=10.0.0.22
IP_REMOTE13=10.0.0.23
IP_REMOTE21=10.0.1.2
IP_REMOTE22=10.0.1.22
ALTQD_PIDFILE=./pid

start_altqd()
{

	$HIJACKING_ALTQ altqd

	sleep 0.1
	if $HIJACKING_ALTQ test ! -f /var/run/altqd.pid; then
		sleep 1
	fi

	$HIJACKING_ALTQ test -f /var/run/altqd.pid
	if [ $? != 0 ]; then
		atf_fail "starting altqd failed"
	fi

	$HIJACKING_ALTQ cat /var/run/altqd.pid > $ALTQD_PIDFILE
}

start_altqd_basic()
{

	export RUMP_SERVER=$SOCK_LOCAL

	$HIJACKING_ALTQ mkdir -p /rump/etc
	$HIJACKING_ALTQ mkdir -p /rump/var/run

	cat > ./altq.conf <<-EOF
	interface shmif0 cbq
	class cbq shmif0 root_class NULL pbandwidth 100
	class cbq shmif0 normal_class root_class pbandwidth 50 default
	    filter shmif0 normal_class $IP_REMOTE11 0 0 0 0
	class cbq shmif0 drop_class root_class pbandwidth 0
	    filter shmif0 drop_class $IP_REMOTE12 0 0 0 0
	EOF
	$DEBUG && cat ./altq.conf
	atf_check -s exit:0 $HIJACKING_ALTQ cp ./altq.conf /rump/etc/altq.conf
	atf_check -s exit:0 $HIJACKING_ALTQ test -f /rump/etc/altq.conf

	start_altqd

	$DEBUG && $HIJACKING_ALTQ altqstat -s
	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out
	atf_check -s exit:0 \
	    -o match:"altqstat: cbq on interface shmif0" \
	    -o match:'Class 1 on Interface shmif0: root_class' \
	    -o match:'Class 2 on Interface shmif0: normal_class' \
	    -o match:'Class 3 on Interface shmif0: ctl_class' \
	    -o match:'Class 4 on Interface shmif0: drop_class' \
	    cat ./out
	rm -f ./out
}

shutdown_altqd()
{
	local pid="$(cat $ALTQD_PIDFILE)"

	if [ -n "$pid" ]; then
		pgrep -x altqd | grep -q $pid
		if [ $? = 0 ]; then
			kill $(cat $ALTQD_PIDFILE)
			sleep 1
		fi
		$DEBUG && pgrep -x altqd
	fi
}

check_counter()
{
	local file=$1
	local name=$2
	local match="$3"

	grep -A 8 ${name}_class $file > $file.$name
	atf_check -s exit:0 -o match:"$match" cat $file.$name
	rm -f $file.$name
}

test_altq_cbq_basic_ipv4()
{
	local ifconfig="atf_check -s exit:0 rump.ifconfig"
	local ping="atf_check -s exit:0 -o ignore rump.ping"
	local opts="-q -c 1 -w 1"

	rump_server_fs_start $SOCK_LOCAL local altq
	rump_server_start $SOCK_REMOTE

	rump_server_add_iface $SOCK_LOCAL shmif0 $BUS
	rump_server_add_iface $SOCK_REMOTE shmif0 $BUS

	export RUMP_SERVER=$SOCK_LOCAL
	$ifconfig shmif0 inet $IP_LOCAL1/24
	export RUMP_SERVER=$SOCK_REMOTE
	$ifconfig shmif0 inet $IP_REMOTE11/24
	$ifconfig shmif0 inet $IP_REMOTE12/24 alias
	$ifconfig -w 10

	export RUMP_SERVER=$SOCK_LOCAL
	# Invoke ARP
	$ping $opts $IP_REMOTE11
	$ping $opts $IP_REMOTE12

	start_altqd_basic

	export RUMP_SERVER=$SOCK_LOCAL
	$ping $opts $IP_REMOTE11

	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out

	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'
	check_counter ./out drop   'pkts: 0'

	export RUMP_SERVER=$SOCK_LOCAL
	atf_check -s not-exit:0 -o ignore -e match:"No buffer space available" \
	    rump.ping $opts $IP_REMOTE12

	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out

	check_counter ./out drop   'drops: 1'
	check_counter ./out drop   'pkts: 0'
	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'

	rm -f ./out

	shutdown_altqd

	rump_server_destroy_ifaces
}

start_altqd_multi_ifaces()
{

	export RUMP_SERVER=$SOCK_LOCAL

	$HIJACKING_ALTQ mkdir -p /rump/etc
	$HIJACKING_ALTQ mkdir -p /rump/var/run

	cat > ./altq.conf <<-EOF
	interface shmif0 cbq
	class cbq shmif0 root_class NULL pbandwidth 100
	class cbq shmif0 normal_class root_class pbandwidth 50 default
	    filter shmif0 normal_class $IP_REMOTE11 0 0 0 0
	class cbq shmif0 drop_class root_class pbandwidth 0
	    filter shmif0 drop_class $IP_REMOTE12 0 0 0 0
	interface shmif1 cbq
	class cbq shmif1 root_class NULL pbandwidth 100
	class cbq shmif1 normal_class root_class pbandwidth 50 default
	    filter shmif1 normal_class $IP_REMOTE21 0 0 0 0
	class cbq shmif1 drop_class root_class pbandwidth 0
	    filter shmif1 drop_class $IP_REMOTE22 0 0 0 0
	EOF
	$DEBUG && cat ./altq.conf
	atf_check -s exit:0 $HIJACKING_ALTQ cp ./altq.conf /rump/etc/altq.conf
	$HIJACKING_ALTQ test -f /rump/etc/altq.conf

	start_altqd

	$DEBUG && $HIJACKING_ALTQ altqstat -s

	$HIJACKING_ALTQ altqstat -c 1 -i shmif0 >./out
	$DEBUG && cat ./out
	atf_check -s exit:0 \
	    -o match:"altqstat: cbq on interface shmif0" \
	    -o match:'Class 1 on Interface shmif0: root_class' \
	    -o match:'Class 2 on Interface shmif0: normal_class' \
	    -o match:'Class 3 on Interface shmif0: ctl_class' \
	    -o match:'Class 4 on Interface shmif0: drop_class' \
	    cat ./out

	$HIJACKING_ALTQ altqstat -c 1 -i shmif1 >./out
	$DEBUG && cat ./out
	atf_check -s exit:0 \
	    -o match:"altqstat: cbq on interface shmif1" \
	    -o match:'Class 1 on Interface shmif1: root_class' \
	    -o match:'Class 2 on Interface shmif1: normal_class' \
	    -o match:'Class 3 on Interface shmif1: ctl_class' \
	    -o match:'Class 4 on Interface shmif1: drop_class' \
	    cat ./out

	rm -f ./out
}

test_altq_cbq_multi_ifaces_ipv4()
{
	local ifconfig="atf_check -s exit:0 rump.ifconfig"
	local ping="atf_check -s exit:0 -o ignore rump.ping"
	local opts="-q -c 1 -w 1"

	rump_server_fs_start $SOCK_LOCAL local altq
	rump_server_start $SOCK_REMOTE

	rump_server_add_iface $SOCK_LOCAL shmif0 $BUS
	rump_server_add_iface $SOCK_LOCAL shmif1 $BUS
	rump_server_add_iface $SOCK_REMOTE shmif0 $BUS

	export RUMP_SERVER=$SOCK_LOCAL
	$ifconfig shmif0 inet $IP_LOCAL1/24
	$ifconfig shmif1 inet $IP_LOCAL2/24
	export RUMP_SERVER=$SOCK_REMOTE
	$ifconfig shmif0 inet $IP_REMOTE11/24
	$ifconfig shmif0 inet $IP_REMOTE12/24 alias
	$ifconfig shmif0 inet $IP_REMOTE21/24 alias
	$ifconfig shmif0 inet $IP_REMOTE22/24 alias
	$ifconfig -w 10

	export RUMP_SERVER=$SOCK_LOCAL
	# Invoke ARP
	$ping $opts $IP_REMOTE11
	$ping $opts $IP_REMOTE12
	$ping $opts $IP_REMOTE21
	$ping $opts $IP_REMOTE22

	start_altqd_multi_ifaces

	export RUMP_SERVER=$SOCK_LOCAL
	$ping $opts $IP_REMOTE11

	$HIJACKING_ALTQ altqstat -c 1 -i shmif0 >./out
	$DEBUG && cat ./out

	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'
	check_counter ./out drop   'pkts: 0'

	$ping $opts $IP_REMOTE21

	$HIJACKING_ALTQ altqstat -c 1 -i shmif1 >./out
	$DEBUG && cat ./out

	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'
	check_counter ./out drop   'pkts: 0'

	export RUMP_SERVER=$SOCK_LOCAL
	atf_check -s not-exit:0 -o ignore -e match:"No buffer space available" \
	    rump.ping $opts $IP_REMOTE12

	$HIJACKING_ALTQ altqstat -c 1 -i shmif0 >./out
	$DEBUG && cat ./out

	check_counter ./out drop   'drops: 1'
	check_counter ./out drop   'pkts: 0'
	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'

	atf_check -s not-exit:0 -o ignore -e match:"No buffer space available" \
	    rump.ping $opts $IP_REMOTE22

	$HIJACKING_ALTQ altqstat -c 1 -i shmif1 >./out
	$DEBUG && cat ./out

	check_counter ./out drop   'drops: 1'
	check_counter ./out drop   'pkts: 0'
	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'

	rm -f ./out

	shutdown_altqd

	rump_server_destroy_ifaces
}

start_altqd_options()
{

	export RUMP_SERVER=$SOCK_LOCAL

	$HIJACKING_ALTQ mkdir -p /rump/etc
	$HIJACKING_ALTQ mkdir -p /rump/var/run

	# - no-tbr and no-control are specified
	# - root_class is the default class
	cat > ./altq.conf <<-EOF
	interface shmif0 cbq no-tbr no-control
	class cbq shmif0 root_class NULL pbandwidth 100 default
	class cbq shmif0 normal_class root_class pbandwidth 50
	    filter shmif0 normal_class $IP_REMOTE11 0 0 0 0
	class cbq shmif0 drop_class root_class pbandwidth 0
	    filter shmif0 drop_class $IP_REMOTE12 0 0 0 0
	EOF
	$DEBUG && cat ./altq.conf
	atf_check -s exit:0 $HIJACKING_ALTQ cp ./altq.conf /rump/etc/altq.conf
	$HIJACKING_ALTQ test -f /rump/etc/altq.conf

	start_altqd

	$DEBUG && $HIJACKING_ALTQ altqstat -s
	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out
	atf_check -s exit:0 \
	    -o match:"altqstat: cbq on interface shmif0" \
	    -o match:'Class 1 on Interface shmif0: root_class' \
	    -o match:'Class 2 on Interface shmif0: normal_class' \
	    -o match:'Class 3 on Interface shmif0: drop_class' \
	    cat ./out
	atf_check -s exit:0 -o not-match:'shmif0: ctl_class' cat ./out

	rm -f ./out
}

test_altq_cbq_options_ipv4()
{
	local ifconfig="atf_check -s exit:0 rump.ifconfig"
	local ping="atf_check -s exit:0 -o ignore rump.ping"
	local opts="-q -c 1 -w 1"

	rump_server_fs_start $SOCK_LOCAL local altq
	rump_server_start $SOCK_REMOTE

	rump_server_add_iface $SOCK_LOCAL shmif0 $BUS
	rump_server_add_iface $SOCK_REMOTE shmif0 $BUS

	export RUMP_SERVER=$SOCK_LOCAL
	$ifconfig shmif0 inet $IP_LOCAL1/24
	export RUMP_SERVER=$SOCK_REMOTE
	$ifconfig shmif0 inet $IP_REMOTE11/24
	$ifconfig shmif0 inet $IP_REMOTE12/24 alias
	$ifconfig shmif0 inet $IP_REMOTE13/24 alias
	$ifconfig -w 10

	export RUMP_SERVER=$SOCK_LOCAL
	# Invoke ARP
	$ping $opts $IP_REMOTE11
	$ping $opts $IP_REMOTE12
	$ping $opts $IP_REMOTE13

	start_altqd_options

	export RUMP_SERVER=$SOCK_LOCAL
	$ping $opts $IP_REMOTE11

	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out

	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'
	check_counter ./out drop   'pkts: 0'

	atf_check -s not-exit:0 -o ignore -e match:"No buffer space available" \
	    rump.ping $opts $IP_REMOTE12

	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out

	check_counter ./out drop   'drops: 1'
	check_counter ./out drop   'pkts: 0'
	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 1'

	# The packet goes to the default class
	$ping $opts $IP_REMOTE13

	$HIJACKING_ALTQ altqstat -c 1 >./out
	$DEBUG && cat ./out

	check_counter ./out drop   'pkts: 0'
	check_counter ./out normal 'pkts: 1'
	check_counter ./out root   'pkts: 2'

	rm -f ./out

	shutdown_altqd

	rump_server_destroy_ifaces
}

add_test_case()
{
	local algo=$1
	local type=$2
	local ipproto=$3

	name="altq_${algo}_${type}_${ipproto}"
	desc="Tests for ALTQ $algo (${type}) on ${ipproto}"

	atf_test_case ${name} cleanup
	eval "
	    ${name}_head() {
	        atf_set descr \"$desc\"
	        atf_set require.progs rump_server altqd altqstat
	    }
	    ${name}_body() {
	        test_altq_${algo}_${type}_${ipproto}
	    }
	    ${name}_cleanup() {
	        shutdown_altqd
	        \$DEBUG && dump
	        cleanup
	    }
	"
	atf_add_test_case ${name}
}

atf_init_test_cases()
{

	add_test_case cbq basic        ipv4
	add_test_case cbq multi_ifaces ipv4
	add_test_case cbq options      ipv4
}
