mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-08-05 16:54:27 +00:00
d74cd9a02f
Make require_cmd() calls explicit about whether commands are needed locally, remotely, or both. Since require_cmd() defaults to local=True, tests should explicitly set local=False when commands are only needed remotely. - socat: Set local=False since it's only needed on remote hosts. - iperf3: Use single call with both local=True and remote=True since it's needed on both hosts. This avoids unnecessary test failures when commands are missing locally but available remotely where actually needed, and consolidates a duplicate require_cmd() call into single call that checks both hosts. Fixes:0d0f4174f6("selftests: drv-net: add a simple TSO test") Fixes:f1e68a1a4a("selftests: drv-net: add require_XYZ() helpers for validating env") Fixes:c76bab22e9("selftests: drv-net: rss_input_xfrm: Check test prerequisites before running") Reviewed-by: Nimrod Oren <noren@nvidia.com> Signed-off-by: Gal Pressman <gal@nvidia.com> Link: https://patch.msgid.link/20250723135454.649342-3-gal@nvidia.com Signed-off-by: Jakub Kicinski <kuba@kernel.org>
241 lines
8.1 KiB
Python
Executable file
241 lines
8.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: GPL-2.0
|
|
|
|
import os
|
|
import random, string, time
|
|
from lib.py import ksft_run, ksft_exit
|
|
from lib.py import ksft_eq, KsftSkipEx, KsftFailEx
|
|
from lib.py import EthtoolFamily, NetDrvEpEnv
|
|
from lib.py import bkg, cmd, wait_port_listen, rand_port
|
|
from lib.py import defer, ethtool, ip
|
|
|
|
no_sleep=False
|
|
|
|
def _test_v4(cfg) -> None:
|
|
if not cfg.addr_v["4"]:
|
|
return
|
|
|
|
cmd("ping -c 1 -W0.5 " + cfg.remote_addr_v["4"])
|
|
cmd("ping -c 1 -W0.5 " + cfg.addr_v["4"], host=cfg.remote)
|
|
cmd("ping -s 65000 -c 1 -W0.5 " + cfg.remote_addr_v["4"])
|
|
cmd("ping -s 65000 -c 1 -W0.5 " + cfg.addr_v["4"], host=cfg.remote)
|
|
|
|
def _test_v6(cfg) -> None:
|
|
if not cfg.addr_v["6"]:
|
|
return
|
|
|
|
cmd("ping -c 1 -W5 " + cfg.remote_addr_v["6"])
|
|
cmd("ping -c 1 -W5 " + cfg.addr_v["6"], host=cfg.remote)
|
|
cmd("ping -s 65000 -c 1 -W0.5 " + cfg.remote_addr_v["6"])
|
|
cmd("ping -s 65000 -c 1 -W0.5 " + cfg.addr_v["6"], host=cfg.remote)
|
|
|
|
def _test_tcp(cfg) -> None:
|
|
cfg.require_cmd("socat", local=False, remote=True)
|
|
|
|
port = rand_port()
|
|
listen_cmd = f"socat -{cfg.addr_ipver} -t 2 -u TCP-LISTEN:{port},reuseport STDOUT"
|
|
|
|
test_string = ''.join(random.choice(string.ascii_lowercase) for _ in range(65536))
|
|
with bkg(listen_cmd, exit_wait=True) as nc:
|
|
wait_port_listen(port)
|
|
|
|
cmd(f"echo {test_string} | socat -t 2 -u STDIN TCP:{cfg.baddr}:{port}",
|
|
shell=True, host=cfg.remote)
|
|
ksft_eq(nc.stdout.strip(), test_string)
|
|
|
|
test_string = ''.join(random.choice(string.ascii_lowercase) for _ in range(65536))
|
|
with bkg(listen_cmd, host=cfg.remote, exit_wait=True) as nc:
|
|
wait_port_listen(port, host=cfg.remote)
|
|
|
|
cmd(f"echo {test_string} | socat -t 2 -u STDIN TCP:{cfg.remote_baddr}:{port}", shell=True)
|
|
ksft_eq(nc.stdout.strip(), test_string)
|
|
|
|
def _schedule_checksum_reset(cfg, netnl) -> None:
|
|
features = ethtool(f"-k {cfg.ifname}", json=True)
|
|
setting = ""
|
|
for side in ["tx", "rx"]:
|
|
f = features[0][side + "-checksumming"]
|
|
if not f["fixed"]:
|
|
setting += " " + side
|
|
setting += " " + ("on" if f["requested"] or f["active"] else "off")
|
|
defer(ethtool, f" -K {cfg.ifname} " + setting)
|
|
|
|
def _set_offload_checksum(cfg, netnl, on) -> None:
|
|
try:
|
|
ethtool(f" -K {cfg.ifname} rx {on} tx {on} ")
|
|
except:
|
|
return
|
|
|
|
def _set_xdp_generic_sb_on(cfg) -> None:
|
|
prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
|
|
cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
|
|
cmd(f"ip link set dev {cfg.ifname} mtu 1500 xdpgeneric obj {prog} sec xdp", shell=True)
|
|
defer(cmd, f"ip link set dev {cfg.ifname} xdpgeneric off")
|
|
|
|
if no_sleep != True:
|
|
time.sleep(10)
|
|
|
|
def _set_xdp_generic_mb_on(cfg) -> None:
|
|
prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
|
|
cmd(f"ip link set dev {cfg.remote_ifname} mtu 9000", shell=True, host=cfg.remote)
|
|
defer(ip, f"link set dev {cfg.remote_ifname} mtu 1500", host=cfg.remote)
|
|
ip("link set dev %s mtu 9000 xdpgeneric obj %s sec xdp.frags" % (cfg.ifname, prog))
|
|
defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdpgeneric off")
|
|
|
|
if no_sleep != True:
|
|
time.sleep(10)
|
|
|
|
def _set_xdp_native_sb_on(cfg) -> None:
|
|
prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
|
|
cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
|
|
cmd(f"ip -j link set dev {cfg.ifname} mtu 1500 xdp obj {prog} sec xdp", shell=True)
|
|
defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdp off")
|
|
xdp_info = ip("-d link show %s" % (cfg.ifname), json=True)[0]
|
|
if xdp_info['xdp']['mode'] != 1:
|
|
"""
|
|
If the interface doesn't support native-mode, it falls back to generic mode.
|
|
The mode value 1 is native and 2 is generic.
|
|
So it raises an exception if mode is not 1(native mode).
|
|
"""
|
|
raise KsftSkipEx('device does not support native-XDP')
|
|
|
|
if no_sleep != True:
|
|
time.sleep(10)
|
|
|
|
def _set_xdp_native_mb_on(cfg) -> None:
|
|
prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
|
|
cmd(f"ip link set dev {cfg.remote_ifname} mtu 9000", shell=True, host=cfg.remote)
|
|
defer(ip, f"link set dev {cfg.remote_ifname} mtu 1500", host=cfg.remote)
|
|
try:
|
|
cmd(f"ip link set dev {cfg.ifname} mtu 9000 xdp obj {prog} sec xdp.frags", shell=True)
|
|
defer(ip, f"link set dev {cfg.ifname} mtu 1500 xdp off")
|
|
except Exception as e:
|
|
raise KsftSkipEx('device does not support native-multi-buffer XDP')
|
|
|
|
if no_sleep != True:
|
|
time.sleep(10)
|
|
|
|
def _set_xdp_offload_on(cfg) -> None:
|
|
prog = cfg.net_lib_dir / "xdp_dummy.bpf.o"
|
|
cmd(f"ip link set dev {cfg.ifname} mtu 1500", shell=True)
|
|
try:
|
|
cmd(f"ip link set dev {cfg.ifname} xdpoffload obj {prog} sec xdp", shell=True)
|
|
except Exception as e:
|
|
raise KsftSkipEx('device does not support offloaded XDP')
|
|
defer(ip, f"link set dev {cfg.ifname} xdpoffload off")
|
|
cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
|
|
|
|
if no_sleep != True:
|
|
time.sleep(10)
|
|
|
|
def get_interface_info(cfg) -> None:
|
|
global no_sleep
|
|
|
|
if cfg.remote_ifname == "":
|
|
raise KsftFailEx('Can not get remote interface')
|
|
local_info = ip("-d link show %s" % (cfg.ifname), json=True)[0]
|
|
if 'parentbus' in local_info and local_info['parentbus'] == "netdevsim":
|
|
no_sleep=True
|
|
if 'linkinfo' in local_info and local_info['linkinfo']['info_kind'] == "veth":
|
|
no_sleep=True
|
|
|
|
def set_interface_init(cfg) -> None:
|
|
cmd(f"ip link set dev {cfg.ifname} mtu 1500", shell=True)
|
|
cmd(f"ip link set dev {cfg.ifname} xdp off ", shell=True)
|
|
cmd(f"ip link set dev {cfg.ifname} xdpgeneric off ", shell=True)
|
|
cmd(f"ip link set dev {cfg.ifname} xdpoffload off", shell=True)
|
|
cmd(f"ip link set dev {cfg.remote_ifname} mtu 1500", shell=True, host=cfg.remote)
|
|
|
|
def test_default_v4(cfg, netnl) -> None:
|
|
cfg.require_ipver("4")
|
|
|
|
_schedule_checksum_reset(cfg, netnl)
|
|
_set_offload_checksum(cfg, netnl, "off")
|
|
_test_v4(cfg)
|
|
_test_tcp(cfg)
|
|
_set_offload_checksum(cfg, netnl, "on")
|
|
_test_v4(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def test_default_v6(cfg, netnl) -> None:
|
|
cfg.require_ipver("6")
|
|
|
|
_schedule_checksum_reset(cfg, netnl)
|
|
_set_offload_checksum(cfg, netnl, "off")
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
_set_offload_checksum(cfg, netnl, "on")
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def test_xdp_generic_sb(cfg, netnl) -> None:
|
|
_schedule_checksum_reset(cfg, netnl)
|
|
_set_xdp_generic_sb_on(cfg)
|
|
_set_offload_checksum(cfg, netnl, "off")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
_set_offload_checksum(cfg, netnl, "on")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def test_xdp_generic_mb(cfg, netnl) -> None:
|
|
_schedule_checksum_reset(cfg, netnl)
|
|
_set_xdp_generic_mb_on(cfg)
|
|
_set_offload_checksum(cfg, netnl, "off")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
_set_offload_checksum(cfg, netnl, "on")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def test_xdp_native_sb(cfg, netnl) -> None:
|
|
_schedule_checksum_reset(cfg, netnl)
|
|
_set_xdp_native_sb_on(cfg)
|
|
_set_offload_checksum(cfg, netnl, "off")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
_set_offload_checksum(cfg, netnl, "on")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def test_xdp_native_mb(cfg, netnl) -> None:
|
|
_schedule_checksum_reset(cfg, netnl)
|
|
_set_xdp_native_mb_on(cfg)
|
|
_set_offload_checksum(cfg, netnl, "off")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
_set_offload_checksum(cfg, netnl, "on")
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def test_xdp_offload(cfg, netnl) -> None:
|
|
_set_xdp_offload_on(cfg)
|
|
_test_v4(cfg)
|
|
_test_v6(cfg)
|
|
_test_tcp(cfg)
|
|
|
|
def main() -> None:
|
|
with NetDrvEpEnv(__file__) as cfg:
|
|
get_interface_info(cfg)
|
|
set_interface_init(cfg)
|
|
ksft_run([test_default_v4,
|
|
test_default_v6,
|
|
test_xdp_generic_sb,
|
|
test_xdp_generic_mb,
|
|
test_xdp_native_sb,
|
|
test_xdp_native_mb,
|
|
test_xdp_offload],
|
|
args=(cfg, EthtoolFamily()))
|
|
ksft_exit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|