Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1#!/usr/bin/env python3
2# SPDX-License-Identifier: GPL-2.0
3
4import os
5import random, string, time
6from lib.py import ksft_run, ksft_exit
7from lib.py import ksft_eq, KsftSkipEx, KsftFailEx
8from lib.py import EthtoolFamily, NetDrvEpEnv
9from lib.py import bkg, cmd, wait_port_listen, rand_port
10from lib.py import defer, ethtool, ip
11
12remote_ifname=""
13no_sleep=False
14
15def _test_v4(cfg) -> None:
16 cfg.require_v4()
17
18 cmd(f"ping -c 1 -W0.5 {cfg.remote_v4}")
19 cmd(f"ping -c 1 -W0.5 {cfg.v4}", host=cfg.remote)
20 cmd(f"ping -s 65000 -c 1 -W0.5 {cfg.remote_v4}")
21 cmd(f"ping -s 65000 -c 1 -W0.5 {cfg.v4}", host=cfg.remote)
22
23def _test_v6(cfg) -> None:
24 cfg.require_v6()
25
26 cmd(f"ping -c 1 -W5 {cfg.remote_v6}")
27 cmd(f"ping -c 1 -W5 {cfg.v6}", host=cfg.remote)
28 cmd(f"ping -s 65000 -c 1 -W0.5 {cfg.remote_v6}")
29 cmd(f"ping -s 65000 -c 1 -W0.5 {cfg.v6}", host=cfg.remote)
30
31def _test_tcp(cfg) -> None:
32 cfg.require_cmd("socat", remote=True)
33
34 port = rand_port()
35 listen_cmd = f"socat -{cfg.addr_ipver} -t 2 -u TCP-LISTEN:{port},reuseport STDOUT"
36
37 test_string = ''.join(random.choice(string.ascii_lowercase) for _ in range(65536))
38 with bkg(listen_cmd, exit_wait=True) as nc:
39 wait_port_listen(port)
40
41 cmd(f"echo {test_string} | socat -t 2 -u STDIN TCP:{cfg.baddr}:{port}",
42 shell=True, host=cfg.remote)
43 ksft_eq(nc.stdout.strip(), test_string)
44
45 test_string = ''.join(random.choice(string.ascii_lowercase) for _ in range(65536))
46 with bkg(listen_cmd, host=cfg.remote, exit_wait=True) as nc:
47 wait_port_listen(port, host=cfg.remote)
48
49 cmd(f"echo {test_string} | socat -t 2 -u STDIN TCP:{cfg.remote_baddr}:{port}", shell=True)
50 ksft_eq(nc.stdout.strip(), test_string)
51
52def _set_offload_checksum(cfg, netnl, on) -> None:
53 try:
54 ethtool(f" -K {cfg.ifname} rx {on} tx {on} ")
55 except:
56 return
57
58def _set_xdp_generic_sb_on(cfg) -> None:
59 test_dir = os.path.dirname(os.path.realpath(__file__))
60 prog = test_dir + "/../../net/lib/xdp_dummy.bpf.o"
61 cmd(f"ip link set dev {remote_ifname} mtu 1500", shell=True, host=cfg.remote)
62 cmd(f"ip link set dev {cfg.ifname} mtu 1500 xdpgeneric obj {prog} sec xdp", shell=True)
63 defer(cmd, f"ip link set dev {cfg.ifname} xdpgeneric off")
64
65 if no_sleep != True:
66 time.sleep(10)
67
68def _set_xdp_generic_mb_on(cfg) -> None:
69 test_dir = os.path.dirname(os.path.realpath(__file__))
70 prog = test_dir + "/../../net/lib/xdp_dummy.bpf.o"
71 cmd(f"ip link set dev {remote_ifname} mtu 9000", shell=True, host=cfg.remote)
72 defer(ip, f"link set dev {remote_ifname} mtu 1500", host=cfg.remote)
73 ip("link set dev %s mtu 9000 xdpgeneric obj %s sec xdp.frags" % (cfg.ifname, prog))
74 defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdpgeneric off")
75
76 if no_sleep != True:
77 time.sleep(10)
78
79def _set_xdp_native_sb_on(cfg) -> None:
80 test_dir = os.path.dirname(os.path.realpath(__file__))
81 prog = test_dir + "/../../net/lib/xdp_dummy.bpf.o"
82 cmd(f"ip link set dev {remote_ifname} mtu 1500", shell=True, host=cfg.remote)
83 cmd(f"ip -j link set dev {cfg.ifname} mtu 1500 xdp obj {prog} sec xdp", shell=True)
84 defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdp off")
85 xdp_info = ip("-d link show %s" % (cfg.ifname), json=True)[0]
86 if xdp_info['xdp']['mode'] != 1:
87 """
88 If the interface doesn't support native-mode, it falls back to generic mode.
89 The mode value 1 is native and 2 is generic.
90 So it raises an exception if mode is not 1(native mode).
91 """
92 raise KsftSkipEx('device does not support native-XDP')
93
94 if no_sleep != True:
95 time.sleep(10)
96
97def _set_xdp_native_mb_on(cfg) -> None:
98 test_dir = os.path.dirname(os.path.realpath(__file__))
99 prog = test_dir + "/../../net/lib/xdp_dummy.bpf.o"
100 cmd(f"ip link set dev {remote_ifname} mtu 9000", shell=True, host=cfg.remote)
101 defer(ip, f"link set dev {remote_ifname} mtu 1500", host=cfg.remote)
102 try:
103 cmd(f"ip link set dev {cfg.ifname} mtu 9000 xdp obj {prog} sec xdp.frags", shell=True)
104 defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdp off")
105 except Exception as e:
106 raise KsftSkipEx('device does not support native-multi-buffer XDP')
107
108 if no_sleep != True:
109 time.sleep(10)
110
111def _set_xdp_offload_on(cfg) -> None:
112 test_dir = os.path.dirname(os.path.realpath(__file__))
113 prog = test_dir + "/../../net/lib/xdp_dummy.bpf.o"
114 cmd(f"ip link set dev {cfg.ifname} mtu 1500", shell=True)
115 try:
116 cmd(f"ip link set dev {cfg.ifname} xdpoffload obj {prog} sec xdp", shell=True)
117 except Exception as e:
118 raise KsftSkipEx('device does not support offloaded XDP')
119 defer(ip, f"link set dev {cfg.ifname} xdpoffload off")
120 cmd(f"ip link set dev {remote_ifname} mtu 1500", shell=True, host=cfg.remote)
121
122 if no_sleep != True:
123 time.sleep(10)
124
125def get_interface_info(cfg) -> None:
126 global remote_ifname
127 global no_sleep
128
129 remote_info = cmd(f"ip -4 -o addr show to {cfg.remote_v4} | awk '{{print $2}}'", shell=True, host=cfg.remote).stdout
130 remote_ifname = remote_info.rstrip('\n')
131 if remote_ifname == "":
132 raise KsftFailEx('Can not get remote interface')
133 local_info = ip("-d link show %s" % (cfg.ifname), json=True)[0]
134 if 'parentbus' in local_info and local_info['parentbus'] == "netdevsim":
135 no_sleep=True
136 if 'linkinfo' in local_info and local_info['linkinfo']['info_kind'] == "veth":
137 no_sleep=True
138
139def set_interface_init(cfg) -> None:
140 cmd(f"ip link set dev {cfg.ifname} mtu 1500", shell=True)
141 cmd(f"ip link set dev {cfg.ifname} xdp off ", shell=True)
142 cmd(f"ip link set dev {cfg.ifname} xdpgeneric off ", shell=True)
143 cmd(f"ip link set dev {cfg.ifname} xdpoffload off", shell=True)
144 cmd(f"ip link set dev {remote_ifname} mtu 1500", shell=True, host=cfg.remote)
145
146def test_default(cfg, netnl) -> None:
147 _set_offload_checksum(cfg, netnl, "off")
148 _test_v4(cfg)
149 _test_v6(cfg)
150 _test_tcp(cfg)
151 _set_offload_checksum(cfg, netnl, "on")
152 _test_v4(cfg)
153 _test_v6(cfg)
154 _test_tcp(cfg)
155
156def test_xdp_generic_sb(cfg, netnl) -> None:
157 _set_xdp_generic_sb_on(cfg)
158 _set_offload_checksum(cfg, netnl, "off")
159 _test_v4(cfg)
160 _test_v6(cfg)
161 _test_tcp(cfg)
162 _set_offload_checksum(cfg, netnl, "on")
163 _test_v4(cfg)
164 _test_v6(cfg)
165 _test_tcp(cfg)
166
167def test_xdp_generic_mb(cfg, netnl) -> None:
168 _set_xdp_generic_mb_on(cfg)
169 _set_offload_checksum(cfg, netnl, "off")
170 _test_v4(cfg)
171 _test_v6(cfg)
172 _test_tcp(cfg)
173 _set_offload_checksum(cfg, netnl, "on")
174 _test_v4(cfg)
175 _test_v6(cfg)
176 _test_tcp(cfg)
177
178def test_xdp_native_sb(cfg, netnl) -> None:
179 _set_xdp_native_sb_on(cfg)
180 _set_offload_checksum(cfg, netnl, "off")
181 _test_v4(cfg)
182 _test_v6(cfg)
183 _test_tcp(cfg)
184 _set_offload_checksum(cfg, netnl, "on")
185 _test_v4(cfg)
186 _test_v6(cfg)
187 _test_tcp(cfg)
188
189def test_xdp_native_mb(cfg, netnl) -> None:
190 _set_xdp_native_mb_on(cfg)
191 _set_offload_checksum(cfg, netnl, "off")
192 _test_v4(cfg)
193 _test_v6(cfg)
194 _test_tcp(cfg)
195 _set_offload_checksum(cfg, netnl, "on")
196 _test_v4(cfg)
197 _test_v6(cfg)
198 _test_tcp(cfg)
199
200def test_xdp_offload(cfg, netnl) -> None:
201 _set_xdp_offload_on(cfg)
202 _test_v4(cfg)
203 _test_v6(cfg)
204 _test_tcp(cfg)
205
206def main() -> None:
207 with NetDrvEpEnv(__file__) as cfg:
208 get_interface_info(cfg)
209 set_interface_init(cfg)
210 ksft_run([test_default,
211 test_xdp_generic_sb,
212 test_xdp_generic_mb,
213 test_xdp_native_sb,
214 test_xdp_native_mb,
215 test_xdp_offload],
216 args=(cfg, EthtoolFamily()))
217 ksft_exit()
218
219
220if __name__ == "__main__":
221 main()