-
Notifications
You must be signed in to change notification settings - Fork 0
/
netpoll_linux.c
57 lines (47 loc) · 1.17 KB
/
netpoll_linux.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#include <errno.h>
#include <sys/epoll.h>
#include "runtime.h"
static int epfd = -1;
void netpollinit(void) {
epfd = epoll_create(1);
if (epfd < 0)
throw("netpollinit: epoll_create failed with errno %d", errno);
setcloexec(epfd);
}
int netpollopen(Event *ep) {
struct epoll_event ev;
ev.events = EPOLLIN | EPOLLOUT | EPOLLRDHUP | EPOLLET;
ev.data.ptr = (void *)ep;
return epoll_ctl(epfd, EPOLL_CTL_ADD, ep->fd, &ev);
}
G *netpoll(long ns) {
struct epoll_event ev[64], *ep;
G *gp;
int n, i, mode, ms = -1;
if (epfd < 0)
return nil;
if (ns > 0)
ms = ns / 1000000L;
retry:
gp = nil;
n = epoll_wait(epfd, ev, len(ev), ms);
if (n < 0) {
if (n == -EINTR)
goto retry;
throw("netpoll: epoll_wait failed with %d", n);
}
#ifdef DEBUG
debug("netpoll: n=%d ts.tv_sec=%ld ts.tv_nsec=%d", n, ts.tv_sec, ts.tv_nsec);
#endif
for (i = 0; i < n; i++) {
ep = ev + i;
mode = 0;
if (ep->events & (EPOLLIN | EPOLLRDHUP | EPOLLHUP | EPOLLERR))
mode |= EV_READ;
if (ep->events & (EPOLLOUT | EPOLLHUP | EPOLLERR))
mode |= EV_WRITE;
if (mode)
netpollready(&gp, (Event *)ep->data.ptr, mode);
}
return gp;
}