PageRenderTime 72ms CodeModel.GetById 18ms app.highlight 48ms RepoModel.GetById 1ms app.codeStats 0ms

/src/unix/ev/event.c

http://github.com/joyent/libuv
C | 402 lines | 278 code | 80 blank | 44 comment | 32 complexity | cb275ba0534c032bb89b4500bc45c648 MD5 | raw file
  1/*
  2 * libevent compatibility layer
  3 *
  4 * Copyright (c) 2007,2008,2009,2010 Marc Alexander Lehmann <libev@schmorp.de>
  5 * All rights reserved.
  6 *
  7 * Redistribution and use in source and binary forms, with or without modifica-
  8 * tion, are permitted provided that the following conditions are met:
  9 *
 10 *   1.  Redistributions of source code must retain the above copyright notice,
 11 *       this list of conditions and the following disclaimer.
 12 *
 13 *   2.  Redistributions in binary form must reproduce the above copyright
 14 *       notice, this list of conditions and the following disclaimer in the
 15 *       documentation and/or other materials provided with the distribution.
 16 *
 17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 18 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER-
 19 * CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO
 20 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPE-
 21 * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 22 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 23 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 24 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTH-
 25 * ERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 26 * OF THE POSSIBILITY OF SUCH DAMAGE.
 27 *
 28 * Alternatively, the contents of this file may be used under the terms of
 29 * the GNU General Public License ("GPL") version 2 or any later version,
 30 * in which case the provisions of the GPL are applicable instead of
 31 * the above. If you wish to allow the use of your version of this file
 32 * only under the terms of the GPL and not to allow others to use your
 33 * version of this file under the BSD license, indicate your decision
 34 * by deleting the provisions above and replace them with the notice
 35 * and other provisions required by the GPL. If you do not delete the
 36 * provisions above, a recipient may use your version of this file under
 37 * either the BSD or the GPL.
 38 */
 39
 40#include <stddef.h>
 41#include <stdlib.h>
 42#include <assert.h>
 43
 44#ifdef EV_EVENT_H
 45# include EV_EVENT_H
 46#else
 47# include "event.h"
 48#endif
 49
 50#if EV_MULTIPLICITY
 51# define dLOOPev struct ev_loop *loop = (struct ev_loop *)ev->ev_base
 52# define dLOOPbase struct ev_loop *loop = (struct ev_loop *)base
 53#else
 54# define dLOOPev
 55# define dLOOPbase
 56#endif
 57
 58/* never accessed, will always be cast from/to ev_loop */
 59struct event_base
 60{
 61  int dummy;
 62};
 63
 64static struct event_base *ev_x_cur;
 65
 66static ev_tstamp
 67ev_tv_get (struct timeval *tv)
 68{
 69  if (tv)
 70    {
 71      ev_tstamp after = tv->tv_sec + tv->tv_usec * 1e-6;
 72      return after ? after : 1e-6;
 73    }
 74  else
 75    return -1.;
 76}
 77
 78#define EVENT_STRINGIFY(s) # s
 79#define EVENT_VERSION(a,b) EVENT_STRINGIFY (a) "." EVENT_STRINGIFY (b)
 80
 81const char *event_get_version (void)
 82{
 83  /* returns ABI, not API or library, version */
 84  return EVENT_VERSION (EV_VERSION_MAJOR, EV_VERSION_MINOR);
 85}
 86
 87const char *event_get_method (void)
 88{
 89  return "libev";
 90}
 91
 92void *event_init (void)
 93{
 94#if EV_MULTIPLICITY
 95  if (ev_x_cur)
 96    ev_x_cur = (struct event_base *)ev_loop_new (EVFLAG_AUTO);
 97  else
 98    ev_x_cur = (struct event_base *)ev_default_loop (EVFLAG_AUTO);
 99#else
100  assert (("libev: multiple event bases not supported when not compiled with EV_MULTIPLICITY", !ev_x_cur));
101
102  ev_x_cur = (struct event_base *)(long)ev_default_loop (EVFLAG_AUTO);
103#endif
104
105  return ev_x_cur;
106}
107
108void event_base_free (struct event_base *base)
109{
110  dLOOPbase;
111
112#if EV_MULTIPLICITY
113  if (!ev_is_default_loop (loop))
114    ev_loop_destroy (loop);
115#endif
116}
117
118int event_dispatch (void)
119{
120  return event_base_dispatch (ev_x_cur);
121}
122
123#ifdef EV_STANDALONE
124void event_set_log_callback (event_log_cb cb)
125{
126  /* nop */
127}
128#endif
129
130int event_loop (int flags)
131{
132  return event_base_loop (ev_x_cur, flags);
133}
134
135int event_loopexit (struct timeval *tv)
136{
137  return event_base_loopexit (ev_x_cur, tv);
138}
139
140static void
141ev_x_cb (struct event *ev, int revents)
142{
143  revents &= EV_READ | EV_WRITE | EV_TIMER | EV_SIGNAL;
144
145  ev->ev_res = revents;
146  ev->ev_callback (ev->ev_fd, (short)revents, ev->ev_arg);
147}
148
149static void
150ev_x_cb_sig (EV_P_ struct ev_signal *w, int revents)
151{
152  struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, iosig.sig));
153
154  if (revents & EV_ERROR)
155    event_del (ev);
156
157  ev_x_cb (ev, revents);
158}
159
160static void
161ev_x_cb_io (EV_P_ struct ev_io *w, int revents)
162{
163  struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, iosig.io));
164
165  if ((revents & EV_ERROR) || !(ev->ev_events & EV_PERSIST))
166    event_del (ev);
167
168  ev_x_cb (ev, revents);
169}
170
171static void
172ev_x_cb_to (EV_P_ struct ev_timer *w, int revents)
173{
174  struct event *ev = (struct event *)(((char *)w) - offsetof (struct event, to));
175
176  event_del (ev);
177
178  ev_x_cb (ev, revents);
179}
180
181void event_set (struct event *ev, int fd, short events, void (*cb)(int, short, void *), void *arg)
182{
183  if (events & EV_SIGNAL)
184    ev_init (&ev->iosig.sig, ev_x_cb_sig);
185  else
186    ev_init (&ev->iosig.io, ev_x_cb_io);
187
188  ev_init (&ev->to, ev_x_cb_to);
189
190  ev->ev_base     = ev_x_cur; /* not threadsafe, but it's how libevent works */
191  ev->ev_fd       = fd;
192  ev->ev_events   = events;
193  ev->ev_pri      = 0;
194  ev->ev_callback = cb;
195  ev->ev_arg      = arg;
196  ev->ev_res      = 0;
197  ev->ev_flags    = EVLIST_INIT;
198}
199
200int event_once (int fd, short events, void (*cb)(int, short, void *), void *arg, struct timeval *tv)
201{
202  return event_base_once (ev_x_cur, fd, events, cb, arg, tv);
203}
204
205int event_add (struct event *ev, struct timeval *tv)
206{
207  dLOOPev;
208
209  if (ev->ev_events & EV_SIGNAL)
210    {
211      if (!ev_is_active (&ev->iosig.sig))
212        {
213          ev_signal_set (&ev->iosig.sig, ev->ev_fd);
214          ev_signal_start (EV_A_ &ev->iosig.sig);
215
216          ev->ev_flags |= EVLIST_SIGNAL;
217        }
218    }
219  else if (ev->ev_events & (EV_READ | EV_WRITE))
220    {
221      if (!ev_is_active (&ev->iosig.io))
222        {
223          ev_io_set (&ev->iosig.io, ev->ev_fd, ev->ev_events & (EV_READ | EV_WRITE));
224          ev_io_start (EV_A_ &ev->iosig.io);
225
226          ev->ev_flags |= EVLIST_INSERTED;
227        }
228    }
229
230  if (tv)
231    {
232      ev->to.repeat = ev_tv_get (tv);
233      ev_timer_again (EV_A_ &ev->to);
234      ev->ev_flags |= EVLIST_TIMEOUT;
235    }
236  else
237    {
238      ev_timer_stop (EV_A_ &ev->to);
239      ev->ev_flags &= ~EVLIST_TIMEOUT;
240    }
241
242  ev->ev_flags |= EVLIST_ACTIVE;
243
244  return 0;
245}
246
247int event_del (struct event *ev)
248{
249  dLOOPev;
250
251  if (ev->ev_events & EV_SIGNAL)
252    ev_signal_stop (EV_A_ &ev->iosig.sig);
253  else if (ev->ev_events & (EV_READ | EV_WRITE))
254    ev_io_stop (EV_A_ &ev->iosig.io);
255
256  if (ev_is_active (&ev->to))
257    ev_timer_stop (EV_A_ &ev->to);
258
259  ev->ev_flags = EVLIST_INIT;
260
261  return 0;
262}
263
264void event_active (struct event *ev, int res, short ncalls)
265{
266  dLOOPev;
267
268  if (res & EV_TIMEOUT)
269    ev_feed_event (EV_A_ &ev->to, res & EV_TIMEOUT);
270
271  if (res & EV_SIGNAL)
272    ev_feed_event (EV_A_ &ev->iosig.sig, res & EV_SIGNAL);
273
274  if (res & (EV_READ | EV_WRITE))
275    ev_feed_event (EV_A_ &ev->iosig.io, res & (EV_READ | EV_WRITE));
276}
277
278int event_pending (struct event *ev, short events, struct timeval *tv)
279{
280  short revents = 0;
281  dLOOPev;
282
283  if (ev->ev_events & EV_SIGNAL)
284    {
285      /* sig */
286      if (ev_is_active (&ev->iosig.sig) || ev_is_pending (&ev->iosig.sig))
287        revents |= EV_SIGNAL;
288    }
289  else if (ev->ev_events & (EV_READ | EV_WRITE))
290    {
291      /* io */
292      if (ev_is_active (&ev->iosig.io) || ev_is_pending (&ev->iosig.io))
293        revents |= ev->ev_events & (EV_READ | EV_WRITE);
294    }
295
296  if (ev->ev_events & EV_TIMEOUT || ev_is_active (&ev->to) || ev_is_pending (&ev->to))
297    {
298      revents |= EV_TIMEOUT;
299
300      if (tv)
301        {
302          ev_tstamp at = ev_now (EV_A);
303
304          tv->tv_sec  = (long)at;
305          tv->tv_usec = (long)((at - (ev_tstamp)tv->tv_sec) * 1e6);
306        }
307    }
308
309  return events & revents;
310}
311
312int event_priority_init (int npri)
313{
314  return event_base_priority_init (ev_x_cur, npri);
315}
316
317int event_priority_set (struct event *ev, int pri)
318{
319  ev->ev_pri = pri;
320
321  return 0;
322}
323
324int event_base_set (struct event_base *base, struct event *ev)
325{
326  ev->ev_base = base;
327
328  return 0;
329}
330
331int event_base_loop (struct event_base *base, int flags)
332{
333  dLOOPbase;
334
335  ev_run (EV_A_ flags);
336
337  return 0;
338}
339
340int event_base_dispatch (struct event_base *base)
341{
342  return event_base_loop (base, 0);
343}
344
345static void
346ev_x_loopexit_cb (int revents, void *base)
347{
348  dLOOPbase;
349
350  ev_break (EV_A_ EVBREAK_ONE);
351}
352
353int event_base_loopexit (struct event_base *base, struct timeval *tv)
354{
355  ev_tstamp after = ev_tv_get (tv);
356  dLOOPbase;
357
358  ev_once (EV_A_ -1, 0, after >= 0. ? after : 0., ev_x_loopexit_cb, (void *)base);
359
360  return 0;
361}
362
363struct ev_x_once
364{
365  int fd;
366  void (*cb)(int, short, void *);
367  void *arg;
368};
369
370static void
371ev_x_once_cb (int revents, void *arg)
372{
373  struct ev_x_once *once = (struct ev_x_once *)arg;
374
375  once->cb (once->fd, (short)revents, once->arg);
376  free (once);
377}
378
379int event_base_once (struct event_base *base, int fd, short events, void (*cb)(int, short, void *), void *arg, struct timeval *tv)
380{
381  struct ev_x_once *once = (struct ev_x_once *)malloc (sizeof (struct ev_x_once));
382  dLOOPbase;
383
384  if (!once)
385    return -1;
386
387  once->fd  = fd;
388  once->cb  = cb;
389  once->arg = arg;
390
391  ev_once (EV_A_ fd, events & (EV_READ | EV_WRITE), ev_tv_get (tv), ev_x_once_cb, (void *)once);
392
393  return 0;
394}
395
396int event_base_priority_init (struct event_base *base, int npri)
397{
398  /*dLOOPbase;*/
399
400  return 0;
401}
402