hpke-dispatch 0.9.0

Runtime algorithm selection for hybrid public key encryption
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use hpke::Serializable;

use crate::Kem;

/// An encoded keypair
#[derive(Debug, Clone, Eq, PartialEq, zeroize::Zeroize)]
pub struct Keypair {
    /// the public key for this keypair
    pub public_key: Vec<u8>,

    /// the private key for this keypair,
    pub private_key: Vec<u8>,
}

impl Keypair {
    /// generate a keypair from a [`Kem`]
    #[must_use]
    pub fn new(kem: Kem) -> Keypair {
        gen_keypair(kem)
    }
}

impl Keypair {
    /// deconstructs the pair into (private, public)
    #[must_use]
    pub fn into_parts(self) -> (Vec<u8>, Vec<u8>) {
        (self.private_key, self.public_key)
    }
}

/// generate a Keypair for the provided asymmetric key encapsulation mechanism ([`Kem`])
#[must_use]
pub fn gen_keypair(kem: Kem) -> Keypair {
    match kem {
        #[cfg(feature = "nistp")]
        Kem::DhP256HkdfSha256 => gen_kp::<hpke::kem::DhP256HkdfSha256>(),

        #[cfg(feature = "nistp")]
        Kem::DhP384HkdfSha384 => gen_kp::<hpke::kem::DhP384HkdfSha384>(),

        #[cfg(feature = "nistp")]
        Kem::DhP521HkdfSha512 => gen_kp::<hpke::kem::DhP521HkdfSha512>(),

        #[cfg(feature = "x25519")]
        Kem::X25519HkdfSha256 => gen_kp::<hpke::kem::X25519HkdfSha256>(),

        #[cfg(feature = "mlkem")]
        Kem::MlKem768 => gen_kp::<hpke::kem::MlKem768>(),

        #[cfg(feature = "mlkem")]
        Kem::MlKem1024 => gen_kp::<hpke::kem::MlKem1024>(),

        #[cfg(all(feature = "mlkem", feature = "x25519"))]
        Kem::XWing => gen_kp::<hpke::kem::XWing>(),

        #[cfg(all(feature = "mlkem", feature = "nistp"))]
        Kem::MlKem768P256 => gen_kp::<hpke::kem::MlKem768P256>(),

        #[cfg(all(feature = "mlkem", feature = "nistp"))]
        Kem::MlKem1024P384 => gen_kp::<hpke::kem::MlKem1024P384>(),
    }
}

fn gen_kp<KemT: hpke::kem::Kem>() -> Keypair {
    let (private_key, public_key) = KemT::gen_keypair();
    let public_key = public_key.to_bytes().to_vec();
    let private_key = private_key.to_bytes().to_vec();

    Keypair {
        public_key,
        private_key,
    }
}