]> git.saurik.com Git - redis.git/blob - src/ae_epoll.c
Merge remote-tracking branch 'origin/unstable' into aof-fixes
[redis.git] / src / ae_epoll.c
1 /* Linux epoll(2) based ae.c module
2 * Copyright (C) 2009-2010 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;
10 } aeApiState;
11
12 static int aeApiCreate(aeEventLoop *eventLoop) {
13 aeApiState *state = zmalloc(sizeof(aeApiState));
14
15 if (!state) return -1;
16 state->events = zmalloc(sizeof(struct epoll_event)*eventLoop->setsize);
17 if (!state->events) {
18 zfree(state);
19 return -1;
20 }
21 state->epfd = epoll_create(1024); /* 1024 is just an hint for the kernel */
22 if (state->epfd == -1) {
23 zfree(state->events);
24 zfree(state);
25 return -1;
26 }
27 eventLoop->apidata = state;
28 return 0;
29 }
30
31 static void aeApiFree(aeEventLoop *eventLoop) {
32 aeApiState *state = eventLoop->apidata;
33
34 close(state->epfd);
35 zfree(state->events);
36 zfree(state);
37 }
38
39 static int aeApiAddEvent(aeEventLoop *eventLoop, int fd, int mask) {
40 aeApiState *state = eventLoop->apidata;
41 struct epoll_event ee;
42 /* If the fd was already monitored for some event, we need a MOD
43 * operation. Otherwise we need an ADD operation. */
44 int op = eventLoop->events[fd].mask == AE_NONE ?
45 EPOLL_CTL_ADD : EPOLL_CTL_MOD;
46
47 ee.events = 0;
48 mask |= eventLoop->events[fd].mask; /* Merge old events */
49 if (mask & AE_READABLE) ee.events |= EPOLLIN;
50 if (mask & AE_WRITABLE) ee.events |= EPOLLOUT;
51 ee.data.u64 = 0; /* avoid valgrind warning */
52 ee.data.fd = fd;
53 if (epoll_ctl(state->epfd,op,fd,&ee) == -1) return -1;
54 return 0;
55 }
56
57 static void aeApiDelEvent(aeEventLoop *eventLoop, int fd, int delmask) {
58 aeApiState *state = eventLoop->apidata;
59 struct epoll_event ee;
60 int mask = eventLoop->events[fd].mask & (~delmask);
61
62 ee.events = 0;
63 if (mask & AE_READABLE) ee.events |= EPOLLIN;
64 if (mask & AE_WRITABLE) ee.events |= EPOLLOUT;
65 ee.data.u64 = 0; /* avoid valgrind warning */
66 ee.data.fd = fd;
67 if (mask != AE_NONE) {
68 epoll_ctl(state->epfd,EPOLL_CTL_MOD,fd,&ee);
69 } else {
70 /* Note, Kernel < 2.6.9 requires a non null event pointer even for
71 * EPOLL_CTL_DEL. */
72 epoll_ctl(state->epfd,EPOLL_CTL_DEL,fd,&ee);
73 }
74 }
75
76 static int aeApiPoll(aeEventLoop *eventLoop, struct timeval *tvp) {
77 aeApiState *state = eventLoop->apidata;
78 int retval, numevents = 0;
79
80 retval = epoll_wait(state->epfd,state->events,eventLoop->setsize,
81 tvp ? (tvp->tv_sec*1000 + tvp->tv_usec/1000) : -1);
82 if (retval > 0) {
83 int j;
84
85 numevents = retval;
86 for (j = 0; j < numevents; j++) {
87 int mask = 0;
88 struct epoll_event *e = state->events+j;
89
90 if (e->events & EPOLLIN) mask |= AE_READABLE;
91 if (e->events & EPOLLOUT) mask |= AE_WRITABLE;
92 eventLoop->fired[j].fd = e->data.fd;
93 eventLoop->fired[j].mask = mask;
94 }
95 }
96 return numevents;
97 }
98
99 static char *aeApiName(void) {
100 return "epoll";
101 }