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

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines