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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines