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", 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 _set_offload_checksum(cfg, netnl, on) -> None:
54    try:
55        ethtool(f" -K {cfg.ifname} rx {on} tx {on} ")
56    except:
57        return
58
59def _set_xdp_generic_sb_on(cfg) -> None:
60    prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
61    cmd(f"ip link set dev {cfg.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    prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
70    cmd(f"ip link set dev {cfg.remote_ifname} mtu 9000", shell=True, host=cfg.remote)
71    defer(ip, f"link set dev {cfg.remote_ifname} mtu 1500", host=cfg.remote)
72    ip("link set dev %s mtu 9000 xdpgeneric obj %s sec xdp.frags" % (cfg.ifname, prog))
73    defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdpgeneric off")
74
75    if no_sleep != True:
76        time.sleep(10)
77
78def _set_xdp_native_sb_on(cfg) -> None:
79    prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
80    cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
81    cmd(f"ip -j link set dev {cfg.ifname} mtu 1500 xdp obj {prog} sec xdp", shell=True)
82    defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdp off")
83    xdp_info = ip("-d link show %s" % (cfg.ifname), json=True)[0]
84    if xdp_info['xdp']['mode'] != 1:
85        """
86        If the interface doesn't support native-mode, it falls back to generic mode.
87        The mode value 1 is native and 2 is generic.
88        So it raises an exception if mode is not 1(native mode).
89        """
90        raise KsftSkipEx('device does not support native-XDP')
91
92    if no_sleep != True:
93        time.sleep(10)
94
95def _set_xdp_native_mb_on(cfg) -> None:
96    prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
97    cmd(f"ip link set dev {cfg.remote_ifname} mtu 9000", shell=True, host=cfg.remote)
98    defer(ip, f"link set dev {cfg.remote_ifname} mtu 1500", host=cfg.remote)
99    try:
100        cmd(f"ip link set dev {cfg.ifname} mtu 9000 xdp obj {prog} sec xdp.frags", shell=True)
101        defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdp off")
102    except Exception as e:
103        raise KsftSkipEx('device does not support native-multi-buffer XDP')
104
105    if no_sleep != True:
106        time.sleep(10)
107
108def _set_xdp_offload_on(cfg) -> None:
109    prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
110    cmd(f"ip link set dev {cfg.ifname} mtu 1500", shell=True)
111    try:
112        cmd(f"ip link set dev {cfg.ifname} xdpoffload obj {prog} sec xdp", shell=True)
113    except Exception as e:
114        raise KsftSkipEx('device does not support offloaded XDP')
115    defer(ip, f"link set dev {cfg.ifname} xdpoffload off")
116    cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
117
118    if no_sleep != True:
119        time.sleep(10)
120
121def get_interface_info(cfg) -> None:
122    global no_sleep
123
124    if cfg.remote_ifname == "":
125        raise KsftFailEx('Can not get remote interface')
126    local_info = ip("-d link show %s" % (cfg.ifname), json=True)[0]
127    if 'parentbus' in local_info and local_info['parentbus'] == "netdevsim":
128        no_sleep=True
129    if 'linkinfo' in local_info and local_info['linkinfo']['info_kind'] == "veth":
130        no_sleep=True
131
132def set_interface_init(cfg) -> None:
133    cmd(f"ip link set dev {cfg.ifname} mtu 1500", shell=True)
134    cmd(f"ip link set dev {cfg.ifname} xdp off ", shell=True)
135    cmd(f"ip link set dev {cfg.ifname} xdpgeneric off ", shell=True)
136    cmd(f"ip link set dev {cfg.ifname} xdpoffload off", shell=True)
137    cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
138
139def test_default_v4(cfg, netnl) -> None:
140    cfg.require_ipver("4")
141
142    _set_offload_checksum(cfg, netnl, "off")
143    _test_v4(cfg)
144    _test_tcp(cfg)
145    _set_offload_checksum(cfg, netnl, "on")
146    _test_v4(cfg)
147    _test_tcp(cfg)
148
149def test_default_v6(cfg, netnl) -> None:
150    cfg.require_ipver("6")
151
152    _set_offload_checksum(cfg, netnl, "off")
153    _test_v6(cfg)
154    _test_tcp(cfg)
155    _set_offload_checksum(cfg, netnl, "on")
156    _test_v6(cfg)
157    _test_tcp(cfg)
158
159def test_xdp_generic_sb(cfg, netnl) -> None:
160    _set_xdp_generic_sb_on(cfg)
161    _set_offload_checksum(cfg, netnl, "off")
162    _test_v4(cfg)
163    _test_v6(cfg)
164    _test_tcp(cfg)
165    _set_offload_checksum(cfg, netnl, "on")
166    _test_v4(cfg)
167    _test_v6(cfg)
168    _test_tcp(cfg)
169
170def test_xdp_generic_mb(cfg, netnl) -> None:
171    _set_xdp_generic_mb_on(cfg)
172    _set_offload_checksum(cfg, netnl, "off")
173    _test_v4(cfg)
174    _test_v6(cfg)
175    _test_tcp(cfg)
176    _set_offload_checksum(cfg, netnl, "on")
177    _test_v4(cfg)
178    _test_v6(cfg)
179    _test_tcp(cfg)
180
181def test_xdp_native_sb(cfg, netnl) -> None:
182    _set_xdp_native_sb_on(cfg)
183    _set_offload_checksum(cfg, netnl, "off")
184    _test_v4(cfg)
185    _test_v6(cfg)
186    _test_tcp(cfg)
187    _set_offload_checksum(cfg, netnl, "on")
188    _test_v4(cfg)
189    _test_v6(cfg)
190    _test_tcp(cfg)
191
192def test_xdp_native_mb(cfg, netnl) -> None:
193    _set_xdp_native_mb_on(cfg)
194    _set_offload_checksum(cfg, netnl, "off")
195    _test_v4(cfg)
196    _test_v6(cfg)
197    _test_tcp(cfg)
198    _set_offload_checksum(cfg, netnl, "on")
199    _test_v4(cfg)
200    _test_v6(cfg)
201    _test_tcp(cfg)
202
203def test_xdp_offload(cfg, netnl) -> None:
204    _set_xdp_offload_on(cfg)
205    _test_v4(cfg)
206    _test_v6(cfg)
207    _test_tcp(cfg)
208
209def main() -> None:
210    with NetDrvEpEnv(__file__) as cfg:
211        get_interface_info(cfg)
212        set_interface_init(cfg)
213        ksft_run([test_default_v4,
214                  test_default_v6,
215                  test_xdp_generic_sb,
216                  test_xdp_generic_mb,
217                  test_xdp_native_sb,
218                  test_xdp_native_mb,
219                  test_xdp_offload],
220                 args=(cfg, EthtoolFamily()))
221    ksft_exit()
222
223
224if __name__ == "__main__":
225    main()
226