diff options
| -rw-r--r-- | framework/java/android/net/wifi/rtt/PasnConfig.java | 1 | ||||
| -rw-r--r-- | framework/tests/src/android/net/wifi/rtt/PasnConfigTest.java | 213 |
2 files changed, 214 insertions, 0 deletions
diff --git a/framework/java/android/net/wifi/rtt/PasnConfig.java b/framework/java/android/net/wifi/rtt/PasnConfig.java index 9688f5eee3..d420154cba 100644 --- a/framework/java/android/net/wifi/rtt/PasnConfig.java +++ b/framework/java/android/net/wifi/rtt/PasnConfig.java @@ -263,6 +263,7 @@ public final class PasnConfig implements Parcelable { dest.writeInt(mBaseAkms); dest.writeInt(mCiphers); dest.writeString(mPassword); + dest.writeParcelable(mWifiSsid, flags); dest.writeByteArray(mPasnComebackCookie); } diff --git a/framework/tests/src/android/net/wifi/rtt/PasnConfigTest.java b/framework/tests/src/android/net/wifi/rtt/PasnConfigTest.java new file mode 100644 index 0000000000..d986e251f3 --- /dev/null +++ b/framework/tests/src/android/net/wifi/rtt/PasnConfigTest.java @@ -0,0 +1,213 @@ +/* + * Copyright (C) 2025 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 android.net.wifi.rtt; + +import static org.junit.Assert.assertArrayEquals; +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNotEquals; +import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertNull; +import static org.junit.Assert.assertTrue; + +import android.net.wifi.WifiSsid; +import android.os.Parcel; + +import androidx.test.filters.SmallTest; + +import org.junit.Test; + +import java.util.Arrays; + +/** + * Tests PasnConfig class. + */ +@SmallTest +public class PasnConfigTest { + private static final int TEST_AKM = PasnConfig.AKM_SAE; + private static final int TEST_CIPHER = PasnConfig.CIPHER_CCMP_128; + private static final String TEST_SSID = "\"Test_SSID\""; + private static final String TEST_PASSWORD = "password"; + private static final byte[] TEST_COOKIE = new byte[]{1, 2, 3}; + + /** + * Verifies builder and getter methods work as expected. + */ + @Test + public void testBuilderAndGetters() { + WifiSsid ssid = WifiSsid.fromString(TEST_SSID); + PasnConfig config = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER) + .setPassword(TEST_PASSWORD) + .setWifiSsid(ssid) + .setPasnComebackCookie(TEST_COOKIE) + .build(); + + assertEquals(TEST_AKM, config.getBaseAkms()); + assertEquals(TEST_CIPHER, config.getCiphers()); + assertEquals(TEST_PASSWORD, config.getPassword()); + assertEquals(ssid, config.getWifiSsid()); + assertArrayEquals(TEST_COOKIE, config.getPasnComebackCookie()); + } + + /** + * Verifies parceling round trip returns an identical object. + */ + @Test + public void testParcelableRoundTrip() { + WifiSsid ssid = WifiSsid.fromString(TEST_SSID); + PasnConfig config = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER) + .setPassword(TEST_PASSWORD) + .setWifiSsid(ssid) + .setPasnComebackCookie(TEST_COOKIE) + .build(); + + Parcel parcel = Parcel.obtain(); + config.writeToParcel(parcel, 0); + parcel.setDataPosition(0); + + PasnConfig fromParcel = PasnConfig.CREATOR.createFromParcel(parcel); + assertEquals(config, fromParcel); + } + + /** + * Tests that null SSID and password result in unauthenticated PASN being used. + */ + @Test + public void testNullSsidAndPassword() { + PasnConfig config = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER).build(); + assertNull(config.getPassword()); + assertNull(config.getWifiSsid()); + } + + /** + * Tests equality and non-equality of objects. + */ + @Test + public void testEqualsAndHashCode() { + WifiSsid ssid1 = WifiSsid.fromString(TEST_SSID); + WifiSsid ssid2 = WifiSsid.fromString("\"Another_SSID\""); + PasnConfig config1 = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER) + .setWifiSsid(ssid1) + .setPassword(TEST_PASSWORD) + .setPasnComebackCookie(TEST_COOKIE) + .build(); + PasnConfig config2 = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER) + .setWifiSsid(ssid1) + .setPassword(TEST_PASSWORD) + .setPasnComebackCookie(TEST_COOKIE) + .build(); + PasnConfig config3 = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER) + .setWifiSsid(ssid2) + .setPassword(TEST_PASSWORD) + .setPasnComebackCookie(TEST_COOKIE) + .build(); + + assertEquals(config1, config2); + assertNotEquals(config1, config3); + assertEquals(config1.hashCode(), config2.hashCode()); + } + + + /** + * Tests toString() method. + */ + @Test + public void testToString() { + WifiSsid ssid = WifiSsid.fromString(TEST_SSID); + PasnConfig config = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER) + .setPassword(TEST_PASSWORD) + .setWifiSsid(ssid) + .setPasnComebackCookie(TEST_COOKIE) + .build(); + + String expectedString = "PasnConfig{" + "mBaseAkms=" + TEST_AKM + ", mCiphers=" + + TEST_CIPHER + ", mPassword='" + TEST_PASSWORD + '\'' + ", mWifiSsid=" + ssid + + ", mPasnComebackCookie=" + Arrays.toString(TEST_COOKIE) + '}'; + assertEquals(expectedString, config.toString()); + } + + + /** + * Verifies that builder methods return a non-null builder instance. + */ + @Test + public void testBuilderMethodsReturnNonNull() { + PasnConfig.Builder builder = new PasnConfig.Builder(TEST_AKM, TEST_CIPHER); + assertNotNull(builder.setPassword(TEST_PASSWORD)); + assertNotNull(builder.setWifiSsid(WifiSsid.fromString(TEST_SSID))); + assertNotNull(builder.setPasnComebackCookie(TEST_COOKIE)); + } + + /** + * Tests the validation when setting an empty comeback cookie. + */ + @Test(expected = IllegalArgumentException.class) + public void testSetPasnComebackCookie_emptyCookie() { + new PasnConfig.Builder(TEST_AKM, TEST_CIPHER).setPasnComebackCookie(new byte[0]); + } + + /** + * Tests the validation when setting a long comeback cookie. + */ + @Test(expected = IllegalArgumentException.class) + public void testSetPasnComebackCookie_longCookie() { + new PasnConfig.Builder(TEST_AKM, TEST_CIPHER).setPasnComebackCookie(new byte[256]); + } + + /** + * Tests the validation when setting a null comeback cookie. + */ + @Test(expected = NullPointerException.class) + public void testSetPasnComebackCookie_nullCookie() { + new PasnConfig.Builder(TEST_AKM, TEST_CIPHER).setPasnComebackCookie(null); + } + + /** + * Tests {@link PasnConfig#getBaseAkmsFromCapabilities(String)} method. + */ + @Test + public void testGetBaseAkmsFromCapabilities() { + assertEquals(PasnConfig.AKM_NONE, PasnConfig.getBaseAkmsFromCapabilities(null)); + assertEquals(PasnConfig.AKM_NONE, PasnConfig.getBaseAkmsFromCapabilities("")); + assertEquals(PasnConfig.AKM_SAE, + PasnConfig.getBaseAkmsFromCapabilities("[RSN-SAE+SAE_EXT_KEY-CCMP]")); + assertEquals(PasnConfig.AKM_SAE, + PasnConfig.getBaseAkmsFromCapabilities("[RSN-PSK+SAE-CCMP]")); + assertEquals(PasnConfig.AKM_FT_PSK_SHA256, + PasnConfig.getBaseAkmsFromCapabilities("[RSN-FT/PSK-CCMP]")); + } + + /** + * Tests {@link PasnConfig#getCiphersFromCapabilities(String)} method. + */ + @Test + public void testGetCiphersFromCapabilities() { + assertEquals(PasnConfig.CIPHER_NONE, PasnConfig.getCiphersFromCapabilities(null)); + assertEquals(PasnConfig.CIPHER_NONE, PasnConfig.getCiphersFromCapabilities("")); + assertEquals(PasnConfig.CIPHER_CCMP_128, + PasnConfig.getCiphersFromCapabilities("[RSN-SAE+SAE_EXT_KEY-CCMP]")); + assertEquals(PasnConfig.CIPHER_GCMP_128, + PasnConfig.getCiphersFromCapabilities("[RSN-SAE+SAE_EXT_KEY-GCMP-128]")); + } + + /** + * Tests {@link PasnConfig#isAkmRequiresPassword(int)} method. + */ + @Test + public void testIsAkmRequiresPassword() { + assertTrue(PasnConfig.isAkmRequiresPassword(PasnConfig.AKM_SAE)); + } +} |