Fix undefined behavior in experimental::coro

Set the cancel member for different coro executors. Fixes chriskohlhoff#1536
terrakuh · Oct 3, 2024 · a744b72 · a744b72
1 parent efdc25a
commit a744b72
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/asio/include/asio/experimental/impl/coro.hpp b/asio/include/asio/experimental/impl/coro.hpp
@@ -1000,6 +1000,7 @@ struct coro<Yield, Return, Executor, Allocator>::awaitable_t
         {
           st->second.state = cancellation_state(
               st->second.slot = st->first.slot());
+          coro_->cancel = &st->second;
         }
 
         E e;
@@ -1021,6 +1022,10 @@ struct coro<Yield, Return, Executor, Allocator>::awaitable_t
         hp.cancel->state.slot().template emplace<cancel_handler>(
             coro_.get_executor(), coro_);
       }
+      else
+      {
+        coro_.coro_->cancel = hp.cancel;
+      }
 
       auto hh = detail::coroutine_handle<
         detail::coro_promise<Yield, Return, Executor, Allocator>>::from_promise(