use std::str::FromStr;
use num_enum::TryFromPrimitive;
use crate::{IdLookupError, Keypair};
#[non_exhaustive]
#[repr(u16)]
#[derive(Copy, Clone, Debug, PartialEq, Eq, TryFromPrimitive)]
#[cfg_attr(
feature = "serde",
derive(serde_crate::Serialize, serde_crate::Deserialize)
)]
#[cfg_attr(feature = "serde", serde(crate = "serde_crate"))]
pub enum Kem {
#[cfg(feature = "nistp")]
DhP256HkdfSha256 = 0x0010,
#[cfg(feature = "nistp")]
DhP384HkdfSha384 = 0x0011,
#[cfg(feature = "nistp")]
DhP521HkdfSha512 = 0x0012,
#[cfg(feature = "x25519")]
X25519HkdfSha256 = 0x0020,
#[cfg(feature = "mlkem")]
MlKem768 = 0x0041,
#[cfg(feature = "mlkem")]
MlKem1024 = 0x0042,
#[cfg(all(feature = "mlkem", feature = "x25519"))]
XWing = 0x647a,
#[cfg(all(feature = "mlkem", feature = "nistp"))]
MlKem768P256 = 0x0050,
#[cfg(all(feature = "mlkem", feature = "nistp"))]
MlKem1024P384 = 0x0051,
}
impl FromStr for Kem {
type Err = IdLookupError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match &*s.to_lowercase().replace('-', "") {
#[cfg(feature = "nistp")]
"p256sha256" | "dhkemp256hkdfsha256" | "p256hkdfsha256" | "dhkem(p256, hkdfsha256)" => {
Ok(Self::DhP256HkdfSha256)
}
#[cfg(feature = "nistp")]
"p384sha384" | "dhkemp384hkdfsha384" | "p384hkdfsha384" | "dhkem(p384, hkdfsha384)" => {
Ok(Self::DhP384HkdfSha384)
}
#[cfg(feature = "nistp")]
"p521sha512" | "dhkemp521hkdfsha512" | "p521hkdfsha512" | "dhkem(p521, hkdfsha512)" => {
Ok(Self::DhP521HkdfSha512)
}
#[cfg(feature = "x25519")]
"x25519sha256"
| "dhkemx25519hkdfsha256"
| "x25519hkdfsha256"
| "dhkem(x25519, hkdfsha256)" => Ok(Self::X25519HkdfSha256),
#[cfg(feature = "mlkem")]
"mlkem768" => Ok(Self::MlKem768),
#[cfg(feature = "mlkem")]
"mlkem1024" => Ok(Self::MlKem1024),
#[cfg(all(feature = "mlkem", feature = "x25519"))]
"xwing" | "mlkem768x25119" | "mlkem768-x25119" => Ok(Self::XWing),
#[cfg(all(feature = "mlkem", feature = "nistp"))]
"mlkem768p256" | "mlkem768-p256" => Ok(Self::MlKem768P256),
#[cfg(all(feature = "mlkem", feature = "nistp"))]
"mlkem1024p384" | "mlkem1024-p384" => Ok(Self::MlKem1024P384),
_ => Err(IdLookupError("kem not recognized")),
}
}
}
impl Kem {
#[must_use]
pub fn gen_keypair(self) -> Keypair {
crate::gen_keypair(self)
}
}
pub const KEM_ALL: &[Kem] = &[
#[cfg(feature = "nistp")]
Kem::DhP256HkdfSha256,
#[cfg(feature = "nistp")]
Kem::DhP384HkdfSha384,
#[cfg(feature = "nistp")]
Kem::DhP521HkdfSha512,
#[cfg(feature = "x25519")]
Kem::X25519HkdfSha256,
#[cfg(feature = "mlkem")]
Kem::MlKem768,
#[cfg(feature = "mlkem")]
Kem::MlKem1024,
#[cfg(all(feature = "mlkem", feature = "x25519"))]
Kem::XWing,
#[cfg(all(feature = "mlkem", feature = "nistp"))]
Kem::MlKem768P256,
#[cfg(all(feature = "mlkem", feature = "nistp"))]
Kem::MlKem1024P384,
];