ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libev/ev.c
(Generate patch)

Comparing libev/ev.c (file contents):
Revision 1.4 by root, Tue Oct 30 23:10:33 2007 UTC vs.
Revision 1.10 by root, Wed Oct 31 07:36:03 2007 UTC

1#include <math.h> 1#include <math.h>
2#include <stdlib.h> 2#include <stdlib.h>
3#include <unistd.h>
4#include <fcntl.h>
5#include <signal.h>
3 6
4#include <stdio.h> 7#include <stdio.h>
5 8
6#include <assert.h> 9#include <assert.h>
7#include <errno.h> 10#include <errno.h>
8#include <sys/time.h> 11#include <sys/time.h>
9#include <time.h> 12#include <time.h>
10 13
14#ifndef HAVE_MONOTONIC
11#ifdef CLOCK_MONOTONIC 15# ifdef CLOCK_MONOTONIC
12# define HAVE_MONOTONIC 1 16# define HAVE_MONOTONIC 1
13#endif 17# endif
18#endif
14 19
15#define HAVE_EPOLL 1 20#ifndef HAVE_SELECT
16#define HAVE_REALTIME 1
17#define HAVE_SELECT 0 21# define HAVE_SELECT 1
22#endif
23
24#ifndef HAVE_EPOLL
25# define HAVE_EPOLL 0
26#endif
27
28#ifndef HAVE_REALTIME
29# define HAVE_REALTIME 1 /* posix requirement, but might be slower */
30#endif
18 31
19#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */ 32#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */
20#define MAX_BLOCKTIME 60. 33#define MAX_BLOCKTIME 60.
21 34
22#include "ev.h" 35#include "ev.h"
27 40
28struct ev_watcher_list { 41struct ev_watcher_list {
29 EV_WATCHER_LIST (ev_watcher_list); 42 EV_WATCHER_LIST (ev_watcher_list);
30}; 43};
31 44
45typedef struct ev_watcher *W;
46typedef struct ev_watcher_list *WL;
47
32static ev_tstamp now, diff; /* monotonic clock */ 48static ev_tstamp now, diff; /* monotonic clock */
33ev_tstamp ev_now; 49ev_tstamp ev_now;
34int ev_method; 50int ev_method;
35 51
36static int have_monotonic; /* runtime */ 52static int have_monotonic; /* runtime */
37 53
38static ev_tstamp method_fudge; /* stupid epoll-returns-early bug */ 54static ev_tstamp method_fudge; /* stupid epoll-returns-early bug */
39static void (*method_reify)(void); 55static void (*method_modify)(int fd, int oev, int nev);
40static void (*method_poll)(ev_tstamp timeout); 56static void (*method_poll)(ev_tstamp timeout);
57
58/*****************************************************************************/
41 59
42ev_tstamp 60ev_tstamp
43ev_time (void) 61ev_time (void)
44{ 62{
45#if HAVE_REALTIME 63#if HAVE_REALTIME
76 base = realloc (base, sizeof (*base) * (newcnt)); \ 94 base = realloc (base, sizeof (*base) * (newcnt)); \
77 init (base + cur, newcnt - cur); \ 95 init (base + cur, newcnt - cur); \
78 cur = newcnt; \ 96 cur = newcnt; \
79 } 97 }
80 98
99/*****************************************************************************/
100
81typedef struct 101typedef struct
82{ 102{
83 struct ev_io *head; 103 struct ev_io *head;
84 unsigned char wev, rev; /* want, received event set */ 104 unsigned char wev, rev; /* want, received event set */
85} ANFD; 105} ANFD;
101 } 121 }
102} 122}
103 123
104typedef struct 124typedef struct
105{ 125{
106 struct ev_watcher *w; 126 W w;
107 int events; 127 int events;
108} ANPENDING; 128} ANPENDING;
109 129
110static ANPENDING *pendings; 130static ANPENDING *pendings;
111static int pendingmax, pendingcnt; 131static int pendingmax, pendingcnt;
112 132
113static void 133static void
114event (struct ev_watcher *w, int events) 134event (W w, int events)
115{ 135{
116 w->pending = ++pendingcnt; 136 w->pending = ++pendingcnt;
117 array_needsize (pendings, pendingmax, pendingcnt, ); 137 array_needsize (pendings, pendingmax, pendingcnt, );
118 pendings [pendingcnt - 1].w = w; 138 pendings [pendingcnt - 1].w = w;
119 pendings [pendingcnt - 1].events = events; 139 pendings [pendingcnt - 1].events = events;
128 for (w = anfd->head; w; w = w->next) 148 for (w = anfd->head; w; w = w->next)
129 { 149 {
130 int ev = w->events & events; 150 int ev = w->events & events;
131 151
132 if (ev) 152 if (ev)
133 event ((struct ev_watcher *)w, ev); 153 event ((W)w, ev);
134 } 154 }
135} 155}
156
157static void
158queue_events (W *events, int eventcnt, int type)
159{
160 int i;
161
162 for (i = 0; i < eventcnt; ++i)
163 event (events [i], type);
164}
165
166/*****************************************************************************/
136 167
137static struct ev_timer **atimers; 168static struct ev_timer **atimers;
138static int atimermax, atimercnt; 169static int atimermax, atimercnt;
139 170
140static struct ev_timer **rtimers; 171static struct ev_timer **rtimers;
179 210
180 timers [k] = w; 211 timers [k] = w;
181 timers [k]->active = k + 1; 212 timers [k]->active = k + 1;
182} 213}
183 214
184static struct ev_signal **signals; 215/*****************************************************************************/
216
217typedef struct
218{
219 struct ev_signal *head;
220 sig_atomic_t gotsig;
221} ANSIG;
222
223static ANSIG *signals;
185static int signalmax; 224static int signalmax;
186 225
226static int sigpipe [2];
227static sig_atomic_t gotsig;
228static struct ev_io sigev;
229
187static void 230static void
188signals_init (struct ev_signal **base, int count) 231signals_init (ANSIG *base, int count)
189{ 232{
190 while (count--) 233 while (count--)
191 *base++ = 0; 234 {
235 base->head = 0;
236 base->gotsig = 0;
237 ++base;
238 }
192} 239}
240
241static void
242sighandler (int signum)
243{
244 signals [signum - 1].gotsig = 1;
245
246 if (!gotsig)
247 {
248 gotsig = 1;
249 write (sigpipe [1], &gotsig, 1);
250 }
251}
252
253static void
254sigcb (struct ev_io *iow, int revents)
255{
256 struct ev_signal *w;
257 int sig;
258
259 gotsig = 0;
260 read (sigpipe [0], &revents, 1);
261
262 for (sig = signalmax; sig--; )
263 if (signals [sig].gotsig)
264 {
265 signals [sig].gotsig = 0;
266
267 for (w = signals [sig].head; w; w = w->next)
268 event ((W)w, EV_SIGNAL);
269 }
270}
271
272static void
273siginit (void)
274{
275 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
276 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
277
278 /* rather than sort out wether we really need nb, set it */
279 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
280 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
281
282 evio_set (&sigev, sigpipe [0], EV_READ);
283 evio_start (&sigev);
284}
285
286/*****************************************************************************/
287
288static struct ev_idle **idles;
289static int idlemax, idlecnt;
290
291static struct ev_check **checks;
292static int checkmax, checkcnt;
293
294/*****************************************************************************/
193 295
194#if HAVE_EPOLL 296#if HAVE_EPOLL
195# include "ev_epoll.c" 297# include "ev_epoll.c"
196#endif 298#endif
197#if HAVE_SELECT 299#if HAVE_SELECT
210 312
211 ev_now = ev_time (); 313 ev_now = ev_time ();
212 now = get_clock (); 314 now = get_clock ();
213 diff = ev_now - now; 315 diff = ev_now - now;
214 316
317 if (pipe (sigpipe))
318 return 0;
319
320 ev_method = EVMETHOD_NONE;
215#if HAVE_EPOLL 321#if HAVE_EPOLL
216 if (epoll_init (flags)) 322 if (ev_method == EVMETHOD_NONE) epoll_init (flags);
217 return ev_method;
218#endif 323#endif
219#if HAVE_SELECT 324#if HAVE_SELECT
220 if (select_init (flags)) 325 if (ev_method == EVMETHOD_NONE) select_init (flags);
221 return ev_method;
222#endif 326#endif
223 327
224 ev_method = EVMETHOD_NONE; 328 if (ev_method)
329 {
330 evw_init (&sigev, sigcb, 0);
331 siginit ();
332 }
333
225 return ev_method; 334 return ev_method;
226} 335}
227 336
337/*****************************************************************************/
338
228void ev_prefork (void) 339void ev_prefork (void)
229{ 340{
230} 341}
231 342
232void ev_postfork_parent (void) 343void ev_postfork_parent (void)
234} 345}
235 346
236void ev_postfork_child (void) 347void ev_postfork_child (void)
237{ 348{
238#if HAVE_EPOLL 349#if HAVE_EPOLL
350 if (ev_method == EVMETHOD_EPOLL)
239 epoll_postfork_child (); 351 epoll_postfork_child ();
240#endif 352#endif
353
354 evio_stop (&sigev);
355 close (sigpipe [0]);
356 close (sigpipe [1]);
357 pipe (sigpipe);
358 siginit ();
359}
360
361/*****************************************************************************/
362
363static void
364fd_reify (void)
365{
366 int i;
367
368 for (i = 0; i < fdchangecnt; ++i)
369 {
370 int fd = fdchanges [i];
371 ANFD *anfd = anfds + fd;
372 struct ev_io *w;
373
374 int wev = 0;
375
376 for (w = anfd->head; w; w = w->next)
377 wev |= w->events;
378
379 if (anfd->wev != wev)
380 {
381 method_modify (fd, anfd->wev, wev);
382 anfd->wev = wev;
383 }
384 }
385
386 fdchangecnt = 0;
241} 387}
242 388
243static void 389static void
244call_pending () 390call_pending ()
245{ 391{
282 { 428 {
283 evtimer_stop (w); /* nonrepeating: stop timer */ 429 evtimer_stop (w); /* nonrepeating: stop timer */
284 --timercnt; /* maybe pass by reference instead? */ 430 --timercnt; /* maybe pass by reference instead? */
285 } 431 }
286 432
287 event ((struct ev_watcher *)w, EV_TIMEOUT); 433 event ((W)w, EV_TIMEOUT);
288 } 434 }
289} 435}
290 436
291static void 437static void
292time_update () 438time_update ()
333void ev_loop (int flags) 479void ev_loop (int flags)
334{ 480{
335 double block; 481 double block;
336 ev_loop_done = flags & EVLOOP_ONESHOT; 482 ev_loop_done = flags & EVLOOP_ONESHOT;
337 483
484 if (checkcnt)
485 {
486 queue_events ((W *)checks, checkcnt, EV_CHECK);
487 call_pending ();
488 }
489
338 do 490 do
339 { 491 {
340 /* update fd-related kernel structures */ 492 /* update fd-related kernel structures */
341 method_reify (); fdchangecnt = 0; 493 fd_reify ();
342 494
343 /* calculate blocking time */ 495 /* calculate blocking time */
344 if (flags & EVLOOP_NONBLOCK) 496 if (flags & EVLOOP_NONBLOCK || idlecnt)
345 block = 0.; 497 block = 0.;
346 else 498 else
347 { 499 {
348 block = MAX_BLOCKTIME; 500 block = MAX_BLOCKTIME;
349 501
365 method_poll (block); 517 method_poll (block);
366 518
367 /* update ev_now, do magic */ 519 /* update ev_now, do magic */
368 time_update (); 520 time_update ();
369 521
370 /* put pending timers into pendign queue and reschedule them */ 522 /* queue pending timers and reschedule them */
371 /* absolute timers first */ 523 /* absolute timers first */
372 timers_reify (atimers, atimercnt, ev_now); 524 timers_reify (atimers, atimercnt, ev_now);
373 /* relative timers second */ 525 /* relative timers second */
374 timers_reify (rtimers, rtimercnt, now); 526 timers_reify (rtimers, rtimercnt, now);
375 527
528 /* queue idle watchers unless io or timers are pending */
529 if (!pendingcnt)
530 queue_events ((W *)idles, idlecnt, EV_IDLE);
531
532 /* queue check and possibly idle watchers */
533 queue_events ((W *)checks, checkcnt, EV_CHECK);
534
376 call_pending (); 535 call_pending ();
377 } 536 }
378 while (!ev_loop_done); 537 while (!ev_loop_done);
379} 538}
380 539
540/*****************************************************************************/
541
381static void 542static void
382wlist_add (struct ev_watcher_list **head, struct ev_watcher_list *elem) 543wlist_add (WL *head, WL elem)
383{ 544{
384 elem->next = *head; 545 elem->next = *head;
385 *head = elem; 546 *head = elem;
386} 547}
387 548
388static void 549static void
389wlist_del (struct ev_watcher_list **head, struct ev_watcher_list *elem) 550wlist_del (WL *head, WL elem)
390{ 551{
391 while (*head) 552 while (*head)
392 { 553 {
393 if (*head == elem) 554 if (*head == elem)
394 { 555 {
399 head = &(*head)->next; 560 head = &(*head)->next;
400 } 561 }
401} 562}
402 563
403static void 564static void
404ev_start (struct ev_watcher *w, int active) 565ev_start (W w, int active)
405{ 566{
406 w->pending = 0; 567 w->pending = 0;
407 w->active = active; 568 w->active = active;
408} 569}
409 570
410static void 571static void
411ev_stop (struct ev_watcher *w) 572ev_stop (W w)
412{ 573{
413 if (w->pending) 574 if (w->pending)
414 pendings [w->pending - 1].w = 0; 575 pendings [w->pending - 1].w = 0;
415 576
416 w->active = 0; 577 w->active = 0;
417 /* nop */ 578 /* nop */
418} 579}
419 580
581/*****************************************************************************/
582
420void 583void
421evio_start (struct ev_io *w) 584evio_start (struct ev_io *w)
422{ 585{
423 if (ev_is_active (w)) 586 if (ev_is_active (w))
424 return; 587 return;
425 588
426 int fd = w->fd; 589 int fd = w->fd;
427 590
428 ev_start ((struct ev_watcher *)w, 1); 591 ev_start ((W)w, 1);
429 array_needsize (anfds, anfdmax, fd + 1, anfds_init); 592 array_needsize (anfds, anfdmax, fd + 1, anfds_init);
430 wlist_add ((struct ev_watcher_list **)&anfds[fd].head, (struct ev_watcher_list *)w); 593 wlist_add ((WL *)&anfds[fd].head, (WL)w);
431 594
432 ++fdchangecnt; 595 ++fdchangecnt;
433 array_needsize (fdchanges, fdchangemax, fdchangecnt, ); 596 array_needsize (fdchanges, fdchangemax, fdchangecnt, );
434 fdchanges [fdchangecnt - 1] = fd; 597 fdchanges [fdchangecnt - 1] = fd;
435} 598}
438evio_stop (struct ev_io *w) 601evio_stop (struct ev_io *w)
439{ 602{
440 if (!ev_is_active (w)) 603 if (!ev_is_active (w))
441 return; 604 return;
442 605
443 wlist_del ((struct ev_watcher_list **)&anfds[w->fd].head, (struct ev_watcher_list *)w); 606 wlist_del ((WL *)&anfds[w->fd].head, (WL)w);
444 ev_stop ((struct ev_watcher *)w); 607 ev_stop ((W)w);
445 608
446 ++fdchangecnt; 609 ++fdchangecnt;
447 array_needsize (fdchanges, fdchangemax, fdchangecnt, ); 610 array_needsize (fdchanges, fdchangemax, fdchangecnt, );
448 fdchanges [fdchangecnt - 1] = w->fd; 611 fdchanges [fdchangecnt - 1] = w->fd;
449} 612}
458 { 621 {
459 /* this formula differs from the one in timer_reify becuse we do not round up */ 622 /* this formula differs from the one in timer_reify becuse we do not round up */
460 if (w->repeat) 623 if (w->repeat)
461 w->at += ceil ((ev_now - w->at) / w->repeat) * w->repeat; 624 w->at += ceil ((ev_now - w->at) / w->repeat) * w->repeat;
462 625
463 ev_start ((struct ev_watcher *)w, ++atimercnt); 626 ev_start ((W)w, ++atimercnt);
464 array_needsize (atimers, atimermax, atimercnt, ); 627 array_needsize (atimers, atimermax, atimercnt, );
465 atimers [atimercnt - 1] = w; 628 atimers [atimercnt - 1] = w;
466 upheap (atimers, atimercnt - 1); 629 upheap (atimers, atimercnt - 1);
467 } 630 }
468 else 631 else
469 { 632 {
470 w->at += now; 633 w->at += now;
471 634
472 ev_start ((struct ev_watcher *)w, ++rtimercnt); 635 ev_start ((W)w, ++rtimercnt);
473 array_needsize (rtimers, rtimermax, rtimercnt, ); 636 array_needsize (rtimers, rtimermax, rtimercnt, );
474 rtimers [rtimercnt - 1] = w; 637 rtimers [rtimercnt - 1] = w;
475 upheap (rtimers, rtimercnt - 1); 638 upheap (rtimers, rtimercnt - 1);
476 } 639 }
477 640
498 rtimers [w->active - 1] = rtimers [rtimercnt]; 661 rtimers [w->active - 1] = rtimers [rtimercnt];
499 downheap (rtimers, rtimercnt, w->active - 1); 662 downheap (rtimers, rtimercnt, w->active - 1);
500 } 663 }
501 } 664 }
502 665
503 ev_stop ((struct ev_watcher *)w); 666 ev_stop ((W)w);
504} 667}
505 668
506void 669void
507evsignal_start (struct ev_signal *w) 670evsignal_start (struct ev_signal *w)
508{ 671{
509 if (ev_is_active (w)) 672 if (ev_is_active (w))
510 return; 673 return;
511 674
512 ev_start ((struct ev_watcher *)w, 1); 675 ev_start ((W)w, 1);
513 array_needsize (signals, signalmax, w->signum, signals_init); 676 array_needsize (signals, signalmax, w->signum, signals_init);
514 wlist_add ((struct ev_watcher_list **)&signals [w->signum - 1], (struct ev_watcher_list *)w); 677 wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w);
678
679 if (!w->next)
680 {
681 struct sigaction sa;
682 sa.sa_handler = sighandler;
683 sigfillset (&sa.sa_mask);
684 sa.sa_flags = 0;
685 sigaction (w->signum, &sa, 0);
686 }
515} 687}
516 688
517void 689void
518evsignal_stop (struct ev_signal *w) 690evsignal_stop (struct ev_signal *w)
519{ 691{
520 if (!ev_is_active (w)) 692 if (!ev_is_active (w))
521 return; 693 return;
522 694
523 wlist_del ((struct ev_watcher_list **)&signals [w->signum - 1], (struct ev_watcher_list *)w); 695 wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w);
524 ev_stop ((struct ev_watcher *)w); 696 ev_stop ((W)w);
525}
526 697
698 if (!signals [w->signum - 1].head)
699 signal (w->signum, SIG_DFL);
700}
701
702void evidle_start (struct ev_idle *w)
703{
704 if (ev_is_active (w))
705 return;
706
707 ev_start ((W)w, ++idlecnt);
708 array_needsize (idles, idlemax, idlecnt, );
709 idles [idlecnt - 1] = w;
710}
711
712void evidle_stop (struct ev_idle *w)
713{
714 idles [w->active - 1] = idles [--idlecnt];
715 ev_stop ((W)w);
716}
717
718void evcheck_start (struct ev_check *w)
719{
720 if (ev_is_active (w))
721 return;
722
723 ev_start ((W)w, ++checkcnt);
724 array_needsize (checks, checkmax, checkcnt, );
725 checks [checkcnt - 1] = w;
726}
727
728void evcheck_stop (struct ev_check *w)
729{
730 checks [w->active - 1] = checks [--checkcnt];
731 ev_stop ((W)w);
732}
733
527/*****************************************************************************/ 734/*****************************************************************************/
735
528#if 1 736#if 0
529 737
530static void 738static void
531sin_cb (struct ev_io *w, int revents) 739sin_cb (struct ev_io *w, int revents)
532{ 740{
533 fprintf (stderr, "sin %d, revents %d\n", w->fd, revents); 741 fprintf (stderr, "sin %d, revents %d\n", w->fd, revents);
539 //fprintf (stderr, "timer %f,%f (%x) (%f) d%p\n", w->at, w->repeat, revents, w->at - ev_time (), w->data); 747 //fprintf (stderr, "timer %f,%f (%x) (%f) d%p\n", w->at, w->repeat, revents, w->at - ev_time (), w->data);
540 evtimer_stop (w); 748 evtimer_stop (w);
541 evtimer_start (w); 749 evtimer_start (w);
542} 750}
543 751
752static void
753scb (struct ev_signal *w, int revents)
754{
755 fprintf (stderr, "signal %x,%d\n", revents, w->signum);
756}
757
758static void
759gcb (struct ev_signal *w, int revents)
760{
761 fprintf (stderr, "generic %x\n", revents);
762}
763
544int main (void) 764int main (void)
545{ 765{
546 struct ev_io sin; 766 struct ev_io sin;
547 767
548 ev_init (0); 768 ev_init (0);
551 evio_set (&sin, 0, EV_READ); 771 evio_set (&sin, 0, EV_READ);
552 evio_start (&sin); 772 evio_start (&sin);
553 773
554 struct ev_timer t[10000]; 774 struct ev_timer t[10000];
555 775
556#if 1 776#if 0
557 int i; 777 int i;
558 for (i = 0; i < 10000; ++i) 778 for (i = 0; i < 10000; ++i)
559 { 779 {
560 struct ev_timer *w = t + i; 780 struct ev_timer *w = t + i;
561 evw_init (w, ocb, i); 781 evw_init (w, ocb, i);
569 struct ev_timer t1; 789 struct ev_timer t1;
570 evw_init (&t1, ocb, 0); 790 evw_init (&t1, ocb, 0);
571 evtimer_set_abs (&t1, 5, 10); 791 evtimer_set_abs (&t1, 5, 10);
572 evtimer_start (&t1); 792 evtimer_start (&t1);
573 793
794 struct ev_signal sig;
795 evw_init (&sig, scb, 65535);
796 evsignal_set (&sig, SIGQUIT);
797 evsignal_start (&sig);
798
799 struct ev_check cw;
800 evw_init (&cw, gcb, 0);
801 evcheck_start (&cw);
802
803 struct ev_idle iw;
804 evw_init (&iw, gcb, 0);
805 evidle_start (&iw);
806
574 ev_loop (0); 807 ev_loop (0);
575 808
576 return 0; 809 return 0;
577} 810}
578 811

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines