Add validation for serialization format

Signed-off-by: trivernis <trivernis@protonmail.com>
main
trivernis 2 years ago
parent a6c8ad8795
commit 3cb1aa0496
Signed by: Trivernis
GPG Key ID: DFFFCC2C7A02DB45

@ -15,10 +15,23 @@ pub trait HydrusSerializable: DeserializeOwned {
fn type_id() -> u64; fn type_id() -> u64;
} }
pub trait ConstNumberTrait {
fn value() -> u64;
}
impl<T> ConstNumberTrait for T
where
T: HydrusSerializable,
{
fn value() -> u64 {
T::type_id()
}
}
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
pub struct SerializableId<T: HydrusSerializable>(u64, PhantomData<T>); pub struct SerializableId<T: ConstNumberTrait>(u64, PhantomData<T>);
impl<'de, T: HydrusSerializable> Deserialize<'de> for SerializableId<T> { impl<'de, T: ConstNumberTrait> Deserialize<'de> for SerializableId<T> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where where
D: Deserializer<'de>, D: Deserializer<'de>,
@ -29,11 +42,11 @@ impl<'de, T: HydrusSerializable> Deserialize<'de> for SerializableId<T> {
struct SerIdVisitor<T>(PhantomData<T>); struct SerIdVisitor<T>(PhantomData<T>);
impl<'de, T: HydrusSerializable> Visitor<'de> for SerIdVisitor<T> { impl<'de, T: ConstNumberTrait> Visitor<'de> for SerIdVisitor<T> {
type Value = SerializableId<T>; type Value = SerializableId<T>;
fn expecting(&self, formatter: &mut Formatter) -> std::fmt::Result { fn expecting(&self, formatter: &mut Formatter) -> std::fmt::Result {
write!(formatter, "an unsigned integer equal to {}", T::type_id()) write!(formatter, "an unsigned integer equal to {}", T::value())
} }
fn visit_u8<E>(self, v: u8) -> Result<Self::Value, E> fn visit_u8<E>(self, v: u8) -> Result<Self::Value, E>
@ -61,9 +74,9 @@ impl<'de, T: HydrusSerializable> Visitor<'de> for SerIdVisitor<T> {
where where
E: Error, E: Error,
{ {
let expected_value = T::type_id(); let expected_value = T::value();
if v != expected_value { if v != expected_value {
Err(E::custom(format!("type not equal to {}", expected_value))) Err(E::custom(format!("value not equal to {}", expected_value)))
} else { } else {
Ok(SerializableId(expected_value, PhantomData)) Ok(SerializableId(expected_value, PhantomData))
} }

@ -1,10 +1,21 @@
use crate::hydrus_serializable::{HydrusSerializable, SerializableId}; use crate::hydrus_serializable::{ConstNumberTrait, HydrusSerializable, SerializableId};
use serde::Deserialize; use serde::Deserialize;
#[derive(Clone, Debug, Deserialize)]
pub struct VersionOne;
impl ConstNumberTrait for VersionOne {
fn value() -> u64 {
1
}
}
#[derive(Clone, Debug, Deserialize)] #[derive(Clone, Debug, Deserialize)]
#[serde(bound = "")] #[serde(bound = "")]
pub struct HydrusSerWrapper<T: HydrusSerializable> { pub struct HydrusSerWrapper<T: HydrusSerializable> {
#[allow(unused)]
pub type_id: SerializableId<T>, pub type_id: SerializableId<T>,
pub version: u8, #[allow(unused)]
pub version: SerializableId<VersionOne>,
pub inner: T, pub inner: T,
} }

Loading…
Cancel
Save