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 _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() 242