itsscb/embassy
1
0
Fork 0
mirror of https://github.com/embassy-rs/embassy.git synced 2025-09-29 21:31:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

embassy-rp: defensive change to ensure wakers are registered

Browse Source 
This change ensures that wakers are registered PRIOR to checking status
in i2c `wait_on` helpers.
This commit is contained in:
James Munns 2025-04-03 19:01:00 +02:00
parent 99536219a3
commit 72832c1550
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
embassy-rp/src/i2c.rs
View File

@ -114,17 +114,19 @@ impl<'d, T: Instance> I2c<'d, T, Async> {
}
/// Calls `f` to check if we are ready or not.
/// If not, `g` is called once the waker is set (to eg enable the required interrupts).
/// If not, `g` is called once(to eg enable the required interrupts).
/// The waker will always be registered prior to calling `f`.
async fn wait_on<F, U, G>(&mut self, mut f: F, mut g: G) -> U
where
F: FnMut(&mut Self) -> Poll<U>,
G: FnMut(&mut Self),
{
future::poll_fn(|cx| {
// Register prior to checking the condition
T::waker().register(cx.waker());
let r = f(self);
if r.is_pending() {
T::waker().register(cx.waker());
g(self);
}
r

6
embassy-rp/src/i2c_slave.rs
View File

@ -159,7 +159,8 @@ impl<'d, T: Instance> I2cSlave<'d, T> {
}
/// Calls `f` to check if we are ready or not.
/// If not, `g` is called once the waker is set (to eg enable the required interrupts).
/// If not, `g` is called once(to eg enable the required interrupts).
/// The waker will always be registered prior to calling `f`.
#[inline(always)]
async fn wait_on<F, U, G>(&mut self, mut f: F, mut g: G) -> U
where
@ -167,10 +168,11 @@ impl<'d, T: Instance> I2cSlave<'d, T> {
G: FnMut(&mut Self),
{
future::poll_fn(|cx| {
// Register prior to checking the condition
T::waker().register(cx.waker());
let r = f(self);
if r.is_pending() {
T::waker().register(cx.waker());
g(self);
}