]>
Commit | Line | Data |
---|---|---|
20ab58c7 SH |
1 | #!/bin/bash |
2 | ||
3 | # lxc: linux Container library | |
4 | ||
5 | # Authors: | |
6 | # Serge Hallyn <serge.hallyn@ubuntu.com> | |
7 | # | |
8 | # This is a test script for the lxc-user-nic program | |
9 | ||
10 | # This library is free software; you can redistribute it and/or | |
11 | # modify it under the terms of the GNU Lesser General Public | |
12 | # License as published by the Free Software Foundation; either | |
13 | # version 2.1 of the License, or (at your option) any later version. | |
14 | ||
15 | # This library is distributed in the hope that it will be useful, | |
16 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
17 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
18 | # Lesser General Public License for more details. | |
19 | ||
20 | # You should have received a copy of the GNU Lesser General Public | |
21 | # License along with this library; if not, write to the Free Software | |
250b1eec | 22 | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
20ab58c7 | 23 | |
061ba5d0 SG |
24 | # This test assumes an Ubuntu host |
25 | ||
0e17b9c6 | 26 | DONE=0 |
8cb8e496 | 27 | LXC_USER_NIC="@LIBEXECDIR@/lxc/lxc-user-nic" |
0e17b9c6 | 28 | |
8befa924 | 29 | cleanup() { |
0e17b9c6 SG |
30 | ( |
31 | set +e | |
32 | ||
c85a40bc | 33 | lxc-stop -n usernic-c1 -k |
0e17b9c6 SG |
34 | lxc-destroy -n usernic-c1 |
35 | ||
061ba5d0 | 36 | sed -i '/usernic-user/d' /run/lxc/nics /etc/lxc/lxc-usernet |
0e17b9c6 SG |
37 | ifconfig usernic-br0 down |
38 | ifconfig usernic-br1 down | |
39 | brctl delbr usernic-br0 | |
40 | brctl delbr usernic-br1 | |
41 | ||
c85a40bc | 42 | run_cmd "lxc-stop -n b1 -k" |
0e17b9c6 SG |
43 | pkill -u $(id -u usernic-user) -9 |
44 | ||
45 | rm -rf /tmp/usernic-test /home/usernic-user /run/user/$(id -u usernic-user) | |
46 | ||
47 | deluser usernic-user | |
48 | ) >/dev/null 2>&1 | |
49 | ||
50 | if [ "$DONE" = "1" ]; then | |
51 | echo "PASS" | |
52 | exit 0 | |
53 | fi | |
54 | ||
55 | echo "FAIL" | |
56 | exit 1 | |
57 | } | |
58 | ||
59 | run_cmd() { | |
198a3f10 SG |
60 | sudo -i -u usernic-user \ |
61 | env http_proxy=${http_proxy:-} https_proxy=${https_proxy:-} \ | |
62 | XDG_RUNTIME_DIR=/run/user/$(id -u usernic-user) $* | |
8befa924 | 63 | } |
20ab58c7 | 64 | |
fd2b7320 SG |
65 | ARCH=i386 |
66 | if type dpkg >/dev/null 2>&1; then | |
67 | ARCH=$(dpkg --print-architecture) | |
68 | fi | |
69 | ||
0e17b9c6 SG |
70 | set -eu |
71 | trap cleanup EXIT SIGHUP SIGINT SIGTERM | |
72 | ||
8befa924 SH |
73 | # create a test user |
74 | deluser usernic-user || true | |
75 | useradd usernic-user | |
76 | sudo mkdir -p /home/usernic-user | |
79d88b03 | 77 | sudo chown usernic-user: /home/usernic-user |
8befa924 | 78 | usermod -v 910000-919999 -w 910000-919999 usernic-user |
8befa924 | 79 | |
0e17b9c6 SG |
80 | mkdir -p /home/usernic-user/.config/lxc/ |
81 | cat > /home/usernic-user/.config/lxc/default.conf << EOF | |
8befa924 | 82 | lxc.network.type = empty |
0e17b9c6 SG |
83 | lxc.id_map = u 0 910000 10000 |
84 | lxc.id_map = g 0 910000 10000 | |
8befa924 | 85 | EOF |
20ab58c7 | 86 | |
42e5c987 | 87 | if which cgm >/dev/null 2>&1; then |
2ba5eb93 SG |
88 | cgm create all usernic-user |
89 | cgm chown all usernic-user $(id -u usernic-user) $(id -g usernic-user) | |
90 | cgm movepid all usernic-user $$ | |
42e5c987 | 91 | elif [ -e /sys/fs/cgroup/cgmanager/sock ]; then |
ef4deb7f | 92 | for d in $(cut -d : -f 2 /proc/self/cgroup); do |
3f458ed0 SG |
93 | dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \ |
94 | --type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.Create \ | |
95 | string:$d string:usernic-user >/dev/null | |
96 | ||
97 | dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \ | |
98 | --type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.Chown \ | |
99 | string:$d string:usernic-user int32:$(id -u usernic-user) int32:$(id -g usernic-user) >/dev/null | |
100 | ||
101 | dbus-send --print-reply --address=unix:path=/sys/fs/cgroup/cgmanager/sock \ | |
102 | --type=method_call /org/linuxcontainers/cgmanager org.linuxcontainers.cgmanager0_0.MovePid \ | |
103 | string:$d string:usernic-user int32:$$ >/dev/null | |
104 | done | |
105 | else | |
106 | for d in /sys/fs/cgroup/*; do | |
107 | [ ! -d $d/lxctest ] && mkdir $d/lxctest | |
79d88b03 | 108 | chown -R usernic-user: $d/lxctest |
3f458ed0 SG |
109 | echo $$ > $d/lxctest/tasks |
110 | done | |
111 | fi | |
0e17b9c6 SG |
112 | |
113 | mkdir -p /run/user/$(id -u usernic-user) | |
79d88b03 | 114 | chown -R usernic-user: /run/user/$(id -u usernic-user) /home/usernic-user |
20ab58c7 | 115 | |
57221f67 SG |
116 | # Copy the download template cache if available |
117 | run_cmd "mkdir -p /home/usernic-user/.cache/lxc" | |
118 | [ -d /var/cache/lxc/download ] && \ | |
119 | cp -R /var/cache/lxc/download /home/usernic-user/.cache/lxc && \ | |
120 | chown -R usernic-user: /home/usernic-user/.cache/lxc | |
121 | ||
122 | ||
0e17b9c6 | 123 | # Create two test bridges |
8befa924 SH |
124 | brctl addbr usernic-br0 |
125 | brctl addbr usernic-br1 | |
126 | ifconfig usernic-br0 0.0.0.0 up | |
127 | ifconfig usernic-br1 0.0.0.0 up | |
128 | ||
fd2b7320 SG |
129 | ARCH=i386 |
130 | if type dpkg >/dev/null 2>&1; then | |
131 | ARCH=$(dpkg --print-architecture) | |
132 | fi | |
133 | ||
8befa924 | 134 | # Create three containers |
fd2b7320 | 135 | run_cmd "lxc-create -t download -n b1 -- -d ubuntu -r trusty -a $ARCH" |
0e17b9c6 SG |
136 | run_cmd "lxc-start -n b1 -d" |
137 | p1=$(run_cmd "lxc-info -n b1 -p -H") | |
8befa924 | 138 | |
615af4ac SH |
139 | lxcpath=/home/usernic-user/.local/share/lxc |
140 | lxcname=b1 | |
141 | ||
8befa924 | 142 | # Assign one veth, should fail as no allowed entries yet |
615af4ac | 143 | if run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p1 veth usernic-br0 xx1"; then |
8befa924 | 144 | echo "FAIL: able to create nic with no entries" |
0e17b9c6 | 145 | exit 1 |
20ab58c7 SH |
146 | fi |
147 | ||
8befa924 | 148 | # Give him a quota of two |
d08363af | 149 | touch /etc/lxc/lxc-usernet |
0e17b9c6 SG |
150 | sed -i '/^usernic-user/d' /etc/lxc/lxc-usernet |
151 | echo "usernic-user veth usernic-br0 2" >> /etc/lxc/lxc-usernet | |
8befa924 SH |
152 | |
153 | # Assign one veth to second bridge, should fail | |
615af4ac | 154 | if run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p1 veth usernic-br1 xx1"; then |
8befa924 | 155 | echo "FAIL: able to create nic with no entries" |
0e17b9c6 | 156 | exit 1 |
8befa924 | 157 | fi |
20ab58c7 | 158 | |
8befa924 | 159 | # Assign two veths, should succeed |
615af4ac | 160 | if ! run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p1 veth usernic-br0 xx2"; then |
8befa924 | 161 | echo "FAIL: unable to create first nic" |
0e17b9c6 | 162 | exit 1 |
8befa924 | 163 | fi |
0e17b9c6 | 164 | |
615af4ac | 165 | if ! run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p1 veth usernic-br0 xx3"; then |
8befa924 | 166 | echo "FAIL: unable to create second nic" |
0e17b9c6 | 167 | exit 1 |
20ab58c7 SH |
168 | fi |
169 | ||
8befa924 | 170 | # Assign one more veth, should fail. |
615af4ac | 171 | if run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p1 veth usernic-br0 xx4"; then |
8befa924 | 172 | echo "FAIL: able to create third nic" |
0e17b9c6 | 173 | exit 1 |
20ab58c7 SH |
174 | fi |
175 | ||
8befa924 | 176 | # Shut down and restart the container, should be able to assign more nics |
c85a40bc | 177 | run_cmd "lxc-stop -n b1 -k" |
0e17b9c6 SG |
178 | run_cmd "lxc-start -n b1 -d" |
179 | p1=$(run_cmd "lxc-info -n b1 -p -H") | |
180 | ||
615af4ac | 181 | if ! run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p1 veth usernic-br0 xx5"; then |
8befa924 SH |
182 | echo "FAIL: unable to create nic after destroying the old" |
183 | cleanup 1 | |
184 | fi | |
185 | ||
c85a40bc | 186 | run_cmd "lxc-stop -n b1 -k" |
8befa924 SH |
187 | |
188 | # Create a root-owned ns | |
189 | lxc-create -t busybox -n usernic-c1 | |
190 | lxc-start -n usernic-c1 -d | |
0e17b9c6 | 191 | p2=$(lxc-info -n usernic-c1 -p -H) |
8befa924 SH |
192 | |
193 | # assign veth to it - should fail | |
615af4ac | 194 | if run_cmd "$LXC_USER_NIC $lxcpath $lxcname $p2 veth usernic-br0 xx6"; then |
8befa924 SH |
195 | echo "FAIL: able to attach nic to root-owned container" |
196 | cleanup 1 | |
20ab58c7 SH |
197 | fi |
198 | ||
199 | echo "All tests passed" | |
0e17b9c6 | 200 | DONE=1 |