selftests: drv-net: store addresses in dict indexed by ipver

Looks like more and more tests want to iterate over IP version,
run the same test over ipv4 and ipv6. The current naming of
members in the env class makes it a bit awkward, we have
separate members for ipv4 and ipv6 parameters.

Store the parameters inside dicts, so that tests can easily
index them with ip version.

Reviewed-by: Willem de Bruijn <willemb@google.com>
Link: https://patch.msgid.link/20250218225426.77726-4-kuba@kernel.org
Signed-off-by: Jakub Kicinski <kuba@kernel.org>
This commit is contained in:
Jakub Kicinski 2025-02-18 14:54:25 -08:00
parent 2aefca8e1f
commit de94e86974
4 changed files with 50 additions and 63 deletions

View file

@ -9,15 +9,12 @@ from lib.py import ksft_run, ksft_exit, KsftSkipEx
from lib.py import EthtoolFamily, NetDrvEpEnv from lib.py import EthtoolFamily, NetDrvEpEnv
from lib.py import bkg, cmd, wait_port_listen from lib.py import bkg, cmd, wait_port_listen
def test_receive(cfg, ipv4=False, extra_args=None): def test_receive(cfg, ipver="6", extra_args=None):
"""Test local nic checksum receive. Remote host sends crafted packets.""" """Test local nic checksum receive. Remote host sends crafted packets."""
if not cfg.have_rx_csum: if not cfg.have_rx_csum:
raise KsftSkipEx(f"Test requires rx checksum offload on {cfg.ifname}") raise KsftSkipEx(f"Test requires rx checksum offload on {cfg.ifname}")
if ipv4: ip_args = f"-{ipver} -S {cfg.remote_addr_v[ipver]} -D {cfg.addr_v[ipver]}"
ip_args = f"-4 -S {cfg.remote_v4} -D {cfg.v4}"
else:
ip_args = f"-6 -S {cfg.remote_v6} -D {cfg.v6}"
rx_cmd = f"{cfg.bin_local} -i {cfg.ifname} -n 100 {ip_args} -r 1 -R {extra_args}" rx_cmd = f"{cfg.bin_local} -i {cfg.ifname} -n 100 {ip_args} -r 1 -R {extra_args}"
tx_cmd = f"{cfg.bin_remote} -i {cfg.ifname} -n 100 {ip_args} -r 1 -T {extra_args}" tx_cmd = f"{cfg.bin_remote} -i {cfg.ifname} -n 100 {ip_args} -r 1 -T {extra_args}"
@ -27,17 +24,14 @@ def test_receive(cfg, ipv4=False, extra_args=None):
cmd(tx_cmd, host=cfg.remote) cmd(tx_cmd, host=cfg.remote)
def test_transmit(cfg, ipv4=False, extra_args=None): def test_transmit(cfg, ipver="6", extra_args=None):
"""Test local nic checksum transmit. Remote host verifies packets.""" """Test local nic checksum transmit. Remote host verifies packets."""
if (not cfg.have_tx_csum_generic and if (not cfg.have_tx_csum_generic and
not (cfg.have_tx_csum_ipv4 and ipv4) and not (cfg.have_tx_csum_ipv4 and ipver == "4") and
not (cfg.have_tx_csum_ipv6 and not ipv4)): not (cfg.have_tx_csum_ipv6 and ipver == "6")):
raise KsftSkipEx(f"Test requires tx checksum offload on {cfg.ifname}") raise KsftSkipEx(f"Test requires tx checksum offload on {cfg.ifname}")
if ipv4: ip_args = f"-{ipver} -S {cfg.addr_v[ipver]} -D {cfg.remote_addr_v[ipver]}"
ip_args = f"-4 -S {cfg.v4} -D {cfg.remote_v4}"
else:
ip_args = f"-6 -S {cfg.v6} -D {cfg.remote_v6}"
# Cannot randomize input when calculating zero checksum # Cannot randomize input when calculating zero checksum
if extra_args != "-U -Z": if extra_args != "-U -Z":
@ -51,26 +45,20 @@ def test_transmit(cfg, ipv4=False, extra_args=None):
cmd(tx_cmd) cmd(tx_cmd)
def test_builder(name, cfg, ipv4=False, tx=False, extra_args=""): def test_builder(name, cfg, ipver="6", tx=False, extra_args=""):
"""Construct specific tests from the common template. """Construct specific tests from the common template.
Most tests follow the same basic pattern, differing only in Most tests follow the same basic pattern, differing only in
Direction of the test and optional flags passed to csum.""" Direction of the test and optional flags passed to csum."""
def f(cfg): def f(cfg):
if ipv4: cfg.require_ipver(ipver)
cfg.require_v4()
else:
cfg.require_v6()
if tx: if tx:
test_transmit(cfg, ipv4, extra_args) test_transmit(cfg, ipver, extra_args)
else: else:
test_receive(cfg, ipv4, extra_args) test_receive(cfg, ipver, extra_args)
if ipv4: f.__name__ = f"ipv{ipver}_" + name
f.__name__ = "ipv4_" + name
else:
f.__name__ = "ipv6_" + name
return f return f
@ -104,15 +92,15 @@ def main() -> None:
cfg.bin_remote = cfg.remote.deploy(cfg.bin_local) cfg.bin_remote = cfg.remote.deploy(cfg.bin_local)
cases = [] cases = []
for ipv4 in [True, False]: for ipver in ["4", "6"]:
cases.append(test_builder("rx_tcp", cfg, ipv4, False, "-t")) cases.append(test_builder("rx_tcp", cfg, ipver, False, "-t"))
cases.append(test_builder("rx_tcp_invalid", cfg, ipv4, False, "-t -E")) cases.append(test_builder("rx_tcp_invalid", cfg, ipver, False, "-t -E"))
cases.append(test_builder("rx_udp", cfg, ipv4, False, "")) cases.append(test_builder("rx_udp", cfg, ipver, False, ""))
cases.append(test_builder("rx_udp_invalid", cfg, ipv4, False, "-E")) cases.append(test_builder("rx_udp_invalid", cfg, ipver, False, "-E"))
cases.append(test_builder("tx_udp_csum_offload", cfg, ipv4, True, "-U")) cases.append(test_builder("tx_udp_csum_offload", cfg, ipver, True, "-U"))
cases.append(test_builder("tx_udp_zero_checksum", cfg, ipv4, True, "-U -Z")) cases.append(test_builder("tx_udp_zero_checksum", cfg, ipver, True, "-U -Z"))
ksft_run(cases=cases, args=(cfg, )) ksft_run(cases=cases, args=(cfg, ))
ksft_exit() ksft_exit()

View file

@ -21,15 +21,15 @@ def require_devmem(cfg):
@ksft_disruptive @ksft_disruptive
def check_rx(cfg) -> None: def check_rx(cfg) -> None:
cfg.require_v6() cfg.require_ipver("6")
require_devmem(cfg) require_devmem(cfg)
port = rand_port() port = rand_port()
listen_cmd = f"./ncdevmem -l -f {cfg.ifname} -s {cfg.v6} -p {port}" listen_cmd = f"./ncdevmem -l -f {cfg.ifname} -s {cfg.addr_v['6']} -p {port}"
with bkg(listen_cmd) as socat: with bkg(listen_cmd) as socat:
wait_port_listen(port) wait_port_listen(port)
cmd(f"echo -e \"hello\\nworld\"| socat -u - TCP6:[{cfg.v6}]:{port}", host=cfg.remote, shell=True) cmd(f"echo -e \"hello\\nworld\"| socat -u - TCP6:[{cfg.addr_v['6']}]:{port}", host=cfg.remote, shell=True)
ksft_eq(socat.stdout.strip(), "hello\nworld") ksft_eq(socat.stdout.strip(), "hello\nworld")

View file

@ -113,6 +113,9 @@ class NetDrvEpEnv(NetDrvEnvBase):
self._ns = None self._ns = None
self._ns_peer = None self._ns_peer = None
self.addr_v = { "4": None, "6": None }
self.remote_addr_v = { "4": None, "6": None }
if "NETIF" in self.env: if "NETIF" in self.env:
if nsim_test is True: if nsim_test is True:
raise KsftXfailEx("Test only works on netdevsim") raise KsftXfailEx("Test only works on netdevsim")
@ -120,10 +123,10 @@ class NetDrvEpEnv(NetDrvEnvBase):
self.dev = ip("-d link show dev " + self.env['NETIF'], json=True)[0] self.dev = ip("-d link show dev " + self.env['NETIF'], json=True)[0]
self.v4 = self.env.get("LOCAL_V4") self.addr_v["4"] = self.env.get("LOCAL_V4")
self.v6 = self.env.get("LOCAL_V6") self.addr_v["6"] = self.env.get("LOCAL_V6")
self.remote_v4 = self.env.get("REMOTE_V4") self.remote_addr_v["4"] = self.env.get("REMOTE_V4")
self.remote_v6 = self.env.get("REMOTE_V6") self.remote_addr_v["6"] = self.env.get("REMOTE_V6")
kind = self.env["REMOTE_TYPE"] kind = self.env["REMOTE_TYPE"]
args = self.env["REMOTE_ARGS"] args = self.env["REMOTE_ARGS"]
else: else:
@ -134,22 +137,22 @@ class NetDrvEpEnv(NetDrvEnvBase):
self.dev = self._ns.nsims[0].dev self.dev = self._ns.nsims[0].dev
self.v4 = self.nsim_v4_pfx + "1" self.addr_v["4"] = self.nsim_v4_pfx + "1"
self.v6 = self.nsim_v6_pfx + "1" self.addr_v["6"] = self.nsim_v6_pfx + "1"
self.remote_v4 = self.nsim_v4_pfx + "2" self.remote_addr_v["4"] = self.nsim_v4_pfx + "2"
self.remote_v6 = self.nsim_v6_pfx + "2" self.remote_addr_v["6"] = self.nsim_v6_pfx + "2"
kind = "netns" kind = "netns"
args = self._netns.name args = self._netns.name
self.remote = Remote(kind, args, src_path) self.remote = Remote(kind, args, src_path)
self.addr = self.v6 if self.v6 else self.v4 self.addr_ipver = "6" if self.addr_v["6"] else "4"
self.remote_addr = self.remote_v6 if self.remote_v6 else self.remote_v4 self.addr = self.addr_v[self.addr_ipver]
self.remote_addr = self.remote_addr_v[self.addr_ipver]
self.addr_ipver = "6" if self.v6 else "4"
# Bracketed addresses, some commands need IPv6 to be inside [] # Bracketed addresses, some commands need IPv6 to be inside []
self.baddr = f"[{self.v6}]" if self.v6 else self.v4 self.baddr = f"[{self.addr_v['6']}]" if self.addr_v["6"] else self.addr_v["4"]
self.remote_baddr = f"[{self.remote_v6}]" if self.remote_v6 else self.remote_v4 self.remote_baddr = f"[{self.remote_addr_v['6']}]" if self.remote_addr_v["6"] else self.remote_addr_v["4"]
self.ifname = self.dev['ifname'] self.ifname = self.dev['ifname']
self.ifindex = self.dev['ifindex'] self.ifindex = self.dev['ifindex']
@ -205,10 +208,10 @@ class NetDrvEpEnv(NetDrvEnvBase):
def resolve_remote_ifc(self): def resolve_remote_ifc(self):
v4 = v6 = None v4 = v6 = None
if self.remote_v4: if self.remote_addr_v["4"]:
v4 = ip("addr show to " + self.remote_v4, json=True, host=self.remote) v4 = ip("addr show to " + self.remote_addr_v["4"], json=True, host=self.remote)
if self.remote_v6: if self.remote_addr_v["6"]:
v6 = ip("addr show to " + self.remote_v6, json=True, host=self.remote) v6 = ip("addr show to " + self.remote_addr_v["6"], json=True, host=self.remote)
if v4 and v6 and v4[0]["ifname"] != v6[0]["ifname"]: if v4 and v6 and v4[0]["ifname"] != v6[0]["ifname"]:
raise Exception("Can't resolve remote interface name, v4 and v6 don't match") raise Exception("Can't resolve remote interface name, v4 and v6 don't match")
if (v4 and len(v4) > 1) or (v6 and len(v6) > 1): if (v4 and len(v4) > 1) or (v6 and len(v6) > 1):
@ -238,13 +241,9 @@ class NetDrvEpEnv(NetDrvEnvBase):
del self.remote del self.remote
self.remote = None self.remote = None
def require_v4(self): def require_ipver(self, ipver):
if not self.v4 or not self.remote_v4: if not self.addr_v[ipver] or not self.remote_addr_v[ipver]:
raise KsftSkipEx("Test requires IPv4 connectivity") raise KsftSkipEx(f"Test requires IPv{ipver} connectivity")
def require_v6(self):
if not self.v6 or not self.remote_v6:
raise KsftSkipEx("Test requires IPv6 connectivity")
def _require_cmd(self, comm, key, host=None): def _require_cmd(self, comm, key, host=None):
cached = self._required_cmd.get(comm, {}) cached = self._required_cmd.get(comm, {})

View file

@ -8,17 +8,17 @@ from lib.py import bkg, cmd, wait_port_listen, rand_port
def test_v4(cfg) -> None: def test_v4(cfg) -> None:
cfg.require_v4() cfg.require_ipver("4")
cmd(f"ping -c 1 -W0.5 {cfg.remote_v4}") cmd("ping -c 1 -W0.5 " + cfg.remote_addr_v["4"])
cmd(f"ping -c 1 -W0.5 {cfg.v4}", host=cfg.remote) cmd("ping -c 1 -W0.5 " + cfg.addr_v["4"], host=cfg.remote)
def test_v6(cfg) -> None: def test_v6(cfg) -> None:
cfg.require_v6() cfg.require_ipver("6")
cmd(f"ping -c 1 -W0.5 {cfg.remote_v6}") cmd("ping -c 1 -W0.5 " + cfg.remote_addr_v["6"])
cmd(f"ping -c 1 -W0.5 {cfg.v6}", host=cfg.remote) cmd("ping -c 1 -W0.5 " + cfg.addr_v["6"], host=cfg.remote)
def test_tcp(cfg) -> None: def test_tcp(cfg) -> None: