ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libev/ev.c
Revision: 1.74
Committed: Tue Nov 6 16:51:20 2007 UTC (16 years, 6 months ago) by root
Content type: text/plain
Branch: MAIN
CVS Tags: rel-0_51
Changes since 1.73: +16 -16 lines
Log Message:
*** empty log message ***

File Contents

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