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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines