| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865 |
- /*
- * libwebsockets - small server side websockets and web server implementation
- *
- * Copyright (C) 2010-2018 Andy Green <[email protected]>
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation:
- * version 2.1 of the License.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
- * MA 02110-1301 USA
- */
- #include "core/private.h"
- #ifndef LWS_BUILD_HASH
- #define LWS_BUILD_HASH "unknown-build-hash"
- #endif
- const struct lws_role_ops *available_roles[] = {
- #if defined(LWS_ROLE_H2)
- &role_ops_h2,
- #endif
- #if defined(LWS_ROLE_H1)
- &role_ops_h1,
- #endif
- #if defined(LWS_ROLE_WS)
- &role_ops_ws,
- #endif
- #if defined(LWS_ROLE_DBUS)
- &role_ops_dbus,
- #endif
- NULL
- };
- const struct lws_event_loop_ops *available_event_libs[] = {
- #if defined(LWS_WITH_POLL)
- &event_loop_ops_poll,
- #endif
- #if defined(LWS_WITH_LIBUV)
- &event_loop_ops_uv,
- #endif
- #if defined(LWS_WITH_LIBEVENT)
- &event_loop_ops_event,
- #endif
- #if defined(LWS_WITH_LIBEV)
- &event_loop_ops_ev,
- #endif
- NULL
- };
- static const char *library_version = LWS_LIBRARY_VERSION " " LWS_BUILD_HASH;
- /**
- * lws_get_library_version: get version and git hash library built from
- *
- * returns a const char * to a string like "1.1 178d78c"
- * representing the library version followed by the git head hash it
- * was built from
- */
- LWS_VISIBLE const char *
- lws_get_library_version(void)
- {
- return library_version;
- }
- int
- lws_role_call_alpn_negotiated(struct lws *wsi, const char *alpn)
- {
- #if defined(LWS_WITH_TLS)
- if (!alpn)
- return 0;
- lwsl_info("%s: '%s'\n", __func__, alpn);
- LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
- if (ar->alpn && !strcmp(ar->alpn, alpn) && ar->alpn_negotiated)
- return ar->alpn_negotiated(wsi, alpn);
- LWS_FOR_EVERY_AVAILABLE_ROLE_END;
- #endif
- return 0;
- }
- #if !defined(LWS_WITHOUT_SERVER)
- int
- lws_role_call_adoption_bind(struct lws *wsi, int type, const char *prot)
- {
- LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
- if (ar->adoption_bind)
- if (ar->adoption_bind(wsi, type, prot))
- return 0;
- LWS_FOR_EVERY_AVAILABLE_ROLE_END;
- /* fall back to raw socket role if, eg, h1 not configured */
- if (role_ops_raw_skt.adoption_bind &&
- role_ops_raw_skt.adoption_bind(wsi, type, prot))
- return 0;
- /* fall back to raw file role if, eg, h1 not configured */
- if (role_ops_raw_file.adoption_bind &&
- role_ops_raw_file.adoption_bind(wsi, type, prot))
- return 0;
- return 1;
- }
- #endif
- #if !defined(LWS_WITHOUT_CLIENT)
- int
- lws_role_call_client_bind(struct lws *wsi,
- const struct lws_client_connect_info *i)
- {
- LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
- if (ar->client_bind) {
- int m = ar->client_bind(wsi, i);
- if (m < 0)
- return m;
- if (m)
- return 0;
- }
- LWS_FOR_EVERY_AVAILABLE_ROLE_END;
- /* fall back to raw socket role if, eg, h1 not configured */
- if (role_ops_raw_skt.client_bind &&
- role_ops_raw_skt.client_bind(wsi, i))
- return 0;
- return 1;
- }
- #endif
- static const char * const mount_protocols[] = {
- "http://",
- "https://",
- "file://",
- "cgi://",
- ">http://",
- ">https://",
- "callback://"
- };
- LWS_VISIBLE void *
- lws_protocol_vh_priv_zalloc(struct lws_vhost *vhost,
- const struct lws_protocols *prot, int size)
- {
- int n = 0;
- /* allocate the vh priv array only on demand */
- if (!vhost->protocol_vh_privs) {
- vhost->protocol_vh_privs = (void **)lws_zalloc(
- vhost->count_protocols * sizeof(void *),
- "protocol_vh_privs");
- if (!vhost->protocol_vh_privs)
- return NULL;
- }
- while (n < vhost->count_protocols && &vhost->protocols[n] != prot)
- n++;
- if (n == vhost->count_protocols) {
- n = 0;
- while (n < vhost->count_protocols &&
- strcmp(vhost->protocols[n].name, prot->name))
- n++;
- if (n == vhost->count_protocols)
- return NULL;
- }
- vhost->protocol_vh_privs[n] = lws_zalloc(size, "vh priv");
- return vhost->protocol_vh_privs[n];
- }
- LWS_VISIBLE void *
- lws_protocol_vh_priv_get(struct lws_vhost *vhost,
- const struct lws_protocols *prot)
- {
- int n = 0;
- if (!vhost || !vhost->protocol_vh_privs || !prot)
- return NULL;
- while (n < vhost->count_protocols && &vhost->protocols[n] != prot)
- n++;
- if (n == vhost->count_protocols) {
- n = 0;
- while (n < vhost->count_protocols &&
- strcmp(vhost->protocols[n].name, prot->name))
- n++;
- if (n == vhost->count_protocols) {
- lwsl_err("%s: unknown protocol %p\n", __func__, prot);
- return NULL;
- }
- }
- return vhost->protocol_vh_privs[n];
- }
- static const struct lws_protocol_vhost_options *
- lws_vhost_protocol_options(struct lws_vhost *vh, const char *name)
- {
- const struct lws_protocol_vhost_options *pvo = vh->pvo;
- if (!name)
- return NULL;
- while (pvo) {
- if (!strcmp(pvo->name, name))
- return pvo;
- pvo = pvo->next;
- }
- return NULL;
- }
- /*
- * inform every vhost that hasn't already done it, that
- * his protocols are initializing
- */
- LWS_VISIBLE int
- lws_protocol_init(struct lws_context *context)
- {
- struct lws_vhost *vh = context->vhost_list;
- const struct lws_protocol_vhost_options *pvo, *pvo1;
- struct lws wsi;
- int n, any = 0;
- if (context->doing_protocol_init)
- return 0;
- context->doing_protocol_init = 1;
- memset(&wsi, 0, sizeof(wsi));
- wsi.context = context;
- lwsl_info("%s\n", __func__);
- while (vh) {
- wsi.vhost = vh;
- /* only do the protocol init once for a given vhost */
- if (vh->created_vhost_protocols ||
- (vh->options & LWS_SERVER_OPTION_SKIP_PROTOCOL_INIT))
- goto next;
- /* initialize supported protocols on this vhost */
- for (n = 0; n < vh->count_protocols; n++) {
- wsi.protocol = &vh->protocols[n];
- if (!vh->protocols[n].name)
- continue;
- pvo = lws_vhost_protocol_options(vh,
- vh->protocols[n].name);
- if (pvo) {
- /*
- * linked list of options specific to
- * vh + protocol
- */
- pvo1 = pvo;
- pvo = pvo1->options;
- while (pvo) {
- lwsl_debug(
- " vhost \"%s\", "
- "protocol \"%s\", "
- "option \"%s\"\n",
- vh->name,
- vh->protocols[n].name,
- pvo->name);
- if (!strcmp(pvo->name, "default")) {
- lwsl_info("Setting default "
- "protocol for vh %s to %s\n",
- vh->name,
- vh->protocols[n].name);
- vh->default_protocol_index = n;
- }
- if (!strcmp(pvo->name, "raw")) {
- lwsl_info("Setting raw "
- "protocol for vh %s to %s\n",
- vh->name,
- vh->protocols[n].name);
- vh->raw_protocol_index = n;
- }
- pvo = pvo->next;
- }
- pvo = pvo1->options;
- }
- #if defined(LWS_WITH_TLS)
- any |= !!vh->tls.ssl_ctx;
- #endif
- /*
- * inform all the protocols that they are doing their
- * one-time initialization if they want to.
- *
- * NOTE the wsi is all zeros except for the context, vh
- * + protocol ptrs so lws_get_context(wsi) etc can work
- */
- if (vh->protocols[n].callback(&wsi,
- LWS_CALLBACK_PROTOCOL_INIT, NULL,
- (void *)pvo, 0)) {
- lws_free(vh->protocol_vh_privs[n]);
- vh->protocol_vh_privs[n] = NULL;
- lwsl_err("%s: protocol %s failed init\n",
- __func__, vh->protocols[n].name);
- return 1;
- }
- }
- vh->created_vhost_protocols = 1;
- next:
- vh = vh->vhost_next;
- }
- context->doing_protocol_init = 0;
- if (!context->protocol_init_done)
- lws_finalize_startup(context);
- context->protocol_init_done = 1;
- if (any)
- lws_tls_check_all_cert_lifetimes(context);
- return 0;
- }
- /* list of supported protocols and callbacks */
- static const struct lws_protocols protocols_dummy[] = {
- /* first protocol must always be HTTP handler */
- {
- "http-only", /* name */
- lws_callback_http_dummy, /* callback */
- 0, /* per_session_data_size */
- 0, /* rx_buffer_size */
- 0, /* id */
- NULL, /* user */
- 0 /* tx_packet_size */
- },
- /*
- * the other protocols are provided by lws plugins
- */
- { NULL, NULL, 0, 0, 0, NULL, 0} /* terminator */
- };
- #ifdef LWS_PLAT_OPTEE
- #undef LWS_HAVE_GETENV
- #endif
- LWS_VISIBLE struct lws_vhost *
- lws_create_vhost(struct lws_context *context,
- const struct lws_context_creation_info *info)
- {
- struct lws_vhost *vh = lws_zalloc(sizeof(*vh), "create vhost"),
- **vh1 = &context->vhost_list;
- const struct lws_http_mount *mounts;
- const struct lws_protocols *pcols = info->protocols;
- const struct lws_protocol_vhost_options *pvo;
- #ifdef LWS_WITH_PLUGINS
- struct lws_plugin *plugin = context->plugin_list;
- #endif
- struct lws_protocols *lwsp;
- int m, f = !info->pvo;
- char buf[20];
- #if !defined(LWS_WITHOUT_CLIENT) && defined(LWS_HAVE_GETENV)
- char *p;
- #endif
- int n;
- if (!vh)
- return NULL;
- #if LWS_MAX_SMP > 1
- pthread_mutex_init(&vh->lock, NULL);
- #endif
- if (!pcols)
- pcols = &protocols_dummy[0];
- vh->context = context;
- if (!info->vhost_name)
- vh->name = "default";
- else
- vh->name = info->vhost_name;
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- vh->http.error_document_404 = info->error_document_404;
- #endif
- if (info->options & LWS_SERVER_OPTION_ONLY_RAW)
- lwsl_info("%s set to only support RAW\n", vh->name);
- vh->iface = info->iface;
- #if !defined(LWS_WITH_ESP32) && \
- !defined(OPTEE_TA) && !defined(WIN32)
- vh->bind_iface = info->bind_iface;
- #endif
- for (vh->count_protocols = 0;
- pcols[vh->count_protocols].callback;
- vh->count_protocols++)
- ;
- vh->options = info->options;
- vh->pvo = info->pvo;
- vh->headers = info->headers;
- vh->user = info->user;
- vh->finalize = info->finalize;
- vh->finalize_arg = info->finalize_arg;
- LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
- if (ar->init_vhost)
- if (ar->init_vhost(vh, info))
- return NULL;
- LWS_FOR_EVERY_AVAILABLE_ROLE_END;
- if (info->keepalive_timeout)
- vh->keepalive_timeout = info->keepalive_timeout;
- else
- vh->keepalive_timeout = 5;
- if (info->timeout_secs_ah_idle)
- vh->timeout_secs_ah_idle = info->timeout_secs_ah_idle;
- else
- vh->timeout_secs_ah_idle = 10;
- #if defined(LWS_WITH_TLS)
- vh->tls.alpn = info->alpn;
- vh->tls.ssl_info_event_mask = info->ssl_info_event_mask;
- if (info->ecdh_curve)
- lws_strncpy(vh->tls.ecdh_curve, info->ecdh_curve,
- sizeof(vh->tls.ecdh_curve));
- /* carefully allocate and take a copy of cert + key paths if present */
- n = 0;
- if (info->ssl_cert_filepath)
- n += (int)strlen(info->ssl_cert_filepath) + 1;
- if (info->ssl_private_key_filepath)
- n += (int)strlen(info->ssl_private_key_filepath) + 1;
- if (n) {
- vh->tls.key_path = vh->tls.alloc_cert_path =
- lws_malloc(n, "vh paths");
- if (info->ssl_cert_filepath) {
- n = (int)strlen(info->ssl_cert_filepath) + 1;
- memcpy(vh->tls.alloc_cert_path,
- info->ssl_cert_filepath, n);
- vh->tls.key_path += n;
- }
- if (info->ssl_private_key_filepath)
- memcpy(vh->tls.key_path, info->ssl_private_key_filepath,
- strlen(info->ssl_private_key_filepath) + 1);
- }
- #endif
- /*
- * give the vhost a unified list of protocols including the
- * ones that came from plugins
- */
- lwsp = lws_zalloc(sizeof(struct lws_protocols) * (vh->count_protocols +
- context->plugin_protocol_count + 1),
- "vhost-specific plugin table");
- if (!lwsp) {
- lwsl_err("OOM\n");
- return NULL;
- }
- m = vh->count_protocols;
- memcpy(lwsp, pcols, sizeof(struct lws_protocols) * m);
- /* for compatibility, all protocols enabled on vhost if only
- * the default vhost exists. Otherwise only vhosts who ask
- * for a protocol get it enabled.
- */
- if (context->options & LWS_SERVER_OPTION_EXPLICIT_VHOSTS)
- f = 0;
- (void)f;
- #ifdef LWS_WITH_PLUGINS
- if (plugin) {
- while (plugin) {
- for (n = 0; n < plugin->caps.count_protocols; n++) {
- /*
- * for compatibility's sake, no pvo implies
- * allow all protocols
- */
- if (f || lws_vhost_protocol_options(vh,
- plugin->caps.protocols[n].name)) {
- memcpy(&lwsp[m],
- &plugin->caps.protocols[n],
- sizeof(struct lws_protocols));
- m++;
- vh->count_protocols++;
- }
- }
- plugin = plugin->list;
- }
- }
- #endif
- if (
- #ifdef LWS_WITH_PLUGINS
- (context->plugin_list) ||
- #endif
- context->options & LWS_SERVER_OPTION_EXPLICIT_VHOSTS)
- vh->protocols = lwsp;
- else {
- vh->protocols = pcols;
- lws_free(lwsp);
- }
- vh->same_vh_protocol_heads = (struct lws_dll_lws *)
- lws_zalloc(sizeof(struct lws_dll_lws) *
- vh->count_protocols, "same vh list");
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- vh->http.mount_list = info->mounts;
- #endif
- #ifdef LWS_WITH_UNIX_SOCK
- if (LWS_UNIX_SOCK_ENABLED(vh)) {
- lwsl_notice("Creating Vhost '%s' path \"%s\", %d protocols\n",
- vh->name, vh->iface, vh->count_protocols);
- } else
- #endif
- {
- switch(info->port) {
- case CONTEXT_PORT_NO_LISTEN:
- strcpy(buf, "(serving disabled)");
- break;
- case CONTEXT_PORT_NO_LISTEN_SERVER:
- strcpy(buf, "(no listener)");
- break;
- default:
- lws_snprintf(buf, sizeof(buf), "port %u", info->port);
- break;
- }
- lwsl_notice("Creating Vhost '%s' %s, %d protocols, IPv6 %s\n",
- vh->name, buf, vh->count_protocols,
- LWS_IPV6_ENABLED(vh) ? "on" : "off");
- }
- mounts = info->mounts;
- while (mounts) {
- (void)mount_protocols[0];
- lwsl_info(" mounting %s%s to %s\n",
- mount_protocols[mounts->origin_protocol],
- mounts->origin, mounts->mountpoint);
- /* convert interpreter protocol names to pointers */
- pvo = mounts->interpret;
- while (pvo) {
- for (n = 0; n < vh->count_protocols; n++) {
- if (strcmp(pvo->value, vh->protocols[n].name))
- continue;
- ((struct lws_protocol_vhost_options *)pvo)->
- value = (const char *)(lws_intptr_t)n;
- break;
- }
- if (n == vh->count_protocols)
- lwsl_err("ignoring unknown interp pr %s\n",
- pvo->value);
- pvo = pvo->next;
- }
- mounts = mounts->mount_next;
- }
- vh->listen_port = info->port;
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- vh->http.http_proxy_port = 0;
- vh->http.http_proxy_address[0] = '\0';
- #endif
- #if defined(LWS_WITH_SOCKS5)
- vh->socks_proxy_port = 0;
- vh->socks_proxy_address[0] = '\0';
- #endif
- #if !defined(LWS_WITHOUT_CLIENT)
- /* either use proxy from info, or try get it from env var */
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- /* http proxy */
- if (info->http_proxy_address) {
- /* override for backwards compatibility */
- if (info->http_proxy_port)
- vh->http.http_proxy_port = info->http_proxy_port;
- lws_set_proxy(vh, info->http_proxy_address);
- } else
- #endif
- {
- #ifdef LWS_HAVE_GETENV
- p = getenv("http_proxy");
- if (p)
- lws_set_proxy(vh, p);
- #endif
- }
- #endif
- #if defined(LWS_WITH_SOCKS5)
- /* socks proxy */
- if (info->socks_proxy_address) {
- /* override for backwards compatibility */
- if (info->socks_proxy_port)
- vh->socks_proxy_port = info->socks_proxy_port;
- lws_set_socks(vh, info->socks_proxy_address);
- } else {
- #ifdef LWS_HAVE_GETENV
- p = getenv("socks_proxy");
- if (p)
- lws_set_socks(vh, p);
- #endif
- }
- #endif
- vh->ka_time = info->ka_time;
- vh->ka_interval = info->ka_interval;
- vh->ka_probes = info->ka_probes;
- if (vh->options & LWS_SERVER_OPTION_STS)
- lwsl_notice(" STS enabled\n");
- #ifdef LWS_WITH_ACCESS_LOG
- if (info->log_filepath) {
- vh->log_fd = lws_open(info->log_filepath,
- O_CREAT | O_APPEND | O_RDWR, 0600);
- if (vh->log_fd == (int)LWS_INVALID_FILE) {
- lwsl_err("unable to open log filepath %s\n",
- info->log_filepath);
- goto bail;
- }
- #ifndef WIN32
- if (context->uid != -1)
- if (chown(info->log_filepath, context->uid,
- context->gid) == -1)
- lwsl_err("unable to chown log file %s\n",
- info->log_filepath);
- #endif
- } else
- vh->log_fd = (int)LWS_INVALID_FILE;
- #endif
- if (lws_context_init_server_ssl(info, vh)) {
- lwsl_err("%s: lws_context_init_server_ssl failed\n", __func__);
- goto bail1;
- }
- if (lws_context_init_client_ssl(info, vh)) {
- lwsl_err("%s: lws_context_init_client_ssl failed\n", __func__);
- goto bail1;
- }
- lws_context_lock(context, "create_vhost");
- n = _lws_vhost_init_server(info, vh);
- lws_context_unlock(context);
- if (n < 0) {
- lwsl_err("init server failed\n");
- goto bail1;
- }
- while (1) {
- if (!(*vh1)) {
- *vh1 = vh;
- break;
- }
- vh1 = &(*vh1)->vhost_next;
- };
- /* for the case we are adding a vhost much later, after server init */
- if (context->protocol_init_done)
- if (lws_protocol_init(context)) {
- lwsl_err("%s: lws_protocol_init failed\n", __func__);
- goto bail1;
- }
- return vh;
- bail1:
- lws_vhost_destroy(vh);
- return NULL;
- #ifdef LWS_WITH_ACCESS_LOG
- bail:
- lws_free(vh);
- #endif
- return NULL;
- }
- LWS_VISIBLE int
- lws_init_vhost_client_ssl(const struct lws_context_creation_info *info,
- struct lws_vhost *vhost)
- {
- struct lws_context_creation_info i;
- memcpy(&i, info, sizeof(i));
- i.port = CONTEXT_PORT_NO_LISTEN;
- return lws_context_init_client_ssl(&i, vhost);
- }
- LWS_VISIBLE void
- lws_cancel_service_pt(struct lws *wsi)
- {
- lws_plat_pipe_signal(wsi);
- }
- LWS_VISIBLE void
- lws_cancel_service(struct lws_context *context)
- {
- struct lws_context_per_thread *pt = &context->pt[0];
- short m = context->count_threads;
- if (context->being_destroyed1)
- return;
- lwsl_info("%s\n", __func__);
- while (m--) {
- if (pt->pipe_wsi)
- lws_plat_pipe_signal(pt->pipe_wsi);
- pt++;
- }
- }
- int
- lws_create_event_pipes(struct lws_context *context)
- {
- struct lws *wsi;
- int n;
- /*
- * Create the pt event pipes... these are unique in that they are
- * not bound to a vhost or protocol (both are NULL)
- */
- for (n = 0; n < context->count_threads; n++) {
- if (context->pt[n].pipe_wsi)
- continue;
- wsi = lws_zalloc(sizeof(*wsi), "event pipe wsi");
- if (!wsi) {
- lwsl_err("%s: Out of mem\n", __func__);
- return 1;
- }
- wsi->context = context;
- lws_role_transition(wsi, 0, LRS_UNCONNECTED, &role_ops_pipe);
- wsi->protocol = NULL;
- wsi->tsi = n;
- wsi->vhost = NULL;
- wsi->event_pipe = 1;
- wsi->desc.sockfd = LWS_SOCK_INVALID;
- context->pt[n].pipe_wsi = wsi;
- context->count_wsi_allocated++;
- if (lws_plat_pipe_create(wsi))
- /*
- * platform code returns 0 if it actually created pipes
- * and initialized pt->dummy_pipe_fds[]. If it used
- * some other mechanism outside of signaling in the
- * normal event loop, we skip treating the pipe as
- * related to dummy_pipe_fds[], adding it to the fds,
- * etc.
- */
- continue;
- wsi->desc.sockfd = context->pt[n].dummy_pipe_fds[0];
- lwsl_debug("event pipe fd %d\n", wsi->desc.sockfd);
- if (context->event_loop_ops->accept)
- if (context->event_loop_ops->accept(wsi))
- return 1;
- if (__insert_wsi_socket_into_fds(context, wsi))
- return 1;
- }
- return 0;
- }
- void
- lws_destroy_event_pipe(struct lws *wsi)
- {
- lwsl_info("%s\n", __func__);
- __remove_wsi_socket_from_fds(wsi);
- if (wsi->context->event_loop_ops->wsi_logical_close) {
- wsi->context->event_loop_ops->wsi_logical_close(wsi);
- lws_plat_pipe_close(wsi);
- wsi->context->count_wsi_allocated--;
- return;
- }
- if (wsi->context->event_loop_ops->destroy_wsi)
- wsi->context->event_loop_ops->destroy_wsi(wsi);
- lws_plat_pipe_close(wsi);
- wsi->context->count_wsi_allocated--;
- lws_free(wsi);
- }
- LWS_VISIBLE struct lws_context *
- lws_create_context(const struct lws_context_creation_info *info)
- {
- struct lws_context *context = NULL;
- struct lws_plat_file_ops *prev;
- #ifndef LWS_NO_DAEMONIZE
- int pid_daemon = get_daemonize_pid();
- #endif
- int n;
- #if defined(__ANDROID__)
- struct rlimit rt;
- #endif
- lwsl_info("Initial logging level %d\n", log_level);
- lwsl_info("Libwebsockets version: %s\n", library_version);
- #ifdef LWS_WITH_IPV6
- if (!lws_check_opt(info->options, LWS_SERVER_OPTION_DISABLE_IPV6))
- lwsl_info("IPV6 compiled in and enabled\n");
- else
- lwsl_info("IPV6 compiled in but disabled\n");
- #else
- lwsl_info("IPV6 not compiled in\n");
- #endif
- lwsl_info(" LWS_DEF_HEADER_LEN : %u\n", LWS_DEF_HEADER_LEN);
- lwsl_info(" LWS_MAX_PROTOCOLS : %u\n", LWS_MAX_PROTOCOLS);
- lwsl_info(" LWS_MAX_SMP : %u\n", LWS_MAX_SMP);
- lwsl_info(" sizeof (*info) : %ld\n", (long)sizeof(*info));
- #if defined(LWS_WITH_STATS)
- lwsl_info(" LWS_WITH_STATS : on\n");
- #endif
- lwsl_info(" SYSTEM_RANDOM_FILEPATH: '%s'\n", SYSTEM_RANDOM_FILEPATH);
- #if defined(LWS_WITH_HTTP2)
- lwsl_info(" HTTP2 support : available\n");
- #else
- lwsl_info(" HTTP2 support : not configured\n");
- #endif
- if (lws_plat_context_early_init())
- return NULL;
- context = lws_zalloc(sizeof(struct lws_context), "context");
- if (!context) {
- lwsl_err("No memory for websocket context\n");
- return NULL;
- }
- #if defined(LWS_WITH_TLS)
- #if defined(LWS_WITH_MBEDTLS)
- context->tls_ops = &tls_ops_mbedtls;
- #else
- context->tls_ops = &tls_ops_openssl;
- #endif
- #endif
- if (info->pt_serv_buf_size)
- context->pt_serv_buf_size = info->pt_serv_buf_size;
- else
- context->pt_serv_buf_size = 4096;
- #if defined(LWS_ROLE_H2)
- role_ops_h2.init_context(context, info);
- #endif
- #if LWS_MAX_SMP > 1
- lws_mutex_refcount_init(&context->mr);
- #endif
- #if defined(LWS_WITH_ESP32)
- context->last_free_heap = esp_get_free_heap_size();
- #endif
- /* default to just the platform fops implementation */
- context->fops_platform.LWS_FOP_OPEN = _lws_plat_file_open;
- context->fops_platform.LWS_FOP_CLOSE = _lws_plat_file_close;
- context->fops_platform.LWS_FOP_SEEK_CUR = _lws_plat_file_seek_cur;
- context->fops_platform.LWS_FOP_READ = _lws_plat_file_read;
- context->fops_platform.LWS_FOP_WRITE = _lws_plat_file_write;
- context->fops_platform.fi[0].sig = NULL;
- /*
- * arrange a linear linked-list of fops starting from context->fops
- *
- * platform fops
- * [ -> fops_zip (copied into context so .next settable) ]
- * [ -> info->fops ]
- */
- context->fops = &context->fops_platform;
- prev = (struct lws_plat_file_ops *)context->fops;
- #if defined(LWS_WITH_ZIP_FOPS)
- /* make a soft copy so we can set .next */
- context->fops_zip = fops_zip;
- prev->next = &context->fops_zip;
- prev = (struct lws_plat_file_ops *)prev->next;
- #endif
- /* if user provided fops, tack them on the end of the list */
- if (info->fops)
- prev->next = info->fops;
- context->reject_service_keywords = info->reject_service_keywords;
- if (info->external_baggage_free_on_destroy)
- context->external_baggage_free_on_destroy =
- info->external_baggage_free_on_destroy;
- context->time_up = time(NULL);
- context->pcontext_finalize = info->pcontext;
- context->simultaneous_ssl_restriction =
- info->simultaneous_ssl_restriction;
- #ifndef LWS_NO_DAEMONIZE
- if (pid_daemon) {
- context->started_with_parent = pid_daemon;
- lwsl_info(" Started with daemon pid %d\n", pid_daemon);
- }
- #endif
- #if defined(__ANDROID__)
- n = getrlimit ( RLIMIT_NOFILE,&rt);
- if (-1 == n) {
- lwsl_err("Get RLIMIT_NOFILE failed!\n");
- return NULL;
- }
- context->max_fds = rt.rlim_cur;
- #else
- context->max_fds = getdtablesize();
- #endif
- if (info->count_threads)
- context->count_threads = info->count_threads;
- else
- context->count_threads = 1;
- if (context->count_threads > LWS_MAX_SMP)
- context->count_threads = LWS_MAX_SMP;
- context->token_limits = info->token_limits;
- context->options = info->options;
- /*
- * set the context event loops ops struct
- *
- * after this, all event_loop actions use the generic ops
- */
- #if defined(LWS_WITH_POLL)
- context->event_loop_ops = &event_loop_ops_poll;
- #endif
- if (lws_check_opt(context->options, LWS_SERVER_OPTION_LIBUV))
- #if defined(LWS_WITH_LIBUV)
- context->event_loop_ops = &event_loop_ops_uv;
- #else
- goto fail_event_libs;
- #endif
- if (lws_check_opt(context->options, LWS_SERVER_OPTION_LIBEV))
- #if defined(LWS_WITH_LIBEV)
- context->event_loop_ops = &event_loop_ops_ev;
- #else
- goto fail_event_libs;
- #endif
- if (lws_check_opt(context->options, LWS_SERVER_OPTION_LIBEVENT))
- #if defined(LWS_WITH_LIBEVENT)
- context->event_loop_ops = &event_loop_ops_event;
- #else
- goto fail_event_libs;
- #endif
- if (!context->event_loop_ops)
- goto fail_event_libs;
- lwsl_info("Using event loop: %s\n", context->event_loop_ops->name);
- #if defined(LWS_WITH_TLS)
- time(&context->tls.last_cert_check_s);
- if (info->alpn)
- context->tls.alpn_default = info->alpn;
- else {
- char *p = context->tls.alpn_discovered, first = 1;
- LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar) {
- if (ar->alpn) {
- if (!first)
- *p++ = ',';
- p += lws_snprintf(p,
- context->tls.alpn_discovered +
- sizeof(context->tls.alpn_discovered) -
- 2 - p, "%s", ar->alpn);
- first = 0;
- }
- } LWS_FOR_EVERY_AVAILABLE_ROLE_END;
- context->tls.alpn_default = context->tls.alpn_discovered;
- }
- lwsl_info("Default ALPN advertisment: %s\n", context->tls.alpn_default);
- #endif
- if (info->timeout_secs)
- context->timeout_secs = info->timeout_secs;
- else
- context->timeout_secs = AWAITING_TIMEOUT;
- context->ws_ping_pong_interval = info->ws_ping_pong_interval;
- lwsl_info(" default timeout (secs): %u\n", context->timeout_secs);
- if (info->max_http_header_data)
- context->max_http_header_data = info->max_http_header_data;
- else
- if (info->max_http_header_data2)
- context->max_http_header_data =
- info->max_http_header_data2;
- else
- context->max_http_header_data = LWS_DEF_HEADER_LEN;
- if (info->max_http_header_pool)
- context->max_http_header_pool = info->max_http_header_pool;
- else
- if (info->max_http_header_pool2)
- context->max_http_header_pool =
- info->max_http_header_pool2;
- else
- context->max_http_header_pool = context->max_fds;
- if (info->fd_limit_per_thread)
- context->fd_limit_per_thread = info->fd_limit_per_thread;
- else
- context->fd_limit_per_thread = context->max_fds /
- context->count_threads;
- /*
- * Allocate the per-thread storage for scratchpad buffers,
- * and header data pool
- */
- for (n = 0; n < context->count_threads; n++) {
- context->pt[n].serv_buf = lws_malloc(context->pt_serv_buf_size,
- "pt_serv_buf");
- if (!context->pt[n].serv_buf) {
- lwsl_err("OOM\n");
- return NULL;
- }
- context->pt[n].context = context;
- context->pt[n].tid = n;
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- context->pt[n].http.ah_list = NULL;
- context->pt[n].http.ah_pool_length = 0;
- #endif
- lws_pt_mutex_init(&context->pt[n]);
- }
- lwsl_info(" Threads: %d each %d fds\n", context->count_threads,
- context->fd_limit_per_thread);
- if (!info->ka_interval && info->ka_time > 0) {
- lwsl_err("info->ka_interval can't be 0 if ka_time used\n");
- return NULL;
- }
- #if defined(LWS_WITH_PEER_LIMITS)
- /* scale the peer hash table according to the max fds for the process,
- * so that the max list depth averages 16. Eg, 1024 fd -> 64,
- * 102400 fd -> 6400
- */
- context->pl_hash_elements =
- (context->count_threads * context->fd_limit_per_thread) / 16;
- context->pl_hash_table = lws_zalloc(sizeof(struct lws_peer *) *
- context->pl_hash_elements, "peer limits hash table");
- context->ip_limit_ah = info->ip_limit_ah;
- context->ip_limit_wsi = info->ip_limit_wsi;
- #endif
- lwsl_info(" mem: context: %5lu B (%ld ctx + (%ld thr x %d))\n",
- (long)sizeof(struct lws_context) +
- (context->count_threads * context->pt_serv_buf_size),
- (long)sizeof(struct lws_context),
- (long)context->count_threads,
- context->pt_serv_buf_size);
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- lwsl_info(" mem: http hdr rsvd: %5lu B (%u thr x (%u + %lu) x %u))\n",
- (long)(context->max_http_header_data +
- sizeof(struct allocated_headers)) *
- context->max_http_header_pool * context->count_threads,
- context->count_threads,
- context->max_http_header_data,
- (long)sizeof(struct allocated_headers),
- context->max_http_header_pool);
- #endif
- n = sizeof(struct lws_pollfd) * context->count_threads *
- context->fd_limit_per_thread;
- context->pt[0].fds = lws_zalloc(n, "fds table");
- if (context->pt[0].fds == NULL) {
- lwsl_err("OOM allocating %d fds\n", context->max_fds);
- goto bail;
- }
- lwsl_info(" mem: pollfd map: %5u\n", n);
- if (info->server_string) {
- context->server_string = info->server_string;
- context->server_string_len = (short)
- strlen(context->server_string);
- }
- #if LWS_MAX_SMP > 1
- /* each thread serves his own chunk of fds */
- for (n = 1; n < (int)info->count_threads; n++)
- context->pt[n].fds = context->pt[n - 1].fds +
- context->fd_limit_per_thread;
- #endif
- if (lws_plat_init(context, info))
- goto bail;
- if (context->event_loop_ops->init_context)
- if (context->event_loop_ops->init_context(context, info))
- goto bail;
- if (context->event_loop_ops->init_pt)
- for (n = 0; n < context->count_threads; n++) {
- void *lp = NULL;
- if (info->foreign_loops)
- lp = info->foreign_loops[n];
- if (context->event_loop_ops->init_pt(context, lp, n))
- goto bail;
- }
- if (lws_create_event_pipes(context))
- goto bail;
- lws_context_init_ssl_library(info);
- context->user_space = info->user;
- /*
- * if he's not saying he'll make his own vhosts later then act
- * compatibly and make a default vhost using the data in the info
- */
- if (!lws_check_opt(info->options, LWS_SERVER_OPTION_EXPLICIT_VHOSTS))
- if (!lws_create_vhost(context, info)) {
- lwsl_err("Failed to create default vhost\n");
- for (n = 0; n < context->count_threads; n++)
- lws_free_set_NULL(context->pt[n].serv_buf);
- #if defined(LWS_WITH_PEER_LIMITS)
- lws_free_set_NULL(context->pl_hash_table);
- #endif
- lws_free_set_NULL(context->pt[0].fds);
- lws_plat_context_late_destroy(context);
- lws_free_set_NULL(context);
- return NULL;
- }
- lws_context_init_extensions(info, context);
- lwsl_info(" mem: per-conn: %5lu bytes + protocol rx buf\n",
- (unsigned long)sizeof(struct lws));
- strcpy(context->canonical_hostname, "unknown");
- lws_server_get_canonical_hostname(context, info);
- context->uid = info->uid;
- context->gid = info->gid;
- #if defined(LWS_HAVE_SYS_CAPABILITY_H) && defined(LWS_HAVE_LIBCAP)
- memcpy(context->caps, info->caps, sizeof(context->caps));
- context->count_caps = info->count_caps;
- #endif
- /*
- * drop any root privs for this process
- * to listen on port < 1023 we would have needed root, but now we are
- * listening, we don't want the power for anything else
- */
- if (!lws_check_opt(info->options, LWS_SERVER_OPTION_EXPLICIT_VHOSTS))
- lws_plat_drop_app_privileges(info);
- /* expedite post-context init (eg, protocols) */
- lws_cancel_service(context);
- #if defined(LWS_WITH_SELFTESTS)
- lws_jws_selftest();
- #endif
- return context;
- bail:
- lws_context_destroy(context);
- return NULL;
- fail_event_libs:
- lwsl_err("Requested event library support not configured, available:\n");
- {
- const struct lws_event_loop_ops **elops = available_event_libs;
- while (*elops) {
- lwsl_err(" - %s\n", (*elops)->name);
- elops++;
- }
- }
- lws_free(context);
- return NULL;
- }
- LWS_VISIBLE LWS_EXTERN void
- lws_context_deprecate(struct lws_context *context, lws_reload_func cb)
- {
- struct lws_vhost *vh = context->vhost_list, *vh1;
- /*
- * "deprecation" means disable the context from accepting any new
- * connections and free up listen sockets to be used by a replacement
- * context.
- *
- * Otherwise the deprecated context remains operational, until its
- * number of connected sockets falls to zero, when it is deleted.
- */
- /* for each vhost, close his listen socket */
- while (vh) {
- struct lws *wsi = vh->lserv_wsi;
- if (wsi) {
- wsi->socket_is_permanently_unusable = 1;
- lws_close_free_wsi(wsi, LWS_CLOSE_STATUS_NOSTATUS, "ctx deprecate");
- wsi->context->deprecation_pending_listen_close_count++;
- /*
- * other vhosts can share the listen port, they
- * point to the same wsi. So zap those too.
- */
- vh1 = context->vhost_list;
- while (vh1) {
- if (vh1->lserv_wsi == wsi)
- vh1->lserv_wsi = NULL;
- vh1 = vh1->vhost_next;
- }
- }
- vh = vh->vhost_next;
- }
- context->deprecated = 1;
- context->deprecation_cb = cb;
- }
- LWS_VISIBLE LWS_EXTERN int
- lws_context_is_deprecated(struct lws_context *context)
- {
- return context->deprecated;
- }
- void
- lws_vhost_destroy1(struct lws_vhost *vh)
- {
- struct lws_context *context = vh->context;
- lwsl_info("%s\n", __func__);
- lws_context_lock(context, "vhost destroy 1"); /* ---------- context { */
- if (vh->being_destroyed)
- goto out;
- lws_vhost_lock(vh); /* -------------- vh { */
- vh->being_destroyed = 1;
- /*
- * PHASE 1: take down or reassign any listen wsi
- *
- * Are there other vhosts that are piggybacking on our listen socket?
- * If so we need to hand the listen socket off to one of the others
- * so it will remain open.
- *
- * If not, leave it attached to the closing vhost, the vh being marked
- * being_destroyed will defeat any service and it will get closed in
- * later phases.
- */
- if (vh->lserv_wsi)
- lws_start_foreach_ll(struct lws_vhost *, v,
- context->vhost_list) {
- if (v != vh &&
- !v->being_destroyed &&
- v->listen_port == vh->listen_port &&
- ((!v->iface && !vh->iface) ||
- (v->iface && vh->iface &&
- !strcmp(v->iface, vh->iface)))) {
- /*
- * this can only be a listen wsi, which is
- * restricted... it has no protocol or other
- * bindings or states. So we can simply
- * swap it to a vhost that has the same
- * iface + port, but is not closing.
- */
- assert(v->lserv_wsi == NULL);
- v->lserv_wsi = vh->lserv_wsi;
- lwsl_notice("%s: listen skt from %s to %s\n",
- __func__, vh->name, v->name);
- if (v->lserv_wsi) {
- lws_vhost_unbind_wsi(vh->lserv_wsi);
- lws_vhost_bind_wsi(v, v->lserv_wsi);
- }
- break;
- }
- } lws_end_foreach_ll(v, vhost_next);
- lws_vhost_unlock(vh); /* } vh -------------- */
- /*
- * lws_check_deferred_free() will notice there is a vhost that is
- * marked for destruction during the next 1s, for all tsi.
- *
- * It will start closing all wsi on this vhost. When the last wsi
- * is closed, it will trigger lws_vhost_destroy2()
- */
- out:
- lws_context_unlock(context); /* --------------------------- context { */
- }
- void
- __lws_vhost_destroy2(struct lws_vhost *vh)
- {
- const struct lws_protocols *protocol = NULL;
- struct lws_context *context = vh->context;
- struct lws_deferred_free *df;
- struct lws wsi;
- int n;
- /*
- * destroy any pending timed events
- */
- while (vh->timed_vh_protocol_list)
- __lws_timed_callback_remove(vh, vh->timed_vh_protocol_list);
- /*
- * let the protocols destroy the per-vhost protocol objects
- */
- memset(&wsi, 0, sizeof(wsi));
- wsi.context = vh->context;
- wsi.vhost = vh; /* not a real bound wsi */
- protocol = vh->protocols;
- if (protocol && vh->created_vhost_protocols) {
- n = 0;
- while (n < vh->count_protocols) {
- wsi.protocol = protocol;
- protocol->callback(&wsi, LWS_CALLBACK_PROTOCOL_DESTROY,
- NULL, NULL, 0);
- protocol++;
- n++;
- }
- }
- /*
- * remove vhost from context list of vhosts
- */
- lws_start_foreach_llp(struct lws_vhost **, pv, context->vhost_list) {
- if (*pv == vh) {
- *pv = vh->vhost_next;
- break;
- }
- } lws_end_foreach_llp(pv, vhost_next);
- /* add ourselves to the pending destruction list */
- vh->vhost_next = vh->context->vhost_pending_destruction_list;
- vh->context->vhost_pending_destruction_list = vh;
- lwsl_info("%s: %p\n", __func__, vh);
- /* if we are still on deferred free list, remove ourselves */
- lws_start_foreach_llp(struct lws_deferred_free **, pdf,
- context->deferred_free_list) {
- if ((*pdf)->payload == vh) {
- df = *pdf;
- *pdf = df->next;
- lws_free(df);
- break;
- }
- } lws_end_foreach_llp(pdf, next);
- /* remove ourselves from the pending destruction list */
- lws_start_foreach_llp(struct lws_vhost **, pv,
- context->vhost_pending_destruction_list) {
- if ((*pv) == vh) {
- *pv = (*pv)->vhost_next;
- break;
- }
- } lws_end_foreach_llp(pv, vhost_next);
- /*
- * Free all the allocations associated with the vhost
- */
- protocol = vh->protocols;
- if (protocol) {
- n = 0;
- while (n < vh->count_protocols) {
- if (vh->protocol_vh_privs &&
- vh->protocol_vh_privs[n]) {
- lws_free(vh->protocol_vh_privs[n]);
- vh->protocol_vh_privs[n] = NULL;
- }
- protocol++;
- n++;
- }
- }
- if (vh->protocol_vh_privs)
- lws_free(vh->protocol_vh_privs);
- lws_ssl_SSL_CTX_destroy(vh);
- lws_free(vh->same_vh_protocol_heads);
- if (context->plugin_list ||
- (context->options & LWS_SERVER_OPTION_EXPLICIT_VHOSTS))
- lws_free((void *)vh->protocols);
- LWS_FOR_EVERY_AVAILABLE_ROLE_START(ar)
- if (ar->destroy_vhost)
- ar->destroy_vhost(vh);
- LWS_FOR_EVERY_AVAILABLE_ROLE_END;
- #ifdef LWS_WITH_ACCESS_LOG
- if (vh->log_fd != (int)LWS_INVALID_FILE)
- close(vh->log_fd);
- #endif
- #if defined (LWS_WITH_TLS)
- lws_free_set_NULL(vh->tls.alloc_cert_path);
- #endif
- #if LWS_MAX_SMP > 1
- pthread_mutex_destroy(&vh->lock);
- #endif
- #if defined(LWS_WITH_UNIX_SOCK)
- if (LWS_UNIX_SOCK_ENABLED(vh)) {
- n = unlink(vh->iface);
- if (n)
- lwsl_info("Closing unix socket %s: errno %d\n",
- vh->iface, errno);
- }
- #endif
- /*
- * although async event callbacks may still come for wsi handles with
- * pending close in the case of asycn event library like libuv,
- * they do not refer to the vhost. So it's safe to free.
- */
- if (vh->finalize)
- vh->finalize(vh, vh->finalize_arg);
- lwsl_info(" %s: Freeing vhost %p\n", __func__, vh);
- memset(vh, 0, sizeof(*vh));
- lws_free(vh);
- }
- /*
- * each service thread calls this once a second or so
- */
- int
- lws_check_deferred_free(struct lws_context *context, int tsi, int force)
- {
- struct lws_context_per_thread *pt;
- int n;
- /*
- * If we see a vhost is being destroyed, forcibly close every wsi on
- * this tsi associated with this vhost. That will include the listen
- * socket if it is still associated with the closing vhost.
- *
- * For SMP, we do this once per tsi per destroyed vhost. The reference
- * counting on the vhost as the bound wsi close will notice that there
- * are no bound wsi left, that vhost destruction can complete,
- * and perform it. It doesn't matter which service thread does that
- * because there is nothing left using the vhost to conflict.
- */
- lws_context_lock(context, "check deferred free"); /* ------ context { */
- lws_start_foreach_ll_safe(struct lws_vhost *, v, context->vhost_list, vhost_next) {
- if (v->being_destroyed
- #if LWS_MAX_SMP > 1
- && !v->close_flow_vs_tsi[tsi]
- #endif
- ) {
- pt = &context->pt[tsi];
- lws_pt_lock(pt, "vhost removal"); /* -------------- pt { */
- #if LWS_MAX_SMP > 1
- v->close_flow_vs_tsi[tsi] = 1;
- #endif
- for (n = 0; (unsigned int)n < pt->fds_count; n++) {
- struct lws *wsi = wsi_from_fd(context, pt->fds[n].fd);
- if (!wsi)
- continue;
- if (wsi->vhost != v)
- continue;
- __lws_close_free_wsi(wsi,
- LWS_CLOSE_STATUS_NOSTATUS_CONTEXT_DESTROY,
- "vh destroy"
- /* no protocol close */);
- n--;
- }
- lws_pt_unlock(pt); /* } pt -------------- */
- }
- } lws_end_foreach_ll_safe(v);
- lws_context_unlock(context); /* } context ------------------- */
- return 0;
- }
- LWS_VISIBLE void
- lws_vhost_destroy(struct lws_vhost *vh)
- {
- struct lws_deferred_free *df = lws_malloc(sizeof(*df), "deferred free");
- struct lws_context *context = vh->context;
- if (!df)
- return;
- lws_context_lock(context, __func__); /* ------ context { */
- lws_vhost_destroy1(vh);
- if (!vh->count_bound_wsi) {
- /*
- * After listen handoff, there are already no wsi bound to this
- * vhost by any pt: nothing can be servicing any wsi belonging
- * to it any more.
- *
- * Finalize the vh destruction immediately
- */
- __lws_vhost_destroy2(vh);
- lws_free(df);
- goto out;
- }
- /* part 2 is deferred to allow all the handle closes to complete */
- df->next = vh->context->deferred_free_list;
- df->deadline = lws_now_secs();
- df->payload = vh;
- vh->context->deferred_free_list = df;
- out:
- lws_context_unlock(context); /* } context ------------------- */
- }
- /*
- * When using an event loop, the context destruction is in three separate
- * parts. This is to cover both internal and foreign event loops cleanly.
- *
- * - lws_context_destroy() simply starts a soft close of all wsi and
- * related allocations. The event loop continues.
- *
- * As the closes complete in the event loop, reference counting is used
- * to determine when everything is closed. It then calls
- * lws_context_destroy2().
- *
- * - lws_context_destroy2() cleans up the rest of the higher-level logical
- * lws pieces like vhosts. If the loop was foreign, it then proceeds to
- * lws_context_destroy3(). If it the loop is internal, it stops the
- * internal loops and waits for lws_context_destroy() to be called again
- * outside the event loop (since we cannot destroy the loop from
- * within the loop). That will cause lws_context_destroy3() to run
- * directly.
- *
- * - lws_context_destroy3() destroys any internal event loops and then
- * destroys the context itself, setting what was info.pcontext to NULL.
- */
- /*
- * destroy the actual context itself
- */
- static void
- lws_context_destroy3(struct lws_context *context)
- {
- struct lws_context **pcontext_finalize = context->pcontext_finalize;
- int n;
- for (n = 0; n < context->count_threads; n++) {
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- struct lws_context_per_thread *pt = &context->pt[n];
- #endif
- if (context->event_loop_ops->destroy_pt)
- context->event_loop_ops->destroy_pt(context, n);
- lws_free_set_NULL(context->pt[n].serv_buf);
- #if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
- while (pt->http.ah_list)
- _lws_destroy_ah(pt, pt->http.ah_list);
- #endif
- }
- if (context->pt[0].fds)
- lws_free_set_NULL(context->pt[0].fds);
- lws_free(context);
- lwsl_info("%s: ctx %p freed\n", __func__, context);
- if (pcontext_finalize)
- *pcontext_finalize = NULL;
- }
- /*
- * really start destroying things
- */
- void
- lws_context_destroy2(struct lws_context *context)
- {
- struct lws_vhost *vh = NULL, *vh1;
- #if defined(LWS_WITH_PEER_LIMITS)
- uint32_t nu;
- #endif
- lwsl_info("%s: ctx %p\n", __func__, context);
- lws_context_lock(context, "context destroy 2"); /* ------ context { */
- context->being_destroyed2 = 1;
- /*
- * free all the per-vhost allocations
- */
- vh = context->vhost_list;
- while (vh) {
- vh1 = vh->vhost_next;
- __lws_vhost_destroy2(vh);
- vh = vh1;
- }
- /* remove ourselves from the pending destruction list */
- while (context->vhost_pending_destruction_list)
- /* removes itself from list */
- __lws_vhost_destroy2(context->vhost_pending_destruction_list);
- lws_stats_log_dump(context);
- lws_ssl_context_destroy(context);
- lws_plat_context_late_destroy(context);
- #if defined(LWS_WITH_PEER_LIMITS)
- for (nu = 0; nu < context->pl_hash_elements; nu++) {
- lws_start_foreach_llp(struct lws_peer **, peer,
- context->pl_hash_table[nu]) {
- struct lws_peer *df = *peer;
- *peer = df->next;
- lws_free(df);
- continue;
- } lws_end_foreach_llp(peer, next);
- }
- lws_free(context->pl_hash_table);
- #endif
- if (context->external_baggage_free_on_destroy)
- free(context->external_baggage_free_on_destroy);
- lws_check_deferred_free(context, 0, 1);
- #if LWS_MAX_SMP > 1
- lws_mutex_refcount_destroy(&context->mr);
- #endif
- if (context->event_loop_ops->destroy_context2)
- if (context->event_loop_ops->destroy_context2(context)) {
- lws_context_unlock(context); /* } context ----------- */
- context->finalize_destroy_after_internal_loops_stopped = 1;
- return;
- }
- if (!context->pt[0].event_loop_foreign) {
- int n;
- for (n = 0; n < context->count_threads; n++)
- if (context->pt[n].inside_service) {
- lws_context_unlock(context); /* } context --- */
- return;
- }
- }
- lws_context_unlock(context); /* } context ------------------- */
- lws_context_destroy3(context);
- }
- /*
- * Begin the context takedown
- */
- LWS_VISIBLE void
- lws_context_destroy(struct lws_context *context)
- {
- volatile struct lws_foreign_thread_pollfd *ftp, *next;
- volatile struct lws_context_per_thread *vpt;
- struct lws_vhost *vh = NULL;
- struct lws wsi;
- int n, m;
- if (!context)
- return;
- if (context->finalize_destroy_after_internal_loops_stopped) {
- if (context->event_loop_ops->destroy_context2)
- context->event_loop_ops->destroy_context2(context);
- lws_context_destroy3(context);
- return;
- }
- if (context->being_destroyed1) {
- if (!context->being_destroyed2) {
- lws_context_destroy2(context);
- return;
- }
- lwsl_info("%s: ctx %p: already being destroyed\n",
- __func__, context);
- lws_context_destroy3(context);
- return;
- }
- lwsl_info("%s: ctx %p\n", __func__, context);
- m = context->count_threads;
- context->being_destroyed = 1;
- context->being_destroyed1 = 1;
- context->requested_kill = 1;
- memset(&wsi, 0, sizeof(wsi));
- wsi.context = context;
- #ifdef LWS_LATENCY
- if (context->worst_latency_info[0])
- lwsl_notice("Worst latency: %s\n", context->worst_latency_info);
- #endif
- while (m--) {
- struct lws_context_per_thread *pt = &context->pt[m];
- vpt = (volatile struct lws_context_per_thread *)pt;
- ftp = vpt->foreign_pfd_list;
- while (ftp) {
- next = ftp->next;
- lws_free((void *)ftp);
- ftp = next;
- }
- vpt->foreign_pfd_list = NULL;
- for (n = 0; (unsigned int)n < context->pt[m].fds_count; n++) {
- struct lws *wsi = wsi_from_fd(context, pt->fds[n].fd);
- if (!wsi)
- continue;
- if (wsi->event_pipe)
- lws_destroy_event_pipe(wsi);
- else
- lws_close_free_wsi(wsi,
- LWS_CLOSE_STATUS_NOSTATUS_CONTEXT_DESTROY,
- "ctx destroy"
- /* no protocol close */);
- n--;
- }
- lws_pt_mutex_destroy(pt);
- }
- /*
- * inform all the protocols that they are done and will have no more
- * callbacks.
- *
- * We can't free things until after the event loop shuts down.
- */
- if (context->protocol_init_done)
- vh = context->vhost_list;
- while (vh) {
- struct lws_vhost *vhn = vh->vhost_next;
- lws_vhost_destroy1(vh);
- vh = vhn;
- }
- lws_plat_context_early_destroy(context);
- /*
- * We face two different needs depending if foreign loop or not.
- *
- * 1) If foreign loop, we really want to advance the destroy_context()
- * past here, and block only for libuv-style async close completion.
- *
- * 2a) If poll, and we exited by ourselves and are calling a final
- * destroy_context() outside of any service already, we want to
- * advance all the way in one step.
- *
- * 2b) If poll, and we are reacting to a SIGINT, service thread(s) may
- * be in poll wait or servicing. We can't advance the
- * destroy_context() to the point it's freeing things; we have to
- * leave that for the final destroy_context() after the service
- * thread(s) are finished calling for service.
- */
- if (context->event_loop_ops->destroy_context1) {
- context->event_loop_ops->destroy_context1(context);
- return;
- }
- lws_context_destroy2(context);
- }
|