use base64ct::Base64UrlUnpadded;
use base64ct::Encoding;
use serde::{de::DeserializeOwned, Serialize};
use sha2::Digest;
use crate::claims::*;
use crate::common::*;
use crate::error::*;
use crate::jwt_header::*;
use crate::token::*;
#[doc(hidden)]
#[derive(Debug, Clone)]
pub struct Edwards25519PublicKey(ed25519_compact::PublicKey);
impl AsRef<ed25519_compact::PublicKey> for Edwards25519PublicKey {
fn as_ref(&self) -> &ed25519_compact::PublicKey {
&self.0
}
}
impl Edwards25519PublicKey {
pub fn from_bytes(raw: &[u8]) -> Result<Self, JWTError> {
let ed25519_pk = ed25519_compact::PublicKey::from_slice(raw);
Ok(Edwards25519PublicKey(
ed25519_pk.map_err(|_| JWTError::InvalidPublicKey)?,
))
}
pub fn from_der(der: &[u8]) -> Result<Self, JWTError> {
let ed25519_pk = ed25519_compact::PublicKey::from_der(der);
Ok(Edwards25519PublicKey(
ed25519_pk.map_err(|_| JWTError::InvalidPublicKey)?,
))
}
pub fn from_pem(pem: &str) -> Result<Self, JWTError> {
let ed25519_pk = ed25519_compact::PublicKey::from_pem(pem);
Ok(Edwards25519PublicKey(
ed25519_pk.map_err(|_| JWTError::InvalidPublicKey)?,
))
}
pub fn to_bytes(&self) -> Vec<u8> {
self.0.as_ref().to_vec()
}
pub fn to_der(&self) -> Vec<u8> {
self.0.to_der()
}
pub fn to_pem(&self) -> String {
self.0.to_pem()
}
}
#[doc(hidden)]
#[derive(Clone)]
pub struct Edwards25519KeyPair {
ed25519_kp: ed25519_compact::KeyPair,
metadata: Option<KeyMetadata>,
}
impl AsRef<ed25519_compact::KeyPair> for Edwards25519KeyPair {
fn as_ref(&self) -> &ed25519_compact::KeyPair {
&self.ed25519_kp
}
}
impl Edwards25519KeyPair {
pub fn from_bytes(raw: &[u8]) -> Result<Self, JWTError> {
let ed25519_kp = ed25519_compact::KeyPair::from_slice(raw)?;
Ok(Edwards25519KeyPair {
ed25519_kp,
metadata: None,
})
}
pub fn from_der(der: &[u8]) -> Result<Self, JWTError> {
let ed25519_kp = match ed25519_compact::KeyPair::from_der(der) {
Ok(kp) => kp,
Err(_) => ed25519_compact::KeyPair::from_seed(
ed25519_compact::SecretKey::from_der(der)?.seed(),
),
};
Ok(Edwards25519KeyPair {
ed25519_kp,
metadata: None,
})
}
pub fn from_pem(pem: &str) -> Result<Self, JWTError> {
let ed25519_kp = match ed25519_compact::KeyPair::from_pem(pem) {
Ok(kp) => kp,
Err(_) => ed25519_compact::KeyPair::from_seed(
ed25519_compact::SecretKey::from_pem(pem)?.seed(),
),
};
Ok(Edwards25519KeyPair {
ed25519_kp,
metadata: None,
})
}
pub fn to_bytes(&self) -> Vec<u8> {
self.ed25519_kp.to_vec()
}
pub fn to_der(&self) -> Vec<u8> {
self.ed25519_kp.sk.to_der()
}
pub fn to_pem(&self) -> String {
self.ed25519_kp.to_pem()
}
pub fn public_key(&self) -> Edwards25519PublicKey {
let ed25519_pk = self.ed25519_kp.pk;
Edwards25519PublicKey(ed25519_pk)
}
pub fn generate() -> Self {
let ed25519_kp = ed25519_compact::KeyPair::from_seed(ed25519_compact::Seed::generate());
Edwards25519KeyPair {
ed25519_kp,
metadata: None,
}
}
}
pub trait EdDSAKeyPairLike {
fn jwt_alg_name() -> &'static str;
fn key_pair(&self) -> &Edwards25519KeyPair;
fn key_id(&self) -> &Option<String>;
fn metadata(&self) -> &Option<KeyMetadata>;
fn attach_metadata(&mut self, metadata: KeyMetadata) -> Result<(), JWTError>;
fn sign<CustomClaims: Serialize + DeserializeOwned>(
&self,
claims: JWTClaims<CustomClaims>,
) -> Result<String, JWTError> {
let jwt_header = JWTHeader::new(Self::jwt_alg_name().to_string(), self.key_id().clone())
.with_metadata(self.metadata());
Token::build(&jwt_header, claims, |authenticated| {
let noise = ed25519_compact::Noise::generate();
let signature = self.key_pair().as_ref().sk.sign(authenticated, Some(noise));
Ok(signature.to_vec())
})
}
}
pub trait EdDSAPublicKeyLike {
fn jwt_alg_name() -> &'static str;
fn public_key(&self) -> &Edwards25519PublicKey;
fn key_id(&self) -> &Option<String>;
fn set_key_id(&mut self, key_id: String);
fn verify_token<CustomClaims: Serialize + DeserializeOwned>(
&self,
token: &str,
options: Option<VerificationOptions>,
) -> Result<JWTClaims<CustomClaims>, JWTError> {
Token::verify(
Self::jwt_alg_name(),
token,
options,
|authenticated, signature| {
let ed25519_signature = ed25519_compact::Signature::from_slice(signature)?;
self.public_key()
.as_ref()
.verify(authenticated, &ed25519_signature)
.map_err(|_| JWTError::InvalidSignature)?;
Ok(())
},
)
}
fn create_key_id(&mut self) -> Result<String, JWTError> {
let mut hasher = sha2::Sha256::new();
hasher.update(self.public_key().to_bytes());
let key_id = Base64UrlUnpadded::encode_string(&hasher.finalize());
self.set_key_id(key_id.clone());
Ok(key_id)
}
}
#[derive(Clone)]
pub struct Ed25519KeyPair {
key_pair: Edwards25519KeyPair,
key_id: Option<String>,
}
#[derive(Debug, Clone)]
pub struct Ed25519PublicKey {
pk: Edwards25519PublicKey,
key_id: Option<String>,
}
impl EdDSAKeyPairLike for Ed25519KeyPair {
fn jwt_alg_name() -> &'static str {
"EdDSA"
}
fn key_pair(&self) -> &Edwards25519KeyPair {
&self.key_pair
}
fn key_id(&self) -> &Option<String> {
&self.key_id
}
fn metadata(&self) -> &Option<KeyMetadata> {
&self.key_pair.metadata
}
fn attach_metadata(&mut self, metadata: KeyMetadata) -> Result<(), JWTError> {
self.key_pair.metadata = Some(metadata);
Ok(())
}
}
impl Ed25519KeyPair {
pub fn from_bytes(raw: &[u8]) -> Result<Self, JWTError> {
Ok(Ed25519KeyPair {
key_pair: Edwards25519KeyPair::from_bytes(raw)?,
key_id: None,
})
}
pub fn from_der(der: &[u8]) -> Result<Self, JWTError> {
Ok(Ed25519KeyPair {
key_pair: Edwards25519KeyPair::from_der(der)?,
key_id: None,
})
}
pub fn from_pem(pem: &str) -> Result<Self, JWTError> {
Ok(Ed25519KeyPair {
key_pair: Edwards25519KeyPair::from_pem(pem)?,
key_id: None,
})
}
pub fn to_bytes(&self) -> Vec<u8> {
self.key_pair.to_bytes()
}
pub fn to_der(&self) -> Vec<u8> {
self.key_pair.to_der()
}
pub fn to_pem(&self) -> String {
self.key_pair.to_pem()
}
pub fn public_key(&self) -> Ed25519PublicKey {
Ed25519PublicKey {
pk: self.key_pair.public_key(),
key_id: self.key_id.clone(),
}
}
pub fn generate() -> Self {
Ed25519KeyPair {
key_pair: Edwards25519KeyPair::generate(),
key_id: None,
}
}
pub fn with_key_id(mut self, key_id: &str) -> Self {
self.key_id = Some(key_id.to_string());
self
}
}
impl EdDSAPublicKeyLike for Ed25519PublicKey {
fn jwt_alg_name() -> &'static str {
"EdDSA"
}
fn public_key(&self) -> &Edwards25519PublicKey {
&self.pk
}
fn key_id(&self) -> &Option<String> {
&self.key_id
}
fn set_key_id(&mut self, key_id: String) {
self.key_id = Some(key_id);
}
}
impl Ed25519PublicKey {
pub fn from_bytes(raw: &[u8]) -> Result<Self, JWTError> {
Ok(Ed25519PublicKey {
pk: Edwards25519PublicKey::from_bytes(raw)?,
key_id: None,
})
}
pub fn from_der(der: &[u8]) -> Result<Self, JWTError> {
Ok(Ed25519PublicKey {
pk: Edwards25519PublicKey::from_der(der)?,
key_id: None,
})
}
pub fn from_pem(pem: &str) -> Result<Self, JWTError> {
Ok(Ed25519PublicKey {
pk: Edwards25519PublicKey::from_pem(pem)?,
key_id: None,
})
}
pub fn to_bytes(&self) -> Vec<u8> {
self.pk.to_bytes()
}
pub fn to_der(&self) -> Vec<u8> {
self.pk.to_der()
}
pub fn to_pem(&self) -> String {
self.pk.to_pem()
}
pub fn with_key_id(mut self, key_id: &str) -> Self {
self.key_id = Some(key_id.to_string());
self
}
pub fn sha256_thumbprint(&self) -> Result<String, JWTError> {
let mut hasher = sha2::Sha256::new();
hasher.update(self.to_der());
Ok(Base64UrlUnpadded::encode_string(&hasher.finalize()))
}
}