| #!/usr/bin/env python3 |
| # |
| # Copyright 2020 - 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 |
| |
| from acts import asserts |
| from acts import utils |
| from acts import signals |
| from acts.controllers.openwrt_lib.openwrt_constants import OpenWrtWifiSecurity |
| from acts.test_decorators import test_tracker_info |
| import acts_contrib.test_utils.wifi.wifi_test_utils as wutils |
| from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest |
| |
| |
| WifiEnums = wutils.WifiEnums |
| |
| |
| class WifiWpaPersonalTest(WifiBaseTest): |
| """Test for WPA Personal. |
| |
| Test Bed Requirement: |
| * One Android device without sim card. |
| * One OpenWrt Wi-Fi AP. |
| """ |
| |
| def setup_class(self): |
| super().setup_class() |
| self.dut = self.android_devices[0] |
| |
| if "OpenWrtAP" in self.user_params: |
| self.openwrt = self.access_points[0] |
| self.configure_openwrt_ap_and_start(wpa1_network=True) |
| |
| req_params = ["OpenWrtAP", "roaming_attn"] |
| opt_params = ["pixel_models", "cnss_diag_file"] |
| self.unpack_userparams(req_params, opt_params) |
| self.wpa_psk_2g = self.wpa1_networks[0]["2g"] |
| self.wpa_psk_5g = self.wpa1_networks[0]["5g"] |
| |
| def setup_test(self): |
| super().setup_test() |
| for ad in self.android_devices: |
| ad.droid.wakeLockAcquireBright() |
| ad.droid.wakeUpNow() |
| wutils.wifi_toggle_state(ad, True) |
| |
| def teardown_test(self): |
| super().teardown_test() |
| for ad in self.android_devices: |
| ad.droid.wakeLockRelease() |
| ad.droid.goToSleepNow() |
| wutils.reset_wifi(self.dut) |
| utils.force_airplane_mode(self.dut, False) |
| |
| def teardown_class(self): |
| super().teardown_class() |
| |
| def on_fail(self, test_name, begin_time): |
| super().on_fail(test_name, begin_time) |
| self.dut.cat_adb_log(test_name, begin_time) |
| self.dut.take_bug_report(test_name, begin_time) |
| |
| def verify_wpa_network_encryption(self, encryption): |
| result = wutils.get_wlan0_link(self.dut) |
| if encryption == "psk+ccmp": |
| asserts.assert_true( |
| result["pairwise_cipher"] == "CCMP" and |
| result["group_cipher"] == "CCMP", |
| "DUT does not connect to {} encryption network".format(encryption)) |
| elif encryption == "psk+tkip": |
| asserts.assert_true( |
| result["pairwise_cipher"] == "TKIP" and |
| result["group_cipher"] == "TKIP", |
| "DUT does not connect to {} encryption network".format(encryption)) |
| elif encryption == "psk+tkip+ccmp": |
| asserts.assert_true( |
| result["pairwise_cipher"] == "CCMP" and |
| result["group_cipher"] == "TKIP", |
| "DUT does not connect to {} encryption network".format(encryption)) |
| |
| ### Tests ### |
| |
| @test_tracker_info(uuid="0c68a772-b70c-47d6-88ab-1b069c1d8005") |
| def test_connect_to_wpa_psk_ccmp_2g(self): |
| """Test connection between DUT and WPA PSK CCMP 2G. |
| |
| Steps: |
| Change AP's security type to "WPA" and cipher to "CCMP". |
| Connect to 2g network. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| |
| @test_tracker_info(uuid="4722dffc-2960-4459-9729-0f8114af2321") |
| def test_connect_to_wpa_psk_ccmp_5g(self): |
| """Test connection between DUT and WPA PSK CCMP 5G. |
| |
| Steps: |
| Change AP's security type to "WPA" and cipher to "CCMP". |
| Connect to 5g network. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_5g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_5g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| |
| @test_tracker_info(uuid="4759503e-ef9c-430b-9306-b96a347ca3de") |
| def test_connect_to_wpa_psk_tkip_2g(self): |
| """Test connection between DUT and WPA PSK TKIP 2G. |
| |
| Steps: |
| Change AP's security type to "WPA" and cipher to "TKIP". |
| Connect to 2g network. |
| """ |
| self.openwrt.log.info("Enable WPA-TKIP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_TKIP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_TKIP) |
| |
| @test_tracker_info(uuid="9c836ca6-af14-4d6b-a98e-227fb29e84ee") |
| def test_connect_to_wpa_psk_tkip_5g(self): |
| """Test connection between DUT and WPA PSK TKIP 5G. |
| |
| Steps: |
| Change AP's security type to "WPA" and cipher to "TKIP". |
| Connect to 5g network. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK TKIP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_TKIP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_5g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_5g) |
| |
| @test_tracker_info(uuid="c03b362b-cd03-4e34-a99a-ef80a9db6db9") |
| def test_connect_to_wpa_psk_tkip_and_ccmp_2g(self): |
| """Test connection between DUT and WPA PSK CCMP+TKIP 2G. |
| |
| Steps: |
| Change AP's security type to "WPA" and cipher to "CCMP and TKIP". |
| Connect to 2g network. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP and TKIP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_TKIP_AND_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption( |
| OpenWrtWifiSecurity.WPA_PSK_TKIP_AND_CCMP) |
| |
| @test_tracker_info(uuid="203d7e7f-536d-4feb-9aa2-648f1f9a685d") |
| def test_connect_to_wpa_psk_tkip_and_ccmp_5g(self): |
| """Test connection between DUT and WPA PSK CCMP+TKIP 5G. |
| |
| Steps: |
| Change AP's security type to "WPA" and cipher to "CCMP and TKIP". |
| Connect to 5g network. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP and TKIP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_TKIP_AND_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_5g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_5g) |
| self.verify_wpa_network_encryption( |
| OpenWrtWifiSecurity.WPA_PSK_TKIP_AND_CCMP) |
| |
| @test_tracker_info(uuid="20a41f61-4fda-4fe9-82ee-482ecd8c82eb") |
| def test_connect_to_wpa_psk_ccmp_2g_after_airplane_mode(self): |
| """Test Wi-Fi reconnection after enabling Airplane Mode. |
| |
| Steps: |
| DUT connect to 2GHz Wi-Fi network. |
| DUT turns ON Airplane Mode. |
| DUT turns ON Wi-Fi. |
| DUT verify internet connection with HTTP ping. |
| DUT turns OFF Airplane Mode. |
| DUT verify internet connection with HTTP ping. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| # Turn ON DUT"s Airplane Mode. |
| self.dut.log.info("Toggle Airplane Mode ON") |
| utils.force_airplane_mode(self.dut, True) |
| self.dut.log.info("Toggle Wi-Fi ON") |
| # Turn ON DUT"s Wi-Fi |
| wutils.wifi_toggle_state(self.dut, True) |
| wutils.wait_for_connect(self.dut, |
| self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.validate_connection(self.dut) |
| utils.force_airplane_mode(self.dut, False) |
| wutils.validate_connection(self.dut) |
| |
| @test_tracker_info(uuid="df89c92b-a30c-4485-ab45-daef5240c027") |
| def test_connect_to_wpa_psk_ccmp_2g_after_wifi_off(self): |
| """Test Wi-Fi reconnection after Turn OFF Wi-Fi. |
| |
| Steps: |
| DUT connect to 2GHz Wi-Fi network. |
| DUT turns OFF Wi-Fi. |
| DUT turns ON Wi-Fi. |
| DUT verify internet connection with HTTP ping. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| self.dut.log.info("Toggle Wi-Fi OFF") |
| # Turn OFF DUT"s Wi-Fi then Turn if ON. |
| wutils.wifi_toggle_state(self.dut, False) |
| wutils.wifi_toggle_state(self.dut, True) |
| wutils.wait_for_connect(self.dut, |
| self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.validate_connection(self.dut) |
| |
| @test_tracker_info(uuid="c591e687-340c-42e6-8d85-58a1f930b6b1") |
| def test_connect_to_wpa_psk_ccmp_2g_after_suspend_resume(self): |
| """Test Wi-Fi reconnection after Suspend. |
| |
| Steps: |
| DUT connect to 2GHz Wi-Fi network. |
| DUT suspend and resume. |
| DUT verify internet connection with HTTP ping. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| self.dut.log.info("Suspend the DUT and wait for 10 seconds") |
| # Suspend and Resume the DUT. |
| self.dut.go_to_sleep() |
| time.sleep(10) |
| self.dut.log.info("Resume the DUT") |
| self.dut.wakeup_screen() |
| wutils.validate_connection(self.dut) |
| |
| @test_tracker_info(uuid="d3e34869-f2ae-4614-983d-19be238d8499") |
| def test_connect_to_wpa_psk_ccmp_2g_after_reboot(self): |
| """Test Wi-Fi reconnection after reboot. |
| |
| Steps: |
| DUT connect to 2GHz Wi-Fi network. |
| DUT reboot. |
| DUT verify internet connection with HTTP ping. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| # Reboot the DUT. |
| self.dut.log.info("Reboot the DUT") |
| self.dut.reboot() |
| self.dut.wait_for_boot_completion() |
| wutils.wait_for_connect(self.dut, |
| self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.validate_connection(self.dut) |
| |
| @test_tracker_info(uuid="ebd8dd7f-dc36-4e99-b18c-5f725a2f88b2") |
| def test_connect_to_wpa_psk_ccmp_2g_after_incorrect_password(self): |
| """Test Wi-Fi reconnection after incorrect password. |
| |
| Steps: |
| DUT connect to 2GHz Wi-Fi network. |
| DUT try to connect to the Wi-Fi network with incorrect password. |
| Connection fail as expected. |
| DUT connect to the Wi-Fi network with correct password. |
| DUT verify internet connection with HTTP ping. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| self.wpa_psk_2g_fail = self.wpa_psk_2g.copy() |
| self.wpa_psk_2g_fail["password"] = "incorrect_password" |
| # Try to connect a Wi-Fi network with incorrect passwlrd. |
| try: |
| self.dut.log.info("Connect to Wi-Fi with wrong password") |
| wutils.wifi_connect(self.dut, self.wpa_psk_2g_fail, num_of_tries=1) |
| except: |
| self.dut.log.info("Connect to Wi-Fi with correct password") |
| wutils.wifi_connect(self.dut, self.wpa_psk_2g) |
| else: |
| raise signals.TestFailure("DUT connect to Wi-Fi with wrong password") |
| self.verify_wpa_network_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.validate_connection(self.dut) |
| |
| @test_tracker_info(uuid="d20fc634-8dcc-4336-9640-2a6907ca1894") |
| def test_connect_to_wpa_psk_ccmp_2g_after_out_of_range(self): |
| """Test Wi-Fi reconnection after out of range. |
| |
| Steps: |
| DUT connect to 2GHz Wi-Fi network. |
| DUT out of Wi-Fi range. |
| Make Wi-Fi network is not visible by DUT. |
| DUT back in Wi-Fi range. |
| Wi-Fi network is visible by DUT. |
| DUT connect to the Wi-Fi network. |
| DUT verify internet connection with HTTP ping. |
| """ |
| self.openwrt.log.info("Enable WPA-PSK CCMP on OpenWrt AP") |
| self.openwrt.set_wpa_encryption(OpenWrtWifiSecurity.WPA_PSK_CCMP) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
| # Make the DUT out of range. |
| wutils.set_attns(self.attenuators, |
| "atten1_off_atten2_off", |
| self.roaming_attn) |
| wutils.start_wifi_connection_scan_and_ensure_network_not_found( |
| self.dut, |
| self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| # Make the DUT back in range. |
| wutils.set_attns(self.attenuators, |
| "atten1_on_atten2_on", |
| self.roaming_attn) |
| wutils.start_wifi_connection_scan_and_ensure_network_found( |
| self.dut, self.wpa_psk_2g[WifiEnums.SSID_KEY]) |
| wutils.connect_to_wifi_network(self.dut, self.wpa_psk_2g) |
