1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#!/bin/bash
set -e
__curdir=$(dirname $(realpath "${BASH_SOURCE[0]}"))
source "$__curdir/env/switch.bash"
setup() {
switch_env_setup --wg --wg-v6 --wg-mtu=1408
if [ "$1" = "mimic" ]; then
for _i in {0..1}; do
ip netns exec ${netns[$_i]} "$__curdir/../out/mimic" run ${veth[$_i]} \
"-flocal=[$(strip_ip_cidr ${veth_ipv6[$_i]})]:${wg_port[$_i]}" &
_mimic[$_i]=$!
done
sleep 1
fi
}
bench() {
local _type="$1"
shift
local _dest
if [ "$_type" = "veth" ]; then
_dest=$(strip_ip_cidr ${veth_ipv6[0]})
else
_dest=$(strip_ip_cidr ${wg_ipv6[0]})
fi
_iperf3_pid_tmp=$(mktemp)
ip netns exec ${netns[0]} iperf3 -s -D -I "$_iperf3_pid_tmp"
ip netns exec ${netns[1]} iperf3 -c $_dest $@
}
cleanup() (
set +e
for _i in {0..1}; do
[ -z ${_mimic[$_i]} ] || kill ${_mimic[$_i]}
done
wait
if [ -n "$_iperf3_pid_tmp" ]; then
kill $(cat $_iperf3_pid_tmp)
fi
switch_env_cleanup
)
case "$1" in
clean | cleanup)
cleanup
exit
;;
veth | wg | mimic)
_type=$1
shift
;;
*)
_type=mimic
;;
esac
if [ "$UID" -ne 0 ]; then
>&2 echo Benchmark needs to be run as root.
exit 1
fi
trap "cleanup; exit 1" SIGINT
_exitcode=0
[ $_exitcode -ne 0 ] || setup $_type || _exitcode=$?
[ $_exitcode -ne 0 ] || bench $_type $@ || _exitcode=$?
cleanup
exit $_exitcode
|
