Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 5fe5f2de authored by Janis Danisevskis's avatar Janis Danisevskis
Browse files

Test for contract between AndroidKeyStoreKey hash and equals. 

Test: atest KeystoreTests
Bug: 196118021

Merged-In: Ic6e60752faa986debe3d325f54242cffaa03b336
Change-Id: Ic6e60752faa986debe3d325f54242cffaa03b336
parent e36fe6bf

keystore/java/android/security/keystore2/AndroidKeyStoreKey.java

+7 −1
Original line number Diff line number Diff line
@@ -22,6 +22,8 @@ import android.system.keystore2.Authorization; 
import android.system.keystore2.Domain;

import android.system.keystore2.KeyDescriptor;



import com.android.internal.annotations.VisibleForTesting;



import java.security.Key;



/**
@@ -46,7 +48,11 @@ public class AndroidKeyStoreKey implements Key { 
    // We do not include this member in comparisons.

    private final KeyStoreSecurityLevel mSecurityLevel;



    AndroidKeyStoreKey(@NonNull KeyDescriptor descriptor,

    /**

     * @hide

     */

    @VisibleForTesting

    public AndroidKeyStoreKey(@NonNull KeyDescriptor descriptor,

            long keyId,

            @NonNull Authorization[] authorizations,

            @NonNull String algorithm,

keystore/tests/src/android/security/keystore2/AndroidKeyStoreSpiTest.java

+114 −0
Original line number Diff line number Diff line
@@ -24,7 +24,13 @@ import static org.mockito.Mockito.when; 


import android.security.KeyStore2;

import android.security.KeyStoreException;

import android.security.KeyStoreSecurityLevel;

import android.system.keystore2.Authorization;

import android.system.keystore2.Domain;

import android.system.keystore2.KeyDescriptor;

import android.system.keystore2.KeyMetadata;



import org.junit.Assert;

import org.junit.Before;

import org.junit.Test;

import org.mockito.Mock;
@@ -52,4 +58,112 @@ public class AndroidKeyStoreSpiTest { 
        verify(mKeystore2).list(anyInt(), anyLong());

    }



    @Mock

    private KeyStoreSecurityLevel mKeystoreSecurityLevel;



    private static KeyDescriptor descriptor() {

        final KeyDescriptor keyDescriptor = new KeyDescriptor();

        keyDescriptor.alias = "key";

        keyDescriptor.blob = null;

        keyDescriptor.domain = Domain.APP;

        keyDescriptor.nspace = -1;

        return keyDescriptor;

    }



    private static KeyMetadata metadata(byte[] cert, byte[] certChain) {

        KeyMetadata metadata = new KeyMetadata();

        metadata.authorizations = new Authorization[0];

        metadata.certificate = cert;

        metadata.certificateChain = certChain;

        metadata.key = descriptor();

        metadata.modificationTimeMs = 0;

        metadata.keySecurityLevel = 1;

        return metadata;

    }



    private static byte[] bytes(String string) {

        return string.getBytes();

    }



    class MyPublicKey extends AndroidKeyStorePublicKey {

        MyPublicKey(String cert, String chain, KeyStoreSecurityLevel securityLevel) {

            super(descriptor(), metadata(cert.getBytes(), chain.getBytes()), "N/A".getBytes(),

                    "RSA", securityLevel);

        }



        @Override

        AndroidKeyStorePrivateKey getPrivateKey() {

            return null;

        }

    }



    private AndroidKeyStorePublicKey makePrivateKeyObject(String cert, String chain) {

        return new MyPublicKey(cert, chain, mKeystoreSecurityLevel);

    }



    @Test

    public void testKeystoreKeysAdhereToContractBetweenEqualsAndHashCode() throws Exception {

        AndroidKeyStoreKey key1 = new AndroidKeyStoreKey(descriptor(), 1, new Authorization[0],

                "RSA", mKeystoreSecurityLevel);

        AndroidKeyStoreKey key2 = new AndroidKeyStoreKey(descriptor(), 2, new Authorization[0],

                "RSA", mKeystoreSecurityLevel);

        AndroidKeyStoreKey key1_clone = new AndroidKeyStoreKey(descriptor(), 1,

                new Authorization[0], "RSA", mKeystoreSecurityLevel);



        assertThat("Identity should yield true", key1.equals(key1));

        Assert.assertEquals("Identity should yield same hash codes",

                key1.hashCode(), key1.hashCode());

        assertThat("Identity should yield true", key2.equals(key2));

        Assert.assertEquals("Identity should yield same hash codes",

                key2.hashCode(), key2.hashCode());

        assertThat("Different keys should differ", !key1.equals(key2));

        Assert.assertNotEquals("Different keys should have different hash codes",

                key1.hashCode(), key2.hashCode());



        assertThat("Same keys should yield true", key1.equals(key1_clone));

        assertThat("Same keys should yield true", key1_clone.equals(key1));

        Assert.assertEquals("Same keys should yield same hash codes",

                key1.hashCode(), key1_clone.hashCode());



        assertThat("anything.equal(null) should yield false", !key1.equals(null));

        assertThat("anything.equal(null) should yield false", !key2.equals(null));

        assertThat("anything.equal(null) should yield false", !key1_clone.equals(null));

    }



    @Test

    public void testKeystorePublicKeysAdhereToContractBetweenEqualsAndHashCode() throws Exception {

        AndroidKeyStorePublicKey key1 = makePrivateKeyObject("myCert1", "myChain1");

        AndroidKeyStorePublicKey key2 = makePrivateKeyObject("myCert2", "myChain1");

        AndroidKeyStorePublicKey key3 = makePrivateKeyObject("myCert1", "myChain3");

        AndroidKeyStorePublicKey key1_clone = makePrivateKeyObject("myCert1", "myChain1");



        assertThat("Identity should yield true", key1.equals(key1));

        Assert.assertEquals("Identity should yield same hash codes",

                key1.hashCode(), key1.hashCode());

        assertThat("Identity should yield true", key2.equals(key2));

        Assert.assertEquals("Identity should yield same hash codes",

                key2.hashCode(), key2.hashCode());

        assertThat("Identity should yield true", key3.equals(key3));

        Assert.assertEquals("Identity should yield same hash codes",

                key3.hashCode(), key3.hashCode());

        assertThat("Different keys should differ", !key1.equals(key2));

        Assert.assertNotEquals("Different keys should have different hash codes",

                key1.hashCode(), key2.hashCode());

        assertThat("Different keys should differ", !key1.equals(key3));

        Assert.assertNotEquals("Different keys should have different hash codes",

                key1.hashCode(), key3.hashCode());

        assertThat("Different keys should differ", !key2.equals(key3));

        Assert.assertNotEquals("Different keys should have different hash codes",

                key2.hashCode(), key3.hashCode());



        assertThat("Same keys should yield true", key1.equals(key1_clone));

        assertThat("Same keys should yield true", key1_clone.equals(key1));

        Assert.assertEquals("Same keys should yield same hash codes",

                key1.hashCode(), key1_clone.hashCode());



        assertThat("anything.equal(null) should yield false", !key1.equals(null));

        assertThat("anything.equal(null) should yield false", !key2.equals(null));

        assertThat("anything.equal(null) should yield false", !key3.equals(null));

        assertThat("anything.equal(null) should yield false", !key1_clone.equals(null));

    }

}