diff options
| -rw-r--r-- | ev.c | 108 | ||||
| -rw-r--r-- | ev.h | 22 | 
2 files changed, 115 insertions, 15 deletions
| @@ -3,6 +3,7 @@  #include <unistd.h>  #include <fcntl.h>  #include <signal.h> +#include <stddef.h>  #include <stdio.h> @@ -127,10 +128,13 @@ static int pendingmax, pendingcnt;  static void  event (W w, int events)  { -  w->pending = ++pendingcnt; -  array_needsize (pendings, pendingmax, pendingcnt, ); -  pendings [pendingcnt - 1].w      = w; -  pendings [pendingcnt - 1].events = events; +  if (w->active) +    { +      w->pending = ++pendingcnt; +      array_needsize (pendings, pendingmax, pendingcnt, ); +      pendings [pendingcnt - 1].w      = w; +      pendings [pendingcnt - 1].events = events; +    }  }  static void @@ -408,6 +412,8 @@ timers_reify ()      {        struct ev_timer *w = timers [0]; +      event ((W)w, EV_TIMEOUT); +        /* first reschedule or stop timer */        if (w->repeat)          { @@ -417,8 +423,6 @@ timers_reify ()          }        else          evtimer_stop (w); /* nonrepeating: stop timer */ - -      event ((W)w, EV_TIMEOUT);      }  } @@ -601,18 +605,24 @@ wlist_del (WL *head, WL elem)  }  static void +ev_clear (W w) +{ +  if (w->pending) +    { +      pendings [w->pending - 1].w = 0; +      w->pending = 0; +    } +} + +static void  ev_start (W w, int active)  { -  w->pending = 0;    w->active = active;  }  static void  ev_stop (W w)  { -  if (w->pending) -    pendings [w->pending - 1].w = 0; -    w->active = 0;  } @@ -638,6 +648,7 @@ evio_start (struct ev_io *w)  void  evio_stop (struct ev_io *w)  { +  ev_clear ((W)w);    if (!ev_is_active (w))      return; @@ -649,7 +660,6 @@ evio_stop (struct ev_io *w)    fdchanges [fdchangecnt - 1] = w->fd;  } -  void  evtimer_start (struct ev_timer *w)  { @@ -669,6 +679,7 @@ evtimer_start (struct ev_timer *w)  void  evtimer_stop (struct ev_timer *w)  { +  ev_clear ((W)w);    if (!ev_is_active (w))      return; @@ -721,6 +732,7 @@ evperiodic_start (struct ev_periodic *w)  void  evperiodic_stop (struct ev_periodic *w)  { +  ev_clear ((W)w);    if (!ev_is_active (w))      return; @@ -756,6 +768,7 @@ evsignal_start (struct ev_signal *w)  void  evsignal_stop (struct ev_signal *w)  { +  ev_clear ((W)w);    if (!ev_is_active (w))      return; @@ -778,6 +791,10 @@ void evidle_start (struct ev_idle *w)  void evidle_stop (struct ev_idle *w)  { +  ev_clear ((W)w); +  if (ev_is_active (w)) +    return; +    idles [w->active - 1] = idles [--idlecnt];    ev_stop ((W)w);  } @@ -794,12 +811,81 @@ void evcheck_start (struct ev_check *w)  void evcheck_stop (struct ev_check *w)  { +  ev_clear ((W)w); +  if (ev_is_active (w)) +    return; +    checks [w->active - 1] = checks [--checkcnt];    ev_stop ((W)w);  }  /*****************************************************************************/ +struct ev_once +{ +  struct ev_io io; +  struct ev_timer to; +  void (*cb)(int revents, void *arg); +  void *arg; +}; + +static void +once_cb (struct ev_once *once, int revents) +{ +  void (*cb)(int revents, void *arg) = once->cb; +  void *arg = once->arg; + +  evio_stop (&once->io); +  evtimer_stop (&once->to); +  free (once); + +  cb (revents, arg); +} + +static void +once_cb_io (struct ev_io *w, int revents) +{ +  once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, io)), revents); +} + +static void +once_cb_to (struct ev_timer *w, int revents) +{ +  once_cb ((struct ev_once *)(((char *)w) - offsetof (struct ev_once, to)), revents); +} + +void +ev_once (int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg) +{ +  struct ev_once *once = malloc (sizeof (struct ev_once)); + +  if (!once) +    cb (EV_ERROR, arg); +  else +    { +      once->cb  = cb; +      once->arg = arg; + +      evw_init (&once->io, once_cb_io); + +      if (fd >= 0) +        { +          evio_set (&once->io, fd, events); +          evio_start (&once->io); +        } + +      evw_init (&once->to, once_cb_to); + +      if (timeout >= 0.) +        { +          evtimer_set (&once->to, timeout, 0.); +          evtimer_start (&once->to); +        } +    } +} + +/*****************************************************************************/ +  #if 0  struct ev_io wio; @@ -12,11 +12,15 @@ typedef double ev_tstamp;  #define EV_SIGNAL  0x08  #define EV_IDLE    0x10  #define EV_CHECK   0x20 +#define EV_ERROR   (0x3f|0x80)  /* can be used to add custom fields to all watchers */  #ifndef EV_COMMON  # define EV_COMMON void *data  #endif +#ifndef EV_PROTOTYPES +# define EV_PROTOTYPES 1 +#endif  /*   * struct member types: @@ -103,8 +107,9 @@ struct ev_check  #define EVMETHOD_NONE   0  #define EVMETHOD_SELECT 1  #define EVMETHOD_EPOLL  2 -int ev_init (int flags); /* returns ev_method */ +#if EV_PROTOTYPES  extern int ev_method; +int ev_init (int flags); /* returns ev_method */  /* these three calls are suitable for plugging into pthread_atfork */  void ev_prefork (void); @@ -113,22 +118,29 @@ void ev_postfork_child (void);  extern ev_tstamp ev_now; /* time w.r.t. timers and the eventloop, updated after each poll */  ev_tstamp ev_time (void); +#endif  #define EVLOOP_NONBLOCK	1 /* do not block/wait */  #define EVLOOP_ONESHOT	2 /* block *once* only */ +#if EV_PROTOTYPES  void ev_loop (int flags);  extern int ev_loop_done; /* set to 1 to break out of event loop, set to 2 to break out of all event loops */ +/* convinience function, wait for a single event, without registering an event watcher */ +/* if timeout is < 0, do wait indefinitely */ +void ev_once (int fd, int events, ev_tstamp timeout, void (*cb)(int revents, void *arg), void *arg); +#endif +  /* these may evaluate ev multiple times, and the other arguments at most once */  /* either use evw_init + evXXX_set, or the evXXX_init macro, below, to first initialise a watcher */ -#define evw_init(ev,cb_)                   do { (ev)->active = 0; (ev)->cb = (cb_); } while (0) +#define evw_init(ev,cb_)                   do { (ev)->active = 0; (ev)->pending = 0; (ev)->cb = (cb_); } while (0)  #define evio_set(ev,fd_,events_)           do { (ev)->fd = (fd_); (ev)->events = (events_); } while (0)  #define evtimer_set(ev,after_,repeat_)     do { (ev)->at = (after_); (ev)->repeat = (repeat_); } while (0)  #define evperiodic_set(ev,at_,interval_)   do { (ev)->at = (at_); (ev)->interval = (interval_); } while (0)  #define evsignal_set(ev,signum_)           do { (ev)->signum = (signum_); } while (0) -#define evcheck_set(ev)                    /* nop, yes this is a serious in-joke */ -#define evidle_set(ev)                     /* nop, yes this is a serious in-joke */ +#define evcheck_set(ev)                    /* nop, yes, this is a serious in-joke */ +#define evidle_set(ev)                     /* nop, yes, this is a serious in-joke */  #define evio_init(ev,cb,fd,events)         do { evw_init ((ev), (cb)); evio_set ((ev),(fd),(events)); } while (0)  #define evtimer_init(ev,cb,after,repeat)   do { evw_init ((ev), (cb)); evtimer_set ((ev),(after),(repeat)); } while (0) @@ -141,6 +153,7 @@ extern int ev_loop_done; /* set to 1 to break out of event loop, set to 2 to bre  /* stopping (enabling, adding) a watcher does nothing if it is already running */  /* stopping (disabling, deleting) a watcher does nothing unless its already running */ +#if EV_PROTOTYPES  void evio_start       (struct ev_io *w);  void evio_stop        (struct ev_io *w); @@ -159,6 +172,7 @@ void evidle_stop      (struct ev_idle *w);  void evcheck_start    (struct ev_check *w);  void evcheck_stop     (struct ev_check *w); +#endif  #endif | 
