--- libev/ev_poll.c 2007/11/02 15:56:19 1.1 +++ libev/ev_poll.c 2007/12/09 02:12:44 1.18 @@ -1,5 +1,5 @@ /* - * libev epoll fd activity backend + * libev poll fd activity backend * * Copyright (c) 2007 Marc Alexander Lehmann * All rights reserved. @@ -29,82 +29,99 @@ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ -#include +#include -static int epoll_fd = -1; +void inline_size +pollidx_init (int *base, int count) +{ + while (count--) + *base++ = -1; +} static void -epoll_modify (int fd, int oev, int nev) +poll_modify (EV_P_ int fd, int oev, int nev) { - int mode = nev ? oev ? EPOLL_CTL_MOD : EPOLL_CTL_ADD : EPOLL_CTL_DEL; + int idx; - struct epoll_event ev; - ev.data.fd = fd; - ev.events = - (nev & EV_READ ? EPOLLIN : 0) - | (nev & EV_WRITE ? EPOLLOUT : 0); + if (oev == nev) + return; - epoll_ctl (epoll_fd, mode, fd, &ev); -} + array_needsize (int, pollidxs, pollidxmax, fd + 1, pollidx_init); -static void -epoll_postfork_child (void) -{ - int fd; + idx = pollidxs [fd]; - epoll_fd = epoll_create (256); - fcntl (epoll_fd, F_SETFD, FD_CLOEXEC); + if (idx < 0) /* need to allocate a new pollfd */ + { + pollidxs [fd] = idx = pollcnt++; + array_needsize (struct pollfd, polls, pollmax, pollcnt, EMPTY2); + polls [idx].fd = fd; + } - /* re-register interest in fds */ - for (fd = 0; fd < anfdmax; ++fd) - if (anfds [fd].events)//D - epoll_modify (fd, EV_NONE, anfds [fd].events); -} + assert (polls [idx].fd == fd); -static struct epoll_event *events; -static int eventmax; + if (nev) + polls [idx].events = + (nev & EV_READ ? POLLIN : 0) + | (nev & EV_WRITE ? POLLOUT : 0); + else /* remove pollfd */ + { + pollidxs [fd] = -1; + + if (expect_true (idx < --pollcnt)) + { + polls [idx] = polls [pollcnt]; + pollidxs [polls [idx].fd] = idx; + } + } +} static void -epoll_poll (ev_tstamp timeout) +poll_poll (EV_P_ ev_tstamp timeout) { - int eventcnt = epoll_wait (epoll_fd, events, eventmax, ceil (timeout * 1000.)); int i; + int res = poll (polls, pollcnt, (int)ceil (timeout * 1000.)); - if (eventcnt < 0) - return; - - for (i = 0; i < eventcnt; ++i) - fd_event ( - events [i].data.fd, - (events [i].events & (EPOLLOUT | EPOLLERR | EPOLLHUP) ? EV_WRITE : 0) - | (events [i].events & (EPOLLIN | EPOLLERR | EPOLLHUP) ? EV_READ : 0) - ); - - /* if the receive array was full, increase its size */ - if (expect_false (eventcnt == eventmax)) + if (expect_false (res < 0)) { - free (events); - eventmax = array_roundsize (events, eventmax << 1); - events = malloc (sizeof (struct epoll_event) * eventmax); + if (errno == EBADF) + fd_ebadf (EV_A); + else if (errno == ENOMEM && !syserr_cb) + fd_enomem (EV_A); + else if (errno != EINTR) + syserr ("(libev) poll"); + + return; } + + for (i = 0; i < pollcnt; ++i) + if (expect_false (polls [i].revents & POLLNVAL)) + fd_kill (EV_A_ polls [i].fd); + else + fd_event ( + EV_A_ + polls [i].fd, + (polls [i].revents & (POLLOUT | POLLERR | POLLHUP) ? EV_WRITE : 0) + | (polls [i].revents & (POLLIN | POLLERR | POLLHUP) ? EV_READ : 0) + ); } -static void -epoll_init (int flags) +int inline_size +poll_init (EV_P_ int flags) { - epoll_fd = epoll_create (256); - - if (epoll_fd < 0) - return; + backend_fudge = 1e-3; /* needed to compensate for select returning early, very conservative */ + backend_modify = poll_modify; + backend_poll = poll_poll; - fcntl (epoll_fd, F_SETFD, FD_CLOEXEC); + pollidxs = 0; pollidxmax = 0; + polls = 0; pollmax = 0; pollcnt = 0; - ev_method = EVMETHOD_EPOLL; - method_fudge = 1e-3; /* needed to compensate for epoll returning early */ - method_modify = epoll_modify; - method_poll = epoll_poll; + return EVBACKEND_POLL; +} - eventmax = 64; /* intiial number of events receivable per poll */ - events = malloc (sizeof (struct epoll_event) * eventmax); +void inline_size +poll_destroy (EV_P) +{ + ev_free (pollidxs); + ev_free (polls); }