connection.cpp File Reference

#include <boost/test/unit_test.hpp>
#include "connection_tu2.hpp"
#include <websocketpp/transport/debug/endpoint.hpp>

Include dependency graph for connection.cpp:

Go to the source code of this file.

Classes
struct	connection_extension
struct	stub_config
struct	debug_config_client
struct	debug_config_client::transport_config
struct	connection_setup

Macros
#define	BOOST_TEST_MODULE   connection

Typedefs
typedef websocketpp::client< debug_config_client >	debug_client
typedef websocketpp::server< debug_config_client >	debug_server

Functions
	BOOST_AUTO_TEST_CASE (basic_http_request)
void	validate_func (server *s, websocketpp::connection_hdl hdl, message_ptr msg)
bool	validate_set_ua (server *s, websocketpp::connection_hdl hdl)
void	http_func (server *s, websocketpp::connection_hdl hdl)
void	defer_http_func (server *s, bool *deferred, websocketpp::connection_hdl hdl)
void	check_on_fail (server *s, websocketpp::lib::error_code ec, bool &called, websocketpp::connection_hdl hdl)
void	on_open_print (server *s, websocketpp::connection_hdl hdl)
void	fail_on_open (websocketpp::connection_hdl)
void	fail_on_http (websocketpp::connection_hdl)
	BOOST_AUTO_TEST_CASE (connection_extensions)
	BOOST_AUTO_TEST_CASE (basic_websocket_request)
	BOOST_AUTO_TEST_CASE (http_request)
	BOOST_AUTO_TEST_CASE (deferred_http_request)
	BOOST_AUTO_TEST_CASE (request_no_server_header)
	BOOST_AUTO_TEST_CASE (request_no_server_header_override)
	BOOST_AUTO_TEST_CASE (basic_client_websocket)
	BOOST_AUTO_TEST_CASE (set_max_message_size)
	BOOST_AUTO_TEST_CASE (websocket_fail_parse_error)
	BOOST_AUTO_TEST_CASE (websocket_fail_invalid_version)
	BOOST_AUTO_TEST_CASE (websocket_fail_unsupported_version)
	BOOST_AUTO_TEST_CASE (websocket_fail_upgrade_required)
	BOOST_AUTO_TEST_CASE (client_handshake_timeout_race1)
	BOOST_AUTO_TEST_CASE (client_handshake_timeout_race2)
	BOOST_AUTO_TEST_CASE (server_handshake_timeout_race1)
	BOOST_AUTO_TEST_CASE (server_handshake_timeout_race2)

Macro Definition Documentation

BOOST_TEST_MODULE

#define BOOST_TEST_MODULE   connection

Definition at line 28 of file connection.cpp.

Typedef Documentation

debug_client

typedef websocketpp::client<debug_config_client> debug_client

Definition at line 144 of file connection.cpp.

debug_server

typedef websocketpp::server<debug_config_client> debug_server

Definition at line 145 of file connection.cpp.

Function Documentation

BOOST_AUTO_TEST_CASE() [1/17]

BOOST_AUTO_TEST_CASE

(

basic_client_websocket

)

Definition at line 301 of file connection.cpp.

                                               {
    std::string uri = "ws://localhost";
 
    //std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nServer: foo\r\nUpgrade: websocket\r\n\r\n";
 
    std::string ref = "GET / HTTP/1.1\r\nConnection: Upgrade\r\nFoo: Bar\r\nHost: localhost\r\nSec-WebSocket-Key: AAAAAAAAAAAAAAAAAAAAAA==\r\nSec-WebSocket-Version: 13\r\nUpgrade: websocket\r\nUser-Agent: foo\r\n\r\n";
 
    std::stringstream output;
 
    client e;
    e.set_access_channels(websocketpp::log::alevel::none);
    e.set_error_channels(websocketpp::log::elevel::none);
    e.set_user_agent("foo");
    e.register_ostream(&output);
 
    client::connection_ptr con;
    websocketpp::lib::error_code ec;
    con = e.get_connection(uri, ec);
    con->append_header("Foo","Bar");
    e.connect(con);
 
    BOOST_CHECK_EQUAL(ref, output.str());
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [2/17]

BOOST_AUTO_TEST_CASE

(

basic_http_request

)

Definition at line 41 of file connection.cpp.

                                           {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 426 Upgrade Required\r\nServer: " +
                         std::string(websocketpp::user_agent)+"\r\n\r\n";
 
    std::string o2 = run_server_test(input);
 
    BOOST_CHECK(o2 == output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [3/17]

BOOST_AUTO_TEST_CASE

(

basic_websocket_request

)

Definition at line 216 of file connection.cpp.

                                                {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nServer: ";
    output+=websocketpp::user_agent;
    output+="\r\nUpgrade: websocket\r\n\r\n";
 
    server s;
    s.set_message_handler(bind(&echo_func,&s,::_1,::_2));
 
    BOOST_CHECK(run_server_test(s,input) == output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [4/17]

BOOST_AUTO_TEST_CASE

(

client_handshake_timeout_race1

)

Definition at line 451 of file connection.cpp.

                                                       {
    debug_client c;
 
    websocketpp::lib::error_code ec;
    debug_client::connection_ptr con = c.get_connection("ws://localhost:9002", ec);
 
    BOOST_CHECK(!ec);
 
    // This test the case where a handshake times out immediately before the 
    // handler that would have completed it gets invoked. This situation happens
    // when clients are connecting to overloaded servers and on servers that are
    // overloaded. 
    c.connect(con);
    
    con->expire_timer(websocketpp::lib::error_code());
    // Fullfil the write to simulate the write completing immediately after
    // timer expires
    con->fullfil_write();
    
    BOOST_CHECK_EQUAL(con->get_ec(), make_error_code(websocketpp::error::open_handshake_timeout));
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [5/17]

BOOST_AUTO_TEST_CASE

(

client_handshake_timeout_race2

)

Definition at line 473 of file connection.cpp.

                                                       {
    debug_client c;
 
    websocketpp::lib::error_code ec;
    debug_client::connection_ptr con = c.get_connection("ws://localhost:9002", ec);
 
    BOOST_CHECK(!ec);
 
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: ICX+Yqv66kxgM0FcWaLWlFLwTAI=\r\nServer: foo\r\nUpgrade: websocket\r\n\r\n";
 
    // This test the case where a handshake times out immediately before the 
    // handler that would have completed it gets invoked. This situation happens
    // when clients are connecting to overloaded servers and on servers that are
    // overloaded. 
    c.connect(con);
    con->fullfil_write();
    
    con->expire_timer(websocketpp::lib::error_code());
    // Read valid handshake to simulate receiving the handshake response
    // immediately after the timer expires
    con->read_all(output.data(),output.size());
    
    BOOST_CHECK_EQUAL(con->get_ec(), make_error_code(websocketpp::error::open_handshake_timeout));
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [6/17]

BOOST_AUTO_TEST_CASE

(

connection_extensions

)

Definition at line 207 of file connection.cpp.

                                              {
    connection_setup env(true);
 
    BOOST_CHECK( env.c.extension_value == 5 );
    BOOST_CHECK( env.c.extension_method() == 5 );
 
    BOOST_CHECK( env.c.is_server() == true );
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [7/17]

BOOST_AUTO_TEST_CASE

(

deferred_http_request

)

Definition at line 240 of file connection.cpp.

                                              {
    std::string input = "GET /foo/bar HTTP/1.1\r\nHost: www.example.com\r\nOrigin: http://www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 200 OK\r\nContent-Length: 8\r\nServer: ";
    output+=websocketpp::user_agent;
    output+="\r\n\r\n/foo/bar";
 
    server s;
    server::connection_ptr con;
    bool deferred = false;
    s.set_http_handler(bind(&defer_http_func,&s, &deferred,::_1));
 
    s.clear_access_channels(websocketpp::log::alevel::all);
    s.clear_error_channels(websocketpp::log::elevel::all);
    
    std::stringstream ostream;
    s.register_ostream(&ostream);
 
    con = s.get_connection();
    con->start();
    
    BOOST_CHECK(!deferred);
    BOOST_CHECK_EQUAL(ostream.str(), "");
    con->read_some(input.data(),input.size());
    BOOST_CHECK(deferred);
    BOOST_CHECK_EQUAL(ostream.str(), "");
 
    con->set_body(con->get_resource());
    con->set_status(websocketpp::http::status_code::ok);
    
    websocketpp::lib::error_code ec;
    s.send_http_response(con->get_handle(),ec);
    BOOST_CHECK_EQUAL(ec, websocketpp::lib::error_code());
    BOOST_CHECK_EQUAL(ostream.str(), output);
    con->send_http_response(ec);
    BOOST_CHECK_EQUAL(ec, make_error_code(websocketpp::error::invalid_state));
    
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [8/17]

BOOST_AUTO_TEST_CASE

(

http_request

)

Definition at line 228 of file connection.cpp.

                                     {
    std::string input = "GET /foo/bar HTTP/1.1\r\nHost: www.example.com\r\nOrigin: http://www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 200 OK\r\nContent-Length: 8\r\nServer: ";
    output+=websocketpp::user_agent;
    output+="\r\n\r\n/foo/bar";
 
    server s;
    s.set_http_handler(bind(&http_func,&s,::_1));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [9/17]

BOOST_AUTO_TEST_CASE

(

request_no_server_header

)

Definition at line 278 of file connection.cpp.

                                                 {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nUpgrade: websocket\r\n\r\n";
 
    server s;
    s.set_user_agent("");
    s.set_message_handler(bind(&echo_func,&s,::_1,::_2));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [10/17]

BOOST_AUTO_TEST_CASE

(

request_no_server_header_override

)

Definition at line 289 of file connection.cpp.

                                                          {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nServer: foo\r\nUpgrade: websocket\r\n\r\n";
 
    server s;
    s.set_user_agent("");
    s.set_message_handler(bind(&echo_func,&s,::_1,::_2));
    s.set_validate_handler(bind(&validate_set_ua,&s,::_1));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [11/17]

BOOST_AUTO_TEST_CASE

(

server_handshake_timeout_race1

)

Definition at line 498 of file connection.cpp.

                                                       {
    debug_server s;
 
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: AAAAAAAAAAAAAAAAAAAAAA==\r\n\r\n";
 
    debug_server::connection_ptr con = s.get_connection();
    con->start();
    
    con->expire_timer(websocketpp::lib::error_code());
    // Read handshake immediately after timer expire
    con->read_all(input.data(), input.size());
    
    BOOST_CHECK_EQUAL(con->get_ec(), make_error_code(websocketpp::error::open_handshake_timeout));
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [12/17]

BOOST_AUTO_TEST_CASE

(

server_handshake_timeout_race2

)

Definition at line 513 of file connection.cpp.

                                                       {
    debug_server s;
 
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: AAAAAAAAAAAAAAAAAAAAAA==\r\n\r\n";
 
    debug_server::connection_ptr con = s.get_connection();
    con->start();
    
    con->read_all(input.data(), input.size());
    
    con->expire_timer(websocketpp::lib::error_code());
    // Complete write immediately after timer expire
    con->fullfil_write();
    
    BOOST_CHECK_EQUAL(con->get_ec(), make_error_code(websocketpp::error::open_handshake_timeout));
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [13/17]

BOOST_AUTO_TEST_CASE

(

set_max_message_size

)

Definition at line 325 of file connection.cpp.

                                             {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n\r\n";
    
    // After the handshake, add a single frame with a message that is too long.
    char frame0[10] = {char(0x82), char(0x83), 0x00, 0x00, 0x00, 0x00, 0x01, 0x01, 0x01};
    input.append(frame0, 10);
    
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nServer: foo\r\nUpgrade: websocket\r\n\r\n";
 
    // After the handshake, add a single frame with a close message with message too big
    // error code.
    char frame1[4] = {char(0x88), 0x19, 0x03, char(0xf1)};
    output.append(frame1, 4);
    output.append("A message was too large");
 
    server s;
    s.set_user_agent("");
    s.set_validate_handler(bind(&validate_set_ua,&s,::_1));
    s.set_max_message_size(2);
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [14/17]

BOOST_AUTO_TEST_CASE

(

websocket_fail_invalid_version

)

Definition at line 360 of file connection.cpp.

                                                       {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: foo\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
 
    server s;
    websocketpp::lib::error_code ec = make_error_code(websocketpp::error::invalid_version);
    bool called = false;
    s.set_fail_handler(bind(&check_on_fail,&s,ec,websocketpp::lib::ref(called),::_1));
 
    run_server_test(s,input,false);
    BOOST_CHECK(called);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [15/17]

BOOST_AUTO_TEST_CASE

(

websocket_fail_parse_error

)

Definition at line 348 of file connection.cpp.

                                                   {
    std::string input = "asdf\r\n\r\n";
 
    server s;
    websocketpp::lib::error_code ec = make_error_code(websocketpp::error::http_parse_error);
    bool called = false;
    s.set_fail_handler(bind(&check_on_fail,&s,ec,websocketpp::lib::ref(called),::_1));
 
    run_server_test(s,input,false);
    BOOST_CHECK(called);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [16/17]

BOOST_AUTO_TEST_CASE

(

websocket_fail_unsupported_version

)

Definition at line 372 of file connection.cpp.

                                                           {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 12\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
 
    server s;
    websocketpp::lib::error_code ec = make_error_code(websocketpp::error::unsupported_version);
    bool called = false;
    s.set_fail_handler(bind(&check_on_fail,&s,ec,websocketpp::lib::ref(called),::_1));
 
    run_server_test(s,input,false);
    BOOST_CHECK(called);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [17/17]

BOOST_AUTO_TEST_CASE

(

websocket_fail_upgrade_required

)

Definition at line 410 of file connection.cpp.

                                                        {
    std::string input = "GET /foo/bar HTTP/1.1\r\nHost: www.example.com\r\nOrigin: http://www.example.com\r\n\r\n";
 
    server s;
    websocketpp::lib::error_code ec = make_error_code(websocketpp::error::upgrade_required);
    bool called = false;
    s.set_fail_handler(bind(&check_on_fail,&s,ec,websocketpp::lib::ref(called),::_1));
 
    run_server_test(s,input,false);
    BOOST_CHECK(called);
}

Here is the call graph for this function:

check_on_fail()

void check_on_fail	(	server *	s,
		websocketpp::lib::error_code	ec,
		bool &	called,
		websocketpp::connection_hdl	hdl )

Definition at line 184 of file connection.cpp.

{
    server::connection_ptr con = s->get_con_from_hdl(hdl);
 
    BOOST_CHECK_EQUAL(ec, con->get_ec());
    called = true;
}

Here is the caller graph for this function:

defer_http_func()

void defer_http_func	(	server *	s,
		bool *	deferred,
		websocketpp::connection_hdl	hdl )

Definition at line 175 of file connection.cpp.

                                                                                {
    *deferred = true;
    
    server::connection_ptr con = s->get_con_from_hdl(hdl);
    
    websocketpp::lib::error_code ec = con->defer_http_response();
    BOOST_CHECK_EQUAL(ec, websocketpp::lib::error_code());
}

Here is the caller graph for this function:

fail_on_http()

void fail_on_http

(

websocketpp::connection_hdl

)

Definition at line 203 of file connection.cpp.

                                             {
    BOOST_CHECK(false);
}

fail_on_open()

void fail_on_open

(

websocketpp::connection_hdl

)

Definition at line 200 of file connection.cpp.

                                             {
    BOOST_CHECK(false);
}

http_func()

void http_func	(	server *	s,
		websocketpp::connection_hdl	hdl )

Definition at line 161 of file connection.cpp.

                                                         {
    using namespace websocketpp::http;
 
    server::connection_ptr con = s->get_con_from_hdl(hdl);
 
    std::string res = con->get_resource();
 
    con->set_body(res);
    con->set_status(status_code::ok);
 
    BOOST_CHECK_EQUAL(con->get_response_code(), status_code::ok);
    BOOST_CHECK_EQUAL(con->get_response_msg(), status_code::get_string(status_code::ok));
}

Here is the caller graph for this function:

on_open_print()

void on_open_print	(	server *	s,
		websocketpp::connection_hdl	hdl )

Definition at line 193 of file connection.cpp.

{
    server::connection_ptr con = s->get_con_from_hdl(hdl);
 
    std::cout << con->get_uri() << std::endl;
}

validate_func()

void validate_func	(	server *	s,
		websocketpp::connection_hdl	hdl,
		message_ptr	msg )

Definition at line 151 of file connection.cpp.

                                                                              {
    s->send(hdl, msg->get_payload(), msg->get_opcode());
}

Here is the call graph for this function:

validate_set_ua()

bool validate_set_ua	(	server *	s,
		websocketpp::connection_hdl	hdl )

Definition at line 155 of file connection.cpp.

                                                               {
    server::connection_ptr con = s->get_con_from_hdl(hdl);
    con->replace_header("Server","foo");
    return true;
}

Here is the caller graph for this function: