1 files changed, 29 insertions, 35 deletions

diff --git a/embassy-stm32f4/src/exti.rs b/embassy-stm32f4/src/exti.rs
index bf1dd3a71..11704e6d1 100644
--- a/embassy-stm32f4/src/exti.rs
+++ b/embassy-stm32f4/src/exti.rs
@@ -1,49 +1,39 @@
+use core::cell::UnsafeCell;
 use core::future::Future;
 use core::mem;
 use core::pin::Pin;
+use cortex_m;
 
 use embassy::interrupt::Interrupt;
 use embassy::traits::gpio::{WaitForFallingEdge, WaitForRisingEdge};
 use embassy::util::InterruptFuture;
 
 use crate::hal::gpio;
-use crate::hal::gpio::{Edge, ExtiPin as HalExtiPin};
+use crate::hal::gpio::Edge;
 use crate::hal::syscfg::SysCfg;
 use crate::pac::EXTI;
 use embedded_hal::digital::v2 as digital;
 
 use crate::interrupt;
 
-pub struct ExtiManager {
-    syscfg: SysCfg,
+pub struct ExtiPin<T: gpio::ExtiPin + WithInterrupt> {
+    pin: T,
+    interrupt: T::Interrupt,
 }
 
-impl<'a> ExtiManager {
-    pub fn new(_exti: EXTI, syscfg: SysCfg) -> Self {
-        Self { syscfg }
-    }
+impl<T: gpio::ExtiPin + WithInterrupt> ExtiPin<T> {
+    fn new(mut pin: T, interrupt: T::Interrupt) -> Self {
+        let mut syscfg: SysCfg = unsafe { mem::transmute(()) };
 
-    pub fn new_pin<T>(&'static self, mut pin: T, interrupt: T::Interrupt) -> ExtiPin<T>
-    where
-        T: HalExtiPin + WithInterrupt,
-    {
-        pin.make_interrupt_source(&mut self.syscfg);
+        cortex_m::interrupt::free(|_| {
+            pin.make_interrupt_source(&mut syscfg);
+        });
 
-        ExtiPin {
-            pin,
-            interrupt,
-            _mgr: self,
-        }
+        Self { pin, interrupt }
     }
 }
 
-pub struct ExtiPin<T: HalExtiPin + WithInterrupt> {
-    pin: T,
-    interrupt: T::Interrupt,
-    _mgr: &'static ExtiManager,
-}
-
-impl<T: HalExtiPin + WithInterrupt + digital::OutputPin> digital::OutputPin for ExtiPin<T> {
+impl<T: gpio::ExtiPin + WithInterrupt + digital::OutputPin> digital::OutputPin for ExtiPin<T> {
     type Error = T::Error;
 
     fn set_low(&mut self) -> Result<(), Self::Error> {
@@ -55,7 +45,7 @@ impl<T: HalExtiPin + WithInterrupt + digital::OutputPin> digital::OutputPin for
     }
 }
 
-impl<T: HalExtiPin + WithInterrupt + digital::StatefulOutputPin> digital::StatefulOutputPin
+impl<T: gpio::ExtiPin + WithInterrupt + digital::StatefulOutputPin> digital::StatefulOutputPin
     for ExtiPin<T>
 {
     fn is_set_low(&self) -> Result<bool, Self::Error> {
@@ -67,7 +57,7 @@ impl<T: HalExtiPin + WithInterrupt + digital::StatefulOutputPin> digital::Statef
     }
 }
 
-impl<T: HalExtiPin + WithInterrupt + digital::ToggleableOutputPin> digital::ToggleableOutputPin
+impl<T: gpio::ExtiPin + WithInterrupt + digital::ToggleableOutputPin> digital::ToggleableOutputPin
     for ExtiPin<T>
 {
     type Error = T::Error;
@@ -77,7 +67,7 @@ impl<T: HalExtiPin + WithInterrupt + digital::ToggleableOutputPin> digital::Togg
     }
 }
 
-impl<T: HalExtiPin + WithInterrupt + digital::InputPin> digital::InputPin for ExtiPin<T> {
+impl<T: gpio::ExtiPin + WithInterrupt + digital::InputPin> digital::InputPin for ExtiPin<T> {
     type Error = T::Error;
 
     fn is_high(&self) -> Result<bool, Self::Error> {
@@ -100,7 +90,7 @@ impl<T: HalExtiPin + WithInterrupt + digital::InputPin> digital::InputPin for Ex
     EXTI15_10_IRQn      EXTI15_10_IRQHandler    Handler for pins connected to line 10 to 15
 */
 
-impl<T: HalExtiPin + WithInterrupt + 'static> WaitForRisingEdge for ExtiPin<T> {
+impl<T: gpio::ExtiPin + WithInterrupt + 'static> WaitForRisingEdge for ExtiPin<T> {
     type Future<'a> = impl Future<Output = ()> + 'a;
 
     fn wait_for_rising_edge<'a>(self: Pin<&'a mut Self>) -> Self::Future<'a> {
@@ -109,10 +99,12 @@ impl<T: HalExtiPin + WithInterrupt + 'static> WaitForRisingEdge for ExtiPin<T> {
         s.pin.clear_interrupt_pending_bit();
         async move {
             let fut = InterruptFuture::new(&mut s.interrupt);
-            let mut exti: EXTI = unsafe { mem::transmute(()) };
+            cortex_m::interrupt::free(|_| {
+                let mut exti: EXTI = unsafe { mem::transmute(()) };
 
-            s.pin.trigger_on_edge(&mut exti, Edge::RISING);
-            s.pin.enable_interrupt(&mut exti);
+                s.pin.trigger_on_edge(&mut exti, Edge::RISING);
+                s.pin.enable_interrupt(&mut exti);
+            });
             fut.await;
 
             s.pin.clear_interrupt_pending_bit();
@@ -120,7 +112,7 @@ impl<T: HalExtiPin + WithInterrupt + 'static> WaitForRisingEdge for ExtiPin<T> {
     }
 }
 
-impl<T: HalExtiPin + WithInterrupt + 'static> WaitForFallingEdge for ExtiPin<T> {
+impl<T: gpio::ExtiPin + WithInterrupt + 'static> WaitForFallingEdge for ExtiPin<T> {
     type Future<'a> = impl Future<Output = ()> + 'a;
 
     fn wait_for_falling_edge<'a>(self: Pin<&'a mut Self>) -> Self::Future<'a> {
@@ -129,10 +121,12 @@ impl<T: HalExtiPin + WithInterrupt + 'static> WaitForFallingEdge for ExtiPin<T>
         s.pin.clear_interrupt_pending_bit();
         async move {
             let fut = InterruptFuture::new(&mut s.interrupt);
-            let mut exti: EXTI = unsafe { mem::transmute(()) };
+            cortex_m::interrupt::free(|_| {
+                let mut exti: EXTI = unsafe { mem::transmute(()) };
 
-            s.pin.trigger_on_edge(&mut exti, Edge::FALLING);
-            s.pin.enable_interrupt(&mut exti);
+                s.pin.trigger_on_edge(&mut exti, Edge::FALLING);
+                s.pin.enable_interrupt(&mut exti);
+            });
             fut.await;
 
             s.pin.clear_interrupt_pending_bit();