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