bitcoin-serialize 0.1.19

tools for serialization
// ---------------- [ File: bitcoin-serialize/src/read_write_data.rs ]
crate::ix!();

/**
  | Convert the reference base type to X,
  | without changing constness or reference
  | type.
  |
  */
pub fn read_write_as_helper<X>(x: &mut X) -> &mut X {
    x
}

/**
  | Lowest-level serialization and conversion.
  | 
  | -----------
  | @note
  | 
  | Sizes of these types are verified in
  | the tests
  |
  */
#[inline]
pub fn ser_writedata8<Stream: Write>(s: &mut Stream, obj: u8) {
    trace!(value = obj, "ser_writedata8");
    s.write_all(&[obj])
        .expect("I/O error while writing u8");
}

#[inline]
pub fn ser_writedata16<Stream: Write>(s: &mut Stream, obj: u16) {
    trace!(value = obj, "ser_writedata16 (LE)");
    s.write_all(&obj.to_le_bytes())
        .expect("I/O error while writing u16");
}

#[inline]
pub fn ser_writedata_16be<Stream: Write>(s: &mut Stream, obj: u16) {
    trace!(value = obj, "ser_writedata_16be (BE)");
    s.write_all(&obj.to_be_bytes())
        .expect("I/O error while writing u16‑be");
}

#[inline]
pub fn ser_writedata32<Stream: Write>(s: &mut Stream, obj: u32) {
    trace!(value = obj, "ser_writedata32 (LE)");
    s.write_all(&obj.to_le_bytes())
        .expect("I/O error while writing u32");
}

#[inline]
pub fn ser_writedata_32be<Stream: Write>(s: &mut Stream, obj: u32) {
    trace!(value = obj, "ser_writedata_32be (BE)");
    s.write_all(&obj.to_be_bytes())
        .expect("I/O error while writing u32‑be");
}

#[inline]
pub fn ser_writedata64<Stream: Write>(s: &mut Stream, obj: u64) {
    trace!(value = obj, "ser_writedata64 (LE)");
    s.write_all(&obj.to_le_bytes())
        .expect("I/O error while writing u64");
}

#[inline]
pub fn ser_readdata8<Stream: Read>(s: &mut Stream) -> u8 {
    let mut buf = [0u8; 1];
    s.read_exact(&mut buf).expect("I/O error while reading u8");
    let v = buf[0];
    trace!(value = v, "ser_readdata8");
    v
}

#[inline]
pub fn ser_readdata16<Stream: Read>(s: &mut Stream) -> u16 {
    let mut buf = [0u8; 2];
    s.read_exact(&mut buf).expect("I/O error while reading u16");
    let v = u16::from_le_bytes(buf);
    trace!(value = v, "ser_readdata16 (LE)");
    v
}

#[inline]
pub fn ser_readdata_16be<Stream: Read>(s: &mut Stream) -> u16 {
    let mut buf = [0u8; 2];
    s.read_exact(&mut buf).expect("I/O error while reading u16‑be");
    let v = u16::from_be_bytes(buf);
    trace!(value = v, "ser_readdata_16be (BE)");
    v
}

#[inline]
pub fn ser_readdata32<Stream: Read>(s: &mut Stream) -> u32 {
    let mut buf = [0u8; 4];
    s.read_exact(&mut buf).expect("I/O error while reading u32");
    let v = u32::from_le_bytes(buf);
    trace!(value = v, "ser_readdata32 (LE)");
    v
}

#[inline]
pub fn ser_readdata_32be<Stream: Read>(s: &mut Stream) -> u32 {
    let mut buf = [0u8; 4];
    s.read_exact(&mut buf).expect("I/O error while reading u32‑be");
    let v = u32::from_be_bytes(buf);
    trace!(value = v, "ser_readdata_32be (BE)");
    v
}

#[inline]
pub fn ser_readdata64<Stream: Read>(s: &mut Stream) -> u64 {
    let mut buf = [0u8; 8];
    s.read_exact(&mut buf).expect("I/O error while reading u64");
    let v = u64::from_le_bytes(buf);
    trace!(value = v, "ser_readdata64 (LE)");
    v
}

#[cfg(test)]
mod read_write_data_tests {
    use super::*;
    use std::io::Cursor;

    macro_rules! roundtrip_scalar {
        ($name:ident, $write_fn:ident, $read_fn:ident, $ty:ty, $val:expr $(,)?) => {
            #[traced_test]
            fn $name() {
                let mut buf = Cursor::new(Vec::<u8>::new());
                $write_fn(&mut buf, $val as $ty);
                buf.set_position(0);
                let decoded = $read_fn(&mut buf);
                assert_eq!(decoded as $ty, $val as $ty);
            }
        };
    }

    roundtrip_scalar!(u8_roundtrip,  ser_writedata8,   ser_readdata8,  u8,  0xAB);
    roundtrip_scalar!(u16_roundtrip, ser_writedata16,  ser_readdata16, u16, 0xBEEF);
    roundtrip_scalar!(u32_roundtrip, ser_writedata32,  ser_readdata32, u32, 0xDEADBEEF);
    roundtrip_scalar!(u64_roundtrip, ser_writedata64,  ser_readdata64, u64, 0x0123_4567_89AB_CDEF);
}