scottk/wlan-lanforge-scripts
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-lanforge-scripts.git synced 2025-11-15 01:24:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

added layer3 in test_ipv4_ttls with optional argument of vap: commit is used in tip

Browse Source 
Signed-off-by: shivamcandela <shivam.thakur@candelatech.com>
This commit is contained in:
shivamcandela
2021-05-20 18:55:11 +05:30
parent c782d292ed
commit fefa5ae77d
2 changed files with 115 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

142
py-scripts/test_ipv4_ttls.py
View File

@@ -17,7 +17,7 @@ from LANforge import LFUtils
import realm import realm
import time import time
import pprint import pprint
from test_ipv4_variable_time import IPV4VariableTime
class TTLSTest(LFCliBase): class TTLSTest(LFCliBase):
def __init__(self, host="localhost", port=8080, def __init__(self, host="localhost", port=8080,
@@ -54,6 +54,7 @@ class TTLSTest(LFCliBase):
network_auth_type="NA", network_auth_type="NA",
anqp_3gpp_cell_net="NA", anqp_3gpp_cell_net="NA",
ieee80211w=1, ieee80211w=1,
vap=True,
hs20_enable=False, hs20_enable=False,
enable_pkc=False, enable_pkc=False,
number_template="00000", number_template="00000",
@@ -109,8 +110,12 @@ class TTLSTest(LFCliBase):
self.debug = _debug_on self.debug = _debug_on
self.local_realm = realm.Realm(lfclient_host=self.host, lfclient_port=self.port) self.local_realm = realm.Realm(lfclient_host=self.host, lfclient_port=self.port)
self.station_profile = self.local_realm.new_station_profile() self.station_profile = self.local_realm.new_station_profile()
self.vap_profile = self.local_realm.new_vap_profile() self.vap = vap
self.vap_profile.vap_name = "TestNet" self.upstream_port = "eth1"
self.upstream_resource = 1
if self.vap:
self.vap_profile = self.local_realm.new_vap_profile()
self.vap_profile.vap_name = "TestNet"
self.station_profile.lfclient_url = self.lfclient_url self.station_profile.lfclient_url = self.lfclient_url
self.station_profile.ssid = self.ssid self.station_profile.ssid = self.ssid
@@ -118,20 +123,47 @@ class TTLSTest(LFCliBase):
self.station_profile.number_template_ = self.number_template self.station_profile.number_template_ = self.number_template
self.station_profile.mode = 0 self.station_profile.mode = 0
# Layer3 Traffic
self.l3_cx_obj_udp = IPV4VariableTime(host=self.host, port=self.port,
create_sta=False, sta_list=self.sta_list, traffic_type="lf_udp",
upstream=self.upstream_port)
self.l3_cx_obj_udp.cx_profile.name_prefix = "udp-"
self.l3_cx_obj_udp.cx_profile.side_a_min_bps = 128000
self.l3_cx_obj_udp.cx_profile.side_a_max_bps = 128000
self.l3_cx_obj_udp.cx_profile.side_b_min_bps = 128000
self.l3_cx_obj_udp.cx_profile.side_b_max_bps = 128000
self.l3_cx_obj_udp.cx_profile.side_a_min_pdu = 1200
self.l3_cx_obj_udp.cx_profile.side_b_min_pdu = 1500
self.l3_cx_obj_udp.cx_profile.report_timer = 1000
self.l3_cx_obj_tcp = IPV4VariableTime(host=self.host, port=self.port,
create_sta=False, sta_list=self.sta_list, traffic_type="lf_tcp",
upstream=self.upstream_port)
self.l3_cx_obj_tcp.cx_profile.name_prefix = "tcp-"
self.l3_cx_obj_tcp.cx_profile.side_a_min_bps = 128000
self.l3_cx_obj_tcp.cx_profile.side_a_max_bps = 128000
self.l3_cx_obj_tcp.cx_profile.side_b_min_bps = 128000
self.l3_cx_obj_tcp.cx_profile.side_b_max_bps = 128000
self.l3_cx_obj_tcp.cx_profile.side_a_min_pdu = 1200
self.l3_cx_obj_tcp.cx_profile.side_b_min_pdu = 1500
self.l3_cx_obj_tcp.cx_profile.report_timer = 1000
def build(self): def build(self):
# Build stations # Build stations
keyphrase = "[BLANK]" keyphrase = "[BLANK]"
self.station_profile.use_security(self.security, self.ssid, passwd=self.password) self.station_profile.use_security(self.security, self.ssid, passwd=self.password)
self.vap_profile.use_security(self.security, self.ssid, passwd=self.password) if self.vap:
self.vap_profile.use_security(self.security, self.ssid, passwd=self.password)
self.station_profile.set_number_template(self.number_template) self.station_profile.set_number_template(self.number_template)
print("Creating stations") print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1) self.station_profile.set_command_flag("add_sta", "create_admin_down", 1)
self.station_profile.set_command_param("set_port", "report_timer", 1500) self.station_profile.set_command_param("set_port", "report_timer", 1500)
self.station_profile.set_command_flag("set_port", "rpt_timer", 1) self.station_profile.set_command_flag("set_port", "rpt_timer", 1)
self.station_profile.set_wifi_extra(key_mgmt=self.key_mgmt, self.station_profile.set_wifi_extra(key_mgmt=self.key_mgmt,
pairwise="DEFAULT", pairwise=self.pairwise,
group="DEFAULT", group=self.group,
psk=self.password, psk=self.password,
eap=self.eap, eap=self.eap,
identity=self.identity, identity=self.identity,
@@ -146,36 +178,40 @@ class TTLSTest(LFCliBase):
if self.hs20_enable: if self.hs20_enable:
self.station_profile.set_command_flag("add_sta", "hs20_enable", 1) self.station_profile.set_command_flag("add_sta", "hs20_enable", 1)
self.vap_profile.set_wifi_extra(key_mgmt=self.key_mgmt, if self.vap:
pairwise="DEFAULT", self.vap_profile.set_wifi_extra(key_mgmt=self.key_mgmt,
group="DEFAULT", pairwise="DEFAULT",
psk=self.password, group="DEFAULT",
eap=self.eap, psk=self.password,
identity=self.identity, eap=self.eap,
passwd=self.ttls_passwd, identity=self.identity,
realm=self.ttls_realm, passwd=self.ttls_passwd,
domain=self.domain, realm=self.ttls_realm,
hessid=self.hessid) domain=self.domain,
self.vap_profile.create(resource=1, hessid=self.hessid)
radio=self.radio, self.vap_profile.create(resource=1,
channel=36, radio=self.radio,
up_=True, channel=36,
debug=False, up_=True,
suppress_related_commands_=True, debug=False,
use_radius=True, suppress_related_commands_=True,
hs20_enable=False) use_radius=True,
hs20_enable=False)
self.station_profile.create(radio=self.radio, self.station_profile.create(radio=self.radio,
sta_names_=self.sta_list, sta_names_=self.sta_list,
debug=self.debug, debug=self.debug,
use_radius=True, use_radius=True,
hs20_enable=False) hs20_enable=False)
self._pass("Station build finished") self._pass("Station build finished")
self.l3_cx_obj_udp.build()
self.l3_cx_obj_tcp.build()
if self.debug: if self.debug:
pprint.pprint(self.station_profile.add_sta_data) pprint.pprint(self.station_profile.add_sta_data)
def start(self, sta_list, print_pass, print_fail): def start(self, sta_list, print_pass, print_fail, wait_time=40):
self.station_profile.admin_up() self.station_profile.admin_up()
self.vap_profile.admin_up(1) if self.vap:
self.vap_profile.admin_up(1)
associated_map = {} associated_map = {}
ip_map = {} ip_map = {}
print("Starting test...") print("Starting test...")
@@ -183,7 +219,7 @@ class TTLSTest(LFCliBase):
for sta_name in sta_list: for sta_name in sta_list:
sta_status = self.json_get("port/1/1/" + sta_name + "?fields=port,alias,ip,ap", debug_=self.debug) sta_status = self.json_get("port/1/1/" + sta_name + "?fields=port,alias,ip,ap", debug_=self.debug)
# print(sta_status) # print(sta_status)
if sta_status is None or sta_status['interface'] is None or sta_status['interface']['ap'] is None: if sta_status is None or sta_status['interface'] is None or sta_status['interface']['ip'] == "0.0.0.0":
continue continue
if len(sta_status['interface']['ap']) == 17 and sta_status['interface']['ap'][-3] == ':': if len(sta_status['interface']['ap']) == 17 and sta_status['interface']['ap'][-3] == ':':
# print("Associated", sta_name, sta_status['interface']['ap'], sta_status['interface']['ip']) # print("Associated", sta_name, sta_status['interface']['ap'], sta_status['interface']['ip'])
@@ -210,6 +246,9 @@ class TTLSTest(LFCliBase):
print("ip_map", ip_map) print("ip_map", ip_map)
print("associated_map", associated_map) print("associated_map", associated_map)
self.l3_cx_obj_udp.start()
self.l3_cx_obj_tcp.start()
time.sleep(wait_time)
# notice that this does not actually generate traffic # notice that this does not actually generate traffic
# please see test_ipv4_variable_time for example of generating traffic # please see test_ipv4_variable_time for example of generating traffic
return self.passes() return self.passes()
@@ -218,14 +257,58 @@ class TTLSTest(LFCliBase):
def stop(self): def stop(self):
# Bring stations down # Bring stations down
self.station_profile.admin_down() self.station_profile.admin_down()
self.vap_profile.admin_down(1) if self.vap:
self.vap_profile.admin_down(1)
self.l3_cx_obj_udp.stop()
self.l3_cx_obj_tcp.stop()
self.collect_endp_stats(self.l3_cx_obj_tcp.cx_profile.created_cx, traffic_type="TCP")
self.collect_endp_stats(self.l3_cx_obj_udp.cx_profile.created_cx, traffic_type="UDP")
def cleanup(self, sta_list): def cleanup(self, sta_list):
self.station_profile.cleanup(sta_list) self.station_profile.cleanup(sta_list)
self.vap_profile.cleanup(1) if self.vap:
self.vap_profile.cleanup(1)
LFUtils.wait_until_ports_disappear(base_url=self.lfclient_url, port_list=sta_list, LFUtils.wait_until_ports_disappear(base_url=self.lfclient_url, port_list=sta_list,
debug=self.debug) debug=self.debug)
def collect_endp_stats(self, endp_map, traffic_type="TCP"):
self.resulting_endpoints = {}
print("Collecting Data")
fields = "?fields=name,tx+bytes,rx+bytes"
for (cx_name, endps) in endp_map.items():
try:
endp_url = "/endp/%s%s" % (endps[0], fields)
endp_json = self.json_get(endp_url)
self.resulting_endpoints[endp_url] = endp_json
ptest_a_tx = endp_json['endpoint']['tx bytes']
ptest_a_rx = endp_json['endpoint']['rx bytes']
# ptest = self.json_get("/endp/%s?fields=tx+bytes,rx+bytes" % cx_names[cx_name]["b"])
endp_url = "/endp/%s%s" % (endps[1], fields)
endp_json = self.json_get(endp_url)
self.resulting_endpoints[endp_url] = endp_json
ptest_b_tx = endp_json['endpoint']['tx bytes']
ptest_b_rx = endp_json['endpoint']['rx bytes']
self.compare_vals("test" + traffic_type + "-A TX", ptest_a_tx)
self.compare_vals("test" + traffic_type + "-A RX", ptest_a_rx)
self.compare_vals("test" + traffic_type + "-B TX", ptest_b_tx)
self.compare_vals("test" + traffic_type + "-B RX", ptest_b_rx)
except Exception as e:
print("Is this the function having the error?")
self.error(e)
def compare_vals(self, name, postVal, print_pass=True, print_fail=True):
# print(f"Comparing {name}")
if postVal > 0:
self._pass("%s %s" % (name, postVal), print_pass)
else:
self._fail("%s did not report traffic: %s" % (name, postVal), print_fail)
def main(): def main():
parser = LFCliBase.create_basic_argparse( parser = LFCliBase.create_basic_argparse(
@@ -291,6 +374,7 @@ test_ipv4_ttls.py:
key_mgmt=args.key_mgmt, key_mgmt=args.key_mgmt,
wpa_psk=args.wpa_psk, wpa_psk=args.wpa_psk,
eap=args.eap, eap=args.eap,
vap=True,
identity=args.identity, identity=args.identity,
ttls_passwd=args.ttls_passwd, ttls_passwd=args.ttls_passwd,
ttls_realm=args.ttls_realm, ttls_realm=args.ttls_realm,

4
py-scripts/test_ipv4_variable_time.py
View File

@@ -128,7 +128,8 @@ class IPV4VariableTime(Realm):
def stop(self): def stop(self):
self.cx_profile.stop_cx() self.cx_profile.stop_cx()
self.station_profile.admin_down() if self.create_sta:
self.station_profile.admin_down()
def pre_cleanup(self): def pre_cleanup(self):
self.cx_profile.cleanup_prefix() self.cx_profile.cleanup_prefix()
@@ -157,7 +158,6 @@ class IPV4VariableTime(Realm):
self.cx_profile.create(endp_type=self.traffic_type, side_a=self.sta_list, self.cx_profile.create(endp_type=self.traffic_type, side_a=self.sta_list,
side_b=self.upstream, side_b=self.upstream,
sleep_time=0) sleep_time=0)
print("Some")
def main(): def main():
parser = Realm.create_basic_argparse( parser = Realm.create_basic_argparse(