| # 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") |