Wire Sysio Wire Sysion 1.0.0
Loading...
Searching...
No Matches
pool.cpp
Go to the documentation of this file.
1/*
2 * Copyright (c) 2011, Peter Thorson. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 * * Redistributions of source code must retain the above copyright
7 * notice, this list of conditions and the following disclaimer.
8 * * Redistributions in binary form must reproduce the above copyright
9 * notice, this list of conditions and the following disclaimer in the
10 * documentation and/or other materials provided with the distribution.
11 * * Neither the name of the WebSocket++ Project nor the
12 * names of its contributors may be used to endorse or promote products
13 * derived from this software without specific prior written permission.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
19 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 *
26 */
27//#define BOOST_TEST_DYN_LINK
28#define BOOST_TEST_MODULE hybi_00_processor
29#include <boost/test/unit_test.hpp>
30
31#include <iostream>
32#include <string>
33
34#include <websocketpp/processors/hybi00.hpp>
35#include <websocketpp/http/request.hpp>
36#include <websocketpp/http/response.hpp>
37
38
39BOOST_AUTO_TEST_CASE( exact_match ) {
40 websocketpp::http::parser::request r;
41 websocketpp::http::parser::response response;
42 websocketpp::processor::hybi00<websocketpp::http::parser::request,websocketpp::http::parser::response> p(false);
43
44 std::string handshake = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nOrigin: http://example.com\r\nSec-WebSocket-Key1: 3e6b263 4 17 80\r\nSec-WebSocket-Key2: 17 9 G`ZD9 2 2b 7X 3 /r90\r\n\r\n";
45
46 r.consume(handshake.c_str(),handshake.size());
47 r.replace_header("Sec-WebSocket-Key3","WjN}|M(6");
48
49 BOOST_CHECK(websocketpp::processor::is_websocket_handshake(r));
50 BOOST_CHECK(websocketpp::processor::get_websocket_version(r) == p.get_version());
51 BOOST_CHECK(p.validate_handshake(r));
52
53 websocketpp::uri_ptr u;
54 bool exception;
55
56 try {
57 u = p.get_uri(r);
58 } catch (const websocketpp::uri_exception& e) {
59 exception = true;
60 }
61
62 BOOST_CHECK(exception == false);
63 BOOST_CHECK(u->get_secure() == false);
64 BOOST_CHECK(u->get_host() == "www.example.com");
65 BOOST_CHECK(u->get_resource() == "/");
66 BOOST_CHECK(u->get_port() == websocketpp::URI_DEFAULT_PORT);
67
68 p.process_handshake(r,response);
69
70 BOOST_CHECK(response.get_header("Connection") == "Upgrade");
71 BOOST_CHECK(response.get_header("Upgrade") == "websocket");
72 BOOST_CHECK(response.get_header("Sec-WebSocket-Origin") == "http://example.com");
73
74 BOOST_CHECK(response.get_header("Sec-WebSocket-Location") == "ws://www.example.com/");
75 BOOST_CHECK(response.get_header("Sec-WebSocket-Key3") == "n`9eBk9z$R8pOtVb");
76}
77
78BOOST_AUTO_TEST_CASE( non_get_method ) {
79 websocketpp::http::parser::request r;
80 websocketpp::processor::hybi00<websocketpp::http::parser::request,websocketpp::http::parser::response> p(false);
81
82 std::string handshake = "POST / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Key1: 3e6b263 4 17 80\r\nSec-WebSocket-Key2: 17 9 G`ZD9 2 2b 7X 3 /r90\r\n\r\n";
83
84 r.consume(handshake.c_str(),handshake.size());
85 r.replace_header("Sec-WebSocket-Key3","janelle!");
86
87 BOOST_CHECK(websocketpp::processor::is_websocket_handshake(r));
88 BOOST_CHECK(websocketpp::processor::get_websocket_version(r) == p.get_version());
89 BOOST_CHECK(!p.validate_handshake(r));
90}
91
92BOOST_AUTO_TEST_CASE( old_http_version ) {
93 websocketpp::http::parser::request r;
94 websocketpp::processor::hybi00<websocketpp::http::parser::request,websocketpp::http::parser::response> p(false);
95
96 std::string handshake = "GET / HTTP/1.0\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Key1: 3e6b263 4 17 80\r\nSec-WebSocket-Key2: 17 9 G`ZD9 2 2b 7X 3 /r90\r\n\r\n";
97
98 r.consume(handshake.c_str(),handshake.size());
99 r.replace_header("Sec-WebSocket-Key3","janelle!");
100
101 BOOST_CHECK(websocketpp::processor::is_websocket_handshake(r));
102 BOOST_CHECK(websocketpp::processor::get_websocket_version(r) == p.get_version());
103 BOOST_CHECK(!p.validate_handshake(r));
104}
105
106BOOST_AUTO_TEST_CASE( missing_handshake_key1 ) {
107 websocketpp::http::parser::request r;
108 websocketpp::processor::hybi00<websocketpp::http::parser::request,websocketpp::http::parser::response> p(false);
109
110 std::string handshake = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Key1: 3e6b263 4 17 80\r\n\r\n";
111
112 r.consume(handshake.c_str(),handshake.size());
113 r.replace_header("Sec-WebSocket-Key3","janelle!");
114
115 BOOST_CHECK(websocketpp::processor::is_websocket_handshake(r));
116 BOOST_CHECK(websocketpp::processor::get_websocket_version(r) == p.get_version());
117 BOOST_CHECK(!p.validate_handshake(r));
118}
119
120BOOST_AUTO_TEST_CASE( missing_handshake_key2 ) {
121 websocketpp::http::parser::request r;
122 websocketpp::processor::hybi00<websocketpp::http::parser::request,websocketpp::http::parser::response> p(false);
123
124 std::string handshake = "GET / HTTP/1.1\r\nHost: www.example.com\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Key2: 17 9 G`ZD9 2 2b 7X 3 /r90\r\n\r\n";
125
126 r.consume(handshake.c_str(),handshake.size());
127 r.replace_header("Sec-WebSocket-Key3","janelle!");
128
129 BOOST_CHECK(websocketpp::processor::is_websocket_handshake(r));
130 BOOST_CHECK(websocketpp::processor::get_websocket_version(r) == p.get_version());
131 BOOST_CHECK(!p.validate_handshake(r));
132}
133
134BOOST_AUTO_TEST_CASE( bad_host ) {
135 websocketpp::http::parser::request r;
136 websocketpp::processor::hybi00<websocketpp::http::parser::request,websocketpp::http::parser::response> p(false);
137 websocketpp::uri_ptr u;
138 bool exception = false;
139
140 std::string handshake = "GET / HTTP/1.1\r\nHost: www.example.com:70000\r\nConnection: upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Key2: 17 9 G`ZD9 2 2b 7X 3 /r90\r\n\r\n";
141
142 r.consume(handshake.c_str(),handshake.size());
143 r.replace_header("Sec-WebSocket-Key3","janelle!");
144
145 BOOST_CHECK(websocketpp::processor::is_websocket_handshake(r));
146 BOOST_CHECK(websocketpp::processor::get_websocket_version(r) == p.get_version());
147 BOOST_CHECK(!p.validate_handshake(r));
148
149 try {
150 u = p.get_uri(r);
151 } catch (const websocketpp::uri_exception& e) {
152 exception = true;
153 }
154
155 BOOST_CHECK(exception == true);
156}
p
const mie::Vuint & p
Definition bn.cpp:27
r
const mie::Vuint & r
Definition bn.cpp:28
websocketpp::http::parser::request
Stores, parses, and manipulates HTTP requests.
Definition request.hpp:50
websocketpp::http::parser::response
Stores, parses, and manipulates HTTP responses.
Definition response.hpp:57
websocketpp::processor::hybi00
Processor for Hybi Draft version 00.
Definition hybi00.hpp:54
websocketpp::processor::get_websocket_version
int get_websocket_version(request_type &r)
Extract the version from a WebSocket handshake request.
Definition processor.hpp:107
websocketpp::processor::is_websocket_handshake
bool is_websocket_handshake(request_type &r)
Determine whether or not a generic HTTP request is a WebSocket handshake.
Definition processor.hpp:68
websocketpp::uri_ptr
lib::shared_ptr< uri > uri_ptr
Pointer to a URI.
Definition uri.hpp:351
BOOST_AUTO_TEST_CASE
BOOST_AUTO_TEST_CASE(exact_match)
Definition pool.cpp:39