ping.py

  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", local=False, 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()