mirror of
https://github.com/araxiaonline/TrinityCore.git
synced 2026-06-16 13:09:50 -04:00
91 lines
2.8 KiB
C++
91 lines
2.8 KiB
C++
/*
|
|
* Copyright (C) 2008-2017 TrinityCore <http://www.trinitycore.org/>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
* option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef SslSocket_h__
|
|
#define SslSocket_h__
|
|
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
#include <boost/asio/ssl/stream.hpp>
|
|
#include <boost/system/error_code.hpp>
|
|
|
|
using boost::asio::ip::tcp;
|
|
namespace boostssl = boost::asio::ssl;
|
|
|
|
template<class SslContext>
|
|
class SslSocket
|
|
{
|
|
public:
|
|
explicit SslSocket(tcp::socket&& socket) : _socket(std::move(socket)), _sslSocket(_socket, SslContext::instance())
|
|
{
|
|
_sslSocket.set_verify_mode(boostssl::verify_none);
|
|
}
|
|
|
|
// adapting tcp::socket api
|
|
void close(boost::system::error_code& error)
|
|
{
|
|
_socket.close(error);
|
|
}
|
|
|
|
void shutdown(boost::asio::socket_base::shutdown_type what, boost::system::error_code& shutdownError)
|
|
{
|
|
_sslSocket.shutdown(shutdownError);
|
|
_socket.shutdown(what, shutdownError);
|
|
}
|
|
|
|
template<typename MutableBufferSequence, typename ReadHandlerType>
|
|
void async_read_some(MutableBufferSequence const& buffers, ReadHandlerType&& handler)
|
|
{
|
|
_sslSocket.async_read_some(buffers, std::move(handler));
|
|
}
|
|
|
|
template<typename ConstBufferSequence, typename WriteHandlerType>
|
|
void async_write_some(ConstBufferSequence const& buffers, WriteHandlerType&& handler)
|
|
{
|
|
_sslSocket.async_write_some(buffers, std::move(handler));
|
|
}
|
|
|
|
template<typename ConstBufferSequence>
|
|
std::size_t write_some(ConstBufferSequence const& buffers, boost::system::error_code& error)
|
|
{
|
|
return _sslSocket.write_some(buffers, error);
|
|
}
|
|
|
|
template<typename SettableSocketOption>
|
|
void set_option(SettableSocketOption const& option, boost::system::error_code& error)
|
|
{
|
|
_socket.set_option(option, error);
|
|
}
|
|
|
|
tcp::socket::endpoint_type remote_endpoint() const
|
|
{
|
|
return _socket.remote_endpoint();
|
|
}
|
|
|
|
// ssl api
|
|
template<typename HandshakeHandlerType>
|
|
void async_handshake(boostssl::stream_base::handshake_type type, HandshakeHandlerType&& handler)
|
|
{
|
|
_sslSocket.async_handshake(type, std::move(handler));
|
|
}
|
|
|
|
private:
|
|
tcp::socket _socket;
|
|
boostssl::stream<tcp::socket&> _sslSocket;
|
|
};
|
|
|
|
#endif // SslSocket_h__
|