use super::sphinx::{
derive_kx_public, derive_kx_shared_secret, gen_kx_secret, KxPublic, KxSecret, SharedSecret,
};
use rand::{CryptoRng, Rng};
use zeroize::Zeroizing;
pub struct KxPair {
secret: Box<Zeroizing<KxSecret>>,
public: KxPublic,
}
impl KxPair {
pub fn gen(rng: &mut (impl Rng + CryptoRng)) -> Self {
gen_kx_secret(rng).into()
}
pub fn public(&self) -> &KxPublic {
&self.public
}
pub fn exchange(&self, their_public: &KxPublic) -> SharedSecret {
derive_kx_shared_secret(their_public, self.secret.as_ref())
}
}
impl From<KxSecret> for KxPair {
fn from(secret: KxSecret) -> Self {
let secret = Box::new(Zeroizing::new(secret));
let public = derive_kx_public(secret.as_ref());
Self { secret, public }
}
}