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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines