| 1 | /* Linux epoll(2) based ae.c module |
| 2 | * Copyright (C) 2009 Salvatore Sanfilippo - antirez@gmail.com |
| 3 | * Released under the BSD license. See the COPYING file for more info. */ |
| 4 | |
| 5 | #include <sys/epoll.h> |
| 6 | |
| 7 | typedef struct aeApiState { |
| 8 | int epfd; |
| 9 | struct epoll_event events[AE_SETSIZE]; |
| 10 | } aeApiState; |
| 11 | |
| 12 | static int aeApiCreate(aeEventLoop *eventLoop) { |
| 13 | aeApiState *state = zmalloc(sizeof(aeApiState)); |
| 14 | |
| 15 | if (!state) return -1; |
| 16 | state->epfd = epoll_create(1024); /* 1024 is just an hint for the kernel */ |
| 17 | if (state->epfd == -1) return -1; |
| 18 | eventLoop->apidata = state; |
| 19 | return 0; |
| 20 | } |
| 21 | |
| 22 | static void aeApiFree(aeEventLoop *eventLoop) { |
| 23 | aeApiState *state = eventLoop->apidata; |
| 24 | |
| 25 | close(state->epfd); |
| 26 | zfree(state); |
| 27 | } |
| 28 | |
| 29 | static int aeApiAddEvent(aeEventLoop *eventLoop, int fd, int mask) { |
| 30 | aeApiState *state = eventLoop->apidata; |
| 31 | struct epoll_event ee; |
| 32 | int op = eventLoop->events[fd].mask == AE_NONE ? |
| 33 | EPOLL_CTL_ADD : EPOLL_CTL_MOD; |
| 34 | |
| 35 | ee.events = 0; |
| 36 | if (mask & AE_READABLE) ee.events |= EPOLLIN; |
| 37 | if (mask & AE_WRITABLE) ee.events |= EPOLLOUT; |
| 38 | if (mask & AE_EXCEPTION) ee.events |= EPOLLPRI; |
| 39 | ee.data.u64 = 0; /* avoid valgrind warning */ |
| 40 | ee.data.fd = fd; |
| 41 | if (epoll_ctl(state->epfd,op,fd,&ee) == -1) return -1; |
| 42 | return 0; |
| 43 | } |
| 44 | |
| 45 | static void aeApiDelEvent(aeEventLoop *eventLoop, int fd, int delmask) { |
| 46 | aeApiState *state = eventLoop->apidata; |
| 47 | struct epoll_event ee; |
| 48 | int mask = eventLoop->events[fd].mask & (~delmask); |
| 49 | |
| 50 | ee.events = 0; |
| 51 | if (mask & AE_READABLE) ee.events |= EPOLLIN; |
| 52 | if (mask & AE_WRITABLE) ee.events |= EPOLLOUT; |
| 53 | if (mask & AE_EXCEPTION) ee.events |= EPOLLPRI; |
| 54 | ee.data.u64 = 0; /* avoid valgrind warning */ |
| 55 | ee.data.fd = fd; |
| 56 | if (mask != AE_NONE) { |
| 57 | epoll_ctl(state->epfd,EPOLL_CTL_MOD,fd,&ee); |
| 58 | } else { |
| 59 | /* Note, Kernel < 2.6.9 requires a non null event pointer even for |
| 60 | * EPOLL_CTL_DEL. */ |
| 61 | epoll_ctl(state->epfd,EPOLL_CTL_DEL,fd,&ee); |
| 62 | } |
| 63 | } |
| 64 | |
| 65 | static int aeApiPoll(aeEventLoop *eventLoop, struct timeval *tvp) { |
| 66 | aeApiState *state = eventLoop->apidata; |
| 67 | int retval, numevents = 0; |
| 68 | |
| 69 | retval = epoll_wait(state->epfd,state->events,AE_SETSIZE, |
| 70 | tvp ? (tvp->tv_sec*1000 + tvp->tv_usec/1000) : -1); |
| 71 | if (retval > 0) { |
| 72 | int j; |
| 73 | |
| 74 | numevents = retval; |
| 75 | for (j = 0; j < numevents; j++) { |
| 76 | int mask = 0; |
| 77 | struct epoll_event *e = state->events+j; |
| 78 | |
| 79 | if (e->events & EPOLLIN) mask |= AE_READABLE; |
| 80 | if (e->events & EPOLLOUT) mask |= AE_WRITABLE; |
| 81 | if (e->events & EPOLLPRI) mask |= AE_EXCEPTION; |
| 82 | eventLoop->fired[j].fd = e->data.fd; |
| 83 | eventLoop->fired[j].mask = mask; |
| 84 | } |
| 85 | } |
| 86 | return numevents; |
| 87 | } |