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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines