#!/usr/bin/env python3# SPDX-License-Identifier: GPL-2.0 import osimport random, string, timefrom lib.py import ksft_run, ksft_exitfrom lib.py import ksft_eq, KsftSkipEx, KsftFailExfrom lib.py import EthtoolFamily, NetDrvEpEnvfrom lib.py import bkg, cmd, wait_port_listen, rand_portfrom 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()