use std::fmt;
use std::marker::PhantomData;
use serde::de::{Error as SerdeError, SeqAccess, Visitor};
use serde::ser::SerializeSeq;
use serde::{Deserialize, Serialize};
use serde_bytes::{ByteBuf as SerdeByteBuf, Bytes as SerdeBytes};
use crate::cursor::Cursor;
use crate::hash::{Hash, HashError};
use crate::identity::{Author, IdentityError, Signature, SigningKey, VerifyingKey};
use crate::logs::LogHeights;
use crate::operation::{Body, Header};
use crate::timestamp::Timestamp;
use crate::topic::TopicError;
use crate::{LogId, Topic};
pub fn serialize_hex<S>(value: &[u8], serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
if serializer.is_human_readable() {
hex::serde::serialize(value, serializer)
} else {
SerdeBytes::new(value).serialize(serializer)
}
}
pub fn deserialize_hex<'de, D>(deserializer: D) -> Result<Vec<u8>, D::Error>
where
D: serde::Deserializer<'de>,
{
if deserializer.is_human_readable() {
hex::serde::deserialize(deserializer)
} else {
let bytes = <SerdeByteBuf>::deserialize(deserializer)?;
Ok(bytes.to_vec())
}
}
impl Serialize for Hash {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serialize_hex(self.as_bytes(), serializer)
}
}
impl<'de> Deserialize<'de> for Hash {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let bytes = deserialize_hex(deserializer)?;
bytes
.as_slice()
.try_into()
.map_err(|err: HashError| serde::de::Error::custom(err.to_string()))
}
}
impl Serialize for SigningKey {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serialize_hex(self.as_bytes(), serializer)
}
}
impl<'de> Deserialize<'de> for SigningKey {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let bytes = deserialize_hex(deserializer)?;
bytes
.as_slice()
.try_into()
.map_err(|err: IdentityError| serde::de::Error::custom(err.to_string()))
}
}
impl Serialize for VerifyingKey {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serialize_hex(self.as_bytes(), serializer)
}
}
impl<'de> Deserialize<'de> for VerifyingKey {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let bytes = deserialize_hex(deserializer)?;
bytes
.as_slice()
.try_into()
.map_err(|err: IdentityError| serde::de::Error::custom(err.to_string()))
}
}
impl Serialize for Signature {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serialize_hex(&self.to_bytes(), serializer)
}
}
impl<'de> Deserialize<'de> for Signature {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let bytes = deserialize_hex(deserializer)?;
bytes
.as_slice()
.try_into()
.map_err(|err: IdentityError| serde::de::Error::custom(err.to_string()))
}
}
impl<E> Serialize for Header<E>
where
E: Serialize,
{
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
let mut seq = serializer.serialize_seq(Some(self.field_count()))?;
seq.serialize_element(&self.version)?;
seq.serialize_element(&self.verifying_key)?;
if let Some(signature) = &self.signature {
seq.serialize_element(signature)?;
}
seq.serialize_element(&self.payload_size)?;
if let Some(hash) = &self.payload_hash {
seq.serialize_element(&hash)?;
}
seq.serialize_element(&self.timestamp)?;
seq.serialize_element(&self.seq_num)?;
if let Some(backlink) = &self.backlink {
seq.serialize_element(backlink)?;
}
seq.serialize_element(&self.extensions)?;
seq.end()
}
}
impl<'de, E> Deserialize<'de> for Header<E>
where
E: Deserialize<'de>,
{
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
struct HeaderVisitor<E> {
_marker: PhantomData<E>,
}
impl<'de, E> Visitor<'de> for HeaderVisitor<E>
where
E: Deserialize<'de>,
{
type Value = Header<E>;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("Header encoded as a sequence")
}
fn visit_seq<A>(self, mut seq: A) -> Result<Self::Value, A::Error>
where
A: SeqAccess<'de>,
{
let version: u64 = seq
.next_element()?
.ok_or(SerdeError::custom("version missing"))?;
let verifying_key: VerifyingKey = seq
.next_element()?
.ok_or(SerdeError::custom("public key missing"))?;
let signature: Signature = seq
.next_element()?
.ok_or(SerdeError::custom("signature missing"))?;
let payload_size: u64 = seq
.next_element()?
.ok_or(SerdeError::custom("payload size missing"))?;
let payload_hash: Option<Hash> = match payload_size {
0 => None,
_ => {
let hash: Hash = seq
.next_element()?
.ok_or(SerdeError::custom("payload hash missing"))?;
Some(hash)
}
};
let timestamp: Timestamp = seq
.next_element()?
.ok_or(SerdeError::custom("timestamp missing"))?;
let seq_num: u64 = seq
.next_element()?
.ok_or(SerdeError::custom("sequence number missing"))?;
let backlink: Option<Hash> = match seq_num {
0 => None,
_ => {
let hash: Hash = seq
.next_element()?
.ok_or(SerdeError::custom("backlink missing"))?;
Some(hash)
}
};
let extensions: E = seq
.next_element()?
.ok_or(SerdeError::custom("extensions missing"))?;
if let Some(remainder) = seq.size_hint()
&& remainder > 0
{
return Err(SerdeError::custom("unexpected excessive fields in header"));
}
Ok(Header {
version,
verifying_key,
signature: Some(signature),
payload_hash,
payload_size,
timestamp,
seq_num,
backlink,
extensions,
})
}
}
deserializer.deserialize_seq(HeaderVisitor::<E> {
_marker: PhantomData,
})
}
}
impl Serialize for Body {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serialize_hex(&self.0, serializer)
}
}
impl<'de> Deserialize<'de> for Body {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let bytes = deserialize_hex(deserializer)?;
Ok(Body(bytes.to_vec()))
}
}
impl Serialize for Topic {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serialize_hex(&self.0, serializer)
}
}
impl<'de> Deserialize<'de> for Topic {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let bytes = deserialize_hex(deserializer)?;
bytes
.as_slice()
.try_into()
.map_err(|err: TopicError| serde::de::Error::custom(err.to_string()))
}
}
impl<A, L> Serialize for Cursor<A, L>
where
A: Author,
L: LogId,
{
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
let mut seq = serializer.serialize_seq(Some(2))?;
seq.serialize_element(self.name())?;
seq.serialize_element(self.state())?;
seq.end()
}
}
impl<'de, A, L> Deserialize<'de> for Cursor<A, L>
where
A: Author,
L: LogId,
{
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
struct CursorVisitor<A, L> {
_marker: PhantomData<(A, L)>,
}
impl<'de, A, L> Visitor<'de> for CursorVisitor<A, L>
where
A: Author,
L: LogId,
{
type Value = Cursor<A, L>;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("Cursor encoded as a sequence")
}
fn visit_seq<T>(self, mut seq: T) -> Result<Self::Value, T::Error>
where
T: SeqAccess<'de>,
{
let name: String = seq
.next_element()?
.ok_or(SerdeError::custom("cursor id missing"))?;
let state: LogHeights<A, L> = seq
.next_element()?
.ok_or(SerdeError::custom("state vector missing"))?;
Ok(Cursor::new(name, state))
}
}
deserializer.deserialize_seq(CursorVisitor::<A, L> {
_marker: PhantomData,
})
}
}
#[cfg(test)]
mod tests {
use serde::de::DeserializeOwned;
use serde::{Deserialize, Serialize};
use crate::Body;
use crate::hash::Hash;
use crate::identity::{SigningKey, VerifyingKey};
use crate::operation::Header;
use super::{deserialize_hex, serialize_hex};
#[derive(Debug, Serialize, Deserialize)]
struct Test(
#[serde(serialize_with = "serialize_hex", deserialize_with = "deserialize_hex")] Vec<u8>,
);
#[test]
fn serialize() {
let mut bytes: Vec<u8> = Vec::new();
let test = Test(vec![1, 2, 3]);
ciborium::ser::into_writer(&test, &mut bytes).unwrap();
assert_eq!(vec![67, 1, 2, 3], bytes);
}
#[test]
fn deserialize() {
let bytes: Vec<u8> = vec![67, 1, 2, 3];
let test: Test = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(test.0, vec![1, 2, 3]);
}
#[test]
fn serialize_hash() {
let mut bytes: Vec<u8> = Vec::new();
let hash = Hash::digest([1, 2, 3]);
ciborium::ser::into_writer(&hash, &mut bytes).unwrap();
assert_eq!(
bytes,
vec![
88, 32, 177, 119, 236, 27, 242, 109, 251, 59, 112, 16, 212, 115, 230, 212, 71, 19,
178, 155, 118, 91, 153, 198, 230, 14, 203, 250, 231, 66, 222, 73, 101, 67
]
);
let json = serde_json::to_string(&hash).unwrap();
assert_eq!(
json,
"\"b177ec1bf26dfb3b7010d473e6d44713b29b765b99c6e60ecbfae742de496543\""
);
}
#[test]
fn deserialize_hash() {
let bytes = [
88, 32, 177, 119, 236, 27, 242, 109, 251, 59, 112, 16, 212, 115, 230, 212, 71, 19, 178,
155, 118, 91, 153, 198, 230, 14, 203, 250, 231, 66, 222, 73, 101, 67,
];
let hash: Hash = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(hash, Hash::digest([1, 2, 3]));
let json = "\"b177ec1bf26dfb3b7010d473e6d44713b29b765b99c6e60ecbfae742de496543\"";
let hash: Hash = serde_json::from_str(json).unwrap();
assert_eq!(hash, Hash::digest([1, 2, 3]));
}
#[test]
fn serialize_verifying_key() {
let mut bytes: Vec<u8> = Vec::new();
let verifying_key = VerifyingKey::from_bytes(&[
215, 90, 152, 1, 130, 177, 10, 183, 213, 75, 254, 211, 201, 100, 7, 58, 14, 225, 114,
243, 218, 166, 35, 37, 175, 2, 26, 104, 247, 7, 81, 26,
])
.unwrap();
ciborium::ser::into_writer(&verifying_key, &mut bytes).unwrap();
assert_eq!(
bytes,
vec![
88, 32, 215, 90, 152, 1, 130, 177, 10, 183, 213, 75, 254, 211, 201, 100, 7, 58, 14,
225, 114, 243, 218, 166, 35, 37, 175, 2, 26, 104, 247, 7, 81, 26,
]
);
let json = serde_json::to_string(&verifying_key).unwrap();
assert_eq!(
json,
"\"d75a980182b10ab7d54bfed3c964073a0ee172f3daa62325af021a68f707511a\""
);
}
fn assert_serde_roundtrip<
E: Clone + std::fmt::Debug + PartialEq + Serialize + DeserializeOwned,
>(
mut header: Header<E>,
signing_key: &SigningKey,
) {
header.sign(signing_key);
let mut bytes = Vec::new();
ciborium::ser::into_writer(&header, &mut bytes).unwrap();
let header_again: Header<E> = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(header, header_again);
}
#[test]
fn serde_roundtrip_operations() {
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
struct CustomExtensions {
custom_field: u64,
}
let extensions = CustomExtensions { custom_field: 12 };
let signing_key = SigningKey::generate();
assert_serde_roundtrip(
Header::<CustomExtensions> {
version: 1,
verifying_key: signing_key.verifying_key(),
payload_size: 123,
payload_hash: Some(Hash::digest(vec![1, 2, 3])),
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: extensions.clone(),
signature: None,
},
&signing_key,
);
assert_serde_roundtrip(
Header::<CustomExtensions> {
version: 1,
verifying_key: signing_key.verifying_key(),
payload_size: 0,
payload_hash: None,
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: extensions,
signature: None,
},
&signing_key,
);
}
#[test]
fn expected_de_error() {
let signing_key = SigningKey::generate();
let mut header = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 2829099,
payload_hash: None,
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: (),
};
header.sign(&signing_key);
let result = ciborium::de::from_reader::<Header<()>, _>(&header.to_bytes()[..]);
assert!(result.is_err());
let mut header = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 0,
payload_hash: Some(Hash::digest([0, 1, 2])),
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: (),
};
header.sign(&signing_key);
let result = ciborium::de::from_reader::<Header<()>, _>(&header.to_bytes()[..]);
assert!(result.is_err());
let mut header = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 0,
payload_hash: None,
timestamp: 0.into(),
seq_num: 0,
backlink: Some(Hash::digest([0, 1, 2])),
extensions: (),
};
header.sign(&signing_key);
let result = ciborium::de::from_reader::<Header<()>, _>(&header.to_bytes()[..]);
assert!(result.is_err());
let mut header = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 0,
payload_hash: None,
timestamp: 0.into(),
seq_num: 10,
backlink: None,
extensions: (),
};
header.sign(&signing_key);
let result = ciborium::de::from_reader::<Header<()>, _>(&header.to_bytes()[..]);
assert!(result.is_err());
}
#[test]
fn serde_header_with_other_types() {
let signing_key = SigningKey::generate();
#[derive(Debug, PartialEq, Serialize, Deserialize)]
struct Message {
header: Header<()>,
body: Body,
}
let body = Body::new(b"hello");
let mut header = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: body.size(),
payload_hash: Some(body.hash()),
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: (),
};
header.sign(&signing_key);
let message = Message { header, body };
let mut bytes = Vec::new();
ciborium::ser::into_writer(&message, &mut bytes).unwrap();
let message_again: Message = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(message_again, message);
}
#[test]
fn fixtures() {
let signing_key = SigningKey::from_bytes(&[
244, 123, 85, 215, 161, 204, 94, 227, 239, 253, 128, 164, 228, 160, 195, 49, 18, 49,
125, 4, 50, 218, 157, 230, 174, 1, 154, 231, 231, 142, 22, 170,
]);
let mut header_0 = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 0,
payload_hash: None,
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: (),
};
header_0.sign(&signing_key);
let bytes = [
135, 1, 88, 32, 228, 21, 196, 25, 12, 199, 241, 100, 122, 89, 46, 191, 142, 95, 144,
92, 42, 222, 249, 148, 139, 23, 91, 43, 92, 17, 225, 69, 17, 181, 22, 32, 88, 64, 42,
11, 253, 219, 220, 200, 239, 31, 142, 159, 42, 215, 225, 66, 212, 199, 224, 81, 213,
150, 90, 253, 202, 2, 201, 94, 12, 1, 167, 36, 158, 173, 165, 8, 136, 9, 73, 19, 163,
174, 10, 96, 73, 198, 119, 18, 195, 129, 13, 114, 121, 16, 81, 155, 179, 182, 112, 123,
160, 63, 147, 206, 219, 7, 0, 0, 0, 246,
];
let header_again: Header<()> = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(header_0, header_again);
let body = Body::new("Hello, Sloth!".as_bytes());
let mut header_0_with_body = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: body.size(),
payload_hash: Some(body.hash()),
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: (),
};
header_0_with_body.sign(&signing_key);
let bytes = [
136, 1, 88, 32, 228, 21, 196, 25, 12, 199, 241, 100, 122, 89, 46, 191, 142, 95, 144,
92, 42, 222, 249, 148, 139, 23, 91, 43, 92, 17, 225, 69, 17, 181, 22, 32, 88, 64, 27,
199, 136, 138, 253, 125, 87, 20, 50, 247, 40, 93, 111, 176, 15, 63, 216, 239, 129, 134,
100, 162, 66, 133, 249, 181, 26, 79, 119, 128, 192, 86, 237, 32, 146, 71, 175, 180,
118, 146, 240, 172, 149, 99, 246, 177, 182, 110, 84, 49, 220, 60, 65, 70, 206, 79, 92,
237, 4, 43, 41, 202, 94, 10, 13, 88, 32, 191, 127, 68, 13, 227, 43, 252, 155, 49, 148,
176, 2, 162, 217, 175, 171, 49, 44, 181, 215, 71, 113, 211, 195, 29, 128, 192, 169, 5,
138, 160, 142, 0, 0, 246,
];
let header_again: Header<()> = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(header_0_with_body, header_again);
let mut header_1 = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 0,
payload_hash: None,
timestamp: 0.into(),
seq_num: 1,
backlink: Some(header_0.hash()),
extensions: (),
};
header_1.sign(&signing_key);
let bytes = [
136, 1, 88, 32, 228, 21, 196, 25, 12, 199, 241, 100, 122, 89, 46, 191, 142, 95, 144,
92, 42, 222, 249, 148, 139, 23, 91, 43, 92, 17, 225, 69, 17, 181, 22, 32, 88, 64, 198,
228, 65, 150, 97, 52, 1, 243, 222, 62, 11, 115, 104, 187, 64, 118, 179, 178, 190, 109,
15, 67, 36, 224, 172, 166, 199, 117, 59, 92, 164, 141, 190, 191, 151, 194, 193, 241,
115, 149, 98, 43, 39, 113, 255, 105, 24, 154, 136, 110, 250, 84, 159, 127, 192, 17,
240, 82, 84, 223, 41, 29, 150, 7, 0, 0, 1, 88, 32, 140, 19, 67, 147, 71, 66, 239, 37,
103, 212, 94, 71, 203, 133, 40, 89, 241, 1, 120, 215, 147, 204, 180, 108, 5, 39, 2,
178, 190, 77, 146, 144, 246,
];
let header_again: Header<()> = ciborium::de::from_reader(&bytes[..]).unwrap();
assert_eq!(header_1, header_again);
}
#[test]
fn decode_non_map_extensions() {
let signing_key = SigningKey::generate();
let mut header = Header::<()> {
version: 1,
verifying_key: signing_key.verifying_key(),
signature: None,
payload_size: 0,
payload_hash: None,
timestamp: 0.into(),
seq_num: 0,
backlink: None,
extensions: (),
};
header.sign(&signing_key);
let result = ciborium::de::from_reader::<Header<()>, _>(&header.to_bytes()[..]);
assert!(result.is_ok());
}
#[test]
fn unexpected_eof_when_incomplete() {
let incomplete = [
137, 1, 88, 32, 228, 21, 196, 25, 12, 199, 241, 100, 122, 89, 46, 191, 142, 95, 144,
];
let result: Result<Header<()>, _> = ciborium::de::from_reader(&incomplete[..]);
assert!(matches!(result, Err(ciborium::de::Error::Io(_))));
}
}