use super::{KeyTypeId, RuntimePublic};
use alloc::vec::Vec;
pub use crate::core::paired_crypto::ecdsa_bls381::*;
use crate::core::{
bls381,
crypto::CryptoType,
ecdsa, ecdsa_bls381,
proof_of_possession::{NonAggregatable, ProofOfPossessionVerifier},
};
mod app {
crate::app_crypto!(super, crate::core::testing::ECDSA_BLS381);
}
#[cfg(feature = "full_crypto")]
pub use app::Pair as AppPair;
pub use app::{
ProofOfPossession as AppProofOfPossession, Public as AppPublic, Signature as AppSignature,
};
impl RuntimePublic for Public {
type Signature = Signature;
type ProofOfPossession = ecdsa_bls381::ProofOfPossession;
fn all(_key_type: KeyTypeId) -> Vec<Self> {
Vec::new()
}
fn generate_pair(key_type: KeyTypeId, seed: Option<Vec<u8>>) -> Self {
crate::io::crypto::ecdsa_bls381_generate(key_type, seed)
}
fn sign<M: AsRef<[u8]>>(&self, _key_type: KeyTypeId, _msg: &M) -> Option<Self::Signature> {
None
}
fn verify<M: AsRef<[u8]>>(&self, _msg: &M, _signature: &Self::Signature) -> bool {
false
}
fn generate_proof_of_possession(
&mut self,
key_type: KeyTypeId,
owner: &[u8],
) -> Option<Self::ProofOfPossession> {
let pub_key_as_bytes = self.to_raw_vec();
let (ecdsa_pub_as_bytes, bls381_pub_as_bytes) = split_pub_key_bytes(&pub_key_as_bytes)?;
let ecdsa_proof_of_possession =
generate_ecdsa_proof_of_possession(key_type, ecdsa_pub_as_bytes, owner)?;
let bls381_proof_of_possession =
generate_bls381_proof_of_possession(key_type, bls381_pub_as_bytes, owner)?;
let combined_proof_of_possession_raw =
combine_proof_of_possession(&ecdsa_proof_of_possession, &bls381_proof_of_possession)?;
Some(Self::ProofOfPossession::from_raw(combined_proof_of_possession_raw))
}
fn verify_proof_of_possession(
&self,
owner: &[u8],
proof_of_possession: &Self::ProofOfPossession,
) -> bool {
let pub_key = AppPublic::from(*self);
<AppPublic as CryptoType>::Pair::verify_proof_of_possession(
owner,
&proof_of_possession,
&pub_key,
)
}
fn to_raw_vec(&self) -> Vec<u8> {
crate::core::crypto::ByteArray::to_raw_vec(self)
}
}
fn split_pub_key_bytes(
pub_key_as_bytes: &[u8],
) -> Option<([u8; ecdsa::PUBLIC_KEY_SERIALIZED_SIZE], [u8; bls381::PUBLIC_KEY_SERIALIZED_SIZE])> {
let ecdsa_pub_as_bytes =
pub_key_as_bytes[..ecdsa::PUBLIC_KEY_SERIALIZED_SIZE].try_into().ok()?;
let bls381_pub_as_bytes =
pub_key_as_bytes[ecdsa::PUBLIC_KEY_SERIALIZED_SIZE..].try_into().ok()?;
Some((ecdsa_pub_as_bytes, bls381_pub_as_bytes))
}
fn generate_ecdsa_proof_of_possession(
key_type: KeyTypeId,
ecdsa_pub_as_bytes: [u8; ecdsa::PUBLIC_KEY_SERIALIZED_SIZE],
owner: &[u8],
) -> Option<ecdsa::Signature> {
let ecdsa_pub = ecdsa::Public::from_raw(ecdsa_pub_as_bytes);
let proof_of_possession_statement = ecdsa::Pair::proof_of_possession_statement(owner);
crate::io::crypto::ecdsa_sign(key_type, &ecdsa_pub, &proof_of_possession_statement)
}
fn generate_bls381_proof_of_possession(
key_type: KeyTypeId,
bls381_pub_as_bytes: [u8; bls381::PUBLIC_KEY_SERIALIZED_SIZE],
owner: &[u8],
) -> Option<bls381::ProofOfPossession> {
let bls381_pub = bls381::Public::from_raw(bls381_pub_as_bytes);
crate::io::crypto::bls381_generate_proof_of_possession(key_type, &bls381_pub, owner)
}
fn combine_proof_of_possession(
ecdsa_proof_of_possession: &ecdsa::Signature,
bls381_proof_of_possession: &bls381::ProofOfPossession,
) -> Option<[u8; ecdsa_bls381::POP_LEN]> {
let mut combined_proof_of_possession_raw = [0u8; ecdsa_bls381::POP_LEN];
combined_proof_of_possession_raw[..ecdsa::SIGNATURE_SERIALIZED_SIZE]
.copy_from_slice(ecdsa_proof_of_possession.as_ref());
combined_proof_of_possession_raw[ecdsa::SIGNATURE_SERIALIZED_SIZE..]
.copy_from_slice(bls381_proof_of_possession.as_ref());
Some(combined_proof_of_possession_raw)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::core::{bls381, crypto::Pair, ecdsa};
fn generate_test_keys(
) -> ([u8; ecdsa::PUBLIC_KEY_SERIALIZED_SIZE], [u8; bls381::PUBLIC_KEY_SERIALIZED_SIZE]) {
let ecdsa_pair = ecdsa::Pair::generate().0;
let bls381_pair = bls381::Pair::generate().0;
let ecdsa_pub = ecdsa_pair.public();
let bls381_pub = bls381_pair.public();
(ecdsa_pub.to_raw_vec().try_into().unwrap(), bls381_pub.to_raw_vec().try_into().unwrap())
}
#[test]
fn test_split_pub_key_bytes() {
let (ecdsa_pub, bls381_pub) = generate_test_keys();
let mut combined_pub_key = Vec::new();
combined_pub_key.extend_from_slice(&ecdsa_pub);
combined_pub_key.extend_from_slice(&bls381_pub);
let result = split_pub_key_bytes(&combined_pub_key).unwrap();
assert_eq!(result.0, ecdsa_pub, "ECDSA public key does not match");
assert_eq!(result.1, bls381_pub, "BLS381 public key does not match");
}
}