blob: e8bc03feec16f5513c39b28634a91e9182a95efa (
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 Executor, class Allocator>
struct AsyncBase
{
using allocator_type = boost::asio::associated_allocator_t<Handler, Allocator>;
using executor_type = boost::asio::associated_executor_t<Handler, Executor>;
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.get_executor());
}
protected:
template <class HandlerT>
AsyncBase(HandlerT&& handler, const Executor& executor)
: m_handler(std::forward<HandlerT>(handler))
, m_work(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.get_executor(), boost::beast::bind_handler(std::move(m_handler), args...))
);
m_work.reset();
}
else
{
m_handler(std::forward<Args>(args)...);
m_work.reset();
}
}
Handler m_handler;
boost::asio::executor_work_guard<executor_type> m_work;
};
}
}
#endif
|
