| /* |
| * Copyright (C) 2016 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.server.wifi; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.fail; |
| |
| import android.test.suitebuilder.annotation.SmallTest; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.mockito.Mock; |
| import org.mockito.MockitoAnnotations; |
| |
| import java.io.BufferedReader; |
| import java.io.IOException; |
| import java.io.StringReader; |
| import java.util.ArrayList; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| /** |
| * Unit tests for {@link com.android.server.wifi.WifiConfigStore}. |
| */ |
| @SmallTest |
| public class WifiConfigStoreTest { |
| private static final String KEY_SSID = "ssid"; |
| private static final String KEY_PSK = "psk"; |
| private static final String KEY_KEY_MGMT = "key_mgmt"; |
| private static final String KEY_PRIORITY = "priority"; |
| private static final String KEY_DISABLED = "disabled"; |
| private static final String KEY_ID_STR = "id_str"; |
| // This is not actually present as a key in the wpa_supplicant.conf file, but |
| // is used in tests to conveniently access the configKey for a test network. |
| private static final String CONFIG_KEY = "configKey"; |
| |
| private static final HashMap<String, String> NETWORK_0_VARS = new HashMap<>(); |
| static { |
| NETWORK_0_VARS.put(KEY_SSID, "\"TestNetwork0\""); |
| NETWORK_0_VARS.put(KEY_KEY_MGMT, "NONE"); |
| NETWORK_0_VARS.put(KEY_PRIORITY, "2"); |
| NETWORK_0_VARS.put(KEY_ID_STR, "" |
| + "\"%7B%22creatorUid%22%3A%221000%22%2C%22configKey%22%3A%22%5C%22" |
| + "TestNetwork0%5C%22NONE%22%7D\""); |
| NETWORK_0_VARS.put(CONFIG_KEY, "\"TestNetwork0\"NONE"); |
| } |
| |
| private static final HashMap<String, String> NETWORK_1_VARS = new HashMap<>(); |
| static { |
| NETWORK_1_VARS.put(KEY_SSID, "\"Test Network 1\""); |
| NETWORK_1_VARS.put(KEY_KEY_MGMT, "NONE"); |
| NETWORK_1_VARS.put(KEY_PRIORITY, "3"); |
| NETWORK_1_VARS.put(KEY_DISABLED, "1"); |
| NETWORK_1_VARS.put(KEY_ID_STR, "" |
| + "\"%7B%22creatorUid%22%3A%221000%22%2C%22configKey%22%3A%22%5C%22" |
| + "Test+Network+1%5C%22NONE%22%7D\""); |
| NETWORK_1_VARS.put(CONFIG_KEY, "\"Test Network 1\"NONE"); |
| } |
| |
| private static final HashMap<String, String> NETWORK_2_VARS = new HashMap<>(); |
| static { |
| NETWORK_2_VARS.put(KEY_SSID, "\"testNetwork2\""); |
| NETWORK_2_VARS.put(KEY_KEY_MGMT, "NONE"); |
| NETWORK_2_VARS.put(KEY_PRIORITY, "4"); |
| NETWORK_2_VARS.put(KEY_DISABLED, "1"); |
| NETWORK_2_VARS.put(KEY_ID_STR, "" |
| + "\"%7B%22creatorUid%22%3A%221000%22%2C%22configKey%22%3A%22%5C%22" |
| + "testNetwork2%5C%22NONE%22%7D\""); |
| NETWORK_2_VARS.put(CONFIG_KEY, "\"testNetwork2\"NONE"); |
| } |
| |
| private static final HashMap<String, String> NETWORK_3_VARS = new HashMap<>(); |
| static { |
| NETWORK_3_VARS.put(KEY_SSID, "\"testwpa2psk\""); |
| NETWORK_3_VARS.put(KEY_PSK, "blahblah"); |
| NETWORK_3_VARS.put(KEY_KEY_MGMT, "WPA-PSK"); |
| NETWORK_3_VARS.put(KEY_PRIORITY, "6"); |
| NETWORK_3_VARS.put(KEY_DISABLED, "1"); |
| NETWORK_3_VARS.put(KEY_ID_STR, "" |
| + "\"%7B%22creatorUid%22%3A%221000%22%2C%22configKey%22%3A%22%5C%22" |
| + "testwpa2psk%5C%22WPA_PSK%22%7D\""); |
| NETWORK_3_VARS.put(CONFIG_KEY, "\"testwpa2psk\"WPA_PSK"); |
| } |
| |
| private static final ArrayList<HashMap<String, String>> NETWORK_VARS = new ArrayList<HashMap<String, String>>(); |
| static { |
| NETWORK_VARS.add(NETWORK_0_VARS); |
| NETWORK_VARS.add(NETWORK_1_VARS); |
| NETWORK_VARS.add(NETWORK_2_VARS); |
| NETWORK_VARS.add(NETWORK_3_VARS); |
| } |
| |
| // Taken from wpa_supplicant.conf of an actual test device. Some fields modified for privacy |
| // reasons. |
| private static final String TEST_WPA_SUPPLICANT_CONF = "" |
| + "ctrl_interface=/data/misc/wifi/sockets\n" |
| + "disable_scan_offload=1\n" |
| + "driver_param=use_p2p_group_interface=1p2p_device=1\n" |
| + "update_config=1\n" |
| + "device_name=testdevice\n" |
| + "manufacturer=TestManufacturer\n" |
| + "model_name=Testxus\n" |
| + "model_number=Testxus\n" |
| + "serial_number=1ABCD12345678912\n" |
| + "device_type=12-3456A456-7\n" |
| + "config_methods=physical_display virtual_push_button\n" |
| + "p2p_no_go_freq=5170-5740\n" |
| + "pmf=1\n" |
| + "external_sim=1\n" |
| + "wowlan_triggers=any\n" |
| + "p2p_search_delay=0\n" |
| + "network={\n" |
| + " " + KEY_SSID + "=" + NETWORK_0_VARS.get(KEY_SSID) + "\n" |
| + " " + KEY_KEY_MGMT + "=" + NETWORK_0_VARS.get(KEY_KEY_MGMT) + "\n" |
| + " " + KEY_PRIORITY + "=" + NETWORK_0_VARS.get(KEY_PRIORITY) + "\n" |
| + " " + KEY_ID_STR + "=" + NETWORK_0_VARS.get(KEY_ID_STR) + "\n" |
| + "}\n" |
| + "\n" |
| + "network={\n" |
| + " " + KEY_SSID + "=" + NETWORK_1_VARS.get(KEY_SSID) + "\n" |
| + " " + KEY_KEY_MGMT + "=" + NETWORK_1_VARS.get(KEY_KEY_MGMT) + "\n" |
| + " " + KEY_PRIORITY + "=" + NETWORK_1_VARS.get(KEY_PRIORITY) + "\n" |
| + " " + KEY_DISABLED + "=" + NETWORK_1_VARS.get(KEY_DISABLED) + "\n" |
| + " " + KEY_ID_STR + "=" + NETWORK_1_VARS.get(KEY_ID_STR) + "\n" |
| + "}\n" |
| + "\n" |
| + "network={\n" |
| + " " + KEY_SSID + "=" + NETWORK_2_VARS.get(KEY_SSID) + "\n" |
| + " " + KEY_KEY_MGMT + "=" + NETWORK_2_VARS.get(KEY_KEY_MGMT) + "\n" |
| + " " + KEY_PRIORITY + "=" + NETWORK_2_VARS.get(KEY_PRIORITY) + "\n" |
| + " " + KEY_DISABLED + "=" + NETWORK_2_VARS.get(KEY_DISABLED) + "\n" |
| + " " + KEY_ID_STR + "=" + NETWORK_2_VARS.get(KEY_ID_STR) + "\n" |
| + "}\n" |
| + "\n" |
| + "network={\n" |
| + " " + KEY_SSID + "=" + NETWORK_3_VARS.get(KEY_SSID) + "\n" |
| + " " + KEY_PSK + "=" + NETWORK_3_VARS.get(KEY_PSK) + "\n" |
| + " " + KEY_KEY_MGMT + "=" + NETWORK_3_VARS.get(KEY_KEY_MGMT) + "\n" |
| + " " + KEY_PRIORITY + "=" + NETWORK_3_VARS.get(KEY_PRIORITY) + "\n" |
| + " " + KEY_DISABLED + "=" + NETWORK_3_VARS.get(KEY_DISABLED) + "\n" |
| + " " + KEY_ID_STR + "=" + NETWORK_3_VARS.get(KEY_ID_STR) + "\n" |
| + "}\n"; |
| |
| @Mock private WifiNative mWifiNative; |
| private MockKeyStore mMockKeyStore; |
| private WifiConfigStore mWifiConfigStore; |
| |
| @Before |
| public void setUp() throws Exception { |
| MockitoAnnotations.initMocks(this); |
| |
| mMockKeyStore = new MockKeyStore(); |
| mWifiConfigStore = new WifiConfigStore(mWifiNative, mMockKeyStore.createMock(), null); |
| } |
| |
| /** |
| * Verifies that readNetworkVariableFromSupplicantFile() properly reads network variables from a |
| * wpa_supplicant.conf file. |
| */ |
| @Test |
| public void readNetworkVariableFromSupplicantFile() throws Exception { |
| Map<String, String> ssidResults = readNetworkVariableFromSupplicantFile(KEY_SSID); |
| assertEquals(NETWORK_VARS.size(), ssidResults.size()); |
| for (HashMap<String, String> single_network_vars : NETWORK_VARS) { |
| assertEquals(ssidResults.get(single_network_vars.get(CONFIG_KEY)), |
| single_network_vars.get(KEY_SSID)); |
| } |
| |
| Map<String, String> pskResults = readNetworkVariableFromSupplicantFile(KEY_PSK); |
| // Only network 3 is secured with a password. |
| assertEquals(1, pskResults.size()); |
| assertEquals(pskResults.get(NETWORK_3_VARS.get(CONFIG_KEY)), |
| NETWORK_3_VARS.get(KEY_PSK)); |
| |
| Map<String, String> keyMgmtResults = readNetworkVariableFromSupplicantFile(KEY_KEY_MGMT); |
| assertEquals(NETWORK_VARS.size(), keyMgmtResults.size()); |
| for (HashMap<String, String> single_network_vars : NETWORK_VARS) { |
| assertEquals(keyMgmtResults.get(single_network_vars.get(CONFIG_KEY)), |
| single_network_vars.get(KEY_KEY_MGMT)); |
| } |
| |
| Map<String, String> priorityResults = readNetworkVariableFromSupplicantFile(KEY_PRIORITY); |
| assertEquals(NETWORK_VARS.size(), priorityResults.size()); |
| for (HashMap<String, String> single_network_vars : NETWORK_VARS) { |
| assertEquals(priorityResults.get(single_network_vars.get(CONFIG_KEY)), |
| single_network_vars.get(KEY_PRIORITY)); |
| } |
| |
| Map<String, String> disabledResults = readNetworkVariableFromSupplicantFile(KEY_DISABLED); |
| // All but network 0 are disabled. |
| assertEquals(NETWORK_VARS.size() - 1, disabledResults.size()); |
| for (int i = 1; i < NETWORK_VARS.size(); ++i) { |
| assertEquals(disabledResults.get(NETWORK_VARS.get(i).get(CONFIG_KEY)), |
| NETWORK_VARS.get(i).get(KEY_DISABLED)); |
| } |
| |
| Map<String, String> idStrResults = readNetworkVariableFromSupplicantFile(KEY_ID_STR); |
| assertEquals(NETWORK_VARS.size(), idStrResults.size()); |
| for (HashMap<String, String> single_network_vars : NETWORK_VARS) { |
| assertEquals(idStrResults.get(single_network_vars.get(CONFIG_KEY)), |
| single_network_vars.get(KEY_ID_STR)); |
| } |
| } |
| |
| /** |
| * Inject |TEST_WPA_SUPPLICANT_CONF| via the helper method readNetworkVariablesFromReader(). |
| */ |
| public Map<String, String> readNetworkVariableFromSupplicantFile(String key) throws Exception { |
| Map<String, String> result = new HashMap<>(); |
| BufferedReader reader = null; |
| try { |
| reader = new BufferedReader(new StringReader(TEST_WPA_SUPPLICANT_CONF)); |
| result = mWifiConfigStore.readNetworkVariablesFromReader(reader, key); |
| } catch (IOException e) { |
| fail("Error reading test supplicant conf string"); |
| } finally { |
| try { |
| if (reader != null) { |
| reader.close(); |
| } |
| } catch (IOException e) { |
| // Just ignore if we can't close the reader. |
| } |
| } |
| return result; |
| } |
| } |
