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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines