use crate::util::{Error, Hash256, Result, Serializable};
use std::io;
use std::io::{Read, Write};
#[cfg(feature = "async")]
use tokio::io::{AsyncRead, AsyncWrite};
pub const INV_VECT_ERROR: u32 = 0;
pub const INV_VECT_TX: u32 = 1;
pub const INV_VECT_BLOCK: u32 = 2;
pub const INV_VECT_FILTERED_BLOCK: u32 = 3;
pub const INV_VECT_COMPACT_BLOCK: u32 = 4;
#[derive(Debug, PartialEq, Eq, Hash, Clone)]
pub struct InvVect {
pub obj_type: u32,
pub hash: Hash256,
}
impl InvVect {
pub const SIZE: usize = 36;
#[must_use]
#[inline]
pub fn size(&self) -> usize {
Self::SIZE
}
}
impl Serializable<InvVect> for InvVect {
fn read(reader: &mut dyn Read) -> Result<InvVect> {
let mut obj_type = [0u8; 4];
reader
.read_exact(&mut obj_type)
.map_err(|e| Error::IOError(e))?;
let obj_type = u32::from_le_bytes(obj_type);
let hash = Hash256::read(reader)?;
Ok(InvVect { obj_type, hash })
}
fn write(&self, writer: &mut dyn Write) -> io::Result<()> {
writer.write_all(&self.obj_type.to_le_bytes())?;
self.hash.write(writer)?;
Ok(())
}
}
#[cfg(feature = "async")]
impl AsyncSerializable<InvVect> for InvVect {
async fn read_async(reader: &mut dyn AsyncRead) -> Result<InvVect> {
let mut obj_type = [0u8; 4];
reader
.read_exact(&mut obj_type)
.await
.map_err(|e| Error::IOError(e))?;
let obj_type = u32::from_le_bytes(obj_type);
let hash = Hash256::read_async(reader).await?;
Ok(InvVect { obj_type, hash })
}
async fn write_async(&self, writer: &mut dyn AsyncWrite) -> io::Result<()> {
writer.write_all(&self.obj_type.to_le_bytes()).await?;
self.hash.write_async(writer).await?;
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use pretty_assertions::assert_eq;
use std::io::Cursor;
#[test]
fn write_read() {
let mut v = Vec::new();
let iv = InvVect {
obj_type: INV_VECT_TX,
hash: Hash256([8; 32]),
};
iv.write(&mut v).unwrap();
assert_eq!(v.len(), iv.size());
assert_eq!(InvVect::read(&mut Cursor::new(&v)).unwrap(), iv);
}
}