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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines