diff --git a/SConstruct b/SConstruct index 20fbf20bca..0c8cfd3d17 100644 --- a/SConstruct +++ b/SConstruct @@ -226,6 +226,9 @@ testee_client = SConscript('#/examples/testee_client/SConscript',variant_dir = b # utility_client utility_client = SConscript('#/examples/utility_client/SConscript',variant_dir = builddir + 'utility_client',duplicate = 0) +# debug_client +debug_client = SConscript('#/examples/debug_client/SConscript',variant_dir = builddir + 'debug_client',duplicate = 0) + # subprotocol_server subprotocol_server = SConscript('#/examples/subprotocol_server/SConscript',variant_dir = builddir + 'subprotocol_server',duplicate = 0) diff --git a/examples/debug_client/SConscript b/examples/debug_client/SConscript new file mode 100644 index 0000000000..781db8371a --- /dev/null +++ b/examples/debug_client/SConscript @@ -0,0 +1,24 @@ +## Debug client example +## + +Import('env') +Import('env_cpp11') +Import('boostlibs') +Import('platform_libs') +Import('polyfill_libs') +Import('tls_libs') + +env = env.Clone () +env_cpp11 = env_cpp11.Clone () + +prgs = [] + +# if a C++11 environment is available build using that, otherwise use boost +if env_cpp11.has_key('WSPP_CPP11_ENABLED'): + ALL_LIBS = boostlibs(['system'],env_cpp11) + [platform_libs] + [polyfill_libs] + [tls_libs] + prgs += env_cpp11.Program('debug_client', ["debug_client.cpp"], LIBS = ALL_LIBS) +else: + ALL_LIBS = boostlibs(['system','random'],env) + [platform_libs] + [polyfill_libs] + [tls_libs] + prgs += env.Program('debug_client', ["debug_client.cpp"], LIBS = ALL_LIBS) + +Return('prgs') diff --git a/examples/debug_client/debug_client.cpp b/examples/debug_client/debug_client.cpp new file mode 100644 index 0000000000..64282fe656 --- /dev/null +++ b/examples/debug_client/debug_client.cpp @@ -0,0 +1,126 @@ +/** + * This example is presently used as a scratch space. It may or may not be broken + * at any given time. + */ + +#include + +#include + +#include +#include + +typedef websocketpp::client client; + +using websocketpp::lib::placeholders::_1; +using websocketpp::lib::placeholders::_2; +using websocketpp::lib::bind; + +// pull out the type of messages sent by our config +typedef websocketpp::config::asio_tls_client::message_type::ptr message_ptr; +typedef websocketpp::lib::shared_ptr context_ptr; +typedef client::connection_ptr connection_ptr; + + + +class perftest { +public: + typedef perftest type; + typedef std::chrono::duration dur_type; + + perftest () { + m_endpoint.set_access_channels(websocketpp::log::alevel::all); + m_endpoint.set_error_channels(websocketpp::log::elevel::all); + + // Initialize ASIO + m_endpoint.init_asio(); + + // Register our handlers + m_endpoint.set_socket_init_handler(bind(&type::on_socket_init,this,::_1)); + m_endpoint.set_tls_init_handler(bind(&type::on_tls_init,this,::_1)); + m_endpoint.set_message_handler(bind(&type::on_message,this,::_1,::_2)); + m_endpoint.set_open_handler(bind(&type::on_open,this,::_1)); + m_endpoint.set_close_handler(bind(&type::on_close,this,::_1)); + } + + void start(std::string uri) { + websocketpp::lib::error_code ec; + client::connection_ptr con = m_endpoint.get_connection(uri, ec); + + if (ec) { + m_endpoint.get_alog().write(websocketpp::log::alevel::app,ec.message()); + } + + //con->set_proxy("http://humupdates.uchicago.edu:8443"); + + m_endpoint.connect(con); + + // Start the ASIO io_service run loop + m_start = std::chrono::high_resolution_clock::now(); + m_endpoint.run(); + } + + void on_socket_init(websocketpp::connection_hdl hdl) { + m_socket_init = std::chrono::high_resolution_clock::now(); + } + + context_ptr on_tls_init(websocketpp::connection_hdl hdl) { + m_tls_init = std::chrono::high_resolution_clock::now(); + context_ptr ctx(new boost::asio::ssl::context(boost::asio::ssl::context::tlsv1)); + + try { + ctx->set_options(boost::asio::ssl::context::default_workarounds | + boost::asio::ssl::context::no_sslv2 | + boost::asio::ssl::context::single_dh_use); + } catch (std::exception& e) { + std::cout << e.what() << std::endl; + } + return ctx; + } + + void on_open(websocketpp::connection_hdl hdl) { + m_open = std::chrono::high_resolution_clock::now(); + m_endpoint.send(hdl, "", websocketpp::frame::opcode::text); + } + void on_message(websocketpp::connection_hdl hdl, message_ptr msg) { + m_message = std::chrono::high_resolution_clock::now(); + m_endpoint.close(hdl,websocketpp::close::status::going_away,""); + } + void on_close(websocketpp::connection_hdl hdl) { + m_close = std::chrono::high_resolution_clock::now(); + + std::cout << "Socket Init: " << std::chrono::duration_cast(m_socket_init-m_start).count() << std::endl; + std::cout << "TLS Init: " << std::chrono::duration_cast(m_tls_init-m_start).count() << std::endl; + std::cout << "Open: " << std::chrono::duration_cast(m_open-m_start).count() << std::endl; + std::cout << "Message: " << std::chrono::duration_cast(m_message-m_start).count() << std::endl; + std::cout << "Close: " << std::chrono::duration_cast(m_close-m_start).count() << std::endl; + } +private: + client m_endpoint; + + std::chrono::high_resolution_clock::time_point m_start; + std::chrono::high_resolution_clock::time_point m_socket_init; + std::chrono::high_resolution_clock::time_point m_tls_init; + std::chrono::high_resolution_clock::time_point m_open; + std::chrono::high_resolution_clock::time_point m_message; + std::chrono::high_resolution_clock::time_point m_close; +}; + +int main(int argc, char* argv[]) { + std::string uri = "wss://echo.websocket.org"; + + if (argc == 2) { + uri = argv[1]; + } + + try { + perftest endpoint; + endpoint.start(uri); + } catch (const std::exception & e) { + std::cout << e.what() << std::endl; + } catch (websocketpp::lib::error_code e) { + std::cout << e.message() << std::endl; + } catch (...) { + std::cout << "other exception" << std::endl; + } +}