Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3
4##############################################################################
5# Defines
6
7WAIT_TIMEOUT=${WAIT_TIMEOUT:=20}
8BUSYWAIT_TIMEOUT=$((WAIT_TIMEOUT * 1000)) # ms
9
10# Kselftest framework requirement - SKIP code is 4.
11ksft_skip=4
12# namespace list created by setup_ns
13NS_LIST=""
14
15##############################################################################
16# Helpers
17busywait()
18{
19 local timeout=$1; shift
20
21 local start_time="$(date -u +%s%3N)"
22 while true
23 do
24 local out
25 out=$("$@")
26 local ret=$?
27 if ((!ret)); then
28 echo -n "$out"
29 return 0
30 fi
31
32 local current_time="$(date -u +%s%3N)"
33 if ((current_time - start_time > timeout)); then
34 echo -n "$out"
35 return 1
36 fi
37 done
38}
39
40cleanup_ns()
41{
42 local ns=""
43 local errexit=0
44 local ret=0
45
46 # disable errexit temporary
47 if [[ $- =~ "e" ]]; then
48 errexit=1
49 set +e
50 fi
51
52 for ns in "$@"; do
53 ip netns delete "${ns}" &> /dev/null
54 if ! busywait $BUSYWAIT_TIMEOUT ip netns list \| grep -vq "^$ns$" &> /dev/null; then
55 echo "Warn: Failed to remove namespace $ns"
56 ret=1
57 fi
58 done
59
60 [ $errexit -eq 1 ] && set -e
61 return $ret
62}
63
64cleanup_all_ns()
65{
66 cleanup_ns $NS_LIST
67}
68
69# setup netns with given names as prefix. e.g
70# setup_ns local remote
71setup_ns()
72{
73 local ns=""
74 local ns_name=""
75 local ns_list=""
76 for ns_name in "$@"; do
77 # Some test may setup/remove same netns multi times
78 if unset ${ns_name} 2> /dev/null; then
79 ns="${ns_name,,}-$(mktemp -u XXXXXX)"
80 eval readonly ${ns_name}="$ns"
81 else
82 eval ns='$'${ns_name}
83 cleanup_ns "$ns"
84
85 fi
86
87 if ! ip netns add "$ns"; then
88 echo "Failed to create namespace $ns_name"
89 cleanup_ns "$ns_list"
90 return $ksft_skip
91 fi
92 ip -n "$ns" link set lo up
93 ns_list="$ns_list $ns"
94 done
95 NS_LIST="$NS_LIST $ns_list"
96}