Implement BackgroundTask::running_tasks()

1 files changed, 26 insertions, 0 deletions

diff --git a/rust/src/backgroundtask.rs b/rust/src/backgroundtask.rs
index 985b8a83..363b91a8 100644
--- a/rust/src/backgroundtask.rs
+++ b/rust/src/backgroundtask.rs
@@ -85,6 +85,15 @@ impl BackgroundTask {
             )
         }
     }
+
+    pub fn running_tasks() -> Array<BackgroundTask> {
+        unsafe {
+            let mut count = 0;
+            let handles = BNGetRunningBackgroundTasks(&mut count);
+            
+            Array::new(handles, count, ())
+        }
+    }
 }
 
 unsafe impl RefCountable for BackgroundTask {
@@ -99,6 +108,23 @@ unsafe impl RefCountable for BackgroundTask {
     }
 }
 
+unsafe impl CoreOwnedArrayProvider for BackgroundTask {
+    type Raw = *mut BNBackgroundTask;
+    type Context = ();
+
+    unsafe fn free(raw: *mut *mut BNBackgroundTask, count: usize, _context: &()) {
+        BNFreeBackgroundTaskList(raw, count);
+    }
+}
+
+unsafe impl<'a> CoreOwnedArrayWrapper<'a> for BackgroundTask {
+    type Wrapped = Guard<'a, BackgroundTask>;
+
+    unsafe fn wrap_raw(raw: &'a *mut BNBackgroundTask, context: &'a ()) -> Guard<'a, BackgroundTask> {
+        Guard::new(BackgroundTask::from_raw(*raw), context)
+    }
+}
+
 impl AsRef<BackgroundTask> for BackgroundTask {
     fn as_ref(&self) -> &Self {
         self