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         else
  88                 context->pt[tsi].ev_loop_foreign = 1;
  89
  90         context->pt[tsi].io_loop_ev = loop;
  91
  92         /*
  93          * Initialize the accept w_accept with the listening socket
  94          * and register a callback for read operations
  95          */
  96         ev_io_init(w_accept, lws_accept_cb, context->pt[tsi].lserv_fd, EV_READ);
  97         ev_io_start(context->pt[tsi].io_loop_ev, w_accept);
  98
  99         /* Register the signal watcher unless the user says not to */
 100         if (context->use_ev_sigint) {
 101                 ev_signal_init(w_sigint, context->lws_ev_sigint_cb, SIGINT);
 102                 ev_signal_start(context->pt[tsi].io_loop_ev, w_sigint);
 103         }
 104         backend = ev_backend(loop);
 105
 106         switch (backend) {
 107         case EVBACKEND_SELECT:
 108                 backend_name = "select";
 109                 break;
 110         case EVBACKEND_POLL:
 111                 backend_name = "poll";
 112                 break;
 113         case EVBACKEND_EPOLL:
 114                 backend_name = "epoll";
 115                 break;
 116         case EVBACKEND_KQUEUE:
 117                 backend_name = "kqueue";
 118                 break;
 119         case EVBACKEND_DEVPOLL:
 120                 backend_name = "/dev/poll";
 121                 break;
 122         case EVBACKEND_PORT:
 123                 backend_name = "Solaris 10 \"port\"";
 124                 break;
 125         default:
 126                 backend_name = "Unknown libev backend";
 127                 break;
 128         }
 129
 130         lwsl_notice(" libev backend: %s\n", backend_name);
 131
 132         return status;
 133 }
 134
 135 void
 136 lws_libev_destroyloop(struct lws_context *context, int tsi)
 137 {
 138         struct lws_context_per_thread *pt = &context->pt[tsi];
 139
 140         if (!(context->options & LWS_SERVER_OPTION_LIBEV))
 141                 return;
 142
 143         if (!pt->io_loop_ev)
 144                 return;
 145
 146         ev_io_stop(pt->io_loop_ev, &pt->w_accept.ev_watcher);
 147         if (context->use_ev_sigint)
 148                 ev_signal_stop(pt->io_loop_ev,
 149                        &pt->w_sigint.ev_watcher);
 150         if (!pt->ev_loop_foreign)
 151                 ev_loop_destroy(pt->io_loop_ev);
 152 }
 153
 154 LWS_VISIBLE void
 155 lws_libev_accept(struct lws *new_wsi, int accept_fd)
 156 {
 157         struct lws_context *context = lws_get_context(new_wsi);
 158         struct ev_io *r = &new_wsi->w_read.ev_watcher;
 159         struct ev_io *w = &new_wsi->w_write.ev_watcher;
 160
 161         if (!LWS_LIBEV_ENABLED(context))
 162                 return;
 163
 164         new_wsi->w_read.context = context;
 165         new_wsi->w_write.context = context;
 166         ev_io_init(r, lws_accept_cb, accept_fd, EV_READ);
 167         ev_io_init(w, lws_accept_cb, accept_fd, EV_WRITE);
 168 }
 169
 170 LWS_VISIBLE void
 171 lws_libev_io(struct lws *wsi, int flags)
 172 {
 173         struct lws_context *context = lws_get_context(wsi);
 174         struct lws_context_per_thread *pt = &wsi->context->pt[(int)wsi->tsi];
 175
 176         if (!LWS_LIBEV_ENABLED(context))
 177                 return;
 178
 179         if (!pt->io_loop_ev || context->being_destroyed)
 180                 return;
 181
 182         assert((flags & (LWS_EV_START | LWS_EV_STOP)) &&
 183                (flags & (LWS_EV_READ | LWS_EV_WRITE)));
 184
 185         if (flags & LWS_EV_START) {
 186                 if (flags & LWS_EV_WRITE)
 187                         ev_io_start(pt->io_loop_ev, &wsi->w_write.ev_watcher);
 188                 if (flags & LWS_EV_READ)
 189                         ev_io_start(pt->io_loop_ev, &wsi->w_read.ev_watcher);
 190         } else {
 191                 if (flags & LWS_EV_WRITE)
 192                         ev_io_stop(pt->io_loop_ev, &wsi->w_write.ev_watcher);
 193                 if (flags & LWS_EV_READ)
 194                         ev_io_stop(pt->io_loop_ev, &wsi->w_read.ev_watcher);
 195         }
 196 }
 197
 198 LWS_VISIBLE int
 199 lws_libev_init_fd_table(struct lws_context *context)
 200 {
 201         int n;
 202
 203         if (!LWS_LIBEV_ENABLED(context))
 204                 return 0;
 205
 206         for (n = 0; n < context->count_threads; n++) {
 207                 context->pt[n].w_accept.context = context;
 208                 context->pt[n].w_sigint.context = context;
 209         }
 210
 211         return 1;
 212 }
 213
 214 LWS_VISIBLE void
 215 lws_libev_run(const struct lws_context *context, int tsi)
 216 {
 217         if (context->pt[tsi].io_loop_ev && LWS_LIBEV_ENABLED(context))
 218                 ev_run(context->pt[tsi].io_loop_ev, 0);
 219 }