mirror of
https://github.com/torvalds/linux.git
synced 2025-11-03 01:59:51 +02:00
fbbf93556f
Add selftest case to check the send and receive throughput. Supported link modes between local NIC driver and partner are varied. Then send and receive throughput is captured and verified. Test uses iperf3 tool. Add iperf3 server/client function in GenerateTraffic class. Signed-off-by: Mohan Prasad J <mohan.prasad@microchip.com> Signed-off-by: Paolo Abeni <pabeni@redhat.com>
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
# SPDX-License-Identifier: GPL-2.0
|
|
|
|
import time
|
|
|
|
from lib.py import ksft_pr, cmd, ip, rand_port, wait_port_listen, bkg
|
|
|
|
class GenerateTraffic:
|
|
def __init__(self, env, port=None):
|
|
env.require_cmd("iperf3", remote=True)
|
|
|
|
self.env = env
|
|
|
|
if port is None:
|
|
port = rand_port()
|
|
self._iperf_server = cmd(f"iperf3 -s -1 -p {port}", background=True)
|
|
wait_port_listen(port)
|
|
time.sleep(0.1)
|
|
self._iperf_client = cmd(f"iperf3 -c {env.addr} -P 16 -p {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 run_remote_test(self, env: object, port=None, command=None):
|
|
if port is None:
|
|
port = rand_port()
|
|
try:
|
|
server_cmd = f"iperf3 -s 1 -p {port} --one-off"
|
|
with bkg(server_cmd, host=env.remote):
|
|
#iperf3 opens TCP connection as default in server
|
|
#-u to be specified in client command for UDP
|
|
wait_port_listen(port, host=env.remote)
|
|
except Exception as e:
|
|
raise Exception(f"Unexpected error occurred while running server command: {e}")
|
|
try:
|
|
client_cmd = f"iperf3 -c {env.remote_addr} -p {port} {command}"
|
|
proc = cmd(client_cmd)
|
|
return proc
|
|
except Exception as e:
|
|
raise Exception(f"Unexpected error occurred while running client command: {e}")
|
|
|
|
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)
|