Implement alloc::sync::{Arc,Weak,UniqueArc} with alloc::raw_rc types

Author: EFanZh

library/alloc/src/raw_rc/raw_rc.rs

@@ -221,7 +221,6 @@ where
         self.weak.into_raw_parts()
     }
 
-    #[cfg(not(no_global_oom_handling))]
     pub(crate) unsafe fn is_unique<R>(&self) -> bool
     where
         R: RefCounter,
@@ -794,6 +793,23 @@ impl<A> RawRc<dyn Any, A> {
     }
 }
 
+#[cfg(not(no_sync))]
+impl<A> RawRc<dyn Any + Send + Sync, A> {
+    pub(crate) fn downcast<T>(self) -> Result<RawRc<T, A>, Self>
+    where
+        T: Any,
+    {
+        if self.as_ref().is::<T>() { Ok(unsafe { self.downcast_unchecked() }) } else { Err(self) }
+    }
+
+    pub(crate) unsafe fn downcast_unchecked<T>(self) -> RawRc<T, A>
+    where
+        T: Any,
+    {
+        unsafe { self.cast() }
+    }
+}
+
 impl<T, A> AsRef<T> for RawRc<T, A>
 where
     T: ?Sized,

library/alloc/src/raw_rc/raw_weak.rs

@@ -192,13 +192,19 @@ where
         !ptr::addr_eq(self.ptr.as_ptr(), other.ptr.as_ptr())
     }
 
+    /// Returns the `RefCounts` object inside the reference-counted allocation if `self` is
+    /// non-dangling.
+    #[cfg(not(no_sync))]
+    pub(crate) fn ref_counts(&self) -> Option<&RefCounts> {
+        (!is_dangling(self.ptr.cast())).then(|| unsafe { self.ref_counts_unchecked() })
+    }
+
     /// Returns the `RefCounts` object inside the reference-counted allocation, assume `self` is
     /// non-dangling.
     ///
     /// # Safety
     ///
     /// `self` is non-dangling.
-    #[cfg(not(no_global_oom_handling))]
     pub(super) unsafe fn ref_counts_unchecked(&self) -> &RefCounts {
         unsafe { self.value_ptr_unchecked().ref_counts_ptr().as_ref() }
     }