#!/bin/sh

. lib/generic.sh

DEV="$(rand_dev)"

ts_ip "$0" "Add $DEV dummy interface" link add dev $DEV type dummy

ts_ip "$0" "Enable $DEV" link set $DEV up

ts_tc "pedit" "Add ingress qdisc" qdisc add dev $DEV ingress

do_pedit() {

	ts_tc "pedit" "Drop ingress qdisc" \

		qdisc del dev $DEV ingress

	ts_tc "pedit" "Add ingress qdisc" \

		qdisc add dev $DEV ingress

	ts_tc "pedit" "Add pedit action $*" \

		filter add dev $DEV parent ffff: \

		u32 match u32 0 0 \

		action pedit munge $@

	ts_tc "pedit" "Show ingress filters" \

		filter show dev $DEV parent ffff:

}

do_pedit offset 12 u32 set 0x12345678

test_on "key #0  at 12: val 12345678 mask 00000000"

do_pedit offset 12 u16 set 0x1234

test_on "key #0  at 12: val 12340000 mask 0000ffff"

do_pedit offset 14 u16 set 0x1234

test_on "key #0  at 12: val 00001234 mask ffff0000"

do_pedit offset 12 u8 set 0x23

test_on "key #0  at 12: val 23000000 mask 00ffffff"

do_pedit offset 13 u8 set 0x23

test_on "key #0  at 12: val 00230000 mask ff00ffff"

do_pedit offset 14 u8 set 0x23

test_on "key #0  at 12: val 00002300 mask ffff00ff"

do_pedit offset 15 u8 set 0x23

test_on "key #0  at 12: val 00000023 mask ffffff00"

do_pedit offset 13 u8 invert

test_on "key #0  at 12: val 00ff0000 mask ffffffff"

do_pedit offset 13 u8 clear

test_on "key #0  at 12: val 00000000 mask ff00ffff"

do_pedit offset 13 u8 preserve

test_on "key #0  at 12: val 00000000 mask ffffffff"

# the following set of tests has been auto-generated by running this little

# shell script:

#

# do_it() {

#	echo "do_pedit $@"

#	tc qd del dev veth0 ingress >/dev/null 2>&1

#	tc qd add dev veth0 ingress >/dev/null 2>&1

#	tc filter add dev veth0 parent ffff: u32 \

#		match u32 0 0 \

#		action pedit munge $@ >/dev/null 2>&1

#	tc filter show dev veth0 parent ffff: | \

#		sed -n 's/^[\t ]*\(key #0.*\)/test_on "\1"/p'

# }

#

# do_it_all() { # (field, val1 [, val2, ...])

#	local field=$1

#	shift

#	for val in $@; do

#		do_it ip $field set $val

#	done

#	for i in preserve invert clear; do

#		do_it ip $field $i

#	done

# }

#

# do_it_all ihl 0x04 0x40

# do_it_all src 1.2.3.4

# do_it_all dst 1.2.3.4

# do_it_all tos 0x1 0x10

# do_it_all protocol 0x23

# do_it_all nofrag 0x23 0xf4

# do_it_all firstfrag 0x03 0xfa

# do_it_all ce 0x23 0x04 0xf3

# do_it_all df 0x23 0x04 0xf3

# do_it_all mf 0x23 0x04 0xf3

# do_it_all dport 0x1234

# do_it_all sport 0x1234

# do_it_all icmp_type 0x23

# do_it_all icmp_code 0x23

do_pedit ip ihl set 0x04

test_on "key #0  at 0: val 04000000 mask f0ffffff"

do_pedit ip ihl set 0x40

test_on "key #0  at 0: val 00000000 mask f0ffffff"

do_pedit ip ihl preserve

test_on "key #0  at 0: val 00000000 mask ffffffff"

do_pedit ip ihl invert

test_on "key #0  at 0: val 0f000000 mask ffffffff"

do_pedit ip ihl clear

test_on "key #0  at 0: val 00000000 mask f0ffffff"

do_pedit ip src set 1.2.3.4

test_on "key #0  at 12: val 01020304 mask 00000000"

do_pedit ip src preserve

test_on "key #0  at 12: val 00000000 mask ffffffff"

do_pedit ip src invert

test_on "key #0  at 12: val ffffffff mask ffffffff"

do_pedit ip src clear

test_on "key #0  at 12: val 00000000 mask 00000000"

do_pedit ip dst set 1.2.3.4

test_on "key #0  at 16: val 01020304 mask 00000000"

do_pedit ip dst preserve

test_on "key #0  at 16: val 00000000 mask ffffffff"

do_pedit ip dst invert

test_on "key #0  at 16: val ffffffff mask ffffffff"

do_pedit ip dst clear

test_on "key #0  at 16: val 00000000 mask 00000000"

do_pedit ip tos set 0x1

test_on "key #0  at 0: val 00010000 mask ff00ffff"

do_pedit ip tos set 0x10

test_on "key #0  at 0: val 00100000 mask ff00ffff"

do_pedit ip tos preserve

test_on "key #0  at 0: val 00000000 mask ffffffff"

do_pedit ip tos invert

test_on "key #0  at 0: val 00ff0000 mask ffffffff"

do_pedit ip tos clear

test_on "key #0  at 0: val 00000000 mask ff00ffff"

do_pedit ip protocol set 0x23

test_on "key #0  at 8: val 00230000 mask ff00ffff"

do_pedit ip protocol preserve

test_on "key #0  at 8: val 00000000 mask ffffffff"

do_pedit ip protocol invert

test_on "key #0  at 8: val 00ff0000 mask ffffffff"

do_pedit ip protocol clear

test_on "key #0  at 8: val 00000000 mask ff00ffff"

do_pedit ip nofrag set 0x23

test_on "key #0  at 4: val 00002300 mask ffffc0ff"

do_pedit ip nofrag set 0xf4

test_on "key #0  at 4: val 00003400 mask ffffc0ff"

do_pedit ip nofrag preserve

test_on "key #0  at 4: val 00000000 mask ffffffff"

do_pedit ip nofrag invert

test_on "key #0  at 4: val 00003f00 mask ffffffff"

do_pedit ip nofrag clear

test_on "key #0  at 4: val 00000000 mask ffffc0ff"

do_pedit ip firstfrag set 0x03

test_on "key #0  at 4: val 00000300 mask ffffe0ff"

do_pedit ip firstfrag set 0xfa

test_on "key #0  at 4: val 00001a00 mask ffffe0ff"

do_pedit ip firstfrag preserve

test_on "key #0  at 4: val 00000000 mask ffffffff"

do_pedit ip firstfrag invert

test_on "key #0  at 4: val 00001f00 mask ffffffff"

do_pedit ip firstfrag clear

test_on "key #0  at 4: val 00000000 mask ffffe0ff"

do_pedit ip ce set 0x23

test_on "key #0  at 4: val 00000000 mask ffff7fff"

do_pedit ip ce set 0x04

test_on "key #0  at 4: val 00000000 mask ffff7fff"

do_pedit ip ce set 0xf3

test_on "key #0  at 4: val 00008000 mask ffff7fff"

do_pedit ip ce preserve

test_on "key #0  at 4: val 00000000 mask ffffffff"

do_pedit ip ce invert

test_on "key #0  at 4: val 00008000 mask ffffffff"

do_pedit ip ce clear

test_on "key #0  at 4: val 00000000 mask ffff7fff"

do_pedit ip df set 0x23

test_on "key #0  at 4: val 00000000 mask ffffbfff"

do_pedit ip df set 0x04

test_on "key #0  at 4: val 00000000 mask ffffbfff"

do_pedit ip df set 0xf3

test_on "key #0  at 4: val 00004000 mask ffffbfff"

do_pedit ip df preserve

test_on "key #0  at 4: val 00000000 mask ffffffff"

do_pedit ip df invert

test_on "key #0  at 4: val 00004000 mask ffffffff"

do_pedit ip df clear

test_on "key #0  at 4: val 00000000 mask ffffbfff"

do_pedit ip mf set 0x23

test_on "key #0  at 4: val 00002000 mask ffffdfff"

do_pedit ip mf set 0x04

test_on "key #0  at 4: val 00000000 mask ffffdfff"

do_pedit ip mf set 0xf3

test_on "key #0  at 4: val 00002000 mask ffffdfff"

do_pedit ip mf preserve

test_on "key #0  at 4: val 00000000 mask ffffffff"

do_pedit ip mf invert

test_on "key #0  at 4: val 00002000 mask ffffffff"

do_pedit ip mf clear

test_on "key #0  at 4: val 00000000 mask ffffdfff"

do_pedit ip dport set 0x1234

test_on "key #0  at 20: val 00001234 mask ffff0000"

do_pedit ip dport preserve

test_on "key #0  at 20: val 00000000 mask ffffffff"

do_pedit ip dport invert

test_on "key #0  at 20: val 0000ffff mask ffffffff"

do_pedit ip dport clear

test_on "key #0  at 20: val 00000000 mask ffff0000"

do_pedit ip sport set 0x1234

test_on "key #0  at 20: val 12340000 mask 0000ffff"

do_pedit ip sport preserve

test_on "key #0  at 20: val 00000000 mask ffffffff"

do_pedit ip sport invert

test_on "key #0  at 20: val ffff0000 mask ffffffff"

do_pedit ip sport clear

test_on "key #0  at 20: val 00000000 mask 0000ffff"

do_pedit ip icmp_type set 0x23

test_on "key #0  at 20: val 23000000 mask 00ffffff"

do_pedit ip icmp_type preserve

test_on "key #0  at 20: val 00000000 mask ffffffff"

do_pedit ip icmp_type invert

test_on "key #0  at 20: val ff000000 mask ffffffff"

do_pedit ip icmp_type clear

test_on "key #0  at 20: val 00000000 mask 00ffffff"

do_pedit ip icmp_code set 0x23

test_on "key #0  at 20: val 23000000 mask 00ffffff"

do_pedit ip icmp_code preserve

test_on "key #0  at 20: val 00000000 mask ffffffff"

do_pedit ip icmp_code invert

test_on "key #0  at 20: val ff000000 mask ffffffff"

do_pedit ip icmp_code clear

test_on "key #0  at 20: val 00000000 mask 00ffffff"