ev_epoll.c

   1 #include <sys/epoll.h>
   2
   3 static int epoll_fd = -1;
   4
   5 static void
   6 epoll_modify (int fd, int oev, int nev)
   7 {
   8   int mode = nev ? oev ? EPOLL_CTL_MOD : EPOLL_CTL_ADD : EPOLL_CTL_DEL;
   9
  10   struct epoll_event ev;
  11   ev.data.fd = fd;
  12   ev.events =
  13       (nev & EV_READ ? EPOLLIN : 0)
  14       | (nev & EV_WRITE ? EPOLLOUT : 0);
  15
  16   epoll_ctl (epoll_fd, mode, fd, &ev);
  17 }
  18
  19 void epoll_postfork_child (void)
  20 {
  21   int fd;
  22
  23   epoll_fd = epoll_create (256);
  24   fcntl (epoll_fd, F_SETFD, FD_CLOEXEC);
  25
  26   /* re-register interest in fds */
  27   for (fd = 0; fd < anfdmax; ++fd)
  28     if (anfds [fd].wev)
  29       epoll_modify (fd, EV_NONE, anfds [fd].wev);
  30 }
  31
  32 static struct epoll_event *events;
  33 static int eventmax;
  34
  35 static void epoll_poll (ev_tstamp timeout)
  36 {
  37   int eventcnt = epoll_wait (epoll_fd, events, eventmax, ceil (timeout * 1000.));
  38   int i;
  39
  40   if (eventcnt < 0)
  41     return;
  42
  43   for (i = 0; i < eventcnt; ++i)
  44     fd_event (
  45       events [i].data.fd,
  46       (events [i].events & (EPOLLOUT | EPOLLERR | EPOLLHUP) ? EV_WRITE : 0)
  47       | (events [i].events & (EPOLLIN | EPOLLERR | EPOLLHUP) ? EV_READ : 0)
  48     );
  49
  50   /* if the receive array was full, increase its size */
  51   if (eventcnt == eventmax)
  52     {
  53       free (events);
  54       eventmax += eventmax >> 1;
  55       events = malloc (sizeof (struct epoll_event) * eventmax);
  56     }
  57 }
  58
  59 void epoll_init (int flags)
  60 {
  61   epoll_fd = epoll_create (256);
  62
  63   if (epoll_fd < 0)
  64     return;
  65
  66   fcntl (epoll_fd, F_SETFD, FD_CLOEXEC);
  67
  68   ev_method     = EVMETHOD_EPOLL;
  69   method_fudge  = 1e-3; /* needed to compensate for epoll returning early */
  70   method_modify = epoll_modify;
  71   method_poll   = epoll_poll;
  72
  73   eventmax = 64; /* intiial number of events receivable per poll */
  74   events = malloc (sizeof (struct epoll_event) * eventmax);
  75 }