| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874 |
- /*
- * 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"
- #include "private-lib-event-libs-libuv.h"
- #define pt_to_priv_uv(_pt) ((struct lws_pt_eventlibs_libuv *)(_pt)->evlib_pt)
- #define wsi_to_priv_uv(_w) ((struct lws_wsi_eventlibs_libuv *)(_w)->evlib_wsi)
- static void
- lws_uv_sultimer_cb(uv_timer_t *timer
- #if UV_VERSION_MAJOR == 0
- , int status
- #endif
- )
- {
- struct lws_pt_eventlibs_libuv *ptpr = lws_container_of(timer,
- struct lws_pt_eventlibs_libuv, sultimer);
- struct lws_context_per_thread *pt = ptpr->pt;
- lws_usec_t us;
- lws_context_lock(pt->context, __func__);
- lws_pt_lock(pt, __func__);
- us = __lws_sul_service_ripe(pt->pt_sul_owner, LWS_COUNT_PT_SUL_OWNERS,
- lws_now_usecs());
- if (us)
- uv_timer_start(&pt_to_priv_uv(pt)->sultimer, lws_uv_sultimer_cb,
- LWS_US_TO_MS(us), 0);
- lws_pt_unlock(pt);
- lws_context_unlock(pt->context);
- }
- static void
- lws_uv_idle(uv_idle_t *handle
- #if UV_VERSION_MAJOR == 0
- , int status
- #endif
- )
- { struct lws_pt_eventlibs_libuv *ptpr = lws_container_of(handle,
- struct lws_pt_eventlibs_libuv, idle);
- struct lws_context_per_thread *pt = ptpr->pt;
- lws_usec_t us;
- lws_service_do_ripe_rxflow(pt);
- lws_context_lock(pt->context, __func__);
- lws_pt_lock(pt, __func__);
- /*
- * is there anybody with pending stuff that needs service forcing?
- */
- if (!lws_service_adjust_timeout(pt->context, 1, pt->tid))
- /* -1 timeout means just do forced service */
- _lws_plat_service_forced_tsi(pt->context, pt->tid);
- /* account for sultimer */
- us = __lws_sul_service_ripe(pt->pt_sul_owner, LWS_COUNT_PT_SUL_OWNERS,
- lws_now_usecs());
- if (us)
- uv_timer_start(&pt_to_priv_uv(pt)->sultimer, lws_uv_sultimer_cb,
- LWS_US_TO_MS(us), 0);
- /* there is nobody who needs service forcing, shut down idle */
- uv_idle_stop(handle);
- lws_pt_unlock(pt);
- lws_context_unlock(pt->context);
- }
- static void
- lws_io_cb(uv_poll_t *watcher, int status, int revents)
- {
- struct lws *wsi = (struct lws *)((uv_handle_t *)watcher)->data;
- struct lws_context *context = wsi->a.context;
- struct lws_context_per_thread *pt = &context->pt[(int)wsi->tsi];
- struct lws_pollfd eventfd;
- lws_context_lock(pt->context, __func__);
- lws_pt_lock(pt, __func__);
- if (pt->is_destroyed)
- goto bail;
- #if defined(WIN32) || defined(_WIN32)
- eventfd.fd = watcher->socket;
- #else
- eventfd.fd = watcher->io_watcher.fd;
- #endif
- eventfd.events = 0;
- eventfd.revents = 0;
- if (status < 0) {
- /*
- * At this point status will be an UV error, like UV_EBADF,
- * we treat all errors as LWS_POLLHUP
- *
- * You might want to return; instead of servicing the fd in
- * some cases */
- if (status == UV_EAGAIN)
- goto bail;
- eventfd.events |= LWS_POLLHUP;
- eventfd.revents |= LWS_POLLHUP;
- } else {
- if (revents & UV_READABLE) {
- eventfd.events |= LWS_POLLIN;
- eventfd.revents |= LWS_POLLIN;
- }
- if (revents & UV_WRITABLE) {
- eventfd.events |= LWS_POLLOUT;
- eventfd.revents |= LWS_POLLOUT;
- }
- }
- lws_pt_unlock(pt);
- lws_context_unlock(pt->context);
- lws_service_fd_tsi(context, &eventfd, wsi->tsi);
- if (pt->destroy_self) {
- lws_context_destroy(pt->context);
- return;
- }
- uv_idle_start(&pt_to_priv_uv(pt)->idle, lws_uv_idle);
- return;
- bail:
- lws_pt_unlock(pt);
- lws_context_unlock(pt->context);
- }
- /*
- * This does not actually stop the event loop. The reason is we have to pass
- * libuv handle closures through its event loop. So this tries to close all
- * wsi, and set a flag; when all the wsi closures are finalized then we
- * actually stop the libuv event loops.
- */
- static void
- lws_libuv_stop(struct lws_context *context)
- {
- struct lws_context_per_thread *pt;
- int n, m;
- lwsl_err("%s\n", __func__);
- if (context->requested_kill) {
- lwsl_err("%s: ignoring\n", __func__);
- return;
- }
- context->requested_kill = 1;
- m = context->count_threads;
- context->being_destroyed = 1;
- /*
- * Phase 1: start the close of every dynamic uv handle
- */
- while (m--) {
- pt = &context->pt[m];
- if (pt->pipe_wsi) {
- uv_poll_stop(wsi_to_priv_uv(pt->pipe_wsi)->w_read.pwatcher);
- lws_destroy_event_pipe(pt->pipe_wsi);
- pt->pipe_wsi = 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;
- lws_close_free_wsi(wsi,
- LWS_CLOSE_STATUS_NOSTATUS_CONTEXT_DESTROY,
- __func__ /* no protocol close */);
- n--;
- }
- }
- lwsl_info("%s: started closing all wsi\n", __func__);
- /* we cannot have completed... there are at least the cancel pipes */
- }
- static void
- lws_uv_signal_handler(uv_signal_t *watcher, int signum)
- {
- struct lws_context *context = watcher->data;
- if (context->eventlib_signal_cb) {
- context->eventlib_signal_cb((void *)watcher, signum);
- return;
- }
- lwsl_err("internal signal handler caught signal %d\n", signum);
- lws_libuv_stop(watcher->data);
- }
- static const int sigs[] = { SIGINT, SIGTERM, SIGSEGV, SIGFPE, SIGHUP };
- /*
- * Closing Phase 2: Close callback for a static UV asset
- */
- static void
- lws_uv_close_cb_sa(uv_handle_t *handle)
- {
- struct lws_context *context =
- LWS_UV_REFCOUNT_STATIC_HANDLE_TO_CONTEXT(handle);
- int n;
- lwsl_info("%s: sa left %d: dyn left: %d\n", __func__,
- context->count_event_loop_static_asset_handles,
- context->count_wsi_allocated);
- /* any static assets left? */
- if (LWS_UV_REFCOUNT_STATIC_HANDLE_DESTROYED(handle) ||
- context->count_wsi_allocated)
- return;
- /*
- * That's it... all wsi were down, and now every
- * static asset lws had a UV handle for is down.
- *
- * Stop the loop so we can get out of here.
- */
- for (n = 0; n < context->count_threads; n++) {
- struct lws_context_per_thread *pt = &context->pt[n];
- if (pt_to_priv_uv(pt)->io_loop && !pt->event_loop_foreign)
- uv_stop(pt_to_priv_uv(pt)->io_loop);
- }
- if (!context->pt[0].event_loop_foreign) {
- lwsl_info("%s: calling lws_context_destroy2\n", __func__);
- lws_context_destroy2(context);
- }
- lwsl_info("%s: all done\n", __func__);
- }
- /*
- * These must be called by protocols that want to use libuv objects directly...
- *
- * .... when the libuv object is created...
- */
- void
- lws_libuv_static_refcount_add(uv_handle_t *h, struct lws_context *context)
- {
- LWS_UV_REFCOUNT_STATIC_HANDLE_NEW(h, context);
- }
- /*
- * ... and in the close callback when the object is closed.
- */
- void
- lws_libuv_static_refcount_del(uv_handle_t *h)
- {
- lws_uv_close_cb_sa(h);
- }
- static void lws_uv_close_cb(uv_handle_t *handle)
- {
- }
- static void lws_uv_walk_cb(uv_handle_t *handle, void *arg)
- {
- if (!uv_is_closing(handle))
- uv_close(handle, lws_uv_close_cb);
- }
- void
- lws_close_all_handles_in_loop(uv_loop_t *loop)
- {
- uv_walk(loop, lws_uv_walk_cb, NULL);
- }
- void
- lws_libuv_stop_without_kill(const struct lws_context *context, int tsi)
- {
- if (pt_to_priv_uv(&context->pt[tsi])->io_loop)
- uv_stop(pt_to_priv_uv(&context->pt[tsi])->io_loop);
- }
- uv_loop_t *
- lws_uv_getloop(struct lws_context *context, int tsi)
- {
- if (pt_to_priv_uv(&context->pt[tsi])->io_loop)
- return pt_to_priv_uv(&context->pt[tsi])->io_loop;
- return NULL;
- }
- int
- lws_libuv_check_watcher_active(struct lws *wsi)
- {
- uv_handle_t *h = (uv_handle_t *)wsi_to_priv_uv(wsi)->w_read.pwatcher;
- if (!h)
- return 0;
- return uv_is_active(h);
- }
- static int
- elops_init_context_uv(struct lws_context *context,
- const struct lws_context_creation_info *info)
- {
- int n;
- context->eventlib_signal_cb = info->signal_cb;
- for (n = 0; n < context->count_threads; n++)
- pt_to_priv_uv(&context->pt[n])->w_sigint.context = context;
- return 0;
- }
- static int
- elops_destroy_context1_uv(struct lws_context *context)
- {
- struct lws_context_per_thread *pt;
- int n, m = 0;
- for (n = 0; n < context->count_threads; n++) {
- int budget = 10000;
- pt = &context->pt[n];
- /* only for internal loops... */
- if (!pt->event_loop_foreign) {
- while (budget-- && (m = uv_run(pt_to_priv_uv(pt)->io_loop,
- UV_RUN_NOWAIT)))
- ;
- if (m)
- lwsl_info("%s: tsi %d: not all closed\n",
- __func__, n);
- }
- }
- /* call destroy2 if internal loop */
- return !context->pt[0].event_loop_foreign;
- }
- static int
- elops_destroy_context2_uv(struct lws_context *context)
- {
- struct lws_context_per_thread *pt;
- int n, internal = 0;
- for (n = 0; n < context->count_threads; n++) {
- pt = &context->pt[n];
- /* only for internal loops... */
- if (!pt->event_loop_foreign && pt_to_priv_uv(pt)->io_loop) {
- internal = 1;
- if (!context->finalize_destroy_after_internal_loops_stopped)
- uv_stop(pt_to_priv_uv(pt)->io_loop);
- else {
- #if UV_VERSION_MAJOR > 0
- uv_loop_close(pt_to_priv_uv(pt)->io_loop);
- #endif
- lws_free_set_NULL(pt_to_priv_uv(pt)->io_loop);
- }
- }
- }
- return internal;
- }
- static int
- elops_wsi_logical_close_uv(struct lws *wsi)
- {
- if (!lws_socket_is_valid(wsi->desc.sockfd) &&
- wsi->role_ops && strcmp(wsi->role_ops->name, "raw-file"))
- return 0;
- if (wsi->listener || wsi->event_pipe) {
- lwsl_debug("%s: %p: %d %d stop listener / pipe poll\n",
- __func__, wsi, wsi->listener, wsi->event_pipe);
- if (wsi_to_priv_uv(wsi)->w_read.pwatcher)
- uv_poll_stop(wsi_to_priv_uv(wsi)->w_read.pwatcher);
- }
- lwsl_debug("%s: lws_libuv_closehandle: wsi %p\n", __func__, wsi);
- /*
- * libuv has to do his own close handle processing asynchronously
- */
- lws_libuv_closehandle(wsi);
- return 1; /* do not complete the wsi close, uv close cb will do it */
- }
- static int
- elops_check_client_connect_ok_uv(struct lws *wsi)
- {
- if (lws_libuv_check_watcher_active(wsi)) {
- lwsl_warn("Waiting for libuv watcher to close\n");
- return 1;
- }
- return 0;
- }
- static void
- lws_libuv_closewsi_m(uv_handle_t* handle)
- {
- lws_sockfd_type sockfd = (lws_sockfd_type)(lws_intptr_t)handle->data;
- lwsl_debug("%s: sockfd %d\n", __func__, sockfd);
- compatible_close(sockfd);
- lws_free(handle);
- }
- static void
- elops_close_handle_manually_uv(struct lws *wsi)
- {
- uv_handle_t *h = (uv_handle_t *)wsi_to_priv_uv(wsi)->w_read.pwatcher;
- lwsl_debug("%s: lws_libuv_closehandle: wsi %p\n", __func__, wsi);
- /*
- * the "manual" variant only closes the handle itself and the
- * related fd. handle->data is the fd.
- */
- h->data = (void *)(lws_intptr_t)wsi->desc.sockfd;
- /*
- * We take responsibility to close / destroy these now.
- * Remove any trace from the wsi.
- */
- wsi->desc.sockfd = LWS_SOCK_INVALID;
- wsi_to_priv_uv(wsi)->w_read.pwatcher = NULL;
- wsi->told_event_loop_closed = 1;
- uv_close(h, lws_libuv_closewsi_m);
- }
- static int
- elops_accept_uv(struct lws *wsi)
- {
- struct lws_context_per_thread *pt = &wsi->a.context->pt[(int)wsi->tsi];
- struct lws_io_watcher_libuv *w_read = &wsi_to_priv_uv(wsi)->w_read;
- w_read->context = wsi->a.context;
- w_read->pwatcher = lws_malloc(sizeof(*w_read->pwatcher), "uvh");
- if (!w_read->pwatcher)
- return -1;
- if (wsi->role_ops->file_handle)
- uv_poll_init(pt_to_priv_uv(pt)->io_loop, w_read->pwatcher,
- (int)(lws_intptr_t)wsi->desc.filefd);
- else
- uv_poll_init_socket(pt_to_priv_uv(pt)->io_loop,
- w_read->pwatcher, wsi->desc.sockfd);
- ((uv_handle_t *)w_read->pwatcher)->data = (void *)wsi;
- return 0;
- }
- static void
- elops_io_uv(struct lws *wsi, int flags)
- {
- struct lws_context_per_thread *pt = &wsi->a.context->pt[(int)wsi->tsi];
- struct lws_io_watcher_libuv *w = &(wsi_to_priv_uv(wsi)->w_read);
- int current_events = w->actual_events & (UV_READABLE | UV_WRITABLE);
- lwsl_debug("%s: %p: %d\n", __func__, wsi, flags);
- /* w->context is set after the loop is initialized */
- if (!pt_to_priv_uv(pt)->io_loop || !w->context) {
- lwsl_info("%s: no io loop yet\n", __func__);
- return;
- }
- if (!((flags & (LWS_EV_START | LWS_EV_STOP)) &&
- (flags & (LWS_EV_READ | LWS_EV_WRITE)))) {
- lwsl_err("%s: assert: flags %d", __func__, flags);
- assert(0);
- }
- if (!w->pwatcher || wsi->told_event_loop_closed) {
- lwsl_info("%s: no watcher\n", __func__);
- return;
- }
- if (flags & LWS_EV_START) {
- if (flags & LWS_EV_WRITE)
- current_events |= UV_WRITABLE;
- if (flags & LWS_EV_READ)
- current_events |= UV_READABLE;
- uv_poll_start(w->pwatcher, current_events, lws_io_cb);
- } else {
- if (flags & LWS_EV_WRITE)
- current_events &= ~UV_WRITABLE;
- if (flags & LWS_EV_READ)
- current_events &= ~UV_READABLE;
- if (!(current_events & (UV_READABLE | UV_WRITABLE)))
- uv_poll_stop(w->pwatcher);
- else
- uv_poll_start(w->pwatcher, current_events, lws_io_cb);
- }
- w->actual_events = current_events;
- }
- static int
- elops_init_vhost_listen_wsi_uv(struct lws *wsi)
- {
- struct lws_context_per_thread *pt;
- struct lws_io_watcher_libuv *w_read;
- int n;
- if (!wsi)
- return 0;
- w_read = &wsi_to_priv_uv(wsi)->w_read;
- if (w_read->context)
- return 0;
- pt = &wsi->a.context->pt[(int)wsi->tsi];
- if (!pt_to_priv_uv(pt)->io_loop)
- return 0;
- w_read->context = wsi->a.context;
- w_read->pwatcher = lws_malloc(sizeof(*w_read->pwatcher), "uvh");
- if (!w_read->pwatcher)
- return -1;
- n = uv_poll_init_socket(pt_to_priv_uv(pt)->io_loop,
- w_read->pwatcher, wsi->desc.sockfd);
- if (n) {
- lwsl_err("uv_poll_init failed %d, sockfd=%p\n", n,
- (void *)(lws_intptr_t)wsi->desc.sockfd);
- return -1;
- }
- ((uv_handle_t *)w_read->pwatcher)->data = (void *)wsi;
- elops_io_uv(wsi, LWS_EV_START | LWS_EV_READ);
- return 0;
- }
- static void
- elops_run_pt_uv(struct lws_context *context, int tsi)
- {
- if (pt_to_priv_uv(&context->pt[tsi])->io_loop)
- uv_run(pt_to_priv_uv(&context->pt[tsi])->io_loop, 0);
- }
- static void
- elops_destroy_pt_uv(struct lws_context *context, int tsi)
- {
- struct lws_context_per_thread *pt = &context->pt[tsi];
- int m, ns;
- lwsl_info("%s: %d\n", __func__, tsi);
- if (!lws_check_opt(context->options, LWS_SERVER_OPTION_LIBUV))
- return;
- if (!pt_to_priv_uv(pt)->io_loop)
- return;
- if (pt->event_loop_destroy_processing_done)
- return;
- pt->event_loop_destroy_processing_done = 1;
- if (!pt->event_loop_foreign) {
- uv_signal_stop(&pt_to_priv_uv(pt)->w_sigint.watcher);
- ns = LWS_ARRAY_SIZE(sigs);
- if (lws_check_opt(context->options,
- LWS_SERVER_OPTION_UV_NO_SIGSEGV_SIGFPE_SPIN))
- ns = 2;
- for (m = 0; m < ns; m++) {
- uv_signal_stop(&pt_to_priv_uv(pt)->signals[m]);
- uv_close((uv_handle_t *)&pt_to_priv_uv(pt)->signals[m],
- lws_uv_close_cb_sa);
- }
- } else
- lwsl_debug("%s: not closing pt signals\n", __func__);
- uv_timer_stop(&pt_to_priv_uv(pt)->sultimer);
- uv_close((uv_handle_t *)&pt_to_priv_uv(pt)->sultimer, lws_uv_close_cb_sa);
- uv_idle_stop(&pt_to_priv_uv(pt)->idle);
- uv_close((uv_handle_t *)&pt_to_priv_uv(pt)->idle, lws_uv_close_cb_sa);
- }
- /*
- * This needs to be called after vhosts have been defined.
- *
- * If later, after server start, another vhost is added, this must be
- * called again to bind the vhost
- */
- int
- elops_init_pt_uv(struct lws_context *context, void *_loop, int tsi)
- {
- struct lws_context_per_thread *pt = &context->pt[tsi];
- struct lws_pt_eventlibs_libuv *ptpriv = pt_to_priv_uv(pt);
- struct lws_vhost *vh = context->vhost_list;
- int status = 0, n, ns, first = 1;
- uv_loop_t *loop = (uv_loop_t *)_loop;
- ptpriv->pt = pt;
- if (!ptpriv->io_loop) {
- if (!loop) {
- loop = lws_malloc(sizeof(*loop), "libuv loop");
- if (!loop) {
- lwsl_err("OOM\n");
- return -1;
- }
- #if UV_VERSION_MAJOR > 0
- uv_loop_init(loop);
- #else
- lwsl_err("This libuv is too old to work...\n");
- return 1;
- #endif
- pt->event_loop_foreign = 0;
- } else {
- lwsl_notice(" Using foreign event loop...\n");
- pt->event_loop_foreign = 1;
- }
- ptpriv->io_loop = loop;
- uv_idle_init(loop, &ptpriv->idle);
- LWS_UV_REFCOUNT_STATIC_HANDLE_NEW(&ptpriv->idle, context);
- uv_idle_start(&ptpriv->idle, lws_uv_idle);
- ns = LWS_ARRAY_SIZE(sigs);
- if (lws_check_opt(context->options,
- LWS_SERVER_OPTION_UV_NO_SIGSEGV_SIGFPE_SPIN))
- ns = 2;
- if (!pt->event_loop_foreign) {
- assert(ns <= (int)LWS_ARRAY_SIZE(ptpriv->signals));
- for (n = 0; n < ns; n++) {
- uv_signal_init(loop, &ptpriv->signals[n]);
- LWS_UV_REFCOUNT_STATIC_HANDLE_NEW(&ptpriv->signals[n],
- context);
- ptpriv->signals[n].data = pt->context;
- uv_signal_start(&ptpriv->signals[n],
- lws_uv_signal_handler, sigs[n]);
- }
- }
- } else
- first = 0;
- /*
- * Initialize the accept wsi read watcher with all the listening sockets
- * and register a callback for read operations
- *
- * We have to do it here because the uv loop(s) are not
- * initialized until after context creation.
- */
- while (vh) {
- if (elops_init_vhost_listen_wsi_uv(vh->lserv_wsi) == -1)
- return -1;
- vh = vh->vhost_next;
- }
- if (!first)
- return status;
- uv_timer_init(ptpriv->io_loop, &ptpriv->sultimer);
- LWS_UV_REFCOUNT_STATIC_HANDLE_NEW(&ptpriv->sultimer, context);
- return status;
- }
- static void
- lws_libuv_closewsi(uv_handle_t* handle)
- {
- struct lws *wsi = (struct lws *)handle->data;
- struct lws_context *context = lws_get_context(wsi);
- struct lws_context_per_thread *pt = &context->pt[(int)wsi->tsi];
- #if defined(LWS_WITH_SERVER)
- int lspd = 0;
- #endif
- lwsl_info("%s: %p\n", __func__, wsi);
- lws_context_lock(context, __func__);
- /*
- * We get called back here for every wsi that closes
- */
- #if defined(LWS_WITH_SERVER)
- if (wsi->role_ops && !strcmp(wsi->role_ops->name, "listen") &&
- wsi->a.context->deprecated) {
- lspd = 1;
- context->deprecation_pending_listen_close_count--;
- if (!context->deprecation_pending_listen_close_count)
- lspd = 2;
- }
- #endif
- lws_pt_lock(pt, __func__);
- __lws_close_free_wsi_final(wsi);
- lws_pt_unlock(pt);
- /* it's our job to close the handle finally */
- lws_free(handle);
- #if defined(LWS_WITH_SERVER)
- if (lspd == 2 && context->deprecation_cb) {
- lwsl_notice("calling deprecation callback\n");
- context->deprecation_cb();
- }
- #endif
- lwsl_info("%s: sa left %d: dyn left: %d (rk %d)\n", __func__,
- context->count_event_loop_static_asset_handles,
- context->count_wsi_allocated, context->requested_kill);
- /*
- * eventually, we closed all the wsi...
- */
- if (context->requested_kill && !context->count_wsi_allocated) {
- struct lws_vhost *vh = context->vhost_list;
- int m;
- /*
- * Start Closing Phase 2: close of static handles
- */
- lwsl_info("%s: all lws dynamic handles down, closing static\n",
- __func__);
- for (m = 0; m < context->count_threads; m++)
- elops_destroy_pt_uv(context, m);
- /* protocols may have initialized libuv objects */
- while (vh) {
- lws_vhost_destroy1(vh);
- vh = vh->vhost_next;
- }
- if (!context->count_event_loop_static_asset_handles &&
- context->pt[0].event_loop_foreign) {
- lwsl_info("%s: call lws_context_destroy2\n", __func__);
- lws_context_unlock(context);
- lws_context_destroy2(context);
- return;
- }
- }
- lws_context_unlock(context);
- }
- void
- lws_libuv_closehandle(struct lws *wsi)
- {
- uv_handle_t* handle;
- struct lws_io_watcher_libuv *w_read = &wsi_to_priv_uv(wsi)->w_read;
- if (!w_read->pwatcher)
- return;
- if (wsi->told_event_loop_closed) {
- // assert(0);
- return;
- }
- lwsl_debug("%s: %p\n", __func__, wsi);
- wsi->told_event_loop_closed = 1;
- /*
- * The normal close path attaches the related wsi as the
- * handle->data.
- */
- handle = (uv_handle_t *)w_read->pwatcher;
- /* ensure we can only do this once */
- w_read->pwatcher = NULL;
- uv_close(handle, lws_libuv_closewsi);
- }
- static const struct lws_event_loop_ops event_loop_ops_uv = {
- /* name */ "libuv",
- /* init_context */ elops_init_context_uv,
- /* destroy_context1 */ elops_destroy_context1_uv,
- /* destroy_context2 */ elops_destroy_context2_uv,
- /* init_vhost_listen_wsi */ elops_init_vhost_listen_wsi_uv,
- /* init_pt */ elops_init_pt_uv,
- /* wsi_logical_close */ elops_wsi_logical_close_uv,
- /* check_client_connect_ok */ elops_check_client_connect_ok_uv,
- /* close_handle_manually */ elops_close_handle_manually_uv,
- /* accept */ elops_accept_uv,
- /* io */ elops_io_uv,
- /* run_pt */ elops_run_pt_uv,
- /* destroy_pt */ elops_destroy_pt_uv,
- /* destroy wsi */ NULL,
- /* flags */ 0,
- /* evlib_size_ctx */ sizeof(struct lws_context_eventlibs_libuv),
- /* evlib_size_pt */ sizeof(struct lws_pt_eventlibs_libuv),
- /* evlib_size_vh */ 0,
- /* evlib_size_wsi */ sizeof(struct lws_io_watcher_libuv),
- };
- #if defined(LWS_WITH_EVLIB_PLUGINS)
- LWS_VISIBLE
- #endif
- const lws_plugin_evlib_t evlib_uv = {
- .hdr = {
- "libuv event loop",
- "lws_evlib_plugin",
- LWS_PLUGIN_API_MAGIC
- },
- .ops = &event_loop_ops_uv
- };
|