118 lines
2.5 KiB
C++
118 lines
2.5 KiB
C++
#pragma once
|
|
#include <boost/asio.hpp>
|
|
#ifdef CROW_ENABLE_SSL
|
|
#include <boost/asio/ssl.hpp>
|
|
#endif
|
|
#include "crow/settings.h"
|
|
#if BOOST_VERSION >= 107000
|
|
#define GET_IO_SERVICE(s) ((boost::asio::io_context&)(s).get_executor().context())
|
|
#else
|
|
#define GET_IO_SERVICE(s) ((s).get_io_service())
|
|
#endif
|
|
namespace crow
|
|
{
|
|
using namespace boost;
|
|
using tcp = asio::ip::tcp;
|
|
|
|
struct SocketAdaptor
|
|
{
|
|
using context = void;
|
|
SocketAdaptor(boost::asio::io_service& io_service, context*)
|
|
: socket_(io_service)
|
|
{
|
|
}
|
|
|
|
boost::asio::io_service& get_io_service()
|
|
{
|
|
return GET_IO_SERVICE(socket_);
|
|
}
|
|
|
|
tcp::socket& raw_socket()
|
|
{
|
|
return socket_;
|
|
}
|
|
|
|
tcp::socket& socket()
|
|
{
|
|
return socket_;
|
|
}
|
|
|
|
tcp::endpoint remote_endpoint()
|
|
{
|
|
return socket_.remote_endpoint();
|
|
}
|
|
|
|
bool is_open()
|
|
{
|
|
return socket_.is_open();
|
|
}
|
|
|
|
void close()
|
|
{
|
|
socket_.close();
|
|
}
|
|
|
|
template <typename F>
|
|
void start(F f)
|
|
{
|
|
f(boost::system::error_code());
|
|
}
|
|
|
|
tcp::socket socket_;
|
|
};
|
|
|
|
#ifdef CROW_ENABLE_SSL
|
|
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_(new ssl_socket_t(io_service, *ctx))
|
|
{
|
|
}
|
|
|
|
boost::asio::ssl::stream<tcp::socket>& socket()
|
|
{
|
|
return *ssl_socket_;
|
|
}
|
|
|
|
tcp::socket::lowest_layer_type&
|
|
raw_socket()
|
|
{
|
|
return ssl_socket_->lowest_layer();
|
|
}
|
|
|
|
tcp::endpoint remote_endpoint()
|
|
{
|
|
return raw_socket().remote_endpoint();
|
|
}
|
|
|
|
bool is_open()
|
|
{
|
|
return raw_socket().is_open();
|
|
}
|
|
|
|
void close()
|
|
{
|
|
raw_socket().close();
|
|
}
|
|
|
|
boost::asio::io_service& get_io_service()
|
|
{
|
|
return GET_IO_SERVICE(raw_socket());
|
|
}
|
|
|
|
template <typename F>
|
|
void start(F f)
|
|
{
|
|
ssl_socket_->async_handshake(boost::asio::ssl::stream_base::server,
|
|
[f](const boost::system::error_code& ec) {
|
|
f(ec);
|
|
});
|
|
}
|
|
|
|
std::unique_ptr<boost::asio::ssl::stream<tcp::socket>> ssl_socket_;
|
|
};
|
|
#endif
|
|
}
|