1/*
2 * libwebsockets - small server side websockets and web server implementation
3 *
4 * Copyright (C) 2020 Andy Green <andy@warmcat.com>
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
22 * IN THE SOFTWARE.
23 *
24 * C++ classes for Secure Streams
25 */
26
27#include <libwebsockets.hxx>
28
29static const char *pcols[] = {
30	"http://",	/* LWSSSP_H1 */
31	"https://",
32	"h2://",	/* LWSSSP_H2 */
33	"h2s://",
34	"ws://",	/* LWSSSP_WS */
35	"wss://",
36	"mqtt://",	/* LWSSSP_MQTT */
37	"mqtts://",
38	"raw://",	/* LWSSSP_RAW */
39	"raws://",
40};
41
42static const uint8_t pcols_len[] = {
43	7, 8, 5, 6, 5, 6, 7, 8, 6, 7
44};
45
46static const uint16_t pcols_port[] = {
47	80, 443, 443, 443, 80, 443, 1883, 8883, 80, 443
48};
49
50lss::lss(lws_ctx_t _ctx, std::string _uri, lsscomp_t _comp, bool _psh,
51	 lws_sscb_rx rx, lws_sscb_tx tx, lws_sscb_state state)
52{
53	const char *p, *urlpath;
54	lws_ss_info_t ssi;
55	int n, port;
56
57	memset(&ssi, 0, sizeof(ssi));
58	memset(&pol, 0, sizeof(pol));
59
60	ctx		= _ctx;
61	comp		= _comp;
62	comp_done	= 0;
63	rxlen		= 0;
64
65	/*
66	 * We have a common stub userdata, our "real" userdata is in the
67	 * derived class members.   The Opaque user pointer points to the
68	 * lss itself.
69	 */
70
71	ssi.handle_offset	    = offsetof(lssPriv, lssPriv::m_ss);
72	ssi.opaque_user_data_offset = offsetof(lssPriv, lssPriv::m_plss);
73
74	ssi.user_alloc	= sizeof(lssPriv);
75	ssi.rx		= rx;
76	ssi.tx		= tx;
77	ssi.state	= state;
78	ssi.policy	= &pol; /* we will provide our own policy */
79
80	/*
81	 * _uri is like "https://warmcat.com:443/index.html"... we need to
82	 * deconstruct it into its policy implications
83	 */
84
85	uri = strdup(_uri.c_str());
86
87	for (n = 0; n < LWS_ARRAY_SIZE(pcols); n++)
88		if (!strncmp(uri, pcols[n], pcols_len[n]))
89			break;
90
91	if (n == LWS_ARRAY_SIZE(pcols))
92		throw lssException("unknown uri protocol://");
93
94	pol.protocol = n >> 1;
95	if (n & 1)
96		pol.flags |= LWSSSPOLF_TLS;
97
98	n = pcols_port[n];
99
100	if (lws_parse_uri(uri, &p, &pol.endpoint, &n, &urlpath))
101		throw lssException("unable to parse uri://");
102
103	pol.port = (uint16_t)n;
104
105	if (pol.protocol <= LWSSSP_WS) {
106		pol.u.http.url = urlpath;
107
108		/*
109		 * These are workarounds for common h2 server noncompliances
110		 */
111
112		pol.flags |= LWSSSPOLF_QUIRK_NGHTTP2_END_STREAM |
113			     LWSSSPOLF_H2_QUIRK_OVERFLOWS_TXCR |
114			     LWSSSPOLF_H2_QUIRK_UNCLEAN_HPACK_STATE;
115
116		if (pol.protocol < LWSSSP_WS)
117			pol.u.http.method = _psh ? "POST" : "GET";
118	}
119
120	us_start = lws_now_usecs();
121
122	if (lws_ss_create(ctx, 0, &ssi, (void *)this, &m_ss, NULL, NULL))
123		goto blow;
124
125	if (pol.protocol <= LWSSSP_WS)
126		lws_ss_client_connect(m_ss);
127
128	return;
129
130blow:
131	if (uri)
132		free(uri);
133	throw lssException("ss creation failed");
134}
135
136lss::~lss()
137{
138	if (uri)
139		free(uri);
140	if (m_ss)
141		lws_ss_destroy(&m_ss);
142}
143
144int lss::call_completion(lws_ss_constate_t state)
145{
146	if (comp_done)
147		return 0;
148	if (!comp)
149		return 0;
150
151	comp_done = 1;
152
153	return comp(this, state, NULL);
154}
155