max/embassy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

optimize interruptfuture

Browse Source 
remove critical secitons, impl. Unpin
This commit is contained in:
xoviat 2021-01-21 10:59:14 -06:00
parent 6503f9dbf5
commit e0183f4495
1 changed files with 18 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
embassy/src/util/signal.rs
View File

@ -86,20 +86,17 @@ pub struct InterruptFuture<'a, I: OwnedInterrupt> {
impl<'a, I: OwnedInterrupt> Drop for InterruptFuture<'a, I> {
fn drop(&mut self) {
cortex_m::interrupt::free(|_| {
self.interrupt.remove_handler();
self.interrupt.disable();
});
self.interrupt.disable();
self.interrupt.remove_handler();
}
}
impl<'a, I: OwnedInterrupt> InterruptFuture<'a, I> {
pub fn new(interrupt: &'a mut I) -> Self {
cortex_m::interrupt::free(|_| {
interrupt.set_handler(Self::interrupt_handler, ptr::null_mut());
interrupt.unpend();
interrupt.enable();
});
interrupt.disable();
interrupt.set_handler(Self::interrupt_handler, ptr::null_mut());
interrupt.unpend();
interrupt.enable();
Self {
interrupt: interrupt,
@ -120,22 +117,21 @@ impl<'a, I: OwnedInterrupt> InterruptFuture<'a, I> {
}
}
impl<'a, I: OwnedInterrupt> Unpin for InterruptFuture<'a, I> {}
impl<'a, I: OwnedInterrupt> Future for InterruptFuture<'a, I> {
type Output = ();
fn poll(self: core::pin::Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<()> {
cortex_m::interrupt::free(|_| unsafe {
let s = self.get_unchecked_mut();
if s.interrupt.is_enabled() {
s.interrupt.set_handler(
Self::interrupt_handler,
executor::raw::task_from_waker(&cx.waker()).cast().as_ptr(),
);
Poll::Pending
} else {
Poll::Ready(())
}
})
let s = unsafe { self.get_unchecked_mut() };
s.interrupt.set_handler(
Self::interrupt_handler,
executor::raw::task_from_waker(&cx.waker()).cast().as_ptr(),
);
if s.interrupt.is_enabled() {
Poll::Pending
} else {
Poll::Ready(())
}
}
}