cs_mwc_bch/messages/
send_cmpct.rs

1use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
2use messages::message::Payload;
3use std::io;
4use std::io::{Read, Write};
5use util::{Result, Serializable};
6
7/// Specifies whether compact blocks are supported
8#[derive(Debug, Default, PartialEq, Eq, Hash, Clone)]
9pub struct SendCmpct {
10    /// Whether compact blocks may be sent
11    pub enable: u8,
12    /// Should always be 1
13    pub version: u64,
14}
15
16impl SendCmpct {
17    /// Size of the SendCmpct payload in bytes
18    pub const SIZE: usize = 9;
19
20    /// Returns whether compact blocks should be used
21    pub fn use_cmpctblock(&self) -> bool {
22        self.enable == 1 && self.version == 1
23    }
24}
25
26impl Serializable<SendCmpct> for SendCmpct {
27    fn read(reader: &mut dyn Read) -> Result<SendCmpct> {
28        let enable = reader.read_u8()?;
29        let version = reader.read_u64::<LittleEndian>()?;
30        Ok(SendCmpct { enable, version })
31    }
32
33    fn write(&self, writer: &mut dyn Write) -> io::Result<()> {
34        writer.write_u8(self.enable)?;
35        writer.write_u64::<LittleEndian>(self.version)
36    }
37}
38
39impl Payload<SendCmpct> for SendCmpct {
40    fn size(&self) -> usize {
41        SendCmpct::SIZE
42    }
43}
44
45#[cfg(test)]
46mod tests {
47    use super::*;
48    use hex;
49    use std::io::Cursor;
50
51    #[test]
52    fn read_bytes() {
53        let b = hex::decode("000100000000000000".as_bytes()).unwrap();
54        let f = SendCmpct::read(&mut Cursor::new(&b)).unwrap();
55        assert!(f.enable == 0);
56        assert!(f.version == 1);
57    }
58
59    #[test]
60    fn write_read() {
61        let mut v = Vec::new();
62        let s = SendCmpct {
63            enable: 1,
64            version: 1,
65        };
66        s.write(&mut v).unwrap();
67        assert!(v.len() == s.size());
68        assert!(SendCmpct::read(&mut Cursor::new(&v)).unwrap() == s);
69    }
70}
cs_mwc_bch/messages/send_cmpct.rs

cs_mwc_bch/messages/
send_cmpct.rs
