src/fdevent_linux_sysepoll.c

   1 #include "first.h"
   2
   3 #include "fdevent_impl.h"
   4 #include "fdevent.h"
   5 #include "buffer.h"
   6
   7 #include <unistd.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10
  11 #ifdef FDEVENT_USE_LINUX_EPOLL
  12
  13 # include <sys/epoll.h>
  14
  15 __attribute_cold__
  16 static void fdevent_linux_sysepoll_free(fdevents *ev) {
  17         close(ev->epoll_fd);
  18         free(ev->epoll_events);
  19 }
  20
  21 static int fdevent_linux_sysepoll_event_del(fdevents *ev, fdnode *fdn) {
  22     return epoll_ctl(ev->epoll_fd, EPOLL_CTL_DEL, fdn->fd, NULL);
  23 }
  24
  25 static int fdevent_linux_sysepoll_event_set(fdevents *ev, fdnode *fdn, int events) {
  26     int op = (-1 == fdn->fde_ndx) ? EPOLL_CTL_ADD : EPOLL_CTL_MOD;
  27     int fd = fdn->fde_ndx = fdn->fd;
  28     struct epoll_event ep;
  29   #ifndef EPOLLRDHUP
  30     events &= ~FDEVENT_RDHUP;
  31   #endif
  32     ep.events = events | EPOLLERR | EPOLLHUP;
  33     ep.data.ptr = fdn;
  34     return epoll_ctl(ev->epoll_fd, op, fd, &ep);
  35 }
  36
  37 static int fdevent_linux_sysepoll_poll(fdevents * const ev, int timeout_ms) {
  38     int n = epoll_wait(ev->epoll_fd, ev->epoll_events, ev->maxfds, timeout_ms);
  39     server * const srv = ev->srv;
  40     for (int i = 0; i < n; ++i) {
  41         fdnode * const fdn = (fdnode *)ev->epoll_events[i].data.ptr;
  42         int revents = ev->epoll_events[i].events;
  43         if ((fdevent_handler)NULL != fdn->handler) {
  44             (*fdn->handler)(srv, fdn->ctx, revents);
  45         }
  46     }
  47     return n;
  48 }
  49
  50 __attribute_cold__
  51 int fdevent_linux_sysepoll_init(fdevents *ev) {
  52         force_assert(EPOLLIN    == FDEVENT_IN);
  53         force_assert(EPOLLPRI   == FDEVENT_PRI);
  54         force_assert(EPOLLOUT   == FDEVENT_OUT);
  55         force_assert(EPOLLERR   == FDEVENT_ERR);
  56         force_assert(EPOLLHUP   == FDEVENT_HUP);
  57       #ifdef EPOLLRDHUP
  58         force_assert(EPOLLRDHUP == FDEVENT_RDHUP);
  59       #endif
  60
  61         ev->type      = FDEVENT_HANDLER_LINUX_SYSEPOLL;
  62         ev->event_set = fdevent_linux_sysepoll_event_set;
  63         ev->event_del = fdevent_linux_sysepoll_event_del;
  64         ev->poll      = fdevent_linux_sysepoll_poll;
  65         ev->free      = fdevent_linux_sysepoll_free;
  66
  67         if (-1 == (ev->epoll_fd = epoll_create(ev->maxfds))) return -1;
  68
  69         fdevent_setfd_cloexec(ev->epoll_fd);
  70
  71         ev->epoll_events = malloc(ev->maxfds * sizeof(*ev->epoll_events));
  72         force_assert(NULL != ev->epoll_events);
  73
  74         return 0;
  75 }
  76
  77 #endif