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