//! Secp256r1 signing scheme implementation. //! //! This module provides both the generic Secp256r1 implementation and a macro to generate //! protocol-specific wrappers. #[cfg(feature = "mocks")] pub mod mocks; use crate::{ Signer as _, Verifier as _, certificate::{AssemblyError, Attestation, Namespace, Scheme, Signers, Subject, Verification}, secp256r1::standard::{PrivateKey, PublicKey, Signature as Secp256r1Signature}, }; #[cfg(not(feature = "std"))] use alloc::{collections::BTreeSet, vec::Vec}; use bytes::BufMut; use commonware_codec::{ Buf, EncodeSize, Error, FixedSize, Read, ReadRangeExt, Write, types::lazy::Lazy, }; use commonware_utils::{ Participant, Widen, iter::NonEmpty, ordered::{BiMap, Quorum, Set}, }; use rand_core::CryptoRng; #[cfg(feature = "std")] use std::collections::BTreeSet; /// Generic Secp256r1 signing scheme implementation parameterized by identity type. /// /// This struct contains the core cryptographic operations without protocol-specific /// context types. It can be reused across different protocols (simplex, aggregation, etc.) /// by wrapping it with protocol-specific trait implementations via the macro. #[derive(Clone, Debug)] pub struct Generic { /// Participants in the committee. pub participants: BiMap, /// Key used for generating signatures. pub signer: Option<(Participant, PrivateKey)>, /// Pre-computed namespace(s) for this subject type. pub namespace: N, } impl Generic { /// Creates a new scheme instance with the provided key material. /// /// Participants have both an identity key and a signing key. The identity key /// is used for participant set ordering and indexing, while the signing key is used for /// signing and verification. /// /// Returns `None` if the provided private key does not match any signing key /// in the participant set. pub fn signer( namespace: &[u8], participants: BiMap, private_key: PrivateKey, ) -> Option { let public_key = private_key.public_key(); let signer = participants .values() .iter() .position(|p| p == &public_key) .map(|index| (Participant::from_usize(index), private_key))?; Some(Self { participants, signer: Some(signer), namespace: N::derive(namespace), }) } /// Builds a verifier that can authenticate signatures and certificates. /// /// Participants have both an identity key and a signing key. The identity key /// is used for participant set ordering and indexing, while the signing key is used for /// verification. pub fn verifier(namespace: &[u8], participants: BiMap) -> Self { Self { participants, signer: None, namespace: N::derive(namespace), } } /// Returns the ordered set of identity keys. pub const fn participants(&self) -> &Set

{ self.participants.keys() } /// Returns the index this instance signs at, or `None` if it cannot sign. pub fn signer_index(&self) -> Option { self.signer.as_ref().map(|(index, _)| *index) } /// Signs a subject and returns the attestation. pub fn sign(&self, subject: impl Subject) -> Option> where S: Scheme, { let (index, private_key) = self.signer.as_ref()?; let signature = private_key.sign(subject.namespace(&self.namespace), &subject.message()); Some(Attestation { signer: *index, signature: signature.into(), }) } /// Verifies a single attestation from a signer. pub fn verify_attestation( &self, subject: impl Subject, attestation: &Attestation, ) -> bool where S: Scheme, { let Some(public_key) = self.participants.value(attestation.signer.into()) else { return false; }; let Some(signature) = attestation.signature.get() else { return false; }; public_key.verify( subject.namespace(&self.namespace), &subject.message(), signature, ) } /// Verifies attestations one-by-one and returns verified attestations and invalid signers. pub fn verify_attestations( &self, _rng: &mut R, subject: impl Subject, attestations: I, ) -> Verification where S: Scheme, R: CryptoRng, I: IntoIterator>, { let namespace = subject.namespace(&self.namespace); let message = subject.message(); let mut invalid = BTreeSet::new(); let mut verified = Vec::new(); for attestation in attestations.into_iter() { let Some(public_key) = self.participants.value(attestation.signer.into()) else { invalid.insert(attestation.signer); continue; }; let Some(signature) = attestation.signature.get() else { invalid.insert(attestation.signer); continue; }; if public_key.verify(namespace, &message, signature) { verified.push(attestation); } else { invalid.insert(attestation.signer); } } Verification::new(verified, invalid.into_iter().collect()) } /// Assembles a certificate from a non-empty collection of attestations. pub fn assemble(&self, attestations: NonEmpty) -> Result where S: Scheme, I: Iterator>, { // Collect the signers and signatures. let mut entries = Vec::new(); for Attestation { signer, signature } in attestations { self.participants .value(signer.into()) .ok_or(AssemblyError::UnknownSigner(signer))?; let signature = signature .get() .cloned() .ok_or(AssemblyError::MalformedSignature(signer))?; entries.push((signer, signature)); } // Sort the signatures by signer index. entries.sort_by_key(|(signer, _)| *signer); let (signer, signatures): (Vec, Vec<_>) = entries.into_iter().unzip(); let signers = Signers::try_from((self.participants.keys(), signer))? .require(self.participants.quorum::())?; let signatures = signatures.into_iter().map(Lazy::from).collect(); Ok(Certificate { signers, signatures, }) } /// Verifies a certificate by checking each signature individually. pub fn verify_certificate( &self, _rng: &mut R, subject: impl Subject, certificate: &Certificate, ) -> bool where S: Scheme, R: CryptoRng, { // If the certificate signers length does not match the participant set, return false. if certificate.signers.len() != self.participants.len() { return false; } // If the certificate signers and signatures counts differ, return false. if certificate.signers.count() != certificate.signatures.len() { return false; } // If the certificate does not meet the quorum, return false. if certificate.signers.count() < Widen::widen(self.participants.quorum::()) { return false; } let namespace = subject.namespace(&self.namespace); let message = subject.message(); for (signer, signature) in certificate.signers.iter().zip(&certificate.signatures) { let Some(public_key) = self.participants.value(signer.into()) else { return false; }; let Some(signature) = signature.get() else { return false; }; if !public_key.verify(namespace, &message, signature) { return false; } } true } pub const fn is_attributable() -> bool { true } pub const fn is_batchable() -> bool { false } /// Returns an upper bound on the encoded certificate size for at most `participants` /// participants. pub fn max_certificate_size(participants: u32) -> usize { crate::certificate::max_individual_certificate_size(participants, Secp256r1Signature::SIZE) } pub const fn certificate_codec_config(&self) -> ::Cfg { self.participants.len() } pub const fn certificate_codec_config_unbounded() -> ::Cfg { u32::MAX as usize } } #[derive(Clone, Debug, PartialEq, Eq, Hash)] pub struct Certificate { /// Bitmap of participant indices that contributed signatures. pub signers: Signers, /// Secp256r1 signatures emitted by the respective participants ordered by signer index. pub signatures: Vec>, } #[cfg(feature = "arbitrary")] impl arbitrary::Arbitrary<'_> for Certificate { fn arbitrary(u: &mut arbitrary::Unstructured<'_>) -> arbitrary::Result { let signers = Signers::arbitrary(u)?; let signatures = (0..signers.count()) .map(|_| u.arbitrary::().map(Lazy::from)) .collect::>>()?; Ok(Self { signers, signatures, }) } } impl Write for Certificate { fn write(&self, writer: &mut impl BufMut) { self.signers.write(writer); self.signatures.write(writer); } } impl EncodeSize for Certificate { fn encode_size(&self) -> usize { self.signers.encode_size() + self.signatures.encode_size() } } impl Read for Certificate { type Cfg = usize; fn read_cfg(reader: &mut impl Buf, participants: &usize) -> Result { let signers = Signers::read_cfg(reader, participants)?; if signers.count() == 0 { return Err(Error::Invalid( "cryptography::secp256r1::certificate::Certificate", "Certificate contains no signers", )); } let signatures = Vec::>::read_range(reader, ..=*participants)?; if signers.count() != signatures.len() { return Err(Error::Invalid( "cryptography::secp256r1::certificate::Certificate", "Signers and signatures counts differ", )); } Ok(Self { signers, signatures, }) } } /// Generates a Secp256r1 signing scheme wrapper for a specific protocol. /// /// This macro creates a complete wrapper struct with constructors, `Scheme` trait /// implementation, and a `fixture` function for testing. /// /// # Parameters /// /// - `$subject`: The subject type used as `Scheme::Subject<'a, D>`. Use `'a` and `D` /// in the subject type to bind to the GAT lifetime and digest type parameters. /// /// - `$namespace`: The namespace type that implements [`Namespace`]. /// This type pre-computes and stores any protocol-specific namespace bytes derived from /// a base namespace. The scheme calls `$namespace::derive(base)` at construction time /// to create the namespace, then passes it to `Subject::namespace()` during signing /// and verification. For simple protocols with only a base namespace, `Vec` can be used directly. /// For protocols with multiple message types, a custom struct can pre-compute all variants. /// /// - `$faults`: The [`Faults`](commonware_utils::Faults) implementation used to compute certificate /// quorums. /// /// # Example /// ```ignore /// // For non-generic subject types with a single namespace: /// impl_certificate_secp256r1!(MySubject, Vec, commonware_utils::N3f1); /// /// // For protocols with generic subject types: /// impl_certificate_secp256r1!( /// Subject<'a, D>, /// Namespace, /// commonware_utils::N3f1, /// ); /// ``` #[macro_export] macro_rules! impl_certificate_secp256r1 { ($subject:ty, $namespace:ty, $faults:ty) => { /// Generates a test fixture with Ed25519 identities and Secp256r1 signing schemes. /// /// Returns a [`commonware_cryptography::certificate::mocks::Fixture`] whose keys and /// scheme instances share a consistent ordering. #[cfg(feature = "mocks")] #[allow(dead_code)] pub fn fixture( rng: &mut R, namespace: &[u8], n: u32, ) -> $crate::certificate::mocks::Fixture> where R: rand_core::CryptoRng, { $crate::secp256r1::certificate::mocks::fixture( rng, namespace, n, Scheme::signer, Scheme::verifier, ) } /// Secp256r1 signing scheme wrapper. #[derive(Clone, Debug)] pub struct Scheme { /// Verification state, plus the signing material of a signer. generic: $crate::secp256r1::certificate::Generic, /// Index of the local identity key for an instance built with [`Self::member`]. member: Option, } impl Scheme

{ /// Creates a new scheme instance with the provided key material. pub fn signer( namespace: &[u8], participants: commonware_utils::ordered::BiMap, private_key: $crate::secp256r1::standard::PrivateKey, ) -> Option { Some(Self { generic: $crate::secp256r1::certificate::Generic::signer( namespace, participants, private_key, )?, member: None, }) } /// Builds a verifier that can authenticate signatures and certificates. /// /// A participant without signing material should use [`Self::member`] instead, so the /// scheme reports its membership. pub fn verifier( namespace: &[u8], participants: commonware_utils::ordered::BiMap, ) -> Self { Self { generic: $crate::secp256r1::certificate::Generic::verifier( namespace, participants, ), member: None, } } /// Builds a committee member that verifies like [`Self::verifier`] but also knows /// its own place in the participant set. /// /// The instance cannot sign, and [`Scheme::me`] returns [`Membership::Member`] at /// the position of the identity key `public_key` in the participant set. Returns /// `None` if `public_key` is not a participant identity key. /// /// [`Scheme::me`]: commonware_cryptography::certificate::Scheme::me /// [`Membership::Member`]: commonware_cryptography::certificate::Membership::Member pub fn member( namespace: &[u8], participants: commonware_utils::ordered::BiMap, public_key: &P, ) -> Option { let member = commonware_utils::ordered::Quorum::index(participants.keys(), public_key)?; Some(Self { generic: $crate::secp256r1::certificate::Generic::verifier( namespace, participants, ), member: Some(member), }) } } impl $crate::certificate::Verifier for Scheme

{ type Subject<'a, D: $crate::Digest> = $subject; type Faults = $faults; type PublicKey = P; type Certificate = $crate::secp256r1::certificate::Certificate; fn verify_certificate( &self, rng: &mut R, subject: Self::Subject<'_, D>, certificate: &Self::Certificate, _strategy: &impl commonware_parallel::Strategy, ) -> bool where R: rand_core::CryptoRng, D: $crate::Digest, { self.generic.verify_certificate::( rng, subject, certificate, ) } fn verify_certificates<'a, R, D, I>( &self, rng: &mut R, certificates: commonware_utils::iter::NonEmpty, _strategy: &impl commonware_parallel::Strategy, ) -> bool where R: rand_core::CryptoRng, D: $crate::Digest, I: Iterator, &'a Self::Certificate)>, { for (subject, certificate) in certificates { if !self.generic.verify_certificate::(rng, subject, certificate) { return false; } } true } fn is_batchable() -> bool { $crate::secp256r1::certificate::Generic::::is_batchable() } fn max_certificate_size(participants: u32) -> usize { $crate::secp256r1::certificate::Generic::::max_certificate_size(participants) } fn certificate_codec_config( &self, ) -> ::Cfg { self.generic.certificate_codec_config() } fn certificate_codec_config_unbounded() -> ::Cfg { $crate::secp256r1::certificate::Generic::::certificate_codec_config_unbounded() } } impl $crate::certificate::Scheme for Scheme

{ type Signature = $crate::secp256r1::standard::Signature; fn me(&self) -> Option<$crate::certificate::Membership> { self.generic .signer_index() .map($crate::certificate::Membership::Signer) .or(self.member.map($crate::certificate::Membership::Member)) } fn participants(&self) -> &commonware_utils::ordered::Set { self.generic.participants() } fn sign( &self, subject: Self::Subject<'_, D>, ) -> Option<$crate::certificate::Attestation> { self.generic.sign(subject) } fn verify_attestation( &self, _rng: &mut R, subject: Self::Subject<'_, D>, attestation: &$crate::certificate::Attestation, _strategy: &impl commonware_parallel::Strategy, ) -> bool where R: rand_core::CryptoRng, D: $crate::Digest, { self.generic .verify_attestation(subject, attestation) } fn verify_attestations( &self, rng: &mut R, subject: Self::Subject<'_, D>, attestations: I, _strategy: &impl commonware_parallel::Strategy, ) -> $crate::certificate::Verification where R: rand_core::CryptoRng, D: $crate::Digest, I: IntoIterator>, { self.generic.verify_attestations( rng, subject, attestations, ) } fn optimistic_assemble<'a, R, D, I, J>( &self, rng: &mut R, subject: Self::Subject<'_, D>, pending: I, verified: J, strategy: &impl commonware_parallel::Strategy, ) -> Result> where R: rand_core::CryptoRng, D: $crate::Digest, I: IntoIterator>, I::IntoIter: ExactSizeIterator + Send, J: IntoIterator>, J::IntoIter: Send, { $crate::certificate::verify_then_assemble::( self, rng, subject, pending, verified, strategy, ) } fn assemble( &self, attestations: commonware_utils::iter::NonEmpty, _strategy: &impl commonware_parallel::Strategy, ) -> Result where I: Iterator> + Send, { self.generic.assemble::(attestations) } fn is_attributable() -> bool { $crate::secp256r1::certificate::Generic::::is_attributable() } } }; } #[cfg(test)] mod tests { use super::*; use crate::{ certificate::{Membership, Scheme as _, Verifier as _}, sha256::Digest as Sha256Digest, }; use bytes::Bytes; use commonware_codec::{Decode, Encode, varint::MAX_U32_VARINT_SIZE}; use commonware_math::algebra::Random; use commonware_parallel::Sequential; use commonware_utils::{Faults, N3f1, TryCollect, non_empty, ordered::BiMap, test_rng}; use rand_core::CryptoRng; use std::sync::{ Arc, atomic::{AtomicUsize, Ordering}, }; const NAMESPACE: &[u8] = b"test-secp256r1"; const MESSAGE: &[u8] = b"test message"; /// Test context type for generic scheme tests. #[derive(Clone, Debug)] pub struct TestSubject { pub message: Bytes, } impl Subject for TestSubject { type Namespace = Vec; fn namespace<'a>(&self, derived: &'a Self::Namespace) -> &'a [u8] { derived.as_ref() } fn message(&self) -> Bytes { self.message.clone() } } // Use the macro to generate the test scheme impl_certificate_secp256r1!(TestSubject, Vec, N3f1); fn setup_signers( rng: &mut impl CryptoRng, n: u32, ) -> (Vec>, Scheme) { let private_keys: Vec<_> = (0..n).map(|_| PrivateKey::random(&mut *rng)).collect(); // For tests, use secp256r1 keys as both identity and signing keys let participants: BiMap = private_keys .iter() .map(|sk| { let pk = sk.public_key(); (pk.clone(), pk) }) .try_collect() .unwrap(); let signers = private_keys .into_iter() .map(|sk| Scheme::signer(NAMESPACE, participants.clone(), sk).unwrap()) .collect(); let verifier = Scheme::verifier(NAMESPACE, participants); (signers, verifier) } /// A member reports its membership, cannot sign, and verifies a quorum certificate, while a /// key outside the committee yields no member. #[test] fn test_member_reports_membership_without_signing() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let participants = verifier.generic.participants.clone(); let subject = TestSubject { message: Bytes::from_static(MESSAGE), }; // A key outside the committee yields no member. let outsider = PrivateKey::random(&mut rng).public_key(); assert!(Scheme::member(NAMESPACE, participants.clone(), &outsider).is_none()); // A member reports its membership but cannot sign. Verifiers report no membership, and // each signer reports the index it signs at. let index = Participant::new(2); let key = participants.keys().key(index).unwrap().clone(); let member = Scheme::member(NAMESPACE, participants, &key).unwrap(); assert_eq!(member.me(), Some(Membership::Member(index))); assert_eq!(member.me().unwrap().signer(), None); assert!(member.sign::(subject.clone()).is_none()); assert_eq!(verifier.me(), None); for scheme in &schemes { let Some(Membership::Signer(signer)) = scheme.me() else { panic!("signer must report signing membership"); }; let attestation = scheme.sign::(subject.clone()).unwrap(); assert_eq!(attestation.signer, signer); } // A member verifies a certificate assembled by a quorum of signers. let attestations: Vec<_> = schemes .iter() .take(Widen::widen(N3f1::quorum(schemes.len()))) .map(|s| s.sign::(subject.clone()).unwrap()) .collect(); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); assert!(member.verify_certificate::<_, Sha256Digest>( &mut rng, subject, &certificate, &Sequential, )); } /// A member is looked up by its identity key, so each identity key yields a member at its /// position and no signing key yields a member. #[test] fn test_member_uses_identity_keys() { let mut rng = test_rng(); // Build a committee whose signing keys differ from its identity keys. let (_, verifier) = setup_signers(&mut rng, 4); let identities = verifier.participants().clone(); let signing_keys: Vec<_> = (0..identities.len()) .map(|_| PrivateKey::random(&mut rng).public_key()) .collect(); let participants: BiMap = identities .iter() .cloned() .zip(signing_keys.iter().cloned()) .try_collect() .unwrap(); // Each identity key yields a member at its position. for (position, identity) in identities.iter().enumerate() { let member = Scheme::member(NAMESPACE, participants.clone(), identity).unwrap(); assert_eq!( member.me(), Some(Membership::Member(Participant::from_usize(position))) ); } // No signing key yields a member. for signing_key in &signing_keys { assert!(Scheme::member(NAMESPACE, participants.clone(), signing_key).is_none()); } } // The counting scheme built inside this test uses only its signer and verifier constructors. #[allow(dead_code)] #[test] fn test_optimistic_assemble_verifies_pending_once() { /// Counts message preparation without changing the signed subject. #[derive(Clone, Debug)] pub struct CountingSubject { message: Bytes, calls: Arc, } impl Subject for CountingSubject { type Namespace = Vec; fn namespace<'a>(&self, derived: &'a Self::Namespace) -> &'a [u8] { derived } fn message(&self) -> Bytes { self.calls.fetch_add(1, Ordering::Relaxed); self.message.clone() } } impl_certificate_secp256r1!(CountingSubject, Vec, N3f1); let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 5); let participants = verifier.generic.participants; let mut schemes: Vec<_> = schemes .into_iter() .map(|scheme| { Scheme::signer( NAMESPACE, participants.clone(), scheme.generic.signer.unwrap().1, ) .unwrap() }) .collect(); schemes.sort_by_key(|scheme| scheme.me().unwrap().index()); let quorum = N3f1::quorum(schemes.len()) as usize; let verifier = Scheme::verifier(NAMESPACE, participants); let calls = Arc::new(AtomicUsize::new(0)); let subject = CountingSubject { message: Bytes::from_static(MESSAGE), calls: Arc::clone(&calls), }; let sign = |scheme: &Scheme, message| { scheme .sign::(CountingSubject { message, calls: Arc::clone(&calls), }) .unwrap() }; // Retained votes have already passed verification for this exact // configuration and subject before the final candidate arrives. let prior = verifier.verify_attestations::<_, Sha256Digest, _>( &mut rng, subject.clone(), schemes[..quorum - 1] .iter() .map(|scheme| sign(scheme, Bytes::from_static(MESSAGE))), &Sequential, ); assert!(prior.invalid.is_empty()); assert_eq!(prior.verified.len(), quorum - 1); // A rejected candidate must verify the new vote once and return its // signer without repeating certificate verification over retained votes. let rejected = sign( schemes.last().unwrap(), Bytes::from_static(b"other subject"), ); let invalid = rejected.signer; calls.store(0, Ordering::Relaxed); let result = verifier .optimistic_assemble::<_, Sha256Digest, _, _>( &mut rng, subject.clone(), [rejected], &prior.verified, &Sequential, ) .expect_err("invalid vote must be reported"); assert!(result.verified.is_empty()); assert_eq!(result.invalid, vec![invalid]); assert_eq!(calls.load(Ordering::Relaxed), 1); // A fresh valid replacement completes the same retained quorum. let replacement = sign(&schemes[quorum - 1], Bytes::from_static(MESSAGE)); let certificate = verifier .optimistic_assemble::<_, Sha256Digest, _, _>( &mut rng, subject.clone(), [replacement.clone()], &prior.verified, &Sequential, ) .ok() .expect("valid replacement must certify"); assert!(verifier.verify_certificate::<_, Sha256Digest>( &mut rng, subject.clone(), &certificate, &Sequential, )); // A verified quorum can assemble without preparing the subject again. let mut prior = prior.verified; prior.push(replacement); calls.store(0, Ordering::Relaxed); let certificate = verifier .optimistic_assemble::<_, Sha256Digest, _, _>( &mut rng, subject.clone(), [], &prior, &Sequential, ) .ok() .expect("verified quorum must certify"); assert_eq!(calls.load(Ordering::Relaxed), 0); assert!(verifier.verify_certificate::<_, Sha256Digest>( &mut rng, subject.clone(), &certificate, &Sequential, )); assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, CountingSubject { message: Bytes::from_static(b"other subject"), ..subject }, &certificate, &Sequential, )); } #[test] fn test_is_attributable() { assert!(Generic::>::is_attributable()); assert!(Scheme::::is_attributable()); } #[test] fn test_is_not_batchable() { assert!(!Generic::>::is_batchable()); assert!(!Scheme::::is_batchable()); } #[test] fn test_sign_vote_roundtrip() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let scheme = &schemes[0]; let attestation = scheme .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap(); assert!(scheme.verify_attestation::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &attestation, &Sequential, )); } #[test] fn test_verifier_cannot_sign() { let mut rng = test_rng(); let (_, verifier) = setup_signers(&mut rng, 4); assert!( verifier .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .is_none() ); } #[test] fn test_verify_attestations_filters_invalid() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 5); let quorum = N3f1::quorum(schemes.len()) as usize; let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let result = schemes[0].verify_attestations::<_, Sha256Digest, _>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, attestations.clone(), &Sequential, ); assert!(result.invalid.is_empty()); assert_eq!(result.verified.len(), quorum); // Corrupt one attestation's signer index. let mut attestations_corrupted = attestations.clone(); attestations_corrupted[0].signer = Participant::new(999); let result = schemes[0].verify_attestations::<_, Sha256Digest, _>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, attestations_corrupted, &Sequential, ); assert_eq!(result.invalid, vec![Participant::new(999)]); assert_eq!(result.verified.len(), quorum - 1); // Corrupt one attestation's signature. let mut attestations_corrupted = attestations; let first_signer = attestations_corrupted[0].signer; attestations_corrupted[0].signature = attestations_corrupted[1].signature.clone(); let result = schemes[0].verify_attestations::<_, Sha256Digest, _>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, attestations_corrupted, &Sequential, ); // Without batch verification, we detect exactly which signer has invalid sig assert_eq!(result.invalid, vec![first_signer]); assert_eq!(result.verified.len(), quorum - 1); } #[test] fn test_assemble_certificate() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let quorum = N3f1::quorum(schemes.len()) as usize; let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Verify certificate has correct number of signers assert_eq!(certificate.signers.count(), quorum); assert_eq!(certificate.signatures.len(), quorum); } #[test] fn test_assemble_certificate_sorts_signers() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); // Get indices and sort them to create attestations in guaranteed reverse order let mut indexed: Vec<_> = (0..3) .map(|i| (schemes[i].me().unwrap().index(), i)) .collect(); indexed.sort_by_key(|(idx, _)| *idx); // Create attestations in reverse sorted order (guaranteed non-sorted) let attestations = vec![ schemes[indexed[2].1] .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap(), schemes[indexed[1].1] .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap(), schemes[indexed[0].1] .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap(), ]; let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Verify signers are sorted by signer index let expected: Vec<_> = indexed.iter().map(|(idx, _)| *idx).collect(); assert_eq!(certificate.signers.iter().collect::>(), expected); } #[test] fn test_verify_certificate() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let quorum = N3f1::quorum(schemes.len()) as usize; let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); assert!(verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); } #[test] fn test_verify_certificate_detects_corruption() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let quorum = N3f1::quorum(schemes.len()) as usize; let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Valid certificate passes assert!(verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); // Corrupted certificate fails let mut corrupted = certificate; corrupted.signatures[0] = corrupted.signatures[1].clone(); assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &corrupted, &Sequential, )); } #[test] fn test_certificate_codec_roundtrip() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let quorum = N3f1::quorum(schemes.len()) as usize; let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); let encoded = certificate.encode(); assert!( encoded.len() <= Scheme::::max_certificate_size(u32::try_from(schemes.len()).unwrap()) ); let decoded = Certificate::decode_cfg(encoded, &schemes.len()).expect("decode certificate"); assert_eq!(decoded, certificate); } #[test] fn test_max_certificate_size() { // The bound reserves the signature-list prefix at its type's maximum width. let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 128); let attestations = schemes.iter().map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); let prefix_slack = MAX_U32_VARINT_SIZE - schemes.len().encode_size(); assert_eq!( certificate.encode().len() + prefix_slack, Scheme::::max_certificate_size(u32::try_from(schemes.len()).unwrap()) ); } #[test] fn test_certificate_rejects_sub_quorum() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let expected = N3f1::quorum(schemes.len()); let found = expected - 1; let found_count = usize::try_from(found).expect("quorum exceeds usize::MAX"); let attestations: Vec<_> = schemes .iter() .take(found_count) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); assert_eq!( schemes[0].assemble(non_empty![@attestations], &Sequential), Err(AssemblyError::InsufficientAttestations(expected, found)) ); } #[test] fn test_certificate_rejects_invalid_signer() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let quorum = usize::try_from(N3f1::quorum(schemes.len())).expect("quorum exceeds usize::MAX"); let mut attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); // Corrupt signer index to be out of range let unknown = Participant::new(999); attestations[0].signer = unknown; assert_eq!( schemes[0].assemble(non_empty![@attestations], &Sequential), Err(AssemblyError::UnknownSigner(unknown)) ); } #[test] fn test_assemble_certificate_rejects_malformed_signature() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let quorum = usize::try_from(N3f1::quorum(schemes.len())).expect("quorum exceeds usize::MAX"); let mut attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let malformed_signer = attestations[0].signer; let mut truncated = Bytes::from_static(&[0u8]); attestations[0].signature = Lazy::deferred(&mut truncated, ()); assert_eq!( schemes[0].assemble(non_empty![@attestations], &Sequential), Err(AssemblyError::MalformedSignature(malformed_signer)) ); } #[test] fn test_verify_certificate_rejects_sub_quorum() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let participants_len = schemes.len(); let attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let mut certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Artificially truncate to below quorum let mut signers: Vec = certificate.signers.iter().collect(); signers.pop(); certificate.signers = Signers::new(participants_len.try_into().unwrap(), signers).unwrap(); certificate.signatures.pop(); assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); } #[test] fn test_verify_certificate_rejects_mismatched_signature_count() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let mut certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Remove one signature but keep signers bitmap unchanged certificate.signatures.pop(); assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); } #[test] fn test_verify_certificates_batch() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let quorum = N3f1::quorum(schemes.len()) as usize; let messages: Vec = [b"msg1".as_slice(), b"msg2".as_slice(), b"msg3".as_slice()] .into_iter() .map(Bytes::copy_from_slice) .collect(); let mut certificates = Vec::new(); for msg in &messages { let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: msg.clone(), }) .unwrap() }) .collect(); certificates.push( schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(), ); } let certs_iter = messages.iter().zip(&certificates).map(|(msg, cert)| { ( TestSubject { message: msg.clone(), }, cert, ) }); assert!(verifier.verify_certificates::<_, Sha256Digest, _>( &mut rng, non_empty![@certs_iter], &Sequential )); } #[test] fn test_verify_certificates_batch_detects_failure() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let quorum = N3f1::quorum(schemes.len()) as usize; let messages: Vec = [b"msg1".as_slice(), b"msg2".as_slice()] .into_iter() .map(Bytes::copy_from_slice) .collect(); let mut certificates = Vec::new(); for msg in &messages { let attestations: Vec<_> = schemes .iter() .take(quorum) .map(|s| { s.sign::(TestSubject { message: msg.clone(), }) .unwrap() }) .collect(); certificates.push( schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(), ); } // Corrupt second certificate certificates[1].signatures[0] = certificates[1].signatures[1].clone(); let certs_iter = messages.iter().zip(&certificates).map(|(msg, cert)| { ( TestSubject { message: msg.clone(), }, cert, ) }); assert!(!verifier.verify_certificates::<_, Sha256Digest, _>( &mut rng, non_empty![@certs_iter], &Sequential )); } #[test] fn test_assemble_certificate_rejects_duplicate_signers() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let mut attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); // Add a duplicate of the last vote let duplicate = attestations.last().unwrap().clone(); let duplicate_signer = duplicate.signer; attestations.push(duplicate); assert_eq!( schemes[0].assemble(non_empty![@attestations], &Sequential), Err(AssemblyError::DuplicateSigner(duplicate_signer)) ); } #[test] fn test_scheme_clone_and_verifier() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let participants = schemes[0].generic.participants.clone(); // Clone a signer let signer = schemes[0].clone(); assert!( signer .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .is_some(), "signer should produce votes" ); // A verifier cannot produce votes let verifier = Scheme::verifier(NAMESPACE, participants); assert!( verifier .sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .is_none(), "verifier should not produce votes" ); } #[test] fn test_certificate_decode_validation() { let mut rng = test_rng(); let (schemes, _) = setup_signers(&mut rng, 4); let participants_len = schemes.len(); let participant_count = u32::try_from(participants_len).expect("participant count exceeds u32::MAX"); let attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Well-formed certificate decodes successfully let encoded = certificate.encode(); let decoded = Certificate::decode_cfg(encoded, &participants_len).expect("decode certificate"); assert_eq!(decoded, certificate); // Certificate with no signers is rejected let empty = Certificate { signers: Signers::new(participant_count, std::iter::empty::()).unwrap(), signatures: Vec::new(), }; assert!(Certificate::decode_cfg(empty.encode(), &participants_len).is_err()); // Certificate with mismatched signature count is rejected let mismatched = Certificate { signers: Signers::new( participant_count, [Participant::new(0), Participant::new(1)], ) .unwrap(), signatures: vec![certificate.signatures[0].clone()], }; assert!(Certificate::decode_cfg(mismatched.encode(), &participants_len).is_err()); // Certificate containing more signers than the participant set is rejected let mut signers = certificate.signers.iter().collect::>(); signers.push(Participant::from_usize(participants_len)); let mut sigs = certificate.signatures.clone(); sigs.push(certificate.signatures[0].clone()); let extended = Certificate { signers: Signers::new(participant_count + 1, signers).unwrap(), signatures: sigs, }; assert!(Certificate::decode_cfg(extended.encode(), &participants_len).is_err()); } #[test] fn test_verify_certificate_rejects_unknown_signer() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let participants_len = schemes.len(); let attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let mut certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Add an unknown signer (out of range) let mut signers: Vec = certificate.signers.iter().collect(); signers.push(Participant::from_usize(participants_len)); certificate.signers = Signers::new((participants_len + 1).try_into().unwrap(), signers).unwrap(); certificate .signatures .push(certificate.signatures[0].clone()); assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); } #[test] fn test_verify_certificate_rejects_invalid_certificate_signers_size() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let participants_len = schemes.len(); let attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let mut certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Valid certificate passes assert!(verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); // Make the signers bitmap size larger (mismatched with participants) let signers: Vec = certificate.signers.iter().collect(); certificate.signers = Signers::new((participants_len + 1).try_into().unwrap(), signers).unwrap(); // Certificate verification should fail due to size mismatch assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); } #[test] fn test_verify_certificate_rejects_signers_size_mismatch() { let mut rng = test_rng(); let (schemes, verifier) = setup_signers(&mut rng, 4); let participants_len = schemes.len(); let attestations: Vec<_> = schemes .iter() .take(3) .map(|s| { s.sign::(TestSubject { message: Bytes::from_static(MESSAGE), }) .unwrap() }) .collect(); let mut certificate = schemes[0] .assemble(non_empty![@attestations], &Sequential) .unwrap(); // Make the signers bitmap size larger than participants let signers: Vec = certificate.signers.iter().collect(); certificate.signers = Signers::new((participants_len + 1).try_into().unwrap(), signers).unwrap(); certificate .signatures .push(certificate.signatures[0].clone()); assert!(!verifier.verify_certificate::<_, Sha256Digest>( &mut rng, TestSubject { message: Bytes::from_static(MESSAGE), }, &certificate, &Sequential, )); } #[cfg(feature = "arbitrary")] mod conformance { use super::*; use commonware_codec::conformance::CodecConformance; commonware_conformance::conformance_tests! { CodecConformance => 1024, } } }