1# 2# Copyright 2016 - The Android Open Source Project 3# 4# Licensed under the Apache License, Version 2.0 (the "License"); 5# you may not use this file except in compliance with the License. 6# You may obtain a copy of the License at 7# 8# http://www.apache.org/licenses/LICENSE-2.0 9# 10# Unless required by applicable law or agreed to in writing, software 11# distributed under the License is distributed on an "AS IS" BASIS, 12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13# See the License for the specific language governing permissions and 14# limitations under the License. 15 16from acts.test_decorators import test_tracker_info 17from acts_contrib.test_utils.wifi import wifi_test_utils as wutils 18from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest 19 20WifiEnums = wutils.WifiEnums 21 22# EAP Macros 23EAP = WifiEnums.Eap 24EapPhase2 = WifiEnums.EapPhase2 25 26# Enterprise Config Macros 27Ent = WifiEnums.Enterprise 28 29 30class WifiEnterpriseRoamingTest(WifiBaseTest): 31 32 def __init__(self, configs): 33 super().__init__(configs) 34 self.enable_packet_log = True 35 36 def setup_class(self): 37 super().setup_class() 38 39 self.dut = self.android_devices[0] 40 req_params = ( 41 "roaming_attn", 42 # Expected time within which roaming should finish, in seconds. 43 "roam_interval", 44 "ca_cert", 45 "client_cert", 46 "client_key", 47 "eap_identity", 48 "eap_password", 49 "device_password", 50 "wifi6_models", 51 "radius_conf_2g", 52 "radius_conf_5g") 53 self.unpack_userparams(req_params) 54 if "AccessPoint" in self.user_params: 55 self.legacy_configure_ap_and_start( 56 mirror_ap=True, 57 ent_network=True, 58 ap_count=2, 59 radius_conf_2g=self.radius_conf_2g, 60 radius_conf_5g=self.radius_conf_5g,) 61 elif "OpenWrtAP" in self.user_params: 62 self.configure_openwrt_ap_and_start( 63 mirror_ap=True, 64 ent_network=True, 65 ap_count=2, 66 radius_conf_2g=self.radius_conf_2g, 67 radius_conf_5g=self.radius_conf_5g,) 68 self.ent_network_2g_a = self.ent_networks[0]["2g"] 69 self.ent_network_2g_b = self.ent_networks[1]["2g"] 70 self.ent_roaming_ssid = self.ent_network_2g_a[WifiEnums.SSID_KEY] 71 if "AccessPoint" in self.user_params: 72 self.bssid_a = self.ent_network_2g_a[WifiEnums.BSSID_KEY.lower()] 73 self.bssid_b = self.ent_network_2g_b[WifiEnums.BSSID_KEY.lower()] 74 elif "OpenWrtAP" in self.user_params: 75 self.bssid_a = self.bssid_map[0]["2g"][self.ent_roaming_ssid] 76 self.bssid_b = self.bssid_map[1]["2g"][self.ent_roaming_ssid] 77 78 self.config_peap = { 79 Ent.EAP: int(EAP.PEAP), 80 Ent.CA_CERT: self.ca_cert, 81 Ent.IDENTITY: self.eap_identity, 82 Ent.PASSWORD: self.eap_password, 83 Ent.PHASE2: int(EapPhase2.MSCHAPV2), 84 WifiEnums.SSID_KEY: self.ent_roaming_ssid 85 } 86 self.config_tls = { 87 Ent.EAP: int(EAP.TLS), 88 Ent.CA_CERT: self.ca_cert, 89 WifiEnums.SSID_KEY: self.ent_roaming_ssid, 90 Ent.CLIENT_CERT: self.client_cert, 91 Ent.PRIVATE_KEY_ID: self.client_key, 92 Ent.IDENTITY: self.eap_identity, 93 } 94 self.config_ttls = { 95 Ent.EAP: int(EAP.TTLS), 96 Ent.CA_CERT: self.ca_cert, 97 Ent.IDENTITY: self.eap_identity, 98 Ent.PASSWORD: self.eap_password, 99 Ent.PHASE2: int(EapPhase2.MSCHAPV2), 100 WifiEnums.SSID_KEY: self.ent_roaming_ssid 101 } 102 self.config_sim = { 103 Ent.EAP: int(EAP.SIM), 104 WifiEnums.SSID_KEY: self.ent_roaming_ssid, 105 } 106 self.attn_a = self.attenuators[0] 107 self.attn_b = self.attenuators[2] 108 if "OpenWrtAP" in self.user_params: 109 self.attn_b = self.attenuators[1] 110 # Set screen lock password so ConfigStore is unlocked. 111 self.dut.droid.setDevicePassword(self.device_password) 112 wutils.set_attns(self.attenuators, "default") 113 114 def teardown_class(self): 115 wutils.reset_wifi(self.dut) 116 self.dut.droid.disableDevicePassword(self.device_password) 117 self.dut.ed.clear_all_events() 118 wutils.set_attns(self.attenuators, "default") 119 120 def setup_test(self): 121 super().setup_test() 122 self.dut.droid.wifiStartTrackingStateChange() 123 self.dut.droid.wakeLockAcquireBright() 124 self.dut.droid.wakeUpNow() 125 wutils.reset_wifi(self.dut) 126 self.dut.ed.clear_all_events() 127 128 def teardown_test(self): 129 super().teardown_test() 130 self.dut.droid.wakeLockRelease() 131 self.dut.droid.goToSleepNow() 132 self.dut.droid.wifiStopTrackingStateChange() 133 wutils.set_attns(self.attenuators, "default") 134 135 def trigger_roaming_and_validate(self, attn_val_name, expected_con): 136 """Sets attenuators to trigger roaming and validate the DUT connected 137 to the BSSID expected. 138 139 Args: 140 attn_val_name: Name of the attenuation value pair to use. 141 expected_con: The expected info of the network to we expect the DUT 142 to roam to. 143 """ 144 wutils.set_attns_steps( 145 self.attenuators, attn_val_name, self.roaming_attn) 146 self.log.info("Wait %ss for roaming to finish.", self.roam_interval) 147 try: 148 self.dut.droid.wakeLockAcquireBright() 149 self.dut.droid.wakeUpNow() 150 wutils.verify_wifi_connection_info(self.dut, expected_con) 151 expected_bssid = expected_con[WifiEnums.BSSID_KEY] 152 self.log.info("Roamed to %s successfully", expected_bssid) 153 finally: 154 self.dut.droid.wifiLockRelease() 155 self.dut.droid.goToSleepNow() 156 157 def roaming_between_a_and_b_logic(self, config): 158 """Test roaming between two enterprise APs. 159 160 Steps: 161 1. Make bssid_a visible, bssid_b not visible. 162 2. Connect to ent_roaming_ssid. Expect DUT to connect to bssid_a. 163 3. Make bssid_a not visible, bssid_b visible. 164 4. Expect DUT to roam to bssid_b. 165 5. Make bssid_a visible, bssid_b not visible. 166 6. Expect DUT to roam back to bssid_a. 167 """ 168 expected_con_to_a = { 169 WifiEnums.SSID_KEY: self.ent_roaming_ssid, 170 WifiEnums.BSSID_KEY: self.bssid_a, 171 } 172 expected_con_to_b = { 173 WifiEnums.SSID_KEY: self.ent_roaming_ssid, 174 WifiEnums.BSSID_KEY: self.bssid_b, 175 } 176 wutils.set_attns_steps( 177 self.attenuators, "AP1_on_AP2_off", self.roaming_attn) 178 wutils.connect_to_wifi_network(self.dut, config) 179 wutils.verify_11ax_wifi_connection( 180 self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) 181 wutils.verify_wifi_connection_info(self.dut, expected_con_to_a) 182 self.log.info("Roaming from %s to %s", self.bssid_a, self.bssid_b) 183 self.trigger_roaming_and_validate("AP1_off_AP2_on", expected_con_to_b) 184 wutils.verify_11ax_wifi_connection( 185 self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) 186 self.log.info("Roaming from %s to %s", self.bssid_b, self.bssid_a) 187 self.trigger_roaming_and_validate("AP1_on_AP2_off", expected_con_to_a) 188 wutils.verify_11ax_wifi_connection( 189 self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) 190 191 """ Tests Begin """ 192 193 @test_tracker_info(uuid="b15e4b3f-841d-428d-87ac-272f29f06e14") 194 def test_roaming_with_config_tls(self): 195 self.roaming_between_a_and_b_logic(self.config_tls) 196 197 @test_tracker_info(uuid="d349cfec-b4af-48b2-88b7-744f5de25d43") 198 def test_roaming_with_config_ttls_none(self): 199 config = dict(self.config_ttls) 200 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value 201 self.roaming_between_a_and_b_logic(config) 202 203 @test_tracker_info(uuid="89b8161c-754e-4138-831d-5fe40f521ce4") 204 def test_roaming_with_config_ttls_pap(self): 205 config = dict(self.config_ttls) 206 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value 207 self.roaming_between_a_and_b_logic(config) 208 209 @test_tracker_info(uuid="d4925470-924b-4d03-8437-83e26b5f2df3") 210 def test_roaming_with_config_ttls_mschap(self): 211 config = dict(self.config_ttls) 212 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value 213 self.roaming_between_a_and_b_logic(config) 214 215 @test_tracker_info(uuid="206b1327-dd9c-4742-8717-e7bf2a04eed6") 216 def test_roaming_with_config_ttls_mschapv2(self): 217 config = dict(self.config_ttls) 218 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value 219 self.roaming_between_a_and_b_logic(config) 220 221 @test_tracker_info(uuid="c2c0168b-2933-4954-af62-fb41f42dc45a") 222 def test_roaming_with_config_ttls_gtc(self): 223 config = dict(self.config_ttls) 224 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value 225 self.roaming_between_a_and_b_logic(config) 226 227 @test_tracker_info(uuid="481c4102-8f5b-4fcd-95cc-5e3285f47985") 228 def test_roaming_with_config_peap_mschapv2(self): 229 config = dict(self.config_peap) 230 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value 231 self.roaming_between_a_and_b_logic(config) 232 233 @test_tracker_info(uuid="404155d4-33a7-42b3-b369-5f2d63d19f16") 234 def test_roaming_with_config_peap_gtc(self): 235 config = dict(self.config_peap) 236 config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value 237 self.roaming_between_a_and_b_logic(config) 238 239 """ Tests End """ 240