nrf/wdt: add handle steal

2 files changed, 9 insertions, 1 deletions

diff --git a/embassy-nrf/src/wdt.rs b/embassy-nrf/src/wdt.rs
index 0c6f2b3fb..cd62d0d60 100644
--- a/embassy-nrf/src/wdt.rs
+++ b/embassy-nrf/src/wdt.rs
@@ -150,4 +150,12 @@ impl WatchdogHandle {
         let idx = self.index as usize;
         ((rd >> idx) & 0x1) == 0
     }
+
+    /// Steal a watchdog handle by index.
+    ///
+    /// Safety: watchdog must be initialized, index must be between 0 and N-1 where
+    /// N is the handle count when initializing.
+    pub unsafe fn steal(index: u8) -> Self {
+        Self { index }
+    }
 }
diff --git a/examples/nrf/src/bin/wdt.rs b/examples/nrf/src/bin/wdt.rs
index 53df7b0b7..eaa0b5ee1 100644
--- a/examples/nrf/src/bin/wdt.rs
+++ b/examples/nrf/src/bin/wdt.rs
@@ -25,7 +25,7 @@ async fn main(_spawner: Spawner, p: Peripherals) {
     // in the WDT interrupt. The core resets 2 ticks after firing the interrupt.
     config.run_during_debug_halt = false;
 
-    let (_wdt, [mut handle]) = match Watchdog::try_new::<1>(p.WDT, config) {
+    let (_wdt, [mut handle]) = match Watchdog::try_new(p.WDT, config) {
         Ok(x) => x,
         Err(_) => {
             info!("Watchdog already active with wrong config, waiting for it to timeout...");