ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libev/ev.c
(Generate patch)

Comparing libev/ev.c (file contents):
Revision 1.10 by root, Wed Oct 31 07:36:03 2007 UTC vs.
Revision 1.118 by root, Fri Nov 16 01:33:54 2007 UTC

1/*
2 * libev event processing core, watcher management
3 *
4 * Copyright (c) 2007 Marc Alexander Lehmann <libev@schmorp.de>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following
16 * disclaimer in the documentation and/or other materials provided
17 * with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
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 */
31
32#ifdef __cplusplus
33extern "C" {
34#endif
35
36#ifndef EV_STANDALONE
37# include "config.h"
38
39# if HAVE_CLOCK_GETTIME
40# ifndef EV_USE_MONOTONIC
41# define EV_USE_MONOTONIC 1
42# endif
43# ifndef EV_USE_REALTIME
44# define EV_USE_REALTIME 1
45# endif
46# endif
47
48# if HAVE_SELECT && HAVE_SYS_SELECT_H && !defined (EV_USE_SELECT)
49# define EV_USE_SELECT 1
50# endif
51
52# if HAVE_POLL && HAVE_POLL_H && !defined (EV_USE_POLL)
53# define EV_USE_POLL 1
54# endif
55
56# if HAVE_EPOLL_CTL && HAVE_SYS_EPOLL_H && !defined (EV_USE_EPOLL)
57# define EV_USE_EPOLL 1
58# endif
59
60# if HAVE_KQUEUE && HAVE_SYS_EVENT_H && HAVE_SYS_QUEUE_H && !defined (EV_USE_KQUEUE)
61# define EV_USE_KQUEUE 1
62# endif
63
64# if HAVE_PORT_H && HAVE_PORT_H && HAVE_PORT_CREATE && !defined (EV_USE_PORT)
65# define EV_USE_PORT 1
66# endif
67
68#endif
69
1#include <math.h> 70#include <math.h>
2#include <stdlib.h> 71#include <stdlib.h>
3#include <unistd.h>
4#include <fcntl.h> 72#include <fcntl.h>
5#include <signal.h> 73#include <stddef.h>
6 74
7#include <stdio.h> 75#include <stdio.h>
8 76
9#include <assert.h> 77#include <assert.h>
10#include <errno.h> 78#include <errno.h>
11#include <sys/time.h> 79#include <sys/types.h>
12#include <time.h> 80#include <time.h>
13 81
14#ifndef HAVE_MONOTONIC 82#include <signal.h>
15# ifdef CLOCK_MONOTONIC 83
16# define HAVE_MONOTONIC 1 84#ifndef _WIN32
85# include <unistd.h>
86# include <sys/time.h>
87# include <sys/wait.h>
88#else
89# define WIN32_LEAN_AND_MEAN
90# include <windows.h>
91# ifndef EV_SELECT_IS_WINSOCKET
92# define EV_SELECT_IS_WINSOCKET 1
17# endif 93# endif
18#endif 94#endif
19 95
96/**/
97
98#ifndef EV_USE_MONOTONIC
99# define EV_USE_MONOTONIC 1
100#endif
101
102#ifndef EV_USE_REALTIME
103# define EV_USE_REALTIME 1
104#endif
105
20#ifndef HAVE_SELECT 106#ifndef EV_USE_SELECT
21# define HAVE_SELECT 1 107# define EV_USE_SELECT 1
108# define EV_SELECT_USE_FD_SET 1
109#endif
110
111#ifndef EV_USE_POLL
112# ifdef _WIN32
113# define EV_USE_POLL 0
114# else
115# define EV_USE_POLL 1
22#endif 116# endif
117#endif
23 118
24#ifndef HAVE_EPOLL 119#ifndef EV_USE_EPOLL
25# define HAVE_EPOLL 0 120# define EV_USE_EPOLL 0
26#endif 121#endif
27 122
123#ifndef EV_USE_KQUEUE
124# define EV_USE_KQUEUE 0
125#endif
126
127#ifndef EV_USE_PORT
128# define EV_USE_PORT 0
129#endif
130
131/**/
132
133/* darwin simply cannot be helped */
134#ifdef __APPLE__
135# undef EV_USE_POLL
136# undef EV_USE_KQUEUE
137#endif
138
139#ifndef CLOCK_MONOTONIC
140# undef EV_USE_MONOTONIC
141# define EV_USE_MONOTONIC 0
142#endif
143
28#ifndef HAVE_REALTIME 144#ifndef CLOCK_REALTIME
29# define HAVE_REALTIME 1 /* posix requirement, but might be slower */ 145# undef EV_USE_REALTIME
146# define EV_USE_REALTIME 0
30#endif 147#endif
148
149#if EV_SELECT_IS_WINSOCKET
150# include <winsock.h>
151#endif
152
153/**/
31 154
32#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */ 155#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */
33#define MAX_BLOCKTIME 60. 156#define MAX_BLOCKTIME 59.731 /* never wait longer than this time (to detect time jumps) */
157#define PID_HASHSIZE 16 /* size of pid hash table, must be power of two */
158/*#define CLEANUP_INTERVAL 300. /* how often to try to free memory and re-check fds */
34 159
160#ifdef EV_H
161# include EV_H
162#else
35#include "ev.h" 163# include "ev.h"
164#endif
36 165
37struct ev_watcher { 166#if __GNUC__ >= 3
38 EV_WATCHER (ev_watcher); 167# define expect(expr,value) __builtin_expect ((expr),(value))
39}; 168# define inline inline
169#else
170# define expect(expr,value) (expr)
171# define inline static
172#endif
40 173
41struct ev_watcher_list { 174#define expect_false(expr) expect ((expr) != 0, 0)
42 EV_WATCHER_LIST (ev_watcher_list); 175#define expect_true(expr) expect ((expr) != 0, 1)
43}; 176
177#define NUMPRI (EV_MAXPRI - EV_MINPRI + 1)
178#define ABSPRI(w) ((w)->priority - EV_MINPRI)
179
180#define EMPTY0 /* required for microsofts broken pseudo-c compiler */
181#define EMPTY2(a,b) /* used to suppress some warnings */
44 182
45typedef struct ev_watcher *W; 183typedef struct ev_watcher *W;
46typedef struct ev_watcher_list *WL; 184typedef struct ev_watcher_list *WL;
185typedef struct ev_watcher_time *WT;
47 186
48static ev_tstamp now, diff; /* monotonic clock */ 187static int have_monotonic; /* did clock_gettime (CLOCK_MONOTONIC) work? */
188
189#ifdef _WIN32
190# include "ev_win32.c"
191#endif
192
193/*****************************************************************************/
194
195static void (*syserr_cb)(const char *msg);
196
197void ev_set_syserr_cb (void (*cb)(const char *msg))
198{
199 syserr_cb = cb;
200}
201
202static void
203syserr (const char *msg)
204{
205 if (!msg)
206 msg = "(libev) system error";
207
208 if (syserr_cb)
209 syserr_cb (msg);
210 else
211 {
212 perror (msg);
213 abort ();
214 }
215}
216
217static void *(*alloc)(void *ptr, long size);
218
219void ev_set_allocator (void *(*cb)(void *ptr, long size))
220{
221 alloc = cb;
222}
223
224static void *
225ev_realloc (void *ptr, long size)
226{
227 ptr = alloc ? alloc (ptr, size) : realloc (ptr, size);
228
229 if (!ptr && size)
230 {
231 fprintf (stderr, "libev: cannot allocate %ld bytes, aborting.", size);
232 abort ();
233 }
234
235 return ptr;
236}
237
238#define ev_malloc(size) ev_realloc (0, (size))
239#define ev_free(ptr) ev_realloc ((ptr), 0)
240
241/*****************************************************************************/
242
243typedef struct
244{
245 WL head;
246 unsigned char events;
247 unsigned char reify;
248#if EV_SELECT_IS_WINSOCKET
249 SOCKET handle;
250#endif
251} ANFD;
252
253typedef struct
254{
255 W w;
256 int events;
257} ANPENDING;
258
259#if EV_MULTIPLICITY
260
261 struct ev_loop
262 {
263 ev_tstamp ev_rt_now;
264 #define ev_rt_now ((loop)->ev_rt_now)
265 #define VAR(name,decl) decl;
266 #include "ev_vars.h"
267 #undef VAR
268 };
269 #include "ev_wrap.h"
270
271 static struct ev_loop default_loop_struct;
272 struct ev_loop *ev_default_loop_ptr;
273
274#else
275
49ev_tstamp ev_now; 276 ev_tstamp ev_rt_now;
50int ev_method; 277 #define VAR(name,decl) static decl;
278 #include "ev_vars.h"
279 #undef VAR
51 280
52static int have_monotonic; /* runtime */ 281 static int ev_default_loop_ptr;
53 282
54static ev_tstamp method_fudge; /* stupid epoll-returns-early bug */ 283#endif
55static void (*method_modify)(int fd, int oev, int nev);
56static void (*method_poll)(ev_tstamp timeout);
57 284
58/*****************************************************************************/ 285/*****************************************************************************/
59 286
60ev_tstamp 287ev_tstamp
61ev_time (void) 288ev_time (void)
62{ 289{
63#if HAVE_REALTIME 290#if EV_USE_REALTIME
64 struct timespec ts; 291 struct timespec ts;
65 clock_gettime (CLOCK_REALTIME, &ts); 292 clock_gettime (CLOCK_REALTIME, &ts);
66 return ts.tv_sec + ts.tv_nsec * 1e-9; 293 return ts.tv_sec + ts.tv_nsec * 1e-9;
67#else 294#else
68 struct timeval tv; 295 struct timeval tv;
69 gettimeofday (&tv, 0); 296 gettimeofday (&tv, 0);
70 return tv.tv_sec + tv.tv_usec * 1e-6; 297 return tv.tv_sec + tv.tv_usec * 1e-6;
71#endif 298#endif
72} 299}
73 300
74static ev_tstamp 301inline ev_tstamp
75get_clock (void) 302get_clock (void)
76{ 303{
77#if HAVE_MONOTONIC 304#if EV_USE_MONOTONIC
78 if (have_monotonic) 305 if (expect_true (have_monotonic))
79 { 306 {
80 struct timespec ts; 307 struct timespec ts;
81 clock_gettime (CLOCK_MONOTONIC, &ts); 308 clock_gettime (CLOCK_MONOTONIC, &ts);
82 return ts.tv_sec + ts.tv_nsec * 1e-9; 309 return ts.tv_sec + ts.tv_nsec * 1e-9;
83 } 310 }
84#endif 311#endif
85 312
86 return ev_time (); 313 return ev_time ();
87} 314}
88 315
316#if EV_MULTIPLICITY
317ev_tstamp
318ev_now (EV_P)
319{
320 return ev_rt_now;
321}
322#endif
323
324#define array_roundsize(type,n) (((n) | 4) & ~3)
325
89#define array_needsize(base,cur,cnt,init) \ 326#define array_needsize(type,base,cur,cnt,init) \
90 if ((cnt) > cur) \ 327 if (expect_false ((cnt) > cur)) \
91 { \ 328 { \
92 int newcnt = cur ? cur << 1 : 16; \ 329 int newcnt = cur; \
93 fprintf (stderr, "resize(" # base ") from %d to %d\n", cur, newcnt);\ 330 do \
331 { \
332 newcnt = array_roundsize (type, newcnt << 1); \
333 } \
334 while ((cnt) > newcnt); \
335 \
94 base = realloc (base, sizeof (*base) * (newcnt)); \ 336 base = (type *)ev_realloc (base, sizeof (type) * (newcnt));\
95 init (base + cur, newcnt - cur); \ 337 init (base + cur, newcnt - cur); \
96 cur = newcnt; \ 338 cur = newcnt; \
97 } 339 }
340
341#define array_slim(type,stem) \
342 if (stem ## max < array_roundsize (stem ## cnt >> 2)) \
343 { \
344 stem ## max = array_roundsize (stem ## cnt >> 1); \
345 base = (type *)ev_realloc (base, sizeof (type) * (stem ## max));\
346 fprintf (stderr, "slimmed down " # stem " to %d\n", stem ## max);/*D*/\
347 }
348
349#define array_free(stem, idx) \
350 ev_free (stem ## s idx); stem ## cnt idx = stem ## max idx = 0;
98 351
99/*****************************************************************************/ 352/*****************************************************************************/
100 353
101typedef struct
102{
103 struct ev_io *head;
104 unsigned char wev, rev; /* want, received event set */
105} ANFD;
106
107static ANFD *anfds;
108static int anfdmax;
109
110static int *fdchanges;
111static int fdchangemax, fdchangecnt;
112
113static void 354static void
114anfds_init (ANFD *base, int count) 355anfds_init (ANFD *base, int count)
115{ 356{
116 while (count--) 357 while (count--)
117 { 358 {
118 base->head = 0; 359 base->head = 0;
119 base->wev = base->rev = EV_NONE; 360 base->events = EV_NONE;
361 base->reify = 0;
362
120 ++base; 363 ++base;
121 } 364 }
122} 365}
123 366
124typedef struct 367void
368ev_feed_event (EV_P_ void *w, int revents)
125{ 369{
126 W w; 370 W w_ = (W)w;
127 int events;
128} ANPENDING;
129 371
130static ANPENDING *pendings; 372 if (w_->pending)
131static int pendingmax, pendingcnt; 373 {
374 pendings [ABSPRI (w_)][w_->pending - 1].events |= revents;
375 return;
376 }
132 377
133static void
134event (W w, int events)
135{
136 w->pending = ++pendingcnt; 378 w_->pending = ++pendingcnt [ABSPRI (w_)];
137 array_needsize (pendings, pendingmax, pendingcnt, ); 379 array_needsize (ANPENDING, pendings [ABSPRI (w_)], pendingmax [ABSPRI (w_)], pendingcnt [ABSPRI (w_)], EMPTY2);
138 pendings [pendingcnt - 1].w = w; 380 pendings [ABSPRI (w_)][w_->pending - 1].w = w_;
139 pendings [pendingcnt - 1].events = events; 381 pendings [ABSPRI (w_)][w_->pending - 1].events = revents;
140} 382}
141 383
142static void 384static void
385queue_events (EV_P_ W *events, int eventcnt, int type)
386{
387 int i;
388
389 for (i = 0; i < eventcnt; ++i)
390 ev_feed_event (EV_A_ events [i], type);
391}
392
393inline void
143fd_event (int fd, int events) 394fd_event (EV_P_ int fd, int revents)
144{ 395{
145 ANFD *anfd = anfds + fd; 396 ANFD *anfd = anfds + fd;
146 struct ev_io *w; 397 struct ev_io *w;
147 398
148 for (w = anfd->head; w; w = w->next) 399 for (w = (struct ev_io *)anfd->head; w; w = (struct ev_io *)((WL)w)->next)
149 { 400 {
150 int ev = w->events & events; 401 int ev = w->events & revents;
151 402
152 if (ev) 403 if (ev)
153 event ((W)w, ev); 404 ev_feed_event (EV_A_ (W)w, ev);
154 } 405 }
155} 406}
156 407
157static void 408void
158queue_events (W *events, int eventcnt, int type) 409ev_feed_fd_event (EV_P_ int fd, int revents)
159{ 410{
160 int i; 411 fd_event (EV_A_ fd, revents);
161
162 for (i = 0; i < eventcnt; ++i)
163 event (events [i], type);
164} 412}
165 413
166/*****************************************************************************/ 414/*****************************************************************************/
167 415
168static struct ev_timer **atimers;
169static int atimermax, atimercnt;
170
171static struct ev_timer **rtimers;
172static int rtimermax, rtimercnt;
173
174static void 416static void
175upheap (struct ev_timer **timers, int k) 417fd_reify (EV_P)
176{
177 struct ev_timer *w = timers [k];
178
179 while (k && timers [k >> 1]->at > w->at)
180 {
181 timers [k] = timers [k >> 1];
182 timers [k]->active = k + 1;
183 k >>= 1;
184 }
185
186 timers [k] = w;
187 timers [k]->active = k + 1;
188
189}
190
191static void
192downheap (struct ev_timer **timers, int N, int k)
193{
194 struct ev_timer *w = timers [k];
195
196 while (k < (N >> 1))
197 {
198 int j = k << 1;
199
200 if (j + 1 < N && timers [j]->at > timers [j + 1]->at)
201 ++j;
202
203 if (w->at <= timers [j]->at)
204 break;
205
206 timers [k] = timers [j];
207 timers [k]->active = k + 1;
208 k = j;
209 }
210
211 timers [k] = w;
212 timers [k]->active = k + 1;
213}
214
215/*****************************************************************************/
216
217typedef struct
218{
219 struct ev_signal *head;
220 sig_atomic_t gotsig;
221} ANSIG;
222
223static ANSIG *signals;
224static int signalmax;
225
226static int sigpipe [2];
227static sig_atomic_t gotsig;
228static struct ev_io sigev;
229
230static void
231signals_init (ANSIG *base, int count)
232{
233 while (count--)
234 {
235 base->head = 0;
236 base->gotsig = 0;
237 ++base;
238 }
239}
240
241static void
242sighandler (int signum)
243{
244 signals [signum - 1].gotsig = 1;
245
246 if (!gotsig)
247 {
248 gotsig = 1;
249 write (sigpipe [1], &gotsig, 1);
250 }
251}
252
253static void
254sigcb (struct ev_io *iow, int revents)
255{
256 struct ev_signal *w;
257 int sig;
258
259 gotsig = 0;
260 read (sigpipe [0], &revents, 1);
261
262 for (sig = signalmax; sig--; )
263 if (signals [sig].gotsig)
264 {
265 signals [sig].gotsig = 0;
266
267 for (w = signals [sig].head; w; w = w->next)
268 event ((W)w, EV_SIGNAL);
269 }
270}
271
272static void
273siginit (void)
274{
275 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
276 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
277
278 /* rather than sort out wether we really need nb, set it */
279 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
280 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
281
282 evio_set (&sigev, sigpipe [0], EV_READ);
283 evio_start (&sigev);
284}
285
286/*****************************************************************************/
287
288static struct ev_idle **idles;
289static int idlemax, idlecnt;
290
291static struct ev_check **checks;
292static int checkmax, checkcnt;
293
294/*****************************************************************************/
295
296#if HAVE_EPOLL
297# include "ev_epoll.c"
298#endif
299#if HAVE_SELECT
300# include "ev_select.c"
301#endif
302
303int ev_init (int flags)
304{
305#if HAVE_MONOTONIC
306 {
307 struct timespec ts;
308 if (!clock_gettime (CLOCK_MONOTONIC, &ts))
309 have_monotonic = 1;
310 }
311#endif
312
313 ev_now = ev_time ();
314 now = get_clock ();
315 diff = ev_now - now;
316
317 if (pipe (sigpipe))
318 return 0;
319
320 ev_method = EVMETHOD_NONE;
321#if HAVE_EPOLL
322 if (ev_method == EVMETHOD_NONE) epoll_init (flags);
323#endif
324#if HAVE_SELECT
325 if (ev_method == EVMETHOD_NONE) select_init (flags);
326#endif
327
328 if (ev_method)
329 {
330 evw_init (&sigev, sigcb, 0);
331 siginit ();
332 }
333
334 return ev_method;
335}
336
337/*****************************************************************************/
338
339void ev_prefork (void)
340{
341}
342
343void ev_postfork_parent (void)
344{
345}
346
347void ev_postfork_child (void)
348{
349#if HAVE_EPOLL
350 if (ev_method == EVMETHOD_EPOLL)
351 epoll_postfork_child ();
352#endif
353
354 evio_stop (&sigev);
355 close (sigpipe [0]);
356 close (sigpipe [1]);
357 pipe (sigpipe);
358 siginit ();
359}
360
361/*****************************************************************************/
362
363static void
364fd_reify (void)
365{ 418{
366 int i; 419 int i;
367 420
368 for (i = 0; i < fdchangecnt; ++i) 421 for (i = 0; i < fdchangecnt; ++i)
369 { 422 {
370 int fd = fdchanges [i]; 423 int fd = fdchanges [i];
371 ANFD *anfd = anfds + fd; 424 ANFD *anfd = anfds + fd;
372 struct ev_io *w; 425 struct ev_io *w;
373 426
374 int wev = 0; 427 int events = 0;
375 428
376 for (w = anfd->head; w; w = w->next) 429 for (w = (struct ev_io *)anfd->head; w; w = (struct ev_io *)((WL)w)->next)
377 wev |= w->events; 430 events |= w->events;
378 431
379 if (anfd->wev != wev) 432#if EV_SELECT_IS_WINSOCKET
433 if (events)
380 { 434 {
381 method_modify (fd, anfd->wev, wev); 435 unsigned long argp;
382 anfd->wev = wev; 436 anfd->handle = _get_osfhandle (fd);
437 assert (("libev only supports socket fds in this configuration", ioctlsocket (anfd->handle, FIONREAD, &argp) == 0));
383 } 438 }
439#endif
440
441 anfd->reify = 0;
442
443 method_modify (EV_A_ fd, anfd->events, events);
444 anfd->events = events;
384 } 445 }
385 446
386 fdchangecnt = 0; 447 fdchangecnt = 0;
387} 448}
388 449
389static void 450static void
390call_pending () 451fd_change (EV_P_ int fd)
452{
453 if (anfds [fd].reify)
454 return;
455
456 anfds [fd].reify = 1;
457
458 ++fdchangecnt;
459 array_needsize (int, fdchanges, fdchangemax, fdchangecnt, EMPTY2);
460 fdchanges [fdchangecnt - 1] = fd;
461}
462
463static void
464fd_kill (EV_P_ int fd)
465{
466 struct ev_io *w;
467
468 while ((w = (struct ev_io *)anfds [fd].head))
469 {
470 ev_io_stop (EV_A_ w);
471 ev_feed_event (EV_A_ (W)w, EV_ERROR | EV_READ | EV_WRITE);
472 }
473}
474
475static int
476fd_valid (int fd)
477{
478#ifdef _WIN32
479 return _get_osfhandle (fd) != -1;
480#else
481 return fcntl (fd, F_GETFD) != -1;
482#endif
483}
484
485/* called on EBADF to verify fds */
486static void
487fd_ebadf (EV_P)
488{
489 int fd;
490
491 for (fd = 0; fd < anfdmax; ++fd)
492 if (anfds [fd].events)
493 if (!fd_valid (fd) == -1 && errno == EBADF)
494 fd_kill (EV_A_ fd);
495}
496
497/* called on ENOMEM in select/poll to kill some fds and retry */
498static void
499fd_enomem (EV_P)
500{
501 int fd;
502
503 for (fd = anfdmax; fd--; )
504 if (anfds [fd].events)
505 {
506 fd_kill (EV_A_ fd);
507 return;
508 }
509}
510
511/* usually called after fork if method needs to re-arm all fds from scratch */
512static void
513fd_rearm_all (EV_P)
514{
515 int fd;
516
517 /* this should be highly optimised to not do anything but set a flag */
518 for (fd = 0; fd < anfdmax; ++fd)
519 if (anfds [fd].events)
520 {
521 anfds [fd].events = 0;
522 fd_change (EV_A_ fd);
523 }
524}
525
526/*****************************************************************************/
527
528static void
529upheap (WT *heap, int k)
530{
531 WT w = heap [k];
532
533 while (k && heap [k >> 1]->at > w->at)
534 {
535 heap [k] = heap [k >> 1];
536 ((W)heap [k])->active = k + 1;
537 k >>= 1;
538 }
539
540 heap [k] = w;
541 ((W)heap [k])->active = k + 1;
542
543}
544
545static void
546downheap (WT *heap, int N, int k)
547{
548 WT w = heap [k];
549
550 while (k < (N >> 1))
551 {
552 int j = k << 1;
553
554 if (j + 1 < N && heap [j]->at > heap [j + 1]->at)
555 ++j;
556
557 if (w->at <= heap [j]->at)
558 break;
559
560 heap [k] = heap [j];
561 ((W)heap [k])->active = k + 1;
562 k = j;
563 }
564
565 heap [k] = w;
566 ((W)heap [k])->active = k + 1;
567}
568
569inline void
570adjustheap (WT *heap, int N, int k)
571{
572 upheap (heap, k);
573 downheap (heap, N, k);
574}
575
576/*****************************************************************************/
577
578typedef struct
579{
580 WL head;
581 sig_atomic_t volatile gotsig;
582} ANSIG;
583
584static ANSIG *signals;
585static int signalmax;
586
587static int sigpipe [2];
588static sig_atomic_t volatile gotsig;
589static struct ev_io sigev;
590
591static void
592signals_init (ANSIG *base, int count)
593{
594 while (count--)
595 {
596 base->head = 0;
597 base->gotsig = 0;
598
599 ++base;
600 }
601}
602
603static void
604sighandler (int signum)
605{
606#if _WIN32
607 signal (signum, sighandler);
608#endif
609
610 signals [signum - 1].gotsig = 1;
611
612 if (!gotsig)
613 {
614 int old_errno = errno;
615 gotsig = 1;
616 write (sigpipe [1], &signum, 1);
617 errno = old_errno;
618 }
619}
620
621void
622ev_feed_signal_event (EV_P_ int signum)
623{
624 WL w;
625
626#if EV_MULTIPLICITY
627 assert (("feeding signal events is only supported in the default loop", loop == ev_default_loop_ptr));
628#endif
629
630 --signum;
631
632 if (signum < 0 || signum >= signalmax)
633 return;
634
635 signals [signum].gotsig = 0;
636
637 for (w = signals [signum].head; w; w = w->next)
638 ev_feed_event (EV_A_ (W)w, EV_SIGNAL);
639}
640
641static void
642sigcb (EV_P_ struct ev_io *iow, int revents)
643{
644 int signum;
645
646 read (sigpipe [0], &revents, 1);
647 gotsig = 0;
648
649 for (signum = signalmax; signum--; )
650 if (signals [signum].gotsig)
651 ev_feed_signal_event (EV_A_ signum + 1);
652}
653
654inline void
655fd_intern (int fd)
656{
657#ifdef _WIN32
658 int arg = 1;
659 ioctlsocket (_get_osfhandle (fd), FIONBIO, &arg);
660#else
661 fcntl (fd, F_SETFD, FD_CLOEXEC);
662 fcntl (fd, F_SETFL, O_NONBLOCK);
663#endif
664}
665
666static void
667siginit (EV_P)
668{
669 fd_intern (sigpipe [0]);
670 fd_intern (sigpipe [1]);
671
672 ev_io_set (&sigev, sigpipe [0], EV_READ);
673 ev_io_start (EV_A_ &sigev);
674 ev_unref (EV_A); /* child watcher should not keep loop alive */
675}
676
677/*****************************************************************************/
678
679static struct ev_child *childs [PID_HASHSIZE];
680
681#ifndef _WIN32
682
683static struct ev_signal childev;
684
685#ifndef WCONTINUED
686# define WCONTINUED 0
687#endif
688
689static void
690child_reap (EV_P_ struct ev_signal *sw, int chain, int pid, int status)
691{
692 struct ev_child *w;
693
694 for (w = (struct ev_child *)childs [chain & (PID_HASHSIZE - 1)]; w; w = (struct ev_child *)((WL)w)->next)
695 if (w->pid == pid || !w->pid)
696 {
697 ev_priority (w) = ev_priority (sw); /* need to do it *now* */
698 w->rpid = pid;
699 w->rstatus = status;
700 ev_feed_event (EV_A_ (W)w, EV_CHILD);
701 }
702}
703
704static void
705childcb (EV_P_ struct ev_signal *sw, int revents)
706{
707 int pid, status;
708
709 if (0 < (pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)))
710 {
711 /* make sure we are called again until all childs have been reaped */
712 ev_feed_event (EV_A_ (W)sw, EV_SIGNAL);
713
714 child_reap (EV_A_ sw, pid, pid, status);
715 child_reap (EV_A_ sw, 0, pid, status); /* this might trigger a watcher twice, but event catches that */
716 }
717}
718
719#endif
720
721/*****************************************************************************/
722
723#if EV_USE_PORT
724# include "ev_port.c"
725#endif
726#if EV_USE_KQUEUE
727# include "ev_kqueue.c"
728#endif
729#if EV_USE_EPOLL
730# include "ev_epoll.c"
731#endif
732#if EV_USE_POLL
733# include "ev_poll.c"
734#endif
735#if EV_USE_SELECT
736# include "ev_select.c"
737#endif
738
739int
740ev_version_major (void)
741{
742 return EV_VERSION_MAJOR;
743}
744
745int
746ev_version_minor (void)
747{
748 return EV_VERSION_MINOR;
749}
750
751/* return true if we are running with elevated privileges and should ignore env variables */
752static int
753enable_secure (void)
754{
755#ifdef _WIN32
756 return 0;
757#else
758 return getuid () != geteuid ()
759 || getgid () != getegid ();
760#endif
761}
762
763unsigned int
764ev_method (EV_P)
765{
766 return method;
767}
768
769static void
770loop_init (EV_P_ unsigned int flags)
771{
772 if (!method)
773 {
774#if EV_USE_MONOTONIC
775 {
776 struct timespec ts;
777 if (!clock_gettime (CLOCK_MONOTONIC, &ts))
778 have_monotonic = 1;
779 }
780#endif
781
782 ev_rt_now = ev_time ();
783 mn_now = get_clock ();
784 now_floor = mn_now;
785 rtmn_diff = ev_rt_now - mn_now;
786
787 if (!(flags & EVFLAG_NOENV) && !enable_secure () && getenv ("LIBEV_FLAGS"))
788 flags = atoi (getenv ("LIBEV_FLAGS"));
789
790 if (!(flags & 0x0000ffff))
791 flags |= 0x0000ffff;
792
793 method = 0;
794#if EV_USE_PORT
795 if (!method && (flags & EVMETHOD_PORT )) method = port_init (EV_A_ flags);
796#endif
797#if EV_USE_KQUEUE
798 if (!method && (flags & EVMETHOD_KQUEUE)) method = kqueue_init (EV_A_ flags);
799#endif
800#if EV_USE_EPOLL
801 if (!method && (flags & EVMETHOD_EPOLL )) method = epoll_init (EV_A_ flags);
802#endif
803#if EV_USE_POLL
804 if (!method && (flags & EVMETHOD_POLL )) method = poll_init (EV_A_ flags);
805#endif
806#if EV_USE_SELECT
807 if (!method && (flags & EVMETHOD_SELECT)) method = select_init (EV_A_ flags);
808#endif
809
810 ev_init (&sigev, sigcb);
811 ev_set_priority (&sigev, EV_MAXPRI);
812 }
813}
814
815void
816loop_destroy (EV_P)
391{ 817{
392 int i; 818 int i;
393 819
394 for (i = 0; i < pendingcnt; ++i) 820#if EV_USE_PORT
821 if (method == EVMETHOD_PORT ) port_destroy (EV_A);
822#endif
823#if EV_USE_KQUEUE
824 if (method == EVMETHOD_KQUEUE) kqueue_destroy (EV_A);
825#endif
826#if EV_USE_EPOLL
827 if (method == EVMETHOD_EPOLL ) epoll_destroy (EV_A);
828#endif
829#if EV_USE_POLL
830 if (method == EVMETHOD_POLL ) poll_destroy (EV_A);
831#endif
832#if EV_USE_SELECT
833 if (method == EVMETHOD_SELECT) select_destroy (EV_A);
834#endif
835
836 for (i = NUMPRI; i--; )
837 array_free (pending, [i]);
838
839 /* have to use the microsoft-never-gets-it-right macro */
840 array_free (fdchange, EMPTY0);
841 array_free (timer, EMPTY0);
842#if EV_PERIODICS
843 array_free (periodic, EMPTY0);
844#endif
845 array_free (idle, EMPTY0);
846 array_free (prepare, EMPTY0);
847 array_free (check, EMPTY0);
848
849 method = 0;
850}
851
852static void
853loop_fork (EV_P)
854{
855#if EV_USE_PORT
856 if (method == EVMETHOD_PORT ) port_fork (EV_A);
857#endif
858#if EV_USE_KQUEUE
859 if (method == EVMETHOD_KQUEUE) kqueue_fork (EV_A);
860#endif
861#if EV_USE_EPOLL
862 if (method == EVMETHOD_EPOLL ) epoll_fork (EV_A);
863#endif
864
865 if (ev_is_active (&sigev))
866 {
867 /* default loop */
868
869 ev_ref (EV_A);
870 ev_io_stop (EV_A_ &sigev);
871 close (sigpipe [0]);
872 close (sigpipe [1]);
873
874 while (pipe (sigpipe))
875 syserr ("(libev) error creating pipe");
876
877 siginit (EV_A);
395 { 878 }
396 ANPENDING *p = pendings + i;
397 879
398 if (p->w) 880 postfork = 0;
881}
882
883#if EV_MULTIPLICITY
884struct ev_loop *
885ev_loop_new (unsigned int flags)
886{
887 struct ev_loop *loop = (struct ev_loop *)ev_malloc (sizeof (struct ev_loop));
888
889 memset (loop, 0, sizeof (struct ev_loop));
890
891 loop_init (EV_A_ flags);
892
893 if (ev_method (EV_A))
894 return loop;
895
896 return 0;
897}
898
899void
900ev_loop_destroy (EV_P)
901{
902 loop_destroy (EV_A);
903 ev_free (loop);
904}
905
906void
907ev_loop_fork (EV_P)
908{
909 postfork = 1;
910}
911
912#endif
913
914#if EV_MULTIPLICITY
915struct ev_loop *
916ev_default_loop_ (unsigned int flags)
917#else
918int
919ev_default_loop (unsigned int flags)
920#endif
921{
922 if (sigpipe [0] == sigpipe [1])
923 if (pipe (sigpipe))
924 return 0;
925
926 if (!ev_default_loop_ptr)
927 {
928#if EV_MULTIPLICITY
929 struct ev_loop *loop = ev_default_loop_ptr = &default_loop_struct;
930#else
931 ev_default_loop_ptr = 1;
932#endif
933
934 loop_init (EV_A_ flags);
935
936 if (ev_method (EV_A))
399 { 937 {
400 p->w->pending = 0; 938 siginit (EV_A);
401 p->w->cb (p->w, p->events); 939
940#ifndef _WIN32
941 ev_signal_init (&childev, childcb, SIGCHLD);
942 ev_set_priority (&childev, EV_MAXPRI);
943 ev_signal_start (EV_A_ &childev);
944 ev_unref (EV_A); /* child watcher should not keep loop alive */
945#endif
402 } 946 }
947 else
948 ev_default_loop_ptr = 0;
949 }
950
951 return ev_default_loop_ptr;
952}
953
954void
955ev_default_destroy (void)
956{
957#if EV_MULTIPLICITY
958 struct ev_loop *loop = ev_default_loop_ptr;
959#endif
960
961#ifndef _WIN32
962 ev_ref (EV_A); /* child watcher */
963 ev_signal_stop (EV_A_ &childev);
964#endif
965
966 ev_ref (EV_A); /* signal watcher */
967 ev_io_stop (EV_A_ &sigev);
968
969 close (sigpipe [0]); sigpipe [0] = 0;
970 close (sigpipe [1]); sigpipe [1] = 0;
971
972 loop_destroy (EV_A);
973}
974
975void
976ev_default_fork (void)
977{
978#if EV_MULTIPLICITY
979 struct ev_loop *loop = ev_default_loop_ptr;
980#endif
981
982 if (method)
983 postfork = 1;
984}
985
986/*****************************************************************************/
987
988static int
989any_pending (EV_P)
990{
991 int pri;
992
993 for (pri = NUMPRI; pri--; )
994 if (pendingcnt [pri])
995 return 1;
996
997 return 0;
998}
999
1000static void
1001call_pending (EV_P)
1002{
1003 int pri;
1004
1005 for (pri = NUMPRI; pri--; )
1006 while (pendingcnt [pri])
1007 {
1008 ANPENDING *p = pendings [pri] + --pendingcnt [pri];
1009
1010 if (p->w)
1011 {
1012 p->w->pending = 0;
1013 EV_CB_INVOKE (p->w, p->events);
1014 }
403 } 1015 }
404
405 pendingcnt = 0;
406} 1016}
407 1017
408static void 1018static void
409timers_reify (struct ev_timer **timers, int timercnt, ev_tstamp now) 1019timers_reify (EV_P)
410{ 1020{
411 while (timercnt && timers [0]->at <= now) 1021 while (timercnt && ((WT)timers [0])->at <= mn_now)
412 { 1022 {
413 struct ev_timer *w = timers [0]; 1023 struct ev_timer *w = timers [0];
1024
1025 assert (("inactive timer on timer heap detected", ev_is_active (w)));
414 1026
415 /* first reschedule or stop timer */ 1027 /* first reschedule or stop timer */
416 if (w->repeat) 1028 if (w->repeat)
417 { 1029 {
418 if (w->is_abs) 1030 assert (("negative ev_timer repeat value found while processing timers", w->repeat > 0.));
419 w->at += floor ((now - w->at) / w->repeat + 1.) * w->repeat; 1031
420 else
421 w->at = now + w->repeat; 1032 ((WT)w)->at += w->repeat;
1033 if (((WT)w)->at < mn_now)
1034 ((WT)w)->at = mn_now;
422 1035
423 assert (w->at > now);
424
425 downheap (timers, timercnt, 0); 1036 downheap ((WT *)timers, timercnt, 0);
426 } 1037 }
427 else 1038 else
1039 ev_timer_stop (EV_A_ w); /* nonrepeating: stop timer */
1040
1041 ev_feed_event (EV_A_ (W)w, EV_TIMEOUT);
1042 }
1043}
1044
1045#if EV_PERIODICS
1046static void
1047periodics_reify (EV_P)
1048{
1049 while (periodiccnt && ((WT)periodics [0])->at <= ev_rt_now)
1050 {
1051 struct ev_periodic *w = periodics [0];
1052
1053 assert (("inactive timer on periodic heap detected", ev_is_active (w)));
1054
1055 /* first reschedule or stop timer */
1056 if (w->reschedule_cb)
428 { 1057 {
429 evtimer_stop (w); /* nonrepeating: stop timer */ 1058 ((WT)w)->at = w->reschedule_cb (w, ev_rt_now + 0.0001);
430 --timercnt; /* maybe pass by reference instead? */ 1059 assert (("ev_periodic reschedule callback returned time in the past", ((WT)w)->at > ev_rt_now));
1060 downheap ((WT *)periodics, periodiccnt, 0);
431 } 1061 }
1062 else if (w->interval)
1063 {
1064 ((WT)w)->at += floor ((ev_rt_now - ((WT)w)->at) / w->interval + 1.) * w->interval;
1065 assert (("ev_periodic timeout in the past detected while processing timers, negative interval?", ((WT)w)->at > ev_rt_now));
1066 downheap ((WT *)periodics, periodiccnt, 0);
1067 }
1068 else
1069 ev_periodic_stop (EV_A_ w); /* nonrepeating: stop timer */
432 1070
433 event ((W)w, EV_TIMEOUT); 1071 ev_feed_event (EV_A_ (W)w, EV_PERIODIC);
434 } 1072 }
435} 1073}
436 1074
437static void 1075static void
438time_update () 1076periodics_reschedule (EV_P)
439{ 1077{
440 int i; 1078 int i;
1079
1080 /* adjust periodics after time jump */
1081 for (i = 0; i < periodiccnt; ++i)
1082 {
1083 struct ev_periodic *w = periodics [i];
1084
1085 if (w->reschedule_cb)
1086 ((WT)w)->at = w->reschedule_cb (w, ev_rt_now);
1087 else if (w->interval)
1088 ((WT)w)->at += ceil ((ev_rt_now - ((WT)w)->at) / w->interval) * w->interval;
1089 }
1090
1091 /* now rebuild the heap */
1092 for (i = periodiccnt >> 1; i--; )
1093 downheap ((WT *)periodics, periodiccnt, i);
1094}
1095#endif
1096
1097inline int
1098time_update_monotonic (EV_P)
1099{
1100 mn_now = get_clock ();
1101
1102 if (expect_true (mn_now - now_floor < MIN_TIMEJUMP * .5))
1103 {
1104 ev_rt_now = rtmn_diff + mn_now;
1105 return 0;
1106 }
1107 else
1108 {
1109 now_floor = mn_now;
441 ev_now = ev_time (); 1110 ev_rt_now = ev_time ();
442 1111 return 1;
443 if (have_monotonic)
444 { 1112 }
445 ev_tstamp odiff = diff; 1113}
446 1114
447 /* detecting time jumps is much more difficult */ 1115static void
448 for (i = 2; --i; ) /* loop a few times, before making important decisions */ 1116time_update (EV_P)
1117{
1118 int i;
1119
1120#if EV_USE_MONOTONIC
1121 if (expect_true (have_monotonic))
1122 {
1123 if (time_update_monotonic (EV_A))
449 { 1124 {
450 now = get_clock (); 1125 ev_tstamp odiff = rtmn_diff;
1126
1127 for (i = 4; --i; ) /* loop a few times, before making important decisions */
1128 {
451 diff = ev_now - now; 1129 rtmn_diff = ev_rt_now - mn_now;
452 1130
453 if (fabs (odiff - diff) < MIN_TIMEJUMP) 1131 if (fabs (odiff - rtmn_diff) < MIN_TIMEJUMP)
454 return; /* all is well */ 1132 return; /* all is well */
455 1133
456 ev_now = ev_time (); 1134 ev_rt_now = ev_time ();
1135 mn_now = get_clock ();
1136 now_floor = mn_now;
1137 }
1138
1139# if EV_PERIODICS
1140 periodics_reschedule (EV_A);
1141# endif
1142 /* no timer adjustment, as the monotonic clock doesn't jump */
1143 /* timers_reschedule (EV_A_ rtmn_diff - odiff) */
457 } 1144 }
1145 }
1146 else
1147#endif
1148 {
1149 ev_rt_now = ev_time ();
458 1150
459 /* time jump detected, reschedule atimers */ 1151 if (expect_false (mn_now > ev_rt_now || mn_now < ev_rt_now - MAX_BLOCKTIME - MIN_TIMEJUMP))
460 for (i = 0; i < atimercnt; ++i)
461 { 1152 {
462 struct ev_timer *w = atimers [i]; 1153#if EV_PERIODICS
463 w->at += ceil ((ev_now - w->at) / w->repeat + 1.) * w->repeat; 1154 periodics_reschedule (EV_A);
1155#endif
1156
1157 /* adjust timers. this is easy, as the offset is the same for all */
1158 for (i = 0; i < timercnt; ++i)
1159 ((WT)timers [i])->at += ev_rt_now - mn_now;
464 } 1160 }
465 }
466 else
467 {
468 if (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP)
469 /* time jump detected, adjust rtimers */
470 for (i = 0; i < rtimercnt; ++i)
471 rtimers [i]->at += ev_now - now;
472 1161
473 now = ev_now; 1162 mn_now = ev_rt_now;
474 } 1163 }
475} 1164}
476 1165
477int ev_loop_done; 1166void
1167ev_ref (EV_P)
1168{
1169 ++activecnt;
1170}
478 1171
1172void
1173ev_unref (EV_P)
1174{
1175 --activecnt;
1176}
1177
1178static int loop_done;
1179
1180void
479void ev_loop (int flags) 1181ev_loop (EV_P_ int flags)
480{ 1182{
481 double block; 1183 double block;
482 ev_loop_done = flags & EVLOOP_ONESHOT; 1184 loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0;
483 1185
484 if (checkcnt) 1186 while (activecnt)
485 { 1187 {
486 queue_events ((W *)checks, checkcnt, EV_CHECK); 1188 /* queue check watchers (and execute them) */
1189 if (expect_false (preparecnt))
1190 {
1191 queue_events (EV_A_ (W *)prepares, preparecnt, EV_PREPARE);
487 call_pending (); 1192 call_pending (EV_A);
488 } 1193 }
489 1194
490 do 1195 /* we might have forked, so reify kernel state if necessary */
491 { 1196 if (expect_false (postfork))
1197 loop_fork (EV_A);
1198
492 /* update fd-related kernel structures */ 1199 /* update fd-related kernel structures */
493 fd_reify (); 1200 fd_reify (EV_A);
494 1201
495 /* calculate blocking time */ 1202 /* calculate blocking time */
1203
1204 /* we only need this for !monotonic clock or timers, but as we basically
1205 always have timers, we just calculate it always */
1206#if EV_USE_MONOTONIC
1207 if (expect_true (have_monotonic))
1208 time_update_monotonic (EV_A);
1209 else
1210#endif
1211 {
1212 ev_rt_now = ev_time ();
1213 mn_now = ev_rt_now;
1214 }
1215
496 if (flags & EVLOOP_NONBLOCK || idlecnt) 1216 if (flags & EVLOOP_NONBLOCK || idlecnt)
497 block = 0.; 1217 block = 0.;
498 else 1218 else
499 { 1219 {
500 block = MAX_BLOCKTIME; 1220 block = MAX_BLOCKTIME;
501 1221
502 if (rtimercnt) 1222 if (timercnt)
503 { 1223 {
504 ev_tstamp to = rtimers [0]->at - get_clock () + method_fudge; 1224 ev_tstamp to = ((WT)timers [0])->at - mn_now + method_fudge;
505 if (block > to) block = to; 1225 if (block > to) block = to;
506 } 1226 }
507 1227
1228#if EV_PERIODICS
508 if (atimercnt) 1229 if (periodiccnt)
509 { 1230 {
510 ev_tstamp to = atimers [0]->at - ev_time () + method_fudge; 1231 ev_tstamp to = ((WT)periodics [0])->at - ev_rt_now + method_fudge;
511 if (block > to) block = to; 1232 if (block > to) block = to;
512 } 1233 }
1234#endif
513 1235
514 if (block < 0.) block = 0.; 1236 if (block < 0.) block = 0.;
515 } 1237 }
516 1238
517 method_poll (block); 1239 method_poll (EV_A_ block);
518 1240
519 /* update ev_now, do magic */ 1241 /* update ev_rt_now, do magic */
520 time_update (); 1242 time_update (EV_A);
521 1243
522 /* queue pending timers and reschedule them */ 1244 /* queue pending timers and reschedule them */
523 /* absolute timers first */ 1245 timers_reify (EV_A); /* relative timers called last */
524 timers_reify (atimers, atimercnt, ev_now); 1246#if EV_PERIODICS
525 /* relative timers second */ 1247 periodics_reify (EV_A); /* absolute timers called first */
526 timers_reify (rtimers, rtimercnt, now); 1248#endif
527 1249
528 /* queue idle watchers unless io or timers are pending */ 1250 /* queue idle watchers unless io or timers are pending */
529 if (!pendingcnt) 1251 if (idlecnt && !any_pending (EV_A))
530 queue_events ((W *)idles, idlecnt, EV_IDLE); 1252 queue_events (EV_A_ (W *)idles, idlecnt, EV_IDLE);
531 1253
532 /* queue check and possibly idle watchers */ 1254 /* queue check watchers, to be executed first */
1255 if (checkcnt)
533 queue_events ((W *)checks, checkcnt, EV_CHECK); 1256 queue_events (EV_A_ (W *)checks, checkcnt, EV_CHECK);
534 1257
535 call_pending (); 1258 call_pending (EV_A);
1259
1260 if (loop_done)
1261 break;
536 } 1262 }
537 while (!ev_loop_done); 1263
1264 if (loop_done != 2)
1265 loop_done = 0;
1266}
1267
1268void
1269ev_unloop (EV_P_ int how)
1270{
1271 loop_done = how;
538} 1272}
539 1273
540/*****************************************************************************/ 1274/*****************************************************************************/
541 1275
542static void 1276inline void
543wlist_add (WL *head, WL elem) 1277wlist_add (WL *head, WL elem)
544{ 1278{
545 elem->next = *head; 1279 elem->next = *head;
546 *head = elem; 1280 *head = elem;
547} 1281}
548 1282
549static void 1283inline void
550wlist_del (WL *head, WL elem) 1284wlist_del (WL *head, WL elem)
551{ 1285{
552 while (*head) 1286 while (*head)
553 { 1287 {
554 if (*head == elem) 1288 if (*head == elem)
559 1293
560 head = &(*head)->next; 1294 head = &(*head)->next;
561 } 1295 }
562} 1296}
563 1297
564static void 1298inline void
1299ev_clear_pending (EV_P_ W w)
1300{
1301 if (w->pending)
1302 {
1303 pendings [ABSPRI (w)][w->pending - 1].w = 0;
1304 w->pending = 0;
1305 }
1306}
1307
1308inline void
565ev_start (W w, int active) 1309ev_start (EV_P_ W w, int active)
566{ 1310{
567 w->pending = 0; 1311 if (w->priority < EV_MINPRI) w->priority = EV_MINPRI;
1312 if (w->priority > EV_MAXPRI) w->priority = EV_MAXPRI;
1313
568 w->active = active; 1314 w->active = active;
1315 ev_ref (EV_A);
569} 1316}
570 1317
571static void 1318inline void
572ev_stop (W w) 1319ev_stop (EV_P_ W w)
573{ 1320{
574 if (w->pending) 1321 ev_unref (EV_A);
575 pendings [w->pending - 1].w = 0;
576
577 w->active = 0; 1322 w->active = 0;
578 /* nop */
579} 1323}
580 1324
581/*****************************************************************************/ 1325/*****************************************************************************/
582 1326
583void 1327void
584evio_start (struct ev_io *w) 1328ev_io_start (EV_P_ struct ev_io *w)
585{ 1329{
1330 int fd = w->fd;
1331
586 if (ev_is_active (w)) 1332 if (ev_is_active (w))
587 return; 1333 return;
588 1334
589 int fd = w->fd; 1335 assert (("ev_io_start called with negative fd", fd >= 0));
590 1336
591 ev_start ((W)w, 1); 1337 ev_start (EV_A_ (W)w, 1);
592 array_needsize (anfds, anfdmax, fd + 1, anfds_init); 1338 array_needsize (ANFD, anfds, anfdmax, fd + 1, anfds_init);
593 wlist_add ((WL *)&anfds[fd].head, (WL)w); 1339 wlist_add ((WL *)&anfds[fd].head, (WL)w);
594 1340
595 ++fdchangecnt; 1341 fd_change (EV_A_ fd);
596 array_needsize (fdchanges, fdchangemax, fdchangecnt, );
597 fdchanges [fdchangecnt - 1] = fd;
598} 1342}
599 1343
600void 1344void
601evio_stop (struct ev_io *w) 1345ev_io_stop (EV_P_ struct ev_io *w)
602{ 1346{
1347 ev_clear_pending (EV_A_ (W)w);
603 if (!ev_is_active (w)) 1348 if (!ev_is_active (w))
604 return; 1349 return;
605 1350
1351 assert (("ev_io_start called with illegal fd (must stay constant after start!)", w->fd >= 0 && w->fd < anfdmax));
1352
606 wlist_del ((WL *)&anfds[w->fd].head, (WL)w); 1353 wlist_del ((WL *)&anfds[w->fd].head, (WL)w);
607 ev_stop ((W)w); 1354 ev_stop (EV_A_ (W)w);
608 1355
609 ++fdchangecnt; 1356 fd_change (EV_A_ w->fd);
610 array_needsize (fdchanges, fdchangemax, fdchangecnt, );
611 fdchanges [fdchangecnt - 1] = w->fd;
612} 1357}
613 1358
614void 1359void
615evtimer_start (struct ev_timer *w) 1360ev_timer_start (EV_P_ struct ev_timer *w)
616{ 1361{
617 if (ev_is_active (w)) 1362 if (ev_is_active (w))
618 return; 1363 return;
619 1364
620 if (w->is_abs) 1365 ((WT)w)->at += mn_now;
1366
1367 assert (("ev_timer_start called with negative timer repeat value", w->repeat >= 0.));
1368
1369 ev_start (EV_A_ (W)w, ++timercnt);
1370 array_needsize (struct ev_timer *, timers, timermax, timercnt, EMPTY2);
1371 timers [timercnt - 1] = w;
1372 upheap ((WT *)timers, timercnt - 1);
1373
1374 assert (("internal timer heap corruption", timers [((W)w)->active - 1] == w));
1375}
1376
1377void
1378ev_timer_stop (EV_P_ struct ev_timer *w)
1379{
1380 ev_clear_pending (EV_A_ (W)w);
1381 if (!ev_is_active (w))
1382 return;
1383
1384 assert (("internal timer heap corruption", timers [((W)w)->active - 1] == w));
1385
1386 if (((W)w)->active < timercnt--)
1387 {
1388 timers [((W)w)->active - 1] = timers [timercnt];
1389 adjustheap ((WT *)timers, timercnt, ((W)w)->active - 1);
621 { 1390 }
622 /* this formula differs from the one in timer_reify becuse we do not round up */ 1391
1392 ((WT)w)->at -= mn_now;
1393
1394 ev_stop (EV_A_ (W)w);
1395}
1396
1397void
1398ev_timer_again (EV_P_ struct ev_timer *w)
1399{
1400 if (ev_is_active (w))
1401 {
623 if (w->repeat) 1402 if (w->repeat)
624 w->at += ceil ((ev_now - w->at) / w->repeat) * w->repeat; 1403 {
625 1404 ((WT)w)->at = mn_now + w->repeat;
626 ev_start ((W)w, ++atimercnt); 1405 adjustheap ((WT *)timers, timercnt, ((W)w)->active - 1);
627 array_needsize (atimers, atimermax, atimercnt, ); 1406 }
628 atimers [atimercnt - 1] = w;
629 upheap (atimers, atimercnt - 1);
630 }
631 else 1407 else
1408 ev_timer_stop (EV_A_ w);
632 { 1409 }
633 w->at += now; 1410 else if (w->repeat)
634
635 ev_start ((W)w, ++rtimercnt);
636 array_needsize (rtimers, rtimermax, rtimercnt, );
637 rtimers [rtimercnt - 1] = w;
638 upheap (rtimers, rtimercnt - 1);
639 } 1411 {
640 1412 w->at = w->repeat;
1413 ev_timer_start (EV_A_ w);
1414 }
641} 1415}
642 1416
1417#if EV_PERIODICS
643void 1418void
644evtimer_stop (struct ev_timer *w) 1419ev_periodic_start (EV_P_ struct ev_periodic *w)
645{ 1420{
1421 if (ev_is_active (w))
1422 return;
1423
1424 if (w->reschedule_cb)
1425 ((WT)w)->at = w->reschedule_cb (w, ev_rt_now);
1426 else if (w->interval)
1427 {
1428 assert (("ev_periodic_start called with negative interval value", w->interval >= 0.));
1429 /* this formula differs from the one in periodic_reify because we do not always round up */
1430 ((WT)w)->at += ceil ((ev_rt_now - ((WT)w)->at) / w->interval) * w->interval;
1431 }
1432
1433 ev_start (EV_A_ (W)w, ++periodiccnt);
1434 array_needsize (struct ev_periodic *, periodics, periodicmax, periodiccnt, EMPTY2);
1435 periodics [periodiccnt - 1] = w;
1436 upheap ((WT *)periodics, periodiccnt - 1);
1437
1438 assert (("internal periodic heap corruption", periodics [((W)w)->active - 1] == w));
1439}
1440
1441void
1442ev_periodic_stop (EV_P_ struct ev_periodic *w)
1443{
1444 ev_clear_pending (EV_A_ (W)w);
646 if (!ev_is_active (w)) 1445 if (!ev_is_active (w))
647 return; 1446 return;
648 1447
649 if (w->is_abs) 1448 assert (("internal periodic heap corruption", periodics [((W)w)->active - 1] == w));
650 { 1449
651 if (w->active < atimercnt--) 1450 if (((W)w)->active < periodiccnt--)
652 {
653 atimers [w->active - 1] = atimers [atimercnt];
654 downheap (atimers, atimercnt, w->active - 1);
655 }
656 } 1451 {
657 else 1452 periodics [((W)w)->active - 1] = periodics [periodiccnt];
1453 adjustheap ((WT *)periodics, periodiccnt, ((W)w)->active - 1);
658 { 1454 }
659 if (w->active < rtimercnt--)
660 {
661 rtimers [w->active - 1] = rtimers [rtimercnt];
662 downheap (rtimers, rtimercnt, w->active - 1);
663 }
664 }
665 1455
666 ev_stop ((W)w); 1456 ev_stop (EV_A_ (W)w);
667} 1457}
668 1458
669void 1459void
670evsignal_start (struct ev_signal *w) 1460ev_periodic_again (EV_P_ struct ev_periodic *w)
1461{
1462 /* TODO: use adjustheap and recalculation */
1463 ev_periodic_stop (EV_A_ w);
1464 ev_periodic_start (EV_A_ w);
1465}
1466#endif
1467
1468void
1469ev_idle_start (EV_P_ struct ev_idle *w)
671{ 1470{
672 if (ev_is_active (w)) 1471 if (ev_is_active (w))
673 return; 1472 return;
674 1473
1474 ev_start (EV_A_ (W)w, ++idlecnt);
1475 array_needsize (struct ev_idle *, idles, idlemax, idlecnt, EMPTY2);
1476 idles [idlecnt - 1] = w;
1477}
1478
1479void
1480ev_idle_stop (EV_P_ struct ev_idle *w)
1481{
1482 ev_clear_pending (EV_A_ (W)w);
1483 if (!ev_is_active (w))
1484 return;
1485
1486 idles [((W)w)->active - 1] = idles [--idlecnt];
1487 ev_stop (EV_A_ (W)w);
1488}
1489
1490void
1491ev_prepare_start (EV_P_ struct ev_prepare *w)
1492{
1493 if (ev_is_active (w))
1494 return;
1495
1496 ev_start (EV_A_ (W)w, ++preparecnt);
1497 array_needsize (struct ev_prepare *, prepares, preparemax, preparecnt, EMPTY2);
1498 prepares [preparecnt - 1] = w;
1499}
1500
1501void
1502ev_prepare_stop (EV_P_ struct ev_prepare *w)
1503{
1504 ev_clear_pending (EV_A_ (W)w);
1505 if (!ev_is_active (w))
1506 return;
1507
1508 prepares [((W)w)->active - 1] = prepares [--preparecnt];
1509 ev_stop (EV_A_ (W)w);
1510}
1511
1512void
1513ev_check_start (EV_P_ struct ev_check *w)
1514{
1515 if (ev_is_active (w))
1516 return;
1517
1518 ev_start (EV_A_ (W)w, ++checkcnt);
1519 array_needsize (struct ev_check *, checks, checkmax, checkcnt, EMPTY2);
1520 checks [checkcnt - 1] = w;
1521}
1522
1523void
1524ev_check_stop (EV_P_ struct ev_check *w)
1525{
1526 ev_clear_pending (EV_A_ (W)w);
1527 if (!ev_is_active (w))
1528 return;
1529
1530 checks [((W)w)->active - 1] = checks [--checkcnt];
1531 ev_stop (EV_A_ (W)w);
1532}
1533
1534#ifndef SA_RESTART
1535# define SA_RESTART 0
1536#endif
1537
1538void
1539ev_signal_start (EV_P_ struct ev_signal *w)
1540{
1541#if EV_MULTIPLICITY
1542 assert (("signal watchers are only supported in the default loop", loop == ev_default_loop_ptr));
1543#endif
1544 if (ev_is_active (w))
1545 return;
1546
1547 assert (("ev_signal_start called with illegal signal number", w->signum > 0));
1548
675 ev_start ((W)w, 1); 1549 ev_start (EV_A_ (W)w, 1);
676 array_needsize (signals, signalmax, w->signum, signals_init); 1550 array_needsize (ANSIG, signals, signalmax, w->signum, signals_init);
677 wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w); 1551 wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w);
678 1552
679 if (!w->next) 1553 if (!((WL)w)->next)
680 { 1554 {
1555#if _WIN32
1556 signal (w->signum, sighandler);
1557#else
681 struct sigaction sa; 1558 struct sigaction sa;
682 sa.sa_handler = sighandler; 1559 sa.sa_handler = sighandler;
683 sigfillset (&sa.sa_mask); 1560 sigfillset (&sa.sa_mask);
684 sa.sa_flags = 0; 1561 sa.sa_flags = SA_RESTART; /* if restarting works we save one iteration */
685 sigaction (w->signum, &sa, 0); 1562 sigaction (w->signum, &sa, 0);
1563#endif
686 } 1564 }
687} 1565}
688 1566
689void 1567void
690evsignal_stop (struct ev_signal *w) 1568ev_signal_stop (EV_P_ struct ev_signal *w)
691{ 1569{
1570 ev_clear_pending (EV_A_ (W)w);
692 if (!ev_is_active (w)) 1571 if (!ev_is_active (w))
693 return; 1572 return;
694 1573
695 wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w); 1574 wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w);
696 ev_stop ((W)w); 1575 ev_stop (EV_A_ (W)w);
697 1576
698 if (!signals [w->signum - 1].head) 1577 if (!signals [w->signum - 1].head)
699 signal (w->signum, SIG_DFL); 1578 signal (w->signum, SIG_DFL);
700} 1579}
701 1580
702void evidle_start (struct ev_idle *w) 1581void
1582ev_child_start (EV_P_ struct ev_child *w)
703{ 1583{
1584#if EV_MULTIPLICITY
1585 assert (("child watchers are only supported in the default loop", loop == ev_default_loop_ptr));
1586#endif
704 if (ev_is_active (w)) 1587 if (ev_is_active (w))
705 return; 1588 return;
706 1589
707 ev_start ((W)w, ++idlecnt); 1590 ev_start (EV_A_ (W)w, 1);
708 array_needsize (idles, idlemax, idlecnt, ); 1591 wlist_add ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
709 idles [idlecnt - 1] = w;
710} 1592}
711 1593
712void evidle_stop (struct ev_idle *w) 1594void
1595ev_child_stop (EV_P_ struct ev_child *w)
713{ 1596{
714 idles [w->active - 1] = idles [--idlecnt]; 1597 ev_clear_pending (EV_A_ (W)w);
715 ev_stop ((W)w);
716}
717
718void evcheck_start (struct ev_check *w)
719{
720 if (ev_is_active (w)) 1598 if (!ev_is_active (w))
721 return; 1599 return;
722 1600
723 ev_start ((W)w, ++checkcnt); 1601 wlist_del ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
724 array_needsize (checks, checkmax, checkcnt, );
725 checks [checkcnt - 1] = w;
726}
727
728void evcheck_stop (struct ev_check *w)
729{
730 checks [w->active - 1] = checks [--checkcnt];
731 ev_stop ((W)w); 1602 ev_stop (EV_A_ (W)w);
732} 1603}
733 1604
734/*****************************************************************************/ 1605/*****************************************************************************/
735 1606
736#if 0 1607struct ev_once
737
738static void
739sin_cb (struct ev_io *w, int revents)
740{ 1608{
741 fprintf (stderr, "sin %d, revents %d\n", w->fd, revents);
742}
743
744static void
745ocb (struct ev_timer *w, int revents)
746{
747 //fprintf (stderr, "timer %f,%f (%x) (%f) d%p\n", w->at, w->repeat, revents, w->at - ev_time (), w->data);
748 evtimer_stop (w);
749 evtimer_start (w);
750}
751
752static void
753scb (struct ev_signal *w, int revents)
754{
755 fprintf (stderr, "signal %x,%d\n", revents, w->signum);
756}
757
758static void
759gcb (struct ev_signal *w, int revents)
760{
761 fprintf (stderr, "generic %x\n", revents);
762}
763
764int main (void)
765{
766 struct ev_io sin; 1609 struct ev_io io;
767
768 ev_init (0);
769
770 evw_init (&sin, sin_cb, 55);
771 evio_set (&sin, 0, EV_READ);
772 evio_start (&sin);
773
774 struct ev_timer t[10000];
775
776#if 0
777 int i;
778 for (i = 0; i < 10000; ++i)
779 {
780 struct ev_timer *w = t + i;
781 evw_init (w, ocb, i);
782 evtimer_set_abs (w, drand48 (), 0.99775533);
783 evtimer_start (w);
784 if (drand48 () < 0.5)
785 evtimer_stop (w);
786 }
787#endif
788
789 struct ev_timer t1; 1610 struct ev_timer to;
790 evw_init (&t1, ocb, 0); 1611 void (*cb)(int revents, void *arg);
791 evtimer_set_abs (&t1, 5, 10); 1612 void *arg;
792 evtimer_start (&t1); 1613};
793 1614
794 struct ev_signal sig; 1615static void
795 evw_init (&sig, scb, 65535); 1616once_cb (EV_P_ struct ev_once *once, int revents)
796 evsignal_set (&sig, SIGQUIT); 1617{
797 evsignal_start (&sig); 1618 void (*cb)(int revents, void *arg) = once->cb;
1619 void *arg = once->arg;
798 1620
799 struct ev_check cw; 1621 ev_io_stop (EV_A_ &once->io);
800 evw_init (&cw, gcb, 0); 1622 ev_timer_stop (EV_A_ &once->to);
801 evcheck_start (&cw); 1623 ev_free (once);
802 1624
803 struct ev_idle iw; 1625 cb (revents, arg);
804 evw_init (&iw, gcb, 0);
805 evidle_start (&iw);
806
807 ev_loop (0);
808
809 return 0;
810} 1626}
811 1627
812#endif 1628static void
1629once_cb_io (EV_P_ struct ev_io *w, int revents)
1630{
1631 once_cb (EV_A_ (struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents);
1632}
813 1633
1634static void
1635once_cb_to (EV_P_ struct ev_timer *w, int revents)
1636{
1637 once_cb (EV_A_ (struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents);
1638}
814 1639
1640void
1641ev_once (EV_P_ int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg)
1642{
1643 struct ev_once *once = (struct ev_once *)ev_malloc (sizeof (struct ev_once));
815 1644
1645 if (!once)
1646 cb (EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT, arg);
1647 else
1648 {
1649 once->cb = cb;
1650 once->arg = arg;
816 1651
1652 ev_init (&once->io, once_cb_io);
1653 if (fd >= 0)
1654 {
1655 ev_io_set (&once->io, fd, events);
1656 ev_io_start (EV_A_ &once->io);
1657 }
1658
1659 ev_init (&once->to, once_cb_to);
1660 if (timeout >= 0.)
1661 {
1662 ev_timer_set (&once->to, timeout, 0.);
1663 ev_timer_start (EV_A_ &once->to);
1664 }
1665 }
1666}
1667
1668#ifdef __cplusplus
1669}
1670#endif
1671

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines