1# SPDX-License-Identifier: GPL-2.0 2 3import re 4import time 5 6from lib.py import ksft_pr, cmd, ip, rand_port, wait_port_listen 7 8class GenerateTraffic: 9 def __init__(self, env, port=None): 10 env.require_cmd("iperf3", local=True, remote=True) 11 12 self.env = env 13 14 self.port = rand_port() if port is None else port 15 self._iperf_server = cmd(f"iperf3 -s -1 -p {self.port}", background=True) 16 wait_port_listen(self.port) 17 time.sleep(0.1) 18 self._iperf_client = cmd(f"iperf3 -c {env.addr} -P 16 -p {self.port} -t 86400", 19 background=True, host=env.remote) 20 21 # Wait for traffic to ramp up 22 if not self._wait_pkts(pps=1000): 23 self.stop(verbose=True) 24 raise Exception("iperf3 traffic did not ramp up") 25 26 def _wait_pkts(self, pkt_cnt=None, pps=None): 27 """ 28 Wait until we've seen pkt_cnt or until traffic ramps up to pps. 29 Only one of pkt_cnt or pss can be specified. 30 """ 31 pkt_start = ip("-s link show dev " + self.env.ifname, json=True)[0]["stats64"]["rx"]["packets"] 32 for _ in range(50): 33 time.sleep(0.1) 34 pkt_now = ip("-s link show dev " + self.env.ifname, json=True)[0]["stats64"]["rx"]["packets"] 35 if pps: 36 if pkt_now - pkt_start > pps / 10: 37 return True 38 pkt_start = pkt_now 39 elif pkt_cnt: 40 if pkt_now - pkt_start > pkt_cnt: 41 return True 42 return False 43 44 def wait_pkts_and_stop(self, pkt_cnt): 45 failed = not self._wait_pkts(pkt_cnt=pkt_cnt) 46 self.stop(verbose=failed) 47 48 def stop(self, verbose=None): 49 self._iperf_client.process(terminate=True) 50 if verbose: 51 ksft_pr(">> Client:") 52 ksft_pr(self._iperf_client.stdout) 53 ksft_pr(self._iperf_client.stderr) 54 self._iperf_server.process(terminate=True) 55 if verbose: 56 ksft_pr(">> Server:") 57 ksft_pr(self._iperf_server.stdout) 58 ksft_pr(self._iperf_server.stderr) 59 self._wait_client_stopped() 60 61 def _wait_client_stopped(self, sleep=0.005, timeout=5): 62 end = time.monotonic() + timeout 63 64 live_port_pattern = re.compile(fr":{self.port:04X} 0[^6] ") 65 66 while time.monotonic() < end: 67 data = cmd("cat /proc/net/tcp*", host=self.env.remote).stdout 68 if not live_port_pattern.search(data): 69 return 70 time.sleep(sleep) 71 raise Exception(f"Waiting for client to stop timed out after {timeout}s") 72