]> git.bitcoin.ninja Git - rust-lightning/commitdiff
Wipe `Notifier` `FutureState` when returning from a waiter.
authorMatt Corallo <git@bluematt.me>
Tue, 15 Nov 2022 00:24:25 +0000 (00:24 +0000)
committerMatt Corallo <git@bluematt.me>
Wed, 16 Nov 2022 00:21:43 +0000 (00:21 +0000)
When we return from one of the wait functions in `Notifier`, we
should also ensure that the next `Future` doesn't start in the
`complete` state, as we have already notified the user, as far as
we're concerned.

This is technically a regression from the previous commit, but as
it is a logically separate change it is in its own commit.

lightning/src/util/wakers.rs

index 6d8b03cbd7c4c66e9c686bad5afbec601df96ee5..976ce69f8859d3e533580a8767de08b040106680 100644 (file)
@@ -33,6 +33,20 @@ pub(crate) struct Notifier {
        condvar: Condvar,
 }
 
+macro_rules! check_woken {
+       ($guard: expr, $retval: expr) => { {
+               if $guard.0 {
+                       $guard.0 = false;
+                       if $guard.1.as_ref().map(|l| l.lock().unwrap().complete).unwrap_or(false) {
+                               // If we're about to return as woken, and the future state is marked complete, wipe
+                               // the future state and let the next future wait until we get a new notify.
+                               $guard.1.take();
+                       }
+                       return $retval;
+               }
+       } }
+}
+
 impl Notifier {
        pub(crate) fn new() -> Self {
                Self {
@@ -57,16 +71,9 @@ impl Notifier {
        pub(crate) fn wait(&self) {
                loop {
                        let mut guard = self.propagate_future_state_to_notify_flag();
-                       if guard.0 {
-                               guard.0 = false;
-                               return;
-                       }
+                       check_woken!(guard, ());
                        guard = self.condvar.wait(guard).unwrap();
-                       let result = guard.0;
-                       if result {
-                               guard.0 = false;
-                               return
-                       }
+                       check_woken!(guard, ());
                }
        }
 
@@ -75,24 +82,20 @@ impl Notifier {
                let current_time = Instant::now();
                loop {
                        let mut guard = self.propagate_future_state_to_notify_flag();
-                       if guard.0 {
-                               guard.0 = false;
-                               return true;
-                       }
+                       check_woken!(guard, true);
                        guard = self.condvar.wait_timeout(guard, max_wait).unwrap().0;
+                       check_woken!(guard, true);
                        // Due to spurious wakeups that can happen on `wait_timeout`, here we need to check if the
                        // desired wait time has actually passed, and if not then restart the loop with a reduced wait
                        // time. Note that this logic can be highly simplified through the use of
                        // `Condvar::wait_while` and `Condvar::wait_timeout_while`, if and when our MSRV is raised to
                        // 1.42.0.
                        let elapsed = current_time.elapsed();
-                       let result = guard.0;
-                       if result || elapsed >= max_wait {
-                               guard.0 = false;
-                               return result;
+                       if elapsed >= max_wait {
+                               return false;
                        }
                        match max_wait.checked_sub(elapsed) {
-                               None => return result,
+                               None => return false,
                                Some(_) => continue
                        }
                }