1 | /* |
1 | /* |
2 | * libevent compatibility layer |
2 | * libevent compatibility layer |
3 | * |
3 | * |
4 | * Copyright (c) 2007 Marc Alexander Lehmann <libev@schmorp.de> |
4 | * Copyright (c) 2007,2008 Marc Alexander Lehmann <libev@schmorp.de> |
5 | * All rights reserved. |
5 | * All rights reserved. |
6 | * |
6 | * |
7 | * Redistribution and use in source and binary forms, with or without |
7 | * Redistribution and use in source and binary forms, with or without modifica- |
8 | * modification, are permitted provided that the following conditions are |
8 | * tion, are permitted provided that the following conditions are met: |
9 | * met: |
9 | * |
|
|
10 | * 1. Redistributions of source code must retain the above copyright notice, |
|
|
11 | * this list of conditions and the following disclaimer. |
|
|
12 | * |
|
|
13 | * 2. Redistributions in binary form must reproduce the above copyright |
|
|
14 | * notice, this list of conditions and the following disclaimer in the |
|
|
15 | * documentation and/or other materials provided with the distribution. |
|
|
16 | * |
|
|
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED |
|
|
18 | * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER- |
|
|
19 | * CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO |
|
|
20 | * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPE- |
|
|
21 | * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
|
|
22 | * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; |
|
|
23 | * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
|
|
24 | * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTH- |
|
|
25 | * ERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED |
|
|
26 | * OF THE POSSIBILITY OF SUCH DAMAGE. |
10 | * |
27 | * |
11 | * * Redistributions of source code must retain the above copyright |
28 | * Alternatively, the contents of this file may be used under the terms of |
12 | * notice, this list of conditions and the following disclaimer. |
29 | * the GNU General Public License ("GPL") version 2 or any later version, |
13 | * |
30 | * in which case the provisions of the GPL are applicable instead of |
14 | * * Redistributions in binary form must reproduce the above |
31 | * the above. If you wish to allow the use of your version of this file |
15 | * copyright notice, this list of conditions and the following |
32 | * only under the terms of the GPL and not to allow others to use your |
16 | * disclaimer in the documentation and/or other materials provided |
33 | * version of this file under the BSD license, indicate your decision |
17 | * with the distribution. |
34 | * by deleting the provisions above and replace them with the notice |
18 | * |
35 | * and other provisions required by the GPL. If you do not delete the |
19 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
36 | * provisions above, a recipient may use your version of this file under |
20 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
37 | * either the BSD or the GPL. |
21 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
|
22 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
|
23 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
|
24 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
|
25 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
|
26 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
|
27 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
|
28 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
|
29 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
30 | */ |
38 | */ |
31 | |
39 | |
32 | #include <stddef.h> |
40 | #include <stddef.h> |
33 | #include <stdlib.h> |
41 | #include <stdlib.h> |
34 | #include <assert.h> |
42 | #include <assert.h> |
35 | |
|
|
36 | #ifndef WIN32 |
|
|
37 | # include <sys/time.h> |
|
|
38 | #endif |
|
|
39 | |
43 | |
40 | #ifdef EV_EVENT_H |
44 | #ifdef EV_EVENT_H |
41 | # include EV_EVENT_H |
45 | # include EV_EVENT_H |
42 | #else |
46 | #else |
43 | # include "event.h" |
47 | # include "event.h" |
… | |
… | |
55 | struct event_base |
59 | struct event_base |
56 | { |
60 | { |
57 | int dummy; |
61 | int dummy; |
58 | }; |
62 | }; |
59 | |
63 | |
60 | static struct event_base *x_cur; |
64 | static struct event_base *ev_x_cur; |
61 | |
65 | |
62 | static void |
66 | static void |
63 | tv_set (struct timeval *tv, ev_tstamp at) |
67 | ev_tv_set (struct timeval *tv, ev_tstamp at) |
64 | { |
68 | { |
65 | tv->tv_sec = (long)at; |
69 | tv->tv_sec = (long)at; |
66 | tv->tv_usec = (long)((at - (ev_tstamp)tv->tv_sec) * 1e6); |
70 | tv->tv_usec = (long)((at - (ev_tstamp)tv->tv_sec) * 1e6); |
67 | } |
71 | } |
68 | |
72 | |
69 | static ev_tstamp |
73 | static ev_tstamp |
70 | tv_get (struct timeval *tv) |
74 | ev_tv_get (struct timeval *tv) |
71 | { |
75 | { |
72 | if (tv) |
76 | if (tv) |
73 | return tv->tv_sec + tv->tv_usec * 1e-6; |
77 | return tv->tv_sec + tv->tv_usec * 1e-6; |
74 | else |
78 | else |
75 | return -1.; |
79 | return -1.; |
… | |
… | |
88 | } |
92 | } |
89 | |
93 | |
90 | void *event_init (void) |
94 | void *event_init (void) |
91 | { |
95 | { |
92 | #if EV_MULTIPLICITY |
96 | #if EV_MULTIPLICITY |
93 | if (x_cur) |
97 | if (ev_x_cur) |
94 | x_cur = (struct event_base *)ev_loop_new (EVFLAG_AUTO); |
98 | ev_x_cur = (struct event_base *)ev_loop_new (EVFLAG_AUTO); |
95 | else |
99 | else |
96 | x_cur = (struct event_base *)ev_default_loop (EVFLAG_AUTO); |
100 | ev_x_cur = (struct event_base *)ev_default_loop (EVFLAG_AUTO); |
97 | #else |
101 | #else |
98 | assert (("multiple event bases not supported when not compiled with EV_MULTIPLICITY", !x_cur)); |
102 | assert (("multiple event bases not supported when not compiled with EV_MULTIPLICITY", !ev_x_cur)); |
99 | |
103 | |
100 | x_cur = (struct event_base *)(long)ev_default_loop (EVFLAG_AUTO); |
104 | ev_x_cur = (struct event_base *)(long)ev_default_loop (EVFLAG_AUTO); |
101 | #endif |
105 | #endif |
102 | |
106 | |
103 | return x_cur; |
107 | return ev_x_cur; |
104 | } |
108 | } |
105 | |
109 | |
106 | void event_base_free (struct event_base *base) |
110 | void event_base_free (struct event_base *base) |
107 | { |
111 | { |
108 | dLOOPbase; |
112 | dLOOPbase; |
… | |
… | |
113 | #endif |
117 | #endif |
114 | } |
118 | } |
115 | |
119 | |
116 | int event_dispatch (void) |
120 | int event_dispatch (void) |
117 | { |
121 | { |
118 | return event_base_dispatch (x_cur); |
122 | return event_base_dispatch (ev_x_cur); |
119 | } |
123 | } |
120 | |
124 | |
121 | #ifdef EV_STANDALONE |
125 | #ifdef EV_STANDALONE |
122 | void event_set_log_callback (event_log_cb cb) |
126 | void event_set_log_callback (event_log_cb cb) |
123 | { |
127 | { |
… | |
… | |
125 | } |
129 | } |
126 | #endif |
130 | #endif |
127 | |
131 | |
128 | int event_loop (int flags) |
132 | int event_loop (int flags) |
129 | { |
133 | { |
130 | return event_base_loop (x_cur, flags); |
134 | return event_base_loop (ev_x_cur, flags); |
131 | } |
135 | } |
132 | |
136 | |
133 | int event_loopexit (struct timeval *tv) |
137 | int event_loopexit (struct timeval *tv) |
134 | { |
138 | { |
135 | return event_base_loopexit (x_cur, tv); |
139 | return event_base_loopexit (ev_x_cur, tv); |
136 | } |
140 | } |
137 | |
141 | |
138 | static void |
142 | static void |
139 | x_cb (struct event *ev, int revents) |
143 | ev_x_cb (struct event *ev, int revents) |
140 | { |
144 | { |
141 | revents &= EV_READ | EV_WRITE | EV_TIMEOUT | EV_SIGNAL; |
145 | revents &= EV_READ | EV_WRITE | EV_TIMEOUT | EV_SIGNAL; |
142 | |
146 | |
143 | ev->ev_res = revents; |
147 | ev->ev_res = revents; |
144 | ev->ev_callback (ev->ev_fd, (short)revents, ev->ev_arg); |
148 | ev->ev_callback (ev->ev_fd, (short)revents, ev->ev_arg); |
145 | } |
149 | } |
146 | |
150 | |
147 | static void |
151 | static void |
148 | x_cb_sig (EV_P_ struct ev_signal *w, int revents) |
152 | ev_x_cb_sig (EV_P_ struct ev_signal *w, int revents) |
149 | { |
153 | { |
150 | struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, iosig.sig)); |
154 | struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, iosig.sig)); |
151 | |
155 | |
152 | if (revents & EV_ERROR) |
156 | if (revents & EV_ERROR) |
153 | event_del (ev); |
157 | event_del (ev); |
154 | |
158 | |
155 | x_cb (ev, revents); |
159 | ev_x_cb (ev, revents); |
156 | } |
160 | } |
157 | |
161 | |
158 | static void |
162 | static void |
159 | x_cb_io (EV_P_ struct ev_io *w, int revents) |
163 | ev_x_cb_io (EV_P_ struct ev_io *w, int revents) |
160 | { |
164 | { |
161 | struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, iosig.io)); |
165 | struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, iosig.io)); |
162 | |
166 | |
163 | if ((revents & EV_ERROR) || !(ev->ev_events & EV_PERSIST)) |
167 | if ((revents & EV_ERROR) || !(ev->ev_events & EV_PERSIST)) |
164 | event_del (ev); |
168 | event_del (ev); |
165 | |
169 | |
166 | x_cb (ev, revents); |
170 | ev_x_cb (ev, revents); |
167 | } |
171 | } |
168 | |
172 | |
169 | static void |
173 | static void |
170 | x_cb_to (EV_P_ struct ev_timer *w, int revents) |
174 | ev_x_cb_to (EV_P_ struct ev_timer *w, int revents) |
171 | { |
175 | { |
172 | struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, to)); |
176 | struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, to)); |
173 | |
177 | |
174 | event_del (ev); |
178 | event_del (ev); |
175 | |
179 | |
176 | x_cb (ev, revents); |
180 | ev_x_cb (ev, revents); |
177 | } |
181 | } |
178 | |
182 | |
179 | void event_set (struct event *ev, int fd, short events, void (*cb)(int, short, void *), void *arg) |
183 | void event_set (struct event *ev, int fd, short events, void (*cb)(int, short, void *), void *arg) |
180 | { |
184 | { |
181 | if (events & EV_SIGNAL) |
185 | if (events & EV_SIGNAL) |
182 | ev_init (&ev->iosig.sig, x_cb_sig); |
186 | ev_init (&ev->iosig.sig, ev_x_cb_sig); |
183 | else |
187 | else |
184 | ev_init (&ev->iosig.io, x_cb_io); |
188 | ev_init (&ev->iosig.io, ev_x_cb_io); |
185 | |
189 | |
186 | ev_init (&ev->to, x_cb_to); |
190 | ev_init (&ev->to, ev_x_cb_to); |
187 | |
191 | |
188 | ev->ev_base = x_cur; /* not threadsafe, but its like libevent works */ |
192 | ev->ev_base = ev_x_cur; /* not threadsafe, but it's how libevent works */ |
189 | ev->ev_fd = fd; |
193 | ev->ev_fd = fd; |
190 | ev->ev_events = events; |
194 | ev->ev_events = events; |
191 | ev->ev_pri = 0; |
195 | ev->ev_pri = 0; |
192 | ev->ev_callback = cb; |
196 | ev->ev_callback = cb; |
193 | ev->ev_arg = arg; |
197 | ev->ev_arg = arg; |
… | |
… | |
195 | ev->ev_flags = EVLIST_INIT; |
199 | ev->ev_flags = EVLIST_INIT; |
196 | } |
200 | } |
197 | |
201 | |
198 | int event_once (int fd, short events, void (*cb)(int, short, void *), void *arg, struct timeval *tv) |
202 | int event_once (int fd, short events, void (*cb)(int, short, void *), void *arg, struct timeval *tv) |
199 | { |
203 | { |
200 | return event_base_once (x_cur, fd, events, cb, arg, tv); |
204 | return event_base_once (ev_x_cur, fd, events, cb, arg, tv); |
201 | } |
205 | } |
202 | |
206 | |
203 | int event_add (struct event *ev, struct timeval *tv) |
207 | int event_add (struct event *ev, struct timeval *tv) |
204 | { |
208 | { |
205 | dLOOPev; |
209 | dLOOPev; |
… | |
… | |
225 | } |
229 | } |
226 | } |
230 | } |
227 | |
231 | |
228 | if (tv) |
232 | if (tv) |
229 | { |
233 | { |
230 | ev->to.repeat = tv_get (tv); |
234 | ev->to.repeat = ev_tv_get (tv); |
231 | ev_timer_again (EV_A_ &ev->to); |
235 | ev_timer_again (EV_A_ &ev->to); |
232 | ev->ev_flags |= EVLIST_TIMEOUT; |
236 | ev->ev_flags |= EVLIST_TIMEOUT; |
233 | } |
237 | } |
234 | else |
238 | else |
235 | { |
239 | { |
… | |
… | |
276 | int event_pending (struct event *ev, short events, struct timeval *tv) |
280 | int event_pending (struct event *ev, short events, struct timeval *tv) |
277 | { |
281 | { |
278 | short revents = 0; |
282 | short revents = 0; |
279 | dLOOPev; |
283 | dLOOPev; |
280 | |
284 | |
281 | |
|
|
282 | if (ev->ev_events & EV_SIGNAL) |
285 | if (ev->ev_events & EV_SIGNAL) |
283 | { |
286 | { |
284 | /* sig */ |
287 | /* sig */ |
285 | if (ev_is_active (&ev->iosig.sig) || ev_is_pending (&ev->iosig.sig)) |
288 | if (ev_is_active (&ev->iosig.sig) || ev_is_pending (&ev->iosig.sig)) |
286 | revents |= EV_SIGNAL; |
289 | revents |= EV_SIGNAL; |
… | |
… | |
295 | if (ev->ev_events & EV_TIMEOUT || ev_is_active (&ev->to) || ev_is_pending (&ev->to)) |
298 | if (ev->ev_events & EV_TIMEOUT || ev_is_active (&ev->to) || ev_is_pending (&ev->to)) |
296 | { |
299 | { |
297 | revents |= EV_TIMEOUT; |
300 | revents |= EV_TIMEOUT; |
298 | |
301 | |
299 | if (tv) |
302 | if (tv) |
300 | tv_set (tv, ev_now (EV_A)); /* not sure if this is right :) */ |
303 | ev_tv_set (tv, ev_now (EV_A)); /* not sure if this is right :) */ |
301 | } |
304 | } |
302 | |
305 | |
303 | return events & revents; |
306 | return events & revents; |
304 | } |
307 | } |
305 | |
308 | |
306 | int event_priority_init (int npri) |
309 | int event_priority_init (int npri) |
307 | { |
310 | { |
308 | return event_base_priority_init (x_cur, npri); |
311 | return event_base_priority_init (ev_x_cur, npri); |
309 | } |
312 | } |
310 | |
313 | |
311 | int event_priority_set (struct event *ev, int pri) |
314 | int event_priority_set (struct event *ev, int pri) |
312 | { |
315 | { |
313 | ev->ev_pri = pri; |
316 | ev->ev_pri = pri; |
… | |
… | |
335 | { |
338 | { |
336 | return event_base_loop (base, 0); |
339 | return event_base_loop (base, 0); |
337 | } |
340 | } |
338 | |
341 | |
339 | static void |
342 | static void |
340 | x_loopexit_cb (int revents, void *base) |
343 | ev_x_loopexit_cb (int revents, void *base) |
341 | { |
344 | { |
342 | dLOOPbase; |
345 | dLOOPbase; |
343 | |
346 | |
344 | ev_unloop (EV_A_ EVUNLOOP_ONE); |
347 | ev_unloop (EV_A_ EVUNLOOP_ONE); |
345 | } |
348 | } |
346 | |
349 | |
347 | int event_base_loopexit (struct event_base *base, struct timeval *tv) |
350 | int event_base_loopexit (struct event_base *base, struct timeval *tv) |
348 | { |
351 | { |
349 | ev_tstamp after = tv_get (tv); |
352 | ev_tstamp after = ev_tv_get (tv); |
350 | dLOOPbase; |
353 | dLOOPbase; |
351 | |
354 | |
352 | ev_once (EV_A_ -1, 0, after >= 0. ? after : 0., x_loopexit_cb, (void *)base); |
355 | ev_once (EV_A_ -1, 0, after >= 0. ? after : 0., ev_x_loopexit_cb, (void *)base); |
353 | |
356 | |
354 | return -1; |
357 | return 0; |
355 | } |
358 | } |
356 | |
359 | |
357 | struct x_once |
360 | struct ev_x_once |
358 | { |
361 | { |
359 | int fd; |
362 | int fd; |
360 | void (*cb)(int, short, void *); |
363 | void (*cb)(int, short, void *); |
361 | void *arg; |
364 | void *arg; |
362 | }; |
365 | }; |
363 | |
366 | |
364 | static void |
367 | static void |
365 | x_once_cb (int revents, void *arg) |
368 | ev_x_once_cb (int revents, void *arg) |
366 | { |
369 | { |
367 | struct x_once *once = (struct x_once *)arg; |
370 | struct ev_x_once *once = (struct ev_x_once *)arg; |
368 | |
371 | |
369 | once->cb (once->fd, (short)revents, once->arg); |
372 | once->cb (once->fd, (short)revents, once->arg); |
370 | free (once); |
373 | free (once); |
371 | } |
374 | } |
372 | |
375 | |
373 | int event_base_once (struct event_base *base, int fd, short events, void (*cb)(int, short, void *), void *arg, struct timeval *tv) |
376 | int event_base_once (struct event_base *base, int fd, short events, void (*cb)(int, short, void *), void *arg, struct timeval *tv) |
374 | { |
377 | { |
375 | struct x_once *once = (struct x_once *)malloc (sizeof (struct x_once)); |
378 | struct ev_x_once *once = (struct ev_x_once *)malloc (sizeof (struct ev_x_once)); |
376 | dLOOPbase; |
379 | dLOOPbase; |
377 | |
380 | |
378 | if (!once) |
381 | if (!once) |
379 | return -1; |
382 | return -1; |
380 | |
383 | |
381 | once->fd = fd; |
384 | once->fd = fd; |
382 | once->cb = cb; |
385 | once->cb = cb; |
383 | once->arg = arg; |
386 | once->arg = arg; |
384 | |
387 | |
385 | ev_once (EV_A_ fd, events & (EV_READ | EV_WRITE), tv_get (tv), x_once_cb, (void *)once); |
388 | ev_once (EV_A_ fd, events & (EV_READ | EV_WRITE), ev_tv_get (tv), ev_x_once_cb, (void *)once); |
386 | |
389 | |
387 | return 0; |
390 | return 0; |
388 | } |
391 | } |
389 | |
392 | |
390 | int event_base_priority_init (struct event_base *base, int npri) |
393 | int event_base_priority_init (struct event_base *base, int npri) |