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_ipver("4") 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 cfg.require_ipver("6") 25 26 cmd("ping -c 1 -W5 " + cfg.remote_addr_v["6"]) 27 cmd("ping -c 1 -W5 " + cfg.addr_v["6"], host=cfg.remote) 28 cmd("ping -s 65000 -c 1 -W0.5 " + cfg.remote_addr_v["6"]) 29 cmd("ping -s 65000 -c 1 -W0.5 " + cfg.addr_v["6"], 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_addr_v['4']} | 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() 222