mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-08-05 16:54:27 +00:00
d74cd9a02f
Make require_cmd() calls explicit about whether commands are needed locally, remotely, or both. Since require_cmd() defaults to local=True, tests should explicitly set local=False when commands are only needed remotely. - socat: Set local=False since it's only needed on remote hosts. - iperf3: Use single call with both local=True and remote=True since it's needed on both hosts. This avoids unnecessary test failures when commands are missing locally but available remotely where actually needed, and consolidates a duplicate require_cmd() call into single call that checks both hosts. Fixes:0d0f4174f6("selftests: drv-net: add a simple TSO test") Fixes:f1e68a1a4a("selftests: drv-net: add require_XYZ() helpers for validating env") Fixes:c76bab22e9("selftests: drv-net: rss_input_xfrm: Check test prerequisites before running") Reviewed-by: Nimrod Oren <noren@nvidia.com> Signed-off-by: Gal Pressman <gal@nvidia.com> Link: https://patch.msgid.link/20250723135454.649342-3-gal@nvidia.com Signed-off-by: Jakub Kicinski <kuba@kernel.org>
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
# SPDX-License-Identifier: GPL-2.0
|
|
|
|
import re
|
|
import time
|
|
|
|
from lib.py import ksft_pr, cmd, ip, rand_port, wait_port_listen
|
|
|
|
class GenerateTraffic:
|
|
def __init__(self, env, port=None):
|
|
env.require_cmd("iperf3", local=True, remote=True)
|
|
|
|
self.env = env
|
|
|
|
self.port = rand_port() if port is None else port
|
|
self._iperf_server = cmd(f"iperf3 -s -1 -p {self.port}", background=True)
|
|
wait_port_listen(self.port)
|
|
time.sleep(0.1)
|
|
self._iperf_client = cmd(f"iperf3 -c {env.addr} -P 16 -p {self.port} -t 86400",
|
|
background=True, host=env.remote)
|
|
|
|
# Wait for traffic to ramp up
|
|
if not self._wait_pkts(pps=1000):
|
|
self.stop(verbose=True)
|
|
raise Exception("iperf3 traffic did not ramp up")
|
|
|
|
def _wait_pkts(self, pkt_cnt=None, pps=None):
|
|
"""
|
|
Wait until we've seen pkt_cnt or until traffic ramps up to pps.
|
|
Only one of pkt_cnt or pss can be specified.
|
|
"""
|
|
pkt_start = ip("-s link show dev " + self.env.ifname, json=True)[0]["stats64"]["rx"]["packets"]
|
|
for _ in range(50):
|
|
time.sleep(0.1)
|
|
pkt_now = ip("-s link show dev " + self.env.ifname, json=True)[0]["stats64"]["rx"]["packets"]
|
|
if pps:
|
|
if pkt_now - pkt_start > pps / 10:
|
|
return True
|
|
pkt_start = pkt_now
|
|
elif pkt_cnt:
|
|
if pkt_now - pkt_start > pkt_cnt:
|
|
return True
|
|
return False
|
|
|
|
def wait_pkts_and_stop(self, pkt_cnt):
|
|
failed = not self._wait_pkts(pkt_cnt=pkt_cnt)
|
|
self.stop(verbose=failed)
|
|
|
|
def stop(self, verbose=None):
|
|
self._iperf_client.process(terminate=True)
|
|
if verbose:
|
|
ksft_pr(">> Client:")
|
|
ksft_pr(self._iperf_client.stdout)
|
|
ksft_pr(self._iperf_client.stderr)
|
|
self._iperf_server.process(terminate=True)
|
|
if verbose:
|
|
ksft_pr(">> Server:")
|
|
ksft_pr(self._iperf_server.stdout)
|
|
ksft_pr(self._iperf_server.stderr)
|
|
self._wait_client_stopped()
|
|
|
|
def _wait_client_stopped(self, sleep=0.005, timeout=5):
|
|
end = time.monotonic() + timeout
|
|
|
|
live_port_pattern = re.compile(fr":{self.port:04X} 0[^6] ")
|
|
|
|
while time.monotonic() < end:
|
|
data = cmd("cat /proc/net/tcp*", host=self.env.remote).stdout
|
|
if not live_port_pattern.search(data):
|
|
return
|
|
time.sleep(sleep)
|
|
raise Exception(f"Waiting for client to stop timed out after {timeout}s")
|