|
|
@ -13,7 +13,7 @@ use tokio::io::{
|
|
|
|
};
|
|
|
|
};
|
|
|
|
use serde::{
|
|
|
|
use serde::{
|
|
|
|
Serialize,
|
|
|
|
Serialize,
|
|
|
|
Deserialize
|
|
|
|
de::DeserializeOwned
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Hash, Copy, PartialOrd, Ord)]
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Hash, Copy, PartialOrd, Ord)]
|
|
|
@ -75,6 +75,19 @@ pub struct SendOpt
|
|
|
|
}
|
|
|
|
}
|
|
|
|
ref_self!(SendOpt);
|
|
|
|
ref_self!(SendOpt);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Hash, Copy, Default)]
|
|
|
|
|
|
|
|
pub struct RecvOpt
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
pub comp: Option<CompressionKind>,
|
|
|
|
|
|
|
|
pub handle_encrypted: Option<EncryptionKind>,
|
|
|
|
|
|
|
|
pub format: SerialFormat,
|
|
|
|
|
|
|
|
/// Max size of the (decompressed) input buffer.
|
|
|
|
|
|
|
|
/// 0 for unlimited.
|
|
|
|
|
|
|
|
//TODO: Do we need this?
|
|
|
|
|
|
|
|
pub max_size: usize,
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
ref_self!(RecvOpt);
|
|
|
|
|
|
|
|
|
|
|
|
/// Default buffer size for encryption transform stream copying.
|
|
|
|
/// Default buffer size for encryption transform stream copying.
|
|
|
|
pub const DEFAULT_BUFSIZE: usize = 4096;
|
|
|
|
pub const DEFAULT_BUFSIZE: usize = 4096;
|
|
|
|
|
|
|
|
|
|
|
@ -104,7 +117,13 @@ where F: AsyncRead + Unpin + ?Sized,
|
|
|
|
Ok((written, read))
|
|
|
|
Ok((written, read))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
async fn ser_singleton_inner<T: Serialize, V: AsyncWrite + Unpin, F>(to: F, value: &T, how: impl AsRef<SendOpt>) -> Result<(V, usize), SendErrorKind>
|
|
|
|
async fn de_singleton_inner<T: DeserializeOwned, S>(from: S, how: impl AsRef<RecvOpt>) -> Result<T, TransformErrorKind>
|
|
|
|
|
|
|
|
where S: AsRef<[u8]> //TODO: Should we use bytes::Buf or something instead?
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
todo!("Deserialise from `from` using `how`.")
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async fn ser_singleton_inner<T: Serialize, V: AsyncWrite + Unpin, F>(to: F, value: &T, how: impl AsRef<SendOpt>) -> Result<(V, usize), TransformErrorKind>
|
|
|
|
where F: FnOnce(&Vec<u8>) -> V
|
|
|
|
where F: FnOnce(&Vec<u8>) -> V
|
|
|
|
{
|
|
|
|
{
|
|
|
|
let how = how.as_ref();
|
|
|
|
let how = how.as_ref();
|
|
|
@ -153,13 +172,25 @@ where F: FnOnce(&Vec<u8>) -> V
|
|
|
|
/// Serialise a single object to a stream with the method described by `how`.
|
|
|
|
/// Serialise a single object to a stream with the method described by `how`.
|
|
|
|
#[inline] pub async fn write_singleton<T: Serialize, S: ?Sized + AsyncWrite + Unpin>(to: &mut S, value: &T, how: &SendOpt) -> Result<usize, SendError>
|
|
|
|
#[inline] pub async fn write_singleton<T: Serialize, S: ?Sized + AsyncWrite + Unpin>(to: &mut S, value: &T, how: &SendOpt) -> Result<usize, SendError>
|
|
|
|
{
|
|
|
|
{
|
|
|
|
ser_singleton_inner(|_| to, value, &how).await
|
|
|
|
let (cont, v) = ser_singleton_inner(|n| Vec::with_capacity(n.len()), value, &how).await
|
|
|
|
.map_err(|k| SendError(Box::new((k, how.to_owned()))))
|
|
|
|
.map_err(|k| SendError(Box::new((k, how.to_owned()))))?;
|
|
|
|
.map(|(_, n)| n)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
let n = async move {
|
|
|
|
|
|
|
|
to.write_all(&(v as u64).to_be_bytes()[..]).await?;
|
|
|
|
|
|
|
|
to.write_all(&cont).await
|
|
|
|
|
|
|
|
.map(|_| std::mem::size_of::<u64>() + cont.len())
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
.await
|
|
|
|
|
|
|
|
.map_err(|k| SendError(Box::new((k.into(), how.to_owned()))))?;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Ok(n)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Kind of error for a send (serialise) or receive (deserialise) operation
|
|
|
|
#[derive(Debug)]
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub enum SendErrorKind
|
|
|
|
pub enum TransformErrorKind
|
|
|
|
{
|
|
|
|
{
|
|
|
|
/// Invalid serialised format
|
|
|
|
/// Invalid serialised format
|
|
|
|
Format,
|
|
|
|
Format,
|
|
|
@ -174,14 +205,19 @@ pub enum SendErrorKind
|
|
|
|
|
|
|
|
|
|
|
|
/// An error when sending / serialising an object.
|
|
|
|
/// An error when sending / serialising an object.
|
|
|
|
#[derive(Debug)]
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct SendError(Box<(SendErrorKind, SendOpt)>);
|
|
|
|
pub struct RecvError(Box<(TransformErrorKind, RecvOpt)>);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// An error when sending / serialising an object.
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
|
|
|
|
pub struct SendError(Box<(TransformErrorKind, SendOpt)>);
|
|
|
|
|
|
|
|
|
|
|
|
impl error::Error for SendError
|
|
|
|
impl error::Error for SendError
|
|
|
|
{
|
|
|
|
{
|
|
|
|
fn source(&self) -> Option<&(dyn error::Error + 'static)> {
|
|
|
|
fn source(&self) -> Option<&(dyn error::Error + 'static)> {
|
|
|
|
Some(match &self.0.0
|
|
|
|
Some(match &self.0.0
|
|
|
|
{
|
|
|
|
{
|
|
|
|
SendErrorKind::IO(io) => io,
|
|
|
|
TransformErrorKind::IO(io) => io,
|
|
|
|
_ => return None,
|
|
|
|
_ => return None,
|
|
|
|
})
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -193,15 +229,15 @@ impl fmt::Display for SendError
|
|
|
|
{
|
|
|
|
{
|
|
|
|
write!(f, "error when serialising object with params {:?}: ", self.0.1)?;
|
|
|
|
write!(f, "error when serialising object with params {:?}: ", self.0.1)?;
|
|
|
|
match self.0.0 {
|
|
|
|
match self.0.0 {
|
|
|
|
SendErrorKind::Format => write!(f, "failed to serialise object to data"),
|
|
|
|
TransformErrorKind::Format => write!(f, "failed to serialise object to data"),
|
|
|
|
SendErrorKind::Compress => write!(f, "failed to compress data"),
|
|
|
|
TransformErrorKind::Compress => write!(f, "failed to compress data"),
|
|
|
|
SendErrorKind::Encrypt => write!(f, "failed to encrypt data"),
|
|
|
|
TransformErrorKind::Encrypt => write!(f, "failed to encrypt data"),
|
|
|
|
SendErrorKind::IO(_) => write!(f, "i/o failure"),
|
|
|
|
TransformErrorKind::IO(_) => write!(f, "i/o failure"),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
impl From<io::Error> for SendErrorKind
|
|
|
|
impl From<io::Error> for TransformErrorKind
|
|
|
|
{
|
|
|
|
{
|
|
|
|
fn from(from: io::Error) -> Self
|
|
|
|
fn from(from: io::Error) -> Self
|
|
|
|
{
|
|
|
|
{
|
|
|
@ -210,7 +246,7 @@ impl From<io::Error> for SendErrorKind
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
impl From<serde_cbor::Error> for SendErrorKind
|
|
|
|
impl From<serde_cbor::Error> for TransformErrorKind
|
|
|
|
{
|
|
|
|
{
|
|
|
|
#[inline] fn from(_: serde_cbor::Error) -> Self
|
|
|
|
#[inline] fn from(_: serde_cbor::Error) -> Self
|
|
|
|
{
|
|
|
|
{
|
|
|
@ -218,7 +254,7 @@ impl From<serde_cbor::Error> for SendErrorKind
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
impl From<serde_json::Error> for SendErrorKind
|
|
|
|
impl From<serde_json::Error> for TransformErrorKind
|
|
|
|
{
|
|
|
|
{
|
|
|
|
#[inline] fn from(_: serde_json::Error) -> Self
|
|
|
|
#[inline] fn from(_: serde_json::Error) -> Self
|
|
|
|
{
|
|
|
|
{
|
|
|
|