server.cpp File Reference

#include <boost/test/unit_test.hpp>
#include <iostream>
#include <websocketpp/config/core.hpp>
#include <websocketpp/server.hpp>

Include dependency graph for server.cpp:

Go to the source code of this file.

Macros
#define	BOOST_TEST_MODULE   server

Typedefs
typedef websocketpp::server< websocketpp::config::core >	server

Functions
std::string	run_server_test (server &s, std::string input)
void	echo_func (server *s, websocketpp::connection_hdl hdl, message_ptr msg)
bool	validate_func_subprotocol (server *s, std::string *out, std::string accept, websocketpp::connection_hdl hdl)
void	open_func_subprotocol (server *s, std::string *out, websocketpp::connection_hdl hdl)
	BOOST_AUTO_TEST_CASE (basic_websocket_request)
	BOOST_AUTO_TEST_CASE (invalid_websocket_version)
	BOOST_AUTO_TEST_CASE (unimplemented_websocket_version)
	BOOST_AUTO_TEST_CASE (list_subprotocol_empty)
	BOOST_AUTO_TEST_CASE (list_subprotocol_one)
	BOOST_AUTO_TEST_CASE (accept_subprotocol_one)
	BOOST_AUTO_TEST_CASE (accept_subprotocol_invalid)
	BOOST_AUTO_TEST_CASE (accept_subprotocol_two)

Macro Definition Documentation

BOOST_TEST_MODULE

#define BOOST_TEST_MODULE   server

Definition at line 28 of file server.cpp.

Typedef Documentation

server

typedef websocketpp::server<websocketpp::config::core> server

Definition at line 38 of file server.cpp.

Function Documentation

BOOST_AUTO_TEST_CASE() [1/8]

BOOST_AUTO_TEST_CASE

(

accept_subprotocol_invalid

)

Definition at line 203 of file server.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\nSec-WebSocket-Protocol: foo\r\n\r\n";
 
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nSec-WebSocket-Protocol: foo\r\nServer: test\r\nUpgrade: websocket\r\n\r\n";
 
    std::string validate;
    std::string open;
 
    server s;
    s.set_user_agent("test");
    s.set_validate_handler(bind(&validate_func_subprotocol,&s,&validate,"foo2",::_1));
    s.set_open_handler(bind(&open_func_subprotocol,&s,&open,::_1));
 
    std::string o;
 
    BOOST_CHECK_THROW(o = run_server_test(s,input), websocketpp::exception);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [2/8]

BOOST_AUTO_TEST_CASE

(

accept_subprotocol_one

)

Definition at line 185 of file server.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\nSec-WebSocket-Protocol: foo\r\n\r\n";
 
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nSec-WebSocket-Protocol: foo\r\nServer: test\r\nUpgrade: websocket\r\n\r\n";
 
    std::string validate;
    std::string open;
 
    server s;
    s.set_user_agent("test");
    s.set_validate_handler(bind(&validate_func_subprotocol,&s,&validate,"foo",::_1));
    s.set_open_handler(bind(&open_func_subprotocol,&s,&open,::_1));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
    BOOST_CHECK_EQUAL(validate, "foo,");
    BOOST_CHECK_EQUAL(open, "foo");
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [3/8]

BOOST_AUTO_TEST_CASE

(

accept_subprotocol_two

)

Definition at line 221 of file server.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\nSec-WebSocket-Protocol: foo, bar\r\n\r\n";
 
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nSec-WebSocket-Protocol: bar\r\nServer: test\r\nUpgrade: websocket\r\n\r\n";
 
    std::string validate;
    std::string open;
 
    server s;
    s.set_user_agent("test");
    s.set_validate_handler(bind(&validate_func_subprotocol,&s,&validate,"bar",::_1));
    s.set_open_handler(bind(&open_func_subprotocol,&s,&open,::_1));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
    BOOST_CHECK_EQUAL(validate, "foo,bar,");
    BOOST_CHECK_EQUAL(open, "bar");
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [4/8]

BOOST_AUTO_TEST_CASE

(

basic_websocket_request

)

Definition at line 120 of file server.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: test\r\nUpgrade: websocket\r\n\r\n";
 
    server s;
    s.set_user_agent("test");
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [5/8]

BOOST_AUTO_TEST_CASE

(

invalid_websocket_version

)

Definition at line 130 of file server.cpp.

                                                  {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: a\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
    std::string output = "HTTP/1.1 400 Bad Request\r\nServer: test\r\n\r\n";
 
    server s;
    s.set_user_agent("test");
    //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() [6/8]

BOOST_AUTO_TEST_CASE

(

list_subprotocol_empty

)

Definition at line 152 of file server.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\nSec-WebSocket-Protocol: foo\r\n\r\n";
 
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nServer: test\r\nUpgrade: websocket\r\n\r\n";
 
    std::string subprotocol;
 
    server s;
    s.set_user_agent("test");
    s.set_open_handler(bind(&open_func_subprotocol,&s,&subprotocol,::_1));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
    BOOST_CHECK_EQUAL(subprotocol, "");
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [7/8]

BOOST_AUTO_TEST_CASE

(

list_subprotocol_one

)

Definition at line 167 of file server.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\nSec-WebSocket-Protocol: foo\r\n\r\n";
 
    std::string output = "HTTP/1.1 101 Switching Protocols\r\nConnection: upgrade\r\nSec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\nServer: test\r\nUpgrade: websocket\r\n\r\n";
 
    std::string validate;
    std::string open;
 
    server s;
    s.set_user_agent("test");
    s.set_validate_handler(bind(&validate_func_subprotocol,&s,&validate,"",::_1));
    s.set_open_handler(bind(&open_func_subprotocol,&s,&open,::_1));
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
    BOOST_CHECK_EQUAL(validate, "foo,");
    BOOST_CHECK_EQUAL(open, "");
}

Here is the call graph for this function:

BOOST_AUTO_TEST_CASE() [8/8]

BOOST_AUTO_TEST_CASE

(

unimplemented_websocket_version

)

Definition at line 141 of file server.cpp.

                                                        {
    std::string input = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 14\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nOrigin: http://www.example.com\r\n\r\n";
 
    std::string output = "HTTP/1.1 400 Bad Request\r\nSec-WebSocket-Version: 0,7,8,13\r\nServer: test\r\n\r\n";
 
    server s;
    s.set_user_agent("test");
 
    BOOST_CHECK_EQUAL(run_server_test(s,input), output);
}

Here is the call graph for this function:

echo_func()

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

Definition at line 86 of file server.cpp.

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

open_func_subprotocol()

void open_func_subprotocol	(	server *	s,
		std::string *	out,
		websocketpp::connection_hdl	hdl )

Definition at line 113 of file server.cpp.

                                                                                     {
    server::connection_ptr con = s->get_con_from_hdl(hdl);
 
    *out = con->get_subprotocol();
}

Here is the caller graph for this function:

run_server_test()

std::string run_server_test	(	server &	s,
		std::string	input )

Definition at line 66 of file server.cpp.

                                                      {
    server::connection_ptr con;
    std::stringstream output;
 
    s.register_ostream(&output);
    s.clear_access_channels(websocketpp::log::alevel::all);
    s.clear_error_channels(websocketpp::log::elevel::all);
 
    con = s.get_connection();
    con->start();
 
    std::stringstream channel;
 
    channel << input;
    channel >> *con;
 
    return output.str();
}

Here is the caller graph for this function:

validate_func_subprotocol()

bool validate_func_subprotocol	(	server *	s,
		std::string *	out,
		std::string	accept,
		websocketpp::connection_hdl	hdl )

Definition at line 90 of file server.cpp.

{
    server::connection_ptr con = s->get_con_from_hdl(hdl);
 
    std::stringstream o;
 
    const std::vector<std::string> & protocols = con->get_requested_subprotocols();
    std::vector<std::string>::const_iterator it;
 
    for (it = protocols.begin(); it != protocols.end(); ++it) {
        o << *it << ",";
    }
 
    *out = o.str();
 
    if (!accept.empty()) {
        con->select_subprotocol(accept);
    }
 
    return true;
}

Here is the caller graph for this function: