110 lines
2.5 KiB
C++
110 lines
2.5 KiB
C++
//
|
|
// Copyright (c) 2016-2019 Vinnie Falco (vinnie dot falco at gmail dot com)
|
|
//
|
|
// 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)
|
|
//
|
|
// Official repository: https://github.com/boostorg/beast
|
|
//
|
|
|
|
#ifndef BOOST_BEAST_WEBSOCKET_IMPL_SSL_HPP
|
|
#define BOOST_BEAST_WEBSOCKET_IMPL_SSL_HPP
|
|
|
|
#include <utility>
|
|
#include <boost/beast/websocket/teardown.hpp>
|
|
#include <boost/asio/compose.hpp>
|
|
#include <boost/asio/coroutine.hpp>
|
|
|
|
namespace boost {
|
|
namespace beast {
|
|
|
|
/*
|
|
|
|
See
|
|
http://stackoverflow.com/questions/32046034/what-is-the-proper-way-to-securely-disconnect-an-asio-ssl-socket/32054476#32054476
|
|
|
|
Behavior of ssl::stream regarding close_notify
|
|
|
|
If the remote host calls async_shutdown then the
|
|
local host's async_read will complete with eof.
|
|
|
|
If both hosts call async_shutdown then the calls
|
|
to async_shutdown will complete with eof.
|
|
|
|
*/
|
|
|
|
template<class AsyncStream>
|
|
void
|
|
teardown(
|
|
role_type role,
|
|
boost::asio::ssl::stream<AsyncStream>& stream,
|
|
error_code& ec)
|
|
{
|
|
stream.shutdown(ec);
|
|
using boost::beast::websocket::teardown;
|
|
error_code ec2;
|
|
teardown(role, stream.next_layer(), ec ? ec2 : ec);
|
|
}
|
|
|
|
namespace detail {
|
|
|
|
template<class AsyncStream>
|
|
struct ssl_shutdown_op
|
|
: boost::asio::coroutine
|
|
{
|
|
ssl_shutdown_op(
|
|
boost::asio::ssl::stream<AsyncStream>& s,
|
|
role_type role)
|
|
: s_(s)
|
|
, role_(role)
|
|
{
|
|
}
|
|
|
|
template<class Self>
|
|
void
|
|
operator()(Self& self, error_code ec = {}, std::size_t = 0)
|
|
{
|
|
BOOST_ASIO_CORO_REENTER(*this)
|
|
{
|
|
BOOST_ASIO_CORO_YIELD
|
|
s_.async_shutdown(std::move(self));
|
|
ec_ = ec;
|
|
|
|
using boost::beast::websocket::async_teardown;
|
|
BOOST_ASIO_CORO_YIELD
|
|
async_teardown(role_, s_.next_layer(), std::move(self));
|
|
if (!ec_)
|
|
ec_ = ec;
|
|
|
|
self.complete(ec_);
|
|
}
|
|
}
|
|
|
|
private:
|
|
boost::asio::ssl::stream<AsyncStream>& s_;
|
|
role_type role_;
|
|
error_code ec_;
|
|
};
|
|
|
|
} // detail
|
|
|
|
template<
|
|
class AsyncStream,
|
|
class TeardownHandler>
|
|
void
|
|
async_teardown(
|
|
role_type role,
|
|
boost::asio::ssl::stream<AsyncStream>& stream,
|
|
TeardownHandler&& handler)
|
|
{
|
|
return boost::asio::async_compose<TeardownHandler, void(error_code)>(
|
|
detail::ssl_shutdown_op<AsyncStream>(stream, role),
|
|
handler,
|
|
stream);
|
|
}
|
|
|
|
} // beast
|
|
} // boost
|
|
|
|
#endif
|