| /* |
| * Copyright (C) 2013, 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. |
| */ |
| |
| package com.android.connectivitymanagertest.functional; |
| |
| import android.net.wifi.WifiConfiguration; |
| import android.net.wifi.WifiConfiguration.AuthAlgorithm; |
| import android.net.wifi.WifiConfiguration.GroupCipher; |
| import android.net.wifi.WifiConfiguration.PairwiseCipher; |
| import android.net.wifi.WifiConfiguration.Protocol; |
| import android.net.wifi.WifiInfo; |
| import android.net.wifi.WifiManager; |
| import android.os.Bundle; |
| import android.test.suitebuilder.annotation.LargeTest; |
| |
| import com.android.connectivitymanagertest.ConnectivityManagerTestBase; |
| import com.android.connectivitymanagertest.WifiAssociationTestRunner; |
| import com.android.connectivitymanagertest.WifiConfigurationHelper; |
| |
| /** |
| * Test Wi-Fi connection with different configuration |
| * To run this tests: |
| * * adb shell am instrument -e ssid <ssid> -e password <password> \ |
| * -e security-type [OPEN|WEP64|WEP128|WPA_TKIP|WPA2_AES] -e frequency-band [2.4|5.0|auto] |
| * -w com.android.connectivitymanagertest/.WifiAssociationTestRunner" |
| */ |
| public class WifiAssociationTest extends ConnectivityManagerTestBase { |
| private enum SecurityType { |
| OPEN, WEP64, WEP128, WPA_TKIP, WPA2_AES |
| } |
| |
| public WifiAssociationTest() { |
| super(WifiAssociationTest.class.getSimpleName()); |
| } |
| |
| /** |
| * Test that the wifi can associate with a given access point. |
| */ |
| @LargeTest |
| public void testWifiAssociation() { |
| WifiAssociationTestRunner runner = (WifiAssociationTestRunner) getInstrumentation(); |
| Bundle arguments = runner.getArguments(); |
| |
| String ssid = arguments.getString("ssid"); |
| assertNotNull("ssid is empty", ssid); |
| |
| String securityTypeStr = arguments.getString("security-type"); |
| assertNotNull("security-type is empty", securityTypeStr); |
| SecurityType securityType = SecurityType.valueOf(securityTypeStr); |
| |
| String password = arguments.getString("password"); |
| |
| String freqStr = arguments.getString("frequency-band"); |
| if (freqStr != null) { |
| setFrequencyBand(freqStr); |
| } |
| |
| assertTrue("enable Wifi failed", enableWifi()); |
| WifiInfo wi = mWifiManager.getConnectionInfo(); |
| logv("%s", wi); |
| assertNotNull("no active wifi info", wi); |
| |
| WifiConfiguration config = getConfig(ssid, securityType, password); |
| |
| logv("Network config: %s", config.toString()); |
| connectToWifi(config); |
| } |
| |
| /** |
| * Set the frequency band and verify that it has been set. |
| */ |
| private void setFrequencyBand(String frequencyBandStr) { |
| int frequencyBand = -1; |
| if ("2.4".equals(frequencyBandStr)) { |
| frequencyBand = WifiManager.WIFI_FREQUENCY_BAND_2GHZ; |
| } else if ("5.0".equals(frequencyBandStr)) { |
| frequencyBand = WifiManager.WIFI_FREQUENCY_BAND_5GHZ; |
| } else if ("auto".equals(frequencyBandStr)) { |
| frequencyBand = WifiManager.WIFI_FREQUENCY_BAND_AUTO; |
| } else { |
| fail("Invalid frequency-band"); |
| } |
| if (mWifiManager.getFrequencyBand() != frequencyBand) { |
| logv("Set frequency band to %s", frequencyBandStr); |
| mWifiManager.setFrequencyBand(frequencyBand, true); |
| } |
| assertEquals("Specified frequency band does not match operational band", |
| frequencyBand, mWifiManager.getFrequencyBand()); |
| } |
| |
| /** |
| * Get the {@link WifiConfiguration} based on ssid, security, and password. |
| */ |
| private WifiConfiguration getConfig(String ssid, SecurityType securityType, String password) { |
| logv("Security type is %s", securityType.toString()); |
| |
| WifiConfiguration config = null; |
| switch (securityType) { |
| case OPEN: |
| config = WifiConfigurationHelper.createOpenConfig(ssid); |
| break; |
| case WEP64: |
| assertNotNull("password is empty", password); |
| // always use hex pair for WEP-40 |
| assertTrue(WifiConfigurationHelper.isHex(password, 10)); |
| config = WifiConfigurationHelper.createWepConfig(ssid, password); |
| config.allowedGroupCiphers.set(GroupCipher.WEP40); |
| break; |
| case WEP128: |
| assertNotNull("password is empty", password); |
| // always use hex pair for WEP-104 |
| assertTrue(WifiConfigurationHelper.isHex(password, 26)); |
| config = WifiConfigurationHelper.createWepConfig(ssid, password); |
| config.allowedGroupCiphers.set(GroupCipher.WEP104); |
| break; |
| case WPA_TKIP: |
| assertNotNull("password is empty", password); |
| config = WifiConfigurationHelper.createPskConfig(ssid, password); |
| config.allowedAuthAlgorithms.set(AuthAlgorithm.OPEN); |
| config.allowedProtocols.set(Protocol.WPA); |
| config.allowedPairwiseCiphers.set(PairwiseCipher.TKIP); |
| config.allowedGroupCiphers.set(GroupCipher.TKIP); |
| break; |
| case WPA2_AES: |
| assertNotNull("password is empty", password); |
| config = WifiConfigurationHelper.createPskConfig(ssid, password); |
| config.allowedAuthAlgorithms.set(AuthAlgorithm.OPEN); |
| config.allowedProtocols.set(Protocol.RSN); |
| config.allowedPairwiseCiphers.set(PairwiseCipher.CCMP); |
| config.allowedGroupCiphers.set(GroupCipher.CCMP); |
| break; |
| default: |
| fail("Not a valid security type: " + securityType); |
| break; |
| } |
| return config; |
| } |
| } |