mirror of
https://github.com/Icinga/icinga2.git
synced 2025-08-24 19:18:15 +02:00
58 lines
1.5 KiB
C++
58 lines
1.5 KiB
C++
//
|
|
// Copyright (c) 2020 Kasper Laudrup (laudrup at stacktrace dot dk)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef BOOST_WINTLS_DETAIL_ASYNC_WRITE_HPP
|
|
#define BOOST_WINTLS_DETAIL_ASYNC_WRITE_HPP
|
|
|
|
#include <boost/wintls/detail/sspi_encrypt.hpp>
|
|
|
|
#include <boost/asio/coroutine.hpp>
|
|
|
|
#include <boost/core/ignore_unused.hpp>
|
|
|
|
namespace boost {
|
|
namespace wintls {
|
|
namespace detail {
|
|
|
|
template <typename NextLayer, typename ConstBufferSequence>
|
|
struct async_write : boost::asio::coroutine {
|
|
async_write(NextLayer& next_layer, const ConstBufferSequence& buffer, detail::sspi_encrypt& encrypt)
|
|
: next_layer_(next_layer)
|
|
, buffer_(buffer)
|
|
, encrypt_(encrypt) {
|
|
}
|
|
|
|
template <typename Self>
|
|
void operator()(Self& self, boost::system::error_code ec = {}, std::size_t length = 0) {
|
|
boost::ignore_unused(length);
|
|
BOOST_ASIO_CORO_REENTER(*this) {
|
|
bytes_consumed_ = encrypt_(buffer_, ec);
|
|
if (ec) {
|
|
self.complete(ec, 0);
|
|
return;
|
|
}
|
|
|
|
BOOST_ASIO_CORO_YIELD {
|
|
net::async_write(next_layer_, encrypt_.buffers, std::move(self));
|
|
}
|
|
self.complete(ec, bytes_consumed_);
|
|
}
|
|
}
|
|
|
|
private:
|
|
NextLayer& next_layer_;
|
|
ConstBufferSequence buffer_;
|
|
detail::sspi_encrypt& encrypt_;
|
|
size_t bytes_consumed_{0};
|
|
};
|
|
|
|
} // detail
|
|
} // namespace wintls
|
|
} // namespace boost
|
|
|
|
#endif // BOOST_WINTLS_DETAIL_ASYNC_WRITE_HPP
|