tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3
4# Author: Taehee Yoo <ap420073@gmail.com>
5#
6# This script evaluates the AMT driver.
7# There are four network-namespaces, LISTENER, SOURCE, GATEWAY, RELAY.
8# The role of LISTENER is to listen multicast traffic.
9# In order to do that, it send IGMP group join message.
10# The role of SOURCE is to send multicast traffic to listener.
11# The role of GATEWAY is to work Gateway role of AMT interface.
12# The role of RELAY is to work Relay role of AMT interface.
13#
14#
15# +------------------------+
16# | LISTENER netns |
17# | |
18# | +------------------+ |
19# | | l_gw | |
20# | | 192.168.0.2/24 | |
21# | | 2001:db8::2/64 | |
22# | +------------------+ |
23# | . |
24# +------------------------+
25# .
26# .
27# +-----------------------------------------------------+
28# | . GATEWAY netns |
29# | . |
30# |+---------------------------------------------------+|
31# || . br0 ||
32# || +------------------+ +------------------+ ||
33# || | gw_l | | amtg | ||
34# || | 192.168.0.1/24 | +--------+---------+ ||
35# || | 2001:db8::1/64 | | ||
36# || +------------------+ | ||
37# |+-------------------------------------|-------------+|
38# | | |
39# | +--------+---------+ |
40# | | gw_relay | |
41# | | 10.0.0.1/24 | |
42# | +------------------+ |
43# | . |
44# +-----------------------------------------------------+
45# .
46# .
47# +-----------------------------------------------------+
48# | RELAY netns . |
49# | +------------------+ |
50# | | relay_gw | |
51# | | 10.0.0.2/24 | |
52# | +--------+---------+ |
53# | | |
54# | | |
55# | +------------------+ +--------+---------+ |
56# | | relay_src | | amtr | |
57# | | 172.17.0.1/24 | +------------------+ |
58# | | 2001:db8:3::1/64 | |
59# | +------------------+ |
60# | . |
61# | . |
62# +-----------------------------------------------------+
63# .
64# .
65# +------------------------+
66# | . |
67# | +------------------+ |
68# | | src_relay | |
69# | | 172.17.0.2/24 | |
70# | | 2001:db8:3::2/64 | |
71# | +------------------+ |
72# | SOURCE netns |
73# +------------------------+
74#==============================================================================
75
76readonly LISTENER=$(mktemp -u listener-XXXXXXXX)
77readonly GATEWAY=$(mktemp -u gateway-XXXXXXXX)
78readonly RELAY=$(mktemp -u relay-XXXXXXXX)
79readonly SOURCE=$(mktemp -u source-XXXXXXXX)
80readonly SMCROUTEDIR="$(mktemp -d)"
81ERR=4
82err=0
83
84exit_cleanup()
85{
86 for ns in "$@"; do
87 ip netns delete "${ns}" 2>/dev/null || true
88 done
89 if [ -f "$SMCROUTEDIR/amt.pid" ]; then
90 smcpid=$(< $SMCROUTEDIR/amt.pid)
91 kill $smcpid
92 fi
93 rm -rf $SMCROUTEDIR
94
95 exit $ERR
96}
97
98create_namespaces()
99{
100 ip netns add "${LISTENER}" || exit_cleanup
101 ip netns add "${GATEWAY}" || exit_cleanup "${LISTENER}"
102 ip netns add "${RELAY}" || exit_cleanup "${LISTENER}" "${GATEWAY}"
103 ip netns add "${SOURCE}" || exit_cleanup "${LISTENER}" "${GATEWAY}" \
104 "${RELAY}"
105}
106
107# The trap function handler
108#
109exit_cleanup_all()
110{
111 exit_cleanup "${LISTENER}" "${GATEWAY}" "${RELAY}" "${SOURCE}"
112}
113
114setup_interface()
115{
116 for ns in "${LISTENER}" "${GATEWAY}" "${RELAY}" "${SOURCE}"; do
117 ip -netns "${ns}" link set dev lo up
118 done;
119
120 ip link add l_gw type veth peer name gw_l
121 ip link add gw_relay type veth peer name relay_gw
122 ip link add relay_src type veth peer name src_relay
123
124 ip link set l_gw netns "${LISTENER}" up
125 ip link set gw_l netns "${GATEWAY}" up
126 ip link set gw_relay netns "${GATEWAY}" up
127 ip link set relay_gw netns "${RELAY}" up
128 ip link set relay_src netns "${RELAY}" up
129 ip link set src_relay netns "${SOURCE}" up mtu 1400
130
131 ip netns exec "${LISTENER}" ip a a 192.168.0.2/24 dev l_gw
132 ip netns exec "${LISTENER}" ip r a default via 192.168.0.1 dev l_gw
133 ip netns exec "${LISTENER}" ip a a 2001:db8::2/64 dev l_gw
134 ip netns exec "${LISTENER}" ip r a default via 2001:db8::1 dev l_gw
135 ip netns exec "${LISTENER}" ip a a 239.0.0.1/32 dev l_gw autojoin
136 ip netns exec "${LISTENER}" ip a a ff0e::5:6/128 dev l_gw autojoin
137
138 ip netns exec "${GATEWAY}" ip a a 192.168.0.1/24 dev gw_l
139 ip netns exec "${GATEWAY}" ip a a 2001:db8::1/64 dev gw_l
140 ip netns exec "${GATEWAY}" ip a a 10.0.0.1/24 dev gw_relay
141 ip netns exec "${GATEWAY}" ip link add br0 type bridge
142 ip netns exec "${GATEWAY}" ip link set br0 up
143 ip netns exec "${GATEWAY}" ip link set gw_l master br0
144 ip netns exec "${GATEWAY}" ip link set gw_l up
145 ip netns exec "${GATEWAY}" ip link add amtg master br0 type amt \
146 mode gateway local 10.0.0.1 discovery 10.0.0.2 dev gw_relay \
147 gateway_port 2268 relay_port 2268
148 ip netns exec "${RELAY}" ip a a 10.0.0.2/24 dev relay_gw
149 ip netns exec "${RELAY}" ip link add amtr type amt mode relay \
150 local 10.0.0.2 dev relay_gw relay_port 2268 max_tunnels 4
151 ip netns exec "${RELAY}" ip a a 172.17.0.1/24 dev relay_src
152 ip netns exec "${RELAY}" ip a a 2001:db8:3::1/64 dev relay_src
153 ip netns exec "${SOURCE}" ip a a 172.17.0.2/24 dev src_relay
154 ip netns exec "${SOURCE}" ip a a 2001:db8:3::2/64 dev src_relay
155 ip netns exec "${SOURCE}" ip r a default via 172.17.0.1 dev src_relay
156 ip netns exec "${SOURCE}" ip r a default via 2001:db8:3::1 dev src_relay
157 ip netns exec "${RELAY}" ip link set amtr up
158 ip netns exec "${GATEWAY}" ip link set amtg up
159}
160
161setup_sysctl()
162{
163 ip netns exec "${RELAY}" sysctl net.ipv4.ip_forward=1 -w -q
164}
165
166setup_iptables()
167{
168 ip netns exec "${RELAY}" iptables -t mangle -I PREROUTING \
169 -d 239.0.0.1 -j TTL --ttl-set 2
170 ip netns exec "${RELAY}" ip6tables -t mangle -I PREROUTING \
171 -j HL --hl-set 2
172}
173
174setup_mcast_routing()
175{
176 ip netns exec "${RELAY}" smcrouted -P $SMCROUTEDIR/amt.pid
177 ip netns exec "${RELAY}" smcroutectl a relay_src \
178 172.17.0.2 239.0.0.1 amtr
179 ip netns exec "${RELAY}" smcroutectl a relay_src \
180 2001:db8:3::2 ff0e::5:6 amtr
181}
182
183test_remote_ip()
184{
185 REMOTE=$(ip netns exec "${GATEWAY}" \
186 ip -d -j link show amtg | jq .[0].linkinfo.info_data.remote)
187 if [ $REMOTE == "\"10.0.0.2\"" ]; then
188 printf "TEST: %-60s [ OK ]\n" "amt discovery"
189 else
190 printf "TEST: %-60s [FAIL]\n" "amt discovery"
191 ERR=1
192 fi
193}
194
195send_mcast_torture4()
196{
197 for i in `seq 10`; do
198 ip netns exec "${SOURCE}" bash -c \
199 'cat /dev/urandom | head -c 100M | nc -w 1 -u 239.0.0.1 4001'
200 echo -n "."
201 done
202}
203
204
205send_mcast_torture6()
206{
207 for i in `seq 10`; do
208 ip netns exec "${SOURCE}" bash -c \
209 'cat /dev/urandom | head -c 100M | nc -w 1 -u ff0e::5:6 6001'
210 echo -n "."
211 done
212}
213
214check_features()
215{
216 ip link help 2>&1 | grep -q amt
217 if [ $? -ne 0 ]; then
218 echo "Missing amt support in iproute2" >&2
219 exit_cleanup
220 fi
221}
222
223test_ipv4_forward()
224{
225 RESULT4=$(ip netns exec "${LISTENER}" timeout 15 socat - UDP4-LISTEN:4000,readbytes=128 || true)
226 if echo "$RESULT4" | grep -q "172.17.0.2"; then
227 printf "TEST: %-60s [ OK ]\n" "IPv4 amt multicast forwarding"
228 exit 0
229 else
230 printf "TEST: %-60s [FAIL]\n" "IPv4 amt multicast forwarding"
231 exit 1
232 fi
233}
234
235test_ipv6_forward()
236{
237 RESULT6=$(ip netns exec "${LISTENER}" timeout 15 socat - UDP6-LISTEN:6000,readbytes=128 || true)
238 if echo "$RESULT6" | grep -q "2001:db8:3::2"; then
239 printf "TEST: %-60s [ OK ]\n" "IPv6 amt multicast forwarding"
240 exit 0
241 else
242 printf "TEST: %-60s [FAIL]\n" "IPv6 amt multicast forwarding"
243 exit 1
244 fi
245}
246
247send_mcast4()
248{
249 sleep 2
250 ip netns exec "${SOURCE}" bash -c \
251 'printf "%s %128s" 172.17.0.2 | nc -w 1 -u 239.0.0.1 4000' &
252}
253
254send_mcast6()
255{
256 sleep 2
257 ip netns exec "${SOURCE}" bash -c \
258 'printf "%s %128s" 2001:db8:3::2 | nc -w 1 -u ff0e::5:6 6000' &
259}
260
261check_features
262
263create_namespaces
264
265set -e
266trap exit_cleanup_all EXIT
267
268setup_interface
269setup_sysctl
270setup_iptables
271setup_mcast_routing
272test_remote_ip
273test_ipv4_forward &
274pid=$!
275send_mcast4
276wait $pid || err=$?
277if [ $err -eq 1 ]; then
278 ERR=1
279fi
280test_ipv6_forward &
281pid=$!
282send_mcast6
283wait $pid || err=$?
284if [ $err -eq 1 ]; then
285 ERR=1
286fi
287printf "TEST: %-50s" "IPv4 amt traffic forwarding torture"
288send_mcast_torture4
289printf " [ OK ]\n"
290printf "TEST: %-50s" "IPv6 amt traffic forwarding torture"
291send_mcast_torture6
292printf " [ OK ]\n"
293sleep 5
294if [ "${ERR}" -eq 1 ]; then
295 echo "Some tests failed." >&2
296else
297 ERR=0
298fi