132 lines
3.4 KiB
Rust
132 lines
3.4 KiB
Rust
use super::{
|
|
fake_aggregate_public_key::FakeAggregatePublicKey, fake_signature::FakeSignature,
|
|
BLS_AGG_SIG_BYTE_SIZE,
|
|
};
|
|
use cached_tree_hash::cached_tree_hash_ssz_encoding_as_vector;
|
|
use serde::de::{Deserialize, Deserializer};
|
|
use serde::ser::{Serialize, Serializer};
|
|
use serde_hex::{encode as hex_encode, PrefixedHexVisitor};
|
|
use ssz::{ssz_encode, Decode, DecodeError};
|
|
use tree_hash::tree_hash_ssz_encoding_as_vector;
|
|
|
|
/// A BLS aggregate signature.
|
|
///
|
|
/// This struct is a wrapper upon a base type and provides helper functions (e.g., SSZ
|
|
/// serialization).
|
|
#[derive(Debug, PartialEq, Clone, Default, Eq)]
|
|
pub struct FakeAggregateSignature {
|
|
bytes: Vec<u8>,
|
|
}
|
|
|
|
impl FakeAggregateSignature {
|
|
/// Creates a new all-zero's signature
|
|
pub fn new() -> Self {
|
|
Self::zero()
|
|
}
|
|
|
|
/// Creates a new all-zero's signature
|
|
pub fn zero() -> Self {
|
|
Self {
|
|
bytes: vec![0; BLS_AGG_SIG_BYTE_SIZE],
|
|
}
|
|
}
|
|
|
|
/// Does glorious nothing.
|
|
pub fn add(&mut self, _signature: &FakeSignature) {
|
|
// Do nothing.
|
|
}
|
|
|
|
/// Does glorious nothing.
|
|
pub fn add_aggregate(&mut self, _agg_sig: &FakeAggregateSignature) {
|
|
// Do nothing.
|
|
}
|
|
|
|
/// _Always_ returns `true`.
|
|
pub fn verify(
|
|
&self,
|
|
_msg: &[u8],
|
|
_domain: u64,
|
|
_aggregate_public_key: &FakeAggregatePublicKey,
|
|
) -> bool {
|
|
true
|
|
}
|
|
|
|
/// _Always_ returns `true`.
|
|
pub fn verify_multiple(
|
|
&self,
|
|
_messages: &[&[u8]],
|
|
_domain: u64,
|
|
_aggregate_public_keys: &[&FakeAggregatePublicKey],
|
|
) -> bool {
|
|
true
|
|
}
|
|
|
|
/// Convert bytes to fake BLS aggregate signature
|
|
pub fn from_bytes(bytes: &[u8]) -> Result<Self, DecodeError> {
|
|
if bytes.len() != BLS_AGG_SIG_BYTE_SIZE {
|
|
Err(DecodeError::InvalidByteLength {
|
|
len: bytes.len(),
|
|
expected: BLS_AGG_SIG_BYTE_SIZE,
|
|
})
|
|
} else {
|
|
Ok(Self {
|
|
bytes: bytes.to_vec(),
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn as_bytes(&self) -> Vec<u8> {
|
|
self.bytes.clone()
|
|
}
|
|
}
|
|
|
|
impl_ssz!(
|
|
FakeAggregateSignature,
|
|
BLS_AGG_SIG_BYTE_SIZE,
|
|
"FakeAggregateSignature"
|
|
);
|
|
|
|
impl Serialize for FakeAggregateSignature {
|
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
|
where
|
|
S: Serializer,
|
|
{
|
|
serializer.serialize_str(&hex_encode(ssz_encode(self)))
|
|
}
|
|
}
|
|
|
|
impl<'de> Deserialize<'de> for FakeAggregateSignature {
|
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
|
where
|
|
D: Deserializer<'de>,
|
|
{
|
|
let bytes = deserializer.deserialize_str(PrefixedHexVisitor)?;
|
|
let obj = <_>::from_ssz_bytes(&bytes[..])
|
|
.map_err(|e| serde::de::Error::custom(format!("invalid ssz ({:?})", e)))?;
|
|
Ok(obj)
|
|
}
|
|
}
|
|
|
|
tree_hash_ssz_encoding_as_vector!(FakeAggregateSignature);
|
|
cached_tree_hash_ssz_encoding_as_vector!(FakeAggregateSignature, 96);
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::super::{Keypair, Signature};
|
|
use super::*;
|
|
use ssz::ssz_encode;
|
|
|
|
#[test]
|
|
pub fn test_ssz_round_trip() {
|
|
let keypair = Keypair::random();
|
|
|
|
let mut original = FakeAggregateSignature::new();
|
|
original.add(&Signature::new(&[42, 42], 0, &keypair.sk));
|
|
|
|
let bytes = ssz_encode(&original);
|
|
let decoded = FakeAggregateSignature::from_ssz_bytes(&bytes).unwrap();
|
|
|
|
assert_eq!(original, decoded);
|
|
}
|
|
}
|