WebSocket++ 0.8.2
C++ websocket client/server library
Loading...
Searching...
No Matches
uri.hpp
1/*
2 * Copyright (c) 2014, 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
28#ifndef WEBSOCKETPP_URI_HPP
29#define WEBSOCKETPP_URI_HPP
30
31#include <websocketpp/error.hpp>
32
33#include <websocketpp/common/memory.hpp>
34#include <websocketpp/common/stdint.hpp>
35
36#include <algorithm>
37#include <sstream>
38#include <string>
39
40namespace websocketpp {
41
42// TODO: figure out why this fixes horrible linking errors.
43
45static uint16_t const uri_default_port = 80;
47static uint16_t const uri_default_secure_port = 443;
48
49class uri {
50public:
51 explicit uri(std::string const & uri_string) : m_valid(false) {
52 std::string::const_iterator it;
53 std::string::const_iterator temp;
54
55 int state = 0;
56
57 it = uri_string.begin();
58 size_t uri_len = uri_string.length();
59
60 if (uri_len >= 7 && std::equal(it,it+6,"wss://")) {
61 m_secure = true;
62 m_scheme = "wss";
63 it += 6;
64 } else if (uri_len >= 6 && std::equal(it,it+5,"ws://")) {
65 m_secure = false;
66 m_scheme = "ws";
67 it += 5;
68 } else if (uri_len >= 8 && std::equal(it,it+7,"http://")) {
69 m_secure = false;
70 m_scheme = "http";
71 it += 7;
72 } else if (uri_len >= 9 && std::equal(it,it+8,"https://")) {
73 m_secure = true;
74 m_scheme = "https";
75 it += 8;
76 } else {
77 return;
78 }
79
80 // extract host.
81 // either a host string
82 // an IPv4 address
83 // or an IPv6 address
84 if (*it == '[') {
85 ++it;
86 // IPv6 literal
87 // extract IPv6 digits until ]
88
89 // TODO: this doesn't work on g++... not sure why
90 //temp = std::find(it,it2,']');
91
92 temp = it;
93 while (temp != uri_string.end()) {
94 if (*temp == ']') {
95 break;
96 }
97 ++temp;
98 }
99
100 if (temp == uri_string.end()) {
101 return;
102 } else {
103 // validate IPv6 literal parts
104 // can contain numbers, a-f and A-F
105 m_host.append(it,temp);
106 }
107 it = temp+1;
108 if (it == uri_string.end()) {
109 state = 2;
110 } else if (*it == '/') {
111 state = 2;
112 ++it;
113 } else if (*it == ':') {
114 state = 1;
115 ++it;
116 } else {
117 // problem
118 return;
119 }
120 } else {
121 // IPv4 or hostname
122 // extract until : or /
123 while (state == 0) {
124 if (it == uri_string.end()) {
125 state = 2;
126 break;
127 } else if (*it == '/') {
128 state = 2;
129 } else if (*it == ':') {
130 // end hostname start port
131 state = 1;
132 } else {
133 m_host += *it;
134 }
135 ++it;
136 }
137 }
138
139 // parse port
140 std::string port;
141 while (state == 1) {
142 if (it == uri_string.end()) {
143 // state is not used after this point presently.
144 // this should be re-enabled if it ever is needed in a future
145 // refactoring
146 //state = 3;
147 break;
148 } else if (*it == '/') {
149 state = 3;
150 } else {
151 port += *it;
152 }
153 ++it;
154 }
155
156 lib::error_code ec;
157 m_port = get_port_from_string(port, ec);
158
159 if (ec) {
160 return;
161 }
162
163 m_resource = "/";
164 m_resource.append(it,uri_string.end());
165
166
167 m_valid = true;
168 }
169
170 uri(bool secure, std::string const & host, uint16_t port,
171 std::string const & resource)
172 : m_scheme(secure ? "wss" : "ws")
173 , m_host(host)
174 , m_resource(resource.empty() ? "/" : resource)
175 , m_port(port)
176 , m_secure(secure)
177 , m_valid(true) {}
178
179 uri(bool secure, std::string const & host, std::string const & resource)
180 : m_scheme(secure ? "wss" : "ws")
181 , m_host(host)
182 , m_resource(resource.empty() ? "/" : resource)
183 , m_port(secure ? uri_default_secure_port : uri_default_port)
184 , m_secure(secure)
185 , m_valid(true) {}
186
187 uri(bool secure, std::string const & host, std::string const & port,
188 std::string const & resource)
189 : m_scheme(secure ? "wss" : "ws")
190 , m_host(host)
191 , m_resource(resource.empty() ? "/" : resource)
192 , m_secure(secure)
193 {
194 lib::error_code ec;
195 m_port = get_port_from_string(port,ec);
196 m_valid = !ec;
197 }
198
199 uri(std::string const & scheme, std::string const & host, uint16_t port,
200 std::string const & resource)
201 : m_scheme(scheme)
202 , m_host(host)
203 , m_resource(resource.empty() ? "/" : resource)
204 , m_port(port)
205 , m_secure(scheme == "wss" || scheme == "https")
206 , m_valid(true) {}
207
208 uri(std::string scheme, std::string const & host, std::string const & resource)
209 : m_scheme(scheme)
210 , m_host(host)
211 , m_resource(resource.empty() ? "/" : resource)
212 , m_port((scheme == "wss" || scheme == "https") ? uri_default_secure_port : uri_default_port)
213 , m_secure(scheme == "wss" || scheme == "https")
214 , m_valid(true) {}
215
216 uri(std::string const & scheme, std::string const & host,
217 std::string const & port, std::string const & resource)
218 : m_scheme(scheme)
219 , m_host(host)
220 , m_resource(resource.empty() ? "/" : resource)
221 , m_secure(scheme == "wss" || scheme == "https")
222 {
223 lib::error_code ec;
224 m_port = get_port_from_string(port,ec);
225 m_valid = !ec;
226 }
227
228 bool get_valid() const {
229 return m_valid;
230 }
231
232 bool get_secure() const {
233 return m_secure;
234 }
235
236 std::string const & get_scheme() const {
237 return m_scheme;
238 }
239
240 std::string const & get_host() const {
241 return m_host;
242 }
243
244 std::string get_host_port() const {
245 if (m_port == (m_secure ? uri_default_secure_port : uri_default_port)) {
246 return m_host;
247 } else {
248 std::stringstream p;
249 p << m_host << ":" << m_port;
250 return p.str();
251 }
252 }
253
254 std::string get_authority() const {
255 std::stringstream p;
256 p << m_host << ":" << m_port;
257 return p.str();
258 }
259
260 uint16_t get_port() const {
261 return m_port;
262 }
263
264 std::string get_port_str() const {
265 std::stringstream p;
266 p << m_port;
267 return p.str();
268 }
269
270 std::string const & get_resource() const {
271 return m_resource;
272 }
273
274 std::string str() const {
275 std::stringstream s;
276
277 s << m_scheme << "://" << m_host;
278
279 if (m_port != (m_secure ? uri_default_secure_port : uri_default_port)) {
280 s << ":" << m_port;
281 }
282
283 s << m_resource;
284 return s.str();
285 }
286
288
294 std::string get_query() const {
295 std::size_t found = m_resource.find('?');
296 if (found != std::string::npos) {
297 return m_resource.substr(found + 1);
298 } else {
299 return "";
300 }
301 }
302
303 // get fragment
304
305 // hi <3
306
307 // get the string representation of this URI
308
309 //std::string base() const; // is this still needed?
310
311 // setter methods set some or all (in the case of parse) based on the input.
312 // These functions throw a uri_exception on failure.
313 /*void set_uri(const std::string& uri);
314
315 void set_secure(bool secure);
316 void set_host(const std::string& host);
317 void set_port(uint16_t port);
318 void set_port(const std::string& port);
319 void set_resource(const std::string& resource);*/
320private:
321 uint16_t get_port_from_string(std::string const & port, lib::error_code &
322 ec) const
323 {
324 ec = lib::error_code();
325
326 if (port.empty()) {
327 return (m_secure ? uri_default_secure_port : uri_default_port);
328 }
329
330 unsigned int t_port = static_cast<unsigned int>(atoi(port.c_str()));
331
332 if (t_port > 65535) {
333 ec = error::make_error_code(error::invalid_port);
334 }
335
336 if (t_port == 0) {
337 ec = error::make_error_code(error::invalid_port);
338 }
339
340 return static_cast<uint16_t>(t_port);
341 }
342
343 std::string m_scheme;
344 std::string m_host;
345 std::string m_resource;
346 uint16_t m_port;
347 bool m_secure;
348 bool m_valid;
349};
350
352typedef lib::shared_ptr<uri> uri_ptr;
353
354} // namespace websocketpp
355
356#endif // WEBSOCKETPP_URI_HPP
std::string get_query() const
Return the query portion.
Definition uri.hpp:294
@ invalid_port
Invalid port in URI.
Definition error.hpp:120
Namespace for the WebSocket++ project.
Definition base64.hpp:41
static uint16_t const uri_default_secure_port
Default port for wss://.
Definition uri.hpp:47
static uint16_t const uri_default_port
Default port for ws://.
Definition uri.hpp:45
lib::shared_ptr< uri > uri_ptr
Pointer to a URI.
Definition uri.hpp:352