limited immediate to initiate state.

Author: klemens-morgenstern

include/boost/async/detail/handler.hpp

@@ -32,17 +32,18 @@ struct completion_handler_noop_executor : executor
   template<typename Fn>
   void execute(Fn && fn) const
   {
+    // only allow it when we're still initializing
     if (completed_immediately &&
-        //avoid recursion. i.e. don't let multiple sup operations complete immediately.
-       *completed_immediately != completed_immediately_t::maybe)
+        ((*completed_immediately == completed_immediately_t::initiating)
+      || (*completed_immediately == completed_immediately_t::maybe)))
     {
       // only use this indicator if the fn will actually call our completion-handler
       // otherwise this was a single op in a composed operation
       *completed_immediately = completed_immediately_t::maybe;
       fn();
       // yes means completion_handler::operator() was called, so we're good.
       if (*completed_immediately != completed_immediately_t::yes)
-        *completed_immediately = completed_immediately_t::no;
+        *completed_immediately = completed_immediately_t::initiating;
     }
     else
     {

test/handler.cpp

@@ -44,6 +44,7 @@ struct immediate_aw
                                    &completed_immediately};
 
     auto exec = asio::get_associated_immediate_executor(ch, h.promise().get_executor());
+    completed_immediately = async::detail::completed_immediately_t::initiating;
     asio::dispatch(exec, std::move(ch));
 
     CHECK(result);