From 4e1a72796c8390d76f42d3b4002ead1141317f7b Mon Sep 17 00:00:00 2001
From: Victor Koenders <gpg@trangar.com>
Date: Sat, 16 Oct 2021 12:08:29 +0200
Subject: [PATCH] Added support for Cow, Rc and Arc

---
 src/features/impl_alloc.rs | 65 +++++++++++++++++++++++++++++++++++---
 tests/alloc.rs             |  9 ++++++
 2 files changed, 70 insertions(+), 4 deletions(-)

diff --git a/src/features/impl_alloc.rs b/src/features/impl_alloc.rs
index bbe3d1f..53fa85b 100644
--- a/src/features/impl_alloc.rs
+++ b/src/features/impl_alloc.rs
@@ -5,7 +5,7 @@ use crate::{
     error::{DecodeError, EncodeError},
     Config,
 };
-use alloc::{boxed::Box, vec::Vec};
+use alloc::{borrow::Cow, boxed::Box, rc::Rc, sync::Arc, vec::Vec};
 
 #[derive(Default)]
 struct VecWriter {
@@ -35,7 +35,7 @@ pub fn encode_to_vec_with_config<E: enc::Encodeable, C: Config>(
     Ok(encoder.into_writer().inner)
 }
 
-impl<'de, T> Decodable for Vec<T>
+impl<T> Decodable for Vec<T>
 where
     T: Decodable,
 {
@@ -62,7 +62,7 @@ where
     }
 }
 
-impl<'de, T> Decodable for Box<T>
+impl<T> Decodable for Box<T>
 where
     T: Decodable,
 {
@@ -81,7 +81,7 @@ where
     }
 }
 
-impl<'de, T> Decodable for Box<[T]>
+impl<T> Decodable for Box<[T]>
 where
     T: Decodable,
 {
@@ -103,3 +103,60 @@ where
         Ok(())
     }
 }
+
+impl<'cow, T> Decodable for Cow<'cow, T>
+where
+    T: Decodable + Clone,
+{
+    fn decode<D: Decode>(decoder: D) -> Result<Self, DecodeError> {
+        let t = T::decode(decoder)?;
+        Ok(Cow::Owned(t))
+    }
+}
+
+impl<'cow, T> Encodeable for Cow<'cow, T>
+where
+    T: Encodeable + Clone,
+{
+    fn encode<E: Encode>(&self, encoder: E) -> Result<(), EncodeError> {
+        self.as_ref().encode(encoder)
+    }
+}
+
+impl<T> Decodable for Rc<T>
+where
+    T: Decodable,
+{
+    fn decode<D: Decode>(decoder: D) -> Result<Self, DecodeError> {
+        let t = T::decode(decoder)?;
+        Ok(Rc::new(t))
+    }
+}
+
+impl<T> Encodeable for Rc<T>
+where
+    T: Encodeable,
+{
+    fn encode<E: Encode>(&self, encoder: E) -> Result<(), EncodeError> {
+        T::encode(self, encoder)
+    }
+}
+
+impl<T> Decodable for Arc<T>
+where
+    T: Decodable,
+{
+    fn decode<D: Decode>(decoder: D) -> Result<Self, DecodeError> {
+        let t = T::decode(decoder)?;
+        Ok(Arc::new(t))
+    }
+}
+
+impl<T> Encodeable for Arc<T>
+where
+    T: Encodeable,
+{
+    fn encode<E: Encode>(&self, encoder: E) -> Result<(), EncodeError> {
+        T::encode(self, encoder)
+    }
+}
diff --git a/tests/alloc.rs b/tests/alloc.rs
index e051d77..2118b65 100644
--- a/tests/alloc.rs
+++ b/tests/alloc.rs
@@ -1,7 +1,12 @@
 #![cfg(feature = "alloc")]
 
+extern crate alloc;
+
 mod utils;
 
+use alloc::borrow::Cow;
+use alloc::rc::Rc;
+use alloc::sync::Arc;
 use utils::the_same;
 
 struct Foo {
@@ -44,4 +49,8 @@ fn test_alloc_commons() {
     the_same::<Vec<u32>>(vec![1, 2, 3, 4, 5]);
     the_same(Box::<u32>::new(5));
     the_same(Box::<[u32]>::from(vec![1, 2, 3, 4, 5]));
+    the_same(Cow::<u32>::Owned(5));
+    the_same(Cow::<u32>::Borrowed(&5));
+    the_same(Rc::<u32>::new(5));
+    the_same(Arc::<u32>::new(5));
 }