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

Comparing libev/ev.c (file contents):
Revision 1.38 by root, Thu Nov 1 15:21:13 2007 UTC vs.
Revision 1.115 by root, Wed Nov 14 04:53:21 2007 UTC

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines