From 3df5d980d0486b422321670c01c75ca10e216882 Mon Sep 17 00:00:00 2001
From: Mohammad Nejati <ashtumashtum@gmail.com>
Date: Thu, 28 Dec 2023 19:49:45 +0000
Subject: [PATCH] Specialize asio::associator for bind_wrapper and
 bind_front_wrapper

---
 .../beast/_experimental/test/impl/stream.hpp  |   1 +
 include/boost/beast/core/async_base.hpp       |   1 +
 .../boost/beast/core/detail/bind_handler.hpp  | 153 +++++-------------
 test/beast/websocket/close.cpp                |   2 +-
 4 files changed, 40 insertions(+), 117 deletions(-)

diff --git a/include/boost/beast/_experimental/test/impl/stream.hpp b/include/boost/beast/_experimental/test/impl/stream.hpp
index 0971a5b0..b102c164 100644
--- a/include/boost/beast/_experimental/test/impl/stream.hpp
+++ b/include/boost/beast/_experimental/test/impl/stream.hpp
@@ -15,6 +15,7 @@
 #include <boost/beast/core/detail/service_base.hpp>
 #include <boost/beast/core/detail/is_invocable.hpp>
 #include <boost/asio/any_io_executor.hpp>
+#include <boost/asio/associated_cancellation_slot.hpp>
 #include <boost/asio/dispatch.hpp>
 #include <boost/asio/post.hpp>
 #include <mutex>
diff --git a/include/boost/beast/core/async_base.hpp b/include/boost/beast/core/async_base.hpp
index d8fe12ec..4f43a5c6 100644
--- a/include/boost/beast/core/async_base.hpp
+++ b/include/boost/beast/core/async_base.hpp
@@ -17,6 +17,7 @@
 #include <boost/beast/core/detail/filtering_cancellation_slot.hpp>
 #include <boost/beast/core/detail/work_guard.hpp>
 #include <boost/asio/associated_allocator.hpp>
+#include <boost/asio/associated_cancellation_slot.hpp>
 #include <boost/asio/associated_executor.hpp>
 #include <boost/asio/associated_immediate_executor.hpp>
 #include <boost/asio/bind_executor.hpp>
diff --git a/include/boost/beast/core/detail/bind_handler.hpp b/include/boost/beast/core/detail/bind_handler.hpp
index 4b1d188d..5215ef20 100644
--- a/include/boost/beast/core/detail/bind_handler.hpp
+++ b/include/boost/beast/core/detail/bind_handler.hpp
@@ -12,9 +12,7 @@
 
 #include <boost/beast/core/error.hpp>
 #include <boost/beast/core/detail/tuple.hpp>
-#include <boost/asio/associated_allocator.hpp>
-#include <boost/asio/associated_cancellation_slot.hpp>
-#include <boost/asio/associated_executor.hpp>
+#include <boost/asio/associator.hpp>
 #include <boost/asio/handler_continuation_hook.hpp>
 #include <boost/core/ignore_unused.hpp>
 #include <boost/mp11/integer_sequence.hpp>
@@ -42,14 +40,9 @@ class bind_wrapper
     Handler h_;
     args_type args_;
 
-    template<class T, class Executor>
-    friend struct net::associated_executor;
-
-    template<class T, class Allocator>
-    friend struct net::associated_allocator;
-
-    template<class T, class CancellationSlot>
-    friend struct net::associated_cancellation_slot;
+    template <template <typename, typename> class Associator,
+        typename T, typename DefaultCandidate>
+    friend struct net::associator;
 
     template<class Arg, class Vals>
     static
@@ -186,15 +179,9 @@ class bind_front_wrapper
     Handler h_;
     detail::tuple<Args...> args_;
 
-    template<class T, class Executor>
-    friend struct net::associated_executor;
-
-    template<class T, class Allocator>
-    friend struct net::associated_allocator;
-
-    template<class T, class CancellationSlot>
-    friend struct net::associated_cancellation_slot;
-
+    template <template <typename, typename> class Associator,
+        typename T, typename DefaultCandidate>
+    friend struct net::associator;
 
     template<std::size_t... I, class... Ts>
     void
@@ -264,112 +251,46 @@ public:
 namespace boost {
 namespace asio {
 
-template<class Handler, class... Args, class Executor>
-struct associated_executor<
-    beast::detail::bind_wrapper<Handler, Args...>, Executor>
+template <template <typename, typename> class Associator,
+    typename Handler, typename... Args, typename DefaultCandidate>
+struct associator<Associator,
+    beast::detail::bind_wrapper<Handler, Args...>, DefaultCandidate>
+    : Associator<Handler, DefaultCandidate>
 {
-    using type = typename
-        associated_executor<Handler, Executor>::type;
-
-    static
-    type
-    get(beast::detail::bind_wrapper<Handler, Args...> const& op,
-        Executor const& ex = Executor{}) noexcept
+    static typename Associator<Handler, DefaultCandidate>::type get(
+        const beast::detail::bind_wrapper<Handler, Args...>& h) noexcept
     {
-        return associated_executor<
-            Handler, Executor>::get(op.h_, ex);
+        return Associator<Handler, DefaultCandidate>::get(h.h_);
+    }
+
+    static auto get(const beast::detail::bind_wrapper<Handler, Args...>& h,
+        const DefaultCandidate& c) noexcept
+    -> decltype(Associator<Handler, DefaultCandidate>::get(h.h_, c))
+    {
+    return Associator<Handler, DefaultCandidate>::get(h.h_, c);
     }
 };
 
-template<class Handler, class... Args, class Executor>
-struct associated_executor<
-    beast::detail::bind_front_wrapper<Handler, Args...>, Executor>
+template <template <typename, typename> class Associator,
+    typename Handler, typename... Args, typename DefaultCandidate>
+struct associator<Associator,
+    beast::detail::bind_front_wrapper<Handler, Args...>, DefaultCandidate>
+    : Associator<Handler, DefaultCandidate>
 {
-    using type = typename
-        associated_executor<Handler, Executor>::type;
-
-    static
-    type
-    get(beast::detail::bind_front_wrapper<Handler, Args...> const& op,
-        Executor const& ex = Executor{}) noexcept
+    static typename Associator<Handler, DefaultCandidate>::type get(
+        const beast::detail::bind_front_wrapper<Handler, Args...>& h) noexcept
     {
-        return associated_executor<
-            Handler, Executor>::get(op.h_, ex);
+        return Associator<Handler, DefaultCandidate>::get(h.h_);
+    }
+
+    static auto get(const beast::detail::bind_front_wrapper<Handler, Args...>& h,
+        const DefaultCandidate& c) noexcept
+    -> decltype(Associator<Handler, DefaultCandidate>::get(h.h_, c))
+    {
+        return Associator<Handler, DefaultCandidate>::get(h.h_, c);
     }
 };
 
-//
-
-template<class Handler, class... Args, class Allocator>
-struct associated_allocator<
-    beast::detail::bind_wrapper<Handler, Args...>, Allocator>
-{
-    using type = typename
-        associated_allocator<Handler, Allocator>::type;
-
-    static
-    type
-    get(beast::detail::bind_wrapper<Handler, Args...> const& op,
-        Allocator const& alloc = Allocator{}) noexcept
-    {
-        return associated_allocator<
-            Handler, Allocator>::get(op.h_, alloc);
-    }
-};
-
-template<class Handler, class... Args, class Allocator>
-struct associated_allocator<
-    beast::detail::bind_front_wrapper<Handler, Args...>, Allocator>
-{
-    using type = typename
-        associated_allocator<Handler, Allocator>::type;
-
-    static
-    type
-    get(beast::detail::bind_front_wrapper<Handler, Args...> const& op,
-        Allocator const& alloc = Allocator{}) noexcept
-    {
-        return associated_allocator<
-            Handler, Allocator>::get(op.h_, alloc);
-    }
-};
-
-template<class Handler, class... Args, class CancellationSlot>
-struct associated_cancellation_slot<
-    beast::detail::bind_wrapper<Handler, Args...>, CancellationSlot>
-{
-    using type = typename
-        associated_cancellation_slot<Handler>::type;
-
-    static
-    type
-    get(beast::detail::bind_wrapper<Handler, Args...> const& op,
-        CancellationSlot const& slot = CancellationSlot{}) noexcept
-    {
-        return associated_cancellation_slot<
-            Handler, CancellationSlot>::get(op.h_, slot);
-    }
-};
-
-template<class Handler, class... Args, class CancellationSlot>
-struct associated_cancellation_slot<
-    beast::detail::bind_front_wrapper<Handler, Args...>, CancellationSlot>
-{
-    using type = typename
-        associated_cancellation_slot<Handler>::type;
-
-    static
-    type
-    get(beast::detail::bind_front_wrapper<Handler, Args...> const& op,
-        CancellationSlot const& slot = CancellationSlot{}) noexcept
-    {
-        return associated_cancellation_slot<
-            Handler, CancellationSlot>::get(op.h_, slot);
-    }
-};
-
-
-
 } // asio
 } // boost
 
diff --git a/test/beast/websocket/close.cpp b/test/beast/websocket/close.cpp
index 958ef460..ddf7a53d 100644
--- a/test/beast/websocket/close.cpp
+++ b/test/beast/websocket/close.cpp
@@ -775,7 +775,7 @@ public:
 
             BEAST_EXPECT(ioc.run() == 0);
             BEAST_EXPECT(count == 3);
-            BEAST_EXPECT(ic == 6);
+            BEAST_EXPECT(ic == 3);
         });
     }