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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines