#!/bin/bash

# test using many netns

# arbitry value of 'many'

HOWMANY=20

IP=$(which ip)

if [ ! -x "$IP" ] ; then

	echo "E: no ip binary" >&2

	exit 1

fi

RULESET="table ip t {

	set s {

		type ipv4_addr

		elements = { 1.1.0.0 }

	}

	chain c {

		ct state new

		udp dport { 12345, 54321 }

		ip saddr @s drop

		jump other

	}

	chain other {

	}

}

table ip6 t {

	set s {

		type ipv6_addr

		elements = { fe00::1 }

	}

	chain c {

		ct state new

		udp dport { 12345, 54321 }

		ip6 saddr @s drop

		jump other

	}

	chain other {

	}

}

table inet t {

	set s {

		type ipv6_addr

		elements = { fe00::1 }

	}

	chain c {

		ct state new

		udp dport { 12345, 54321 }

		ip6 saddr @s drop

		jump other

	}

	chain other {

	}

}

table bridge t {

	chain c {

		jump other

	}

	chain other {

		accept

	}

}

table arp t {

	chain c {

		jump other

	}

	chain other {

		accept

	}

}"

function test_netns()

{

	local NETNS_NAME=$1

	$IP netns add $NETNS_NAME

	if [ $? -ne 0 ] ; then

		echo "E: unable to create netns" >&2

		exit 1

	fi

	$IP netns exec $NETNS_NAME $NFT -f - <<< "$RULESET"

	if [ $? -ne 0 ] ; then

		echo "E: unable to load ruleset in netns" >&2

		$IP netns del $NETNS_NAME

		exit 1

	fi

	KERNEL_RULESET="$($IP netns exec $NETNS_NAME $NFT list ruleset)"

	if [ "$RULESET" != "$KERNEL_RULESET" ] ; then

		echo "E: ruleset in netns $NETNS_NAME differs from the loaded" >&2

	        DIFF="$(which diff)"

	        [ -x $DIFF ] && $DIFF -u <(echo "$RULESET") <(echo "$KERNEL_RULESET")

		$IP netns del $NETNS_NAME

	        exit 1

	fi

	$IP netns del $NETNS_NAME

}

for i in $(seq 1 $HOWMANY) ; do

	NETNS_NAME="$netns${i}_$(basename "$0")"

	test_netns $NETNS_NAME

done

exit 0