#!/bin/bash

# test loading a ruleset with the 'action object' pattern

set -e

FAMILIES="ip ip6 inet arp bridge"

generate1()
{
	local family=$1
	echo "
	add table $family t
	add chain $family t c
	add rule $family t c accept
	add set $family t s {type inet_service;}
	add element $family t s {8080}
	insert rule $family t c meta l4proto tcp tcp dport @s accept
	add rule $family t c meta l4proto tcp tcp dport {9090, 8080}
	add map $family t m {type inet_service:verdict;}
	add element $family t m {10080:drop}
	insert rule $family t c meta l4proto tcp tcp dport vmap @m
	add rule $family t c meta l4proto udp udp sport vmap {1111:accept, 2222:drop}
	"
}

generate2()
{
	local family=$1
	echo "
	flush chain $family t c
	delete element $family t m {10080:drop}
	delete element $family t s {8080}
	delete chain $family t c
	delete table $family t
	"
}

RULESET=$(for family in $FAMILIES ; do
	generate1 $family
done)

$NFT -f - <<< "$RULESET"
if [ $? -ne 0 ] ; then
	echo "E: unable to load ruleset 1" >&2
	exit 1
fi

RULESET=$(for family in $FAMILIES ; do
	generate2 $family
done)

$NFT -f - <<< "$RULESET"
if [ $? -ne 0 ] ; then
	echo "E: unable to load ruleset 2" >&2
	exit 1
fi

exit 0