use core::fmt;
use serde::de::{Error, Visitor};
use serde::{Deserializer, Serialize, Serializer};
#[cfg(feature = "alloc")]
use ::{alloc::vec::Vec, serde::Deserialize};
#[cfg(feature = "zeroize")]
use zeroize::Zeroize;
pub fn serialize_hex_lower_or_bin<S, T>(value: &T, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
T: AsRef<[u8]>,
{
if serializer.is_human_readable() {
crate::serialize_hex::<_, _, false>(value, serializer)
} else {
value.as_ref().serialize(serializer)
}
}
pub fn serialize_hex_upper_or_bin<S, T>(value: &T, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
T: AsRef<[u8]>,
{
if serializer.is_human_readable() {
crate::serialize_hex::<_, _, true>(value, serializer)
} else {
value.as_ref().serialize(serializer)
}
}
pub fn deserialize_hex_or_bin<'de, D>(buffer: &mut [u8], deserializer: D) -> Result<&[u8], D::Error>
where
D: Deserializer<'de>,
{
if deserializer.is_human_readable() {
struct StrVisitor<'b>(&'b mut [u8]);
impl<'de, 'b> Visitor<'de> for StrVisitor<'b> {
type Value = &'b [u8];
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(
formatter,
"a string with a maximum length of {}",
self.0.len() * 2
)
}
fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
where
E: Error,
{
base16ct::mixed::decode(v, self.0).map_err(E::custom)
}
}
deserializer.deserialize_str(StrVisitor(buffer))
} else {
struct SliceVisitor<'b>(&'b mut [u8]);
impl<'de, 'b> Visitor<'de> for SliceVisitor<'b> {
type Value = &'b [u8];
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(
formatter,
"a slice with a maximum length of {}",
self.0.len()
)
}
fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
where
E: Error,
{
if v.len() <= self.0.len() {
let buffer = &mut self.0[..v.len()];
buffer.copy_from_slice(v);
return Ok(buffer);
}
Err(E::invalid_length(v.len(), &self))
}
#[cfg(feature = "alloc")]
fn visit_byte_buf<E>(self, mut v: Vec<u8>) -> Result<Self::Value, E>
where
E: Error,
{
if v.len() <= self.0.len() {
let buffer = &mut self.0[..v.len()];
buffer.swap_with_slice(&mut v);
return Ok(buffer);
}
Err(E::invalid_length(v.len(), &self))
}
}
deserializer.deserialize_byte_buf(SliceVisitor(buffer))
}
}
#[cfg(feature = "alloc")]
pub fn deserialize_hex_or_bin_vec<'de, D>(deserializer: D) -> Result<Vec<u8>, D::Error>
where
D: Deserializer<'de>,
{
if deserializer.is_human_readable() {
struct StrVisitor;
impl<'de> Visitor<'de> for StrVisitor {
type Value = Vec<u8>;
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(formatter, "a string")
}
fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
where
E: Error,
{
base16ct::mixed::decode_vec(v).map_err(E::custom)
}
}
deserializer.deserialize_str(StrVisitor)
} else {
Vec::deserialize(deserializer)
}
}
#[cfg(feature = "alloc")]
pub type HexLowerOrBin = HexOrBin<false>;
#[cfg(feature = "alloc")]
pub type HexUpperOrBin = HexOrBin<true>;
#[cfg(feature = "alloc")]
#[derive(Clone, Debug, Default, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub struct HexOrBin<const UPPERCASE: bool>(pub Vec<u8>);
#[cfg(feature = "alloc")]
impl<const UPPERCASE: bool> AsRef<[u8]> for HexOrBin<UPPERCASE> {
fn as_ref(&self) -> &[u8] {
self.0.as_ref()
}
}
#[cfg(feature = "alloc")]
impl<const UPPERCASE: bool> From<&[u8]> for HexOrBin<UPPERCASE> {
fn from(bytes: &[u8]) -> HexOrBin<UPPERCASE> {
Self(bytes.into())
}
}
#[cfg(feature = "alloc")]
impl<const UPPERCASE: bool> From<Vec<u8>> for HexOrBin<UPPERCASE> {
fn from(vec: Vec<u8>) -> HexOrBin<UPPERCASE> {
Self(vec)
}
}
#[cfg(feature = "alloc")]
impl<const UPPERCASE: bool> From<HexOrBin<UPPERCASE>> for Vec<u8> {
fn from(vec: HexOrBin<UPPERCASE>) -> Vec<u8> {
vec.0
}
}
#[cfg(feature = "alloc")]
impl<const UPPERCASE: bool> Serialize for HexOrBin<UPPERCASE> {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
if UPPERCASE {
serialize_hex_upper_or_bin(self, serializer)
} else {
serialize_hex_lower_or_bin(self, serializer)
}
}
}
#[cfg(feature = "alloc")]
impl<'de, const UPPERCASE: bool> Deserialize<'de> for HexOrBin<UPPERCASE> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
deserialize_hex_or_bin_vec(deserializer).map(Self)
}
}
#[cfg(all(feature = "alloc", feature = "zeroize"))]
impl<const UPPERCASE: bool> Zeroize for HexOrBin<UPPERCASE> {
fn zeroize(&mut self) {
self.0.as_mut_slice().zeroize();
}
}