chore: Add a feature gate to allow user disabling the NaN check

near · Sep 3, 2024 · fd4e597 · fd4e597
1 parent f16cd07
commit fd4e597
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/borsh/Cargo.toml b/borsh/Cargo.toml
@@ -54,3 +54,4 @@ std = []
 # Be sure that this is what you want before enabling this feature.
 rc = []
 de_strict_order = []
+allow_nan = []
diff --git a/borsh/src/de/mod.rs b/borsh/src/de/mod.rs
@@ -282,6 +282,7 @@ macro_rules! impl_for_float {
                     .read_exact(&mut buf)
                     .map_err(unexpected_eof_to_unexpected_length_of_input)?;
                 let res = $type::from_bits($int_type::from_le_bytes(buf.try_into().unwrap()));
+                #[cfg(not(feature = "allow_nan"))]
                 if res.is_nan() {
                     return Err(Error::new(
                         ErrorKind::InvalidData,

diff --git a/borsh/src/ser/mod.rs b/borsh/src/ser/mod.rs
@@ -138,6 +138,7 @@ macro_rules! impl_for_float {
         impl BorshSerialize for $type {
             #[inline]
             fn serialize<W: Write>(&self, writer: &mut W) -> Result<()> {
+                #[cfg(not(feature = "allow_nan"))]
                 if self.is_nan() {
                     return Err(Error::new(ErrorKind::InvalidData, FLOAT_NAN_ERR));
                 }