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

Comparing libev/ev.c (file contents):
Revision 1.34 by root, Thu Nov 1 11:43:11 2007 UTC vs.
Revision 1.46 by root, Sat Nov 3 09:20:12 2007 UTC

1/* 1/*
2 * libev event processing core, watcher management
3 *
2 * Copyright (c) 2007 Marc Alexander Lehmann <libev@schmorp.de> 4 * Copyright (c) 2007 Marc Alexander Lehmann <libev@schmorp.de>
3 * All rights reserved. 5 * All rights reserved.
4 * 6 *
5 * Redistribution and use in source and binary forms, with or without 7 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are 8 * modification, are permitted provided that the following conditions are
40#include <stdio.h> 42#include <stdio.h>
41 43
42#include <assert.h> 44#include <assert.h>
43#include <errno.h> 45#include <errno.h>
44#include <sys/types.h> 46#include <sys/types.h>
47#ifndef WIN32
45#include <sys/wait.h> 48# include <sys/wait.h>
49#endif
46#include <sys/time.h> 50#include <sys/time.h>
47#include <time.h> 51#include <time.h>
48 52
53/**/
54
49#ifndef EV_USE_MONOTONIC 55#ifndef EV_USE_MONOTONIC
50# ifdef CLOCK_MONOTONIC
51# define EV_USE_MONOTONIC 1 56# define EV_USE_MONOTONIC 1
52# endif
53#endif 57#endif
54 58
55#ifndef EV_USE_SELECT 59#ifndef EV_USE_SELECT
56# define EV_USE_SELECT 1 60# define EV_USE_SELECT 1
57#endif 61#endif
58 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
59#ifndef EV_USE_EPOLL 67#ifndef EV_USE_EPOLL
60# define EV_USE_EPOLL 0 68# define EV_USE_EPOLL 0
61#endif 69#endif
62 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
63#ifndef CLOCK_REALTIME 86#ifndef CLOCK_REALTIME
87# undef EV_USE_REALTIME
64# define EV_USE_REALTIME 0 88# define EV_USE_REALTIME 0
65#endif 89#endif
66#ifndef EV_USE_REALTIME 90
67# define EV_USE_REALTIME 1 /* posix requirement, but might be slower */ 91/**/
68#endif
69 92
70#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) */
71#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) */
72#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 */
73#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 */
74 97
75#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)
76 113
77typedef struct ev_watcher *W; 114typedef struct ev_watcher *W;
78typedef struct ev_watcher_list *WL; 115typedef struct ev_watcher_list *WL;
79typedef struct ev_watcher_time *WT; 116typedef struct ev_watcher_time *WT;
80 117
81static ev_tstamp now, diff; /* monotonic clock */ 118static ev_tstamp now_floor, now, diff; /* monotonic clock */
82ev_tstamp ev_now; 119ev_tstamp ev_now;
83int ev_method; 120int ev_method;
84 121
85static int have_monotonic; /* runtime */ 122static int have_monotonic; /* runtime */
86 123
106 143
107static ev_tstamp 144static ev_tstamp
108get_clock (void) 145get_clock (void)
109{ 146{
110#if EV_USE_MONOTONIC 147#if EV_USE_MONOTONIC
111 if (have_monotonic) 148 if (expect_true (have_monotonic))
112 { 149 {
113 struct timespec ts; 150 struct timespec ts;
114 clock_gettime (CLOCK_MONOTONIC, &ts); 151 clock_gettime (CLOCK_MONOTONIC, &ts);
115 return ts.tv_sec + ts.tv_nsec * 1e-9; 152 return ts.tv_sec + ts.tv_nsec * 1e-9;
116 } 153 }
120} 157}
121 158
122#define array_roundsize(base,n) ((n) | 4 & ~3) 159#define array_roundsize(base,n) ((n) | 4 & ~3)
123 160
124#define array_needsize(base,cur,cnt,init) \ 161#define array_needsize(base,cur,cnt,init) \
125 if ((cnt) > cur) \ 162 if (expect_false ((cnt) > cur)) \
126 { \ 163 { \
127 int newcnt = cur; \ 164 int newcnt = cur; \
128 do \ 165 do \
129 { \ 166 { \
130 newcnt = array_roundsize (base, newcnt << 1); \ 167 newcnt = array_roundsize (base, newcnt << 1); \
165{ 202{
166 W w; 203 W w;
167 int events; 204 int events;
168} ANPENDING; 205} ANPENDING;
169 206
170static ANPENDING *pendings; 207static ANPENDING *pendings [NUMPRI];
171static int pendingmax, pendingcnt; 208static int pendingmax [NUMPRI], pendingcnt [NUMPRI];
172 209
173static void 210static void
174event (W w, int events) 211event (W w, int events)
175{ 212{
176 if (w->pending) 213 if (w->pending)
177 { 214 {
178 pendings [w->pending - 1].events |= events; 215 pendings [ABSPRI (w)][w->pending - 1].events |= events;
179 return; 216 return;
180 } 217 }
181 218
182 w->pending = ++pendingcnt; 219 w->pending = ++pendingcnt [ABSPRI (w)];
183 array_needsize (pendings, pendingmax, pendingcnt, ); 220 array_needsize (pendings [ABSPRI (w)], pendingmax [ABSPRI (w)], pendingcnt [ABSPRI (w)], );
184 pendings [pendingcnt - 1].w = w; 221 pendings [ABSPRI (w)][w->pending - 1].w = w;
185 pendings [pendingcnt - 1].events = events; 222 pendings [ABSPRI (w)][w->pending - 1].events = events;
186} 223}
187 224
188static void 225static void
189queue_events (W *events, int eventcnt, int type) 226queue_events (W *events, int eventcnt, int type)
190{ 227{
253 ++fdchangecnt; 290 ++fdchangecnt;
254 array_needsize (fdchanges, fdchangemax, fdchangecnt, ); 291 array_needsize (fdchanges, fdchangemax, fdchangecnt, );
255 fdchanges [fdchangecnt - 1] = fd; 292 fdchanges [fdchangecnt - 1] = fd;
256} 293}
257 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
258/* called on EBADF to verify fds */ 308/* called on EBADF to verify fds */
259static void 309static void
260fd_recheck (void) 310fd_ebadf (void)
261{ 311{
262 int fd; 312 int fd;
263 313
264 for (fd = 0; fd < anfdmax; ++fd) 314 for (fd = 0; fd < anfdmax; ++fd)
265 if (anfds [fd].events) 315 if (anfds [fd].events)
266 if (fcntl (fd, F_GETFD) == -1 && errno == EBADF) 316 if (fcntl (fd, F_GETFD) == -1 && errno == EBADF)
267 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)
268 { 328 {
269 ev_io_stop (anfds [fd].head); 329 close (fd);
270 event ((W)anfds [fd].head, EV_ERROR | EV_READ | EV_WRITE); 330 fd_kill (fd);
331 return;
271 } 332 }
272} 333}
273 334
274/*****************************************************************************/ 335/*****************************************************************************/
275 336
276static struct ev_timer **timers; 337static struct ev_timer **timers;
361 422
362static void 423static void
363sigcb (struct ev_io *iow, int revents) 424sigcb (struct ev_io *iow, int revents)
364{ 425{
365 struct ev_signal *w; 426 struct ev_signal *w;
366 int sig; 427 int signum;
367 428
368 read (sigpipe [0], &revents, 1); 429 read (sigpipe [0], &revents, 1);
369 gotsig = 0; 430 gotsig = 0;
370 431
371 for (sig = signalmax; sig--; ) 432 for (signum = signalmax; signum--; )
372 if (signals [sig].gotsig) 433 if (signals [signum].gotsig)
373 { 434 {
374 signals [sig].gotsig = 0; 435 signals [signum].gotsig = 0;
375 436
376 for (w = signals [sig].head; w; w = w->next) 437 for (w = signals [signum].head; w; w = w->next)
377 event ((W)w, EV_SIGNAL); 438 event ((W)w, EV_SIGNAL);
378 } 439 }
379} 440}
380 441
381static void 442static void
382siginit (void) 443siginit (void)
383{ 444{
445#ifndef WIN32
384 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC); 446 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
385 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC); 447 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
386 448
387 /* rather than sort out wether we really need nb, set it */ 449 /* rather than sort out wether we really need nb, set it */
388 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK); 450 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
389 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK); 451 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
452#endif
390 453
391 ev_io_set (&sigev, sigpipe [0], EV_READ); 454 ev_io_set (&sigev, sigpipe [0], EV_READ);
392 ev_io_start (&sigev); 455 ev_io_start (&sigev);
393} 456}
394 457
406/*****************************************************************************/ 469/*****************************************************************************/
407 470
408static struct ev_child *childs [PID_HASHSIZE]; 471static struct ev_child *childs [PID_HASHSIZE];
409static struct ev_signal childev; 472static struct ev_signal childev;
410 473
474#ifndef WIN32
475
411#ifndef WCONTINUED 476#ifndef WCONTINUED
412# define WCONTINUED 0 477# define WCONTINUED 0
413#endif 478#endif
414 479
415static void 480static void
418 struct ev_child *w; 483 struct ev_child *w;
419 int pid, status; 484 int pid, status;
420 485
421 while ((pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)) != -1) 486 while ((pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)) != -1)
422 for (w = childs [pid & (PID_HASHSIZE - 1)]; w; w = w->next) 487 for (w = childs [pid & (PID_HASHSIZE - 1)]; w; w = w->next)
423 if (w->pid == pid || w->pid == -1) 488 if (w->pid == pid || !w->pid)
424 { 489 {
490 w->rpid = pid;
425 w->status = status; 491 w->rstatus = status;
426 event ((W)w, EV_CHILD); 492 event ((W)w, EV_CHILD);
427 } 493 }
428} 494}
429 495
430/*****************************************************************************/ 496#endif
431 497
498/*****************************************************************************/
499
500#if EV_USE_KQUEUE
501# include "ev_kqueue.c"
502#endif
432#if EV_USE_EPOLL 503#if EV_USE_EPOLL
433# include "ev_epoll.c" 504# include "ev_epoll.c"
434#endif 505#endif
506#if EV_USE_POLL
507# include "ev_poll.c"
508#endif
435#if EV_USE_SELECT 509#if EV_USE_SELECT
436# include "ev_select.c" 510# include "ev_select.c"
437#endif 511#endif
438 512
439int 513int
446ev_version_minor (void) 520ev_version_minor (void)
447{ 521{
448 return EV_VERSION_MINOR; 522 return EV_VERSION_MINOR;
449} 523}
450 524
525/* return true if we are running with elevated privileges and ignore env variables */
526static int
527enable_secure ()
528{
529 return getuid () != geteuid ()
530 || getgid () != getegid ();
531}
532
451int ev_init (int flags) 533int ev_init (int methods)
452{ 534{
453 if (!ev_method) 535 if (!ev_method)
454 { 536 {
455#if EV_USE_MONOTONIC 537#if EV_USE_MONOTONIC
456 { 538 {
458 if (!clock_gettime (CLOCK_MONOTONIC, &ts)) 540 if (!clock_gettime (CLOCK_MONOTONIC, &ts))
459 have_monotonic = 1; 541 have_monotonic = 1;
460 } 542 }
461#endif 543#endif
462 544
463 ev_now = ev_time (); 545 ev_now = ev_time ();
464 now = get_clock (); 546 now = get_clock ();
547 now_floor = now;
465 diff = ev_now - now; 548 diff = ev_now - now;
466 549
467 if (pipe (sigpipe)) 550 if (pipe (sigpipe))
468 return 0; 551 return 0;
469 552
553 if (methods == EVMETHOD_AUTO)
554 if (!enable_secure () && getenv ("LIBEV_METHODS"))
555 methods = atoi (getenv ("LIBEV_METHODS"));
556 else
470 ev_method = EVMETHOD_NONE; 557 methods = EVMETHOD_ANY;
558
559 ev_method = 0;
560#if EV_USE_KQUEUE
561 if (!ev_method && (methods & EVMETHOD_KQUEUE)) kqueue_init (methods);
562#endif
471#if EV_USE_EPOLL 563#if EV_USE_EPOLL
472 if (ev_method == EVMETHOD_NONE) epoll_init (flags); 564 if (!ev_method && (methods & EVMETHOD_EPOLL )) epoll_init (methods);
565#endif
566#if EV_USE_POLL
567 if (!ev_method && (methods & EVMETHOD_POLL )) poll_init (methods);
473#endif 568#endif
474#if EV_USE_SELECT 569#if EV_USE_SELECT
475 if (ev_method == EVMETHOD_NONE) select_init (flags); 570 if (!ev_method && (methods & EVMETHOD_SELECT)) select_init (methods);
476#endif 571#endif
477 572
478 if (ev_method) 573 if (ev_method)
479 { 574 {
480 ev_watcher_init (&sigev, sigcb); 575 ev_watcher_init (&sigev, sigcb);
481 siginit (); 576 siginit ();
482 577
578#ifndef WIN32
483 ev_signal_init (&childev, childcb, SIGCHLD); 579 ev_signal_init (&childev, childcb, SIGCHLD);
484 ev_signal_start (&childev); 580 ev_signal_start (&childev);
581#endif
485 } 582 }
486 } 583 }
487 584
488 return ev_method; 585 return ev_method;
489} 586}
490 587
491/*****************************************************************************/ 588/*****************************************************************************/
492 589
493void 590void
494ev_prefork (void) 591ev_fork_prepare (void)
495{ 592{
496 /* nop */ 593 /* nop */
497} 594}
498 595
499void 596void
500ev_postfork_parent (void) 597ev_fork_parent (void)
501{ 598{
502 /* nop */ 599 /* nop */
503} 600}
504 601
505void 602void
506ev_postfork_child (void) 603ev_fork_child (void)
507{ 604{
508#if EV_USE_EPOLL 605#if EV_USE_EPOLL
509 if (ev_method == EVMETHOD_EPOLL) 606 if (ev_method == EVMETHOD_EPOLL)
510 epoll_postfork_child (); 607 epoll_postfork_child ();
511#endif 608#endif
520/*****************************************************************************/ 617/*****************************************************************************/
521 618
522static void 619static void
523call_pending (void) 620call_pending (void)
524{ 621{
622 int pri;
623
624 for (pri = NUMPRI; pri--; )
525 while (pendingcnt) 625 while (pendingcnt [pri])
526 { 626 {
527 ANPENDING *p = pendings + --pendingcnt; 627 ANPENDING *p = pendings [pri] + --pendingcnt [pri];
528 628
529 if (p->w) 629 if (p->w)
530 { 630 {
531 p->w->pending = 0; 631 p->w->pending = 0;
532 p->w->cb (p->w, p->events); 632 p->w->cb (p->w, p->events);
533 } 633 }
534 } 634 }
535} 635}
536 636
537static void 637static void
538timers_reify (void) 638timers_reify (void)
539{ 639{
599 } 699 }
600 } 700 }
601 } 701 }
602} 702}
603 703
704static int
705time_update_monotonic (void)
706{
707 now = get_clock ();
708
709 if (expect_true (now - now_floor < MIN_TIMEJUMP * .5))
710 {
711 ev_now = now + diff;
712 return 0;
713 }
714 else
715 {
716 now_floor = now;
717 ev_now = ev_time ();
718 return 1;
719 }
720}
721
604static void 722static void
605time_update (void) 723time_update (void)
606{ 724{
607 int i; 725 int i;
608 726
609 ev_now = ev_time (); 727#if EV_USE_MONOTONIC
610
611 if (have_monotonic) 728 if (expect_true (have_monotonic))
612 { 729 {
730 if (time_update_monotonic ())
731 {
613 ev_tstamp odiff = diff; 732 ev_tstamp odiff = diff;
614 733
615 for (i = 4; --i; ) /* loop a few times, before making important decisions */ 734 for (i = 4; --i; ) /* loop a few times, before making important decisions */
616 { 735 {
617 now = get_clock ();
618 diff = ev_now - now; 736 diff = ev_now - now;
619 737
620 if (fabs (odiff - diff) < MIN_TIMEJUMP) 738 if (fabs (odiff - diff) < MIN_TIMEJUMP)
621 return; /* all is well */ 739 return; /* all is well */
622 740
623 ev_now = ev_time (); 741 ev_now = ev_time ();
742 now = get_clock ();
743 now_floor = now;
624 } 744 }
625 745
626 periodics_reschedule (diff - odiff); 746 periodics_reschedule (diff - odiff);
627 /* no timer adjustment, as the monotonic clock doesn't jump */ 747 /* no timer adjustment, as the monotonic clock doesn't jump */
748 }
628 } 749 }
629 else 750 else
751#endif
630 { 752 {
753 ev_now = ev_time ();
754
631 if (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP) 755 if (expect_false (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP))
632 { 756 {
633 periodics_reschedule (ev_now - now); 757 periodics_reschedule (ev_now - now);
634 758
635 /* adjust timers. this is easy, as the offset is the same for all */ 759 /* adjust timers. this is easy, as the offset is the same for all */
636 for (i = 0; i < timercnt; ++i) 760 for (i = 0; i < timercnt; ++i)
649 ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0; 773 ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0;
650 774
651 do 775 do
652 { 776 {
653 /* queue check watchers (and execute them) */ 777 /* queue check watchers (and execute them) */
654 if (preparecnt) 778 if (expect_false (preparecnt))
655 { 779 {
656 queue_events ((W *)prepares, preparecnt, EV_PREPARE); 780 queue_events ((W *)prepares, preparecnt, EV_PREPARE);
657 call_pending (); 781 call_pending ();
658 } 782 }
659 783
662 786
663 /* calculate blocking time */ 787 /* calculate blocking time */
664 788
665 /* we only need this for !monotonic clockor timers, but as we basically 789 /* we only need this for !monotonic clockor timers, but as we basically
666 always have timers, we just calculate it always */ 790 always have timers, we just calculate it always */
791#if EV_USE_MONOTONIC
792 if (expect_true (have_monotonic))
793 time_update_monotonic ();
794 else
795#endif
796 {
667 ev_now = ev_time (); 797 ev_now = ev_time ();
798 now = ev_now;
799 }
668 800
669 if (flags & EVLOOP_NONBLOCK || idlecnt) 801 if (flags & EVLOOP_NONBLOCK || idlecnt)
670 block = 0.; 802 block = 0.;
671 else 803 else
672 { 804 {
673 block = MAX_BLOCKTIME; 805 block = MAX_BLOCKTIME;
674 806
675 if (timercnt) 807 if (timercnt)
676 { 808 {
677 ev_tstamp to = timers [0]->at - (have_monotonic ? get_clock () : ev_now) + method_fudge; 809 ev_tstamp to = timers [0]->at - now + method_fudge;
678 if (block > to) block = to; 810 if (block > to) block = to;
679 } 811 }
680 812
681 if (periodiccnt) 813 if (periodiccnt)
682 { 814 {
739static void 871static void
740ev_clear_pending (W w) 872ev_clear_pending (W w)
741{ 873{
742 if (w->pending) 874 if (w->pending)
743 { 875 {
744 pendings [w->pending - 1].w = 0; 876 pendings [ABSPRI (w)][w->pending - 1].w = 0;
745 w->pending = 0; 877 w->pending = 0;
746 } 878 }
747} 879}
748 880
749static void 881static void
750ev_start (W w, int active) 882ev_start (W w, int active)
751{ 883{
884 if (w->priority < EV_MINPRI) w->priority = EV_MINPRI;
885 if (w->priority > EV_MAXPRI) w->priority = EV_MAXPRI;
886
752 w->active = active; 887 w->active = active;
753} 888}
754 889
755static void 890static void
756ev_stop (W w) 891ev_stop (W w)
761/*****************************************************************************/ 896/*****************************************************************************/
762 897
763void 898void
764ev_io_start (struct ev_io *w) 899ev_io_start (struct ev_io *w)
765{ 900{
901 int fd = w->fd;
902
766 if (ev_is_active (w)) 903 if (ev_is_active (w))
767 return; 904 return;
768
769 int fd = w->fd;
770 905
771 assert (("ev_io_start called with negative fd", fd >= 0)); 906 assert (("ev_io_start called with negative fd", fd >= 0));
772 907
773 ev_start ((W)w, 1); 908 ev_start ((W)w, 1);
774 array_needsize (anfds, anfdmax, fd + 1, anfds_init); 909 array_needsize (anfds, anfdmax, fd + 1, anfds_init);

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines