diff options
Diffstat (limited to 'include/socket_adaptors.h')
-rw-r--r-- | include/socket_adaptors.h | 24 |
1 files changed, 19 insertions, 5 deletions
diff --git a/include/socket_adaptors.h b/include/socket_adaptors.h index 201360c..634bd4b 100644 --- a/include/socket_adaptors.h +++ b/include/socket_adaptors.h @@ -1,5 +1,8 @@ #pragma once #include <boost/asio.hpp> +#ifdef CROW_ENABLE_SSL +#include <boost/asio/ssl.hpp> +#endif #include "settings.h" namespace crow { @@ -14,6 +17,11 @@ namespace crow { } + boost::asio::io_service& get_io_service() + { + return socket_.get_io_service(); + } + tcp::socket& raw_socket() { return socket_; @@ -52,20 +60,21 @@ namespace crow struct SSLAdaptor { using context = boost::asio::ssl::context; + using ssl_socket_t = boost::asio::ssl::stream<tcp::socket>; SSLAdaptor(boost::asio::io_service& io_service, context* ctx) - : ssl_socket_(io_service, *ctx) + : ssl_socket_(new ssl_socket_t(io_service, *ctx)) { } boost::asio::ssl::stream<tcp::socket>& socket() { - return ssl_socket_; + return *ssl_socket_; } tcp::socket::lowest_layer_type& raw_socket() { - return ssl_socket_.lowest_layer(); + return ssl_socket_->lowest_layer(); } tcp::endpoint remote_endpoint() @@ -83,16 +92,21 @@ namespace crow raw_socket().close(); } + boost::asio::io_service& get_io_service() + { + return raw_socket().get_io_service(); + } + template <typename F> void start(F f) { - ssl_socket_.async_handshake(boost::asio::ssl::stream_base::server, + ssl_socket_->async_handshake(boost::asio::ssl::stream_base::server, [f](const boost::system::error_code& ec) { f(ec); }); } - boost::asio::ssl::stream<tcp::socket> ssl_socket_; + std::unique_ptr<boost::asio::ssl::stream<tcp::socket>> ssl_socket_; }; #endif } |