| 1 | |
| 2 | |
| 3 |
|
| 4 | import os
|
| 5 | import random, string, time
|
| 6 | from lib.py import ksft_run, ksft_exit
|
| 7 | from lib.py import ksft_eq, KsftSkipEx, KsftFailEx
|
| 8 | from lib.py import EthtoolFamily, NetDrvEpEnv
|
| 9 | from lib.py import bkg, cmd, wait_port_listen, rand_port
|
| 10 | from lib.py import defer, ethtool, ip
|
| 11 |
|
| 12 | no_sleep=False
|
| 13 |
|
| 14 | def _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 |
|
| 23 | def _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 |
|
| 32 | def _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 |
|
| 53 | def _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 |
|
| 63 | def _set_offload_checksum(cfg, netnl, on) -> None:
|
| 64 | try:
|
| 65 | ethtool(f" -K {cfg.ifname} rx {on} tx {on} ")
|
| 66 | except:
|
| 67 | return
|
| 68 |
|
| 69 | def _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 |
|
| 78 | def _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 |
|
| 88 | def _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 |
|
| 105 | def _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 |
|
| 118 | def _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 |
|
| 131 | def 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 |
|
| 142 | def 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 |
|
| 149 | def 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 |
|
| 160 | def 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 |
|
| 171 | def 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 |
|
| 183 | def 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 |
|
| 195 | def 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 |
|
| 207 | def 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 |
|
| 219 | def 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 |
|
| 225 | def 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 |
|
| 240 | if __name__ == "__main__":
|
| 241 | main()
|