src/ioset-kevent.c

   1 /* ioset kqueue()/kevent() backend for srvx
   2  * Copyright 2008 srvx Development Team
   3  *
   4  * This file is part of srvx.
   5  *
   6  * srvx is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program 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
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with srvx; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
  19  */
  20
  21 #include "ioset-impl.h"
  22 #include "common.h"
  23 #include "log.h"
  24
  25 #ifdef HAVE_SYS_EVENT_H
  26 # include <sys/event.h>
  27 #endif
  28
  29 #define MAX_EVENTS 16
  30
  31 extern int clock_skew;
  32 static int kq_fd;
  33
  34 static int
  35 ioset_kevent_init(void)
  36 {
  37     kq_fd = kqueue();
  38     return kq_fd >= 0;
  39 }
  40
  41 static void
  42 ioset_kevent_add(struct io_fd *fd)
  43 {
  44     struct kevent changes[2];
  45     int nchanges = 0;
  46     int res;
  47
  48     EV_SET(&changes[nchanges++], fd->fd, EVFILT_READ, EV_ADD, 0, 0, fd);
  49     if (fd_wants_writes(fd)) {
  50         EV_SET(&changes[nchanges++], fd->fd, EVFILT_WRITE, EV_ADD, 0, 0, fd);
  51     }
  52     res = kevent(kq_fd, changes, nchanges, NULL, 0, NULL);
  53     if (res < 0) {
  54         log_module(MAIN_LOG, LOG_ERROR, "kevent() add failed: %s", strerror(errno));
  55     }
  56 }
  57
  58 static void
  59 ioset_kevent_remove(struct io_fd *fd, int closed)
  60 {
  61     if (!closed) {
  62         struct kevent changes[2];
  63         int nchanges = 0;
  64         int res;
  65
  66         EV_SET(&changes[nchanges++], fd->fd, EVFILT_READ, EV_DELETE, 0, 0, fd);
  67         EV_SET(&changes[nchanges++], fd->fd, EVFILT_WRITE, EV_DELETE, 0, 0, fd);
  68         res = kevent(kq_fd, changes, nchanges, NULL, 0, NULL);
  69         if (res < 0) {
  70             log_module(MAIN_LOG, LOG_ERROR, "kevent() remove failed: %s", strerror(errno));
  71         }
  72     }
  73 }
  74
  75 static void
  76 ioset_kevent_update(struct io_fd *fd)
  77 {
  78     ioset_kevent_add(fd);
  79 }
  80
  81 static void
  82 ioset_kevent_cleanup(void)
  83 {
  84     close(kq_fd);
  85 }
  86
  87 static int
  88 ioset_kevent_loop(struct timeval *timeout)
  89 {
  90     struct kevent events[MAX_EVENTS];
  91     struct timespec ts;
  92     struct timespec *pts;
  93     int is_write;
  94     int is_read;
  95     int res;
  96     int ii;
  97
  98     /* Try to get events from the kernel. */
  99     if (timeout) {
 100         ts.tv_sec = timeout->tv_sec;
 101         ts.tv_nsec = timeout->tv_usec * 1000;
 102         pts = &ts;
 103     } else {
 104         pts = NULL;
 105     }
 106     res = kevent(kq_fd, NULL, 0, events, MAX_EVENTS, pts);
 107     if (res < 0) {
 108         log_module(MAIN_LOG, LOG_ERROR, "kevent() poll failed: %s", strerror(errno));
 109         return 1;
 110     }
 111
 112     /* Process the events we got. */
 113     for (ii = 0; ii < res; ++ii) {
 114         is_write = events[ii].filter == EVFILT_WRITE;
 115         is_read = events[ii].filter == EVFILT_READ;
 116         ioset_events(events[ii].udata, is_read, is_write);
 117     }
 118
 119     return 0;
 120 }
 121
 122 struct io_engine io_engine_kevent = {
 123     .name = "kevent",
 124     .init = ioset_kevent_init,
 125     .add = ioset_kevent_add,
 126     .remove = ioset_kevent_remove,
 127     .update = ioset_kevent_update,
 128     .loop = ioset_kevent_loop,
 129     .cleanup = ioset_kevent_cleanup,
 130 };