bolt-proto 0.12.0

Bolt protocol primitives: values, messages, and serialization / deserialization.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use std::panic::UnwindSafe;

use bytes::{Buf, Bytes};

use crate::{
    error::{DeserializationError, DeserializeResult, SerializeResult},
    value::{MARKER_MEDIUM_STRUCT, MARKER_SMALL_STRUCT, MARKER_TINY_STRUCT},
};

pub(crate) trait BoltValue: Sized {
    fn marker(&self) -> SerializeResult<u8>;

    fn serialize(self) -> SerializeResult<Bytes>;

    fn deserialize<B: Buf + UnwindSafe>(bytes: B) -> DeserializeResult<(Self, B)>;
}

pub(crate) trait BoltStructure: BoltValue {
    fn signature(&self) -> u8;
}

/// Returns size and signature. Might panic - use this inside a catch_unwind block
pub(crate) fn get_structure_info(
    marker: u8,
    bytes: &mut impl Buf,
) -> DeserializeResult<(usize, u8)> {
    let size = match marker {
        marker if (MARKER_TINY_STRUCT..=(MARKER_TINY_STRUCT | 0x0F)).contains(&marker) => {
            0x0F & marker as usize
        }
        MARKER_SMALL_STRUCT => bytes.get_u8() as usize,
        MARKER_MEDIUM_STRUCT => bytes.get_u16() as usize,
        _ => return Err(DeserializationError::InvalidMarkerByte(marker)),
    };
    let signature = bytes.get_u8();
    Ok((size, signature))
}