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 29 static 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 42 static const uint8_t pcols_len[] = { 43 7, 8, 5, 6, 5, 6, 7, 8, 6, 7 44 }; 45 46 static const uint16_t pcols_port[] = { 47 80, 443, 443, 443, 80, 443, 1883, 8883, 80, 443 48 }; 49 lss(lws_ctx_t _ctx,std::string _uri,lsscomp_t _comp,bool _psh,lws_sscb_rx rx,lws_sscb_tx tx,lws_sscb_state state)50 lss::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 130 blow: 131 if (uri) 132 free(uri); 133 throw lssException("ss creation failed"); 134 } 135 ~lss()136 lss::~lss() 137 { 138 if (uri) 139 free(uri); 140 if (m_ss) 141 lws_ss_destroy(&m_ss); 142 } 143 call_completion(lws_ss_constate_t state)144 int 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