From a9862a149a956d6290ba4620e7733aaaef787b4f Mon Sep 17 00:00:00 2001 From: wconner Date: Thu, 27 Jul 2023 08:04:14 -0700 Subject: Ensure that generated HPKE private keys are encoded to the correct length. Also, added utility methods for determining HPKE key lengths from KEM id. PiperOrigin-RevId: 551532702 --- .../src/test/java/com/google/crypto/tink/hybrid/internal/BUILD.bazel | 1 + .../google/crypto/tink/hybrid/internal/HpkePrivateKeyManagerTest.java | 4 ++++ 2 files changed, 5 insertions(+) (limited to 'java_src/src/test/java/com/google') diff --git a/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/BUILD.bazel b/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/BUILD.bazel index 5ca1e21d1..a3397c31a 100644 --- a/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/BUILD.bazel +++ b/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/BUILD.bazel @@ -152,6 +152,7 @@ java_test( "//src/main/java/com/google/crypto/tink/hybrid:hybrid_encrypt_wrapper", "//src/main/java/com/google/crypto/tink/hybrid/internal:hpke_encrypt", "//src/main/java/com/google/crypto/tink/hybrid/internal:hpke_private_key_manager", + "//src/main/java/com/google/crypto/tink/hybrid/internal:hpke_util", "//src/main/java/com/google/crypto/tink/internal:key_type_manager", "//src/main/java/com/google/crypto/tink/subtle:random", "//src/main/java/com/google/crypto/tink/testing:test_util", diff --git a/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/HpkePrivateKeyManagerTest.java b/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/HpkePrivateKeyManagerTest.java index e92bb5d24..34f11ccfc 100644 --- a/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/HpkePrivateKeyManagerTest.java +++ b/java_src/src/test/java/com/google/crypto/tink/hybrid/internal/HpkePrivateKeyManagerTest.java @@ -152,7 +152,11 @@ public final class HpkePrivateKeyManagerTest { assertThat(key.getVersion()).isEqualTo(manager.getVersion()); assertThat(key.getPublicKey().getParams()).isEqualTo(format.getParams()); assertThat(key.getPublicKey().getPublicKey()).isNotEmpty(); + assertThat(key.getPublicKey().getPublicKey().toByteArray().length) + .isEqualTo(HpkeUtil.getEncodedPublicKeyLength(format.getParams().getKem())); assertThat(key.getPrivateKey()).isNotEmpty(); + assertThat(key.getPrivateKey().toByteArray().length) + .isEqualTo(HpkeUtil.getEncodedPrivateKeyLength(format.getParams().getKem())); } @Theory -- cgit v1.2.3