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.127 by root, Sun Nov 18 02:17:57 2007 UTC

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines