mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-01-20 12:37:13 +01:00
270 lines
8.8 KiB
C++
270 lines
8.8 KiB
C++
/*
|
|
* Copyright (c) 2014, Peter Thorson. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* * Neither the name of the WebSocket++ Project nor the
|
|
* names of its contributors may be used to endorse or promote products
|
|
* derived from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
|
|
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*/
|
|
|
|
#ifndef WEBSOCKETPP_ENDPOINT_IMPL_HPP
|
|
#define WEBSOCKETPP_ENDPOINT_IMPL_HPP
|
|
|
|
#include <string>
|
|
|
|
namespace websocketpp {
|
|
|
|
template <typename connection, typename config>
|
|
typename endpoint<connection,config>::connection_ptr
|
|
endpoint<connection,config>::create_connection() {
|
|
m_alog->write(log::alevel::devel,"create_connection");
|
|
//scoped_lock_type lock(m_state_lock);
|
|
|
|
/*if (m_state == STOPPING || m_state == STOPPED) {
|
|
return connection_ptr();
|
|
}*/
|
|
|
|
//scoped_lock_type guard(m_mutex);
|
|
// Create a connection on the heap and manage it using a shared pointer
|
|
connection_ptr con = lib::make_shared<connection_type>(m_is_server,
|
|
m_user_agent, m_alog, m_elog, lib::ref(m_rng));
|
|
|
|
connection_weak_ptr w(con);
|
|
|
|
// Create a weak pointer on the heap using that shared_ptr.
|
|
// Cast that weak pointer to void* and manage it using another shared_ptr
|
|
// connection_hdl hdl(reinterpret_cast<void*>(new connection_weak_ptr(con)));
|
|
|
|
con->set_handle(w);
|
|
|
|
// Copy default handlers from the endpoint
|
|
con->set_open_handler(m_open_handler);
|
|
con->set_close_handler(m_close_handler);
|
|
con->set_fail_handler(m_fail_handler);
|
|
con->set_ping_handler(m_ping_handler);
|
|
con->set_pong_handler(m_pong_handler);
|
|
con->set_pong_timeout_handler(m_pong_timeout_handler);
|
|
con->set_interrupt_handler(m_interrupt_handler);
|
|
con->set_http_handler(m_http_handler);
|
|
con->set_validate_handler(m_validate_handler);
|
|
con->set_message_handler(m_message_handler);
|
|
|
|
if (m_open_handshake_timeout_dur != config::timeout_open_handshake) {
|
|
con->set_open_handshake_timeout(m_open_handshake_timeout_dur);
|
|
}
|
|
if (m_close_handshake_timeout_dur != config::timeout_close_handshake) {
|
|
con->set_close_handshake_timeout(m_close_handshake_timeout_dur);
|
|
}
|
|
if (m_pong_timeout_dur != config::timeout_pong) {
|
|
con->set_pong_timeout(m_pong_timeout_dur);
|
|
}
|
|
if (m_max_message_size != config::max_message_size) {
|
|
con->set_max_message_size(m_max_message_size);
|
|
}
|
|
con->set_max_http_body_size(m_max_http_body_size);
|
|
|
|
lib::error_code ec;
|
|
|
|
ec = transport_type::init(con);
|
|
if (ec) {
|
|
m_elog->write(log::elevel::fatal,ec.message());
|
|
return connection_ptr();
|
|
}
|
|
|
|
return con;
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::interrupt(connection_hdl hdl, lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
|
|
m_alog->write(log::alevel::devel,"Interrupting connection");
|
|
|
|
ec = con->interrupt();
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::interrupt(connection_hdl hdl) {
|
|
lib::error_code ec;
|
|
interrupt(hdl,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::pause_reading(connection_hdl hdl, lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
|
|
ec = con->pause_reading();
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::pause_reading(connection_hdl hdl) {
|
|
lib::error_code ec;
|
|
pause_reading(hdl,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::resume_reading(connection_hdl hdl, lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
|
|
ec = con->resume_reading();
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::resume_reading(connection_hdl hdl) {
|
|
lib::error_code ec;
|
|
resume_reading(hdl,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send_http_response(connection_hdl hdl,
|
|
lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
con->send_http_response(ec);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send_http_response(connection_hdl hdl) {
|
|
lib::error_code ec;
|
|
send_http_response(hdl,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send(connection_hdl hdl, std::string const & payload,
|
|
frame::opcode::value op, lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
|
|
ec = con->send(payload,op);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send(connection_hdl hdl, std::string const & payload,
|
|
frame::opcode::value op)
|
|
{
|
|
lib::error_code ec;
|
|
send(hdl,payload,op,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send(connection_hdl hdl, void const * payload,
|
|
size_t len, frame::opcode::value op, lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
ec = con->send(payload,len,op);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send(connection_hdl hdl, void const * payload,
|
|
size_t len, frame::opcode::value op)
|
|
{
|
|
lib::error_code ec;
|
|
send(hdl,payload,len,op,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send(connection_hdl hdl, message_ptr msg,
|
|
lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
ec = con->send(msg);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::send(connection_hdl hdl, message_ptr msg) {
|
|
lib::error_code ec;
|
|
send(hdl,msg,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::close(connection_hdl hdl, close::status::value
|
|
const code, std::string const & reason,
|
|
lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
con->close(code,reason,ec);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::close(connection_hdl hdl, close::status::value
|
|
const code, std::string const & reason)
|
|
{
|
|
lib::error_code ec;
|
|
close(hdl,code,reason,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::ping(connection_hdl hdl, std::string const &
|
|
payload, lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
con->ping(payload,ec);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::ping(connection_hdl hdl, std::string const & payload)
|
|
{
|
|
lib::error_code ec;
|
|
ping(hdl,payload,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::pong(connection_hdl hdl, std::string const & payload,
|
|
lib::error_code & ec)
|
|
{
|
|
connection_ptr con = get_con_from_hdl(hdl,ec);
|
|
if (ec) {return;}
|
|
con->pong(payload,ec);
|
|
}
|
|
|
|
template <typename connection, typename config>
|
|
void endpoint<connection,config>::pong(connection_hdl hdl, std::string const & payload)
|
|
{
|
|
lib::error_code ec;
|
|
pong(hdl,payload,ec);
|
|
if (ec) { throw exception(ec); }
|
|
}
|
|
|
|
} // namespace websocketpp
|
|
|
|
#endif // WEBSOCKETPP_ENDPOINT_IMPL_HPP
|