Test for contract between AndroidKeyStoreKey hash and equals.

Test: atest KeystoreTests
Bug: 196118021

Merged-In: Ic6e60752faa986debe3d325f54242cffaa03b336
Change-Id: Ic6e60752faa986debe3d325f54242cffaa03b336
This commit is contained in:
Janis Danisevskis
2021-09-01 14:47:20 -07:00
parent 3a37085ede
commit 486adb9aa8
2 changed files with 121 additions and 1 deletions

View File

@@ -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,

View File

@@ -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));
}
}