#!/usr/bin/env python3 # # Copyright (C) 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. """ Script for testing WiFi connection and disconnection in a loop """ import time from acts import signals from acts.controllers.access_point import setup_ap from acts.controllers.ap_lib import hostapd_constants from acts.controllers.ap_lib import hostapd_security from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest from acts_contrib.test_utils.abstract_devices.wlan_device import create_wlan_device from acts.utils import rand_ascii_str class ConnectionStressTest(WifiBaseTest): # Default number of test iterations here. # Override using parameter in config file. # Eg: "connection_stress_test_iterations": "50" num_of_iterations = 10 channel_2G = hostapd_constants.AP_DEFAULT_CHANNEL_2G channel_5G = hostapd_constants.AP_DEFAULT_CHANNEL_5G def setup_class(self): super().setup_class() self.ssid = rand_ascii_str(10) self.fd = self.fuchsia_devices[0] self.dut = create_wlan_device(self.fd) self.access_point = self.access_points[0] self.num_of_iterations = int( self.user_params.get("connection_stress_test_iterations", self.num_of_iterations)) self.log.info('iterations: %d' % self.num_of_iterations) def teardown_test(self): self.dut.reset_wifi() self.download_ap_logs() self.access_point.stop_all_aps() def on_fail(self, test_name, begin_time): super().on_fail(test_name, begin_time) self.access_point.stop_all_aps() def start_ap(self, profile, channel, security=None): """Starts an Access Point Args: profile: Profile name such as 'whirlwind' channel: Channel to operate on """ self.log.info('Profile: %s, Channel: %d' % (profile, channel)) setup_ap(access_point=self.access_point, profile_name=profile, channel=channel, ssid=self.ssid, security=security) def connect_disconnect(self, ap_config, ssid=None, password=None, negative_test=False): """Helper to start an AP, connect DUT to it and disconnect Args: ap_config: Dictionary contaning profile name and channel ssid: ssid to connect to password: password for the ssid to connect to """ security_mode = ap_config.get('security_mode', None) target_security = hostapd_constants.SECURITY_STRING_TO_DEFAULT_TARGET_SECURITY.get( security_mode, None) if security_mode: security_profile = hostapd_security.Security( security_mode=ap_config['security_mode'], password=ap_config['password']) else: security_profile = None # Start AP self.start_ap(ap_config['profile'], ap_config['channel'], security=security_profile) failed = False # Connect and Disconnect several times for x in range(0, self.num_of_iterations): if not ssid: ssid = self.ssid if negative_test: if not self.dut.associate(ssid, target_pwd=password, target_security=target_security): self.log.info( 'Attempt %d. Did not associate as expected.' % x) else: self.log.error('Attempt %d. Negative test successfully ' 'associated. Fail.' % x) failed = True else: # Connect if self.dut.associate(ssid, target_pwd=password): self.log.info('Attempt %d. Successfully associated' % x) else: self.log.error('Attempt %d. Failed to associate.' % x) failed = True # Disconnect self.dut.disconnect() # Wait a second before trying again time.sleep(1) # Stop AP self.access_point.stop_all_aps() if failed: raise signals.TestFailure( 'One or more association attempt failed.') def test_whirlwind_2g(self): self.connect_disconnect({ 'profile': 'whirlwind', 'channel': self.channel_2G, 'security_mode': None }) def test_whirlwind_5g(self): self.connect_disconnect({ 'profile': 'whirlwind', 'channel': self.channel_5G, 'security_mode': None }) def test_whirlwind_11ab_2g(self): self.connect_disconnect({ 'profile': 'whirlwind_11ab_legacy', 'channel': self.channel_2G, 'security_mode': None }) def test_whirlwind_11ab_5g(self): self.connect_disconnect({ 'profile': 'whirlwind_11ab_legacy', 'channel': self.channel_5G, 'security_mode': None }) def test_whirlwind_11ag_2g(self): self.connect_disconnect({ 'profile': 'whirlwind_11ag_legacy', 'channel': self.channel_2G, 'security_mode': None }) def test_whirlwind_11ag_5g(self): self.connect_disconnect({ 'profile': 'whirlwind_11ag_legacy', 'channel': self.channel_5G, 'security_mode': None }) def test_wrong_ssid_whirlwind_2g(self): self.connect_disconnect( { 'profile': 'whirlwind', 'channel': self.channel_2G, 'security_mode': None }, ssid=rand_ascii_str(20), negative_test=True) def test_wrong_ssid_whirlwind_5g(self): self.connect_disconnect( { 'profile': 'whirlwind', 'channel': self.channel_5G, 'security_mode': None }, ssid=rand_ascii_str(20), negative_test=True) def test_wrong_password_whirlwind_2g(self): self.connect_disconnect( { 'profile': 'whirlwind', 'channel': self.channel_2G, 'security_mode': hostapd_constants.WPA2_STRING, 'password': rand_ascii_str(10) }, password=rand_ascii_str(20), negative_test=True) def test_wrong_password_whirlwind_5g(self): self.connect_disconnect( { 'profile': 'whirlwind', 'channel': self.channel_5G, 'security_mode': hostapd_constants.WPA2_STRING, 'password': rand_ascii_str(10) }, password=rand_ascii_str(20), negative_test=True)