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

Comparing libev/ev.c (file contents):
Revision 1.39 by root, Thu Nov 1 17:17:32 2007 UTC vs.
Revision 1.45 by root, Sat Nov 3 09:19:58 2007 UTC

42#include <stdio.h> 42#include <stdio.h>
43 43
44#include <assert.h> 44#include <assert.h>
45#include <errno.h> 45#include <errno.h>
46#include <sys/types.h> 46#include <sys/types.h>
47#ifndef WIN32
47#include <sys/wait.h> 48# include <sys/wait.h>
49#endif
48#include <sys/time.h> 50#include <sys/time.h>
49#include <time.h> 51#include <time.h>
50 52
53/**/
54
51#ifndef EV_USE_MONOTONIC 55#ifndef EV_USE_MONOTONIC
52# define EV_USE_MONOTONIC 1 56# define EV_USE_MONOTONIC 1
53#endif 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/**/
54 80
55#ifndef CLOCK_MONOTONIC 81#ifndef CLOCK_MONOTONIC
56# undef EV_USE_MONOTONIC 82# undef EV_USE_MONOTONIC
57# define EV_USE_MONOTONIC 0 83# define EV_USE_MONOTONIC 0
58#endif 84#endif
59 85
60#ifndef EV_USE_SELECT
61# define EV_USE_SELECT 1
62#endif
63
64#ifndef EV_USE_EPOLL
65# define EV_USE_EPOLL 0
66#endif
67
68#ifndef CLOCK_REALTIME 86#ifndef CLOCK_REALTIME
87# undef EV_USE_REALTIME
69# define EV_USE_REALTIME 0 88# define EV_USE_REALTIME 0
70#endif 89#endif
71#ifndef EV_USE_REALTIME 90
72# define EV_USE_REALTIME 1 /* posix requirement, but might be slower */ 91/**/
73#endif
74 92
75#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */ 93#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */
76#define MAX_BLOCKTIME 59.731 /* never wait longer than this time (to detetc time jumps) */ 94#define MAX_BLOCKTIME 59.731 /* never wait longer than this time (to detect time jumps) */
77#define PID_HASHSIZE 16 /* size of pid hash table, must be power of two */ 95#define PID_HASHSIZE 16 /* size of pid hash table, must be power of two */
78#define CLEANUP_INTERVAL (MAX_BLOCKTIME * 5.) /* how often to try to free memory and re-check fds */ 96/*#define CLEANUP_INTERVAL 300. /* how often to try to free memory and re-check fds */
79 97
80#include "ev.h" 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)
81 113
82typedef struct ev_watcher *W; 114typedef struct ev_watcher *W;
83typedef struct ev_watcher_list *WL; 115typedef struct ev_watcher_list *WL;
84typedef struct ev_watcher_time *WT; 116typedef struct ev_watcher_time *WT;
85 117
86static ev_tstamp now, diff; /* monotonic clock */ 118static ev_tstamp now_floor, now, diff; /* monotonic clock */
87ev_tstamp ev_now; 119ev_tstamp ev_now;
88int ev_method; 120int ev_method;
89 121
90static int have_monotonic; /* runtime */ 122static int have_monotonic; /* runtime */
91 123
111 143
112static ev_tstamp 144static ev_tstamp
113get_clock (void) 145get_clock (void)
114{ 146{
115#if EV_USE_MONOTONIC 147#if EV_USE_MONOTONIC
116 if (have_monotonic) 148 if (expect_true (have_monotonic))
117 { 149 {
118 struct timespec ts; 150 struct timespec ts;
119 clock_gettime (CLOCK_MONOTONIC, &ts); 151 clock_gettime (CLOCK_MONOTONIC, &ts);
120 return ts.tv_sec + ts.tv_nsec * 1e-9; 152 return ts.tv_sec + ts.tv_nsec * 1e-9;
121 } 153 }
125} 157}
126 158
127#define array_roundsize(base,n) ((n) | 4 & ~3) 159#define array_roundsize(base,n) ((n) | 4 & ~3)
128 160
129#define array_needsize(base,cur,cnt,init) \ 161#define array_needsize(base,cur,cnt,init) \
130 if ((cnt) > cur) \ 162 if (expect_false ((cnt) > cur)) \
131 { \ 163 { \
132 int newcnt = cur; \ 164 int newcnt = cur; \
133 do \ 165 do \
134 { \ 166 { \
135 newcnt = array_roundsize (base, newcnt << 1); \ 167 newcnt = array_roundsize (base, newcnt << 1); \
170{ 202{
171 W w; 203 W w;
172 int events; 204 int events;
173} ANPENDING; 205} ANPENDING;
174 206
175static ANPENDING *pendings; 207static ANPENDING *pendings [NUMPRI];
176static int pendingmax, pendingcnt; 208static int pendingmax [NUMPRI], pendingcnt [NUMPRI];
177 209
178static void 210static void
179event (W w, int events) 211event (W w, int events)
180{ 212{
181 if (w->pending) 213 if (w->pending)
182 { 214 {
183 pendings [w->pending - 1].events |= events; 215 pendings [ABSPRI (w)][w->pending - 1].events |= events;
184 return; 216 return;
185 } 217 }
186 218
187 w->pending = ++pendingcnt; 219 w->pending = ++pendingcnt [ABSPRI (w)];
188 array_needsize (pendings, pendingmax, pendingcnt, ); 220 array_needsize (pendings [ABSPRI (w)], pendingmax [ABSPRI (w)], pendingcnt [ABSPRI (w)], );
189 pendings [pendingcnt - 1].w = w; 221 pendings [ABSPRI (w)][w->pending - 1].w = w;
190 pendings [pendingcnt - 1].events = events; 222 pendings [ABSPRI (w)][w->pending - 1].events = events;
191} 223}
192 224
193static void 225static void
194queue_events (W *events, int eventcnt, int type) 226queue_events (W *events, int eventcnt, int type)
195{ 227{
258 ++fdchangecnt; 290 ++fdchangecnt;
259 array_needsize (fdchanges, fdchangemax, fdchangecnt, ); 291 array_needsize (fdchanges, fdchangemax, fdchangecnt, );
260 fdchanges [fdchangecnt - 1] = fd; 292 fdchanges [fdchangecnt - 1] = fd;
261} 293}
262 294
295static void
296fd_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
263/* called on EBADF to verify fds */ 308/* called on EBADF to verify fds */
264static void 309static void
265fd_recheck (void) 310fd_ebadf (void)
266{ 311{
267 int fd; 312 int fd;
268 313
269 for (fd = 0; fd < anfdmax; ++fd) 314 for (fd = 0; fd < anfdmax; ++fd)
270 if (anfds [fd].events) 315 if (anfds [fd].events)
271 if (fcntl (fd, F_GETFD) == -1 && errno == EBADF) 316 if (fcntl (fd, F_GETFD) == -1 && errno == EBADF)
272 while (anfds [fd].head) 317 fd_kill (fd);
318}
319
320/* called on ENOMEM in select/poll to kill some fds and retry */
321static void
322fd_enomem (void)
323{
324 int fd = anfdmax;
325
326 while (fd--)
327 if (anfds [fd].events)
273 { 328 {
274 ev_io_stop (anfds [fd].head); 329 close (fd);
275 event ((W)anfds [fd].head, EV_ERROR | EV_READ | EV_WRITE); 330 fd_kill (fd);
331 return;
276 } 332 }
277} 333}
278 334
279/*****************************************************************************/ 335/*****************************************************************************/
280 336
281static struct ev_timer **timers; 337static struct ev_timer **timers;
384} 440}
385 441
386static void 442static void
387siginit (void) 443siginit (void)
388{ 444{
445#ifndef WIN32
389 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC); 446 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
390 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC); 447 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
391 448
392 /* rather than sort out wether we really need nb, set it */ 449 /* rather than sort out wether we really need nb, set it */
393 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK); 450 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
394 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK); 451 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
452#endif
395 453
396 ev_io_set (&sigev, sigpipe [0], EV_READ); 454 ev_io_set (&sigev, sigpipe [0], EV_READ);
397 ev_io_start (&sigev); 455 ev_io_start (&sigev);
398} 456}
399 457
410 468
411/*****************************************************************************/ 469/*****************************************************************************/
412 470
413static struct ev_child *childs [PID_HASHSIZE]; 471static struct ev_child *childs [PID_HASHSIZE];
414static struct ev_signal childev; 472static struct ev_signal childev;
473
474#ifndef WIN32
415 475
416#ifndef WCONTINUED 476#ifndef WCONTINUED
417# define WCONTINUED 0 477# define WCONTINUED 0
418#endif 478#endif
419 479
430 w->status = status; 490 w->status = status;
431 event ((W)w, EV_CHILD); 491 event ((W)w, EV_CHILD);
432 } 492 }
433} 493}
434 494
435/*****************************************************************************/ 495#endif
436 496
497/*****************************************************************************/
498
499#if EV_USE_KQUEUE
500# include "ev_kqueue.c"
501#endif
437#if EV_USE_EPOLL 502#if EV_USE_EPOLL
438# include "ev_epoll.c" 503# include "ev_epoll.c"
439#endif 504#endif
505#if EV_USE_POLL
506# include "ev_poll.c"
507#endif
440#if EV_USE_SELECT 508#if EV_USE_SELECT
441# include "ev_select.c" 509# include "ev_select.c"
442#endif 510#endif
443 511
444int 512int
451ev_version_minor (void) 519ev_version_minor (void)
452{ 520{
453 return EV_VERSION_MINOR; 521 return EV_VERSION_MINOR;
454} 522}
455 523
524/* return true if we are running with elevated privileges and ignore env variables */
525static int
526enable_secure ()
527{
528 return getuid () != geteuid ()
529 || getgid () != getegid ();
530}
531
456int ev_init (int flags) 532int ev_init (int methods)
457{ 533{
458 if (!ev_method) 534 if (!ev_method)
459 { 535 {
460#if EV_USE_MONOTONIC 536#if EV_USE_MONOTONIC
461 { 537 {
463 if (!clock_gettime (CLOCK_MONOTONIC, &ts)) 539 if (!clock_gettime (CLOCK_MONOTONIC, &ts))
464 have_monotonic = 1; 540 have_monotonic = 1;
465 } 541 }
466#endif 542#endif
467 543
468 ev_now = ev_time (); 544 ev_now = ev_time ();
469 now = get_clock (); 545 now = get_clock ();
546 now_floor = now;
470 diff = ev_now - now; 547 diff = ev_now - now;
471 548
472 if (pipe (sigpipe)) 549 if (pipe (sigpipe))
473 return 0; 550 return 0;
474 551
552 if (methods == EVMETHOD_AUTO)
553 if (!enable_secure () && getenv ("LIBEV_METHODS"))
554 methods = atoi (getenv ("LIBEV_METHODS"));
555 else
475 ev_method = EVMETHOD_NONE; 556 methods = EVMETHOD_ANY;
557
558 ev_method = 0;
559#if EV_USE_KQUEUE
560 if (!ev_method && (methods & EVMETHOD_KQUEUE)) kqueue_init (methods);
561#endif
476#if EV_USE_EPOLL 562#if EV_USE_EPOLL
477 if (ev_method == EVMETHOD_NONE) epoll_init (flags); 563 if (!ev_method && (methods & EVMETHOD_EPOLL )) epoll_init (methods);
564#endif
565#if EV_USE_POLL
566 if (!ev_method && (methods & EVMETHOD_POLL )) poll_init (methods);
478#endif 567#endif
479#if EV_USE_SELECT 568#if EV_USE_SELECT
480 if (ev_method == EVMETHOD_NONE) select_init (flags); 569 if (!ev_method && (methods & EVMETHOD_SELECT)) select_init (methods);
481#endif 570#endif
482 571
483 if (ev_method) 572 if (ev_method)
484 { 573 {
485 ev_watcher_init (&sigev, sigcb); 574 ev_watcher_init (&sigev, sigcb);
486 siginit (); 575 siginit ();
487 576
577#ifndef WIN32
488 ev_signal_init (&childev, childcb, SIGCHLD); 578 ev_signal_init (&childev, childcb, SIGCHLD);
489 ev_signal_start (&childev); 579 ev_signal_start (&childev);
580#endif
490 } 581 }
491 } 582 }
492 583
493 return ev_method; 584 return ev_method;
494} 585}
525/*****************************************************************************/ 616/*****************************************************************************/
526 617
527static void 618static void
528call_pending (void) 619call_pending (void)
529{ 620{
621 int pri;
622
623 for (pri = NUMPRI; pri--; )
530 while (pendingcnt) 624 while (pendingcnt [pri])
531 { 625 {
532 ANPENDING *p = pendings + --pendingcnt; 626 ANPENDING *p = pendings [pri] + --pendingcnt [pri];
533 627
534 if (p->w) 628 if (p->w)
535 { 629 {
536 p->w->pending = 0; 630 p->w->pending = 0;
537 p->w->cb (p->w, p->events); 631 p->w->cb (p->w, p->events);
538 } 632 }
539 } 633 }
540} 634}
541 635
542static void 636static void
543timers_reify (void) 637timers_reify (void)
544{ 638{
604 } 698 }
605 } 699 }
606 } 700 }
607} 701}
608 702
703static int
704time_update_monotonic (void)
705{
706 now = get_clock ();
707
708 if (expect_true (now - now_floor < MIN_TIMEJUMP * .5))
709 {
710 ev_now = now + diff;
711 return 0;
712 }
713 else
714 {
715 now_floor = now;
716 ev_now = ev_time ();
717 return 1;
718 }
719}
720
609static void 721static void
610time_update (void) 722time_update (void)
611{ 723{
612 int i; 724 int i;
613 725
614 ev_now = ev_time (); 726#if EV_USE_MONOTONIC
615
616 if (have_monotonic) 727 if (expect_true (have_monotonic))
617 { 728 {
729 if (time_update_monotonic ())
730 {
618 ev_tstamp odiff = diff; 731 ev_tstamp odiff = diff;
619 732
620 for (i = 4; --i; ) /* loop a few times, before making important decisions */ 733 for (i = 4; --i; ) /* loop a few times, before making important decisions */
621 { 734 {
622 now = get_clock ();
623 diff = ev_now - now; 735 diff = ev_now - now;
624 736
625 if (fabs (odiff - diff) < MIN_TIMEJUMP) 737 if (fabs (odiff - diff) < MIN_TIMEJUMP)
626 return; /* all is well */ 738 return; /* all is well */
627 739
628 ev_now = ev_time (); 740 ev_now = ev_time ();
741 now = get_clock ();
742 now_floor = now;
629 } 743 }
630 744
631 periodics_reschedule (diff - odiff); 745 periodics_reschedule (diff - odiff);
632 /* no timer adjustment, as the monotonic clock doesn't jump */ 746 /* no timer adjustment, as the monotonic clock doesn't jump */
747 }
633 } 748 }
634 else 749 else
750#endif
635 { 751 {
752 ev_now = ev_time ();
753
636 if (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP) 754 if (expect_false (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP))
637 { 755 {
638 periodics_reschedule (ev_now - now); 756 periodics_reschedule (ev_now - now);
639 757
640 /* adjust timers. this is easy, as the offset is the same for all */ 758 /* adjust timers. this is easy, as the offset is the same for all */
641 for (i = 0; i < timercnt; ++i) 759 for (i = 0; i < timercnt; ++i)
654 ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0; 772 ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0;
655 773
656 do 774 do
657 { 775 {
658 /* queue check watchers (and execute them) */ 776 /* queue check watchers (and execute them) */
659 if (preparecnt) 777 if (expect_false (preparecnt))
660 { 778 {
661 queue_events ((W *)prepares, preparecnt, EV_PREPARE); 779 queue_events ((W *)prepares, preparecnt, EV_PREPARE);
662 call_pending (); 780 call_pending ();
663 } 781 }
664 782
667 785
668 /* calculate blocking time */ 786 /* calculate blocking time */
669 787
670 /* we only need this for !monotonic clockor timers, but as we basically 788 /* we only need this for !monotonic clockor timers, but as we basically
671 always have timers, we just calculate it always */ 789 always have timers, we just calculate it always */
790#if EV_USE_MONOTONIC
791 if (expect_true (have_monotonic))
792 time_update_monotonic ();
793 else
794#endif
795 {
672 ev_now = ev_time (); 796 ev_now = ev_time ();
797 now = ev_now;
798 }
673 799
674 if (flags & EVLOOP_NONBLOCK || idlecnt) 800 if (flags & EVLOOP_NONBLOCK || idlecnt)
675 block = 0.; 801 block = 0.;
676 else 802 else
677 { 803 {
678 block = MAX_BLOCKTIME; 804 block = MAX_BLOCKTIME;
679 805
680 if (timercnt) 806 if (timercnt)
681 { 807 {
682 ev_tstamp to = timers [0]->at - (have_monotonic ? get_clock () : ev_now) + method_fudge; 808 ev_tstamp to = timers [0]->at - now + method_fudge;
683 if (block > to) block = to; 809 if (block > to) block = to;
684 } 810 }
685 811
686 if (periodiccnt) 812 if (periodiccnt)
687 { 813 {
744static void 870static void
745ev_clear_pending (W w) 871ev_clear_pending (W w)
746{ 872{
747 if (w->pending) 873 if (w->pending)
748 { 874 {
749 pendings [w->pending - 1].w = 0; 875 pendings [ABSPRI (w)][w->pending - 1].w = 0;
750 w->pending = 0; 876 w->pending = 0;
751 } 877 }
752} 878}
753 879
754static void 880static void
755ev_start (W w, int active) 881ev_start (W w, int active)
756{ 882{
883 if (w->priority < EV_MINPRI) w->priority = EV_MINPRI;
884 if (w->priority > EV_MAXPRI) w->priority = EV_MAXPRI;
885
757 w->active = active; 886 w->active = active;
758} 887}
759 888
760static void 889static void
761ev_stop (W w) 890ev_stop (W w)

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines