2 * uloop - event loop implementation
4 * Copyright (C) 2010-2016 Felix Fietkau <nbd@openwrt.org>
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 static int uloop_init_pollfd(void)
20 struct timespec timeout
= { 0, 0 };
21 struct kevent ev
= {};
30 EV_SET(&ev
, SIGCHLD
, EVFILT_SIGNAL
, EV_ADD
, 0, 0, 0);
31 kevent(poll_fd
, &ev
, 1, NULL
, 0, &timeout
);
37 static uint16_t get_flags(unsigned int flags
, unsigned int mask
)
45 if (flags
& ULOOP_EDGE_TRIGGER
)
51 static struct kevent events
[ULOOP_MAX_EVENTS
];
53 static int register_kevent(struct uloop_fd
*fd
, unsigned int flags
)
55 struct timespec timeout
= { 0, 0 };
62 if (flags
& ULOOP_EDGE_DEFER
)
63 flags
&= ~ULOOP_EDGE_TRIGGER
;
65 changed
= flags
^ fd
->flags
;
66 if (changed
& ULOOP_EDGE_TRIGGER
)
72 if (changed
& ULOOP_READ
) {
73 kflags
= get_flags(flags
, ULOOP_READ
);
74 EV_SET(&ev
[nev
++], fd
->fd
, EVFILT_READ
, kflags
, 0, 0, fd
);
77 if (changed
& ULOOP_WRITE
) {
78 kflags
= get_flags(flags
, ULOOP_WRITE
);
79 EV_SET(&ev
[nev
++], fd
->fd
, EVFILT_WRITE
, kflags
, 0, 0, fd
);
85 if (kevent(poll_fd
, ev
, nev
, NULL
, fl
, &timeout
) == -1)
91 static int register_poll(struct uloop_fd
*fd
, unsigned int flags
)
93 if (flags
& ULOOP_EDGE_TRIGGER
)
94 flags
|= ULOOP_EDGE_DEFER
;
96 flags
&= ~ULOOP_EDGE_DEFER
;
98 return register_kevent(fd
, flags
);
101 static int __uloop_fd_delete(struct uloop_fd
*fd
)
103 return register_poll(fd
, 0);
106 static int64_t get_timestamp_us(void)
108 #ifdef CLOCK_MONOTONIC
109 struct timespec ts
= { 0, 0 };
111 clock_gettime(CLOCK_MONOTONIC
, &ts
);
113 return ts
.tv_sec
* 1000000 + ts
.tv_nsec
/ 1000;
115 struct timeval tv
= { 0, 0 };
117 gettimeofday(&tv
, NULL
);
119 return tv
.tv_sec
* 1000000 + tv
.tv_usec
;
123 static int uloop_fetch_events(int timeout
)
129 ts
.tv_sec
= timeout
/ 1000;
130 ts
.tv_nsec
= (timeout
% 1000) * 1000000;
133 nfds
= kevent(poll_fd
, NULL
, 0, events
, ARRAY_SIZE(events
), timeout
>= 0 ? &ts
: NULL
);
134 for (n
= 0; n
< nfds
; n
++) {
135 if (events
[n
].filter
== EVFILT_TIMER
) {
136 struct uloop_interval
*tm
= events
[n
].udata
;
138 tm
->priv
.time
.fired
= get_timestamp_us();
139 tm
->expirations
+= events
[n
].data
;
145 struct uloop_fd_event
*cur
= &cur_fds
[n
];
146 struct uloop_fd
*u
= events
[n
].udata
;
153 if (events
[n
].flags
& EV_ERROR
) {
155 if (!(u
->flags
& ULOOP_ERROR_CB
))
159 if(events
[n
].filter
== EVFILT_READ
)
161 else if (events
[n
].filter
== EVFILT_WRITE
)
164 if (events
[n
].flags
& EV_EOF
)
170 if (u
->flags
& ULOOP_EDGE_DEFER
) {
171 u
->flags
&= ~ULOOP_EDGE_DEFER
;
172 u
->flags
|= ULOOP_EDGE_TRIGGER
;
173 register_kevent(u
, u
->flags
);
179 static int timer_register(struct uloop_interval
*tm
, unsigned int msecs
)
183 tm
->priv
.time
.msecs
= msecs
;
184 tm
->priv
.time
.fired
= get_timestamp_us();
186 EV_SET(&ev
, (uintptr_t)tm
, EVFILT_TIMER
, EV_ADD
, NOTE_USECONDS
, msecs
* 1000, tm
);
188 return kevent(poll_fd
, &ev
, 1, NULL
, 0, NULL
);
191 static int timer_remove(struct uloop_interval
*tm
)
195 EV_SET(&ev
, (uintptr_t)tm
, EVFILT_TIMER
, EV_DELETE
, 0, 0, NULL
);
197 return kevent(poll_fd
, &ev
, 1, NULL
, 0, NULL
);
200 static int64_t timer_next(struct uloop_interval
*tm
)
202 int64_t t1
= tm
->priv
.time
.fired
;
203 int64_t t2
= get_timestamp_us();
206 t1
+= tm
->priv
.time
.msecs
* 1000;
208 return (t1
- t2
) / 1000;