new: DowncastArc and its docs (#244)

kernel/src/libs/casting.rs

@@ -0,0 +1,75 @@
+// Copyright (C) DragonOS Community  longjin 2023
+
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License
+// as published by the Free Software Foundation; either version 2
+// of the License, or (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+// Or you can visit https://www.gnu.org/licenses/gpl-2.0.html
+#![allow(dead_code)]
+
+use core::any::Any;
+
+use alloc::sync::Arc;
+
+/// @brief 将Arc<dyn xxx>转换为Arc<具体类型>的trait
+///
+/// 用法：
+///
+/// ```rust
+/// trait Base: Any + Send + Sync + Debug {
+///     fn get_name(&self) -> String;
+/// }
+///
+/// struct A {
+///    name: String,
+/// }
+///
+/// impl DowncastArc for dyn Base {
+///     fn as_any_arc(self: Arc<Self>) -> Arc<dyn Any> {
+///         return self;
+///     }
+/// }
+///
+/// impl Base for A {
+///    fn get_name(&self) -> String {
+///       return self.name.clone();
+///   }
+/// }
+///
+/// fn test() {
+///     let a = A { name: "a".to_string() };
+
+///     let a_arc: Arc<dyn Base> = Arc::new(a) as Arc<dyn Base>;
+///     let a_arc2: Option<Arc<A>> = a_arc.downcast_arc::<A>();
+///     assert!(a_arc2.is_some());
+/// }
+/// ```
+trait DowncastArc: Any + Send + Sync {
+    /// 请在具体类型中实现这个函数，返回self
+    fn as_any_arc(self: Arc<Self>) -> Arc<dyn Any>;
+
+    /// @brief 将Arc<dyn xxx>转换为Arc<具体类型>
+    ///
+    /// 如果Arc<dyn xxx>是Arc<具体类型>，则返回Some(Arc<具体类型>)，否则返回None
+    ///
+    /// @param self Arc<dyn xxx>
+    fn downcast_arc<T: Any + Send + Sync>(self: Arc<Self>) -> Option<Arc<T>> {
+        let x: Arc<dyn Any> = self.as_any_arc();
+        if x.is::<T>() {
+            // into_raw不会改变引用计数
+            let p = Arc::into_raw(x);
+            let new = unsafe { Arc::from_raw(p as *const T) };
+            return Some(new);
+        }
+        return None;
+    }
+}

kernel/src/libs/mod.rs

@@ -1,4 +1,5 @@
 pub mod atomic;
+pub mod casting;
 pub mod ffi_convert;
 pub mod list;
 pub mod lockref;