28 #ifndef WEBSOCKETPP_TRANSPORT_SECURITY_TLS_HPP 29 #define WEBSOCKETPP_TRANSPORT_SECURITY_TLS_HPP 31 #include <websocketpp/transport/asio/security/base.hpp> 33 #include <websocketpp/uri.hpp> 35 #include <websocketpp/common/asio_ssl.hpp> 36 #include <websocketpp/common/asio.hpp> 37 #include <websocketpp/common/connection_hdl.hpp> 38 #include <websocketpp/common/functional.hpp> 39 #include <websocketpp/common/memory.hpp> 81 explicit connection() {
88 return shared_from_this();
104 return m_socket->lowest_layer();
112 return m_socket->next_layer();
132 m_socket_init_handler = h;
145 m_tls_init_handler = h;
161 lib::asio::error_code aec;
162 lib::asio::ip::tcp::endpoint ep = m_socket->lowest_layer().remote_endpoint(aec);
165 ec = error::make_error_code(error::pass_through);
166 s <<
"Error getting remote endpoint: " << aec
167 <<
" (" << aec.message() <<
")";
170 ec = lib::error_code();
188 if (!m_tls_init_handler) {
189 return socket::make_error_code(socket::error::missing_tls_init_handler);
191 m_context = m_tls_init_handler(m_hdl);
194 return socket::make_error_code(socket::error::invalid_tls_context);
196 m_socket = lib::make_shared<socket_type>(
197 _WEBSOCKETPP_REF(*service),lib::ref(*m_context));
199 m_io_service = service;
201 m_is_server = is_server;
203 return lib::error_code();
235 #if OPENSSL_VERSION_NUMBER >= 0x90812f
240 long res = SSL_set_tlsext_host_name(
241 get_socket().native_handle(), m_uri->get_host().c_str());
243 callback(socket::make_error_code(socket::error::tls_failed_sni_hostname));
248 if (m_socket_init_handler) {
249 m_socket_init_handler(m_hdl,get_socket());
252 callback(lib::error_code());
264 m_ec = socket::make_error_code(socket::error::tls_handshake_timeout);
268 m_socket->async_handshake(
269 get_handshake_type(),
270 m_strand->wrap(lib::bind(
271 &type::handle_init, get_shared(),
273 lib::placeholders::_1
277 m_socket->async_handshake(
278 get_handshake_type(),
280 &type::handle_init, get_shared(),
282 lib::placeholders::_1
299 void handle_init(init_handler callback,lib::asio::error_code
const & ec) {
301 m_ec = socket::make_error_code(socket::error::tls_handshake_failed);
303 m_ec = lib::error_code();
309 lib::error_code get_ec()
const {
323 lib::asio::error_code ec;
324 get_raw_socket().cancel(ec);
328 void async_shutdown(socket::shutdown_handler callback) {
330 m_socket->async_shutdown(m_strand->wrap(callback));
332 m_socket->async_shutdown(callback);
355 template <
typename ErrorCodeType>
357 if (ec.category() == lib::asio::error::get_ssl_category()) {
360 return make_error_code(transport::error::tls_error);
364 return make_error_code(transport::error::pass_through);
382 socket_type::handshake_type get_handshake_type() {
384 return lib::asio::ssl::stream_base::server;
386 return lib::asio::ssl::stream_base::client;
392 context_ptr m_context;
397 lib::error_code m_ec;
399 connection_hdl m_hdl;
400 socket_init_handler m_socket_init_handler;
401 tls_init_handler m_tls_init_handler;
420 explicit endpoint() {}
439 m_socket_init_handler = h;
452 m_tls_init_handler = h;
465 scon->set_socket_init_handler(m_socket_init_handler);
466 scon->set_tls_init_handler(m_tls_init_handler);
467 return lib::error_code();
471 socket_init_handler m_socket_init_handler;
472 tls_init_handler m_tls_init_handler;
socket_type & get_socket()
Retrieve a pointer to the wrapped socket.
lib::asio::error_code cancel_socket()
Cancel all async operations on this socket.
void pre_init(init_handler callback)
Pre-initialize security policy.
connection type
Type of this connection socket component.
void set_handle(connection_hdl hdl)
Sets the connection handle.
bool is_secure() const
Check whether or not this connection is secure.
lib::function< lib::shared_ptr< lib::asio::ssl::context >connection_hdl)> tls_init_handler
The signature of the tls_init_handler for this socket policy.
lib::error_code translate_ec(ErrorCodeType ec)
Translate any security policy specific information about an error code.
lib::error_code init(socket_con_ptr scon)
Initialize a connection.
socket_type::next_layer_type & get_next_layer()
Retrieve a pointer to the layer below the ssl stream.
std::string get_remote_endpoint(lib::error_code &ec) const
Get the remote endpoint address.
connection socket_con_type
The type of the corresponding connection socket component.
void set_tls_init_handler(tls_init_handler h)
Set TLS init handler.
lib::shared_ptr< lib::asio::ssl::context > context_ptr
Type of a shared pointer to the ASIO TLS context being used.
void set_uri(uri_ptr u)
Set hostname hook.
void set_socket_init_handler(socket_init_handler h)
Set socket init handler.
lib::asio::io_service * io_service_ptr
Type of a pointer to the ASIO io_service being used.
ptr get_shared()
Get a shared pointer to this component.
lib::error_code init_asio(io_service_ptr service, strand_ptr strand, bool is_server)
Perform one time initializations.
socket_type::lowest_layer_type & get_raw_socket()
Retrieve a pointer to the underlying socket.
endpoint type
The type of this endpoint socket component.
void handle_accept(connection_ptr con, lib::error_code const &ec)
Handler callback for start_accept.
void set_tls_init_handler(tls_init_handler h)
Set TLS init handler.
void set_socket_init_handler(socket_init_handler h)
Set the socket initialization handler.
void post_init(init_handler callback)
Post-initialize security policy.
TLS enabled Asio endpoint socket component.
lib::error_code translate_ec(lib::error_code ec)
bool is_secure() const
Checks whether the endpoint creates secure connections.
socket_con_type::ptr socket_con_ptr
lib::function< void(lib::error_code const &)> connect_handler
The type and signature of the callback passed to the connect method.