diff options
Diffstat (limited to 'src/transport/wire.rs')
| -rw-r--r-- | src/transport/wire.rs | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/src/transport/wire.rs b/src/transport/wire.rs index 97f8819..88b2916 100644 --- a/src/transport/wire.rs +++ b/src/transport/wire.rs @@ -14,7 +14,7 @@ use super::{EncodedMessage, MessageKind, ParsedMessage}; /// The wire format's version, the first byte of every payload. Bump it with every change /// to a message type or a codec (`golden_bytes` fails then). -pub(crate) const VERSION: u8 = 40; +pub(crate) const VERSION: u8 = 41; /// What the second byte of a payload says. fn tag(kind: MessageKind) -> u8 { @@ -36,6 +36,7 @@ fn tag(kind: MessageKind) -> u8 { MessageKind::Neighbour => 16, MessageKind::NeighbourReply => 17, MessageKind::Disconnect => 18, + MessageKind::Signatures => 19, } } @@ -97,6 +98,7 @@ pub(crate) fn encode(message: &ParsedMessage) -> Result<Vec<u8>, WireError> { ParsedMessage::NeighbourReply(answer) | ParsedMessage::Disconnect(answer) => { rmp_serde::encode::write(&mut payload, answer)? } + ParsedMessage::Signatures(message) => rmp_serde::encode::write(&mut payload, message)?, } Ok(payload) } @@ -176,6 +178,8 @@ pub(crate) fn decode(payload: &[u8]) -> Result<ParsedMessage, WireError> { ParsedMessage::NeighbourReply(rmp_serde::from_slice(body)?) } else if is(MessageKind::Disconnect) { ParsedMessage::Disconnect(rmp_serde::from_slice(body)?) + } else if is(MessageKind::Signatures) { + ParsedMessage::Signatures(rmp_serde::from_slice(body)?) } else { return Err(WireError::UnknownKind(kind)); }) @@ -535,6 +539,11 @@ pub(crate) fn samples() -> Vec<ParsedMessage> { epoch_seq: 1, agreed: false, }), + ParsedMessage::Signatures(crate::transport::SignaturesMessage { + epoch, + epoch_seq: 1, + signatures: serde_bytes::ByteBuf::from(vec![9; 96]), + }), ] } @@ -566,7 +575,7 @@ mod tests { #[test] fn every_kind_round_trips() { let samples = samples(); - assert_eq!(samples.len(), 17, "a kind is missing from `samples`"); + assert_eq!(samples.len(), 18, "a kind is missing from `samples`"); for message in samples { let bytes = encode(&message).unwrap(); let decoded = decode(&bytes).unwrap(); |
