792 lines
34 KiB
Python
792 lines
34 KiB
Python
|
#!/usr/bin/env python3.4
|
||
|
|
#
|
||
|
|
# Copyright 2016 - The Android Open Source Project
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
import time
|
||
|
|
import acts.utils
|
||
|
|
|
||
|
|
from acts import asserts
|
||
|
|
from acts import signals
|
||
|
|
from acts.test_decorators import test_tracker_info
|
||
|
|
from acts_contrib.test_utils.wifi import wifi_test_utils as wutils
|
||
|
|
from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest
|
||
|
|
|
||
|
|
WifiEnums = wutils.WifiEnums
|
||
|
|
DEFAULT_TIMEOUT = 10
|
||
|
|
|
||
|
|
# EAP Macros
|
||
|
|
EAP = WifiEnums.Eap
|
||
|
|
EapPhase2 = WifiEnums.EapPhase2
|
||
|
|
# Enterprise Config Macros
|
||
|
|
Ent = WifiEnums.Enterprise
|
||
|
|
|
||
|
|
|
||
|
|
class WifiEnterpriseTest(WifiBaseTest):
|
||
|
|
def setup_class(self):
|
||
|
|
super().setup_class()
|
||
|
|
self.enable_packet_log = True
|
||
|
|
|
||
|
|
self.dut = self.android_devices[0]
|
||
|
|
# If running in a setup with attenuators, set attenuation on all
|
||
|
|
# channels to zero.
|
||
|
|
if getattr(self, "attenuators", []):
|
||
|
|
for a in self.attenuators:
|
||
|
|
a.set_atten(0)
|
||
|
|
required_userparam_names = (
|
||
|
|
"ca_cert", "client_cert", "client_key", "eap_identity",
|
||
|
|
"eap_password", "invalid_ca_cert", "invalid_client_cert",
|
||
|
|
"invalid_client_key", "device_password", "radius_conf_2g",
|
||
|
|
"radius_conf_5g", "radius_conf_pwd", "wifi6_models")
|
||
|
|
self.unpack_userparams(required_userparam_names,
|
||
|
|
roaming_consortium_ids=None,
|
||
|
|
plmn=None,
|
||
|
|
ocsp=0)
|
||
|
|
|
||
|
|
if "AccessPoint" in self.user_params:
|
||
|
|
self.legacy_configure_ap_and_start(
|
||
|
|
ent_network=True,
|
||
|
|
radius_conf_2g=self.radius_conf_2g,
|
||
|
|
radius_conf_5g=self.radius_conf_5g,
|
||
|
|
ent_network_pwd=True,
|
||
|
|
radius_conf_pwd=self.radius_conf_pwd,
|
||
|
|
wpa_network=True,
|
||
|
|
ap_count=2,
|
||
|
|
)
|
||
|
|
elif "OpenWrtAP" in self.user_params:
|
||
|
|
self.configure_openwrt_ap_and_start(
|
||
|
|
ent_network=True,
|
||
|
|
radius_conf_2g=self.radius_conf_2g,
|
||
|
|
radius_conf_5g=self.radius_conf_5g,
|
||
|
|
ent_network_pwd=True,
|
||
|
|
radius_conf_pwd=self.radius_conf_pwd,
|
||
|
|
wpa_network=True,
|
||
|
|
ap_count=2,
|
||
|
|
)
|
||
|
|
self.ent_network_2g = self.ent_networks[0]["2g"]
|
||
|
|
self.ent_network_5g = self.ent_networks[0]["5g"]
|
||
|
|
self.ent_network_pwd = self.ent_networks_pwd[0]["2g"]
|
||
|
|
if hasattr(self, "reference_networks") and \
|
||
|
|
isinstance(self.reference_networks, list):
|
||
|
|
self.wpa_psk_2g = self.reference_networks[0]["2g"]
|
||
|
|
self.wpa_psk_5g = self.reference_networks[0]["5g"]
|
||
|
|
|
||
|
|
# Default configs for EAP networks.
|
||
|
|
self.config_peap0 = {
|
||
|
|
Ent.EAP: int(EAP.PEAP),
|
||
|
|
Ent.CA_CERT: self.ca_cert,
|
||
|
|
Ent.IDENTITY: self.eap_identity,
|
||
|
|
Ent.PASSWORD: self.eap_password,
|
||
|
|
Ent.PHASE2: int(EapPhase2.MSCHAPV2),
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_5g[WifiEnums.SSID_KEY],
|
||
|
|
Ent.OCSP: self.ocsp,
|
||
|
|
}
|
||
|
|
self.config_peap1 = dict(self.config_peap0)
|
||
|
|
self.config_peap1[WifiEnums.SSID_KEY] = \
|
||
|
|
self.ent_network_2g[WifiEnums.SSID_KEY]
|
||
|
|
self.config_tls = {
|
||
|
|
Ent.EAP: int(EAP.TLS),
|
||
|
|
Ent.CA_CERT: self.ca_cert,
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY],
|
||
|
|
Ent.CLIENT_CERT: self.client_cert,
|
||
|
|
Ent.PRIVATE_KEY_ID: self.client_key,
|
||
|
|
Ent.IDENTITY: self.eap_identity,
|
||
|
|
Ent.OCSP: self.ocsp,
|
||
|
|
}
|
||
|
|
self.config_ttls = {
|
||
|
|
Ent.EAP: int(EAP.TTLS),
|
||
|
|
Ent.CA_CERT: self.ca_cert,
|
||
|
|
Ent.IDENTITY: self.eap_identity,
|
||
|
|
Ent.PASSWORD: self.eap_password,
|
||
|
|
Ent.PHASE2: int(EapPhase2.MSCHAPV2),
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY],
|
||
|
|
Ent.OCSP: self.ocsp,
|
||
|
|
}
|
||
|
|
self.config_pwd = {
|
||
|
|
Ent.EAP: int(EAP.PWD),
|
||
|
|
Ent.IDENTITY: self.eap_identity,
|
||
|
|
Ent.PASSWORD: self.eap_password,
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_pwd[WifiEnums.SSID_KEY],
|
||
|
|
}
|
||
|
|
self.config_sim = {
|
||
|
|
Ent.EAP: int(EAP.SIM),
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY],
|
||
|
|
}
|
||
|
|
self.config_aka = {
|
||
|
|
Ent.EAP: int(EAP.AKA),
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY],
|
||
|
|
}
|
||
|
|
self.config_aka_prime = {
|
||
|
|
Ent.EAP: int(EAP.AKA_PRIME),
|
||
|
|
WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY],
|
||
|
|
}
|
||
|
|
|
||
|
|
# Set screen lock password so ConfigStore is unlocked.
|
||
|
|
self.dut.droid.setDevicePassword(self.device_password)
|
||
|
|
|
||
|
|
def teardown_class(self):
|
||
|
|
wutils.reset_wifi(self.dut)
|
||
|
|
self.dut.droid.disableDevicePassword(self.device_password)
|
||
|
|
self.dut.ed.clear_all_events()
|
||
|
|
|
||
|
|
def setup_test(self):
|
||
|
|
super().setup_test()
|
||
|
|
self.dut.droid.wifiStartTrackingStateChange()
|
||
|
|
self.dut.droid.wakeLockAcquireBright()
|
||
|
|
self.dut.droid.wakeUpNow()
|
||
|
|
wutils.reset_wifi(self.dut)
|
||
|
|
self.dut.ed.clear_all_events()
|
||
|
|
|
||
|
|
def teardown_test(self):
|
||
|
|
super().teardown_test()
|
||
|
|
self.dut.droid.wakeLockRelease()
|
||
|
|
self.dut.droid.goToSleepNow()
|
||
|
|
self.dut.droid.wifiStopTrackingStateChange()
|
||
|
|
# Turn off airplane mode
|
||
|
|
acts.utils.force_airplane_mode(self.dut, False)
|
||
|
|
wutils.set_attns(self.attenuators, "default")
|
||
|
|
|
||
|
|
"""Helper Functions"""
|
||
|
|
|
||
|
|
def eap_negative_connect_logic(self, config, ad):
|
||
|
|
"""Tries to connect to an enterprise network with invalid credentials
|
||
|
|
and expect a failure.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
config: A dict representing an invalid EAP credential.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
True if connection failed as expected, False otherwise.
|
||
|
|
"""
|
||
|
|
with asserts.assert_raises(signals.TestFailure, extras=config):
|
||
|
|
verdict = wutils.wifi_connect(ad, config)
|
||
|
|
asserts.explicit_pass("Connection failed as expected.")
|
||
|
|
|
||
|
|
def gen_negative_configs(self, config, neg_params):
|
||
|
|
"""Generic function used to generate negative configs.
|
||
|
|
|
||
|
|
For all the valid configurations, if a param in the neg_params also
|
||
|
|
exists in a config, a copy of the config is made with an invalid value
|
||
|
|
of the param.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
config: A valid configuration.
|
||
|
|
neg_params: A dict that has all the invalid values.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
An invalid configurations generated based on the valid
|
||
|
|
configuration. Each invalid configuration has a different invalid
|
||
|
|
field.
|
||
|
|
"""
|
||
|
|
negative_config = dict(config)
|
||
|
|
if negative_config in [
|
||
|
|
self.config_sim, self.config_aka, self.config_aka_prime
|
||
|
|
]:
|
||
|
|
negative_config[WifiEnums.SSID_KEY] = 'wrong_hostapd_ssid'
|
||
|
|
for k, v in neg_params.items():
|
||
|
|
# Skip negative test for TLS's identity field since it's not
|
||
|
|
# used for auth.
|
||
|
|
if config[Ent.EAP] == EAP.TLS and k == Ent.IDENTITY:
|
||
|
|
continue
|
||
|
|
if k in config:
|
||
|
|
negative_config[k] = v
|
||
|
|
negative_config["invalid_field"] = k
|
||
|
|
return negative_config
|
||
|
|
|
||
|
|
def gen_negative_eap_configs(self, config):
|
||
|
|
"""Generates invalid configurations for different EAP authentication
|
||
|
|
types.
|
||
|
|
|
||
|
|
For all the valid EAP configurations, if a param that is part of the
|
||
|
|
authentication info exists in a config, a copy of the config is made
|
||
|
|
with an invalid value of the param.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
A valid network configration
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
An invalid EAP configuration.
|
||
|
|
"""
|
||
|
|
neg_params = {
|
||
|
|
Ent.CLIENT_CERT: self.invalid_client_cert,
|
||
|
|
Ent.CA_CERT: self.invalid_ca_cert,
|
||
|
|
Ent.PRIVATE_KEY_ID: self.invalid_client_key,
|
||
|
|
Ent.IDENTITY: "fake_identity",
|
||
|
|
Ent.PASSWORD: "wrong_password"
|
||
|
|
}
|
||
|
|
return self.gen_negative_configs(config, neg_params)
|
||
|
|
|
||
|
|
def eap_connect_toggle_wifi(self, config, *args):
|
||
|
|
"""Connects to an enterprise network, toggles wifi state and ensures
|
||
|
|
that the device reconnects.
|
||
|
|
|
||
|
|
This logic expect the enterprise network to have Internet access.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
config: A dict representing a wifi enterprise configuration.
|
||
|
|
args: args to be passed to |wutils.eap_connect|.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
True if the connection is successful and Internet access works.
|
||
|
|
"""
|
||
|
|
ad = args[0]
|
||
|
|
wutils.wifi_connect(ad, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
wutils.toggle_wifi_and_wait_for_reconnection(ad,
|
||
|
|
config,
|
||
|
|
num_of_tries=5)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
def toggle_out_of_range_stress(self, stress_count=3):
|
||
|
|
"""toggle_out_of_range_stress."""
|
||
|
|
current_network = self.dut.droid.wifiGetConnectionInfo()
|
||
|
|
self.log.info("Current network: {}".format(current_network))
|
||
|
|
for count in range(stress_count):
|
||
|
|
# move the DUT out of range
|
||
|
|
self.attenuators[0].set_atten(95)
|
||
|
|
self.attenuators[1].set_atten(95)
|
||
|
|
self.attenuators[2].set_atten(95)
|
||
|
|
self.attenuators[3].set_atten(95)
|
||
|
|
time.sleep(20)
|
||
|
|
try:
|
||
|
|
wutils.start_wifi_connection_scan(self.dut)
|
||
|
|
wifi_results = self.dut.droid.wifiGetScanResults()
|
||
|
|
self.log.debug("Scan result {}".format(wifi_results))
|
||
|
|
time.sleep(20)
|
||
|
|
current_network = self.dut.droid.wifiGetConnectionInfo()
|
||
|
|
self.log.info("Current network: {}".format(current_network))
|
||
|
|
asserts.assert_true(
|
||
|
|
('network_id' in current_network and
|
||
|
|
current_network['network_id'] == -1),
|
||
|
|
"Device is connected to network {}".format(current_network))
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
finally:
|
||
|
|
self.dut.droid.wifiLockRelease()
|
||
|
|
# move the DUT back in range
|
||
|
|
wutils.set_attns(self.attenuators, "default")
|
||
|
|
time.sleep(30)
|
||
|
|
try:
|
||
|
|
wutils.start_wifi_connection_scan(self.dut)
|
||
|
|
wifi_results = self.dut.droid.wifiGetScanResults()
|
||
|
|
self.log.debug("Scan result {}".format(wifi_results))
|
||
|
|
time.sleep(20)
|
||
|
|
current_network = self.dut.droid.wifiGetConnectionInfo()
|
||
|
|
self.log.info("Current network: {}".format(current_network))
|
||
|
|
asserts.assert_true(
|
||
|
|
('network_id' in current_network and
|
||
|
|
current_network['network_id'] != -1),
|
||
|
|
"Device is disconnected to network {}".format(current_network))
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
finally:
|
||
|
|
self.dut.droid.wifiLockRelease()
|
||
|
|
|
||
|
|
def check_connection(self, network_bssid):
|
||
|
|
"""Check current wifi connection networks.
|
||
|
|
Args:
|
||
|
|
network_bssid: Network bssid to which connection.
|
||
|
|
Returns:
|
||
|
|
True if connection to given network happen, else return False.
|
||
|
|
"""
|
||
|
|
time.sleep(10) #time for connection state to be updated
|
||
|
|
self.log.info("Check network for {}".format(network_bssid))
|
||
|
|
current_network = self.dut.droid.wifiGetConnectionInfo()
|
||
|
|
self.log.debug("Current network: {}".format(current_network))
|
||
|
|
if WifiEnums.BSSID_KEY in current_network:
|
||
|
|
return current_network[WifiEnums.BSSID_KEY] == network_bssid
|
||
|
|
return False
|
||
|
|
|
||
|
|
""" Tests """
|
||
|
|
|
||
|
|
# EAP connect tests
|
||
|
|
""" Test connecting to enterprise networks of different authentication
|
||
|
|
types.
|
||
|
|
|
||
|
|
The authentication types tested are:
|
||
|
|
EAP-TLS
|
||
|
|
EAP-PEAP with different phase2 types.
|
||
|
|
EAP-TTLS with different phase2 types.
|
||
|
|
|
||
|
|
Procedures:
|
||
|
|
For each enterprise wifi network
|
||
|
|
1. Connect to the network.
|
||
|
|
2. Send a GET request to a website and check response.
|
||
|
|
|
||
|
|
Expect:
|
||
|
|
Successful connection and Internet access through the enterprise
|
||
|
|
networks.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="4e720cac-ea17-4de7-a540-8dc7c49f9713")
|
||
|
|
def test_eap_connect_with_config_tls(self):
|
||
|
|
wutils.wifi_connect(self.dut, self.config_tls)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="10e3a5e9-0018-4162-a9fa-b41500f13340")
|
||
|
|
def test_eap_connect_with_config_pwd(self):
|
||
|
|
wutils.wifi_connect(self.dut, self.config_pwd)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="b4513f78-a1c4-427f-bfc7-2a6b3da714b5")
|
||
|
|
def test_eap_connect_with_config_sim(self):
|
||
|
|
wutils.wifi_connect(self.dut, self.config_sim)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="7d390e30-cb67-4b55-bf00-567adad2d9b0")
|
||
|
|
def test_eap_connect_with_config_aka(self):
|
||
|
|
wutils.wifi_connect(self.dut, self.config_aka)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="742f921b-27c3-4b68-a3ca-88e64fe79c1d")
|
||
|
|
def test_eap_connect_with_config_aka_prime(self):
|
||
|
|
wutils.wifi_connect(self.dut, self.config_aka_prime)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="d34e30f3-6ef6-459f-b47a-e78ed90ce4c6")
|
||
|
|
def test_eap_connect_with_config_ttls_none(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="0dca3a15-472e-427c-8e06-4e38088ee973")
|
||
|
|
def test_eap_connect_with_config_ttls_pap(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="47c4b459-2cb1-4fc7-b4e7-82534e8e090e")
|
||
|
|
def test_eap_connect_with_config_ttls_mschap(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="fdb286c7-8069-481d-baf0-c5dd7a31ff03")
|
||
|
|
def test_eap_connect_with_config_ttls_mschapv2(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="d9315962-7987-4ee7-905d-6972c78ce8a1")
|
||
|
|
def test_eap_connect_with_config_ttls_gtc(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="90a67bd3-30da-4daf-8ab0-d964d7ad19be")
|
||
|
|
def test_eap_connect_with_config_peap0_mschapv2(self):
|
||
|
|
config = dict(self.config_peap0)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="3c451ba4-0c83-4eef-bc95-db4c21893008")
|
||
|
|
def test_eap_connect_with_config_peap0_gtc(self):
|
||
|
|
config = dict(self.config_peap0)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="6b45157d-0325-417a-af18-11af5d240d79")
|
||
|
|
def test_eap_connect_with_config_peap1_mschapv2(self):
|
||
|
|
config = dict(self.config_peap1)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="1663decc-71ae-4f95-a027-8a6dbf9c337f")
|
||
|
|
def test_eap_connect_with_config_peap1_gtc(self):
|
||
|
|
config = dict(self.config_peap1)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
wutils.wifi_connect(self.dut, config)
|
||
|
|
wutils.verify_11ax_wifi_connection(
|
||
|
|
self.dut, self.wifi6_models, "wifi6_ap" in self.user_params)
|
||
|
|
|
||
|
|
# EAP connect negative tests
|
||
|
|
""" Test connecting to enterprise networks.
|
||
|
|
|
||
|
|
Procedures:
|
||
|
|
For each enterprise wifi network
|
||
|
|
1. Connect to the network with invalid credentials.
|
||
|
|
|
||
|
|
Expect:
|
||
|
|
Fail to establish connection.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="b2a91f1f-ccd7-4bd1-ab81-19aab3d8ee38")
|
||
|
|
def test_eap_connect_negative_with_config_tls(self):
|
||
|
|
config = self.gen_negative_eap_configs(self.config_tls)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="b7fb8517-5d52-468e-890a-40ea24129bf1")
|
||
|
|
def test_network_selection_status_wpa2_eap_tls_invalid_cert(self):
|
||
|
|
config = self.gen_negative_eap_configs(self.config_tls)
|
||
|
|
try:
|
||
|
|
wutils.connect_to_wifi_network(self.dut, config)
|
||
|
|
asserts.fail(
|
||
|
|
"WPA2 EAP TLS worked with invalid cert. Expected to fail.")
|
||
|
|
except:
|
||
|
|
asserts.assert_true(
|
||
|
|
self.dut.droid.wifiIsNetworkTemporaryDisabledForNetwork(config),
|
||
|
|
"WiFi network is not temporary disabled.")
|
||
|
|
asserts.explicit_pass(
|
||
|
|
"Connection failed with correct network selection status.")
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="6466abde-1d16-4168-9dd8-1e7a0a19889b")
|
||
|
|
def test_eap_connect_negative_with_config_pwd(self):
|
||
|
|
config = self.gen_negative_eap_configs(self.config_pwd)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="d7742a2a-85b0-409a-99d8-47711ddc5612")
|
||
|
|
def test_eap_connect_negative_with_config_sim(self):
|
||
|
|
config = self.gen_negative_eap_configs(self.config_sim)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="0ec0de93-cab3-4f41-960b-c0af64ff48c4")
|
||
|
|
def test_eap_connect_negative_with_config_aka(self):
|
||
|
|
config = self.gen_negative_eap_configs(self.config_aka)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="bb640ea4-32a6-48ea-87c9-f7128fffbbf6")
|
||
|
|
def test_eap_connect_negative_with_config_aka_prime(self):
|
||
|
|
config = self.gen_negative_eap_configs(self.config_aka_prime)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="86336ada-0ced-45a4-8a22-c4aa23c81111")
|
||
|
|
def test_eap_connect_negative_with_config_ttls_none(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="71e0498d-9973-4958-94bd-79051c328920")
|
||
|
|
def test_eap_connect_negative_with_config_ttls_pap(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="c04142a8-b204-4d2d-98dc-150b16c8397e")
|
||
|
|
def test_eap_connect_negative_with_config_ttls_mschap(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="625e7aa5-e3e6-4bbe-98c0-5aad8ca1555b")
|
||
|
|
def test_eap_connect_negative_with_config_ttls_mschapv2(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="24ea0d80-0a3f-41c2-8e05-d6387e589058")
|
||
|
|
def test_eap_connect_negative_with_config_ttls_gtc(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="b7c1f0f8-6338-4501-8e1d-c9b136aaba88")
|
||
|
|
def test_eap_connect_negative_with_config_peap0_mschapv2(self):
|
||
|
|
config = dict(self.config_peap0)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="9cf83dcb-38ad-4f75-9ea9-98de1cfaf7f3")
|
||
|
|
def test_eap_connect_negative_with_config_peap0_gtc(self):
|
||
|
|
config = dict(self.config_peap0)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="89bb2b6b-d073-402a-bdc1-68ac5f8752a3")
|
||
|
|
def test_eap_connect_negative_with_config_peap1_mschapv2(self):
|
||
|
|
config = dict(self.config_peap1)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="2252a864-9ff7-43b5-82d9-afe57d1f5e5f")
|
||
|
|
def test_eap_connect_negative_with_config_peap1_gtc(self):
|
||
|
|
config = dict(self.config_peap1)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
config = self.gen_negative_eap_configs(config)
|
||
|
|
self.eap_negative_connect_logic(config, self.dut)
|
||
|
|
|
||
|
|
# EAP connect config store tests
|
||
|
|
""" Test connecting to enterprise networks of different authentication
|
||
|
|
types after wifi toggle.
|
||
|
|
|
||
|
|
The authentication types tested are:
|
||
|
|
EAP-TLS
|
||
|
|
EAP-PEAP with different phase2 types.
|
||
|
|
EAP-TTLS with different phase2 types.
|
||
|
|
|
||
|
|
Procedures:
|
||
|
|
For each enterprise wifi network
|
||
|
|
1. Connect to the network.
|
||
|
|
2. Send a GET request to a website and check response.
|
||
|
|
3. Toggle wifi.
|
||
|
|
4. Ensure that the device reconnects to the same network.
|
||
|
|
|
||
|
|
Expect:
|
||
|
|
Successful connection and Internet access through the enterprise
|
||
|
|
networks.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="2a933b7f-27d7-4201-a34f-25b9d8072a8c")
|
||
|
|
def test_eap_connect_config_store_with_config_tls(self):
|
||
|
|
self.eap_connect_toggle_wifi(self.config_tls, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="08dc071b-9fea-408a-a3f6-d3493869f6d4")
|
||
|
|
def test_eap_connect_config_store_with_config_pwd(self):
|
||
|
|
self.eap_connect_toggle_wifi(self.config_pwd, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="230cb03e-58bc-41cb-b9b3-7215c2ab2325")
|
||
|
|
def test_eap_connect_config_store_with_config_sim(self):
|
||
|
|
self.eap_connect_toggle_wifi(self.config_sim, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="dfc3e59c-2309-4598-8c23-bb3fe95ef89f")
|
||
|
|
def test_eap_connect_config_store_with_config_aka(self):
|
||
|
|
self.eap_connect_toggle_wifi(self.config_aka, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="6050a1d1-4f3a-476d-bf93-638abd066790")
|
||
|
|
def test_eap_connect_config_store_with_config_aka_prime(self):
|
||
|
|
self.eap_connect_toggle_wifi(self.config_aka_prime, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="03108057-cc44-4a80-8331-77c93694099c")
|
||
|
|
def test_eap_connect_config_store_with_config_ttls_none(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="53dd8195-e272-4589-a261-b8fa3607ad8d")
|
||
|
|
def test_eap_connect_config_store_with_config_ttls_pap(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="640f697b-9c62-4b19-bd76-53b236a152e0")
|
||
|
|
def test_eap_connect_config_store_with_config_ttls_mschap(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="f0243684-fae0-46f3-afbd-bf525fc712e2")
|
||
|
|
def test_eap_connect_config_store_with_config_ttls_mschapv2(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="49ec7202-3b00-49c3-970a-201360888c74")
|
||
|
|
def test_eap_connect_config_store_with_config_ttls_gtc(self):
|
||
|
|
config = dict(self.config_ttls)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="1c6abfa3-f344-4e28-b891-5481ab79efcf")
|
||
|
|
def test_eap_connect_config_store_with_config_peap0_mschapv2(self):
|
||
|
|
config = dict(self.config_peap0)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="2815bc76-49fa-43a5-a4b6-84788f9809d5")
|
||
|
|
def test_eap_connect_config_store_with_config_peap0_gtc(self):
|
||
|
|
config = dict(self.config_peap0)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="e93f7472-6895-4e36-bff2-9b2dcfd07ad0")
|
||
|
|
def test_eap_connect_config_store_with_config_peap1_mschapv2(self):
|
||
|
|
config = dict(self.config_peap1)
|
||
|
|
config[
|
||
|
|
WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="6da72fa0-b858-4475-9559-46fe052d0d64")
|
||
|
|
def test_eap_connect_config_store_with_config_peap1_gtc(self):
|
||
|
|
config = dict(self.config_peap1)
|
||
|
|
config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value
|
||
|
|
self.eap_connect_toggle_wifi(config, self.dut)
|
||
|
|
|
||
|
|
# Airplane mode on with wifi connect tests
|
||
|
|
""" Test connecting to enterprise networks of different authentication
|
||
|
|
types after airplane mode on.
|
||
|
|
|
||
|
|
The authentication types tested are:
|
||
|
|
EAP-SIM
|
||
|
|
EAP-AKA
|
||
|
|
EAP-AKA_PRIME
|
||
|
|
|
||
|
|
Procedures:
|
||
|
|
For each enterprise wifi network
|
||
|
|
1. Turn on Airplane mode
|
||
|
|
2. Toggle wifi..
|
||
|
|
3. Ensure that the device connects to the enterprise network.
|
||
|
|
|
||
|
|
Expect:
|
||
|
|
Successful connection and Internet access through the enterprise
|
||
|
|
networks with Airplane mode on.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="54b96a6c-f366-421c-9a72-80d7ee8fac8f")
|
||
|
|
def test_eap_connect_with_config_sim_airplane_on(self):
|
||
|
|
self.log.info("Toggling Airplane mode ON")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, True),
|
||
|
|
"Can not turn on airplane mode on: %s" % self.dut.serial)
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
wutils.wifi_toggle_state(self.dut, True)
|
||
|
|
wutils.wifi_connect(self.dut, self.config_sim)
|
||
|
|
self.log.info("Toggling Airplane mode OFF")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, False),
|
||
|
|
"Can not turn OFF airplane mode: %s" % self.dut.serial)
|
||
|
|
wutils.start_wifi_connection_scan_and_ensure_network_found(
|
||
|
|
self.dut, self.config_sim[WifiEnums.SSID_KEY])
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="344f63f6-7c99-4507-8036-757f9f911d20")
|
||
|
|
def test_eap_connect_with_config_aka_airplane_on(self):
|
||
|
|
self.log.info("Toggling Airplane mode ON")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, True),
|
||
|
|
"Can not turn on airplane mode on: %s" % self.dut.serial)
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
wutils.wifi_toggle_state(self.dut, True)
|
||
|
|
wutils.wifi_connect(self.dut, self.config_aka)
|
||
|
|
self.log.info("Toggling Airplane mode OFF")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, False),
|
||
|
|
"Can not turn OFF airplane mode: %s" % self.dut.serial)
|
||
|
|
wutils.start_wifi_connection_scan_and_ensure_network_found(
|
||
|
|
self.dut, self.config_aka[WifiEnums.SSID_KEY])
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="5502b8c8-89d7-4ce9-afee-cae50e71f5f4")
|
||
|
|
def test_eap_connect_with_config_aka_prime_airplane_on(self):
|
||
|
|
self.log.info("Toggling Airplane mode ON")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, True),
|
||
|
|
"Can not turn on airplane mode on: %s" % self.dut.serial)
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
wutils.wifi_toggle_state(self.dut, True)
|
||
|
|
wutils.wifi_connect(self.dut, self.config_aka_prime)
|
||
|
|
self.log.info("Toggling Airplane mode OFF")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, False),
|
||
|
|
"Can not turn OFF airplane mode: %s" % self.dut.serial)
|
||
|
|
wutils.start_wifi_connection_scan_and_ensure_network_found(
|
||
|
|
self.dut, self.config_aka_prime[WifiEnums.SSID_KEY])
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="360a6fec-f4ee-4ecd-9b15-e836c977d6db")
|
||
|
|
def test_connect_eap_sim_network_out_of_range_back(self):
|
||
|
|
"""Test connecting to enterprise networks to do out of range
|
||
|
|
then back in range 3 times
|
||
|
|
1. Connecting EAP-SIM network
|
||
|
|
2. Move DUT out of range then back in range 3 times
|
||
|
|
3. Check that device is connected to network.
|
||
|
|
"""
|
||
|
|
wutils.wifi_connect(self.dut, self.config_sim)
|
||
|
|
self.toggle_out_of_range_stress()
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="9fb71afb-5599-4ca1-b458-09752c40bb0d")
|
||
|
|
def test_eap_sim_network_out_of_range_back_airplane(self):
|
||
|
|
"""Test connecting to enterprise networks with airplne mode on
|
||
|
|
to do out of range then back in range 3 times
|
||
|
|
1. Turn on airplane mode
|
||
|
|
2. Connecting EAP-SIM network
|
||
|
|
3. Move DUT out of range then back in range 3 times
|
||
|
|
4. Check that device is connected to network.
|
||
|
|
"""
|
||
|
|
self.log.debug("Toggling Airplane mode ON")
|
||
|
|
asserts.assert_true(
|
||
|
|
acts.utils.force_airplane_mode(self.dut, True),
|
||
|
|
"Can not turn on airplane mode on: %s" % self.dut.serial)
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
self.log.debug("Toggling WiFi mode ON")
|
||
|
|
wutils.wifi_toggle_state(self.dut, True)
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
wutils.wifi_connect(self.dut, self.config_sim)
|
||
|
|
self.toggle_out_of_range_stress()
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="9e899c55-1a62-498c-bbf1-e9472e42e84f")
|
||
|
|
def test_eap_sim_network_reboot(self):
|
||
|
|
"""Test connecting to enterprise networks with airplne mode on
|
||
|
|
to do out of range then back in range 3 times
|
||
|
|
1. Connecting EAP-SIM network
|
||
|
|
3. Check that device is connected to network after reboot.
|
||
|
|
"""
|
||
|
|
self.dut.droid.disableDevicePassword(self.device_password)
|
||
|
|
wutils.wifi_connect(self.dut, self.config_sim)
|
||
|
|
current_network = self.dut.droid.wifiGetConnectionInfo()
|
||
|
|
self.log.info("Current network: {}".format(current_network))
|
||
|
|
self.dut.reboot()
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
self.check_connection(self.config_sim)
|
||
|
|
|
||
|
|
@test_tracker_info(uuid="8e7465fb-5b16-4abb-92d8-a2c79355e377")
|
||
|
|
def test_connect_to_EAP_SIM_network_switch_to_WPA2(self):
|
||
|
|
"""Test connecting PSK's AP1 and one EAP AP2 network switch test
|
||
|
|
1. Connect to a PSK's AP1 before connect to EAP-SIM AP2 network.
|
||
|
|
2. Out of PSK's AP1 range.
|
||
|
|
3. Connect to EAP-SIM network, then in AP1 range to switch WPA2-PSK network.
|
||
|
|
"""
|
||
|
|
attn1 = self.attenuators[0]
|
||
|
|
attn2 = self.attenuators[2]
|
||
|
|
if "OpenWrtAP" in self.user_params:
|
||
|
|
attn2 = self.attenuators[1]
|
||
|
|
ap1_network = self.config_sim
|
||
|
|
ap2_network = self.reference_networks[1]["2g"]
|
||
|
|
attn1.set_atten(0)
|
||
|
|
attn2.set_atten(0)
|
||
|
|
wutils.connect_to_wifi_network(self.dut, ap2_network)
|
||
|
|
#Enable EAP network signal
|
||
|
|
attn2.set_atten(95)
|
||
|
|
time.sleep(DEFAULT_TIMEOUT)
|
||
|
|
wutils.connect_to_wifi_network(self.dut, ap1_network)
|
||
|
|
current_network = self.dut.droid.wifiGetConnectionInfo()
|
||
|
|
self.log.info("Current network: {}".format(current_network))
|
||
|
|
#Enable SSID1 network signal
|
||
|
|
attn1.set_atten(95)
|
||
|
|
attn2.set_atten(0)
|
||
|
|
time.sleep(20)
|
||
|
|
self.check_connection(ap2_network)
|