1#!/usr/bin/env python3 2# SPDX-License-Identifier: GPL-2.0 3 4from lib.py import ksft_run, ksft_exit 5from lib.py import ksft_eq 6from lib.py import NetDrvEpEnv 7from lib.py import bkg, cmd, wait_port_listen, rand_port 8 9 10def test_v4(cfg) -> None: 11 cfg.require_ipver("4") 12 13 cmd("ping -c 1 -W0.5 " + cfg.remote_addr_v["4"]) 14 cmd("ping -c 1 -W0.5 " + cfg.addr_v["4"], host=cfg.remote) 15 16 17def test_v6(cfg) -> None: 18 cfg.require_ipver("6") 19 20 cmd("ping -c 1 -W0.5 " + cfg.remote_addr_v["6"]) 21 cmd("ping -c 1 -W0.5 " + cfg.addr_v["6"], host=cfg.remote) 22 23 24def test_tcp(cfg) -> None: 25 cfg.require_cmd("socat", remote=True) 26 27 port = rand_port() 28 listen_cmd = f"socat -{cfg.addr_ipver} -t 2 -u TCP-LISTEN:{port},reuseport STDOUT" 29 30 with bkg(listen_cmd, exit_wait=True) as nc: 31 wait_port_listen(port) 32 33 cmd(f"echo ping | socat -t 2 -u STDIN TCP:{cfg.baddr}:{port}", 34 shell=True, host=cfg.remote) 35 ksft_eq(nc.stdout.strip(), "ping") 36 37 with bkg(listen_cmd, host=cfg.remote, exit_wait=True) as nc: 38 wait_port_listen(port, host=cfg.remote) 39 40 cmd(f"echo ping | socat -t 2 -u STDIN TCP:{cfg.remote_baddr}:{port}", shell=True) 41 ksft_eq(nc.stdout.strip(), "ping") 42 43 44def main() -> None: 45 with NetDrvEpEnv(__file__) as cfg: 46 ksft_run(globs=globals(), case_pfx={"test_"}, args=(cfg, )) 47 ksft_exit() 48 49 50if __name__ == "__main__": 51 main() 52