1use serde::{Deserialize, Serialize, Serializer, ser::SerializeTuple};
2
3use crate::{MembershipDigest, RegisteredParty, StmResult, membership_commitment::MerkleTreeLeaf};
4
5use super::{SignatureError, SingleSignature};
6
7#[derive(Debug, Clone, Hash, Deserialize, Eq, PartialEq, Ord, PartialOrd)]
9pub struct SingleSignatureWithRegisteredParty {
10 pub sig: SingleSignature,
12 pub reg_party: RegisteredParty,
14}
15
16impl SingleSignatureWithRegisteredParty {
17 pub fn to_bytes(&self) -> Vec<u8> {
22 let mut out = Vec::new();
23 out.extend_from_slice(&self.reg_party.as_bytes_for_merkle_tree());
24 out.extend_from_slice(&self.sig.to_bytes());
25
26 out
27 }
28 pub fn from_bytes<D: MembershipDigest>(
30 bytes: &[u8],
31 ) -> StmResult<SingleSignatureWithRegisteredParty> {
32 let reg_party = RegisteredParty::from_bytes(
33 bytes.get(0..104).ok_or(SignatureError::SerializationError)?,
34 )?;
35 let sig = SingleSignature::from_bytes::<D>(
36 bytes.get(104..).ok_or(SignatureError::SerializationError)?,
37 )?;
38
39 Ok(SingleSignatureWithRegisteredParty { sig, reg_party })
40 }
41}
42
43impl Serialize for SingleSignatureWithRegisteredParty {
44 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
45 where
46 S: Serializer,
47 {
48 let mut tuple = serializer.serialize_tuple(2)?;
49 tuple.serialize_element(&self.sig)?;
50 tuple.serialize_element(&self.reg_party)?;
51 tuple.end()
52 }
53}
54
55#[cfg(test)]
56mod tests {
57 mod golden {
58 use rand_chacha::ChaCha20Rng;
59 use rand_core::SeedableRng;
60
61 use crate::{
62 ClosedKeyRegistration, KeyRegistration, MithrilMembershipDigest, Parameters, Signer,
63 SingleSignatureWithRegisteredParty,
64 signature_scheme::{BlsSigningKey, BlsVerificationKeyProofOfPossession},
65 };
66
67 const GOLDEN_JSON: &str = r#"
68 [
69 {
70 "sigma": [
71 149, 157, 201, 187, 140, 54, 0, 128, 209, 88, 16, 203, 61, 78, 77, 98,
72 161, 133, 58, 152, 29, 74, 217, 113, 64, 100, 10, 161, 186, 167, 133, 114,
73 211, 153, 218, 56, 223, 84, 105, 242, 41, 54, 224, 170, 208, 185, 126, 83
74 ],
75 "indexes": [1, 4, 5, 8],
76 "signer_index": 1
77 },
78 [
79 [
80 143, 161, 255, 48, 78, 57, 204, 220, 25, 221, 164, 252, 248, 14, 56, 126,
81 186, 135, 228, 188, 145, 181, 52, 200, 97, 99, 213, 46, 0, 199, 193, 89,
82 187, 88, 29, 135, 173, 244, 86, 36, 83, 54, 67, 164, 6, 137, 94, 72, 6,
83 105, 128, 128, 93, 48, 176, 11, 4, 246, 138, 48, 180, 133, 90, 142, 192,
84 24, 193, 111, 142, 31, 76, 111, 110, 234, 153, 90, 208, 192, 31, 124, 95,
85 102, 49, 158, 99, 52, 220, 165, 94, 251, 68, 69, 121, 16, 224, 194
86 ],
87 1
88 ]
89 ]
90 "#;
91
92 fn golden_value() -> SingleSignatureWithRegisteredParty {
93 let mut rng = ChaCha20Rng::from_seed([0u8; 32]);
94 let msg = [0u8; 16];
95 let params = Parameters {
96 m: 10,
97 k: 5,
98 phi_f: 0.8,
99 };
100 let sk_1 = BlsSigningKey::generate(&mut rng);
101 let sk_2 = BlsSigningKey::generate(&mut rng);
102 let pk_1 = BlsVerificationKeyProofOfPossession::from(&sk_1);
103 let pk_2 = BlsVerificationKeyProofOfPossession::from(&sk_2);
104 let mut key_reg = KeyRegistration::init();
105 key_reg.register(1, pk_1).unwrap();
106 key_reg.register(1, pk_2).unwrap();
107 let closed_key_reg: ClosedKeyRegistration<MithrilMembershipDigest> = key_reg.close();
108 let signer = Signer::set_signer(1, 1, params, sk_1, pk_1.vk, closed_key_reg.clone());
109 let signature = signer.sign(&msg).unwrap();
110 SingleSignatureWithRegisteredParty {
111 sig: signature,
112 reg_party: closed_key_reg.reg_parties[0],
113 }
114 }
115
116 #[test]
117 fn golden_conversions() {
118 let value = serde_json::from_str(GOLDEN_JSON)
119 .expect("This JSON deserialization should not fail");
120 assert_eq!(golden_value(), value);
121
122 let serialized =
123 serde_json::to_string(&value).expect("This JSON serialization should not fail");
124 let golden_serialized = serde_json::to_string(&golden_value())
125 .expect("This JSON serialization should not fail");
126 assert_eq!(golden_serialized, serialized);
127 }
128 }
129}