ev.c

   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       gotsig = 1;
 419       write (sigpipe [1], &signum, 1);
 420     }
 421 }
 422
 423 static void
 424 sigcb (struct ev_io *iow, int revents)
 425 {
 426   struct ev_signal *w;
 427   int signum;
 428
 429   read (sigpipe [0], &revents, 1);
 430   gotsig = 0;
 431
 432   for (signum = signalmax; signum--; )
 433     if (signals [signum].gotsig)
 434       {
 435         signals [signum].gotsig = 0;
 436
 437         for (w = signals [signum].head; w; w = w->next)
 438           event ((W)w, EV_SIGNAL);
 439       }
 440 }
 441
 442 static void
 443 siginit (void)
 444 {
 445 #ifndef WIN32
 446   fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
 447   fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
 448
 449   /* rather than sort out wether we really need nb, set it */
 450   fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
 451   fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
 452 #endif
 453
 454   ev_io_set (&sigev, sigpipe [0], EV_READ);
 455   ev_io_start (&sigev);
 456 }
 457
 458 /*****************************************************************************/
 459
 460 static struct ev_idle **idles;
 461 static int idlemax, idlecnt;
 462
 463 static struct ev_prepare **prepares;
 464 static int preparemax, preparecnt;
 465
 466 static struct ev_check **checks;
 467 static int checkmax, checkcnt;
 468
 469 /*****************************************************************************/
 470
 471 static struct ev_child *childs [PID_HASHSIZE];
 472 static struct ev_signal childev;
 473
 474 #ifndef WIN32
 475
 476 #ifndef WCONTINUED
 477 # define WCONTINUED 0
 478 #endif
 479
 480 static void
 481 childcb (struct ev_signal *sw, int revents)
 482 {
 483   struct ev_child *w;
 484   int pid, status;
 485
 486   while ((pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)) != -1)
 487     for (w = childs [pid & (PID_HASHSIZE - 1)]; w; w = w->next)
 488       if (w->pid == pid || !w->pid)
 489         {
 490           w->rpid    = pid;
 491           w->rstatus = status;
 492           event ((W)w, EV_CHILD);
 493         }
 494 }
 495
 496 #endif
 497
 498 /*****************************************************************************/
 499
 500 #if EV_USE_KQUEUE
 501 # include "ev_kqueue.c"
 502 #endif
 503 #if EV_USE_EPOLL
 504 # include "ev_epoll.c"
 505 #endif
 506 #if EV_USE_POLL
 507 # include "ev_poll.c"
 508 #endif
 509 #if EV_USE_SELECT
 510 # include "ev_select.c"
 511 #endif
 512
 513 int
 514 ev_version_major (void)
 515 {
 516   return EV_VERSION_MAJOR;
 517 }
 518
 519 int
 520 ev_version_minor (void)
 521 {
 522   return EV_VERSION_MINOR;
 523 }
 524
 525 /* return true if we are running with elevated privileges and ignore env variables */
 526 static int
 527 enable_secure ()
 528 {
 529   return getuid () != geteuid ()
 530       || getgid () != getegid ();
 531 }
 532
 533 int ev_init (int methods)
 534 {
 535   if (!ev_method)
 536     {
 537 #if EV_USE_MONOTONIC
 538       {
 539         struct timespec ts;
 540         if (!clock_gettime (CLOCK_MONOTONIC, &ts))
 541           have_monotonic = 1;
 542       }
 543 #endif
 544
 545       ev_now    = ev_time ();
 546       now       = get_clock ();
 547       now_floor = now;
 548       diff      = ev_now - now;
 549
 550       if (pipe (sigpipe))
 551         return 0;
 552
 553       if (methods == EVMETHOD_AUTO)
 554           if (!enable_secure () && getenv ("LIBEV_METHODS"))
 555             methods = atoi (getenv ("LIBEV_METHODS"));
 556           else
 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
 563 #if EV_USE_EPOLL
 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);
 568 #endif
 569 #if EV_USE_SELECT
 570       if (!ev_method && (methods & EVMETHOD_SELECT)) select_init (methods);
 571 #endif
 572
 573       if (ev_method)
 574         {
 575           ev_watcher_init (&sigev, sigcb);
 576           siginit ();
 577
 578 #ifndef WIN32
 579           ev_signal_init (&childev, childcb, SIGCHLD);
 580           ev_signal_start (&childev);
 581 #endif
 582         }
 583     }
 584
 585   return ev_method;
 586 }
 587
 588 /*****************************************************************************/
 589
 590 void
 591 ev_fork_prepare (void)
 592 {
 593   /* nop */
 594 }
 595
 596 void
 597 ev_fork_parent (void)
 598 {
 599   /* nop */
 600 }
 601
 602 void
 603 ev_fork_child (void)
 604 {
 605 #if EV_USE_EPOLL
 606   if (ev_method == EVMETHOD_EPOLL)
 607     epoll_postfork_child ();
 608 #endif
 609
 610   ev_io_stop (&sigev);
 611   close (sigpipe [0]);
 612   close (sigpipe [1]);
 613   pipe (sigpipe);
 614   siginit ();
 615 }
 616
 617 /*****************************************************************************/
 618
 619 static void
 620 call_pending (void)
 621 {
 622   int pri;
 623
 624   for (pri = NUMPRI; pri--; )
 625     while (pendingcnt [pri])
 626       {
 627         ANPENDING *p = pendings [pri] + --pendingcnt [pri];
 628
 629         if (p->w)
 630           {
 631             p->w->pending = 0;
 632             p->w->cb (p->w, p->events);
 633           }
 634       }
 635 }
 636
 637 static void
 638 timers_reify (void)
 639 {
 640   while (timercnt && timers [0]->at <= now)
 641     {
 642       struct ev_timer *w = timers [0];
 643
 644       /* first reschedule or stop timer */
 645       if (w->repeat)
 646         {
 647           assert (("negative ev_timer repeat value found while processing timers", w->repeat > 0.));
 648           w->at = now + w->repeat;
 649           downheap ((WT *)timers, timercnt, 0);
 650         }
 651       else
 652         ev_timer_stop (w); /* nonrepeating: stop timer */
 653
 654       event ((W)w, EV_TIMEOUT);
 655     }
 656 }
 657
 658 static void
 659 periodics_reify (void)
 660 {
 661   while (periodiccnt && periodics [0]->at <= ev_now)
 662     {
 663       struct ev_periodic *w = periodics [0];
 664
 665       /* first reschedule or stop timer */
 666       if (w->interval)
 667         {
 668           w->at += floor ((ev_now - w->at) / w->interval + 1.) * w->interval;
 669           assert (("ev_periodic timeout in the past detected while processing timers, negative interval?", w->at > ev_now));
 670           downheap ((WT *)periodics, periodiccnt, 0);
 671         }
 672       else
 673         ev_periodic_stop (w); /* nonrepeating: stop timer */
 674
 675       event ((W)w, EV_PERIODIC);
 676     }
 677 }
 678
 679 static void
 680 periodics_reschedule (ev_tstamp diff)
 681 {
 682   int i;
 683
 684   /* adjust periodics after time jump */
 685   for (i = 0; i < periodiccnt; ++i)
 686     {
 687       struct ev_periodic *w = periodics [i];
 688
 689       if (w->interval)
 690         {
 691           ev_tstamp diff = ceil ((ev_now - w->at) / w->interval) * w->interval;
 692
 693           if (fabs (diff) >= 1e-4)
 694             {
 695               ev_periodic_stop (w);
 696               ev_periodic_start (w);
 697
 698               i = 0; /* restart loop, inefficient, but time jumps should be rare */
 699             }
 700         }
 701     }
 702 }
 703
 704 static int
 705 time_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
 722 static void
 723 time_update (void)
 724 {
 725   int i;
 726
 727 #if EV_USE_MONOTONIC
 728   if (expect_true (have_monotonic))
 729     {
 730       if (time_update_monotonic ())
 731         {
 732           ev_tstamp odiff = diff;
 733
 734           for (i = 4; --i; ) /* loop a few times, before making important decisions */
 735             {
 736               diff = ev_now - now;
 737
 738               if (fabs (odiff - diff) < MIN_TIMEJUMP)
 739                 return; /* all is well */
 740
 741               ev_now    = ev_time ();
 742               now       = get_clock ();
 743               now_floor = now;
 744             }
 745
 746           periodics_reschedule (diff - odiff);
 747           /* no timer adjustment, as the monotonic clock doesn't jump */
 748         }
 749     }
 750   else
 751 #endif
 752     {
 753       ev_now = ev_time ();
 754
 755       if (expect_false (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP))
 756         {
 757           periodics_reschedule (ev_now - now);
 758
 759           /* adjust timers. this is easy, as the offset is the same for all */
 760           for (i = 0; i < timercnt; ++i)
 761             timers [i]->at += diff;
 762         }
 763
 764       now = ev_now;
 765     }
 766 }
 767
 768 int ev_loop_done;
 769
 770 void ev_loop (int flags)
 771 {
 772   double block;
 773   ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0;
 774
 775   do
 776     {
 777       /* queue check watchers (and execute them) */
 778       if (expect_false (preparecnt))
 779         {
 780           queue_events ((W *)prepares, preparecnt, EV_PREPARE);
 781           call_pending ();
 782         }
 783
 784       /* update fd-related kernel structures */
 785       fd_reify ();
 786
 787       /* calculate blocking time */
 788
 789       /* we only need this for !monotonic clockor timers, but as we basically
 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         {
 797           ev_now = ev_time ();
 798           now    = ev_now;
 799         }
 800
 801       if (flags & EVLOOP_NONBLOCK || idlecnt)
 802         block = 0.;
 803       else
 804         {
 805           block = MAX_BLOCKTIME;
 806
 807           if (timercnt)
 808             {
 809               ev_tstamp to = timers [0]->at - now + method_fudge;
 810               if (block > to) block = to;
 811             }
 812
 813           if (periodiccnt)
 814             {
 815               ev_tstamp to = periodics [0]->at - ev_now + method_fudge;
 816               if (block > to) block = to;
 817             }
 818
 819           if (block < 0.) block = 0.;
 820         }
 821
 822       method_poll (block);
 823
 824       /* update ev_now, do magic */
 825       time_update ();
 826
 827       /* queue pending timers and reschedule them */
 828       timers_reify (); /* relative timers called last */
 829       periodics_reify (); /* absolute timers called first */
 830
 831       /* queue idle watchers unless io or timers are pending */
 832       if (!pendingcnt)
 833         queue_events ((W *)idles, idlecnt, EV_IDLE);
 834
 835       /* queue check watchers, to be executed first */
 836       if (checkcnt)
 837         queue_events ((W *)checks, checkcnt, EV_CHECK);
 838
 839       call_pending ();
 840     }
 841   while (!ev_loop_done);
 842
 843   if (ev_loop_done != 2)
 844     ev_loop_done = 0;
 845 }
 846
 847 /*****************************************************************************/
 848
 849 static void
 850 wlist_add (WL *head, WL elem)
 851 {
 852   elem->next = *head;
 853   *head = elem;
 854 }
 855
 856 static void
 857 wlist_del (WL *head, WL elem)
 858 {
 859   while (*head)
 860     {
 861       if (*head == elem)
 862         {
 863           *head = elem->next;
 864           return;
 865         }
 866
 867       head = &(*head)->next;
 868     }
 869 }
 870
 871 static void
 872 ev_clear_pending (W w)
 873 {
 874   if (w->pending)
 875     {
 876       pendings [ABSPRI (w)][w->pending - 1].w = 0;
 877       w->pending = 0;
 878     }
 879 }
 880
 881 static void
 882 ev_start (W w, int active)
 883 {
 884   if (w->priority < EV_MINPRI) w->priority = EV_MINPRI;
 885   if (w->priority > EV_MAXPRI) w->priority = EV_MAXPRI;
 886
 887   w->active = active;
 888 }
 889
 890 static void
 891 ev_stop (W w)
 892 {
 893   w->active = 0;
 894 }
 895
 896 /*****************************************************************************/
 897
 898 void
 899 ev_io_start (struct ev_io *w)
 900 {
 901   int fd = w->fd;
 902
 903   if (ev_is_active (w))
 904     return;
 905
 906   assert (("ev_io_start called with negative fd", fd >= 0));
 907
 908   ev_start ((W)w, 1);
 909   array_needsize (anfds, anfdmax, fd + 1, anfds_init);
 910   wlist_add ((WL *)&anfds[fd].head, (WL)w);
 911
 912   fd_change (fd);
 913 }
 914
 915 void
 916 ev_io_stop (struct ev_io *w)
 917 {
 918   ev_clear_pending ((W)w);
 919   if (!ev_is_active (w))
 920     return;
 921
 922   wlist_del ((WL *)&anfds[w->fd].head, (WL)w);
 923   ev_stop ((W)w);
 924
 925   fd_change (w->fd);
 926 }
 927
 928 void
 929 ev_timer_start (struct ev_timer *w)
 930 {
 931   if (ev_is_active (w))
 932     return;
 933
 934   w->at += now;
 935
 936   assert (("ev_timer_start called with negative timer repeat value", w->repeat >= 0.));
 937
 938   ev_start ((W)w, ++timercnt);
 939   array_needsize (timers, timermax, timercnt, );
 940   timers [timercnt - 1] = w;
 941   upheap ((WT *)timers, timercnt - 1);
 942 }
 943
 944 void
 945 ev_timer_stop (struct ev_timer *w)
 946 {
 947   ev_clear_pending ((W)w);
 948   if (!ev_is_active (w))
 949     return;
 950
 951   if (w->active < timercnt--)
 952     {
 953       timers [w->active - 1] = timers [timercnt];
 954       downheap ((WT *)timers, timercnt, w->active - 1);
 955     }
 956
 957   w->at = w->repeat;
 958
 959   ev_stop ((W)w);
 960 }
 961
 962 void
 963 ev_timer_again (struct ev_timer *w)
 964 {
 965   if (ev_is_active (w))
 966     {
 967       if (w->repeat)
 968         {
 969           w->at = now + w->repeat;
 970           downheap ((WT *)timers, timercnt, w->active - 1);
 971         }
 972       else
 973         ev_timer_stop (w);
 974     }
 975   else if (w->repeat)
 976     ev_timer_start (w);
 977 }
 978
 979 void
 980 ev_periodic_start (struct ev_periodic *w)
 981 {
 982   if (ev_is_active (w))
 983     return;
 984
 985   assert (("ev_periodic_start called with negative interval value", w->interval >= 0.));
 986
 987   /* this formula differs from the one in periodic_reify because we do not always round up */
 988   if (w->interval)
 989     w->at += ceil ((ev_now - w->at) / w->interval) * w->interval;
 990
 991   ev_start ((W)w, ++periodiccnt);
 992   array_needsize (periodics, periodicmax, periodiccnt, );
 993   periodics [periodiccnt - 1] = w;
 994   upheap ((WT *)periodics, periodiccnt - 1);
 995 }
 996
 997 void
 998 ev_periodic_stop (struct ev_periodic *w)
 999 {
1000   ev_clear_pending ((W)w);
1001   if (!ev_is_active (w))
1002     return;
1003
1004   if (w->active < periodiccnt--)
1005     {
1006       periodics [w->active - 1] = periodics [periodiccnt];
1007       downheap ((WT *)periodics, periodiccnt, w->active - 1);
1008     }
1009
1010   ev_stop ((W)w);
1011 }
1012
1013 void
1014 ev_signal_start (struct ev_signal *w)
1015 {
1016   if (ev_is_active (w))
1017     return;
1018
1019   assert (("ev_signal_start called with illegal signal number", w->signum > 0));
1020
1021   ev_start ((W)w, 1);
1022   array_needsize (signals, signalmax, w->signum, signals_init);
1023   wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w);
1024
1025   if (!w->next)
1026     {
1027       struct sigaction sa;
1028       sa.sa_handler = sighandler;
1029       sigfillset (&sa.sa_mask);
1030       sa.sa_flags = 0;
1031       sigaction (w->signum, &sa, 0);
1032     }
1033 }
1034
1035 void
1036 ev_signal_stop (struct ev_signal *w)
1037 {
1038   ev_clear_pending ((W)w);
1039   if (!ev_is_active (w))
1040     return;
1041
1042   wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w);
1043   ev_stop ((W)w);
1044
1045   if (!signals [w->signum - 1].head)
1046     signal (w->signum, SIG_DFL);
1047 }
1048
1049 void
1050 ev_idle_start (struct ev_idle *w)
1051 {
1052   if (ev_is_active (w))
1053     return;
1054
1055   ev_start ((W)w, ++idlecnt);
1056   array_needsize (idles, idlemax, idlecnt, );
1057   idles [idlecnt - 1] = w;
1058 }
1059
1060 void
1061 ev_idle_stop (struct ev_idle *w)
1062 {
1063   ev_clear_pending ((W)w);
1064   if (ev_is_active (w))
1065     return;
1066
1067   idles [w->active - 1] = idles [--idlecnt];
1068   ev_stop ((W)w);
1069 }
1070
1071 void
1072 ev_prepare_start (struct ev_prepare *w)
1073 {
1074   if (ev_is_active (w))
1075     return;
1076
1077   ev_start ((W)w, ++preparecnt);
1078   array_needsize (prepares, preparemax, preparecnt, );
1079   prepares [preparecnt - 1] = w;
1080 }
1081
1082 void
1083 ev_prepare_stop (struct ev_prepare *w)
1084 {
1085   ev_clear_pending ((W)w);
1086   if (ev_is_active (w))
1087     return;
1088
1089   prepares [w->active - 1] = prepares [--preparecnt];
1090   ev_stop ((W)w);
1091 }
1092
1093 void
1094 ev_check_start (struct ev_check *w)
1095 {
1096   if (ev_is_active (w))
1097     return;
1098
1099   ev_start ((W)w, ++checkcnt);
1100   array_needsize (checks, checkmax, checkcnt, );
1101   checks [checkcnt - 1] = w;
1102 }
1103
1104 void
1105 ev_check_stop (struct ev_check *w)
1106 {
1107   ev_clear_pending ((W)w);
1108   if (ev_is_active (w))
1109     return;
1110
1111   checks [w->active - 1] = checks [--checkcnt];
1112   ev_stop ((W)w);
1113 }
1114
1115 void
1116 ev_child_start (struct ev_child *w)
1117 {
1118   if (ev_is_active (w))
1119     return;
1120
1121   ev_start ((W)w, 1);
1122   wlist_add ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
1123 }
1124
1125 void
1126 ev_child_stop (struct ev_child *w)
1127 {
1128   ev_clear_pending ((W)w);
1129   if (ev_is_active (w))
1130     return;
1131
1132   wlist_del ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
1133   ev_stop ((W)w);
1134 }
1135
1136 /*****************************************************************************/
1137
1138 struct ev_once
1139 {
1140   struct ev_io io;
1141   struct ev_timer to;
1142   void (*cb)(int revents, void *arg);
1143   void *arg;
1144 };
1145
1146 static void
1147 once_cb (struct ev_once *once, int revents)
1148 {
1149   void (*cb)(int revents, void *arg) = once->cb;
1150   void *arg = once->arg;
1151
1152   ev_io_stop (&once->io);
1153   ev_timer_stop (&once->to);
1154   free (once);
1155
1156   cb (revents, arg);
1157 }
1158
1159 static void
1160 once_cb_io (struct ev_io *w, int revents)
1161 {
1162   once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents);
1163 }
1164
1165 static void
1166 once_cb_to (struct ev_timer *w, int revents)
1167 {
1168   once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents);
1169 }
1170
1171 void
1172 ev_once (int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg)
1173 {
1174   struct ev_once *once = malloc (sizeof (struct ev_once));
1175
1176   if (!once)
1177     cb (EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT, arg);
1178   else
1179     {
1180       once->cb  = cb;
1181       once->arg = arg;
1182
1183       ev_watcher_init (&once->io, once_cb_io);
1184       if (fd >= 0)
1185         {
1186           ev_io_set (&once->io, fd, events);
1187           ev_io_start (&once->io);
1188         }
1189
1190       ev_watcher_init (&once->to, once_cb_to);
1191       if (timeout >= 0.)
1192         {
1193           ev_timer_set (&once->to, timeout, 0.);
1194           ev_timer_start (&once->to);
1195         }
1196     }
1197 }
1198
1199 /*****************************************************************************/
1200
1201 #if 0
1202
1203 struct ev_io wio;
1204
1205 static void
1206 sin_cb (struct ev_io *w, int revents)
1207 {
1208   fprintf (stderr, "sin %d, revents %d\n", w->fd, revents);
1209 }
1210
1211 static void
1212 ocb (struct ev_timer *w, int revents)
1213 {
1214   //fprintf (stderr, "timer %f,%f (%x) (%f) d%p\n", w->at, w->repeat, revents, w->at - ev_time (), w->data);
1215   ev_timer_stop (w);
1216   ev_timer_start (w);
1217 }
1218
1219 static void
1220 scb (struct ev_signal *w, int revents)
1221 {
1222   fprintf (stderr, "signal %x,%d\n", revents, w->signum);
1223   ev_io_stop (&wio);
1224   ev_io_start (&wio);
1225 }
1226
1227 static void
1228 gcb (struct ev_signal *w, int revents)
1229 {
1230   fprintf (stderr, "generic %x\n", revents);
1231
1232 }
1233
1234 int main (void)
1235 {
1236   ev_init (0);
1237
1238   ev_io_init (&wio, sin_cb, 0, EV_READ);
1239   ev_io_start (&wio);
1240
1241   struct ev_timer t[10000];
1242
1243 #if 0
1244   int i;
1245   for (i = 0; i < 10000; ++i)
1246     {
1247       struct ev_timer *w = t + i;
1248       ev_watcher_init (w, ocb, i);
1249       ev_timer_init_abs (w, ocb, drand48 (), 0.99775533);
1250       ev_timer_start (w);
1251       if (drand48 () < 0.5)
1252         ev_timer_stop (w);
1253     }
1254 #endif
1255
1256   struct ev_timer t1;
1257   ev_timer_init (&t1, ocb, 5, 10);
1258   ev_timer_start (&t1);
1259
1260   struct ev_signal sig;
1261   ev_signal_init (&sig, scb, SIGQUIT);
1262   ev_signal_start (&sig);
1263
1264   struct ev_check cw;
1265   ev_check_init (&cw, gcb);
1266   ev_check_start (&cw);
1267
1268   struct ev_idle iw;
1269   ev_idle_init (&iw, gcb);
1270   ev_idle_start (&iw);
1271
1272   ev_loop (0);
1273
1274   return 0;
1275 }
1276
1277 #endif
1278
1279
1280
1281