#!/usr/bin/env python3.4
#
# Copyright 2018 - 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 itertools
import pprint
import queue
import time
import acts.base_test
import acts.signals as signals
import acts.test_utils.wifi.wifi_test_utils as wutils
import acts.utils
from acts import asserts
from acts.test_decorators import test_tracker_info
from acts.test_utils.wifi.WifiBaseTest import WifiBaseTest
WifiEnums = wutils.WifiEnums
# Default timeout used for reboot, toggle WiFi and Airplane mode,
# for the system to settle down after the operation.
DEFAULT_TIMEOUT = 10
WIFICOND_KILL_SHELL_COMMAND = "killall wificond"
WIFI_VENDOR_HAL_KILL_SHELL_COMMAND = "killall android.hardware.wifi@1.0-service"
SUPPLICANT_KILL_SHELL_COMMAND = "killall wpa_supplicant"
class WifiCrashTest(WifiBaseTest):
"""Crash Tests for wifi stack.
Test Bed Requirement:
* One Android device
* One Wi-Fi network visible to the device.
"""
def __init__(self, controllers):
WifiBaseTest.__init__(self, controllers)
def setup_class(self):
self.dut = self.android_devices[0]
wutils.wifi_test_device_init(self.dut)
req_params = []
opt_param = ["reference_networks"]
self.unpack_userparams(
req_param_names=req_params, opt_param_names=opt_param)
if "AccessPoint" in self.user_params:
self.legacy_configure_ap_and_start()
asserts.assert_true(
len(self.reference_networks) > 0,
"Need at least one reference network with psk.")
self.network = self.reference_networks[0]["2g"]
def setup_test(self):
self.dut.droid.wakeLockAcquireBright()
self.dut.droid.wakeUpNow()
wutils.wifi_toggle_state(self.dut, True)
def teardown_test(self):
self.dut.droid.wakeLockRelease()
self.dut.droid.goToSleepNow()
wutils.reset_wifi(self.dut)
def on_fail(self, test_name, begin_time):
self.dut.take_bug_report(test_name, begin_time)
self.dut.cat_adb_log(test_name, begin_time)
def teardown_class(self):
if "AccessPoint" in self.user_params:
del self.user_params["reference_networks"]
"""Helper Functions"""
"""Tests"""
@test_tracker_info(uuid="b87fd23f-9bfc-406b-a5b2-17ce6be6c780")
def test_wifi_framework_crash_reconnect(self):
"""Connect to a network, crash framework, then ensure
we connect back to the previously connected network.
Steps:
1. Connect to a network.
2. Restart framework.
3. Reconnect to the previous network.
"""
wutils.wifi_connect(self.dut, self.network, num_of_tries=3)
# Restart framework
self.log.info("Crashing framework")
self.dut.restart_runtime()
# We won't get the disconnect broadcast because framework crashed.
# wutils.wait_for_disconnect(self.dut)
time.sleep(DEFAULT_TIMEOUT)
wifi_info = self.dut.droid.wifiGetConnectionInfo()
if wifi_info[WifiEnums.SSID_KEY] != self.network[WifiEnums.SSID_KEY]:
raise signals.TestFailure("Device did not connect to the"
" network after crashing framework.")
@test_tracker_info(uuid="33f9e4f6-29b8-4116-8f9b-5b13d93b4bcb")
def test_wifi_cond_crash_reconnect(self):
"""Connect to a network, crash wificond, then ensure
we connect back to the previously connected network.
Steps:
1. Connect to a network.
2. Crash wificond.
3. Ensure we get a disconnect.
4. Ensure we reconnect to the previous network.
"""
wutils.wifi_connect(self.dut, self.network, num_of_tries=3)
# Restart wificond
self.log.info("Crashing wificond")
self.dut.adb.shell(WIFICOND_KILL_SHELL_COMMAND)
wutils.wait_for_disconnect(self.dut)
time.sleep(DEFAULT_TIMEOUT)
wifi_info = self.dut.droid.wifiGetConnectionInfo()
if wifi_info[WifiEnums.SSID_KEY] != self.network[WifiEnums.SSID_KEY]:
raise signals.TestFailure("Device did not connect to the"
" network after crashing wificond.")
@test_tracker_info(uuid="463e3d7b-b0b7-4843-b83b-5613a71ae2ac")
def test_wifi_vendorhal_crash_reconnect(self):
"""Connect to a network, crash wifi HAL, then ensure
we connect back to the previously connected network.
Steps:
1. Connect to a network.
2. Crash wifi HAL.
3. Ensure we get a disconnect.
4. Ensure we reconnect to the previous network.
"""
wutils.wifi_connect(self.dut, self.network, num_of_tries=3)
# Restart wificond
self.log.info("Crashing wifi HAL")
self.dut.adb.shell(WIFI_VENDOR_HAL_KILL_SHELL_COMMAND)
wutils.wait_for_disconnect(self.dut)
time.sleep(DEFAULT_TIMEOUT)
wifi_info = self.dut.droid.wifiGetConnectionInfo()
if wifi_info[WifiEnums.SSID_KEY] != self.network[WifiEnums.SSID_KEY]:
raise signals.TestFailure("Device did not connect to the"
" network after crashing wifi HAL.")
@test_tracker_info(uuid="7c5cd1fc-8f8d-494c-beaf-4eb61b48917b")
def test_wpa_supplicant_crash_reconnect(self):
"""Connect to a network, crash wpa_supplicant, then ensure
we connect back to the previously connected network.
Steps:
1. Connect to a network.
2. Crash wpa_supplicant.
3. Ensure we get a disconnect.
4. Ensure we reconnect to the previous network.
"""
wutils.wifi_connect(self.dut, self.network, num_of_tries=3)
# Restart wificond
self.log.info("Crashing wpa_supplicant")
self.dut.adb.shell(SUPPLICANT_KILL_SHELL_COMMAND)
wutils.wait_for_disconnect(self.dut)
time.sleep(DEFAULT_TIMEOUT)
wifi_info = self.dut.droid.wifiGetConnectionInfo()
if wifi_info[WifiEnums.SSID_KEY] != self.network[WifiEnums.SSID_KEY]:
raise signals.TestFailure("Device did not connect to the"
" network after crashing wpa_supplicant.")