mono/metadata/threadpool-ms-io-kqueue.c

   1
   2 #if defined(HAVE_KQUEUE)
   3
   4 #include <sys/types.h>
   5 #include <sys/event.h>
   6 #include <sys/time.h>
   7
   8 #if defined(HOST_WIN32)
   9 /* We assume that kqueue is not available on windows */
  10 #error
  11 #endif
  12
  13 #define KQUEUE_NEVENTS 128
  14
  15 static gint kqueue_fd;
  16 static struct kevent *kqueue_events;
  17
  18 static gboolean
  19 kqueue_init (gint wakeup_pipe_fd)
  20 {
  21         struct kevent event;
  22
  23         kqueue_fd = kqueue ();
  24         if (kqueue_fd == -1) {
  25                 g_warning ("kqueue_init: kqueue () failed, error (%d) %s", errno, g_strerror (errno));
  26                 return FALSE;
  27         }
  28
  29         EV_SET (&event, wakeup_pipe_fd, EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
  30         if (kevent (kqueue_fd, &event, 1, NULL, 0, NULL) == -1) {
  31                 g_warning ("kqueue_init: kevent () failed, error (%d) %s", errno, g_strerror (errno));
  32                 close (kqueue_fd);
  33                 return FALSE;
  34         }
  35
  36         kqueue_events = g_new0 (struct kevent, KQUEUE_NEVENTS);
  37
  38         return TRUE;
  39 }
  40
  41 static void
  42 kqueue_cleanup (void)
  43 {
  44         g_free (kqueue_events);
  45         close (kqueue_fd);
  46 }
  47
  48 static void
  49 kqueue_update_add (gint fd, gint events, gboolean is_new)
  50 {
  51         struct kevent event;
  52
  53         if ((events & MONO_POLLIN) != 0)
  54                 EV_SET (&event, fd, EVFILT_READ, EV_ADD | EV_ENABLE | EV_ONESHOT, 0, 0, 0);
  55         if ((events & MONO_POLLOUT) != 0)
  56                 EV_SET (&event, fd, EVFILT_WRITE, EV_ADD | EV_ENABLE | EV_ONESHOT, 0, 0, 0);
  57
  58         if (kevent (kqueue_fd, &event, 1, NULL, 0, NULL) == -1)
  59                 g_warning ("kqueue_update_add: kevent(update) failed, error (%d) %s", errno, g_strerror (errno));
  60 }
  61
  62 static gint
  63 kqueue_event_wait (void)
  64 {
  65         gint ready;
  66
  67         ready = kevent (kqueue_fd, NULL, 0, kqueue_events, KQUEUE_NEVENTS, NULL);
  68         if (ready == -1) {
  69                 switch (errno) {
  70                 case EINTR:
  71                         mono_thread_internal_check_for_interruption_critical (mono_thread_internal_current ());
  72                         ready = 0;
  73                         break;
  74                 default:
  75                         g_warning ("kqueue_event_wait: kevent () failed, error (%d) %s", errno, g_strerror (errno));
  76                         break;
  77                 }
  78         }
  79
  80         return ready;
  81 }
  82
  83 static gint
  84 kqueue_event_get_fd_at (gint i, gint *events)
  85 {
  86         g_assert (events);
  87
  88         *events = ((kqueue_events [i].filter == EVFILT_READ || (kqueue_events [i].flags & EV_ERROR) != 0) ? MONO_POLLIN : 0)
  89                     | ((kqueue_events [i].filter == EVFILT_WRITE || (kqueue_events [i].flags & EV_ERROR) != 0) ? MONO_POLLOUT : 0);
  90
  91         return kqueue_events [i].ident;
  92 }
  93
  94 static gint
  95 kqueue_event_get_fd_max (void)
  96 {
  97         return KQUEUE_NEVENTS;
  98 }
  99
 100 static void
 101 kqueue_event_reset_fd_at (gint i, gint events)
 102 {
 103         if (kqueue_events [i].filter == EVFILT_READ && (events & MONO_POLLIN) != 0) {
 104                 EV_SET (&kqueue_events [i], kqueue_events [i].ident, EVFILT_READ, EV_ADD | EV_ENABLE | EV_ONESHOT, 0, 0, 0);
 105                 if (kevent (kqueue_fd, &kqueue_events [i], 1, NULL, 0, NULL) == -1) {
 106                         g_warning ("kqueue_event_reset_fd_at: kevent (read) failed, error (%d) %s", errno, g_strerror (errno));
 107                 }
 108         }
 109         if (kqueue_events [i].filter == EVFILT_WRITE && (events & MONO_POLLOUT) != 0) {
 110                 EV_SET (&kqueue_events [i], kqueue_events [i].ident, EVFILT_WRITE, EV_ADD | EV_ENABLE | EV_ONESHOT, 0, 0, 0);
 111                 if (kevent (kqueue_fd, &kqueue_events [i], 1, NULL, 0, NULL) == -1) {
 112                         g_warning ("kqueue_event_reset_fd_at: kevent (write) failed, error (%d) %s", errno, g_strerror (errno));
 113                 }
 114         }
 115 }
 116
 117 static ThreadPoolIOBackend backend_kqueue = {
 118         .init = kqueue_init,
 119         .cleanup = kqueue_cleanup,
 120         .update_add = kqueue_update_add,
 121         .event_wait = kqueue_event_wait,
 122         .event_get_fd_max = kqueue_event_get_fd_max,
 123         .event_get_fd_at = kqueue_event_get_fd_at,
 124         .event_reset_fd_at = kqueue_event_reset_fd_at,
 125 };
 126
 127 #endif