| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417 |
- /*
- * libwebsockets - small server side websockets and web server implementation
- *
- * Copyright (C) 2010 - 2019 Andy Green <[email protected]>
- *
- * Permission is hereby granted, free of charge, to any person obtaining a copy
- * of this software and associated documentation files (the "Software"), to
- * deal in the Software without restriction, including without limitation the
- * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
- * sell copies of the Software, and to permit persons to whom the Software is
- * furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in
- * all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
- * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
- * IN THE SOFTWARE.
- */
- #include "private-lib-core.h"
- #if defined (_DEBUG)
- void lwsi_set_role(struct lws *wsi, lws_wsi_state_t role)
- {
- wsi->wsistate = (wsi->wsistate & (~LWSI_ROLE_MASK)) | role;
- lwsl_debug("lwsi_set_role(%p, 0x%lx)\n", wsi,
- (unsigned long)wsi->wsistate);
- }
- void lwsi_set_state(struct lws *wsi, lws_wsi_state_t lrs)
- {
- wsi->wsistate = (wsi->wsistate & (~LRS_MASK)) | lrs;
- lwsl_debug("lwsi_set_state(%p, 0x%lx)\n", wsi,
- (unsigned long)wsi->wsistate);
- }
- #endif
- void
- lws_vhost_bind_wsi(struct lws_vhost *vh, struct lws *wsi)
- {
- if (wsi->a.vhost == vh)
- return;
- lws_context_lock(vh->context, __func__); /* ---------- context { */
- wsi->a.vhost = vh;
- vh->count_bound_wsi++;
- lws_context_unlock(vh->context); /* } context ---------- */
- lwsl_debug("%s: vh %s: wsi %s/%s, count_bound_wsi %d\n", __func__,
- vh->name, wsi->role_ops ? wsi->role_ops->name : "none",
- wsi->a.protocol ? wsi->a.protocol->name : "none",
- vh->count_bound_wsi);
- assert(wsi->a.vhost->count_bound_wsi > 0);
- }
- void
- lws_vhost_unbind_wsi(struct lws *wsi)
- {
- if (!wsi->a.vhost)
- return;
- lws_context_lock(wsi->a.context, __func__); /* ---------- context { */
- assert(wsi->a.vhost->count_bound_wsi > 0);
- wsi->a.vhost->count_bound_wsi--;
- lwsl_debug("%s: vh %s: count_bound_wsi %d\n", __func__,
- wsi->a.vhost->name, wsi->a.vhost->count_bound_wsi);
- if (!wsi->a.vhost->count_bound_wsi &&
- wsi->a.vhost->being_destroyed) {
- /*
- * We have closed all wsi that were bound to this vhost
- * by any pt: nothing can be servicing any wsi belonging
- * to it any more.
- *
- * Finalize the vh destruction
- */
- __lws_vhost_destroy2(wsi->a.vhost);
- }
- wsi->a.vhost = NULL;
- lws_context_unlock(wsi->a.context); /* } context ---------- */
- }
- struct lws *
- lws_get_network_wsi(struct lws *wsi)
- {
- if (!wsi)
- return NULL;
- #if defined(LWS_WITH_HTTP2) || defined(LWS_ROLE_MQTT)
- if (!wsi->mux_substream
- #if defined(LWS_WITH_CLIENT)
- && !wsi->client_mux_substream
- #endif
- )
- return wsi;
- while (wsi->mux.parent_wsi)
- wsi = wsi->mux.parent_wsi;
- #endif
- return wsi;
- }
- const struct lws_protocols *
- lws_vhost_name_to_protocol(struct lws_vhost *vh, const char *name)
- {
- int n;
- for (n = 0; n < vh->count_protocols; n++)
- if (vh->protocols[n].name && !strcmp(name, vh->protocols[n].name))
- return &vh->protocols[n];
- return NULL;
- }
- int
- lws_callback_all_protocol(struct lws_context *context,
- const struct lws_protocols *protocol, int reason)
- {
- struct lws_context_per_thread *pt = &context->pt[0];
- unsigned int n, m = context->count_threads;
- struct lws *wsi;
- while (m--) {
- for (n = 0; n < pt->fds_count; n++) {
- wsi = wsi_from_fd(context, pt->fds[n].fd);
- if (!wsi)
- continue;
- if (wsi->a.protocol == protocol)
- protocol->callback(wsi, reason, wsi->user_space,
- NULL, 0);
- }
- pt++;
- }
- return 0;
- }
- int
- lws_callback_all_protocol_vhost_args(struct lws_vhost *vh,
- const struct lws_protocols *protocol, int reason,
- void *argp, size_t len)
- {
- struct lws_context *context = vh->context;
- struct lws_context_per_thread *pt = &context->pt[0];
- unsigned int n, m = context->count_threads;
- struct lws *wsi;
- while (m--) {
- for (n = 0; n < pt->fds_count; n++) {
- wsi = wsi_from_fd(context, pt->fds[n].fd);
- if (!wsi)
- continue;
- if (wsi->a.vhost == vh && (wsi->a.protocol == protocol ||
- !protocol))
- wsi->a.protocol->callback(wsi, reason,
- wsi->user_space, argp, len);
- }
- pt++;
- }
- return 0;
- }
- int
- lws_callback_all_protocol_vhost(struct lws_vhost *vh,
- const struct lws_protocols *protocol, int reason)
- {
- return lws_callback_all_protocol_vhost_args(vh, protocol, reason, NULL, 0);
- }
- int
- lws_callback_vhost_protocols(struct lws *wsi, int reason, void *in, int len)
- {
- int n;
- for (n = 0; n < wsi->a.vhost->count_protocols; n++)
- if (wsi->a.vhost->protocols[n].callback(wsi, reason, NULL, in, len))
- return 1;
- return 0;
- }
- int
- lws_callback_vhost_protocols_vhost(struct lws_vhost *vh, int reason, void *in,
- size_t len)
- {
- int n;
- struct lws *wsi = lws_zalloc(sizeof(*wsi), "fake wsi");
- if (!wsi)
- return 1;
- wsi->a.context = vh->context;
- lws_vhost_bind_wsi(vh, wsi);
- for (n = 0; n < wsi->a.vhost->count_protocols; n++) {
- wsi->a.protocol = &vh->protocols[n];
- if (wsi->a.protocol->callback(wsi, reason, NULL, in, len)) {
- lws_free(wsi);
- return 1;
- }
- }
- lws_free(wsi);
- return 0;
- }
- int
- lws_rx_flow_control(struct lws *wsi, int _enable)
- {
- struct lws_context_per_thread *pt = &wsi->a.context->pt[(int)wsi->tsi];
- int en = _enable;
- // h2 ignores rx flow control atm
- if (lwsi_role_h2(wsi) || wsi->mux_substream ||
- lwsi_role_h2_ENCAPSULATION(wsi))
- return 0; // !!!
- lwsl_info("%s: %p 0x%x\n", __func__, wsi, _enable);
- if (!(_enable & LWS_RXFLOW_REASON_APPLIES)) {
- /*
- * convert user bool style to bitmap style... in user simple
- * bool style _enable = 0 = flow control it, = 1 = allow rx
- */
- en = LWS_RXFLOW_REASON_APPLIES | LWS_RXFLOW_REASON_USER_BOOL;
- if (_enable & 1)
- en |= LWS_RXFLOW_REASON_APPLIES_ENABLE_BIT;
- }
- lws_pt_lock(pt, __func__);
- /* any bit set in rxflow_bitmap DISABLEs rxflow control */
- if (en & LWS_RXFLOW_REASON_APPLIES_ENABLE_BIT)
- wsi->rxflow_bitmap &= ~(en & 0xff);
- else
- wsi->rxflow_bitmap |= en & 0xff;
- if ((LWS_RXFLOW_PENDING_CHANGE | (!wsi->rxflow_bitmap)) ==
- wsi->rxflow_change_to)
- goto skip;
- wsi->rxflow_change_to = LWS_RXFLOW_PENDING_CHANGE |
- (!wsi->rxflow_bitmap);
- lwsl_info("%s: %p: bitmap 0x%x: en 0x%x, ch 0x%x\n", __func__, wsi,
- wsi->rxflow_bitmap, en, wsi->rxflow_change_to);
- if (_enable & LWS_RXFLOW_REASON_FLAG_PROCESS_NOW ||
- !wsi->rxflow_will_be_applied) {
- en = __lws_rx_flow_control(wsi);
- lws_pt_unlock(pt);
- return en;
- }
- skip:
- lws_pt_unlock(pt);
- return 0;
- }
- void
- lws_rx_flow_allow_all_protocol(const struct lws_context *context,
- const struct lws_protocols *protocol)
- {
- const struct lws_context_per_thread *pt = &context->pt[0];
- struct lws *wsi;
- unsigned int n, m = context->count_threads;
- while (m--) {
- for (n = 0; n < pt->fds_count; n++) {
- wsi = wsi_from_fd(context, pt->fds[n].fd);
- if (!wsi)
- continue;
- if (wsi->a.protocol == protocol)
- lws_rx_flow_control(wsi, LWS_RXFLOW_ALLOW);
- }
- pt++;
- }
- }
- int user_callback_handle_rxflow(lws_callback_function callback_function,
- struct lws *wsi,
- enum lws_callback_reasons reason, void *user,
- void *in, size_t len)
- {
- int n;
- wsi->rxflow_will_be_applied = 1;
- n = callback_function(wsi, reason, user, in, len);
- wsi->rxflow_will_be_applied = 0;
- if (!n)
- n = __lws_rx_flow_control(wsi);
- return n;
- }
- int
- __lws_rx_flow_control(struct lws *wsi)
- {
- struct lws *wsic = wsi->child_list;
- // h2 ignores rx flow control atm
- if (lwsi_role_h2(wsi) || wsi->mux_substream ||
- lwsi_role_h2_ENCAPSULATION(wsi))
- return 0; // !!!
- /* if he has children, do those if they were changed */
- while (wsic) {
- if (wsic->rxflow_change_to & LWS_RXFLOW_PENDING_CHANGE)
- __lws_rx_flow_control(wsic);
- wsic = wsic->sibling_list;
- }
- /* there is no pending change */
- if (!(wsi->rxflow_change_to & LWS_RXFLOW_PENDING_CHANGE))
- return 0;
- /* stuff is still buffered, not ready to really accept new input */
- if (lws_buflist_next_segment_len(&wsi->buflist, NULL)) {
- /* get ourselves called back to deal with stashed buffer */
- lws_callback_on_writable(wsi);
- // return 0;
- }
- /* now the pending is cleared, we can change rxflow state */
- wsi->rxflow_change_to &= ~LWS_RXFLOW_PENDING_CHANGE;
- lwsl_info("rxflow: wsi %p change_to %d\n", wsi,
- wsi->rxflow_change_to & LWS_RXFLOW_ALLOW);
- /* adjust the pollfd for this wsi */
- if (wsi->rxflow_change_to & LWS_RXFLOW_ALLOW) {
- lwsl_info("%s: reenable POLLIN\n", __func__);
- // lws_buflist_describe(&wsi->buflist, NULL, __func__);
- if (__lws_change_pollfd(wsi, 0, LWS_POLLIN)) {
- lwsl_info("%s: fail\n", __func__);
- return -1;
- }
- } else
- if (__lws_change_pollfd(wsi, LWS_POLLIN, 0))
- return -1;
- return 0;
- }
- const struct lws_protocols *
- lws_get_protocol(struct lws *wsi)
- {
- return wsi->a.protocol;
- }
- int
- lws_ensure_user_space(struct lws *wsi)
- {
- if (!wsi->a.protocol)
- return 0;
- /* allocate the per-connection user memory (if any) */
- if (wsi->a.protocol->per_session_data_size && !wsi->user_space) {
- wsi->user_space = lws_zalloc(
- wsi->a.protocol->per_session_data_size, "user space");
- if (wsi->user_space == NULL) {
- lwsl_err("%s: OOM\n", __func__);
- return 1;
- }
- } else
- lwsl_debug("%s: %p protocol pss %lu, user_space=%p\n", __func__,
- wsi, (long)wsi->a.protocol->per_session_data_size,
- wsi->user_space);
- return 0;
- }
- void *
- lws_adjust_protocol_psds(struct lws *wsi, size_t new_size)
- {
- ((struct lws_protocols *)lws_get_protocol(wsi))->per_session_data_size =
- new_size;
- if (lws_ensure_user_space(wsi))
- return NULL;
- return wsi->user_space;
- }
- int
- lws_get_tsi(struct lws *wsi)
- {
- return (int)wsi->tsi;
- }
- int
- lws_is_ssl(struct lws *wsi)
- {
- #if defined(LWS_WITH_TLS)
- return wsi->tls.use_ssl & LCCSCF_USE_SSL;
- #else
- (void)wsi;
- return 0;
- #endif
- }
- #if defined(LWS_WITH_TLS) && !defined(LWS_WITH_MBEDTLS)
- lws_tls_conn*
- lws_get_ssl(struct lws *wsi)
- {
- return wsi->tls.ssl;
- }
- #endif
- int
- lws_partial_buffered(struct lws *wsi)
- {
- return lws_has_buffered_out(wsi);
- }
- lws_fileofs_t
- lws_get_peer_write_allowance(struct lws *wsi)
- {
- if (!wsi->role_ops->tx_credit)
- return -1;
- return wsi->role_ops->tx_credit(wsi, LWSTXCR_US_TO_PEER, 0);
- }
- void
- lws_role_transition(struct lws *wsi, enum lwsi_role role, enum lwsi_state state,
- const struct lws_role_ops *ops)
- {
- #if (_LWS_ENABLED_LOGS & LLL_DEBUG)
- const char *name = "(unset)";
- #endif
- wsi->wsistate = role | state;
- if (ops)
- wsi->role_ops = ops;
- #if (_LWS_ENABLED_LOGS & LLL_DEBUG)
- if (wsi->role_ops)
- name = wsi->role_ops->name;
- lwsl_debug("%s: %p: wsistate 0x%lx, ops %s\n", __func__, wsi,
- (unsigned long)wsi->wsistate, name);
- #endif
- }
- int
- lws_parse_uri(char *p, const char **prot, const char **ads, int *port,
- const char **path)
- {
- const char *end;
- char unix_skt = 0;
- /* cut up the location into address, port and path */
- *prot = p;
- while (*p && (*p != ':' || p[1] != '/' || p[2] != '/'))
- p++;
- if (!*p) {
- end = p;
- p = (char *)*prot;
- *prot = end;
- } else {
- *p = '\0';
- p += 3;
- }
- if (*p == '+') /* unix skt */
- unix_skt = 1;
- *ads = p;
- if (!strcmp(*prot, "http") || !strcmp(*prot, "ws"))
- *port = 80;
- else if (!strcmp(*prot, "https") || !strcmp(*prot, "wss"))
- *port = 443;
- if (*p == '[') {
- ++(*ads);
- while (*p && *p != ']')
- p++;
- if (*p)
- *p++ = '\0';
- } else
- while (*p && *p != ':' && (unix_skt || *p != '/'))
- p++;
- if (*p == ':') {
- *p++ = '\0';
- *port = atoi(p);
- while (*p && *p != '/')
- p++;
- }
- *path = "/";
- if (*p) {
- *p++ = '\0';
- if (*p)
- *path = p;
- }
- return 0;
- }
- /* ... */
- const char *
- lws_get_urlarg_by_name(struct lws *wsi, const char *name, char *buf, int len)
- {
- int n = 0, sl = (int)strlen(name);
- while (lws_hdr_copy_fragment(wsi, buf, len,
- WSI_TOKEN_HTTP_URI_ARGS, n) >= 0) {
- if (!strncmp(buf, name, sl))
- return buf + sl;
- n++;
- }
- return NULL;
- }
- #if defined(LWS_WITHOUT_EXTENSIONS)
- /* we need to provide dummy callbacks for internal exts
- * so user code runs when faced with a lib compiled with
- * extensions disabled.
- */
- int
- lws_extension_callback_pm_deflate(struct lws_context *context,
- const struct lws_extension *ext,
- struct lws *wsi,
- enum lws_extension_callback_reasons reason,
- void *user, void *in, size_t len)
- {
- (void)context;
- (void)ext;
- (void)wsi;
- (void)reason;
- (void)user;
- (void)in;
- (void)len;
- return 0;
- }
- int
- lws_set_extension_option(struct lws *wsi, const char *ext_name,
- const char *opt_name, const char *opt_val)
- {
- return -1;
- }
- #endif
- int
- lws_is_cgi(struct lws *wsi) {
- #ifdef LWS_WITH_CGI
- return !!wsi->http.cgi;
- #else
- return 0;
- #endif
- }
- const struct lws_protocol_vhost_options *
- lws_pvo_search(const struct lws_protocol_vhost_options *pvo, const char *name)
- {
- while (pvo) {
- if (!strcmp(pvo->name, name))
- break;
- pvo = pvo->next;
- }
- return pvo;
- }
- int
- lws_pvo_get_str(void *in, const char *name, const char **result)
- {
- const struct lws_protocol_vhost_options *pv =
- lws_pvo_search((const struct lws_protocol_vhost_options *)in,
- name);
- if (!pv)
- return 1;
- *result = (const char *)pv->value;
- return 0;
- }
- int
- lws_broadcast(struct lws_context_per_thread *pt, int reason, void *in, size_t len)
- {
- struct lws_vhost *v = pt->context->vhost_list;
- lws_fakewsi_def_plwsa(pt);
- int n, ret = 0;
- lws_fakewsi_prep_plwsa_ctx(pt->context);
- #if !defined(LWS_PLAT_FREERTOS) && LWS_MAX_SMP > 1
- ((struct lws *)plwsa)->tsi = (int)(pt - &pt->context->pt[0]);
- #endif
- while (v) {
- const struct lws_protocols *p = v->protocols;
- plwsa->vhost = v; /* not a real bound wsi */
- for (n = 0; n < v->count_protocols; n++) {
- plwsa->protocol = p;
- if (p->callback &&
- p->callback((struct lws *)plwsa, reason, NULL, in, len))
- ret |= 1;
- p++;
- }
- v = v->vhost_next;
- }
- return ret;
- }
- void *
- lws_wsi_user(struct lws *wsi)
- {
- return wsi->user_space;
- }
- int
- lws_wsi_tsi(struct lws *wsi)
- {
- return wsi->tsi;
- }
- void
- lws_set_wsi_user(struct lws *wsi, void *data)
- {
- if (!wsi->user_space_externally_allocated && wsi->user_space)
- lws_free(wsi->user_space);
- wsi->user_space_externally_allocated = 1;
- wsi->user_space = data;
- }
- struct lws *
- lws_get_parent(const struct lws *wsi)
- {
- return wsi->parent;
- }
- struct lws *
- lws_get_child(const struct lws *wsi)
- {
- return wsi->child_list;
- }
- void *
- lws_get_opaque_parent_data(const struct lws *wsi)
- {
- return wsi->opaque_parent_data;
- }
- void
- lws_set_opaque_parent_data(struct lws *wsi, void *data)
- {
- wsi->opaque_parent_data = data;
- }
- void *
- lws_get_opaque_user_data(const struct lws *wsi)
- {
- return wsi->a.opaque_user_data;
- }
- void
- lws_set_opaque_user_data(struct lws *wsi, void *data)
- {
- wsi->a.opaque_user_data = data;
- }
- int
- lws_get_child_pending_on_writable(const struct lws *wsi)
- {
- return wsi->parent_pending_cb_on_writable;
- }
- void
- lws_clear_child_pending_on_writable(struct lws *wsi)
- {
- wsi->parent_pending_cb_on_writable = 0;
- }
- const char *
- lws_get_vhost_name(struct lws_vhost *vhost)
- {
- return vhost->name;
- }
- int
- lws_get_vhost_port(struct lws_vhost *vhost)
- {
- return vhost->listen_port;
- }
- void *
- lws_get_vhost_user(struct lws_vhost *vhost)
- {
- return vhost->user;
- }
- const char *
- lws_get_vhost_iface(struct lws_vhost *vhost)
- {
- return vhost->iface;
- }
- lws_sockfd_type
- lws_get_socket_fd(struct lws *wsi)
- {
- if (!wsi)
- return -1;
- return wsi->desc.sockfd;
- }
- struct lws_vhost *
- lws_vhost_get(struct lws *wsi)
- {
- return wsi->a.vhost;
- }
- struct lws_vhost *
- lws_get_vhost(struct lws *wsi)
- {
- return wsi->a.vhost;
- }
- const struct lws_protocols *
- lws_protocol_get(struct lws *wsi)
- {
- return wsi->a.protocol;
- }
- #if defined(LWS_WITH_UDP)
- const struct lws_udp *
- lws_get_udp(const struct lws *wsi)
- {
- return wsi->udp;
- }
- #endif
- struct lws_context *
- lws_get_context(const struct lws *wsi)
- {
- return wsi->a.context;
- }
- #if defined(LWS_WITH_CLIENT)
- int
- _lws_generic_transaction_completed_active_conn(struct lws **_wsi, char take_vh_lock)
- {
- struct lws *wnew, *wsi = *_wsi;
- /*
- * Are we constitutionally capable of having a queue, ie, we are on
- * the "active client connections" list?
- *
- * If not, that's it for us.
- */
- if (lws_dll2_is_detached(&wsi->dll_cli_active_conns))
- return 0; /* no new transaction */
- /*
- * With h1 queuing, the original "active client" moves his attributes
- * like fd, ssl, queue and active client list entry to the next guy in
- * the queue before closing... it's because the user code knows the
- * individual wsi and the action must take place in the correct wsi
- * context. Note this means we don't truly pipeline headers.
- *
- * Trying to keep the original "active client" in place to do the work
- * of the wsi breaks down when dealing with queued POSTs otherwise; it's
- * also competing with the real mux child arrangements and complicating
- * the code.
- *
- * For that reason, see if we have any queued child now...
- */
- if (!wsi->dll2_cli_txn_queue_owner.head) {
- /*
- * Nothing pipelined... we should hang around a bit
- * in case something turns up... otherwise we'll close
- */
- lwsl_info("%s: nothing pipelined waiting\n", __func__);
- lwsi_set_state(wsi, LRS_IDLING);
- lws_set_timeout(wsi, PENDING_TIMEOUT_CLIENT_CONN_IDLE,
- wsi->keep_warm_secs);
- return 0; /* no new transaction right now */
- }
- /*
- * We have a queued child wsi we should bequeath our assets to, before
- * closing ourself
- */
- if (take_vh_lock)
- lws_vhost_lock(wsi->a.vhost);
- wnew = lws_container_of(wsi->dll2_cli_txn_queue_owner.head, struct lws,
- dll2_cli_txn_queue);
- assert(wsi != wnew);
- lws_dll2_remove(&wnew->dll2_cli_txn_queue);
- assert(lws_socket_is_valid(wsi->desc.sockfd));
- __lws_change_pollfd(wsi, LWS_POLLOUT | LWS_POLLIN, 0);
- /* copy the fd */
- wnew->desc = wsi->desc;
- assert(lws_socket_is_valid(wnew->desc.sockfd));
- /* disconnect the fd from association with old wsi */
- if (__remove_wsi_socket_from_fds(wsi))
- return -1;
- sanity_assert_no_wsi_traces(wsi->a.context, wsi);
- sanity_assert_no_sockfd_traces(wsi->a.context, wsi->desc.sockfd);
- wsi->desc.sockfd = LWS_SOCK_INVALID;
- __lws_wsi_remove_from_sul(wsi);
- /*
- * ... we're doing some magic here in terms of handing off the socket
- * that has been active to a wsi that has not yet itself been active...
- * depending on the event lib we may need to give a magic spark to the
- * new guy and snuff out the old guy's magic spark at that level as well
- */
- #if defined(LWS_WITH_EVENT_LIBS)
- if (wsi->a.context->event_loop_ops->destroy_wsi)
- wsi->a.context->event_loop_ops->destroy_wsi(wsi);
- if (wsi->a.context->event_loop_ops->sock_accept)
- wsi->a.context->event_loop_ops->sock_accept(wnew);
- #endif
- /* point the fd table entry to new guy */
- assert(lws_socket_is_valid(wnew->desc.sockfd));
- if (__insert_wsi_socket_into_fds(wsi->a.context, wnew))
- return -1;
- #if defined(LWS_WITH_TLS)
- /* pass on the tls */
- wnew->tls = wsi->tls;
- wsi->tls.client_bio = NULL;
- wsi->tls.ssl = NULL;
- wsi->tls.use_ssl = 0;
- #endif
- /* take over his copy of his endpoint as an active connection */
- wnew->cli_hostname_copy = wsi->cli_hostname_copy;
- wsi->cli_hostname_copy = NULL;
- wnew->keep_warm_secs = wsi->keep_warm_secs;
- /*
- * selected queued guy now replaces the original leader on the
- * active client conn list
- */
- lws_dll2_remove(&wsi->dll_cli_active_conns);
- lws_dll2_add_tail(&wnew->dll_cli_active_conns,
- &wsi->a.vhost->dll_cli_active_conns_owner);
- /* move any queued guys to queue on new active conn */
- lws_start_foreach_dll_safe(struct lws_dll2 *, d, d1,
- wsi->dll2_cli_txn_queue_owner.head) {
- struct lws *ww = lws_container_of(d, struct lws,
- dll2_cli_txn_queue);
- lws_dll2_remove(&ww->dll2_cli_txn_queue);
- lws_dll2_add_tail(&ww->dll2_cli_txn_queue,
- &wnew->dll2_cli_txn_queue_owner);
- } lws_end_foreach_dll_safe(d, d1);
- if (take_vh_lock)
- lws_vhost_unlock(wsi->a.vhost);
- /*
- * The original leader who passed on all his powers already can die...
- * in the call stack above us there are guys who still want to touch
- * him, so have him die next time around the event loop, not now.
- */
- wsi->already_did_cce = 1; /* so the close doesn't trigger a CCE */
- lws_set_timeout(wsi, 1, LWS_TO_KILL_ASYNC);
- /* after the first one, they can only be coming from the queue */
- wnew->transaction_from_pipeline_queue = 1;
- lwsl_notice("%s: pipeline queue passed wsi %p on to queued wsi %p\n",
- __func__, wsi, wnew);
- *_wsi = wnew; /* inform caller we swapped */
- return 1; /* new transaction */
- }
- #endif
- int LWS_WARN_UNUSED_RESULT
- lws_raw_transaction_completed(struct lws *wsi)
- {
- if (lws_has_buffered_out(wsi)) {
- /*
- * ...so he tried to send something large, but it went out
- * as a partial, but he immediately called us to say he wants
- * to close the connection.
- *
- * Defer the close until the last part of the partial is sent.
- *
- */
- lwsl_debug("%s: %p: deferring due to partial\n", __func__, wsi);
- wsi->close_when_buffered_out_drained = 1;
- lws_callback_on_writable(wsi);
- return 0;
- }
- return -1;
- }
- int
- lws_bind_protocol(struct lws *wsi, const struct lws_protocols *p,
- const char *reason)
- {
- // if (wsi->a.protocol == p)
- // return 0;
- const struct lws_protocols *vp = wsi->a.vhost->protocols, *vpo;
- if (wsi->a.protocol && wsi->protocol_bind_balance) {
- wsi->a.protocol->callback(wsi,
- wsi->role_ops->protocol_unbind_cb[!!lwsi_role_server(wsi)],
- wsi->user_space, (void *)reason, 0);
- wsi->protocol_bind_balance = 0;
- }
- if (!wsi->user_space_externally_allocated)
- lws_free_set_NULL(wsi->user_space);
- lws_same_vh_protocol_remove(wsi);
- wsi->a.protocol = p;
- if (!p)
- return 0;
- if (lws_ensure_user_space(wsi))
- return 1;
- if (p > vp && p < &vp[wsi->a.vhost->count_protocols])
- lws_same_vh_protocol_insert(wsi, (int)(p - vp));
- else {
- int n = wsi->a.vhost->count_protocols;
- int hit = 0;
- vpo = vp;
- while (n--) {
- if (p->name && vp->name && !strcmp(p->name, vp->name)) {
- hit = 1;
- lws_same_vh_protocol_insert(wsi, (int)(vp - vpo));
- break;
- }
- vp++;
- }
- if (!hit)
- lwsl_err("%s: %p is not in vhost '%s' protocols list\n",
- __func__, p, wsi->a.vhost->name);
- }
- if (wsi->a.protocol->callback(wsi, wsi->role_ops->protocol_bind_cb[
- !!lwsi_role_server(wsi)],
- wsi->user_space, NULL, 0))
- return 1;
- wsi->protocol_bind_balance = 1;
- return 0;
- }
- void
- lws_http_close_immortal(struct lws *wsi)
- {
- struct lws *nwsi;
- if (!wsi->mux_substream)
- return;
- assert(wsi->mux_stream_immortal);
- wsi->mux_stream_immortal = 0;
- nwsi = lws_get_network_wsi(wsi);
- lwsl_debug("%s: %p %p %d\n", __func__, wsi, nwsi,
- nwsi->immortal_substream_count);
- assert(nwsi->immortal_substream_count);
- nwsi->immortal_substream_count--;
- if (!nwsi->immortal_substream_count)
- /*
- * since we closed the only immortal stream on this nwsi, we
- * need to reapply a normal timeout regime to the nwsi
- */
- lws_set_timeout(nwsi, PENDING_TIMEOUT_HTTP_KEEPALIVE_IDLE,
- wsi->a.vhost->keepalive_timeout ?
- wsi->a.vhost->keepalive_timeout : 31);
- }
- void
- lws_mux_mark_immortal(struct lws *wsi)
- {
- struct lws *nwsi;
- lws_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
- if (!wsi->mux_substream
- #if defined(LWS_WITH_CLIENT)
- && !wsi->client_mux_substream
- #endif
- ) {
- lwsl_err("%s: not h2 substream\n", __func__);
- return;
- }
- nwsi = lws_get_network_wsi(wsi);
- if (!nwsi)
- return;
- lwsl_debug("%s: %p %p %d\n", __func__, wsi, nwsi,
- nwsi->immortal_substream_count);
- wsi->mux_stream_immortal = 1;
- assert(nwsi->immortal_substream_count < 255); /* largest count */
- nwsi->immortal_substream_count++;
- if (nwsi->immortal_substream_count == 1)
- lws_set_timeout(nwsi, NO_PENDING_TIMEOUT, 0);
- }
- int
- lws_http_mark_sse(struct lws *wsi)
- {
- if (!wsi)
- return 0;
- lws_http_headers_detach(wsi);
- lws_mux_mark_immortal(wsi);
- if (wsi->mux_substream)
- wsi->h2_stream_carries_sse = 1;
- return 0;
- }
- #if defined(LWS_WITH_CLIENT)
- const char *
- lws_wsi_client_stash_item(struct lws *wsi, int stash_idx, int hdr_idx)
- {
- /* try the generic client stash */
- if (wsi->stash)
- return wsi->stash->cis[stash_idx];
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- /* if not, use the ah stash if applicable */
- return lws_hdr_simple_ptr(wsi, hdr_idx);
- #else
- return NULL;
- #endif
- }
- #endif
- #if defined(LWS_ROLE_H2) || defined(LWS_ROLE_MQTT)
- void
- lws_wsi_mux_insert(struct lws *wsi, struct lws *parent_wsi, int sid)
- {
- lwsl_info("%s: wsi %p, par %p: assign sid %d (curr %d)\n", __func__,
- wsi, parent_wsi, sid, wsi->mux.my_sid);
- if (wsi->mux.my_sid && wsi->mux.my_sid != (unsigned int)sid)
- assert(0);
- wsi->mux.my_sid = sid;
- wsi->mux.parent_wsi = parent_wsi;
- wsi->role_ops = parent_wsi->role_ops;
- /* new guy's sibling is whoever was the first child before */
- wsi->mux.sibling_list = parent_wsi->mux.child_list;
- /* first child is now the new guy */
- parent_wsi->mux.child_list = wsi;
- parent_wsi->mux.child_count++;
- }
- struct lws *
- lws_wsi_mux_from_id(struct lws *parent_wsi, unsigned int sid)
- {
- lws_start_foreach_ll(struct lws *, wsi, parent_wsi->mux.child_list) {
- if (wsi->mux.my_sid == sid)
- return wsi;
- } lws_end_foreach_ll(wsi, mux.sibling_list);
- return NULL;
- }
- void
- lws_wsi_mux_dump_children(struct lws *wsi)
- {
- #if defined(_DEBUG)
- if (!wsi->mux.parent_wsi || !lwsl_visible(LLL_INFO))
- return;
- lws_start_foreach_llp(struct lws **, w,
- wsi->mux.parent_wsi->mux.child_list) {
- lwsl_info(" \\---- child %s %p\n",
- (*w)->role_ops ? (*w)->role_ops->name : "?", *w);
- assert(*w != (*w)->mux.sibling_list);
- } lws_end_foreach_llp(w, mux.sibling_list);
- #endif
- }
- void
- lws_wsi_mux_close_children(struct lws *wsi, int reason)
- {
- struct lws *wsi2;
- struct lws **w;
- if (!wsi->mux.child_list)
- return;
- w = &wsi->mux.child_list;
- while (*w) {
- lwsl_info(" closing child %p\n", *w);
- /* disconnect from siblings */
- wsi2 = (*w)->mux.sibling_list;
- assert (wsi2 != *w);
- (*w)->mux.sibling_list = NULL;
- (*w)->socket_is_permanently_unusable = 1;
- __lws_close_free_wsi(*w, reason, "mux child recurse");
- *w = wsi2;
- }
- }
- void
- lws_wsi_mux_sibling_disconnect(struct lws *wsi)
- {
- struct lws *wsi2;
- lws_start_foreach_llp(struct lws **, w,
- wsi->mux.parent_wsi->mux.child_list) {
- /* disconnect from siblings */
- if (*w == wsi) {
- wsi2 = (*w)->mux.sibling_list;
- (*w)->mux.sibling_list = NULL;
- *w = wsi2;
- lwsl_debug(" %p disentangled from sibling %p\n",
- wsi, wsi2);
- break;
- }
- } lws_end_foreach_llp(w, mux.sibling_list);
- wsi->mux.parent_wsi->mux.child_count--;
- wsi->mux.parent_wsi = NULL;
- }
- void
- lws_wsi_mux_dump_waiting_children(struct lws *wsi)
- {
- #if defined(_DEBUG)
- lwsl_info("%s: %p: children waiting for POLLOUT service:\n",
- __func__, wsi);
- wsi = wsi->mux.child_list;
- while (wsi) {
- lwsl_info(" %c %p: sid %u: 0x%x %s %s\n",
- wsi->mux.requested_POLLOUT ? '*' : ' ',
- wsi, wsi->mux.my_sid, lwsi_state(wsi),
- wsi->role_ops->name,
- wsi->a.protocol ? wsi->a.protocol->name : "noprotocol");
- wsi = wsi->mux.sibling_list;
- }
- #endif
- }
- int
- lws_wsi_mux_mark_parents_needing_writeable(struct lws *wsi)
- {
- struct lws /* *network_wsi = lws_get_network_wsi(wsi), */ *wsi2;
- //int already = network_wsi->mux.requested_POLLOUT;
- /* mark everybody above him as requesting pollout */
- wsi2 = wsi;
- while (wsi2) {
- wsi2->mux.requested_POLLOUT = 1;
- lwsl_info("%s: mark wsi: %p, sid %u, pending writable\n",
- __func__, wsi2, wsi2->mux.my_sid);
- wsi2 = wsi2->mux.parent_wsi;
- }
- return 0; // already;
- }
- struct lws *
- lws_wsi_mux_move_child_to_tail(struct lws **wsi2)
- {
- struct lws *w = *wsi2;
- while (w) {
- if (!w->mux.sibling_list) { /* w is the current last */
- lwsl_debug("w=%p, *wsi2 = %p\n", w, *wsi2);
- if (w == *wsi2) /* we are already last */
- break;
- /* last points to us as new last */
- w->mux.sibling_list = *wsi2;
- /* guy pointing to us until now points to
- * our old next */
- *wsi2 = (*wsi2)->mux.sibling_list;
- /* we point to nothing because we are last */
- w->mux.sibling_list->mux.sibling_list = NULL;
- /* w becomes us */
- w = w->mux.sibling_list;
- break;
- }
- w = w->mux.sibling_list;
- }
- /* clear the waiting for POLLOUT on the guy that was chosen */
- if (w)
- w->mux.requested_POLLOUT = 0;
- return w;
- }
- int
- lws_wsi_mux_action_pending_writeable_reqs(struct lws *wsi)
- {
- struct lws *w = wsi->mux.child_list;
- while (w) {
- if (w->mux.requested_POLLOUT) {
- if (lws_change_pollfd(wsi, 0, LWS_POLLOUT))
- return -1;
- return 0;
- }
- w = w->mux.sibling_list;
- }
- if (lws_change_pollfd(wsi, LWS_POLLOUT, 0))
- return -1;
- return 0;
- }
- int
- lws_wsi_txc_check_skint(struct lws_tx_credit *txc, int32_t tx_cr)
- {
- if (txc->tx_cr <= 0) {
- /*
- * If other side is not able to cope with us sending any DATA
- * so no matter if we have POLLOUT on our side if it's DATA we
- * want to send.
- */
- if (!txc->skint)
- lwsl_info("%s: %p: skint (%d)\n", __func__, txc,
- (int)txc->tx_cr);
- txc->skint = 1;
- return 1;
- }
- if (txc->skint)
- lwsl_info("%s: %p: unskint (%d)\n", __func__, txc,
- (int)txc->tx_cr);
- txc->skint = 0;
- return 0;
- }
- #if defined(_DEBUG)
- void
- lws_wsi_txc_describe(struct lws_tx_credit *txc, const char *at, uint32_t sid)
- {
- lwsl_info("%s: %p: %s: sid %d: %speer-to-us: %d, us-to-peer: %d\n",
- __func__, txc, at, (int)sid, txc->skint ? "SKINT, " : "",
- (int)txc->peer_tx_cr_est, (int)txc->tx_cr);
- }
- #endif
- int
- lws_wsi_tx_credit(struct lws *wsi, char peer_to_us, int add)
- {
- if (wsi->role_ops && wsi->role_ops->tx_credit)
- return wsi->role_ops->tx_credit(wsi, peer_to_us, add);
- return 0;
- }
- /*
- * Let the protocol know about incoming tx credit window updates if it's
- * managing the flow control manually (it may want to proxy this information)
- */
- int
- lws_wsi_txc_report_manual_txcr_in(struct lws *wsi, int32_t bump)
- {
- if (!wsi->txc.manual)
- /*
- * If we don't care about managing it manually, no need to
- * report it
- */
- return 0;
- return user_callback_handle_rxflow(wsi->a.protocol->callback,
- wsi, LWS_CALLBACK_WSI_TX_CREDIT_GET,
- wsi->user_space, NULL, (size_t)bump);
- }
- #if defined(LWS_WITH_CLIENT)
- int
- lws_wsi_mux_apply_queue(struct lws *wsi)
- {
- /* we have a transaction queue that wants to pipeline */
- lws_vhost_lock(wsi->a.vhost);
- lws_start_foreach_dll_safe(struct lws_dll2 *, d, d1,
- wsi->dll2_cli_txn_queue_owner.head) {
- struct lws *w = lws_container_of(d, struct lws,
- dll2_cli_txn_queue);
- #if defined(LWS_ROLE_H2)
- if (lwsi_role_http(wsi) &&
- lwsi_state(w) == LRS_H2_WAITING_TO_SEND_HEADERS) {
- lwsl_info("%s: cli pipeq %p to be h2\n", __func__, w);
- lwsi_set_state(w, LRS_H1C_ISSUE_HANDSHAKE2);
- /* remove ourselves from client queue */
- lws_dll2_remove(&w->dll2_cli_txn_queue);
- /* attach ourselves as an h2 stream */
- lws_wsi_h2_adopt(wsi, w);
- }
- #endif
- #if defined(LWS_ROLE_MQTT)
- if (lwsi_role_mqtt(wsi) &&
- lwsi_state(wsi) == LRS_ESTABLISHED) {
- lwsl_info("%s: cli pipeq %p to be mqtt\n", __func__, w);
- /* remove ourselves from client queue */
- lws_dll2_remove(&w->dll2_cli_txn_queue);
- /* attach ourselves as an h2 stream */
- lws_wsi_mqtt_adopt(wsi, w);
- }
- #endif
- } lws_end_foreach_dll_safe(d, d1);
- lws_vhost_unlock(wsi->a.vhost);
- return 0;
- }
- #endif
- #endif
|