Commit c08e8bae authored by Taehee Yoo's avatar Taehee Yoo Committed by David S. Miller
Browse files

selftests: add amt interface selftest script 



This is selftest script for amt interface.
This script includes basic forwarding scenarion and torture scenario.

Signed-off-by: default avatarTaehee Yoo <ap420073@gmail.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent b75f7095

tools/testing/selftests/net/Makefile

+1 −0
Original line number Diff line number Diff line
@@ -22,6 +22,7 @@ TEST_PROGS += devlink_port_split.py 
TEST_PROGS += drop_monitor_tests.sh

TEST_PROGS += vrf_route_leaking.sh

TEST_PROGS += bareudp.sh

TEST_PROGS += amt.sh

TEST_PROGS += unicast_extensions.sh

TEST_PROGS += udpgro_fwd.sh

TEST_PROGS += veth.sh

tools/testing/selftests/net/amt.sh

0 → 100644
+284 −0
Original line number Diff line number Diff line 
#!/bin/sh

# SPDX-License-Identifier: GPL-2.0



# Author: Taehee Yoo <ap420073@gmail.com>

#

# This script evaluates the AMT driver.

# There are four network-namespaces, LISTENER, SOURCE, GATEWAY, RELAY.

# The role of LISTENER is to listen multicast traffic.

# In order to do that, it send IGMP group join message.

# The role of SOURCE is to send multicast traffic to listener.

# The role of GATEWAY is to work Gateway role of AMT interface.

# The role of RELAY is to work Relay role of AMT interface.

#

#

#       +------------------------+

#       |    LISTENER netns      |

#       |                        |

#       |  +------------------+  |

#       |  |       l_gw       |  |

#       |  |  192.168.0.2/24  |  |

#       |  |  2001:db8::2/64  |  |

#       |  +------------------+  |

#       |            .           |

#       +------------------------+

#                    .

#                    .

#       +-----------------------------------------------------+

#       |            .         GATEWAY netns                  |

#       |            .                                        |

#       |+---------------------------------------------------+|

#       ||           .          br0                          ||

#       || +------------------+       +------------------+   ||

#       || |       gw_l       |       |       amtg       |   ||

#       || |  192.168.0.1/24  |       +--------+---------+   ||

#       || |  2001:db8::1/64  |                |             ||

#       || +------------------+                |             ||

#       |+-------------------------------------|-------------+|

#       |                                      |              |

#       |                             +--------+---------+    |

#       |                             |     gw_relay     |    |

#       |                             |    10.0.0.1/24   |    |

#       |                             +------------------+    |

#       |                                      .              |

#       +-----------------------------------------------------+

#                                              .

#                                              .

#       +-----------------------------------------------------+

#       |                       RELAY netns    .              |

#       |                             +------------------+    |

#       |                             |     relay_gw     |    |

#       |                             |    10.0.0.2/24   |    |

#       |                             +--------+---------+    |

#       |                                      |              |

#       |                                      |              |

#       |  +------------------+       +--------+---------+    |

#       |  |     relay_src    |       |       amtr       |    |

#       |  |   172.17.0.1/24  |       +------------------+    |

#       |  | 2001:db8:3::1/64 |                               |

#       |  +------------------+                               |

#       |            .                                        |

#       |            .                                        |

#       +-----------------------------------------------------+

#                    .

#                    .

#       +------------------------+

#       |            .           |

#       |  +------------------+  |

#       |  |     src_relay    |  |

#       |  |   172.17.0.2/24  |  |

#       |  | 2001:db8:3::2/64 |  |

#       |  +------------------+  |

#       |      SOURCE netns      |

#       +------------------------+

#==============================================================================



readonly LISTENER=$(mktemp -u listener-XXXXXXXX)

readonly GATEWAY=$(mktemp -u gateway-XXXXXXXX)

readonly RELAY=$(mktemp -u relay-XXXXXXXX)

readonly SOURCE=$(mktemp -u source-XXXXXXXX)

ERR=4

err=0



exit_cleanup()

{

	for ns in "$@"; do

		ip netns delete "${ns}" 2>/dev/null || true

	done



	exit $ERR

}



create_namespaces()

{

	ip netns add "${LISTENER}" || exit_cleanup

	ip netns add "${GATEWAY}" || exit_cleanup "${LISTENER}"

	ip netns add "${RELAY}" || exit_cleanup "${LISTENER}" "${GATEWAY}"

	ip netns add "${SOURCE}" || exit_cleanup "${LISTENER}" "${GATEWAY}" \

		"${RELAY}"

}



# The trap function handler

#

exit_cleanup_all()

{

	exit_cleanup "${LISTENER}" "${GATEWAY}" "${RELAY}" "${SOURCE}"

}



setup_interface()

{

	for ns in "${LISTENER}" "${GATEWAY}" "${RELAY}" "${SOURCE}"; do

		ip -netns "${ns}" link set dev lo up

	done;



	ip link add l_gw type veth peer name gw_l

	ip link add gw_relay type veth peer name relay_gw

	ip link add relay_src type veth peer name src_relay



	ip link set l_gw netns "${LISTENER}" up

	ip link set gw_l netns "${GATEWAY}" up

	ip link set gw_relay netns "${GATEWAY}" up

	ip link set relay_gw netns "${RELAY}" up

	ip link set relay_src netns "${RELAY}" up

	ip link set src_relay netns "${SOURCE}" up mtu 1400



	ip netns exec "${LISTENER}" ip a a 192.168.0.2/24 dev l_gw

	ip netns exec "${LISTENER}" ip r a default via 192.168.0.1 dev l_gw

	ip netns exec "${LISTENER}" ip a a 2001:db8::2/64 dev l_gw

	ip netns exec "${LISTENER}" ip r a default via 2001:db8::1 dev l_gw

	ip netns exec "${LISTENER}" ip a a 239.0.0.1/32 dev l_gw autojoin

	ip netns exec "${LISTENER}" ip a a ff0e::5:6/128 dev l_gw autojoin



	ip netns exec "${GATEWAY}" ip a a 192.168.0.1/24 dev gw_l

	ip netns exec "${GATEWAY}" ip a a 2001:db8::1/64 dev gw_l

	ip netns exec "${GATEWAY}" ip a a 10.0.0.1/24 dev gw_relay

	ip netns exec "${GATEWAY}" ip link add br0 type bridge

	ip netns exec "${GATEWAY}" ip link set br0 up

	ip netns exec "${GATEWAY}" ip link set gw_l master br0

	ip netns exec "${GATEWAY}" ip link set gw_l up

	ip netns exec "${GATEWAY}" ip link add amtg master br0 type amt \

		mode gateway local 10.0.0.1 discovery 10.0.0.2 dev gw_relay \

		gateway_port 2268 relay_port 2268

	ip netns exec "${RELAY}" ip a a 10.0.0.2/24 dev relay_gw

	ip netns exec "${RELAY}" ip link add amtr type amt mode relay \

		local 10.0.0.2 dev relay_gw relay_port 2268 max_tunnels 4

	ip netns exec "${RELAY}" ip a a 172.17.0.1/24 dev relay_src

	ip netns exec "${RELAY}" ip a a 2001:db8:3::1/64 dev relay_src

	ip netns exec "${SOURCE}" ip a a 172.17.0.2/24 dev src_relay

	ip netns exec "${SOURCE}" ip a a 2001:db8:3::2/64 dev src_relay

	ip netns exec "${SOURCE}" ip r a default via 172.17.0.1 dev src_relay

	ip netns exec "${SOURCE}" ip r a default via 2001:db8:3::1 dev src_relay

	ip netns exec "${RELAY}" ip link set amtr up

	ip netns exec "${GATEWAY}" ip link set amtg up

}



setup_sysctl()

{

	ip netns exec "${RELAY}" sysctl net.ipv4.ip_forward=1 -w -q

}



setup_iptables()

{

	ip netns exec "${RELAY}" iptables -t mangle -I PREROUTING \

		-d 239.0.0.1 -j TTL --ttl-set 2

	ip netns exec "${RELAY}" ip6tables -t mangle -I PREROUTING \

		-j HL --hl-set 2

}



setup_mcast_routing()

{

	ip netns exec "${RELAY}" smcrouted

	ip netns exec "${RELAY}" smcroutectl a relay_src \

		172.17.0.2 239.0.0.1 amtr

	ip netns exec "${RELAY}" smcroutectl a relay_src \

		2001:db8:3::2 ff0e::5:6 amtr

}



test_remote_ip()

{

	REMOTE=$(ip netns exec "${GATEWAY}" \

		ip -d -j link show amtg | jq .[0].linkinfo.info_data.remote)

	if [ $REMOTE == "\"10.0.0.2\"" ]; then

		printf "TEST: %-60s  [ OK ]\n" "amt discovery"

	else

		printf "TEST: %-60s  [FAIL]\n" "amt discovery"

		ERR=1

	fi

}



send_mcast_torture4()

{

	ip netns exec "${SOURCE}" bash -c \

		'cat /dev/urandom | head -c 1G | nc -w 1 -u 239.0.0.1 4001'

}





send_mcast_torture6()

{

	ip netns exec "${SOURCE}" bash -c \

		'cat /dev/urandom | head -c 1G | nc -w 1 -u ff0e::5:6 6001'

}



check_features()

{

        ip link help 2>&1 | grep -q amt

        if [ $? -ne 0 ]; then

                echo "Missing amt support in iproute2" >&2

                exit_cleanup

        fi

}



test_ipv4_forward()

{

	RESULT4=$(ip netns exec "${LISTENER}" nc -w 1 -l -u 239.0.0.1 4000)

	if [ "$RESULT4" == "172.17.0.2" ]; then

		printf "TEST: %-60s  [ OK ]\n" "IPv4 amt multicast forwarding"

		exit 0

	else

		printf "TEST: %-60s  [FAIL]\n" "IPv4 amt multicast forwarding"

		exit 1

	fi

}



test_ipv6_forward()

{

	RESULT6=$(ip netns exec "${LISTENER}" nc -w 1 -l -u ff0e::5:6 6000)

	if [ "$RESULT6" == "2001:db8:3::2" ]; then

		printf "TEST: %-60s  [ OK ]\n" "IPv6 amt multicast forwarding"

		exit 0

	else

		printf "TEST: %-60s  [FAIL]\n" "IPv6 amt multicast forwarding"

		exit 1

	fi

}



send_mcast4()

{

	sleep 2

	ip netns exec "${SOURCE}" bash -c \

		'echo 172.17.0.2 | nc -w 1 -u 239.0.0.1 4000' &

}



send_mcast6()

{

	sleep 2

	ip netns exec "${SOURCE}" bash -c \

		'echo 2001:db8:3::2 | nc -w 1 -u ff0e::5:6 6000' &

}



check_features



create_namespaces



set -e

trap exit_cleanup_all EXIT



setup_interface

setup_sysctl

setup_iptables

setup_mcast_routing

test_remote_ip

test_ipv4_forward &

pid=$!

send_mcast4

wait $pid || err=$?

if [ $err -eq 1 ]; then

	ERR=1

fi

test_ipv6_forward &

pid=$!

send_mcast6

wait $pid || err=$?

if [ $err -eq 1 ]; then

	ERR=1

fi

send_mcast_torture4

printf "TEST: %-60s  [ OK ]\n" "IPv4 amt traffic forwarding torture"

send_mcast_torture6

printf "TEST: %-60s  [ OK ]\n" "IPv6 amt traffic forwarding torture"

sleep 5

if [ "${ERR}" -eq 1 ]; then

        echo "Some tests failed." >&2

else

        ERR=0

fi

tools/testing/selftests/net/config

+1 −0
Original line number Diff line number Diff line
@@ -44,3 +44,4 @@ CONFIG_NET_ACT_MIRRED=m 
CONFIG_BAREUDP=m

CONFIG_IPV6_IOAM6_LWTUNNEL=y

CONFIG_CRYPTO_SM4=y

CONFIG_AMT=m