Add missing k256 kat_tests

rozbb · Aug 11, 2024 · 73dc13d · 73dc13d
1 parent 1c14d38
commit 73dc13d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/kat_tests.rs b/src/kat_tests.rs
@@ -88,6 +88,18 @@ impl TestableKem for DhP521HkdfSha512 {
     }
 }
 
+impl TestableKem for Dh256HkdfSha256 {
+    type EphemeralKey = <Dh256HkdfSha256 as KemTrait>::PrivateKey;
+
+    fn encap_with_eph(
+        pk_recip: &Self::PublicKey,
+        sender_id_keypair: Option<(&Self::PrivateKey, &Self::PublicKey)>,
+        sk_eph: Self::EphemeralKey,
+    ) -> Result<(SharedSecret<Self>, Self::EncappedKey), HpkeError> {
+        kem::dhk256_hkdfsha256::encap_with_eph(pk_recip, sender_id_keypair, sk_eph)
+    }
+}
+
 /// Asserts that the given serializable values are equal
 macro_rules! assert_serializable_eq {
     ($a:expr, $b:expr, $args:tt) => {

diff --git a/src/lib.rs b/src/lib.rs
@@ -102,7 +102,8 @@ pub(crate) use alloc::vec::Vec;
     feature = "x25519",
     feature = "p256",
     feature = "p384",
-    feature = "p521"
+    feature = "p521",
+    feature = "k256"
 ))]
 mod kat_tests;