]> git.llucax.com Git - software/libev.git/blob - ev.c
*** empty log message ***
[software/libev.git] / ev.c
1 /*
2  * Copyright (c) 2007 Marc Alexander Lehmann <libev@schmorp.de>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are
7  * met:
8  *
9  *     * Redistributions of source code must retain the above copyright
10  *       notice, this list of conditions and the following disclaimer.
11  *
12  *     * Redistributions in binary form must reproduce the above
13  *       copyright notice, this list of conditions and the following
14  *       disclaimer in the documentation and/or other materials provided
15  *       with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29 #if EV_USE_CONFIG_H
30 # include "config.h"
31 #endif
32
33 #include <math.h>
34 #include <stdlib.h>
35 #include <unistd.h>
36 #include <fcntl.h>
37 #include <signal.h>
38 #include <stddef.h>
39
40 #include <stdio.h>
41
42 #include <assert.h>
43 #include <errno.h>
44 #include <sys/types.h>
45 #include <sys/wait.h>
46 #include <sys/time.h>
47 #include <time.h>
48
49 #ifndef EV_USE_MONOTONIC
50 # ifdef CLOCK_MONOTONIC
51 #  define EV_USE_MONOTONIC 1
52 # endif
53 #endif
54
55 #ifndef EV_USE_SELECT
56 # define EV_USE_SELECT 1
57 #endif
58
59 #ifndef EV_USE_EPOLL
60 # define EV_USE_EPOLL 0
61 #endif
62
63 #ifndef CLOCK_REALTIME
64 # define EV_USE_REALTIME 0
65 #endif
66 #ifndef EV_USE_REALTIME
67 # define EV_USE_REALTIME 1 /* posix requirement, but might be slower */
68 #endif
69
70 #define MIN_TIMEJUMP  1. /* minimum timejump that gets detected (if monotonic clock available) */
71 #define MAX_BLOCKTIME 59.731 /* never wait longer than this time (to detetc time jumps) */
72 #define PID_HASHSIZE  16 /* size of pid hash table, must be power of two */
73 #define CLEANUP_INTERVAL (MAX_BLOCKTIME * 5.) /* how often to try to free memory and re-check fds */
74
75 #include "ev.h"
76
77 typedef struct ev_watcher *W;
78 typedef struct ev_watcher_list *WL;
79 typedef struct ev_watcher_time *WT;
80
81 static ev_tstamp now, diff; /* monotonic clock */
82 ev_tstamp ev_now;
83 int ev_method;
84
85 static int have_monotonic; /* runtime */
86
87 static ev_tstamp method_fudge; /* stupid epoll-returns-early bug */
88 static void (*method_modify)(int fd, int oev, int nev);
89 static void (*method_poll)(ev_tstamp timeout);
90
91 /*****************************************************************************/
92
93 ev_tstamp
94 ev_time (void)
95 {
96 #if EV_USE_REALTIME
97   struct timespec ts;
98   clock_gettime (CLOCK_REALTIME, &ts);
99   return ts.tv_sec + ts.tv_nsec * 1e-9;
100 #else
101   struct timeval tv;
102   gettimeofday (&tv, 0);
103   return tv.tv_sec + tv.tv_usec * 1e-6;
104 #endif
105 }
106
107 static ev_tstamp
108 get_clock (void)
109 {
110 #if EV_USE_MONOTONIC
111   if (have_monotonic)
112     {
113       struct timespec ts;
114       clock_gettime (CLOCK_MONOTONIC, &ts);
115       return ts.tv_sec + ts.tv_nsec * 1e-9;
116     }
117 #endif
118
119   return ev_time ();
120 }
121
122 #define array_roundsize(base,n) ((n) | 4 & ~3)
123
124 #define array_needsize(base,cur,cnt,init)               \
125   if ((cnt) > cur)                                      \
126     {                                                   \
127       int newcnt = cur;                                 \
128       do                                                \
129         {                                               \
130           newcnt = array_roundsize (base, newcnt << 1); \
131         }                                               \
132       while ((cnt) > newcnt);                           \
133                                                         \
134       base = realloc (base, sizeof (*base) * (newcnt)); \
135       init (base + cur, newcnt - cur);                  \
136       cur = newcnt;                                     \
137     }
138
139 /*****************************************************************************/
140
141 typedef struct
142 {
143   struct ev_io *head;
144   int events;
145 } ANFD;
146
147 static ANFD *anfds;
148 static int anfdmax;
149
150 static void
151 anfds_init (ANFD *base, int count)
152 {
153   while (count--)
154     {
155       base->head   = 0;
156       base->events = EV_NONE;
157       ++base;
158     }
159 }
160
161 typedef struct
162 {
163   W w;
164   int events;
165 } ANPENDING;
166
167 static ANPENDING *pendings;
168 static int pendingmax, pendingcnt;
169
170 static void
171 event (W w, int events)
172 {
173   if (w->pending)
174     {
175       pendings [w->pending - 1].events |= events;
176       return;
177     }
178
179   w->pending = ++pendingcnt;
180   array_needsize (pendings, pendingmax, pendingcnt, );
181   pendings [pendingcnt - 1].w      = w;
182   pendings [pendingcnt - 1].events = events;
183 }
184
185 static void
186 queue_events (W *events, int eventcnt, int type)
187 {
188   int i;
189
190   for (i = 0; i < eventcnt; ++i)
191     event (events [i], type);
192 }
193
194 static void
195 fd_event (int fd, int events)
196 {
197   ANFD *anfd = anfds + fd;
198   struct ev_io *w;
199
200   for (w = anfd->head; w; w = w->next)
201     {
202       int ev = w->events & events;
203
204       if (ev)
205         event ((W)w, ev);
206     }
207 }
208
209 /*****************************************************************************/
210
211 static int *fdchanges;
212 static int fdchangemax, fdchangecnt;
213
214 static void
215 fd_reify (void)
216 {
217   int i;
218
219   for (i = 0; i < fdchangecnt; ++i)
220     {
221       int fd = fdchanges [i];
222       ANFD *anfd = anfds + fd;
223       struct ev_io *w;
224
225       int events = 0;
226
227       for (w = anfd->head; w; w = w->next)
228         events |= w->events;
229
230       anfd->events &= ~EV_REIFY;
231
232       if (anfd->events != events)
233         {
234           method_modify (fd, anfd->events, events);
235           anfd->events = events;
236         }
237     }
238
239   fdchangecnt = 0;
240 }
241
242 static void
243 fd_change (int fd)
244 {
245   if (anfds [fd].events & EV_REIFY || fdchangecnt < 0)
246     return;
247
248   anfds [fd].events |= EV_REIFY;
249
250   ++fdchangecnt;
251   array_needsize (fdchanges, fdchangemax, fdchangecnt, );
252   fdchanges [fdchangecnt - 1] = fd;
253 }
254
255 /* called on EBADF to verify fds */
256 static void
257 fd_recheck (void)
258 {
259   int fd;
260
261   for (fd = 0; fd < anfdmax; ++fd)
262     if (anfds [fd].events)
263       if (fcntl (fd, F_GETFD) == -1 && errno == EBADF)
264         while (anfds [fd].head)
265           {
266             ev_io_stop (anfds [fd].head);
267             event ((W)anfds [fd].head, EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT);
268           }
269 }
270
271 /*****************************************************************************/
272
273 static struct ev_timer **timers;
274 static int timermax, timercnt;
275
276 static struct ev_periodic **periodics;
277 static int periodicmax, periodiccnt;
278
279 static void
280 upheap (WT *timers, int k)
281 {
282   WT w = timers [k];
283
284   while (k && timers [k >> 1]->at > w->at)
285     {
286       timers [k] = timers [k >> 1];
287       timers [k]->active = k + 1;
288       k >>= 1;
289     }
290
291   timers [k] = w;
292   timers [k]->active = k + 1;
293
294 }
295
296 static void
297 downheap (WT *timers, int N, int k)
298 {
299   WT w = timers [k];
300
301   while (k < (N >> 1))
302     {
303       int j = k << 1;
304
305       if (j + 1 < N && timers [j]->at > timers [j + 1]->at)
306         ++j;
307
308       if (w->at <= timers [j]->at)
309         break;
310
311       timers [k] = timers [j];
312       timers [k]->active = k + 1;
313       k = j;
314     }
315
316   timers [k] = w;
317   timers [k]->active = k + 1;
318 }
319
320 /*****************************************************************************/
321
322 typedef struct
323 {
324   struct ev_signal *head;
325   sig_atomic_t gotsig;
326 } ANSIG;
327
328 static ANSIG *signals;
329 static int signalmax;
330
331 static int sigpipe [2];
332 static sig_atomic_t gotsig;
333 static struct ev_io sigev;
334
335 static void
336 signals_init (ANSIG *base, int count)
337 {
338   while (count--)
339     {
340       base->head   = 0;
341       base->gotsig = 0;
342       ++base;
343     }
344 }
345
346 static void
347 sighandler (int signum)
348 {
349   signals [signum - 1].gotsig = 1;
350
351   if (!gotsig)
352     {
353       gotsig = 1;
354       write (sigpipe [1], &gotsig, 1);
355     }
356 }
357
358 static void
359 sigcb (struct ev_io *iow, int revents)
360 {
361   struct ev_signal *w;
362   int sig;
363
364   gotsig = 0;
365   read (sigpipe [0], &revents, 1);
366
367   for (sig = signalmax; sig--; )
368     if (signals [sig].gotsig)
369       {
370         signals [sig].gotsig = 0;
371
372         for (w = signals [sig].head; w; w = w->next)
373           event ((W)w, EV_SIGNAL);
374       }
375 }
376
377 static void
378 siginit (void)
379 {
380   fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
381   fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
382
383   /* rather than sort out wether we really need nb, set it */
384   fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
385   fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
386
387   ev_io_set (&sigev, sigpipe [0], EV_READ);
388   ev_io_start (&sigev);
389 }
390
391 /*****************************************************************************/
392
393 static struct ev_idle **idles;
394 static int idlemax, idlecnt;
395
396 static struct ev_prepare **prepares;
397 static int preparemax, preparecnt;
398
399 static struct ev_check **checks;
400 static int checkmax, checkcnt;
401
402 /*****************************************************************************/
403
404 static struct ev_child *childs [PID_HASHSIZE];
405 static struct ev_signal childev;
406
407 #ifndef WCONTINUED
408 # define WCONTINUED 0
409 #endif
410
411 static void
412 childcb (struct ev_signal *sw, int revents)
413 {
414   struct ev_child *w;
415   int pid, status;
416
417   while ((pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)) != -1)
418     for (w = childs [pid & (PID_HASHSIZE - 1)]; w; w = w->next)
419       if (w->pid == pid || w->pid == -1)
420         {
421           w->status = status;
422           event ((W)w, EV_CHILD);
423         }
424 }
425
426 /*****************************************************************************/
427
428 #if EV_USE_EPOLL
429 # include "ev_epoll.c"
430 #endif
431 #if EV_USE_SELECT
432 # include "ev_select.c"
433 #endif
434
435 int
436 ev_version_major (void)
437 {
438   return EV_VERSION_MAJOR;
439 }
440
441 int
442 ev_version_minor (void)
443 {
444   return EV_VERSION_MINOR;
445 }
446
447 int ev_init (int flags)
448 {
449   if (!ev_method)
450     {
451 #if EV_USE_MONOTONIC
452       {
453         struct timespec ts;
454         if (!clock_gettime (CLOCK_MONOTONIC, &ts))
455           have_monotonic = 1;
456       }
457 #endif
458
459       ev_now = ev_time ();
460       now    = get_clock ();
461       diff   = ev_now - now;
462
463       if (pipe (sigpipe))
464         return 0;
465
466       ev_method = EVMETHOD_NONE;
467 #if EV_USE_EPOLL
468       if (ev_method == EVMETHOD_NONE) epoll_init (flags);
469 #endif
470 #if EV_USE_SELECT
471       if (ev_method == EVMETHOD_NONE) select_init (flags);
472 #endif
473
474       if (ev_method)
475         {
476           ev_watcher_init (&sigev, sigcb);
477           siginit ();
478
479           ev_signal_init (&childev, childcb, SIGCHLD);
480           ev_signal_start (&childev);
481         }
482     }
483
484   return ev_method;
485 }
486
487 /*****************************************************************************/
488
489 void
490 ev_prefork (void)
491 {
492   /* nop */
493 }
494
495 void
496 ev_postfork_parent (void)
497 {
498   /* nop */
499 }
500
501 void
502 ev_postfork_child (void)
503 {
504 #if EV_USE_EPOLL
505   if (ev_method == EVMETHOD_EPOLL)
506     epoll_postfork_child ();
507 #endif
508
509   ev_io_stop (&sigev);
510   close (sigpipe [0]);
511   close (sigpipe [1]);
512   pipe (sigpipe);
513   siginit ();
514 }
515
516 /*****************************************************************************/
517
518 static void
519 call_pending (void)
520 {
521   while (pendingcnt)
522     {
523       ANPENDING *p = pendings + --pendingcnt;
524
525       if (p->w)
526         {
527           p->w->pending = 0;
528           p->w->cb (p->w, p->events);
529         }
530     }
531 }
532
533 static void
534 timers_reify (void)
535 {
536   while (timercnt && timers [0]->at <= now)
537     {
538       struct ev_timer *w = timers [0];
539
540       /* first reschedule or stop timer */
541       if (w->repeat)
542         {
543           w->at = now + w->repeat;
544           assert (("timer timeout in the past, negative repeat?", w->at > now));
545           downheap ((WT *)timers, timercnt, 0);
546         }
547       else
548         ev_timer_stop (w); /* nonrepeating: stop timer */
549
550       event ((W)w, EV_TIMEOUT);
551     }
552 }
553
554 static void
555 periodics_reify (void)
556 {
557   while (periodiccnt && periodics [0]->at <= ev_now)
558     {
559       struct ev_periodic *w = periodics [0];
560
561       /* first reschedule or stop timer */
562       if (w->interval)
563         {
564           w->at += floor ((ev_now - w->at) / w->interval + 1.) * w->interval;
565           assert (("periodic timeout in the past, negative interval?", w->at > ev_now));
566           downheap ((WT *)periodics, periodiccnt, 0);
567         }
568       else
569         ev_periodic_stop (w); /* nonrepeating: stop timer */
570
571       event ((W)w, EV_TIMEOUT);
572     }
573 }
574
575 static void
576 periodics_reschedule (ev_tstamp diff)
577 {
578   int i;
579
580   /* adjust periodics after time jump */
581   for (i = 0; i < periodiccnt; ++i)
582     {
583       struct ev_periodic *w = periodics [i];
584
585       if (w->interval)
586         {
587           ev_tstamp diff = ceil ((ev_now - w->at) / w->interval) * w->interval;
588
589           if (fabs (diff) >= 1e-4)
590             {
591               ev_periodic_stop (w);
592               ev_periodic_start (w);
593
594               i = 0; /* restart loop, inefficient, but time jumps should be rare */
595             }
596         }
597     }
598 }
599
600 static void
601 time_update (void)
602 {
603   int i;
604
605   ev_now = ev_time ();
606
607   if (have_monotonic)
608     {
609       ev_tstamp odiff = diff;
610
611       for (i = 4; --i; ) /* loop a few times, before making important decisions */
612         {
613           now = get_clock ();
614           diff = ev_now - now;
615
616           if (fabs (odiff - diff) < MIN_TIMEJUMP)
617             return; /* all is well */
618
619           ev_now = ev_time ();
620         }
621
622       periodics_reschedule (diff - odiff);
623       /* no timer adjustment, as the monotonic clock doesn't jump */
624     }
625   else
626     {
627       if (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP)
628         {
629           periodics_reschedule (ev_now - now);
630
631           /* adjust timers. this is easy, as the offset is the same for all */
632           for (i = 0; i < timercnt; ++i)
633             timers [i]->at += diff;
634         }
635
636       now = ev_now;
637     }
638 }
639
640 int ev_loop_done;
641
642 void ev_loop (int flags)
643 {
644   double block;
645   ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0;
646
647   do
648     {
649       /* queue check watchers (and execute them) */
650       if (preparecnt)
651         {
652           queue_events ((W *)prepares, preparecnt, EV_PREPARE);
653           call_pending ();
654         }
655
656       /* update fd-related kernel structures */
657       fd_reify ();
658
659       /* calculate blocking time */
660
661       /* we only need this for !monotonic clockor timers, but as we basically
662          always have timers, we just calculate it always */
663       ev_now = ev_time ();
664
665       if (flags & EVLOOP_NONBLOCK || idlecnt)
666         block = 0.;
667       else
668         {
669           block = MAX_BLOCKTIME;
670
671           if (timercnt)
672             {
673               ev_tstamp to = timers [0]->at - (have_monotonic ? get_clock () : ev_now) + method_fudge;
674               if (block > to) block = to;
675             }
676
677           if (periodiccnt)
678             {
679               ev_tstamp to = periodics [0]->at - ev_now + method_fudge;
680               if (block > to) block = to;
681             }
682
683           if (block < 0.) block = 0.;
684         }
685
686       method_poll (block);
687
688       /* update ev_now, do magic */
689       time_update ();
690
691       /* queue pending timers and reschedule them */
692       timers_reify (); /* relative timers called last */
693       periodics_reify (); /* absolute timers called first */
694
695       /* queue idle watchers unless io or timers are pending */
696       if (!pendingcnt)
697         queue_events ((W *)idles, idlecnt, EV_IDLE);
698
699       /* queue check watchers, to be executed first */
700       if (checkcnt)
701         queue_events ((W *)checks, checkcnt, EV_CHECK);
702
703       call_pending ();
704     }
705   while (!ev_loop_done);
706
707   if (ev_loop_done != 2)
708     ev_loop_done = 0;
709 }
710
711 /*****************************************************************************/
712
713 static void
714 wlist_add (WL *head, WL elem)
715 {
716   elem->next = *head;
717   *head = elem;
718 }
719
720 static void
721 wlist_del (WL *head, WL elem)
722 {
723   while (*head)
724     {
725       if (*head == elem)
726         {
727           *head = elem->next;
728           return;
729         }
730
731       head = &(*head)->next;
732     }
733 }
734
735 static void
736 ev_clear (W w)
737 {
738   if (w->pending)
739     {
740       pendings [w->pending - 1].w = 0;
741       w->pending = 0;
742     }
743 }
744
745 static void
746 ev_start (W w, int active)
747 {
748   w->active = active;
749 }
750
751 static void
752 ev_stop (W w)
753 {
754   w->active = 0;
755 }
756
757 /*****************************************************************************/
758
759 void
760 ev_io_start (struct ev_io *w)
761 {
762   if (ev_is_active (w))
763     return;
764
765   int fd = w->fd;
766
767   ev_start ((W)w, 1);
768   array_needsize (anfds, anfdmax, fd + 1, anfds_init);
769   wlist_add ((WL *)&anfds[fd].head, (WL)w);
770
771   fd_change (fd);
772 }
773
774 void
775 ev_io_stop (struct ev_io *w)
776 {
777   ev_clear ((W)w);
778   if (!ev_is_active (w))
779     return;
780
781   wlist_del ((WL *)&anfds[w->fd].head, (WL)w);
782   ev_stop ((W)w);
783
784   fd_change (w->fd);
785 }
786
787 void
788 ev_timer_start (struct ev_timer *w)
789 {
790   if (ev_is_active (w))
791     return;
792
793   w->at += now;
794
795   assert (("timer repeat value less than zero not allowed", w->repeat >= 0.));
796
797   ev_start ((W)w, ++timercnt);
798   array_needsize (timers, timermax, timercnt, );
799   timers [timercnt - 1] = w;
800   upheap ((WT *)timers, timercnt - 1);
801 }
802
803 void
804 ev_timer_stop (struct ev_timer *w)
805 {
806   ev_clear ((W)w);
807   if (!ev_is_active (w))
808     return;
809
810   if (w->active < timercnt--)
811     {
812       timers [w->active - 1] = timers [timercnt];
813       downheap ((WT *)timers, timercnt, w->active - 1);
814     }
815
816   w->at = w->repeat;
817
818   ev_stop ((W)w);
819 }
820
821 void
822 ev_timer_again (struct ev_timer *w)
823 {
824   if (ev_is_active (w))
825     {
826       if (w->repeat)
827         {
828           w->at = now + w->repeat;
829           downheap ((WT *)timers, timercnt, w->active - 1);
830         }
831       else
832         ev_timer_stop (w);
833     }
834   else if (w->repeat)
835     ev_timer_start (w);
836 }
837
838 void
839 ev_periodic_start (struct ev_periodic *w)
840 {
841   if (ev_is_active (w))
842     return;
843
844   assert (("periodic interval value less than zero not allowed", w->interval >= 0.));
845
846   /* this formula differs from the one in periodic_reify because we do not always round up */
847   if (w->interval)
848     w->at += ceil ((ev_now - w->at) / w->interval) * w->interval;
849
850   ev_start ((W)w, ++periodiccnt);
851   array_needsize (periodics, periodicmax, periodiccnt, );
852   periodics [periodiccnt - 1] = w;
853   upheap ((WT *)periodics, periodiccnt - 1);
854 }
855
856 void
857 ev_periodic_stop (struct ev_periodic *w)
858 {
859   ev_clear ((W)w);
860   if (!ev_is_active (w))
861     return;
862
863   if (w->active < periodiccnt--)
864     {
865       periodics [w->active - 1] = periodics [periodiccnt];
866       downheap ((WT *)periodics, periodiccnt, w->active - 1);
867     }
868
869   ev_stop ((W)w);
870 }
871
872 void
873 ev_signal_start (struct ev_signal *w)
874 {
875   if (ev_is_active (w))
876     return;
877
878   ev_start ((W)w, 1);
879   array_needsize (signals, signalmax, w->signum, signals_init);
880   wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w);
881
882   if (!w->next)
883     {
884       struct sigaction sa;
885       sa.sa_handler = sighandler;
886       sigfillset (&sa.sa_mask);
887       sa.sa_flags = 0;
888       sigaction (w->signum, &sa, 0);
889     }
890 }
891
892 void
893 ev_signal_stop (struct ev_signal *w)
894 {
895   ev_clear ((W)w);
896   if (!ev_is_active (w))
897     return;
898
899   wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w);
900   ev_stop ((W)w);
901
902   if (!signals [w->signum - 1].head)
903     signal (w->signum, SIG_DFL);
904 }
905
906 void
907 ev_idle_start (struct ev_idle *w)
908 {
909   if (ev_is_active (w))
910     return;
911
912   ev_start ((W)w, ++idlecnt);
913   array_needsize (idles, idlemax, idlecnt, );
914   idles [idlecnt - 1] = w;
915 }
916
917 void
918 ev_idle_stop (struct ev_idle *w)
919 {
920   ev_clear ((W)w);
921   if (ev_is_active (w))
922     return;
923
924   idles [w->active - 1] = idles [--idlecnt];
925   ev_stop ((W)w);
926 }
927
928 void
929 ev_prepare_start (struct ev_prepare *w)
930 {
931   if (ev_is_active (w))
932     return;
933
934   ev_start ((W)w, ++preparecnt);
935   array_needsize (prepares, preparemax, preparecnt, );
936   prepares [preparecnt - 1] = w;
937 }
938
939 void
940 ev_prepare_stop (struct ev_prepare *w)
941 {
942   ev_clear ((W)w);
943   if (ev_is_active (w))
944     return;
945
946   prepares [w->active - 1] = prepares [--preparecnt];
947   ev_stop ((W)w);
948 }
949
950 void
951 ev_check_start (struct ev_check *w)
952 {
953   if (ev_is_active (w))
954     return;
955
956   ev_start ((W)w, ++checkcnt);
957   array_needsize (checks, checkmax, checkcnt, );
958   checks [checkcnt - 1] = w;
959 }
960
961 void
962 ev_check_stop (struct ev_check *w)
963 {
964   ev_clear ((W)w);
965   if (ev_is_active (w))
966     return;
967
968   checks [w->active - 1] = checks [--checkcnt];
969   ev_stop ((W)w);
970 }
971
972 void
973 ev_child_start (struct ev_child *w)
974 {
975   if (ev_is_active (w))
976     return;
977
978   ev_start ((W)w, 1);
979   wlist_add ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
980 }
981
982 void
983 ev_child_stop (struct ev_child *w)
984 {
985   ev_clear ((W)w);
986   if (ev_is_active (w))
987     return;
988
989   wlist_del ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
990   ev_stop ((W)w);
991 }
992
993 /*****************************************************************************/
994
995 struct ev_once
996 {
997   struct ev_io io;
998   struct ev_timer to;
999   void (*cb)(int revents, void *arg);
1000   void *arg;
1001 };
1002
1003 static void
1004 once_cb (struct ev_once *once, int revents)
1005 {
1006   void (*cb)(int revents, void *arg) = once->cb;
1007   void *arg = once->arg;
1008
1009   ev_io_stop (&once->io);
1010   ev_timer_stop (&once->to);
1011   free (once);
1012
1013   cb (revents, arg);
1014 }
1015
1016 static void
1017 once_cb_io (struct ev_io *w, int revents)
1018 {
1019   once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents);
1020 }
1021
1022 static void
1023 once_cb_to (struct ev_timer *w, int revents)
1024 {
1025   once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents);
1026 }
1027
1028 void
1029 ev_once (int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg)
1030 {
1031   struct ev_once *once = malloc (sizeof (struct ev_once));
1032
1033   if (!once)
1034     cb (EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT, arg);
1035   else
1036     {
1037       once->cb  = cb;
1038       once->arg = arg;
1039
1040       ev_watcher_init (&once->io, once_cb_io);
1041       if (fd >= 0)
1042         {
1043           ev_io_set (&once->io, fd, events);
1044           ev_io_start (&once->io);
1045         }
1046
1047       ev_watcher_init (&once->to, once_cb_to);
1048       if (timeout >= 0.)
1049         {
1050           ev_timer_set (&once->to, timeout, 0.);
1051           ev_timer_start (&once->to);
1052         }
1053     }
1054 }
1055
1056 /*****************************************************************************/
1057
1058 #if 0
1059
1060 struct ev_io wio;
1061
1062 static void
1063 sin_cb (struct ev_io *w, int revents)
1064 {
1065   fprintf (stderr, "sin %d, revents %d\n", w->fd, revents);
1066 }
1067
1068 static void
1069 ocb (struct ev_timer *w, int revents)
1070 {
1071   //fprintf (stderr, "timer %f,%f (%x) (%f) d%p\n", w->at, w->repeat, revents, w->at - ev_time (), w->data);
1072   ev_timer_stop (w);
1073   ev_timer_start (w);
1074 }
1075
1076 static void
1077 scb (struct ev_signal *w, int revents)
1078 {
1079   fprintf (stderr, "signal %x,%d\n", revents, w->signum);
1080   ev_io_stop (&wio);
1081   ev_io_start (&wio);
1082 }
1083
1084 static void
1085 gcb (struct ev_signal *w, int revents)
1086 {
1087   fprintf (stderr, "generic %x\n", revents);
1088
1089 }
1090
1091 int main (void)
1092 {
1093   ev_init (0);
1094
1095   ev_io_init (&wio, sin_cb, 0, EV_READ);
1096   ev_io_start (&wio);
1097
1098   struct ev_timer t[10000];
1099
1100 #if 0
1101   int i;
1102   for (i = 0; i < 10000; ++i)
1103     {
1104       struct ev_timer *w = t + i;
1105       ev_watcher_init (w, ocb, i);
1106       ev_timer_init_abs (w, ocb, drand48 (), 0.99775533);
1107       ev_timer_start (w);
1108       if (drand48 () < 0.5)
1109         ev_timer_stop (w);
1110     }
1111 #endif
1112
1113   struct ev_timer t1;
1114   ev_timer_init (&t1, ocb, 5, 10);
1115   ev_timer_start (&t1);
1116
1117   struct ev_signal sig;
1118   ev_signal_init (&sig, scb, SIGQUIT);
1119   ev_signal_start (&sig);
1120
1121   struct ev_check cw;
1122   ev_check_init (&cw, gcb);
1123   ev_check_start (&cw);
1124
1125   struct ev_idle iw;
1126   ev_idle_init (&iw, gcb);
1127   ev_idle_start (&iw);
1128
1129   ev_loop (0);
1130
1131   return 0;
1132 }
1133
1134 #endif
1135
1136
1137
1138