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