aboutsummaryrefslogtreecommitdiffstats
path: root/src/lib/tls/asio/asio_async_base.h
blob: f476f71550bc8f4976b9265973900f370139220d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/*
* TLS ASIO Stream Helper
* (C) 2018-2019 Jack Lloyd
*     2018-2019 Hannes Rantzsch, Tim Oesterreich, Rene Meusel
*
* Botan is released under the Simplified BSD License (see license.txt)
*/

#ifndef BOTAN_ASIO_ASYNC_BASE_H_
#define BOTAN_ASIO_ASYNC_BASE_H_

#include <boost/beast/core/bind_handler.hpp>

#include <botan/internal/asio_includes.h>

namespace Botan {

namespace TLS {

template <class Handler, class Executor1, class Allocator>
struct AsyncBase
   {
      using allocator_type = boost::asio::associated_allocator_t<Handler, Allocator>;
      using executor_type = boost::asio::associated_executor_t<Handler, Executor1>;

      allocator_type get_allocator() const noexcept
         {
         return boost::asio::get_associated_allocator(m_handler);
         }

      executor_type get_executor() const noexcept
         {
         return boost::asio::get_associated_executor(m_handler, m_work_guard_1.get_executor());
         }

   protected:
      template <class HandlerT>
      AsyncBase(HandlerT&& handler, const Executor1& executor)
         : m_handler(std::forward<HandlerT>(handler))
         , m_work_guard_1(executor)
         {
         }

      template<class... Args>
      void invoke(bool isContinuation, Args&& ... args)
         {
         if(!isContinuation)
            {
            // \note(toesterreich): Is this ok to do with bind_handler? Do we need placeholders?
            boost::asio::post(boost::asio::bind_executor(
                                 m_work_guard_1.get_executor(), boost::beast::bind_handler(std::move(m_handler), args...))
                             );

            m_work_guard_1.reset();
            }
         else
            {
            m_handler(std::forward<Args>(args)...);
            m_work_guard_1.reset();
            }
         }

      Handler m_handler;
      boost::asio::executor_work_guard<Executor1> m_work_guard_1;
   };
}
}

#endif