lib/libev.c

   1 /*
   2  * libwebsockets - small server side websockets and web server implementation
   3  *
   4  * Copyright (C) 2010-2014 Andy Green <andy@warmcat.com>
   5  *
   6  *  This library is free software; you can redistribute it and/or
   7  *  modify it under the terms of the GNU Lesser General Public
   8  *  License as published by the Free Software Foundation:
   9  *  version 2.1 of the License.
  10  *
  11  *  This library is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  *  Lesser General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU Lesser General Public
  17  *  License along with this library; if not, write to the Free Software
  18  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  19  *  MA  02110-1301  USA
  20  */
  21
  22 #include "private-libwebsockets.h"
  23
  24 void lws_feature_status_libev(struct lws_context_creation_info *info)
  25 {
  26         if (info->options & LWS_SERVER_OPTION_LIBEV)
  27                 lwsl_notice("libev support compiled in and enabled\n");
  28         else
  29                 lwsl_notice("libev support compiled in but disabled\n");
  30 }
  31
  32 static void
  33 lws_accept_cb(struct ev_loop *loop, struct ev_io *watcher, int revents)
  34 {
  35         struct lws_io_watcher *lws_io = container_of(watcher,
  36                                         struct lws_io_watcher, ev_watcher);
  37         struct lws_context *context = lws_io->context;
  38         struct lws_pollfd eventfd;
  39
  40         if (revents & EV_ERROR)
  41                 return;
  42
  43         eventfd.fd = watcher->fd;
  44         eventfd.events = 0;
  45         eventfd.revents = EV_NONE;
  46         if (revents & EV_READ) {
  47                 eventfd.events |= LWS_POLLIN;
  48                 eventfd.revents |= LWS_POLLIN;
  49         }
  50         if (revents & EV_WRITE) {
  51                 eventfd.events |= LWS_POLLOUT;
  52                 eventfd.revents |= LWS_POLLOUT;
  53         }
  54         lws_service_fd(context, &eventfd);
  55 }
  56
  57 LWS_VISIBLE void
  58 lws_ev_sigint_cb(struct ev_loop *loop, struct ev_signal *watcher, int revents)
  59 {
  60         ev_break(loop, EVBREAK_ALL);
  61 }
  62
  63 LWS_VISIBLE int
  64 lws_ev_sigint_cfg(struct lws_context *context, int use_ev_sigint,
  65                   lws_ev_signal_cb_t *cb)
  66 {
  67         context->use_ev_sigint = use_ev_sigint;
  68         if (cb)
  69                 context->lws_ev_sigint_cb = cb;
  70         else
  71                 context->lws_ev_sigint_cb = &lws_ev_sigint_cb;
  72
  73         return 0;
  74 }
  75
  76 LWS_VISIBLE int
  77 lws_ev_initloop(struct lws_context *context, struct ev_loop *loop, int tsi)
  78 {
  79         struct ev_signal *w_sigint = &context->pt[tsi].w_sigint.ev_watcher;
  80         struct ev_io *w_accept = &context->pt[tsi].w_accept.ev_watcher;
  81         const char * backend_name;
  82         int status = 0;
  83         int backend;
  84
  85         if (!loop)
  86                 loop = ev_loop_new(0);
  87
  88         context->pt[tsi].io_loop_ev = loop;
  89
  90         /*
  91          * Initialize the accept w_accept with the listening socket
  92          * and register a callback for read operations
  93          */
  94         ev_io_init(w_accept, lws_accept_cb, context->pt[tsi].lserv_fd, EV_READ);
  95         ev_io_start(context->pt[tsi].io_loop_ev, w_accept);
  96
  97         /* Register the signal watcher unless the user says not to */
  98         if (context->use_ev_sigint) {
  99                 ev_signal_init(w_sigint, context->lws_ev_sigint_cb, SIGINT);
 100                 ev_signal_start(context->pt[tsi].io_loop_ev, w_sigint);
 101         }
 102         backend = ev_backend(loop);
 103
 104         switch (backend) {
 105         case EVBACKEND_SELECT:
 106                 backend_name = "select";
 107                 break;
 108         case EVBACKEND_POLL:
 109                 backend_name = "poll";
 110                 break;
 111         case EVBACKEND_EPOLL:
 112                 backend_name = "epoll";
 113                 break;
 114         case EVBACKEND_KQUEUE:
 115                 backend_name = "kqueue";
 116                 break;
 117         case EVBACKEND_DEVPOLL:
 118                 backend_name = "/dev/poll";
 119                 break;
 120         case EVBACKEND_PORT:
 121                 backend_name = "Solaris 10 \"port\"";
 122                 break;
 123         default:
 124                 backend_name = "Unknown libev backend";
 125                 break;
 126         }
 127
 128         lwsl_notice(" libev backend: %s\n", backend_name);
 129
 130         return status;
 131 }
 132
 133 void
 134 lws_libev_destroyloop(struct lws_context *context, int tsi)
 135 {
 136         struct lws_context_per_thread *pt = &context->pt[tsi];
 137
 138         if (!(context->options & LWS_SERVER_OPTION_LIBEV))
 139                 return;
 140
 141         ev_io_stop(pt->io_loop_ev, &pt->w_accept.ev_watcher);
 142         if (context->use_ev_sigint)
 143                 ev_signal_stop(pt->io_loop_ev,
 144                        &pt->w_sigint.ev_watcher);
 145         if (!pt->ev_loop_foreign)
 146                 ev_loop_destroy(pt->io_loop_ev);
 147 }
 148
 149 LWS_VISIBLE void
 150 lws_libev_accept(struct lws *new_wsi, int accept_fd)
 151 {
 152         struct lws_context *context = lws_get_context(new_wsi);
 153         struct ev_io *r = &new_wsi->w_read.ev_watcher;
 154         struct ev_io *w = &new_wsi->w_write.ev_watcher;
 155
 156         if (!LWS_LIBEV_ENABLED(context))
 157                 return;
 158
 159         new_wsi->w_read.context = context;
 160         new_wsi->w_write.context = context;
 161         ev_io_init(r, lws_accept_cb, accept_fd, EV_READ);
 162         ev_io_init(w, lws_accept_cb, accept_fd, EV_WRITE);
 163 }
 164
 165 LWS_VISIBLE void
 166 lws_libev_io(struct lws *wsi, int flags)
 167 {
 168         struct lws_context *context = lws_get_context(wsi);
 169         struct lws_context_per_thread *pt = &wsi->context->pt[(int)wsi->tsi];
 170
 171         if (!LWS_LIBEV_ENABLED(context))
 172                 return;
 173
 174         if (!pt->io_loop_ev || context->being_destroyed)
 175                 return;
 176
 177         assert((flags & (LWS_EV_START | LWS_EV_STOP)) &&
 178                (flags & (LWS_EV_READ | LWS_EV_WRITE)));
 179
 180         if (flags & LWS_EV_START) {
 181                 if (flags & LWS_EV_WRITE)
 182                         ev_io_start(pt->io_loop_ev, &wsi->w_write.ev_watcher);
 183                 if (flags & LWS_EV_READ)
 184                         ev_io_start(pt->io_loop_ev, &wsi->w_read.ev_watcher);
 185         } else {
 186                 if (flags & LWS_EV_WRITE)
 187                         ev_io_stop(pt->io_loop_ev, &wsi->w_write.ev_watcher);
 188                 if (flags & LWS_EV_READ)
 189                         ev_io_stop(pt->io_loop_ev, &wsi->w_read.ev_watcher);
 190         }
 191 }
 192
 193 LWS_VISIBLE int
 194 lws_libev_init_fd_table(struct lws_context *context)
 195 {
 196         int n;
 197
 198         if (!LWS_LIBEV_ENABLED(context))
 199                 return 0;
 200
 201         for (n = 0; n < context->count_threads; n++) {
 202                 context->pt[n].w_accept.context = context;
 203                 context->pt[n].w_sigint.context = context;
 204         }
 205
 206         return 1;
 207 }
 208
 209 LWS_VISIBLE void
 210 lws_libev_run(const struct lws_context *context, int tsi)
 211 {
 212         if (context->pt[tsi].io_loop_ev && LWS_LIBEV_ENABLED(context))
 213                 ev_run(context->pt[tsi].io_loop_ev, 0);
 214 }