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

tos-plus: Support non-wifi station test as well.

Browse Source
This commit is contained in:
Ben Greear
2020-04-09 14:37:07 -07:00
parent 2ce51d7089
commit 61e95cc270
1 changed files with 57 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
lf_tos_plus_test.py
View File

@@ -35,6 +35,9 @@ are very likely more than the PDU count)
--cx "1.wiphy0 1.sta0 anAC 1.eth1 tcp 1472 56000 2000000 BK" \ --cx "1.wiphy0 1.sta0 anAC 1.eth1 tcp 1472 56000 2000000 BK" \
--sniffer_radios "1.wiphy2" --sniffer_radios "1.wiphy2"
# You can also create connections between non-station endpoints.
./lf_tos_plus_test.py --dur 1 --lfmgr 192.168.100.156 --txpkts 99 \
--cx "NA 1.rddVR0 NA 1.rddVR1 tcp MTU 1000000 2000000 0" --wait_sniffer 1
make sure pexpect, pandas is installed: make sure pexpect, pandas is installed:
$ sudo yum install python3-pandas $ sudo yum install python3-pandas
@@ -115,6 +118,8 @@ def main():
global sniffer_radios global sniffer_radios
global wait_sniffer global wait_sniffer
do_sniff = False
parser = argparse.ArgumentParser(description="ToS++ report Script") 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("--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("--radio", type=str, action='append', help="Radio tuple: radio nss channel")
@@ -125,13 +130,14 @@ def main():
parser.add_argument("--passwd", type=str, help="AP's password if using PSK authentication, skip this argement for OPEN") 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") parser.add_argument("--txpkts", type=str, help="Optional: Packets (PDUs) to send before stopping data connections Default (0) means infinite")
parser.add_argument("--sniffer_radios", type=str, help="Optional: list of radios to sniff wifi traffic \"1.wiphy2 1.wiphy4\"") parser.add_argument("--sniffer_radios", type=str, help="Optional: list of radios to sniff wifi traffic \"1.wiphy2 1.wiphy4\"")
parser.add_argument("--wait_sniffer", type=str, help="Optional: 1 means wait on sniffer to finish before existing script\"") parser.add_argument("--wait_sniffer", type=str, help="Optional: 1 means wait on sniffer to finish before existing script.\"")
args = None args = None
try: try:
args = parser.parse_args() args = parser.parse_args()
cx_strs = args.cx.copy() cx_strs = args.cx.copy()
radio_strs = args.radio.copy() if (args.radio != None):
radio_strs = args.radio.copy()
if (args.lfmgr != None): if (args.lfmgr != None):
lfmgr = args.lfmgr lfmgr = args.lfmgr
if (args.duration != None): if (args.duration != None):
@@ -147,7 +153,9 @@ def main():
txpkts = args.txpkts txpkts = args.txpkts
if (args.sniffer_radios != None): if (args.sniffer_radios != None):
sniffer_radios = args.sniffer_radios sniffer_radios = args.sniffer_radios
do_sniff = True
if (args.wait_sniffer != None): if (args.wait_sniffer != None):
do_sniff = True
wait_sniffer = args.wait_sniffer == "1" wait_sniffer = args.wait_sniffer == "1"
filehandler = None filehandler = None
except Exception as e: except Exception as e:
@@ -344,19 +352,25 @@ def main():
if not ukey in upstreams: if not ukey in upstreams:
upstreams.append(ukey) upstreams.append(ukey)
rad_resource = "1" if radio != "NA":
rad_name = radio; rad_resource = "1"
if radio[0].isdigit(): rad_name = radio;
tmpa = radio.split(".", 1); if radio[0].isdigit():
rad_resource = tmpa[0]; tmpa = radio.split(".", 1);
rad_name = tmpa[1]; rad_resource = tmpa[0];
rad_name = tmpa[1];
# Create or update station with requested mode # Create or update station with requested mode
subprocess.run(["./lf_associate_ap.pl", "--mgr", lfmgr, "--resource", rad_resource, "--action", "add", subprocess.run(["./lf_associate_ap.pl", "--mgr", lfmgr, "--resource", rad_resource, "--action", "add",
"--radio", rad_name, "--ssid", ssid, "--passphrase", passwd, "--security", security, "--radio", rad_name, "--ssid", ssid, "--passphrase", passwd, "--security", security,
"--first_sta", sta_name, "--first_ip", "DHCP", "--wifi_mode", mode, "--num_stations", "1"]) "--first_sta", sta_name, "--first_ip", "DHCP", "--wifi_mode", mode, "--num_stations", "1"])
else:
# Assume it should be sniffed.
ukey = "%s.%s"%(sta_resource, sta_name)
if not ukey in upstreams:
upstreams.append(ukey)
# Up station # Up station / A-side Port
subprocess.run(["./lf_portmod.pl", "--manager", lfmgr, "--card", sta_resource, "--port_name", sta_name, subprocess.run(["./lf_portmod.pl", "--manager", lfmgr, "--card", sta_resource, "--port_name", sta_name,
"--set_ifstate", "up"]); "--set_ifstate", "up"]);
@@ -382,18 +396,27 @@ def main():
#print("IP %s Status %s"%(_ip, _status)) #print("IP %s Status %s"%(_ip, _status))
if (_status == "Authorized"): if sta_name.startswith("wlan") or sta_name.startswith("sta"):
if (_status == "Authorized"):
if ((_ip != None) and (_ip != "0.0.0.0")):
print("Station is associated with IP address.")
break
else:
if (not wait_ip_print):
print("Waiting for station %s.%s to get IP Address."%(sta_resource, sta_name))
wait_ip_print = True
else:
if (not wait_assoc_print):
print("Waiting up to 180s for station %s.%s to associate."%(sta_resource, sta_name))
wait_assoc_print = True
else:
if ((_ip != None) and (_ip != "0.0.0.0")): if ((_ip != None) and (_ip != "0.0.0.0")):
print("Station is associated with IP address.") print("Port is associated with IP address.")
break break
else: else:
if (not wait_ip_print): if (not wait_ip_print):
print("Waiting for station %s.%s to get IP Address."%(sta_resource, sta_name)) print("Waiting for port %s.%s to get IP Address."%(sta_resource, sta_name))
wait_ip_print = True wait_ip_print = True
else:
if (not wait_assoc_print):
print("Waiting up to 180s for station %s.%s to associate."%(sta_resource, sta_name))
wait_assoc_print = True
i = i + 1 i = i + 1
# We wait a fairly long time since AP will take a long time to start on a CAC channel. # We wait a fairly long time since AP will take a long time to start on a CAC channel.
@@ -462,26 +485,28 @@ def main():
count = count + 1 count = count + 1
# Start sniffer? # Start sniffer?
if sniffer_radios != "": if do_sniff:
radios = sniffer_radios.split()
lfstations = "" lfstations = ""
lfupstreams = "" lfupstreams = ""
ri = 0 if sniffer_radios != "":
for r in radios: radios = sniffer_radios.split()
lfstations = lfstations + " " + stations[ri] ri = 0
ri = ri + 1 for r in radios:
lfstations = lfstations + " " + stations[ri]
ri = ri + 1
for u in upstreams: for u in upstreams:
lfupstreams = lfupstreams + " " + u lfupstreams = lfupstreams + " " + u
# Add 15 seconds to capture length in case it takes a bit of time to start all # Add 15 seconds to capture length in case it takes a bit of time to start all
# of the connections. # of the connections.
if lfupstreams == "": cmd = ["./lf_sniff.py", "--lfmgr", lfmgr, "--duration", "%f"%((dur + 15) / 60)]
subprocess.run(["./lf_sniff.py", "--lfmgr", lfmgr, "--duration", "%f"%((dur + 15) / 60), "--station", lfstations, if lfstations != "":
"--sniffer_radios", sniffer_radios])# capture_output=True); cmd.extend(["--station", lfstations, "--sniffer_radios", sniffer_radios])
else: if lfupstreams != "":
subprocess.run(["./lf_sniff.py", "--lfmgr", lfmgr, "--duration", "%f"%((dur + 15) / 60), "--station", lfstations, cmd.extend(["--upstreams", lfupstreams])
"--sniffer_radios", sniffer_radios, "--upstreams", lfupstreams])# capture_output=True);
subprocess.run(cmd)
sniff_done_at = time.time() + dur + 15; sniff_done_at = time.time() + dur + 15;