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

Comparing libev/ev.c (file contents):
Revision 1.40 by root, Fri Nov 2 11:02:23 2007 UTC vs.
Revision 1.85 by root, Sat Nov 10 03:13:50 2007 UTC

26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 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. 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 */ 30 */
31#if EV_USE_CONFIG_H 31#ifndef EV_STANDALONE
32# include "config.h" 32# include "config.h"
33
34# if HAVE_CLOCK_GETTIME
35# define EV_USE_MONOTONIC 1
36# define EV_USE_REALTIME 1
37# endif
38
39# if HAVE_SELECT && HAVE_SYS_SELECT_H
40# define EV_USE_SELECT 1
41# endif
42
43# if HAVE_POLL && HAVE_POLL_H
44# define EV_USE_POLL 1
45# endif
46
47# if HAVE_EPOLL && HAVE_EPOLL_CTL && HAVE_SYS_EPOLL_H
48# define EV_USE_EPOLL 1
49# endif
50
51# if HAVE_KQUEUE && HAVE_WORKING_KQUEUE && HAVE_SYS_EVENT_H && HAVE_SYS_QUEUE_H
52# define EV_USE_KQUEUE 1
53# endif
54
33#endif 55#endif
34 56
35#include <math.h> 57#include <math.h>
36#include <stdlib.h> 58#include <stdlib.h>
37#include <unistd.h>
38#include <fcntl.h> 59#include <fcntl.h>
39#include <signal.h>
40#include <stddef.h> 60#include <stddef.h>
41 61
42#include <stdio.h> 62#include <stdio.h>
43 63
44#include <assert.h> 64#include <assert.h>
45#include <errno.h> 65#include <errno.h>
46#include <sys/types.h> 66#include <sys/types.h>
47#include <sys/wait.h>
48#include <sys/time.h>
49#include <time.h> 67#include <time.h>
50 68
69#include <signal.h>
70
71#ifndef WIN32
72# include <unistd.h>
73# include <sys/time.h>
74# include <sys/wait.h>
75#endif
51/**/ 76/**/
52 77
53#ifndef EV_USE_MONOTONIC 78#ifndef EV_USE_MONOTONIC
54# define EV_USE_MONOTONIC 1 79# define EV_USE_MONOTONIC 1
55#endif 80#endif
56 81
57#ifndef EV_USE_SELECT 82#ifndef EV_USE_SELECT
58# define EV_USE_SELECT 1 83# define EV_USE_SELECT 1
59#endif 84#endif
60 85
86#ifndef EV_USE_POLL
87# define EV_USE_POLL 0 /* poll is usually slower than select, and not as well tested */
88#endif
89
61#ifndef EV_USE_EPOLL 90#ifndef EV_USE_EPOLL
62# define EV_USE_EPOLL 0 91# define EV_USE_EPOLL 0
92#endif
93
94#ifndef EV_USE_KQUEUE
95# define EV_USE_KQUEUE 0
96#endif
97
98#ifndef EV_USE_WIN32
99# ifdef WIN32
100# define EV_USE_WIN32 0 /* it does not exist, use select */
101# undef EV_USE_SELECT
102# define EV_USE_SELECT 1
103# else
104# define EV_USE_WIN32 0
105# endif
63#endif 106#endif
64 107
65#ifndef EV_USE_REALTIME 108#ifndef EV_USE_REALTIME
66# define EV_USE_REALTIME 1 109# define EV_USE_REALTIME 1
67#endif 110#endif
83#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */ 126#define MIN_TIMEJUMP 1. /* minimum timejump that gets detected (if monotonic clock available) */
84#define MAX_BLOCKTIME 59.731 /* never wait longer than this time (to detect time jumps) */ 127#define MAX_BLOCKTIME 59.731 /* never wait longer than this time (to detect time jumps) */
85#define PID_HASHSIZE 16 /* size of pid hash table, must be power of two */ 128#define PID_HASHSIZE 16 /* size of pid hash table, must be power of two */
86/*#define CLEANUP_INTERVAL 300. /* how often to try to free memory and re-check fds */ 129/*#define CLEANUP_INTERVAL 300. /* how often to try to free memory and re-check fds */
87 130
131#ifdef EV_H
132# include EV_H
133#else
88#include "ev.h" 134# include "ev.h"
135#endif
89 136
90#if __GNUC__ >= 3 137#if __GNUC__ >= 3
91# define expect(expr,value) __builtin_expect ((expr),(value)) 138# define expect(expr,value) __builtin_expect ((expr),(value))
92# define inline inline 139# define inline inline
93#else 140#else
96#endif 143#endif
97 144
98#define expect_false(expr) expect ((expr) != 0, 0) 145#define expect_false(expr) expect ((expr) != 0, 0)
99#define expect_true(expr) expect ((expr) != 0, 1) 146#define expect_true(expr) expect ((expr) != 0, 1)
100 147
148#define NUMPRI (EV_MAXPRI - EV_MINPRI + 1)
149#define ABSPRI(w) ((w)->priority - EV_MINPRI)
150
101typedef struct ev_watcher *W; 151typedef struct ev_watcher *W;
102typedef struct ev_watcher_list *WL; 152typedef struct ev_watcher_list *WL;
103typedef struct ev_watcher_time *WT; 153typedef struct ev_watcher_time *WT;
104 154
105static ev_tstamp now_floor, now, diff; /* monotonic clock */ 155static int have_monotonic; /* did clock_gettime (CLOCK_MONOTONIC) work? */
106ev_tstamp ev_now;
107int ev_method;
108 156
109static int have_monotonic; /* runtime */ 157#include "ev_win32.c"
110
111static ev_tstamp method_fudge; /* stupid epoll-returns-early bug */
112static void (*method_modify)(int fd, int oev, int nev);
113static void (*method_poll)(ev_tstamp timeout);
114 158
115/*****************************************************************************/ 159/*****************************************************************************/
116 160
117ev_tstamp 161static void (*syserr_cb)(const char *msg);
162
163void ev_set_syserr_cb (void (*cb)(const char *msg))
164{
165 syserr_cb = cb;
166}
167
168static void
169syserr (const char *msg)
170{
171 if (!msg)
172 msg = "(libev) system error";
173
174 if (syserr_cb)
175 syserr_cb (msg);
176 else
177 {
178 perror (msg);
179 abort ();
180 }
181}
182
183static void *(*alloc)(void *ptr, long size);
184
185void ev_set_allocator (void *(*cb)(void *ptr, long size))
186{
187 alloc = cb;
188}
189
190static void *
191ev_realloc (void *ptr, long size)
192{
193 ptr = alloc ? alloc (ptr, size) : realloc (ptr, size);
194
195 if (!ptr && size)
196 {
197 fprintf (stderr, "libev: cannot allocate %ld bytes, aborting.", size);
198 abort ();
199 }
200
201 return ptr;
202}
203
204#define ev_malloc(size) ev_realloc (0, (size))
205#define ev_free(ptr) ev_realloc ((ptr), 0)
206
207/*****************************************************************************/
208
209typedef struct
210{
211 WL head;
212 unsigned char events;
213 unsigned char reify;
214} ANFD;
215
216typedef struct
217{
218 W w;
219 int events;
220} ANPENDING;
221
222#if EV_MULTIPLICITY
223
224 struct ev_loop
225 {
226 #define VAR(name,decl) decl;
227 #include "ev_vars.h"
228 #undef VAR
229 };
230 #include "ev_wrap.h"
231
232 struct ev_loop default_loop_struct;
233 static struct ev_loop *default_loop;
234
235#else
236
237 #define VAR(name,decl) static decl;
238 #include "ev_vars.h"
239 #undef VAR
240
241 static int default_loop;
242
243#endif
244
245/*****************************************************************************/
246
247inline ev_tstamp
118ev_time (void) 248ev_time (void)
119{ 249{
120#if EV_USE_REALTIME 250#if EV_USE_REALTIME
121 struct timespec ts; 251 struct timespec ts;
122 clock_gettime (CLOCK_REALTIME, &ts); 252 clock_gettime (CLOCK_REALTIME, &ts);
126 gettimeofday (&tv, 0); 256 gettimeofday (&tv, 0);
127 return tv.tv_sec + tv.tv_usec * 1e-6; 257 return tv.tv_sec + tv.tv_usec * 1e-6;
128#endif 258#endif
129} 259}
130 260
131static ev_tstamp 261inline ev_tstamp
132get_clock (void) 262get_clock (void)
133{ 263{
134#if EV_USE_MONOTONIC 264#if EV_USE_MONOTONIC
135 if (expect_true (have_monotonic)) 265 if (expect_true (have_monotonic))
136 { 266 {
141#endif 271#endif
142 272
143 return ev_time (); 273 return ev_time ();
144} 274}
145 275
276#if EV_MULTIPLICITY
277ev_tstamp
278ev_now (EV_P)
279{
280 return ev_rt_now;
281}
282#endif
283
146#define array_roundsize(base,n) ((n) | 4 & ~3) 284#define array_roundsize(type,n) ((n) | 4 & ~3)
147 285
148#define array_needsize(base,cur,cnt,init) \ 286#define array_needsize(type,base,cur,cnt,init) \
149 if (expect_false ((cnt) > cur)) \ 287 if (expect_false ((cnt) > cur)) \
150 { \ 288 { \
151 int newcnt = cur; \ 289 int newcnt = cur; \
152 do \ 290 do \
153 { \ 291 { \
154 newcnt = array_roundsize (base, newcnt << 1); \ 292 newcnt = array_roundsize (type, newcnt << 1); \
155 } \ 293 } \
156 while ((cnt) > newcnt); \ 294 while ((cnt) > newcnt); \
157 \ 295 \
158 base = realloc (base, sizeof (*base) * (newcnt)); \ 296 base = (type *)ev_realloc (base, sizeof (type) * (newcnt));\
159 init (base + cur, newcnt - cur); \ 297 init (base + cur, newcnt - cur); \
160 cur = newcnt; \ 298 cur = newcnt; \
161 } 299 }
300
301#define array_slim(type,stem) \
302 if (stem ## max < array_roundsize (stem ## cnt >> 2)) \
303 { \
304 stem ## max = array_roundsize (stem ## cnt >> 1); \
305 base = (type *)ev_realloc (base, sizeof (type) * (stem ## max));\
306 fprintf (stderr, "slimmed down " # stem " to %d\n", stem ## max);/*D*/\
307 }
308
309/* microsoft's pseudo-c is quite far from C as the rest of the world and the standard knows it */
310/* bringing us everlasting joy in form of stupid extra macros that are not required in C */
311#define array_free_microshit(stem) \
312 ev_free (stem ## s); stem ## cnt = stem ## max = 0;
313
314#define array_free(stem, idx) \
315 ev_free (stem ## s idx); stem ## cnt idx = stem ## max idx = 0;
162 316
163/*****************************************************************************/ 317/*****************************************************************************/
164
165typedef struct
166{
167 struct ev_io *head;
168 unsigned char events;
169 unsigned char reify;
170} ANFD;
171
172static ANFD *anfds;
173static int anfdmax;
174 318
175static void 319static void
176anfds_init (ANFD *base, int count) 320anfds_init (ANFD *base, int count)
177{ 321{
178 while (count--) 322 while (count--)
183 327
184 ++base; 328 ++base;
185 } 329 }
186} 330}
187 331
188typedef struct 332void
333ev_feed_event (EV_P_ void *w, int revents)
189{ 334{
190 W w; 335 W w_ = (W)w;
191 int events;
192} ANPENDING;
193 336
194static ANPENDING *pendings;
195static int pendingmax, pendingcnt;
196
197static void
198event (W w, int events)
199{
200 if (w->pending) 337 if (w_->pending)
201 { 338 {
202 pendings [w->pending - 1].events |= events; 339 pendings [ABSPRI (w_)][w_->pending - 1].events |= revents;
203 return; 340 return;
204 } 341 }
205 342
206 w->pending = ++pendingcnt; 343 w_->pending = ++pendingcnt [ABSPRI (w_)];
207 array_needsize (pendings, pendingmax, pendingcnt, ); 344 array_needsize (ANPENDING, pendings [ABSPRI (w_)], pendingmax [ABSPRI (w_)], pendingcnt [ABSPRI (w_)], (void));
208 pendings [pendingcnt - 1].w = w; 345 pendings [ABSPRI (w_)][w_->pending - 1].w = w_;
209 pendings [pendingcnt - 1].events = events; 346 pendings [ABSPRI (w_)][w_->pending - 1].events = revents;
210} 347}
211 348
212static void 349static void
213queue_events (W *events, int eventcnt, int type) 350queue_events (EV_P_ W *events, int eventcnt, int type)
214{ 351{
215 int i; 352 int i;
216 353
217 for (i = 0; i < eventcnt; ++i) 354 for (i = 0; i < eventcnt; ++i)
218 event (events [i], type); 355 ev_feed_event (EV_A_ events [i], type);
219} 356}
220 357
221static void 358inline void
222fd_event (int fd, int events) 359fd_event (EV_P_ int fd, int revents)
223{ 360{
224 ANFD *anfd = anfds + fd; 361 ANFD *anfd = anfds + fd;
225 struct ev_io *w; 362 struct ev_io *w;
226 363
227 for (w = anfd->head; w; w = w->next) 364 for (w = (struct ev_io *)anfd->head; w; w = (struct ev_io *)((WL)w)->next)
228 { 365 {
229 int ev = w->events & events; 366 int ev = w->events & revents;
230 367
231 if (ev) 368 if (ev)
232 event ((W)w, ev); 369 ev_feed_event (EV_A_ (W)w, ev);
233 } 370 }
371}
372
373void
374ev_feed_fd_event (EV_P_ int fd, int revents)
375{
376 fd_event (EV_A_ fd, revents);
234} 377}
235 378
236/*****************************************************************************/ 379/*****************************************************************************/
237 380
238static int *fdchanges;
239static int fdchangemax, fdchangecnt;
240
241static void 381static void
242fd_reify (void) 382fd_reify (EV_P)
243{ 383{
244 int i; 384 int i;
245 385
246 for (i = 0; i < fdchangecnt; ++i) 386 for (i = 0; i < fdchangecnt; ++i)
247 { 387 {
249 ANFD *anfd = anfds + fd; 389 ANFD *anfd = anfds + fd;
250 struct ev_io *w; 390 struct ev_io *w;
251 391
252 int events = 0; 392 int events = 0;
253 393
254 for (w = anfd->head; w; w = w->next) 394 for (w = (struct ev_io *)anfd->head; w; w = (struct ev_io *)((WL)w)->next)
255 events |= w->events; 395 events |= w->events;
256 396
257 anfd->reify = 0; 397 anfd->reify = 0;
258 398
259 if (anfd->events != events)
260 {
261 method_modify (fd, anfd->events, events); 399 method_modify (EV_A_ fd, anfd->events, events);
262 anfd->events = events; 400 anfd->events = events;
263 }
264 } 401 }
265 402
266 fdchangecnt = 0; 403 fdchangecnt = 0;
267} 404}
268 405
269static void 406static void
270fd_change (int fd) 407fd_change (EV_P_ int fd)
271{ 408{
272 if (anfds [fd].reify || fdchangecnt < 0) 409 if (anfds [fd].reify)
273 return; 410 return;
274 411
275 anfds [fd].reify = 1; 412 anfds [fd].reify = 1;
276 413
277 ++fdchangecnt; 414 ++fdchangecnt;
278 array_needsize (fdchanges, fdchangemax, fdchangecnt, ); 415 array_needsize (int, fdchanges, fdchangemax, fdchangecnt, (void));
279 fdchanges [fdchangecnt - 1] = fd; 416 fdchanges [fdchangecnt - 1] = fd;
280} 417}
281 418
419static void
420fd_kill (EV_P_ int fd)
421{
422 struct ev_io *w;
423
424 while ((w = (struct ev_io *)anfds [fd].head))
425 {
426 ev_io_stop (EV_A_ w);
427 ev_feed_event (EV_A_ (W)w, EV_ERROR | EV_READ | EV_WRITE);
428 }
429}
430
431static int
432fd_valid (int fd)
433{
434#ifdef WIN32
435 return !!win32_get_osfhandle (fd);
436#else
437 return fcntl (fd, F_GETFD) != -1;
438#endif
439}
440
282/* called on EBADF to verify fds */ 441/* called on EBADF to verify fds */
283static void 442static void
284fd_recheck (void) 443fd_ebadf (EV_P)
285{ 444{
286 int fd; 445 int fd;
287 446
288 for (fd = 0; fd < anfdmax; ++fd) 447 for (fd = 0; fd < anfdmax; ++fd)
289 if (anfds [fd].events) 448 if (anfds [fd].events)
290 if (fcntl (fd, F_GETFD) == -1 && errno == EBADF) 449 if (!fd_valid (fd) == -1 && errno == EBADF)
291 while (anfds [fd].head) 450 fd_kill (EV_A_ fd);
451}
452
453/* called on ENOMEM in select/poll to kill some fds and retry */
454static void
455fd_enomem (EV_P)
456{
457 int fd;
458
459 for (fd = anfdmax; fd--; )
460 if (anfds [fd].events)
292 { 461 {
293 ev_io_stop (anfds [fd].head); 462 fd_kill (EV_A_ fd);
294 event ((W)anfds [fd].head, EV_ERROR | EV_READ | EV_WRITE); 463 return;
295 } 464 }
465}
466
467/* usually called after fork if method needs to re-arm all fds from scratch */
468static void
469fd_rearm_all (EV_P)
470{
471 int fd;
472
473 /* this should be highly optimised to not do anything but set a flag */
474 for (fd = 0; fd < anfdmax; ++fd)
475 if (anfds [fd].events)
476 {
477 anfds [fd].events = 0;
478 fd_change (EV_A_ fd);
479 }
296} 480}
297 481
298/*****************************************************************************/ 482/*****************************************************************************/
299 483
300static struct ev_timer **timers;
301static int timermax, timercnt;
302
303static struct ev_periodic **periodics;
304static int periodicmax, periodiccnt;
305
306static void 484static void
307upheap (WT *timers, int k) 485upheap (WT *heap, int k)
308{ 486{
309 WT w = timers [k]; 487 WT w = heap [k];
310 488
311 while (k && timers [k >> 1]->at > w->at) 489 while (k && heap [k >> 1]->at > w->at)
312 { 490 {
313 timers [k] = timers [k >> 1]; 491 heap [k] = heap [k >> 1];
314 timers [k]->active = k + 1; 492 ((W)heap [k])->active = k + 1;
315 k >>= 1; 493 k >>= 1;
316 } 494 }
317 495
318 timers [k] = w; 496 heap [k] = w;
319 timers [k]->active = k + 1; 497 ((W)heap [k])->active = k + 1;
320 498
321} 499}
322 500
323static void 501static void
324downheap (WT *timers, int N, int k) 502downheap (WT *heap, int N, int k)
325{ 503{
326 WT w = timers [k]; 504 WT w = heap [k];
327 505
328 while (k < (N >> 1)) 506 while (k < (N >> 1))
329 { 507 {
330 int j = k << 1; 508 int j = k << 1;
331 509
332 if (j + 1 < N && timers [j]->at > timers [j + 1]->at) 510 if (j + 1 < N && heap [j]->at > heap [j + 1]->at)
333 ++j; 511 ++j;
334 512
335 if (w->at <= timers [j]->at) 513 if (w->at <= heap [j]->at)
336 break; 514 break;
337 515
338 timers [k] = timers [j]; 516 heap [k] = heap [j];
339 timers [k]->active = k + 1; 517 ((W)heap [k])->active = k + 1;
340 k = j; 518 k = j;
341 } 519 }
342 520
343 timers [k] = w; 521 heap [k] = w;
344 timers [k]->active = k + 1; 522 ((W)heap [k])->active = k + 1;
523}
524
525inline void
526adjustheap (WT *heap, int N, int k, ev_tstamp at)
527{
528 ev_tstamp old_at = heap [k]->at;
529 heap [k]->at = at;
530
531 if (old_at < at)
532 downheap (heap, N, k);
533 else
534 upheap (heap, k);
345} 535}
346 536
347/*****************************************************************************/ 537/*****************************************************************************/
348 538
349typedef struct 539typedef struct
350{ 540{
351 struct ev_signal *head; 541 WL head;
352 sig_atomic_t volatile gotsig; 542 sig_atomic_t volatile gotsig;
353} ANSIG; 543} ANSIG;
354 544
355static ANSIG *signals; 545static ANSIG *signals;
356static int signalmax; 546static int signalmax;
372} 562}
373 563
374static void 564static void
375sighandler (int signum) 565sighandler (int signum)
376{ 566{
567#if WIN32
568 signal (signum, sighandler);
569#endif
570
377 signals [signum - 1].gotsig = 1; 571 signals [signum - 1].gotsig = 1;
378 572
379 if (!gotsig) 573 if (!gotsig)
380 { 574 {
575 int old_errno = errno;
381 gotsig = 1; 576 gotsig = 1;
577#ifdef WIN32
578 send (sigpipe [1], &signum, 1, MSG_DONTWAIT);
579#else
382 write (sigpipe [1], &signum, 1); 580 write (sigpipe [1], &signum, 1);
581#endif
582 errno = old_errno;
383 } 583 }
384} 584}
385 585
586void
587ev_feed_signal_event (EV_P_ int signum)
588{
589 WL w;
590
591#if EV_MULTIPLICITY
592 assert (("feeding signal events is only supported in the default loop", loop == default_loop));
593#endif
594
595 --signum;
596
597 if (signum < 0 || signum >= signalmax)
598 return;
599
600 signals [signum].gotsig = 0;
601
602 for (w = signals [signum].head; w; w = w->next)
603 ev_feed_event (EV_A_ (W)w, EV_SIGNAL);
604}
605
386static void 606static void
387sigcb (struct ev_io *iow, int revents) 607sigcb (EV_P_ struct ev_io *iow, int revents)
388{ 608{
389 struct ev_signal *w;
390 int signum; 609 int signum;
391 610
611#ifdef WIN32
612 recv (sigpipe [0], &revents, 1, MSG_DONTWAIT);
613#else
392 read (sigpipe [0], &revents, 1); 614 read (sigpipe [0], &revents, 1);
615#endif
393 gotsig = 0; 616 gotsig = 0;
394 617
395 for (signum = signalmax; signum--; ) 618 for (signum = signalmax; signum--; )
396 if (signals [signum].gotsig) 619 if (signals [signum].gotsig)
397 { 620 ev_feed_signal_event (EV_A_ signum + 1);
398 signals [signum].gotsig = 0;
399
400 for (w = signals [signum].head; w; w = w->next)
401 event ((W)w, EV_SIGNAL);
402 }
403} 621}
404 622
405static void 623static void
406siginit (void) 624siginit (EV_P)
407{ 625{
626#ifndef WIN32
408 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC); 627 fcntl (sigpipe [0], F_SETFD, FD_CLOEXEC);
409 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC); 628 fcntl (sigpipe [1], F_SETFD, FD_CLOEXEC);
410 629
411 /* rather than sort out wether we really need nb, set it */ 630 /* rather than sort out wether we really need nb, set it */
412 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK); 631 fcntl (sigpipe [0], F_SETFL, O_NONBLOCK);
413 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK); 632 fcntl (sigpipe [1], F_SETFL, O_NONBLOCK);
633#endif
414 634
415 ev_io_set (&sigev, sigpipe [0], EV_READ); 635 ev_io_set (&sigev, sigpipe [0], EV_READ);
416 ev_io_start (&sigev); 636 ev_io_start (EV_A_ &sigev);
637 ev_unref (EV_A); /* child watcher should not keep loop alive */
417} 638}
418 639
419/*****************************************************************************/ 640/*****************************************************************************/
420 641
421static struct ev_idle **idles;
422static int idlemax, idlecnt;
423
424static struct ev_prepare **prepares;
425static int preparemax, preparecnt;
426
427static struct ev_check **checks;
428static int checkmax, checkcnt;
429
430/*****************************************************************************/
431
432static struct ev_child *childs [PID_HASHSIZE]; 642static struct ev_child *childs [PID_HASHSIZE];
643
644#ifndef WIN32
645
433static struct ev_signal childev; 646static struct ev_signal childev;
434 647
435#ifndef WCONTINUED 648#ifndef WCONTINUED
436# define WCONTINUED 0 649# define WCONTINUED 0
437#endif 650#endif
438 651
439static void 652static void
440childcb (struct ev_signal *sw, int revents) 653child_reap (EV_P_ struct ev_signal *sw, int chain, int pid, int status)
441{ 654{
442 struct ev_child *w; 655 struct ev_child *w;
656
657 for (w = (struct ev_child *)childs [chain & (PID_HASHSIZE - 1)]; w; w = (struct ev_child *)((WL)w)->next)
658 if (w->pid == pid || !w->pid)
659 {
660 ev_priority (w) = ev_priority (sw); /* need to do it *now* */
661 w->rpid = pid;
662 w->rstatus = status;
663 ev_feed_event (EV_A_ (W)w, EV_CHILD);
664 }
665}
666
667static void
668childcb (EV_P_ struct ev_signal *sw, int revents)
669{
443 int pid, status; 670 int pid, status;
444 671
445 while ((pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)) != -1) 672 if (0 < (pid = waitpid (-1, &status, WNOHANG | WUNTRACED | WCONTINUED)))
446 for (w = childs [pid & (PID_HASHSIZE - 1)]; w; w = w->next) 673 {
447 if (w->pid == pid || !w->pid) 674 /* make sure we are called again until all childs have been reaped */
448 { 675 ev_feed_event (EV_A_ (W)sw, EV_SIGNAL);
449 w->status = status; 676
450 event ((W)w, EV_CHILD); 677 child_reap (EV_A_ sw, pid, pid, status);
451 } 678 child_reap (EV_A_ sw, 0, pid, status); /* this might trigger a watcher twice, but event catches that */
679 }
452} 680}
681
682#endif
453 683
454/*****************************************************************************/ 684/*****************************************************************************/
455 685
686#if EV_USE_KQUEUE
687# include "ev_kqueue.c"
688#endif
456#if EV_USE_EPOLL 689#if EV_USE_EPOLL
457# include "ev_epoll.c" 690# include "ev_epoll.c"
458#endif 691#endif
692#if EV_USE_POLL
693# include "ev_poll.c"
694#endif
459#if EV_USE_SELECT 695#if EV_USE_SELECT
460# include "ev_select.c" 696# include "ev_select.c"
461#endif 697#endif
462 698
463int 699int
470ev_version_minor (void) 706ev_version_minor (void)
471{ 707{
472 return EV_VERSION_MINOR; 708 return EV_VERSION_MINOR;
473} 709}
474 710
475int ev_init (int flags) 711/* return true if we are running with elevated privileges and should ignore env variables */
712static int
713enable_secure (void)
476{ 714{
715#ifdef WIN32
716 return 0;
717#else
718 return getuid () != geteuid ()
719 || getgid () != getegid ();
720#endif
721}
722
723int
724ev_method (EV_P)
725{
726 return method;
727}
728
729static void
730loop_init (EV_P_ int methods)
731{
477 if (!ev_method) 732 if (!method)
478 { 733 {
479#if EV_USE_MONOTONIC 734#if EV_USE_MONOTONIC
480 { 735 {
481 struct timespec ts; 736 struct timespec ts;
482 if (!clock_gettime (CLOCK_MONOTONIC, &ts)) 737 if (!clock_gettime (CLOCK_MONOTONIC, &ts))
483 have_monotonic = 1; 738 have_monotonic = 1;
484 } 739 }
485#endif 740#endif
486 741
487 ev_now = ev_time (); 742 ev_rt_now = ev_time ();
488 now = get_clock (); 743 mn_now = get_clock ();
489 now_floor = now; 744 now_floor = mn_now;
490 diff = ev_now - now; 745 rtmn_diff = ev_rt_now - mn_now;
491 746
492 if (pipe (sigpipe)) 747 if (methods == EVMETHOD_AUTO)
493 return 0; 748 if (!enable_secure () && getenv ("LIBEV_METHODS"))
494 749 methods = atoi (getenv ("LIBEV_METHODS"));
750 else
495 ev_method = EVMETHOD_NONE; 751 methods = EVMETHOD_ANY;
752
753 method = 0;
754#if EV_USE_WIN32
755 if (!method && (methods & EVMETHOD_WIN32 )) method = win32_init (EV_A_ methods);
756#endif
757#if EV_USE_KQUEUE
758 if (!method && (methods & EVMETHOD_KQUEUE)) method = kqueue_init (EV_A_ methods);
759#endif
496#if EV_USE_EPOLL 760#if EV_USE_EPOLL
497 if (ev_method == EVMETHOD_NONE) epoll_init (flags); 761 if (!method && (methods & EVMETHOD_EPOLL )) method = epoll_init (EV_A_ methods);
762#endif
763#if EV_USE_POLL
764 if (!method && (methods & EVMETHOD_POLL )) method = poll_init (EV_A_ methods);
498#endif 765#endif
499#if EV_USE_SELECT 766#if EV_USE_SELECT
500 if (ev_method == EVMETHOD_NONE) select_init (flags); 767 if (!method && (methods & EVMETHOD_SELECT)) method = select_init (EV_A_ methods);
501#endif 768#endif
502 769
770 ev_init (&sigev, sigcb);
771 ev_set_priority (&sigev, EV_MAXPRI);
772 }
773}
774
775void
776loop_destroy (EV_P)
777{
778 int i;
779
780#if EV_USE_WIN32
781 if (method == EVMETHOD_WIN32 ) win32_destroy (EV_A);
782#endif
783#if EV_USE_KQUEUE
784 if (method == EVMETHOD_KQUEUE) kqueue_destroy (EV_A);
785#endif
786#if EV_USE_EPOLL
787 if (method == EVMETHOD_EPOLL ) epoll_destroy (EV_A);
788#endif
789#if EV_USE_POLL
790 if (method == EVMETHOD_POLL ) poll_destroy (EV_A);
791#endif
792#if EV_USE_SELECT
793 if (method == EVMETHOD_SELECT) select_destroy (EV_A);
794#endif
795
796 for (i = NUMPRI; i--; )
797 array_free (pending, [i]);
798
799 /* have to use the microsoft-never-gets-it-right macro */
800 array_free_microshit (fdchange);
801 array_free_microshit (timer);
802 array_free_microshit (periodic);
803 array_free_microshit (idle);
804 array_free_microshit (prepare);
805 array_free_microshit (check);
806
807 method = 0;
808}
809
810static void
811loop_fork (EV_P)
812{
813#if EV_USE_EPOLL
814 if (method == EVMETHOD_EPOLL ) epoll_fork (EV_A);
815#endif
816#if EV_USE_KQUEUE
817 if (method == EVMETHOD_KQUEUE) kqueue_fork (EV_A);
818#endif
819
820 if (ev_is_active (&sigev))
821 {
822 /* default loop */
823
824 ev_ref (EV_A);
825 ev_io_stop (EV_A_ &sigev);
826 close (sigpipe [0]);
827 close (sigpipe [1]);
828
829 while (pipe (sigpipe))
830 syserr ("(libev) error creating pipe");
831
832 siginit (EV_A);
833 }
834
835 postfork = 0;
836}
837
838#if EV_MULTIPLICITY
839struct ev_loop *
840ev_loop_new (int methods)
841{
842 struct ev_loop *loop = (struct ev_loop *)ev_malloc (sizeof (struct ev_loop));
843
844 memset (loop, 0, sizeof (struct ev_loop));
845
846 loop_init (EV_A_ methods);
847
848 if (ev_method (EV_A))
849 return loop;
850
851 return 0;
852}
853
854void
855ev_loop_destroy (EV_P)
856{
857 loop_destroy (EV_A);
858 ev_free (loop);
859}
860
861void
862ev_loop_fork (EV_P)
863{
864 postfork = 1;
865}
866
867#endif
868
869#if EV_MULTIPLICITY
870struct ev_loop *
871#else
872int
873#endif
874ev_default_loop (int methods)
875{
876 if (sigpipe [0] == sigpipe [1])
877 if (pipe (sigpipe))
878 return 0;
879
880 if (!default_loop)
881 {
882#if EV_MULTIPLICITY
883 struct ev_loop *loop = default_loop = &default_loop_struct;
884#else
885 default_loop = 1;
886#endif
887
888 loop_init (EV_A_ methods);
889
503 if (ev_method) 890 if (ev_method (EV_A))
504 { 891 {
505 ev_watcher_init (&sigev, sigcb);
506 siginit (); 892 siginit (EV_A);
507 893
894#ifndef WIN32
508 ev_signal_init (&childev, childcb, SIGCHLD); 895 ev_signal_init (&childev, childcb, SIGCHLD);
896 ev_set_priority (&childev, EV_MAXPRI);
509 ev_signal_start (&childev); 897 ev_signal_start (EV_A_ &childev);
898 ev_unref (EV_A); /* child watcher should not keep loop alive */
899#endif
510 } 900 }
901 else
902 default_loop = 0;
511 } 903 }
512 904
513 return ev_method; 905 return default_loop;
906}
907
908void
909ev_default_destroy (void)
910{
911#if EV_MULTIPLICITY
912 struct ev_loop *loop = default_loop;
913#endif
914
915#ifndef WIN32
916 ev_ref (EV_A); /* child watcher */
917 ev_signal_stop (EV_A_ &childev);
918#endif
919
920 ev_ref (EV_A); /* signal watcher */
921 ev_io_stop (EV_A_ &sigev);
922
923 close (sigpipe [0]); sigpipe [0] = 0;
924 close (sigpipe [1]); sigpipe [1] = 0;
925
926 loop_destroy (EV_A);
927}
928
929void
930ev_default_fork (void)
931{
932#if EV_MULTIPLICITY
933 struct ev_loop *loop = default_loop;
934#endif
935
936 if (method)
937 postfork = 1;
514} 938}
515 939
516/*****************************************************************************/ 940/*****************************************************************************/
517 941
518void
519ev_fork_prepare (void)
520{
521 /* nop */
522}
523
524void
525ev_fork_parent (void)
526{
527 /* nop */
528}
529
530void
531ev_fork_child (void)
532{
533#if EV_USE_EPOLL
534 if (ev_method == EVMETHOD_EPOLL)
535 epoll_postfork_child ();
536#endif
537
538 ev_io_stop (&sigev);
539 close (sigpipe [0]);
540 close (sigpipe [1]);
541 pipe (sigpipe);
542 siginit ();
543}
544
545/*****************************************************************************/
546
547static void 942static int
943any_pending (EV_P)
944{
945 int pri;
946
947 for (pri = NUMPRI; pri--; )
948 if (pendingcnt [pri])
949 return 1;
950
951 return 0;
952}
953
954static void
548call_pending (void) 955call_pending (EV_P)
549{ 956{
957 int pri;
958
959 for (pri = NUMPRI; pri--; )
550 while (pendingcnt) 960 while (pendingcnt [pri])
551 { 961 {
552 ANPENDING *p = pendings + --pendingcnt; 962 ANPENDING *p = pendings [pri] + --pendingcnt [pri];
553 963
554 if (p->w) 964 if (p->w)
555 { 965 {
556 p->w->pending = 0; 966 p->w->pending = 0;
557 p->w->cb (p->w, p->events); 967 EV_CB_INVOKE (p->w, p->events);
558 } 968 }
559 } 969 }
560} 970}
561 971
562static void 972static void
563timers_reify (void) 973timers_reify (EV_P)
564{ 974{
565 while (timercnt && timers [0]->at <= now) 975 while (timercnt && ((WT)timers [0])->at <= mn_now)
566 { 976 {
567 struct ev_timer *w = timers [0]; 977 struct ev_timer *w = timers [0];
978
979 assert (("inactive timer on timer heap detected", ev_is_active (w)));
568 980
569 /* first reschedule or stop timer */ 981 /* first reschedule or stop timer */
570 if (w->repeat) 982 if (w->repeat)
571 { 983 {
572 assert (("negative ev_timer repeat value found while processing timers", w->repeat > 0.)); 984 assert (("negative ev_timer repeat value found while processing timers", w->repeat > 0.));
573 w->at = now + w->repeat; 985 ((WT)w)->at = mn_now + w->repeat;
574 downheap ((WT *)timers, timercnt, 0); 986 downheap ((WT *)timers, timercnt, 0);
575 } 987 }
576 else 988 else
577 ev_timer_stop (w); /* nonrepeating: stop timer */ 989 ev_timer_stop (EV_A_ w); /* nonrepeating: stop timer */
578 990
579 event ((W)w, EV_TIMEOUT); 991 ev_feed_event (EV_A_ (W)w, EV_TIMEOUT);
580 } 992 }
581} 993}
582 994
583static void 995static void
584periodics_reify (void) 996periodics_reify (EV_P)
585{ 997{
586 while (periodiccnt && periodics [0]->at <= ev_now) 998 while (periodiccnt && ((WT)periodics [0])->at <= ev_rt_now)
587 { 999 {
588 struct ev_periodic *w = periodics [0]; 1000 struct ev_periodic *w = periodics [0];
589 1001
1002 assert (("inactive timer on periodic heap detected", ev_is_active (w)));
1003
590 /* first reschedule or stop timer */ 1004 /* first reschedule or stop timer */
591 if (w->interval) 1005 if (w->reschedule_cb)
592 { 1006 {
1007 ev_tstamp at = ((WT)w)->at = w->reschedule_cb (w, ev_rt_now + 0.0001);
1008
1009 assert (("ev_periodic reschedule callback returned time in the past", ((WT)w)->at > ev_rt_now));
1010 downheap ((WT *)periodics, periodiccnt, 0);
1011 }
1012 else if (w->interval)
1013 {
593 w->at += floor ((ev_now - w->at) / w->interval + 1.) * w->interval; 1014 ((WT)w)->at += floor ((ev_rt_now - ((WT)w)->at) / w->interval + 1.) * w->interval;
594 assert (("ev_periodic timeout in the past detected while processing timers, negative interval?", w->at > ev_now)); 1015 assert (("ev_periodic timeout in the past detected while processing timers, negative interval?", ((WT)w)->at > ev_rt_now));
595 downheap ((WT *)periodics, periodiccnt, 0); 1016 downheap ((WT *)periodics, periodiccnt, 0);
596 } 1017 }
597 else 1018 else
598 ev_periodic_stop (w); /* nonrepeating: stop timer */ 1019 ev_periodic_stop (EV_A_ w); /* nonrepeating: stop timer */
599 1020
600 event ((W)w, EV_PERIODIC); 1021 ev_feed_event (EV_A_ (W)w, EV_PERIODIC);
601 } 1022 }
602} 1023}
603 1024
604static void 1025static void
605periodics_reschedule (ev_tstamp diff) 1026periodics_reschedule (EV_P)
606{ 1027{
607 int i; 1028 int i;
608 1029
609 /* adjust periodics after time jump */ 1030 /* adjust periodics after time jump */
610 for (i = 0; i < periodiccnt; ++i) 1031 for (i = 0; i < periodiccnt; ++i)
611 { 1032 {
612 struct ev_periodic *w = periodics [i]; 1033 struct ev_periodic *w = periodics [i];
613 1034
1035 if (w->reschedule_cb)
1036 ((WT)w)->at = w->reschedule_cb (w, ev_rt_now);
614 if (w->interval) 1037 else if (w->interval)
615 {
616 ev_tstamp diff = ceil ((ev_now - w->at) / w->interval) * w->interval; 1038 ((WT)w)->at += ceil ((ev_rt_now - ((WT)w)->at) / w->interval) * w->interval;
617
618 if (fabs (diff) >= 1e-4)
619 {
620 ev_periodic_stop (w);
621 ev_periodic_start (w);
622
623 i = 0; /* restart loop, inefficient, but time jumps should be rare */
624 }
625 }
626 } 1039 }
627}
628 1040
629static int 1041 /* now rebuild the heap */
1042 for (i = periodiccnt >> 1; i--; )
1043 downheap ((WT *)periodics, periodiccnt, i);
1044}
1045
1046inline int
630time_update_monotonic (void) 1047time_update_monotonic (EV_P)
631{ 1048{
632 now = get_clock (); 1049 mn_now = get_clock ();
633 1050
634 if (expect_true (now - now_floor < MIN_TIMEJUMP * .5)) 1051 if (expect_true (mn_now - now_floor < MIN_TIMEJUMP * .5))
635 { 1052 {
636 ev_now = now + diff; 1053 ev_rt_now = rtmn_diff + mn_now;
637 return 0; 1054 return 0;
638 } 1055 }
639 else 1056 else
640 { 1057 {
641 now_floor = now; 1058 now_floor = mn_now;
642 ev_now = ev_time (); 1059 ev_rt_now = ev_time ();
643 return 1; 1060 return 1;
644 } 1061 }
645} 1062}
646 1063
647static void 1064static void
648time_update (void) 1065time_update (EV_P)
649{ 1066{
650 int i; 1067 int i;
651 1068
652#if EV_USE_MONOTONIC 1069#if EV_USE_MONOTONIC
653 if (expect_true (have_monotonic)) 1070 if (expect_true (have_monotonic))
654 { 1071 {
655 if (time_update_monotonic ()) 1072 if (time_update_monotonic (EV_A))
656 { 1073 {
657 ev_tstamp odiff = diff; 1074 ev_tstamp odiff = rtmn_diff;
658 1075
659 for (i = 4; --i; ) /* loop a few times, before making important decisions */ 1076 for (i = 4; --i; ) /* loop a few times, before making important decisions */
660 { 1077 {
661 diff = ev_now - now; 1078 rtmn_diff = ev_rt_now - mn_now;
662 1079
663 if (fabs (odiff - diff) < MIN_TIMEJUMP) 1080 if (fabs (odiff - rtmn_diff) < MIN_TIMEJUMP)
664 return; /* all is well */ 1081 return; /* all is well */
665 1082
666 ev_now = ev_time (); 1083 ev_rt_now = ev_time ();
667 now = get_clock (); 1084 mn_now = get_clock ();
668 now_floor = now; 1085 now_floor = mn_now;
669 } 1086 }
670 1087
671 periodics_reschedule (diff - odiff); 1088 periodics_reschedule (EV_A);
672 /* no timer adjustment, as the monotonic clock doesn't jump */ 1089 /* no timer adjustment, as the monotonic clock doesn't jump */
1090 /* timers_reschedule (EV_A_ rtmn_diff - odiff) */
673 } 1091 }
674 } 1092 }
675 else 1093 else
676#endif 1094#endif
677 { 1095 {
678 ev_now = ev_time (); 1096 ev_rt_now = ev_time ();
679 1097
680 if (expect_false (now > ev_now || now < ev_now - MAX_BLOCKTIME - MIN_TIMEJUMP)) 1098 if (expect_false (mn_now > ev_rt_now || mn_now < ev_rt_now - MAX_BLOCKTIME - MIN_TIMEJUMP))
681 { 1099 {
682 periodics_reschedule (ev_now - now); 1100 periodics_reschedule (EV_A);
683 1101
684 /* adjust timers. this is easy, as the offset is the same for all */ 1102 /* adjust timers. this is easy, as the offset is the same for all */
685 for (i = 0; i < timercnt; ++i) 1103 for (i = 0; i < timercnt; ++i)
686 timers [i]->at += diff; 1104 ((WT)timers [i])->at += ev_rt_now - mn_now;
687 } 1105 }
688 1106
689 now = ev_now; 1107 mn_now = ev_rt_now;
690 } 1108 }
691} 1109}
692 1110
693int ev_loop_done; 1111void
1112ev_ref (EV_P)
1113{
1114 ++activecnt;
1115}
694 1116
1117void
1118ev_unref (EV_P)
1119{
1120 --activecnt;
1121}
1122
1123static int loop_done;
1124
1125void
695void ev_loop (int flags) 1126ev_loop (EV_P_ int flags)
696{ 1127{
697 double block; 1128 double block;
698 ev_loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0; 1129 loop_done = flags & (EVLOOP_ONESHOT | EVLOOP_NONBLOCK) ? 1 : 0;
699 1130
700 do 1131 do
701 { 1132 {
702 /* queue check watchers (and execute them) */ 1133 /* queue check watchers (and execute them) */
703 if (expect_false (preparecnt)) 1134 if (expect_false (preparecnt))
704 { 1135 {
705 queue_events ((W *)prepares, preparecnt, EV_PREPARE); 1136 queue_events (EV_A_ (W *)prepares, preparecnt, EV_PREPARE);
706 call_pending (); 1137 call_pending (EV_A);
707 } 1138 }
708 1139
1140 /* we might have forked, so reify kernel state if necessary */
1141 if (expect_false (postfork))
1142 loop_fork (EV_A);
1143
709 /* update fd-related kernel structures */ 1144 /* update fd-related kernel structures */
710 fd_reify (); 1145 fd_reify (EV_A);
711 1146
712 /* calculate blocking time */ 1147 /* calculate blocking time */
713 1148
714 /* we only need this for !monotonic clockor timers, but as we basically 1149 /* we only need this for !monotonic clock or timers, but as we basically
715 always have timers, we just calculate it always */ 1150 always have timers, we just calculate it always */
716#if EV_USE_MONOTONIC 1151#if EV_USE_MONOTONIC
717 if (expect_true (have_monotonic)) 1152 if (expect_true (have_monotonic))
718 time_update_monotonic (); 1153 time_update_monotonic (EV_A);
719 else 1154 else
720#endif 1155#endif
721 { 1156 {
722 ev_now = ev_time (); 1157 ev_rt_now = ev_time ();
723 now = ev_now; 1158 mn_now = ev_rt_now;
724 } 1159 }
725 1160
726 if (flags & EVLOOP_NONBLOCK || idlecnt) 1161 if (flags & EVLOOP_NONBLOCK || idlecnt)
727 block = 0.; 1162 block = 0.;
728 else 1163 else
729 { 1164 {
730 block = MAX_BLOCKTIME; 1165 block = MAX_BLOCKTIME;
731 1166
732 if (timercnt) 1167 if (timercnt)
733 { 1168 {
734 ev_tstamp to = timers [0]->at - now + method_fudge; 1169 ev_tstamp to = ((WT)timers [0])->at - mn_now + method_fudge;
735 if (block > to) block = to; 1170 if (block > to) block = to;
736 } 1171 }
737 1172
738 if (periodiccnt) 1173 if (periodiccnt)
739 { 1174 {
740 ev_tstamp to = periodics [0]->at - ev_now + method_fudge; 1175 ev_tstamp to = ((WT)periodics [0])->at - ev_rt_now + method_fudge;
741 if (block > to) block = to; 1176 if (block > to) block = to;
742 } 1177 }
743 1178
744 if (block < 0.) block = 0.; 1179 if (block < 0.) block = 0.;
745 } 1180 }
746 1181
747 method_poll (block); 1182 method_poll (EV_A_ block);
748 1183
749 /* update ev_now, do magic */ 1184 /* update ev_rt_now, do magic */
750 time_update (); 1185 time_update (EV_A);
751 1186
752 /* queue pending timers and reschedule them */ 1187 /* queue pending timers and reschedule them */
753 timers_reify (); /* relative timers called last */ 1188 timers_reify (EV_A); /* relative timers called last */
754 periodics_reify (); /* absolute timers called first */ 1189 periodics_reify (EV_A); /* absolute timers called first */
755 1190
756 /* queue idle watchers unless io or timers are pending */ 1191 /* queue idle watchers unless io or timers are pending */
757 if (!pendingcnt) 1192 if (idlecnt && !any_pending (EV_A))
758 queue_events ((W *)idles, idlecnt, EV_IDLE); 1193 queue_events (EV_A_ (W *)idles, idlecnt, EV_IDLE);
759 1194
760 /* queue check watchers, to be executed first */ 1195 /* queue check watchers, to be executed first */
761 if (checkcnt) 1196 if (checkcnt)
762 queue_events ((W *)checks, checkcnt, EV_CHECK); 1197 queue_events (EV_A_ (W *)checks, checkcnt, EV_CHECK);
763 1198
764 call_pending (); 1199 call_pending (EV_A);
765 } 1200 }
766 while (!ev_loop_done); 1201 while (activecnt && !loop_done);
767 1202
768 if (ev_loop_done != 2) 1203 if (loop_done != 2)
769 ev_loop_done = 0; 1204 loop_done = 0;
1205}
1206
1207void
1208ev_unloop (EV_P_ int how)
1209{
1210 loop_done = how;
770} 1211}
771 1212
772/*****************************************************************************/ 1213/*****************************************************************************/
773 1214
774static void 1215inline void
775wlist_add (WL *head, WL elem) 1216wlist_add (WL *head, WL elem)
776{ 1217{
777 elem->next = *head; 1218 elem->next = *head;
778 *head = elem; 1219 *head = elem;
779} 1220}
780 1221
781static void 1222inline void
782wlist_del (WL *head, WL elem) 1223wlist_del (WL *head, WL elem)
783{ 1224{
784 while (*head) 1225 while (*head)
785 { 1226 {
786 if (*head == elem) 1227 if (*head == elem)
791 1232
792 head = &(*head)->next; 1233 head = &(*head)->next;
793 } 1234 }
794} 1235}
795 1236
796static void 1237inline void
797ev_clear_pending (W w) 1238ev_clear_pending (EV_P_ W w)
798{ 1239{
799 if (w->pending) 1240 if (w->pending)
800 { 1241 {
801 pendings [w->pending - 1].w = 0; 1242 pendings [ABSPRI (w)][w->pending - 1].w = 0;
802 w->pending = 0; 1243 w->pending = 0;
803 } 1244 }
804} 1245}
805 1246
806static void 1247inline void
807ev_start (W w, int active) 1248ev_start (EV_P_ W w, int active)
808{ 1249{
1250 if (w->priority < EV_MINPRI) w->priority = EV_MINPRI;
1251 if (w->priority > EV_MAXPRI) w->priority = EV_MAXPRI;
1252
809 w->active = active; 1253 w->active = active;
1254 ev_ref (EV_A);
810} 1255}
811 1256
812static void 1257inline void
813ev_stop (W w) 1258ev_stop (EV_P_ W w)
814{ 1259{
1260 ev_unref (EV_A);
815 w->active = 0; 1261 w->active = 0;
816} 1262}
817 1263
818/*****************************************************************************/ 1264/*****************************************************************************/
819 1265
820void 1266void
821ev_io_start (struct ev_io *w) 1267ev_io_start (EV_P_ struct ev_io *w)
822{ 1268{
823 int fd = w->fd; 1269 int fd = w->fd;
824 1270
825 if (ev_is_active (w)) 1271 if (ev_is_active (w))
826 return; 1272 return;
827 1273
828 assert (("ev_io_start called with negative fd", fd >= 0)); 1274 assert (("ev_io_start called with negative fd", fd >= 0));
829 1275
830 ev_start ((W)w, 1); 1276 ev_start (EV_A_ (W)w, 1);
831 array_needsize (anfds, anfdmax, fd + 1, anfds_init); 1277 array_needsize (ANFD, anfds, anfdmax, fd + 1, anfds_init);
832 wlist_add ((WL *)&anfds[fd].head, (WL)w); 1278 wlist_add ((WL *)&anfds[fd].head, (WL)w);
833 1279
834 fd_change (fd); 1280 fd_change (EV_A_ fd);
835} 1281}
836 1282
837void 1283void
838ev_io_stop (struct ev_io *w) 1284ev_io_stop (EV_P_ struct ev_io *w)
839{ 1285{
840 ev_clear_pending ((W)w); 1286 ev_clear_pending (EV_A_ (W)w);
841 if (!ev_is_active (w)) 1287 if (!ev_is_active (w))
842 return; 1288 return;
843 1289
844 wlist_del ((WL *)&anfds[w->fd].head, (WL)w); 1290 wlist_del ((WL *)&anfds[w->fd].head, (WL)w);
845 ev_stop ((W)w); 1291 ev_stop (EV_A_ (W)w);
846 1292
847 fd_change (w->fd); 1293 fd_change (EV_A_ w->fd);
848} 1294}
849 1295
850void 1296void
851ev_timer_start (struct ev_timer *w) 1297ev_timer_start (EV_P_ struct ev_timer *w)
852{ 1298{
853 if (ev_is_active (w)) 1299 if (ev_is_active (w))
854 return; 1300 return;
855 1301
856 w->at += now; 1302 ((WT)w)->at += mn_now;
857 1303
858 assert (("ev_timer_start called with negative timer repeat value", w->repeat >= 0.)); 1304 assert (("ev_timer_start called with negative timer repeat value", w->repeat >= 0.));
859 1305
860 ev_start ((W)w, ++timercnt); 1306 ev_start (EV_A_ (W)w, ++timercnt);
861 array_needsize (timers, timermax, timercnt, ); 1307 array_needsize (struct ev_timer *, timers, timermax, timercnt, (void));
862 timers [timercnt - 1] = w; 1308 timers [timercnt - 1] = w;
863 upheap ((WT *)timers, timercnt - 1); 1309 upheap ((WT *)timers, timercnt - 1);
864}
865 1310
1311 assert (("internal timer heap corruption", timers [((W)w)->active - 1] == w));
1312}
1313
866void 1314void
867ev_timer_stop (struct ev_timer *w) 1315ev_timer_stop (EV_P_ struct ev_timer *w)
868{ 1316{
869 ev_clear_pending ((W)w); 1317 ev_clear_pending (EV_A_ (W)w);
870 if (!ev_is_active (w)) 1318 if (!ev_is_active (w))
871 return; 1319 return;
872 1320
1321 assert (("internal timer heap corruption", timers [((W)w)->active - 1] == w));
1322
873 if (w->active < timercnt--) 1323 if (((W)w)->active < timercnt--)
874 { 1324 {
875 timers [w->active - 1] = timers [timercnt]; 1325 timers [((W)w)->active - 1] = timers [timercnt];
876 downheap ((WT *)timers, timercnt, w->active - 1); 1326 downheap ((WT *)timers, timercnt, ((W)w)->active - 1);
877 } 1327 }
878 1328
879 w->at = w->repeat; 1329 ((WT)w)->at = w->repeat;
880 1330
881 ev_stop ((W)w); 1331 ev_stop (EV_A_ (W)w);
882} 1332}
883 1333
884void 1334void
885ev_timer_again (struct ev_timer *w) 1335ev_timer_again (EV_P_ struct ev_timer *w)
886{ 1336{
887 if (ev_is_active (w)) 1337 if (ev_is_active (w))
888 { 1338 {
889 if (w->repeat) 1339 if (w->repeat)
890 {
891 w->at = now + w->repeat;
892 downheap ((WT *)timers, timercnt, w->active - 1); 1340 adjustheap ((WT *)timers, timercnt, ((W)w)->active - 1, mn_now + w->repeat);
893 }
894 else 1341 else
895 ev_timer_stop (w); 1342 ev_timer_stop (EV_A_ w);
896 } 1343 }
897 else if (w->repeat) 1344 else if (w->repeat)
898 ev_timer_start (w); 1345 ev_timer_start (EV_A_ w);
899} 1346}
900 1347
901void 1348void
902ev_periodic_start (struct ev_periodic *w) 1349ev_periodic_start (EV_P_ struct ev_periodic *w)
903{ 1350{
904 if (ev_is_active (w)) 1351 if (ev_is_active (w))
905 return; 1352 return;
906 1353
1354 if (w->reschedule_cb)
1355 ((WT)w)->at = w->reschedule_cb (w, ev_rt_now);
1356 else if (w->interval)
1357 {
907 assert (("ev_periodic_start called with negative interval value", w->interval >= 0.)); 1358 assert (("ev_periodic_start called with negative interval value", w->interval >= 0.));
908
909 /* this formula differs from the one in periodic_reify because we do not always round up */ 1359 /* this formula differs from the one in periodic_reify because we do not always round up */
910 if (w->interval)
911 w->at += ceil ((ev_now - w->at) / w->interval) * w->interval; 1360 ((WT)w)->at += ceil ((ev_rt_now - ((WT)w)->at) / w->interval) * w->interval;
1361 }
912 1362
913 ev_start ((W)w, ++periodiccnt); 1363 ev_start (EV_A_ (W)w, ++periodiccnt);
914 array_needsize (periodics, periodicmax, periodiccnt, ); 1364 array_needsize (struct ev_periodic *, periodics, periodicmax, periodiccnt, (void));
915 periodics [periodiccnt - 1] = w; 1365 periodics [periodiccnt - 1] = w;
916 upheap ((WT *)periodics, periodiccnt - 1); 1366 upheap ((WT *)periodics, periodiccnt - 1);
917}
918 1367
1368 assert (("internal periodic heap corruption", periodics [((W)w)->active - 1] == w));
1369}
1370
919void 1371void
920ev_periodic_stop (struct ev_periodic *w) 1372ev_periodic_stop (EV_P_ struct ev_periodic *w)
921{ 1373{
922 ev_clear_pending ((W)w); 1374 ev_clear_pending (EV_A_ (W)w);
923 if (!ev_is_active (w)) 1375 if (!ev_is_active (w))
924 return; 1376 return;
925 1377
1378 assert (("internal periodic heap corruption", periodics [((W)w)->active - 1] == w));
1379
926 if (w->active < periodiccnt--) 1380 if (((W)w)->active < periodiccnt--)
927 { 1381 {
928 periodics [w->active - 1] = periodics [periodiccnt]; 1382 periodics [((W)w)->active - 1] = periodics [periodiccnt];
929 downheap ((WT *)periodics, periodiccnt, w->active - 1); 1383 downheap ((WT *)periodics, periodiccnt, ((W)w)->active - 1);
930 } 1384 }
931 1385
932 ev_stop ((W)w); 1386 ev_stop (EV_A_ (W)w);
933} 1387}
934 1388
935void 1389void
1390ev_periodic_again (EV_P_ struct ev_periodic *w)
1391{
1392 /* TODO: use adjustheap and recalculation */
1393 ev_periodic_stop (EV_A_ w);
1394 ev_periodic_start (EV_A_ w);
1395}
1396
1397void
936ev_signal_start (struct ev_signal *w) 1398ev_idle_start (EV_P_ struct ev_idle *w)
937{ 1399{
938 if (ev_is_active (w)) 1400 if (ev_is_active (w))
939 return; 1401 return;
940 1402
1403 ev_start (EV_A_ (W)w, ++idlecnt);
1404 array_needsize (struct ev_idle *, idles, idlemax, idlecnt, (void));
1405 idles [idlecnt - 1] = w;
1406}
1407
1408void
1409ev_idle_stop (EV_P_ struct ev_idle *w)
1410{
1411 ev_clear_pending (EV_A_ (W)w);
1412 if (ev_is_active (w))
1413 return;
1414
1415 idles [((W)w)->active - 1] = idles [--idlecnt];
1416 ev_stop (EV_A_ (W)w);
1417}
1418
1419void
1420ev_prepare_start (EV_P_ struct ev_prepare *w)
1421{
1422 if (ev_is_active (w))
1423 return;
1424
1425 ev_start (EV_A_ (W)w, ++preparecnt);
1426 array_needsize (struct ev_prepare *, prepares, preparemax, preparecnt, (void));
1427 prepares [preparecnt - 1] = w;
1428}
1429
1430void
1431ev_prepare_stop (EV_P_ struct ev_prepare *w)
1432{
1433 ev_clear_pending (EV_A_ (W)w);
1434 if (ev_is_active (w))
1435 return;
1436
1437 prepares [((W)w)->active - 1] = prepares [--preparecnt];
1438 ev_stop (EV_A_ (W)w);
1439}
1440
1441void
1442ev_check_start (EV_P_ struct ev_check *w)
1443{
1444 if (ev_is_active (w))
1445 return;
1446
1447 ev_start (EV_A_ (W)w, ++checkcnt);
1448 array_needsize (struct ev_check *, checks, checkmax, checkcnt, (void));
1449 checks [checkcnt - 1] = w;
1450}
1451
1452void
1453ev_check_stop (EV_P_ struct ev_check *w)
1454{
1455 ev_clear_pending (EV_A_ (W)w);
1456 if (ev_is_active (w))
1457 return;
1458
1459 checks [((W)w)->active - 1] = checks [--checkcnt];
1460 ev_stop (EV_A_ (W)w);
1461}
1462
1463#ifndef SA_RESTART
1464# define SA_RESTART 0
1465#endif
1466
1467void
1468ev_signal_start (EV_P_ struct ev_signal *w)
1469{
1470#if EV_MULTIPLICITY
1471 assert (("signal watchers are only supported in the default loop", loop == default_loop));
1472#endif
1473 if (ev_is_active (w))
1474 return;
1475
941 assert (("ev_signal_start called with illegal signal number", w->signum > 0)); 1476 assert (("ev_signal_start called with illegal signal number", w->signum > 0));
942 1477
943 ev_start ((W)w, 1); 1478 ev_start (EV_A_ (W)w, 1);
944 array_needsize (signals, signalmax, w->signum, signals_init); 1479 array_needsize (ANSIG, signals, signalmax, w->signum, signals_init);
945 wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w); 1480 wlist_add ((WL *)&signals [w->signum - 1].head, (WL)w);
946 1481
947 if (!w->next) 1482 if (!((WL)w)->next)
948 { 1483 {
1484#if WIN32
1485 signal (w->signum, sighandler);
1486#else
949 struct sigaction sa; 1487 struct sigaction sa;
950 sa.sa_handler = sighandler; 1488 sa.sa_handler = sighandler;
951 sigfillset (&sa.sa_mask); 1489 sigfillset (&sa.sa_mask);
952 sa.sa_flags = 0; 1490 sa.sa_flags = SA_RESTART; /* if restarting works we save one iteration */
953 sigaction (w->signum, &sa, 0); 1491 sigaction (w->signum, &sa, 0);
1492#endif
954 } 1493 }
955} 1494}
956 1495
957void 1496void
958ev_signal_stop (struct ev_signal *w) 1497ev_signal_stop (EV_P_ struct ev_signal *w)
959{ 1498{
960 ev_clear_pending ((W)w); 1499 ev_clear_pending (EV_A_ (W)w);
961 if (!ev_is_active (w)) 1500 if (!ev_is_active (w))
962 return; 1501 return;
963 1502
964 wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w); 1503 wlist_del ((WL *)&signals [w->signum - 1].head, (WL)w);
965 ev_stop ((W)w); 1504 ev_stop (EV_A_ (W)w);
966 1505
967 if (!signals [w->signum - 1].head) 1506 if (!signals [w->signum - 1].head)
968 signal (w->signum, SIG_DFL); 1507 signal (w->signum, SIG_DFL);
969} 1508}
970 1509
971void 1510void
972ev_idle_start (struct ev_idle *w) 1511ev_child_start (EV_P_ struct ev_child *w)
973{ 1512{
1513#if EV_MULTIPLICITY
1514 assert (("child watchers are only supported in the default loop", loop == default_loop));
1515#endif
974 if (ev_is_active (w)) 1516 if (ev_is_active (w))
975 return; 1517 return;
976 1518
977 ev_start ((W)w, ++idlecnt); 1519 ev_start (EV_A_ (W)w, 1);
978 array_needsize (idles, idlemax, idlecnt, ); 1520 wlist_add ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
979 idles [idlecnt - 1] = w;
980} 1521}
981 1522
982void 1523void
983ev_idle_stop (struct ev_idle *w) 1524ev_child_stop (EV_P_ struct ev_child *w)
984{ 1525{
985 ev_clear_pending ((W)w); 1526 ev_clear_pending (EV_A_ (W)w);
986 if (ev_is_active (w)) 1527 if (ev_is_active (w))
987 return; 1528 return;
988 1529
989 idles [w->active - 1] = idles [--idlecnt];
990 ev_stop ((W)w);
991}
992
993void
994ev_prepare_start (struct ev_prepare *w)
995{
996 if (ev_is_active (w))
997 return;
998
999 ev_start ((W)w, ++preparecnt);
1000 array_needsize (prepares, preparemax, preparecnt, );
1001 prepares [preparecnt - 1] = w;
1002}
1003
1004void
1005ev_prepare_stop (struct ev_prepare *w)
1006{
1007 ev_clear_pending ((W)w);
1008 if (ev_is_active (w))
1009 return;
1010
1011 prepares [w->active - 1] = prepares [--preparecnt];
1012 ev_stop ((W)w);
1013}
1014
1015void
1016ev_check_start (struct ev_check *w)
1017{
1018 if (ev_is_active (w))
1019 return;
1020
1021 ev_start ((W)w, ++checkcnt);
1022 array_needsize (checks, checkmax, checkcnt, );
1023 checks [checkcnt - 1] = w;
1024}
1025
1026void
1027ev_check_stop (struct ev_check *w)
1028{
1029 ev_clear_pending ((W)w);
1030 if (ev_is_active (w))
1031 return;
1032
1033 checks [w->active - 1] = checks [--checkcnt];
1034 ev_stop ((W)w);
1035}
1036
1037void
1038ev_child_start (struct ev_child *w)
1039{
1040 if (ev_is_active (w))
1041 return;
1042
1043 ev_start ((W)w, 1);
1044 wlist_add ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
1045}
1046
1047void
1048ev_child_stop (struct ev_child *w)
1049{
1050 ev_clear_pending ((W)w);
1051 if (ev_is_active (w))
1052 return;
1053
1054 wlist_del ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w); 1530 wlist_del ((WL *)&childs [w->pid & (PID_HASHSIZE - 1)], (WL)w);
1055 ev_stop ((W)w); 1531 ev_stop (EV_A_ (W)w);
1056} 1532}
1057 1533
1058/*****************************************************************************/ 1534/*****************************************************************************/
1059 1535
1060struct ev_once 1536struct ev_once
1064 void (*cb)(int revents, void *arg); 1540 void (*cb)(int revents, void *arg);
1065 void *arg; 1541 void *arg;
1066}; 1542};
1067 1543
1068static void 1544static void
1069once_cb (struct ev_once *once, int revents) 1545once_cb (EV_P_ struct ev_once *once, int revents)
1070{ 1546{
1071 void (*cb)(int revents, void *arg) = once->cb; 1547 void (*cb)(int revents, void *arg) = once->cb;
1072 void *arg = once->arg; 1548 void *arg = once->arg;
1073 1549
1074 ev_io_stop (&once->io); 1550 ev_io_stop (EV_A_ &once->io);
1075 ev_timer_stop (&once->to); 1551 ev_timer_stop (EV_A_ &once->to);
1076 free (once); 1552 ev_free (once);
1077 1553
1078 cb (revents, arg); 1554 cb (revents, arg);
1079} 1555}
1080 1556
1081static void 1557static void
1082once_cb_io (struct ev_io *w, int revents) 1558once_cb_io (EV_P_ struct ev_io *w, int revents)
1083{ 1559{
1084 once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents); 1560 once_cb (EV_A_ (struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents);
1085} 1561}
1086 1562
1087static void 1563static void
1088once_cb_to (struct ev_timer *w, int revents) 1564once_cb_to (EV_P_ struct ev_timer *w, int revents)
1089{ 1565{
1090 once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents); 1566 once_cb (EV_A_ (struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents);
1091} 1567}
1092 1568
1093void 1569void
1094ev_once (int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg) 1570ev_once (EV_P_ int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg)
1095{ 1571{
1096 struct ev_once *once = malloc (sizeof (struct ev_once)); 1572 struct ev_once *once = (struct ev_once *)ev_malloc (sizeof (struct ev_once));
1097 1573
1098 if (!once) 1574 if (!once)
1099 cb (EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT, arg); 1575 cb (EV_ERROR | EV_READ | EV_WRITE | EV_TIMEOUT, arg);
1100 else 1576 else
1101 { 1577 {
1102 once->cb = cb; 1578 once->cb = cb;
1103 once->arg = arg; 1579 once->arg = arg;
1104 1580
1105 ev_watcher_init (&once->io, once_cb_io); 1581 ev_init (&once->io, once_cb_io);
1106 if (fd >= 0) 1582 if (fd >= 0)
1107 { 1583 {
1108 ev_io_set (&once->io, fd, events); 1584 ev_io_set (&once->io, fd, events);
1109 ev_io_start (&once->io); 1585 ev_io_start (EV_A_ &once->io);
1110 } 1586 }
1111 1587
1112 ev_watcher_init (&once->to, once_cb_to); 1588 ev_init (&once->to, once_cb_to);
1113 if (timeout >= 0.) 1589 if (timeout >= 0.)
1114 { 1590 {
1115 ev_timer_set (&once->to, timeout, 0.); 1591 ev_timer_set (&once->to, timeout, 0.);
1116 ev_timer_start (&once->to); 1592 ev_timer_start (EV_A_ &once->to);
1117 } 1593 }
1118 } 1594 }
1119} 1595}
1120 1596
1121/*****************************************************************************/
1122
1123#if 0
1124
1125struct ev_io wio;
1126
1127static void
1128sin_cb (struct ev_io *w, int revents)
1129{
1130 fprintf (stderr, "sin %d, revents %d\n", w->fd, revents);
1131}
1132
1133static void
1134ocb (struct ev_timer *w, int revents)
1135{
1136 //fprintf (stderr, "timer %f,%f (%x) (%f) d%p\n", w->at, w->repeat, revents, w->at - ev_time (), w->data);
1137 ev_timer_stop (w);
1138 ev_timer_start (w);
1139}
1140
1141static void
1142scb (struct ev_signal *w, int revents)
1143{
1144 fprintf (stderr, "signal %x,%d\n", revents, w->signum);
1145 ev_io_stop (&wio);
1146 ev_io_start (&wio);
1147}
1148
1149static void
1150gcb (struct ev_signal *w, int revents)
1151{
1152 fprintf (stderr, "generic %x\n", revents);
1153
1154}
1155
1156int main (void)
1157{
1158 ev_init (0);
1159
1160 ev_io_init (&wio, sin_cb, 0, EV_READ);
1161 ev_io_start (&wio);
1162
1163 struct ev_timer t[10000];
1164
1165#if 0
1166 int i;
1167 for (i = 0; i < 10000; ++i)
1168 {
1169 struct ev_timer *w = t + i;
1170 ev_watcher_init (w, ocb, i);
1171 ev_timer_init_abs (w, ocb, drand48 (), 0.99775533);
1172 ev_timer_start (w);
1173 if (drand48 () < 0.5)
1174 ev_timer_stop (w);
1175 }
1176#endif
1177
1178 struct ev_timer t1;
1179 ev_timer_init (&t1, ocb, 5, 10);
1180 ev_timer_start (&t1);
1181
1182 struct ev_signal sig;
1183 ev_signal_init (&sig, scb, SIGQUIT);
1184 ev_signal_start (&sig);
1185
1186 struct ev_check cw;
1187 ev_check_init (&cw, gcb);
1188 ev_check_start (&cw);
1189
1190 struct ev_idle iw;
1191 ev_idle_init (&iw, gcb);
1192 ev_idle_start (&iw);
1193
1194 ev_loop (0);
1195
1196 return 0;
1197}
1198
1199#endif
1200
1201
1202
1203

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines