scottk/wlan-lanforge-scripts
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-lanforge-scripts.git synced 2025-10-31 18:58:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

tos-plus: Allow specifying pkts-to-send

Browse Source 
In this case, the script will wait until all connections are stopped
(or duration timer expires), and then exit the script.
This commit is contained in:
Ben Greear
2020-04-07 07:08:38 -07:00
parent 1939665d5a
commit 330b58b212
2 changed files with 75 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lf_firemod.pl
View File

@@ -54,6 +54,7 @@ our $max_speed = "0";
our $rcv_mcast = "YES";
our $min_pkt_sz = "-1";
our $max_pkt_sz = "0";
our $pkts_to_send = "";
our $use_csums = "NO"; # Use LANforge checksums in payload?
our $ttl = 32;
our $report_timer = 5000;
@@ -108,6 +109,7 @@ $0 [ --action {
# --endp_vals rx_bytes (Rx Bytes)
[--ip_port {-1 (let LF choose, AUTO) | 0 (let OS choose, ANY) | specific IP port}]
[--max_pkt_sz {maximum payload size in bytes}]
[--pkts_to_send {Packets (PDUs) to send before stopping the connection. 0 means infinite}]
[--max_speed {speed in bps, 0 means same as mininum value}]
[--min_pkt_sz {minimum payload size in bytes}]
[--mcast_addr {multicast address, for example: 224.4.5.6}]
@@ -143,7 +145,7 @@ Example:
$0 --action create_endp --endp_name mcast_xmit_1 --speed 154000 \\
--endp_type mc_udp --mcast_addr 224.9.9.8 --mcast_port 9998 \\
--rcv_mcast NO --port_name eth1 \\
--min_pkt_sz 1072 --max_pkt_sz 1472 \\
--min_pkt_sz 1072 --max_pkt_sz 1472 --pkts_to_send 9999\\
--use_csums NO --ttl 32 \\
--quiet no --report_timer 1000
@@ -219,6 +221,7 @@ GetOptions
'max_speed=s' => \$::speed,
'min_pkt_sz=s' => \$::min_pkt_sz,
'max_pkt_sz=s' => \$::max_pkt_sz,
'pkts_to_send=s' => \$::pkts_to_send,
'mcast_addr=s' => \$::mcast_addr,
'mcast_port=s' => \$::mcast_port,
'lfmgr_port|mgr_port|port|p=i' => \$::lfmgr_port,
@@ -728,6 +731,11 @@ sub create_endp {
$cmd = $::utils->fmt_cmd("set_mc_endp", $::endp_name, $::ttl, $::mcast_addr, $::mcast_port, $::rcv_mcast);
$::utils->doCmd($cmd);
if ($pkts_to_send ne "") {
$cmd = $::utils->fmt_cmd("set_endp_details", $::endp_name, "NA", "NA", "NA", $::pkts_to_send);
$::utils->doCmd($cmd);
}
$cmd = "set_endp_report_timer $::endp_name $::report_timer";
$::utils->doCmd($cmd);
}
@@ -768,6 +776,11 @@ sub create_endp {
$bad_ppm, $::multicon);
$::utils->doCmd($cmd);
if ($pkts_to_send ne "") {
$cmd = $::utils->fmt_cmd("set_endp_details", $::endp_name, "NA", "NA", "NA", $::pkts_to_send);
$::utils->doCmd($cmd);
}
$cmd = "set_endp_report_timer $my_endp_name $::report_timer";
$::utils->doCmd($cmd);

73
lf_tos_plus_test.py
View File

@@ -26,6 +26,7 @@ Supported modes are: a, b, g, abg, abgn, bgn, bg, abgnAC, anAC, an, bgnAC, abgnA
# Run connections on wlan0 and sta0, set radio wiphy0 to use any frequency
# and 2 spatial streams.
./lf_tos_plus_test.py --lfmgr 192.168.100.178 --ssid testme --passwd mypsk \
[ --txpkts 10000 ]
--radio "1.wiphy0 2 0"
--cx "1.wiphy0 1.wlan0 an 1.eth1 udp 1024 1000000 56000 BK" \
--cx "1.wiphy0 1.sta0 anAC 1.eth1 udp 1472 56000 1000000 BK" ..
@@ -64,6 +65,7 @@ passwd = ""
ssid = "Test-SSID"
security = "open"
radio_strs = [] # Radios to modify: radio nss channel
txpkts = 0 # 0 == Run forever
# rssi_adjust = (current_nf - nf_at_calibration)
@@ -75,7 +77,8 @@ def usage():
print("--lfmgr: LANforge manager IP address")
print("--duration: Duration to run traffic, in minutes")
print("--ssid: AP's SSID")
print("--passwd: Optional password (do not add this option for OPEN)")
print("--passwd: Optional: password (do not add this option for OPEN)")
print("--txpkts: Optional: amount of packets to transmit (and then stop the data connections)")
print("-h|--help")
def main():
@@ -87,15 +90,17 @@ def main():
global ssid
global security
global radio_strs
global txpkts
parser = argparse.ArgumentParser(description="ToS++ report Script")
parser.add_argument("--cx", type=str, action='append', help="Connection tuple: station-radio station-port mode upstream-port protocol pkt-size speed_ul speed_dl QoS")
parser.add_argument("--radio", type=str, action='append', help="Radio tuple: radio nss channel")
parser.add_argument("--lfmgr", type=str, help="LANforge Manager IP address")
parser.add_argument("--outfile", type=str, help="Output file for csv data")
parser.add_argument("--duration", type=float, help="Duration to run traffic, in minutes")
parser.add_argument("--duration", type=float, help="Duration to run traffic, in minutes. If txpkts is specified, that may stop the test earlier.")
parser.add_argument("--ssid", type=str, help="AP's SSID")
parser.add_argument("--passwd", type=str, help="AP's password if using PSK authentication, skip this argement for OPEN")
parser.add_argument("--txpkts", type=str, help="Optional: Packets (PDUs) to send before stopping data connections Default (0) means infinite")
args = None
try:
@@ -113,6 +118,8 @@ def main():
security = "wpa2"
if (args.outfile != None):
outfile = args.outfile
if (args.txpkts != None):
txpkts = args.txpkts
filehandler = None
except Exception as e:
logging.exception(e);
@@ -234,6 +241,7 @@ def main():
sta_to_mode = {} # map station name to mode
cxnames = [] # list of all cxnames
endpnames = []
endp_to_port = {}
endp_to_pktsz = {}
endp_to_proto = {}
@@ -375,6 +383,8 @@ def main():
enb = "scr-tos+-%i-B"%count
cxnames.append(cxn)
endpnames.append(ena)
endpnames.append(enb)
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--action", "do_cmd",
"--cmd", "rm_cx ALL %s"%cxn], stderr=PIPE, stdout=PIPE);
@@ -401,10 +411,10 @@ def main():
# Now, create the new connection
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--resource", "%s"%sta_resource, "--action", "create_endp", "--port_name", sta_name,
"--endp_type", cx_proto, "--endp_name", ena, "--speed", "%s"%cx_speed_ul, "--report_timer", "1000", "--tos", t,
"--min_pkt_sz", pkt_sz, "--multicon", "1"])#, capture_output=True);
"--min_pkt_sz", pkt_sz, "--multicon", "1", "--pkts_to_send", txpkts])#, capture_output=True);
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--resource", "%s"%u_resource, "--action", "create_endp", "--port_name", u_name,
"--endp_type", cx_proto, "--endp_name", enb, "--speed", "%s"%cx_speed_dl, "--report_timer", "1000", "--tos", t,
"--min_pkt_sz", pkt_sz, "--multicon", "1"])# capture_output=True);
"--min_pkt_sz", pkt_sz, "--multicon", "1", "--pkts_to_send", txpkts])# capture_output=True);
# Enable Multi-Helper
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--action", "do_cmd", "--cmd",
@@ -414,17 +424,56 @@ def main():
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--action", "do_cmd", "--cmd",
"add_cx %s default_tm %s %s"%(cxn, ena, enb)])# capture_output=True);
# Start traffic
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--action", "do_cmd",
"--cmd", "set_cx_state all %s RUNNING"%cxn]);
count = count + 1
# Traffic is started, wait requested amount of time
print("Waiting %s seconds to let traffic run for a bit"%(dur))
time.sleep(dur)
# All traffic connects are created, now start them all
for cxn in cxnames:
# Start traffic
subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--action", "do_cmd",
"--cmd", "set_cx_state all %s RUNNING"%cxn]);
# Traffic is started, wait requested amount of time
stop_at = time.time() + dur;
if txpkts == 0:
print("Waiting %s seconds to let traffic run for a bit"%(dur))
time.sleep(dur)
else:
# Wait until connections are done transmitting and all are stopped
print("Waiting until all connections have finished transmitting %s and have stopped themselves."%txpkts);
done = False
while not done:
if time.time() > stop_at:
print("Duration expired, stop waiting for Endpoints to quiesce.")
break
foundone = False
for ename in endpnames:
#print("Checking endpoint: %s"%ename)
endp_stats = subprocess.run(["./lf_firemod.pl", "--manager", lfmgr, "--endp_name", ename,
"--endp_vals", "Endpoint-flags"], stderr=PIPE, stdout=PIPE);
ess = endp_stats.stdout.decode('utf-8', 'ignore');
for line in ess.splitlines():
#print("endp-stats line: %s"%line)
m = re.search('Endpoint-flags:\s+(.*)', line)
if (m != None):
flags = m.group(1)
if not "NOT_RUNNING" in flags:
foundone = True
#print("Flags, was running: %s"%flags)
break
else:
#print("Flags, was not running: %s"%flags)
if foundone:
break
if not foundone:
print("All endpoints stopped, continuing on.")
done = True
break
sleep(3) # wait 3 seconds, then poll again
# Gather probe results and record data, verify NSS, BW, Channel
sta_stats = {} # Key is resource.station, holds array of values