add `claim_mut`

2 files changed, 47 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index dfc5e52..7b170fc 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -83,6 +83,19 @@ where
         unsafe { &*self.data.get() }
     }
 
+    /// Like [Resource.claim](struct.Resource.html#method.claim) but returns a
+    /// `&mut-` reference
+    pub fn claim_mut<'task, PRIORITY>(
+        &'static self,
+        _priority: &'task mut P<PRIORITY>,
+    ) -> &'task mut T
+    where
+        CEILING: Cmp<PRIORITY, Output = Equal>,
+        P<PRIORITY>: Priority,
+    {
+        unsafe { &mut *self.data.get() }
+    }
+
     /// Locks the resource for the duration of the critical section `f`
     ///
     /// For the duration of the critical section, tasks whose priority level is
diff --git a/tests/cfail/claim_mut.rs b/tests/cfail/claim_mut.rs
new file mode 100644
index 0000000..0d1dffd
--- /dev/null
+++ b/tests/cfail/claim_mut.rs
@@ -0,0 +1,34 @@
+#![feature(const_fn)]
+
+extern crate cortex_m_srp;
+
+use cortex_m_srp::{C2, P2, Resource};
+
+static R1: Resource<i32, C2> = Resource::new(0);
+
+fn j1(mut prio: P2) {
+    // OK only one `&mut-` reference to the data
+    let r1: &mut i32 = R1.claim_mut(&mut prio);
+}
+
+fn j2(prio: P2) {
+    // OK two `&-` references to the same data
+    let r1: &i32 = R1.claim(&prio);
+    let another_r1: &i32 = R1.claim(&prio);
+}
+
+fn j3(mut prio: P2) {
+    // CAN'T have a `&-` reference and a `&mut-` reference to the same data
+    let r1: &i32 = R1.claim(&prio);
+    let another_r1: &mut i32 = R1.claim_mut(&mut prio);
+    //~^ error
+}
+
+fn j4(mut prio: P2) {
+    // CAN'T have two `&mut-` references to the same data
+    let r1: &mut i32 = R1.claim_mut(&mut prio);
+    let another_r1: &mut i32 = R1.claim_mut(&mut prio);
+    //~^ error
+}
+
+fn main() {}