]> git.llucax.com Git - software/libev.git/blob - ev.c
7c278f36c5c32256fd3be7a17d79f4190d91dc9a
[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 # ifdef EV_CONFIG_H
38 #  include EV_CONFIG_H
39 # else
40 #  include "config.h"
41 # endif
42
43 # if HAVE_CLOCK_GETTIME
44 #  ifndef EV_USE_MONOTONIC
45 #   define EV_USE_MONOTONIC 1
46 #  endif
47 #  ifndef EV_USE_REALTIME
48 #   define EV_USE_REALTIME  1
49 #  endif
50 # else
51 #  ifndef EV_USE_MONOTONIC
52 #   define EV_USE_MONOTONIC 0
53 #  endif
54 #  ifndef EV_USE_REALTIME
55 #   define EV_USE_REALTIME  0
56 #  endif
57 # endif
58
59 # ifndef EV_USE_SELECT
60 #  if HAVE_SELECT && HAVE_SYS_SELECT_H
61 #   define EV_USE_SELECT 1
62 #  else
63 #   define EV_USE_SELECT 0
64 #  endif
65 # endif
66
67 # ifndef EV_USE_POLL
68 #  if HAVE_POLL && HAVE_POLL_H
69 #   define EV_USE_POLL 1
70 #  else
71 #   define EV_USE_POLL 0
72 #  endif
73 # endif
74    
75 # ifndef EV_USE_EPOLL
76 #  if HAVE_EPOLL_CTL && HAVE_SYS_EPOLL_H
77 #   define EV_USE_EPOLL 1
78 #  else
79 #   define EV_USE_EPOLL 0
80 #  endif
81 # endif
82    
83 # ifndef EV_USE_KQUEUE
84 #  if HAVE_KQUEUE && HAVE_SYS_EVENT_H && HAVE_SYS_QUEUE_H
85 #   define EV_USE_KQUEUE 1
86 #  else
87 #   define EV_USE_KQUEUE 0
88 #  endif
89 # endif
90    
91 # ifndef EV_USE_PORT
92 #  if HAVE_PORT_H && HAVE_PORT_CREATE
93 #   define EV_USE_PORT 1
94 #  else
95 #   define EV_USE_PORT 0
96 #  endif
97 # endif
98
99 # ifndef EV_USE_INOTIFY
100 #  if HAVE_INOTIFY_INIT && HAVE_SYS_INOTIFY_H
101 #   define EV_USE_INOTIFY 1
102 #  else
103 #   define EV_USE_INOTIFY 0
104 #  endif
105 # endif
106
107 #endif
108
109 #include <math.h>
110 #include <stdlib.h>
111 #include <fcntl.h>
112 #include <stddef.h>
113
114 #include <stdio.h>
115
116 #include <assert.h>
117 #include <errno.h>
118 #include <sys/types.h>
119 #include <time.h>
120
121 #include <signal.h>
122
123 #ifdef EV_H
124 # include EV_H
125 #else
126 # include "ev.h"
127 #endif
128
129 #ifndef _WIN32
130 # include <sys/time.h>
131 # include <sys/wait.h>
132 # include <unistd.h>
133 #else
134 # define WIN32_LEAN_AND_MEAN
135 # include <windows.h>
136 # ifndef EV_SELECT_IS_WINSOCKET
137 #  define EV_SELECT_IS_WINSOCKET 1
138 # endif
139 #endif
140
141 /**/
142
143 #ifndef EV_USE_MONOTONIC
144 # define EV_USE_MONOTONIC 0
145 #endif
146
147 #ifndef EV_USE_REALTIME
148 # define EV_USE_REALTIME 0
149 #endif
150
151 #ifndef EV_USE_SELECT
152 # define EV_USE_SELECT 1
153 #endif
154
155 #ifndef EV_USE_POLL
156 # ifdef _WIN32
157 #  define EV_USE_POLL 0
158 # else
159 #  define EV_USE_POLL 1
160 # endif
161 #endif
162
163 #ifndef EV_USE_EPOLL
164 # define EV_USE_EPOLL 0
165 #endif
166
167 #ifndef EV_USE_KQUEUE
168 # define EV_USE_KQUEUE 0
169 #endif
170
171 #ifndef EV_USE_PORT
172 # define EV_USE_PORT 0
173 #endif
174
175 #ifndef EV_USE_INOTIFY
176 # define EV_USE_INOTIFY 0
177 #endif
178
179 #ifndef EV_PID_HASHSIZE
180 # if EV_MINIMAL
181 #  define EV_PID_HASHSIZE 1
182 # else
183 #  define EV_PID_HASHSIZE 16
184 # endif
185 #endif
186
187 #ifndef EV_INOTIFY_HASHSIZE
188 # if EV_MINIMAL
189 #  define EV_INOTIFY_HASHSIZE 1
190 # else
191 #  define EV_INOTIFY_HASHSIZE 16
192 # endif
193 #endif
194
195 /**/
196
197 #ifndef CLOCK_MONOTONIC
198 # undef EV_USE_MONOTONIC
199 # define EV_USE_MONOTONIC 0
200 #endif
201
202 #ifndef CLOCK_REALTIME
203 # undef EV_USE_REALTIME
204 # define EV_USE_REALTIME 0
205 #endif
206
207 #if EV_SELECT_IS_WINSOCKET
208 # include <winsock.h>
209 #endif
210
211 #if !EV_STAT_ENABLE
212 # define EV_USE_INOTIFY 0
213 #endif
214
215 #if EV_USE_INOTIFY
216 # include <sys/inotify.h>
217 #endif
218
219 /**/
220
221 #define MIN_TIMEJUMP  1. /* minimum timejump that gets detected (if monotonic clock available) */
222 #define MAX_BLOCKTIME 59.743 /* never wait longer than this time (to detect time jumps) */
223 /*#define CLEANUP_INTERVAL (MAX_BLOCKTIME * 5.) /* how often to try to free memory and re-check fds */
224
225 #if __GNUC__ >= 3
226 # define expect(expr,value)         __builtin_expect ((expr),(value))
227 # define noinline                   __attribute__ ((noinline))
228 #else
229 # define expect(expr,value)         (expr)
230 # define noinline
231 # if __STDC_VERSION__ < 199901L
232 #  define inline
233 # endif
234 #endif
235
236 #define expect_false(expr) expect ((expr) != 0, 0)
237 #define expect_true(expr)  expect ((expr) != 0, 1)
238 #define inline_size        static inline
239
240 #if EV_MINIMAL
241 # define inline_speed      static noinline
242 #else
243 # define inline_speed      static inline
244 #endif
245
246 #define NUMPRI    (EV_MAXPRI - EV_MINPRI + 1)
247 #define ABSPRI(w) (((W)w)->priority - EV_MINPRI)
248
249 #define EMPTY       /* required for microsofts broken pseudo-c compiler */
250 #define EMPTY2(a,b) /* used to suppress some warnings */
251
252 typedef ev_watcher *W;
253 typedef ev_watcher_list *WL;
254 typedef ev_watcher_time *WT;
255
256 static int have_monotonic; /* did clock_gettime (CLOCK_MONOTONIC) work? */
257
258 #ifdef _WIN32
259 # include "ev_win32.c"
260 #endif
261
262 /*****************************************************************************/
263
264 static void (*syserr_cb)(const char *msg);
265
266 void
267 ev_set_syserr_cb (void (*cb)(const char *msg))
268 {
269   syserr_cb = cb;
270 }
271
272 static void noinline
273 syserr (const char *msg)
274 {
275   if (!msg)
276     msg = "(libev) system error";
277
278   if (syserr_cb)
279     syserr_cb (msg);
280   else
281     {
282       perror (msg);
283       abort ();
284     }
285 }
286
287 static void *(*alloc)(void *ptr, long size);
288
289 void
290 ev_set_allocator (void *(*cb)(void *ptr, long size))
291 {
292   alloc = cb;
293 }
294
295 inline_speed void *
296 ev_realloc (void *ptr, long size)
297 {
298   ptr = alloc ? alloc (ptr, size) : realloc (ptr, size);
299
300   if (!ptr && size)
301     {
302       fprintf (stderr, "libev: cannot allocate %ld bytes, aborting.", size);
303       abort ();
304     }
305
306   return ptr;
307 }
308
309 #define ev_malloc(size) ev_realloc (0, (size))
310 #define ev_free(ptr)    ev_realloc ((ptr), 0)
311
312 /*****************************************************************************/
313
314 typedef struct
315 {
316   WL head;
317   unsigned char events;
318   unsigned char reify;
319 #if EV_SELECT_IS_WINSOCKET
320   SOCKET handle;
321 #endif
322 } ANFD;
323
324 typedef struct
325 {
326   W w;
327   int events;
328 } ANPENDING;
329
330 #if EV_USE_INOTIFY
331 typedef struct
332 {
333   WL head;
334 } ANFS;
335 #endif
336
337 #if EV_MULTIPLICITY
338
339   struct ev_loop
340   {
341     ev_tstamp ev_rt_now;
342     #define ev_rt_now ((loop)->ev_rt_now)
343     #define VAR(name,decl) decl;
344       #include "ev_vars.h"
345     #undef VAR
346   };
347   #include "ev_wrap.h"
348
349   static struct ev_loop default_loop_struct;
350   struct ev_loop *ev_default_loop_ptr;
351
352 #else
353
354   ev_tstamp ev_rt_now;
355   #define VAR(name,decl) static decl;
356     #include "ev_vars.h"
357   #undef VAR
358
359   static int ev_default_loop_ptr;
360
361 #endif
362
363 /*****************************************************************************/
364
365 ev_tstamp
366 ev_time (void)
367 {
368 #if EV_USE_REALTIME
369   struct timespec ts;
370   clock_gettime (CLOCK_REALTIME, &ts);
371   return ts.tv_sec + ts.tv_nsec * 1e-9;
372 #else
373   struct timeval tv;
374   gettimeofday (&tv, 0);
375   return tv.tv_sec + tv.tv_usec * 1e-6;
376 #endif
377 }
378
379 ev_tstamp inline_size
380 get_clock (void)
381 {
382 #if EV_USE_MONOTONIC
383   if (expect_true (have_monotonic))
384     {
385       struct timespec ts;
386       clock_gettime (CLOCK_MONOTONIC, &ts);
387       return ts.tv_sec + ts.tv_nsec * 1e-9;
388     }
389 #endif
390
391   return ev_time ();
392 }
393
394 #if EV_MULTIPLICITY
395 ev_tstamp
396 ev_now (EV_P)
397 {
398   return ev_rt_now;
399 }
400 #endif
401
402 int inline_size
403 array_nextsize (int elem, int cur, int cnt)
404 {
405   int ncur = cur + 1;
406
407   do
408     ncur <<= 1;
409   while (cnt > ncur);
410
411   /* if size > 4096, round to 4096 - 4 * longs to accomodate malloc overhead */
412   if (elem * ncur > 4096)
413     {
414       ncur *= elem;
415       ncur = (ncur + elem + 4095 + sizeof (void *) * 4) & ~4095;
416       ncur = ncur - sizeof (void *) * 4;
417       ncur /= elem;
418     }
419
420   return ncur;
421 }
422
423 static noinline void *
424 array_realloc (int elem, void *base, int *cur, int cnt)
425 {
426   *cur = array_nextsize (elem, *cur, cnt);
427   return ev_realloc (base, elem * *cur);
428 }
429
430 #define array_needsize(type,base,cur,cnt,init)                  \
431   if (expect_false ((cnt) > (cur)))                             \
432     {                                                           \
433       int ocur_ = (cur);                                        \
434       (base) = (type *)array_realloc                            \
435          (sizeof (type), (base), &(cur), (cnt));                \
436       init ((base) + (ocur_), (cur) - ocur_);                   \
437     }
438
439 #if 0
440 #define array_slim(type,stem)                                   \
441   if (stem ## max < array_roundsize (stem ## cnt >> 2))         \
442     {                                                           \
443       stem ## max = array_roundsize (stem ## cnt >> 1);         \
444       base = (type *)ev_realloc (base, sizeof (type) * (stem ## max));\
445       fprintf (stderr, "slimmed down " # stem " to %d\n", stem ## max);/*D*/\
446     }
447 #endif
448
449 #define array_free(stem, idx) \
450   ev_free (stem ## s idx); stem ## cnt idx = stem ## max idx = 0;
451
452 /*****************************************************************************/
453
454 void noinline
455 ev_feed_event (EV_P_ void *w, int revents)
456 {
457   W w_ = (W)w;
458   int pri = ABSPRI (w_);
459
460   if (expect_false (w_->pending))
461     pendings [pri][w_->pending - 1].events |= revents;
462   else
463     {
464       w_->pending = ++pendingcnt [pri];
465       array_needsize (ANPENDING, pendings [pri], pendingmax [pri], w_->pending, EMPTY2);
466       pendings [pri][w_->pending - 1].w      = w_;
467       pendings [pri][w_->pending - 1].events = revents;
468     }
469 }
470
471 void inline_size
472 queue_events (EV_P_ W *events, int eventcnt, int type)
473 {
474   int i;
475
476   for (i = 0; i < eventcnt; ++i)
477     ev_feed_event (EV_A_ events [i], type);
478 }
479
480 /*****************************************************************************/
481
482 void inline_size
483 anfds_init (ANFD *base, int count)
484 {
485   while (count--)
486     {
487       base->head   = 0;
488       base->events = EV_NONE;
489       base->reify  = 0;
490
491       ++base;
492     }
493 }
494
495 void inline_speed
496 fd_event (EV_P_ int fd, int revents)
497 {
498   ANFD *anfd = anfds + fd;
499   ev_io *w;
500
501   for (w = (ev_io *)anfd->head; w; w = (ev_io *)((WL)w)->next)
502     {
503       int ev = w->events & revents;
504
505       if (ev)
506         ev_feed_event (EV_A_ (W)w, ev);
507     }
508 }
509
510 void
511 ev_feed_fd_event (EV_P_ int fd, int revents)
512 {
513   if (fd >= 0 && fd < anfdmax)
514     fd_event (EV_A_ fd, revents);
515 }
516
517 void inline_size
518 fd_reify (EV_P)
519 {
520   int i;
521
522   for (i = 0; i < fdchangecnt; ++i)
523     {
524       int fd = fdchanges [i];
525       ANFD *anfd = anfds + fd;
526       ev_io *w;
527
528       int events = 0;
529
530       for (w = (ev_io *)anfd->head; w; w = (ev_io *)((WL)w)->next)
531         events |= w->events;
532
533 #if EV_SELECT_IS_WINSOCKET
534       if (events)
535         {
536           unsigned long argp;
537           anfd->handle = _get_osfhandle (fd);
538           assert (("libev only supports socket fds in this configuration", ioctlsocket (anfd->handle, FIONREAD, &argp) == 0));
539         }
540 #endif
541
542       anfd->reify = 0;
543
544       backend_modify (EV_A_ fd, anfd->events, events);
545       anfd->events = events;
546     }
547
548   fdchangecnt = 0;
549 }
550
551 void inline_size
552 fd_change (EV_P_ int fd)
553 {
554   if (expect_false (anfds [fd].reify))
555     return;
556
557   anfds [fd].reify = 1;
558
559   ++fdchangecnt;
560   array_needsize (int, fdchanges, fdchangemax, fdchangecnt, EMPTY2);
561   fdchanges [fdchangecnt - 1] = fd;
562 }
563
564 void inline_speed
565 fd_kill (EV_P_ int fd)
566 {
567   ev_io *w;
568
569   while ((w = (ev_io *)anfds [fd].head))
570     {
571       ev_io_stop (EV_A_ w);
572       ev_feed_event (EV_A_ (W)w, EV_ERROR | EV_READ | EV_WRITE);
573     }
574 }
575
576 int inline_size
577 fd_valid (int fd)
578 {
579 #ifdef _WIN32
580   return _get_osfhandle (fd) != -1;
581 #else
582   return fcntl (fd, F_GETFD) != -1;
583 #endif
584 }
585
586 /* called on EBADF to verify fds */
587 static void noinline
588 fd_ebadf (EV_P)
589 {
590   int fd;
591
592   for (fd = 0; fd < anfdmax; ++fd)
593     if (anfds [fd].events)
594       if (!fd_valid (fd) == -1 && errno == EBADF)
595         fd_kill (EV_A_ fd);
596 }
597
598 /* called on ENOMEM in select/poll to kill some fds and retry */
599 static void noinline
600 fd_enomem (EV_P)
601 {
602   int fd;
603
604   for (fd = anfdmax; fd--; )
605     if (anfds [fd].events)
606       {
607         fd_kill (EV_A_ fd);
608         return;
609       }
610 }
611
612 /* usually called after fork if backend needs to re-arm all fds from scratch */
613 static void noinline
614 fd_rearm_all (EV_P)
615 {
616   int fd;
617
618   for (fd = 0; fd < anfdmax; ++fd)
619     if (anfds [fd].events)
620       {
621         anfds [fd].events = 0;
622         fd_change (EV_A_ fd);
623       }
624 }
625
626 /*****************************************************************************/
627
628 void inline_speed
629 upheap (WT *heap, int k)
630 {
631   WT w = heap [k];
632
633   while (k && heap [k >> 1]->at > w->at)
634     {
635       heap [k] = heap [k >> 1];
636       ((W)heap [k])->active = k + 1;
637       k >>= 1;
638     }
639
640   heap [k] = w;
641   ((W)heap [k])->active = k + 1;
642
643 }
644
645 void inline_speed
646 downheap (WT *heap, int N, int k)
647 {
648   WT w = heap [k];
649
650   while (k < (N >> 1))
651     {
652       int j = k << 1;
653
654       if (j + 1 < N && heap [j]->at > heap [j + 1]->at)
655         ++j;
656
657       if (w->at <= heap [j]->at)
658         break;
659
660       heap [k] = heap [j];
661       ((W)heap [k])->active = k + 1;
662       k = j;
663     }
664
665   heap [k] = w;
666   ((W)heap [k])->active = k + 1;
667 }
668
669 void inline_size
670 adjustheap (WT *heap, int N, int k)
671 {
672   upheap (heap, k);
673   downheap (heap, N, k);
674 }
675
676 /*****************************************************************************/
677
678 typedef struct
679 {
680   WL head;
681   sig_atomic_t volatile gotsig;
682 } ANSIG;
683
684 static ANSIG *signals;
685 static int signalmax;
686
687 static int sigpipe [2];
688 static sig_atomic_t volatile gotsig;
689 static ev_io sigev;
690
691 void inline_size
692 signals_init (ANSIG *base, int count)
693 {
694   while (count--)
695     {
696       base->head   = 0;
697       base->gotsig = 0;
698
699       ++base;
700     }
701 }
702
703 static void
704 sighandler (int signum)
705 {
706 #if _WIN32
707   signal (signum, sighandler);
708 #endif
709
710   signals [signum - 1].gotsig = 1;
711
712   if (!gotsig)
713     {
714       int old_errno = errno;
715       gotsig = 1;
716       write (sigpipe [1], &signum, 1);
717       errno = old_errno;
718     }
719 }
720
721 void noinline
722 ev_feed_signal_event (EV_P_ int signum)
723 {
724   WL w;
725
726 #if EV_MULTIPLICITY
727   assert (("feeding signal events is only supported in the default loop", loop == ev_default_loop_ptr));
728 #endif
729
730   --signum;
731
732   if (signum < 0 || signum >= signalmax)
733     return;
734
735   signals [signum].gotsig = 0;
736
737   for (w = signals [signum].head; w; w = w->next)
738     ev_feed_event (EV_A_ (W)w, EV_SIGNAL);
739 }
740
741 static void
742 sigcb (EV_P_ ev_io *iow, int revents)
743 {
744   int signum;
745
746   read (sigpipe [0], &revents, 1);
747   gotsig = 0;
748
749   for (signum = signalmax; signum--; )
750     if (signals [signum].gotsig)
751       ev_feed_signal_event (EV_A_ signum + 1);
752 }
753
754 void inline_speed
755 fd_intern (int fd)
756 {
757 #ifdef _WIN32
758   int arg = 1;
759   ioctlsocket (_get_osfhandle (fd), FIONBIO, &arg);
760 #else
761   fcntl (fd, F_SETFD, FD_CLOEXEC);
762   fcntl (fd, F_SETFL, O_NONBLOCK);
763 #endif
764 }
765
766 static void noinline
767 siginit (EV_P)
768 {
769   fd_intern (sigpipe [0]);
770   fd_intern (sigpipe [1]);
771
772   ev_io_set (&sigev, sigpipe [0], EV_READ);
773   ev_io_start (EV_A_ &sigev);
774   ev_unref (EV_A); /* child watcher should not keep loop alive */
775 }
776
777 /*****************************************************************************/
778
779 static ev_child *childs [EV_PID_HASHSIZE];
780
781 #ifndef _WIN32
782
783 static ev_signal childev;
784
785 void inline_speed
786 child_reap (EV_P_ ev_signal *sw, int chain, int pid, int status)
787 {
788   ev_child *w;
789
790   for (w = (ev_child *)childs [chain & (EV_PID_HASHSIZE - 1)]; w; w = (ev_child *)((WL)w)->next)
791     if (w->pid == pid || !w->pid)
792       {
793         ev_set_priority (w, ev_priority (sw)); /* need to do it *now* */
794         w->rpid    = pid;
795         w->rstatus = status;
796         ev_feed_event (EV_A_ (W)w, EV_CHILD);
797       }
798 }
799
800 #ifndef WCONTINUED
801 # define WCONTINUED 0
802 #endif
803
804 static void
805 childcb (EV_P_ ev_signal *sw, int revents)
806 {
807   int pid, status;
808
809   /* some systems define WCONTINUED but then fail to support it (linux 2.4) */
810   if (0 >= (pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)))
811     if (!WCONTINUED
812         || errno != EINVAL
813         || 0 >= (pid = waitpid (-1, &status, WNOHANG | WUNTRACED)))
814       return;
815
816   /* make sure we are called again until all childs have been reaped */
817   /* we need to do it this way so that the callback gets called before we continue */
818   ev_feed_event (EV_A_ (W)sw, EV_SIGNAL);
819
820   child_reap (EV_A_ sw, pid, pid, status);
821   if (EV_PID_HASHSIZE > 1)
822     child_reap (EV_A_ sw, 0, pid, status); /* this might trigger a watcher twice, but feed_event catches that */
823 }
824
825 #endif
826
827 /*****************************************************************************/
828
829 #if EV_USE_PORT
830 # include "ev_port.c"
831 #endif
832 #if EV_USE_KQUEUE
833 # include "ev_kqueue.c"
834 #endif
835 #if EV_USE_EPOLL
836 # include "ev_epoll.c"
837 #endif
838 #if EV_USE_POLL
839 # include "ev_poll.c"
840 #endif
841 #if EV_USE_SELECT
842 # include "ev_select.c"
843 #endif
844
845 int
846 ev_version_major (void)
847 {
848   return EV_VERSION_MAJOR;
849 }
850
851 int
852 ev_version_minor (void)
853 {
854   return EV_VERSION_MINOR;
855 }
856
857 /* return true if we are running with elevated privileges and should ignore env variables */
858 int inline_size
859 enable_secure (void)
860 {
861 #ifdef _WIN32
862   return 0;
863 #else
864   return getuid () != geteuid ()
865       || getgid () != getegid ();
866 #endif
867 }
868
869 unsigned int
870 ev_supported_backends (void)
871 {
872   unsigned int flags = 0;
873
874   if (EV_USE_PORT  ) flags |= EVBACKEND_PORT;
875   if (EV_USE_KQUEUE) flags |= EVBACKEND_KQUEUE;
876   if (EV_USE_EPOLL ) flags |= EVBACKEND_EPOLL;
877   if (EV_USE_POLL  ) flags |= EVBACKEND_POLL;
878   if (EV_USE_SELECT) flags |= EVBACKEND_SELECT;
879   
880   return flags;
881 }
882
883 unsigned int
884 ev_recommended_backends (void)
885 {
886   unsigned int flags = ev_supported_backends ();
887
888 #ifndef __NetBSD__
889   /* kqueue is borked on everything but netbsd apparently */
890   /* it usually doesn't work correctly on anything but sockets and pipes */
891   flags &= ~EVBACKEND_KQUEUE;
892 #endif
893 #ifdef __APPLE__
894   // flags &= ~EVBACKEND_KQUEUE; for documentation
895   flags &= ~EVBACKEND_POLL;
896 #endif
897
898   return flags;
899 }
900
901 unsigned int
902 ev_embeddable_backends (void)
903 {
904   return EVBACKEND_EPOLL
905        | EVBACKEND_KQUEUE
906        | EVBACKEND_PORT;
907 }
908
909 unsigned int
910 ev_backend (EV_P)
911 {
912   return backend;
913 }
914
915 unsigned int
916 ev_loop_count (EV_P)
917 {
918   return loop_count;
919 }
920
921 static void noinline
922 loop_init (EV_P_ unsigned int flags)
923 {
924   if (!backend)
925     {
926 #if EV_USE_MONOTONIC
927       {
928         struct timespec ts;
929         if (!clock_gettime (CLOCK_MONOTONIC, &ts))
930           have_monotonic = 1;
931       }
932 #endif
933
934       ev_rt_now = ev_time ();
935       mn_now    = get_clock ();
936       now_floor = mn_now;
937       rtmn_diff = ev_rt_now - mn_now;
938
939       /* pid check not overridable via env */
940 #ifndef _WIN32
941       if (flags & EVFLAG_FORKCHECK)
942         curpid = getpid ();
943 #endif
944
945       if (!(flags & EVFLAG_NOENV)
946           && !enable_secure ()
947           && getenv ("LIBEV_FLAGS"))
948         flags = atoi (getenv ("LIBEV_FLAGS"));
949
950       if (!(flags & 0x0000ffffUL))
951         flags |= ev_recommended_backends ();
952
953       backend = 0;
954       backend_fd = -1;
955 #if EV_USE_INOTIFY
956       fs_fd = -2;
957 #endif
958
959 #if EV_USE_PORT
960       if (!backend && (flags & EVBACKEND_PORT  )) backend = port_init   (EV_A_ flags);
961 #endif
962 #if EV_USE_KQUEUE
963       if (!backend && (flags & EVBACKEND_KQUEUE)) backend = kqueue_init (EV_A_ flags);
964 #endif
965 #if EV_USE_EPOLL
966       if (!backend && (flags & EVBACKEND_EPOLL )) backend = epoll_init  (EV_A_ flags);
967 #endif
968 #if EV_USE_POLL
969       if (!backend && (flags & EVBACKEND_POLL  )) backend = poll_init   (EV_A_ flags);
970 #endif
971 #if EV_USE_SELECT
972       if (!backend && (flags & EVBACKEND_SELECT)) backend = select_init (EV_A_ flags);
973 #endif
974
975       ev_init (&sigev, sigcb);
976       ev_set_priority (&sigev, EV_MAXPRI);
977     }
978 }
979
980 static void noinline
981 loop_destroy (EV_P)
982 {
983   int i;
984
985 #if EV_USE_INOTIFY
986   if (fs_fd >= 0)
987     close (fs_fd);
988 #endif
989
990   if (backend_fd >= 0)
991     close (backend_fd);
992
993 #if EV_USE_PORT
994   if (backend == EVBACKEND_PORT  ) port_destroy   (EV_A);
995 #endif
996 #if EV_USE_KQUEUE
997   if (backend == EVBACKEND_KQUEUE) kqueue_destroy (EV_A);
998 #endif
999 #if EV_USE_EPOLL
1000   if (backend == EVBACKEND_EPOLL ) epoll_destroy  (EV_A);
1001 #endif
1002 #if EV_USE_POLL
1003   if (backend == EVBACKEND_POLL  ) poll_destroy   (EV_A);
1004 #endif
1005 #if EV_USE_SELECT
1006   if (backend == EVBACKEND_SELECT) select_destroy (EV_A);
1007 #endif
1008
1009   for (i = NUMPRI; i--; )
1010     {
1011       array_free (pending, [i]);
1012 #if EV_IDLE_ENABLE
1013       array_free (idle, [i]);
1014 #endif
1015     }
1016
1017   /* have to use the microsoft-never-gets-it-right macro */
1018   array_free (fdchange, EMPTY);
1019   array_free (timer, EMPTY);
1020 #if EV_PERIODIC_ENABLE
1021   array_free (periodic, EMPTY);
1022 #endif
1023   array_free (prepare, EMPTY);
1024   array_free (check, EMPTY);
1025
1026   backend = 0;
1027 }
1028
1029 void inline_size infy_fork (EV_P);
1030
1031 void inline_size
1032 loop_fork (EV_P)
1033 {
1034 #if EV_USE_PORT
1035   if (backend == EVBACKEND_PORT  ) port_fork   (EV_A);
1036 #endif
1037 #if EV_USE_KQUEUE
1038   if (backend == EVBACKEND_KQUEUE) kqueue_fork (EV_A);
1039 #endif
1040 #if EV_USE_EPOLL
1041   if (backend == EVBACKEND_EPOLL ) epoll_fork  (EV_A);
1042 #endif
1043 #if EV_USE_INOTIFY
1044   infy_fork (EV_A);
1045 #endif
1046
1047   if (ev_is_active (&sigev))
1048     {
1049       /* default loop */
1050
1051       ev_ref (EV_A);
1052       ev_io_stop (EV_A_ &sigev);
1053       close (sigpipe [0]);
1054       close (sigpipe [1]);
1055
1056       while (pipe (sigpipe))
1057         syserr ("(libev) error creating pipe");
1058
1059       siginit (EV_A);
1060     }
1061
1062   postfork = 0;
1063 }
1064
1065 #if EV_MULTIPLICITY
1066 struct ev_loop *
1067 ev_loop_new (unsigned int flags)
1068 {
1069   struct ev_loop *loop = (struct ev_loop *)ev_malloc (sizeof (struct ev_loop));
1070
1071   memset (loop, 0, sizeof (struct ev_loop));
1072
1073   loop_init (EV_A_ flags);
1074
1075   if (ev_backend (EV_A))
1076     return loop;
1077
1078   return 0;
1079 }
1080
1081 void
1082 ev_loop_destroy (EV_P)
1083 {
1084   loop_destroy (EV_A);
1085   ev_free (loop);
1086 }
1087
1088 void
1089 ev_loop_fork (EV_P)
1090 {
1091   postfork = 1;
1092 }
1093
1094 #endif
1095
1096 #if EV_MULTIPLICITY
1097 struct ev_loop *
1098 ev_default_loop_init (unsigned int flags)
1099 #else
1100 int
1101 ev_default_loop (unsigned int flags)
1102 #endif
1103 {
1104   if (sigpipe [0] == sigpipe [1])
1105     if (pipe (sigpipe))
1106       return 0;
1107
1108   if (!ev_default_loop_ptr)
1109     {
1110 #if EV_MULTIPLICITY
1111       struct ev_loop *loop = ev_default_loop_ptr = &default_loop_struct;
1112 #else
1113       ev_default_loop_ptr = 1;
1114 #endif
1115
1116       loop_init (EV_A_ flags);
1117
1118       if (ev_backend (EV_A))
1119         {
1120           siginit (EV_A);
1121
1122 #ifndef _WIN32
1123           ev_signal_init (&childev, childcb, SIGCHLD);
1124           ev_set_priority (&childev, EV_MAXPRI);
1125           ev_signal_start (EV_A_ &childev);
1126           ev_unref (EV_A); /* child watcher should not keep loop alive */
1127 #endif
1128         }
1129       else
1130         ev_default_loop_ptr = 0;
1131     }
1132
1133   return ev_default_loop_ptr;
1134 }
1135
1136 void
1137 ev_default_destroy (void)
1138 {
1139 #if EV_MULTIPLICITY
1140   struct ev_loop *loop = ev_default_loop_ptr;
1141 #endif
1142
1143 #ifndef _WIN32
1144   ev_ref (EV_A); /* child watcher */
1145   ev_signal_stop (EV_A_ &childev);
1146 #endif
1147
1148   ev_ref (EV_A); /* signal watcher */
1149   ev_io_stop (EV_A_ &sigev);
1150
1151   close (sigpipe [0]); sigpipe [0] = 0;
1152   close (sigpipe [1]); sigpipe [1] = 0;
1153
1154   loop_destroy (EV_A);
1155 }
1156
1157 void
1158 ev_default_fork (void)
1159 {
1160 #if EV_MULTIPLICITY
1161   struct ev_loop *loop = ev_default_loop_ptr;
1162 #endif
1163
1164   if (backend)
1165     postfork = 1;
1166 }
1167
1168 /*****************************************************************************/
1169
1170 void
1171 ev_invoke (EV_P_ void *w, int revents)
1172 {
1173   EV_CB_INVOKE ((W)w, revents);
1174 }
1175
1176 void inline_speed
1177 call_pending (EV_P)
1178 {
1179   int pri;
1180
1181   for (pri = NUMPRI; pri--; )
1182     while (pendingcnt [pri])
1183       {
1184         ANPENDING *p = pendings [pri] + --pendingcnt [pri];
1185
1186         if (expect_true (p->w))
1187           {
1188             /*assert (("non-pending watcher on pending list", p->w->pending));*/
1189
1190             p->w->pending = 0;
1191             EV_CB_INVOKE (p->w, p->events);
1192           }
1193       }
1194 }
1195
1196 void inline_size
1197 timers_reify (EV_P)
1198 {
1199   while (timercnt && ((WT)timers [0])->at <= mn_now)
1200     {
1201       ev_timer *w = timers [0];
1202
1203       /*assert (("inactive timer on timer heap detected", ev_is_active (w)));*/
1204
1205       /* first reschedule or stop timer */
1206       if (w->repeat)
1207         {
1208           assert (("negative ev_timer repeat value found while processing timers", w->repeat > 0.));
1209
1210           ((WT)w)->at += w->repeat;
1211           if (((WT)w)->at < mn_now)
1212             ((WT)w)->at = mn_now;
1213
1214           downheap ((WT *)timers, timercnt, 0);
1215         }
1216       else
1217         ev_timer_stop (EV_A_ w); /* nonrepeating: stop timer */
1218
1219       ev_feed_event (EV_A_ (W)w, EV_TIMEOUT);
1220     }
1221 }
1222
1223 #if EV_PERIODIC_ENABLE
1224 void inline_size
1225 periodics_reify (EV_P)
1226 {
1227   while (periodiccnt && ((WT)periodics [0])->at <= ev_rt_now)
1228     {
1229       ev_periodic *w = periodics [0];
1230
1231       /*assert (("inactive timer on periodic heap detected", ev_is_active (w)));*/
1232
1233       /* first reschedule or stop timer */
1234       if (w->reschedule_cb)
1235         {
1236           ((WT)w)->at = w->reschedule_cb (w, ev_rt_now + 0.0001220703125 /* 1/8192 */);
1237           assert (("ev_periodic reschedule callback returned time in the past", ((WT)w)->at > ev_rt_now));
1238           downheap ((WT *)periodics, periodiccnt, 0);
1239         }
1240       else if (w->interval)
1241         {
1242           ((WT)w)->at = w->offset + (floor ((ev_rt_now - w->offset) / w->interval) + 1.) * w->interval;
1243           assert (("ev_periodic timeout in the past detected while processing timers, negative interval?", ((WT)w)->at > ev_rt_now));
1244           downheap ((WT *)periodics, periodiccnt, 0);
1245         }
1246       else
1247         ev_periodic_stop (EV_A_ w); /* nonrepeating: stop timer */
1248
1249       ev_feed_event (EV_A_ (W)w, EV_PERIODIC);
1250     }
1251 }
1252
1253 static void noinline
1254 periodics_reschedule (EV_P)
1255 {
1256   int i;
1257
1258   /* adjust periodics after time jump */
1259   for (i = 0; i < periodiccnt; ++i)
1260     {
1261       ev_periodic *w = periodics [i];
1262
1263       if (w->reschedule_cb)
1264         ((WT)w)->at = w->reschedule_cb (w, ev_rt_now);
1265       else if (w->interval)
1266         ((WT)w)->at = w->offset + ceil ((ev_rt_now - w->offset) / w->interval) * w->interval;
1267     }
1268
1269   /* now rebuild the heap */
1270   for (i = periodiccnt >> 1; i--; )
1271     downheap ((WT *)periodics, periodiccnt, i);
1272 }
1273 #endif
1274
1275 #if EV_IDLE_ENABLE
1276 void inline_size
1277 idle_reify (EV_P)
1278 {
1279   if (expect_false (idleall))
1280     {
1281       int pri;
1282
1283       for (pri = NUMPRI; pri--; )
1284         {
1285           if (pendingcnt [pri])
1286             break;
1287
1288           if (idlecnt [pri])
1289             {
1290               queue_events (EV_A_ (W *)idles [pri], idlecnt [pri], EV_IDLE);
1291               break;
1292             }
1293         }
1294     }
1295 }
1296 #endif
1297
1298 int inline_size
1299 time_update_monotonic (EV_P)
1300 {
1301   mn_now = get_clock ();
1302
1303   if (expect_true (mn_now - now_floor < MIN_TIMEJUMP * .5))
1304     {
1305       ev_rt_now = rtmn_diff + mn_now;
1306       return 0;
1307     }
1308   else
1309     {
1310       now_floor = mn_now;
1311       ev_rt_now = ev_time ();
1312       return 1;
1313     }
1314 }
1315
1316 void inline_size
1317 time_update (EV_P)
1318 {
1319   int i;
1320
1321 #if EV_USE_MONOTONIC
1322   if (expect_true (have_monotonic))
1323     {
1324       if (time_update_monotonic (EV_A))
1325         {
1326           ev_tstamp odiff = rtmn_diff;
1327
1328           /* loop a few times, before making important decisions.
1329            * on the choice of "4": one iteration isn't enough,
1330            * in case we get preempted during the calls to
1331            * ev_time and get_clock. a second call is almost guaranteed
1332            * to succeed in that case, though. and looping a few more times
1333            * doesn't hurt either as we only do this on time-jumps or
1334            * in the unlikely event of having been preempted here.
1335            */
1336           for (i = 4; --i; )
1337             {
1338               rtmn_diff = ev_rt_now - mn_now;
1339
1340               if (fabs (odiff - rtmn_diff) < MIN_TIMEJUMP)
1341                 return; /* all is well */
1342
1343               ev_rt_now = ev_time ();
1344               mn_now    = get_clock ();
1345               now_floor = mn_now;
1346             }
1347
1348 # if EV_PERIODIC_ENABLE
1349           periodics_reschedule (EV_A);
1350 # endif
1351           /* no timer adjustment, as the monotonic clock doesn't jump */
1352           /* timers_reschedule (EV_A_ rtmn_diff - odiff) */
1353         }
1354     }
1355   else
1356 #endif
1357     {
1358       ev_rt_now = ev_time ();
1359
1360       if (expect_false (mn_now > ev_rt_now || mn_now < ev_rt_now - MAX_BLOCKTIME - MIN_TIMEJUMP))
1361         {
1362 #if EV_PERIODIC_ENABLE
1363           periodics_reschedule (EV_A);
1364 #endif
1365
1366           /* adjust timers. this is easy, as the offset is the same for all of them */
1367           for (i = 0; i < timercnt; ++i)
1368             ((WT)timers [i])->at += ev_rt_now - mn_now;
1369         }
1370
1371       mn_now = ev_rt_now;
1372     }
1373 }
1374
1375 void
1376 ev_ref (EV_P)
1377 {
1378   ++activecnt;
1379 }
1380
1381 void
1382 ev_unref (EV_P)
1383 {
1384   --activecnt;
1385 }
1386
1387 static int loop_done;
1388
1389 void
1390 ev_loop (EV_P_ int flags)
1391 {
1392   loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK)
1393             ? EVUNLOOP_ONE
1394             : EVUNLOOP_CANCEL;
1395
1396   call_pending (EV_A); /* in case we recurse, ensure ordering stays nice and clean */
1397
1398   do
1399     {
1400 #ifndef _WIN32
1401       if (expect_false (curpid)) /* penalise the forking check even more */
1402         if (expect_false (getpid () != curpid))
1403           {
1404             curpid = getpid ();
1405             postfork = 1;
1406           }
1407 #endif
1408
1409 #if EV_FORK_ENABLE
1410       /* we might have forked, so queue fork handlers */
1411       if (expect_false (postfork))
1412         if (forkcnt)
1413           {
1414             queue_events (EV_A_ (W *)forks, forkcnt, EV_FORK);
1415             call_pending (EV_A);
1416           }
1417 #endif
1418
1419       /* queue prepare watchers (and execute them) */
1420       if (expect_false (preparecnt))
1421         {
1422           queue_events (EV_A_ (W *)prepares, preparecnt, EV_PREPARE);
1423           call_pending (EV_A);
1424         }
1425
1426       if (expect_false (!activecnt))
1427         break;
1428
1429       /* we might have forked, so reify kernel state if necessary */
1430       if (expect_false (postfork))
1431         loop_fork (EV_A);
1432
1433       /* update fd-related kernel structures */
1434       fd_reify (EV_A);
1435
1436       /* calculate blocking time */
1437       {
1438         ev_tstamp block;
1439
1440         if (expect_false (flags & EVLOOP_NONBLOCK || idleall || !activecnt))
1441           block = 0.; /* do not block at all */
1442         else
1443           {
1444             /* update time to cancel out callback processing overhead */
1445 #if EV_USE_MONOTONIC
1446             if (expect_true (have_monotonic))
1447               time_update_monotonic (EV_A);
1448             else
1449 #endif
1450               {
1451                 ev_rt_now = ev_time ();
1452                 mn_now    = ev_rt_now;
1453               }
1454
1455             block = MAX_BLOCKTIME;
1456
1457             if (timercnt)
1458               {
1459                 ev_tstamp to = ((WT)timers [0])->at - mn_now + backend_fudge;
1460                 if (block > to) block = to;
1461               }
1462
1463 #if EV_PERIODIC_ENABLE
1464             if (periodiccnt)
1465               {
1466                 ev_tstamp to = ((WT)periodics [0])->at - ev_rt_now + backend_fudge;
1467                 if (block > to) block = to;
1468               }
1469 #endif
1470
1471             if (expect_false (block < 0.)) block = 0.;
1472           }
1473
1474         ++loop_count;
1475         backend_poll (EV_A_ block);
1476       }
1477
1478       /* update ev_rt_now, do magic */
1479       time_update (EV_A);
1480
1481       /* queue pending timers and reschedule them */
1482       timers_reify (EV_A); /* relative timers called last */
1483 #if EV_PERIODIC_ENABLE
1484       periodics_reify (EV_A); /* absolute timers called first */
1485 #endif
1486
1487 #if EV_IDLE_ENABLE
1488       /* queue idle watchers unless other events are pending */
1489       idle_reify (EV_A);
1490 #endif
1491
1492       /* queue check watchers, to be executed first */
1493       if (expect_false (checkcnt))
1494         queue_events (EV_A_ (W *)checks, checkcnt, EV_CHECK);
1495
1496       call_pending (EV_A);
1497
1498     }
1499   while (expect_true (activecnt && !loop_done));
1500
1501   if (loop_done == EVUNLOOP_ONE)
1502     loop_done = EVUNLOOP_CANCEL;
1503 }
1504
1505 void
1506 ev_unloop (EV_P_ int how)
1507 {
1508   loop_done = how;
1509 }
1510
1511 /*****************************************************************************/
1512
1513 void inline_size
1514 wlist_add (WL *head, WL elem)
1515 {
1516   elem->next = *head;
1517   *head = elem;
1518 }
1519
1520 void inline_size
1521 wlist_del (WL *head, WL elem)
1522 {
1523   while (*head)
1524     {
1525       if (*head == elem)
1526         {
1527           *head = elem->next;
1528           return;
1529         }
1530
1531       head = &(*head)->next;
1532     }
1533 }
1534
1535 void inline_speed
1536 clear_pending (EV_P_ W w)
1537 {
1538   if (w->pending)
1539     {
1540       pendings [ABSPRI (w)][w->pending - 1].w = 0;
1541       w->pending = 0;
1542     }
1543 }
1544
1545 int
1546 ev_clear_pending (EV_P_ void *w)
1547 {
1548   W w_ = (W)w;
1549   int pending = w_->pending;
1550
1551   if (expect_true (pending))
1552     {
1553       ANPENDING *p = pendings [ABSPRI (w_)] + pending - 1;
1554       w_->pending = 0;
1555       p->w = 0;
1556       return p->events;
1557     }
1558   else
1559     return 0;
1560 }
1561
1562 void inline_size
1563 pri_adjust (EV_P_ W w)
1564 {
1565   int pri = w->priority;
1566   pri = pri < EV_MINPRI ? EV_MINPRI : pri;
1567   pri = pri > EV_MAXPRI ? EV_MAXPRI : pri;
1568   w->priority = pri;
1569 }
1570
1571 void inline_speed
1572 ev_start (EV_P_ W w, int active)
1573 {
1574   pri_adjust (EV_A_ w);
1575   w->active = active;
1576   ev_ref (EV_A);
1577 }
1578
1579 void inline_size
1580 ev_stop (EV_P_ W w)
1581 {
1582   ev_unref (EV_A);
1583   w->active = 0;
1584 }
1585
1586 /*****************************************************************************/
1587
1588 void noinline
1589 ev_io_start (EV_P_ ev_io *w)
1590 {
1591   int fd = w->fd;
1592
1593   if (expect_false (ev_is_active (w)))
1594     return;
1595
1596   assert (("ev_io_start called with negative fd", fd >= 0));
1597
1598   ev_start (EV_A_ (W)w, 1);
1599   array_needsize (ANFD, anfds, anfdmax, fd + 1, anfds_init);
1600   wlist_add ((WL *)&anfds[fd].head, (WL)w);
1601
1602   fd_change (EV_A_ fd);
1603 }
1604
1605 void noinline
1606 ev_io_stop (EV_P_ ev_io *w)
1607 {
1608   clear_pending (EV_A_ (W)w);
1609   if (expect_false (!ev_is_active (w)))
1610     return;
1611
1612   assert (("ev_io_start called with illegal fd (must stay constant after start!)", w->fd >= 0 && w->fd < anfdmax));
1613
1614   wlist_del ((WL *)&anfds[w->fd].head, (WL)w);
1615   ev_stop (EV_A_ (W)w);
1616
1617   fd_change (EV_A_ w->fd);
1618 }
1619
1620 void noinline
1621 ev_timer_start (EV_P_ ev_timer *w)
1622 {
1623   if (expect_false (ev_is_active (w)))
1624     return;
1625
1626   ((WT)w)->at += mn_now;
1627
1628   assert (("ev_timer_start called with negative timer repeat value", w->repeat >= 0.));
1629
1630   ev_start (EV_A_ (W)w, ++timercnt);
1631   array_needsize (ev_timer *, timers, timermax, timercnt, EMPTY2);
1632   timers [timercnt - 1] = w;
1633   upheap ((WT *)timers, timercnt - 1);
1634
1635   /*assert (("internal timer heap corruption", timers [((W)w)->active - 1] == w));*/
1636 }
1637
1638 void noinline
1639 ev_timer_stop (EV_P_ ev_timer *w)
1640 {
1641   clear_pending (EV_A_ (W)w);
1642   if (expect_false (!ev_is_active (w)))
1643     return;
1644
1645   assert (("internal timer heap corruption", timers [((W)w)->active - 1] == w));
1646
1647   {
1648     int active = ((W)w)->active;
1649
1650     if (expect_true (--active < --timercnt))
1651       {
1652         timers [active] = timers [timercnt];
1653         adjustheap ((WT *)timers, timercnt, active);
1654       }
1655   }
1656
1657   ((WT)w)->at -= mn_now;
1658
1659   ev_stop (EV_A_ (W)w);
1660 }
1661
1662 void noinline
1663 ev_timer_again (EV_P_ ev_timer *w)
1664 {
1665   if (ev_is_active (w))
1666     {
1667       if (w->repeat)
1668         {
1669           ((WT)w)->at = mn_now + w->repeat;
1670           adjustheap ((WT *)timers, timercnt, ((W)w)->active - 1);
1671         }
1672       else
1673         ev_timer_stop (EV_A_ w);
1674     }
1675   else if (w->repeat)
1676     {
1677       w->at = w->repeat;
1678       ev_timer_start (EV_A_ w);
1679     }
1680 }
1681
1682 #if EV_PERIODIC_ENABLE
1683 void noinline
1684 ev_periodic_start (EV_P_ ev_periodic *w)
1685 {
1686   if (expect_false (ev_is_active (w)))
1687     return;
1688
1689   if (w->reschedule_cb)
1690     ((WT)w)->at = w->reschedule_cb (w, ev_rt_now);
1691   else if (w->interval)
1692     {
1693       assert (("ev_periodic_start called with negative interval value", w->interval >= 0.));
1694       /* this formula differs from the one in periodic_reify because we do not always round up */
1695       ((WT)w)->at = w->offset + ceil ((ev_rt_now - w->offset) / w->interval) * w->interval;
1696     }
1697   else
1698     ((WT)w)->at = w->offset;
1699
1700   ev_start (EV_A_ (W)w, ++periodiccnt);
1701   array_needsize (ev_periodic *, periodics, periodicmax, periodiccnt, EMPTY2);
1702   periodics [periodiccnt - 1] = w;
1703   upheap ((WT *)periodics, periodiccnt - 1);
1704
1705   /*assert (("internal periodic heap corruption", periodics [((W)w)->active - 1] == w));*/
1706 }
1707
1708 void noinline
1709 ev_periodic_stop (EV_P_ ev_periodic *w)
1710 {
1711   clear_pending (EV_A_ (W)w);
1712   if (expect_false (!ev_is_active (w)))
1713     return;
1714
1715   assert (("internal periodic heap corruption", periodics [((W)w)->active - 1] == w));
1716
1717   {
1718     int active = ((W)w)->active;
1719
1720     if (expect_true (--active < --periodiccnt))
1721       {
1722         periodics [active] = periodics [periodiccnt];
1723         adjustheap ((WT *)periodics, periodiccnt, active);
1724       }
1725   }
1726
1727   ev_stop (EV_A_ (W)w);
1728 }
1729
1730 void noinline
1731 ev_periodic_again (EV_P_ ev_periodic *w)
1732 {
1733   /* TODO: use adjustheap and recalculation */
1734   ev_periodic_stop (EV_A_ w);
1735   ev_periodic_start (EV_A_ w);
1736 }
1737 #endif
1738
1739 #ifndef SA_RESTART
1740 # define SA_RESTART 0
1741 #endif
1742
1743 void noinline
1744 ev_signal_start (EV_P_ ev_signal *w)
1745 {
1746 #if EV_MULTIPLICITY
1747   assert (("signal watchers are only supported in the default loop", loop == ev_default_loop_ptr));
1748 #endif
1749   if (expect_false (ev_is_active (w)))
1750     return;
1751
1752   assert (("ev_signal_start called with illegal signal number", w->signum > 0));
1753
1754   ev_start (EV_A_ (W)w, 1);
1755   array_needsize (ANSIG, signals, signalmax, w->signum, signals_init);
1756   wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w);
1757
1758   if (!((WL)w)->next)
1759     {
1760 #if _WIN32
1761       signal (w->signum, sighandler);
1762 #else
1763       struct sigaction sa;
1764       sa.sa_handler = sighandler;
1765       sigfillset (&sa.sa_mask);
1766       sa.sa_flags = SA_RESTART; /* if restarting works we save one iteration */
1767       sigaction (w->signum, &sa, 0);
1768 #endif
1769     }
1770 }
1771
1772 void noinline
1773 ev_signal_stop (EV_P_ ev_signal *w)
1774 {
1775   clear_pending (EV_A_ (W)w);
1776   if (expect_false (!ev_is_active (w)))
1777     return;
1778
1779   wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w);
1780   ev_stop (EV_A_ (W)w);
1781
1782   if (!signals [w->signum - 1].head)
1783     signal (w->signum, SIG_DFL);
1784 }
1785
1786 void
1787 ev_child_start (EV_P_ ev_child *w)
1788 {
1789 #if EV_MULTIPLICITY
1790   assert (("child watchers are only supported in the default loop", loop == ev_default_loop_ptr));
1791 #endif
1792   if (expect_false (ev_is_active (w)))
1793     return;
1794
1795   ev_start (EV_A_ (W)w, 1);
1796   wlist_add ((WL *)&childs [w->pid & (EV_PID_HASHSIZE - 1)], (WL)w);
1797 }
1798
1799 void
1800 ev_child_stop (EV_P_ ev_child *w)
1801 {
1802   clear_pending (EV_A_ (W)w);
1803   if (expect_false (!ev_is_active (w)))
1804     return;
1805
1806   wlist_del ((WL *)&childs [w->pid & (EV_PID_HASHSIZE - 1)], (WL)w);
1807   ev_stop (EV_A_ (W)w);
1808 }
1809
1810 #if EV_STAT_ENABLE
1811
1812 # ifdef _WIN32
1813 #  undef lstat
1814 #  define lstat(a,b) _stati64 (a,b)
1815 # endif
1816
1817 #define DEF_STAT_INTERVAL 5.0074891
1818 #define MIN_STAT_INTERVAL 0.1074891
1819
1820 static void noinline stat_timer_cb (EV_P_ ev_timer *w_, int revents);
1821
1822 #if EV_USE_INOTIFY
1823 # define EV_INOTIFY_BUFSIZE 8192
1824
1825 static void noinline
1826 infy_add (EV_P_ ev_stat *w)
1827 {
1828   w->wd = inotify_add_watch (fs_fd, w->path, IN_ATTRIB | IN_DELETE_SELF | IN_MOVE_SELF | IN_MODIFY | IN_DONT_FOLLOW | IN_MASK_ADD);
1829
1830   if (w->wd < 0)
1831     {
1832       ev_timer_start (EV_A_ &w->timer); /* this is not race-free, so we still need to recheck periodically */
1833
1834       /* monitor some parent directory for speedup hints */
1835       if ((errno == ENOENT || errno == EACCES) && strlen (w->path) < 4096)
1836         {
1837           char path [4096];
1838           strcpy (path, w->path);
1839
1840           do
1841             {
1842               int mask = IN_MASK_ADD | IN_DELETE_SELF | IN_MOVE_SELF
1843                        | (errno == EACCES ? IN_ATTRIB : IN_CREATE | IN_MOVED_TO);
1844
1845               char *pend = strrchr (path, '/');
1846
1847               if (!pend)
1848                 break; /* whoops, no '/', complain to your admin */
1849
1850               *pend = 0;
1851               w->wd = inotify_add_watch (fs_fd, path, mask);
1852             } 
1853           while (w->wd < 0 && (errno == ENOENT || errno == EACCES));
1854         }
1855     }
1856   else
1857     ev_timer_stop (EV_A_ &w->timer); /* we can watch this in a race-free way */
1858
1859   if (w->wd >= 0)
1860     wlist_add (&fs_hash [w->wd & (EV_INOTIFY_HASHSIZE - 1)].head, (WL)w);
1861 }
1862
1863 static void noinline
1864 infy_del (EV_P_ ev_stat *w)
1865 {
1866   int slot;
1867   int wd = w->wd;
1868
1869   if (wd < 0)
1870     return;
1871
1872   w->wd = -2;
1873   slot = wd & (EV_INOTIFY_HASHSIZE - 1);
1874   wlist_del (&fs_hash [slot].head, (WL)w);
1875
1876   /* remove this watcher, if others are watching it, they will rearm */
1877   inotify_rm_watch (fs_fd, wd);
1878 }
1879
1880 static void noinline
1881 infy_wd (EV_P_ int slot, int wd, struct inotify_event *ev)
1882 {
1883   if (slot < 0)
1884     /* overflow, need to check for all hahs slots */
1885     for (slot = 0; slot < EV_INOTIFY_HASHSIZE; ++slot)
1886       infy_wd (EV_A_ slot, wd, ev);
1887   else
1888     {
1889       WL w_;
1890
1891       for (w_ = fs_hash [slot & (EV_INOTIFY_HASHSIZE - 1)].head; w_; )
1892         {
1893           ev_stat *w = (ev_stat *)w_;
1894           w_ = w_->next; /* lets us remove this watcher and all before it */
1895
1896           if (w->wd == wd || wd == -1)
1897             {
1898               if (ev->mask & (IN_IGNORED | IN_UNMOUNT | IN_DELETE_SELF))
1899                 {
1900                   w->wd = -1;
1901                   infy_add (EV_A_ w); /* re-add, no matter what */
1902                 }
1903
1904               stat_timer_cb (EV_A_ &w->timer, 0);
1905             }
1906         }
1907     }
1908 }
1909
1910 static void
1911 infy_cb (EV_P_ ev_io *w, int revents)
1912 {
1913   char buf [EV_INOTIFY_BUFSIZE];
1914   struct inotify_event *ev = (struct inotify_event *)buf;
1915   int ofs;
1916   int len = read (fs_fd, buf, sizeof (buf));
1917
1918   for (ofs = 0; ofs < len; ofs += sizeof (struct inotify_event) + ev->len)
1919     infy_wd (EV_A_ ev->wd, ev->wd, ev);
1920 }
1921
1922 void inline_size
1923 infy_init (EV_P)
1924 {
1925   if (fs_fd != -2)
1926     return;
1927
1928   fs_fd = inotify_init ();
1929
1930   if (fs_fd >= 0)
1931     {
1932       ev_io_init (&fs_w, infy_cb, fs_fd, EV_READ);
1933       ev_set_priority (&fs_w, EV_MAXPRI);
1934       ev_io_start (EV_A_ &fs_w);
1935     }
1936 }
1937
1938 void inline_size
1939 infy_fork (EV_P)
1940 {
1941   int slot;
1942
1943   if (fs_fd < 0)
1944     return;
1945
1946   close (fs_fd);
1947   fs_fd = inotify_init ();
1948
1949   for (slot = 0; slot < EV_INOTIFY_HASHSIZE; ++slot)
1950     {
1951       WL w_ = fs_hash [slot].head;
1952       fs_hash [slot].head = 0;
1953
1954       while (w_)
1955         {
1956           ev_stat *w = (ev_stat *)w_;
1957           w_ = w_->next; /* lets us add this watcher */
1958
1959           w->wd = -1;
1960
1961           if (fs_fd >= 0)
1962             infy_add (EV_A_ w); /* re-add, no matter what */
1963           else
1964             ev_timer_start (EV_A_ &w->timer);
1965         }
1966
1967     }
1968 }
1969
1970 #endif
1971
1972 void
1973 ev_stat_stat (EV_P_ ev_stat *w)
1974 {
1975   if (lstat (w->path, &w->attr) < 0)
1976     w->attr.st_nlink = 0;
1977   else if (!w->attr.st_nlink)
1978     w->attr.st_nlink = 1;
1979 }
1980
1981 static void noinline
1982 stat_timer_cb (EV_P_ ev_timer *w_, int revents)
1983 {
1984   ev_stat *w = (ev_stat *)(((char *)w_) - offsetof (ev_stat, timer));
1985
1986   /* we copy this here each the time so that */
1987   /* prev has the old value when the callback gets invoked */
1988   w->prev = w->attr;
1989   ev_stat_stat (EV_A_ w);
1990
1991   /* memcmp doesn't work on netbsd, they.... do stuff to their struct stat */
1992   if (
1993     w->prev.st_dev      != w->attr.st_dev
1994     || w->prev.st_ino   != w->attr.st_ino
1995     || w->prev.st_mode  != w->attr.st_mode
1996     || w->prev.st_nlink != w->attr.st_nlink
1997     || w->prev.st_uid   != w->attr.st_uid
1998     || w->prev.st_gid   != w->attr.st_gid
1999     || w->prev.st_rdev  != w->attr.st_rdev
2000     || w->prev.st_size  != w->attr.st_size
2001     || w->prev.st_atime != w->attr.st_atime
2002     || w->prev.st_mtime != w->attr.st_mtime
2003     || w->prev.st_ctime != w->attr.st_ctime
2004   ) {
2005       #if EV_USE_INOTIFY
2006         infy_del (EV_A_ w);
2007         infy_add (EV_A_ w);
2008         ev_stat_stat (EV_A_ w); /* avoid race... */
2009       #endif
2010
2011       ev_feed_event (EV_A_ w, EV_STAT);
2012     }
2013 }
2014
2015 void
2016 ev_stat_start (EV_P_ ev_stat *w)
2017 {
2018   if (expect_false (ev_is_active (w)))
2019     return;
2020
2021   /* since we use memcmp, we need to clear any padding data etc. */
2022   memset (&w->prev, 0, sizeof (ev_statdata));
2023   memset (&w->attr, 0, sizeof (ev_statdata));
2024
2025   ev_stat_stat (EV_A_ w);
2026
2027   if (w->interval < MIN_STAT_INTERVAL)
2028     w->interval = w->interval ? MIN_STAT_INTERVAL : DEF_STAT_INTERVAL;
2029
2030   ev_timer_init (&w->timer, stat_timer_cb, w->interval, w->interval);
2031   ev_set_priority (&w->timer, ev_priority (w));
2032
2033 #if EV_USE_INOTIFY
2034   infy_init (EV_A);
2035
2036   if (fs_fd >= 0)
2037     infy_add (EV_A_ w);
2038   else
2039 #endif
2040     ev_timer_start (EV_A_ &w->timer);
2041
2042   ev_start (EV_A_ (W)w, 1);
2043 }
2044
2045 void
2046 ev_stat_stop (EV_P_ ev_stat *w)
2047 {
2048   clear_pending (EV_A_ (W)w);
2049   if (expect_false (!ev_is_active (w)))
2050     return;
2051
2052 #if EV_USE_INOTIFY
2053   infy_del (EV_A_ w);
2054 #endif
2055   ev_timer_stop (EV_A_ &w->timer);
2056
2057   ev_stop (EV_A_ (W)w);
2058 }
2059 #endif
2060
2061 #if EV_IDLE_ENABLE
2062 void
2063 ev_idle_start (EV_P_ ev_idle *w)
2064 {
2065   if (expect_false (ev_is_active (w)))
2066     return;
2067
2068   pri_adjust (EV_A_ (W)w);
2069
2070   {
2071     int active = ++idlecnt [ABSPRI (w)];
2072
2073     ++idleall;
2074     ev_start (EV_A_ (W)w, active);
2075
2076     array_needsize (ev_idle *, idles [ABSPRI (w)], idlemax [ABSPRI (w)], active, EMPTY2);
2077     idles [ABSPRI (w)][active - 1] = w;
2078   }
2079 }
2080
2081 void
2082 ev_idle_stop (EV_P_ ev_idle *w)
2083 {
2084   clear_pending (EV_A_ (W)w);
2085   if (expect_false (!ev_is_active (w)))
2086     return;
2087
2088   {
2089     int active = ((W)w)->active;
2090
2091     idles [ABSPRI (w)][active - 1] = idles [ABSPRI (w)][--idlecnt [ABSPRI (w)]];
2092     ((W)idles [ABSPRI (w)][active - 1])->active = active;
2093
2094     ev_stop (EV_A_ (W)w);
2095     --idleall;
2096   }
2097 }
2098 #endif
2099
2100 void
2101 ev_prepare_start (EV_P_ ev_prepare *w)
2102 {
2103   if (expect_false (ev_is_active (w)))
2104     return;
2105
2106   ev_start (EV_A_ (W)w, ++preparecnt);
2107   array_needsize (ev_prepare *, prepares, preparemax, preparecnt, EMPTY2);
2108   prepares [preparecnt - 1] = w;
2109 }
2110
2111 void
2112 ev_prepare_stop (EV_P_ ev_prepare *w)
2113 {
2114   clear_pending (EV_A_ (W)w);
2115   if (expect_false (!ev_is_active (w)))
2116     return;
2117
2118   {
2119     int active = ((W)w)->active;
2120     prepares [active - 1] = prepares [--preparecnt];
2121     ((W)prepares [active - 1])->active = active;
2122   }
2123
2124   ev_stop (EV_A_ (W)w);
2125 }
2126
2127 void
2128 ev_check_start (EV_P_ ev_check *w)
2129 {
2130   if (expect_false (ev_is_active (w)))
2131     return;
2132
2133   ev_start (EV_A_ (W)w, ++checkcnt);
2134   array_needsize (ev_check *, checks, checkmax, checkcnt, EMPTY2);
2135   checks [checkcnt - 1] = w;
2136 }
2137
2138 void
2139 ev_check_stop (EV_P_ ev_check *w)
2140 {
2141   clear_pending (EV_A_ (W)w);
2142   if (expect_false (!ev_is_active (w)))
2143     return;
2144
2145   {
2146     int active = ((W)w)->active;
2147     checks [active - 1] = checks [--checkcnt];
2148     ((W)checks [active - 1])->active = active;
2149   }
2150
2151   ev_stop (EV_A_ (W)w);
2152 }
2153
2154 #if EV_EMBED_ENABLE
2155 void noinline
2156 ev_embed_sweep (EV_P_ ev_embed *w)
2157 {
2158   ev_loop (w->loop, EVLOOP_NONBLOCK);
2159 }
2160
2161 static void
2162 embed_cb (EV_P_ ev_io *io, int revents)
2163 {
2164   ev_embed *w = (ev_embed *)(((char *)io) - offsetof (ev_embed, io));
2165
2166   if (ev_cb (w))
2167     ev_feed_event (EV_A_ (W)w, EV_EMBED);
2168   else
2169     ev_embed_sweep (loop, w);
2170 }
2171
2172 void
2173 ev_embed_start (EV_P_ ev_embed *w)
2174 {
2175   if (expect_false (ev_is_active (w)))
2176     return;
2177
2178   {
2179     struct ev_loop *loop = w->loop;
2180     assert (("loop to be embedded is not embeddable", backend & ev_embeddable_backends ()));
2181     ev_io_init (&w->io, embed_cb, backend_fd, EV_READ);
2182   }
2183
2184   ev_set_priority (&w->io, ev_priority (w));
2185   ev_io_start (EV_A_ &w->io);
2186
2187   ev_start (EV_A_ (W)w, 1);
2188 }
2189
2190 void
2191 ev_embed_stop (EV_P_ ev_embed *w)
2192 {
2193   clear_pending (EV_A_ (W)w);
2194   if (expect_false (!ev_is_active (w)))
2195     return;
2196
2197   ev_io_stop (EV_A_ &w->io);
2198
2199   ev_stop (EV_A_ (W)w);
2200 }
2201 #endif
2202
2203 #if EV_FORK_ENABLE
2204 void
2205 ev_fork_start (EV_P_ ev_fork *w)
2206 {
2207   if (expect_false (ev_is_active (w)))
2208     return;
2209
2210   ev_start (EV_A_ (W)w, ++forkcnt);
2211   array_needsize (ev_fork *, forks, forkmax, forkcnt, EMPTY2);
2212   forks [forkcnt - 1] = w;
2213 }
2214
2215 void
2216 ev_fork_stop (EV_P_ ev_fork *w)
2217 {
2218   clear_pending (EV_A_ (W)w);
2219   if (expect_false (!ev_is_active (w)))
2220     return;
2221
2222   {
2223     int active = ((W)w)->active;
2224     forks [active - 1] = forks [--forkcnt];
2225     ((W)forks [active - 1])->active = active;
2226   }
2227
2228   ev_stop (EV_A_ (W)w);
2229 }
2230 #endif
2231
2232 /*****************************************************************************/
2233
2234 struct ev_once
2235 {
2236   ev_io io;
2237   ev_timer to;
2238   void (*cb)(int revents, void *arg);
2239   void *arg;
2240 };
2241
2242 static void
2243 once_cb (EV_P_ struct ev_once *once, int revents)
2244 {
2245   void (*cb)(int revents, void *arg) = once->cb;
2246   void *arg = once->arg;
2247
2248   ev_io_stop (EV_A_ &once->io);
2249   ev_timer_stop (EV_A_ &once->to);
2250   ev_free (once);
2251
2252   cb (revents, arg);
2253 }
2254
2255 static void
2256 once_cb_io (EV_P_ ev_io *w, int revents)
2257 {
2258   once_cb (EV_A_ (struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents);
2259 }
2260
2261 static void
2262 once_cb_to (EV_P_ ev_timer *w, int revents)
2263 {
2264   once_cb (EV_A_ (struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents);
2265 }
2266
2267 void
2268 ev_once (EV_P_ int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg)
2269 {
2270   struct ev_once *once = (struct ev_once *)ev_malloc (sizeof (struct ev_once));
2271
2272   if (expect_false (!once))
2273     {
2274       cb (EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT, arg);
2275       return;
2276     }
2277
2278   once->cb  = cb;
2279   once->arg = arg;
2280
2281   ev_init (&once->io, once_cb_io);
2282   if (fd >= 0)
2283     {
2284       ev_io_set (&once->io, fd, events);
2285       ev_io_start (EV_A_ &once->io);
2286     }
2287
2288   ev_init (&once->to, once_cb_to);
2289   if (timeout >= 0.)
2290     {
2291       ev_timer_set (&once->to, timeout, 0.);
2292       ev_timer_start (EV_A_ &once->to);
2293     }
2294 }
2295
2296 #ifdef __cplusplus
2297 }
2298 #endif
2299