166 lines
6.8 KiB
C++
166 lines
6.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.
|
|
*/
|
|
|
|
/** ====== WARNING ========
|
|
* This example is presently used as a scratch space. It may or may not be broken
|
|
* at any given time.
|
|
*/
|
|
|
|
#include <websocketpp/config/asio_client.hpp>
|
|
|
|
#include <websocketpp/client.hpp>
|
|
|
|
#include <iostream>
|
|
#include <chrono>
|
|
|
|
typedef websocketpp::client<websocketpp::config::asio_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<boost::asio::ssl::context> context_ptr;
|
|
typedef client::connection_ptr connection_ptr;
|
|
|
|
|
|
|
|
class perftest {
|
|
public:
|
|
typedef perftest type;
|
|
typedef std::chrono::duration<int,std::micro> 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));
|
|
m_endpoint.set_fail_handler(bind(&type::on_fail,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) {
|
|
m_socket_init = std::chrono::high_resolution_clock::now();
|
|
}
|
|
|
|
context_ptr on_tls_init(websocketpp::connection_hdl) {
|
|
m_tls_init = std::chrono::high_resolution_clock::now();
|
|
context_ptr ctx = websocketpp::lib::make_shared<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::no_sslv3 |
|
|
boost::asio::ssl::context::single_dh_use);
|
|
} catch (std::exception& e) {
|
|
std::cout << e.what() << std::endl;
|
|
}
|
|
return ctx;
|
|
}
|
|
|
|
void on_fail(websocketpp::connection_hdl hdl) {
|
|
client::connection_ptr con = m_endpoint.get_con_from_hdl(hdl);
|
|
|
|
std::cout << "Fail handler" << std::endl;
|
|
std::cout << con->get_state() << std::endl;
|
|
std::cout << con->get_local_close_code() << std::endl;
|
|
std::cout << con->get_local_close_reason() << std::endl;
|
|
std::cout << con->get_remote_close_code() << std::endl;
|
|
std::cout << con->get_remote_close_reason() << std::endl;
|
|
std::cout << con->get_ec() << " - " << con->get_ec().message() << std::endl;
|
|
}
|
|
|
|
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) {
|
|
m_message = std::chrono::high_resolution_clock::now();
|
|
m_endpoint.close(hdl,websocketpp::close::status::going_away,"");
|
|
}
|
|
void on_close(websocketpp::connection_hdl) {
|
|
m_close = std::chrono::high_resolution_clock::now();
|
|
|
|
std::cout << "Socket Init: " << std::chrono::duration_cast<dur_type>(m_socket_init-m_start).count() << std::endl;
|
|
std::cout << "TLS Init: " << std::chrono::duration_cast<dur_type>(m_tls_init-m_start).count() << std::endl;
|
|
std::cout << "Open: " << std::chrono::duration_cast<dur_type>(m_open-m_start).count() << std::endl;
|
|
std::cout << "Message: " << std::chrono::duration_cast<dur_type>(m_message-m_start).count() << std::endl;
|
|
std::cout << "Close: " << std::chrono::duration_cast<dur_type>(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;
|
|
}
|
|
}
|