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