|
| 1 | +#!/bin/sh |
| 2 | +# SPDX-License-Identifier: GPL-2.0 |
| 3 | +# |
| 4 | +# End-to-end ipvs test suite |
| 5 | +# Topology: |
| 6 | +#--------------------------------------------------------------+ |
| 7 | +# | | |
| 8 | +# ns0 | ns1 | |
| 9 | +# ----------- | ----------- ----------- | |
| 10 | +# | veth01 | --------- | veth10 | | veth12 | | |
| 11 | +# ----------- peer ----------- ----------- | |
| 12 | +# | | | | |
| 13 | +# ----------- | | | |
| 14 | +# | br0 | |----------------- peer |--------------| |
| 15 | +# ----------- | | | |
| 16 | +# | | | | |
| 17 | +# ---------- peer ---------- ----------- | |
| 18 | +# | veth02 | --------- | veth20 | | veth21 | | |
| 19 | +# ---------- | ---------- ----------- | |
| 20 | +# | ns2 | |
| 21 | +# | | |
| 22 | +#--------------------------------------------------------------+ |
| 23 | +# |
| 24 | +# We assume that all network driver are loaded |
| 25 | +# |
| 26 | + |
| 27 | +# Kselftest framework requirement - SKIP code is 4. |
| 28 | +ksft_skip=4 |
| 29 | +ret=0 |
| 30 | +GREEN='\033[0;92m' |
| 31 | +RED='\033[0;31m' |
| 32 | +NC='\033[0m' # No Color |
| 33 | + |
| 34 | +readonly port=8080 |
| 35 | + |
| 36 | +readonly vip_v4=207.175.44.110 |
| 37 | +readonly cip_v4=10.0.0.2 |
| 38 | +readonly gip_v4=10.0.0.1 |
| 39 | +readonly dip_v4=172.16.0.1 |
| 40 | +readonly rip_v4=172.16.0.2 |
| 41 | +readonly sip_v4=10.0.0.3 |
| 42 | + |
| 43 | +readonly infile="$(mktemp)" |
| 44 | +readonly outfile="$(mktemp)" |
| 45 | +readonly datalen=32 |
| 46 | + |
| 47 | +sysipvsnet="/proc/sys/net/ipv4/vs/" |
| 48 | +if [ ! -d $sysipvsnet ]; then |
| 49 | + modprobe -q ip_vs |
| 50 | + if [ $? -ne 0 ]; then |
| 51 | + echo "skip: could not run test without ipvs module" |
| 52 | + exit $ksft_skip |
| 53 | + fi |
| 54 | +fi |
| 55 | + |
| 56 | +ip -Version > /dev/null 2>&1 |
| 57 | +if [ $? -ne 0 ]; then |
| 58 | + echo "SKIP: Could not run test without ip tool" |
| 59 | + exit $ksft_skip |
| 60 | +fi |
| 61 | + |
| 62 | +ipvsadm -v > /dev/null 2>&1 |
| 63 | +if [ $? -ne 0 ]; then |
| 64 | + echo "SKIP: Could not run test without ipvsadm" |
| 65 | + exit $ksft_skip |
| 66 | +fi |
| 67 | + |
| 68 | +setup() { |
| 69 | + ip netns add ns0 |
| 70 | + ip netns add ns1 |
| 71 | + ip netns add ns2 |
| 72 | + |
| 73 | + ip link add veth01 netns ns0 type veth peer name veth10 netns ns1 |
| 74 | + ip link add veth02 netns ns0 type veth peer name veth20 netns ns2 |
| 75 | + ip link add veth12 netns ns1 type veth peer name veth21 netns ns2 |
| 76 | + |
| 77 | + ip netns exec ns0 ip link set veth01 up |
| 78 | + ip netns exec ns0 ip link set veth02 up |
| 79 | + ip netns exec ns0 ip link add br0 type bridge |
| 80 | + ip netns exec ns0 ip link set veth01 master br0 |
| 81 | + ip netns exec ns0 ip link set veth02 master br0 |
| 82 | + ip netns exec ns0 ip link set br0 up |
| 83 | + ip netns exec ns0 ip addr add ${cip_v4}/24 dev br0 |
| 84 | + |
| 85 | + ip netns exec ns1 ip link set lo up |
| 86 | + ip netns exec ns1 ip link set veth10 up |
| 87 | + ip netns exec ns1 ip addr add ${gip_v4}/24 dev veth10 |
| 88 | + ip netns exec ns1 ip link set veth12 up |
| 89 | + ip netns exec ns1 ip addr add ${dip_v4}/24 dev veth12 |
| 90 | + |
| 91 | + ip netns exec ns2 ip link set lo up |
| 92 | + ip netns exec ns2 ip link set veth21 up |
| 93 | + ip netns exec ns2 ip addr add ${rip_v4}/24 dev veth21 |
| 94 | + ip netns exec ns2 ip link set veth20 up |
| 95 | + ip netns exec ns2 ip addr add ${sip_v4}/24 dev veth20 |
| 96 | + |
| 97 | + sleep 1 |
| 98 | + |
| 99 | + dd if=/dev/urandom of="${infile}" bs="${datalen}" count=1 status=none |
| 100 | +} |
| 101 | + |
| 102 | +cleanup() { |
| 103 | + for i in 0 1 2 |
| 104 | + do |
| 105 | + ip netns del ns$i > /dev/null 2>&1 |
| 106 | + done |
| 107 | + |
| 108 | + if [ -f "${outfile}" ]; then |
| 109 | + rm "${outfile}" |
| 110 | + fi |
| 111 | + if [ -f "${infile}" ]; then |
| 112 | + rm "${infile}" |
| 113 | + fi |
| 114 | +} |
| 115 | + |
| 116 | +server_listen() { |
| 117 | + ip netns exec ns2 nc -l -p 8080 > "${outfile}" & |
| 118 | + server_pid=$! |
| 119 | + sleep 0.2 |
| 120 | +} |
| 121 | + |
| 122 | +client_connect() { |
| 123 | + ip netns exec ns0 timeout 2 nc -w 1 ${vip_v4} ${port} < "${infile}" |
| 124 | +} |
| 125 | + |
| 126 | +verify_data() { |
| 127 | + wait "${server_pid}" |
| 128 | + cmp "$infile" "$outfile" 2>/dev/null |
| 129 | +} |
| 130 | + |
| 131 | +test_service() { |
| 132 | + server_listen |
| 133 | + client_connect |
| 134 | + verify_data |
| 135 | +} |
| 136 | + |
| 137 | + |
| 138 | +test_dr() { |
| 139 | + ip netns exec ns0 ip route add ${vip_v4} via ${gip_v4} dev br0 |
| 140 | + |
| 141 | + ip netns exec ns1 sysctl -qw net.ipv4.ip_forward=1 |
| 142 | + ip netns exec ns1 ipvsadm -A -t ${vip_v4}:${port} -s rr |
| 143 | + ip netns exec ns1 ipvsadm -a -t ${vip_v4}:${port} -r ${rip_v4}:${port} |
| 144 | + ip netns exec ns1 ip addr add ${vip_v4}/32 dev lo:1 |
| 145 | + |
| 146 | + # avoid incorrect arp response |
| 147 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.all.arp_ignore=1 |
| 148 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.all.arp_announce=2 |
| 149 | + # avoid reverse route lookup |
| 150 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.all.rp_filter=0 |
| 151 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.veth21.rp_filter=0 |
| 152 | + ip netns exec ns2 ip addr add ${vip_v4}/32 dev lo:1 |
| 153 | + |
| 154 | + test_service |
| 155 | +} |
| 156 | + |
| 157 | +test_nat() { |
| 158 | + ip netns exec ns0 ip route add ${vip_v4} via ${gip_v4} dev br0 |
| 159 | + |
| 160 | + ip netns exec ns1 sysctl -qw net.ipv4.ip_forward=1 |
| 161 | + ip netns exec ns1 ipvsadm -A -t ${vip_v4}:${port} -s rr |
| 162 | + ip netns exec ns1 ipvsadm -a -m -t ${vip_v4}:${port} -r ${rip_v4}:${port} |
| 163 | + ip netns exec ns1 ip addr add ${vip_v4}/32 dev lo:1 |
| 164 | + |
| 165 | + ip netns exec ns2 ip link del veth20 |
| 166 | + ip netns exec ns2 ip route add default via ${dip_v4} dev veth21 |
| 167 | + |
| 168 | + test_service |
| 169 | +} |
| 170 | + |
| 171 | +test_tun() { |
| 172 | + ip netns exec ns0 ip route add ${vip_v4} via ${gip_v4} dev br0 |
| 173 | + |
| 174 | + ip netns exec ns1 modprobe ipip |
| 175 | + ip netns exec ns1 ip link set tunl0 up |
| 176 | + ip netns exec ns1 sysctl -qw net.ipv4.ip_forward=0 |
| 177 | + ip netns exec ns1 sysctl -qw net.ipv4.conf.all.send_redirects=0 |
| 178 | + ip netns exec ns1 sysctl -qw net.ipv4.conf.default.send_redirects=0 |
| 179 | + ip netns exec ns1 ipvsadm -A -t ${vip_v4}:${port} -s rr |
| 180 | + ip netns exec ns1 ipvsadm -a -i -t ${vip_v4}:${port} -r ${rip_v4}:${port} |
| 181 | + ip netns exec ns1 ip addr add ${vip_v4}/32 dev lo:1 |
| 182 | + |
| 183 | + ip netns exec ns2 modprobe ipip |
| 184 | + ip netns exec ns2 ip link set tunl0 up |
| 185 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.all.arp_ignore=1 |
| 186 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.all.arp_announce=2 |
| 187 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.all.rp_filter=0 |
| 188 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.tunl0.rp_filter=0 |
| 189 | + ip netns exec ns2 sysctl -qw net.ipv4.conf.veth21.rp_filter=0 |
| 190 | + ip netns exec ns2 ip addr add ${vip_v4}/32 dev lo:1 |
| 191 | + |
| 192 | + test_service |
| 193 | +} |
| 194 | + |
| 195 | +run_tests() { |
| 196 | + local errors= |
| 197 | + |
| 198 | + echo "Testing DR mode..." |
| 199 | + cleanup |
| 200 | + setup |
| 201 | + test_dr |
| 202 | + errors=$(( $errors + $? )) |
| 203 | + |
| 204 | + echo "Testing NAT mode..." |
| 205 | + cleanup |
| 206 | + setup |
| 207 | + test_nat |
| 208 | + errors=$(( $errors + $? )) |
| 209 | + |
| 210 | + echo "Testing Tunnel mode..." |
| 211 | + cleanup |
| 212 | + setup |
| 213 | + test_tun |
| 214 | + errors=$(( $errors + $? )) |
| 215 | + |
| 216 | + return $errors |
| 217 | +} |
| 218 | + |
| 219 | +trap cleanup EXIT |
| 220 | + |
| 221 | +run_tests |
| 222 | + |
| 223 | +if [ $? -ne 0 ]; then |
| 224 | + echo -e "$(basename $0): ${RED}FAIL${NC}" |
| 225 | + exit 1 |
| 226 | +fi |
| 227 | +echo -e "$(basename $0): ${GREEN}PASS${NC}" |
| 228 | +exit 0 |
0 commit comments