Miroirs/rkvm
1
0
Fork 0
mirror of https://github.com/htrefil/rkvm.git synced 2025-01-30 20:34:13 +01:00
Code Issues Projects Releases Packages Wiki Activity

Trace message sizes

Browse source
This commit is contained in:
Jan Trefil 2023-05-13 18:18:19 +02:00
parent 046c5e71ba
commit 0da5dc759f
3 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Cargo.lock generated
View file

@ -806,6 +806,7 @@ dependencies = [
"async-trait",
"bincode",
"hmac",
"log",
"rand",
"rkvm-input",
"serde",

1
rkvm-net/Cargo.toml
View file

@ -16,3 +16,4 @@ thiserror = "1.0.40"
hmac = "0.12.1"
sha2 = "0.10.6"
rand = "0.8.5"
log = "0.4.11"

10
rkvm-net/src/message.rs
View file

@ -19,9 +19,13 @@ impl<T: DeserializeOwned + Serialize + Sync> Message for T {
let mut data = vec![0; length.into()];
stream.read_exact(&mut data).await?;
options()
let data = options()
.deserialize(&data)
.map_err(|err| Error::new(ErrorKind::InvalidData, err))
.map_err(|err| Error::new(ErrorKind::InvalidData, err))?;
log::trace!("Read {} bytes", 2 + length);
Ok(data)
}
async fn encode<W: AsyncWrite + Send + Unpin>(&self, stream: &mut W) -> Result<(), Error> {
@ -37,6 +41,8 @@ impl<T: DeserializeOwned + Serialize + Sync> Message for T {
stream.write_u16(length).await?;
stream.write_all(&data).await?;
log::trace!("Wrote {} bytes", 2 + data.len());
Ok(())
}
}