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 gint
  19 KQUEUE_INIT_FD (gint fd, gint events, gint flags)
  20 {
  21         struct kevent event;
  22         EV_SET (&event, fd, events, flags, 0, 0, 0);
  23         return kevent (kqueue_fd, &event, 1, NULL, 0, NULL);
  24 }
  25
  26 static gboolean
  27 kqueue_init (gint wakeup_pipe_fd)
  28 {
  29         kqueue_fd = kqueue ();
  30         if (kqueue_fd == -1) {
  31                 g_error ("kqueue_init: kqueue () failed, error (%d) %s", errno, g_strerror (errno));
  32                 return FALSE;
  33         }
  34
  35         if (KQUEUE_INIT_FD (wakeup_pipe_fd, EVFILT_READ, EV_ADD | EV_ENABLE) == -1) {
  36                 g_error ("kqueue_init: kevent () failed, error (%d) %s", errno, g_strerror (errno));
  37                 close (kqueue_fd);
  38                 return FALSE;
  39         }
  40
  41         kqueue_events = g_new0 (struct kevent, KQUEUE_NEVENTS);
  42
  43         return TRUE;
  44 }
  45
  46 static void
  47 kqueue_register_fd (gint fd, gint events, gboolean is_new)
  48 {
  49         if (events & EVENT_IN) {
  50                 if (KQUEUE_INIT_FD (fd, EVFILT_READ, EV_ADD | EV_ENABLE) == -1)
  51                         g_error ("kqueue_register_fd: kevent(read,enable) failed, error (%d) %s", errno, g_strerror (errno));
  52         } else {
  53                 if (KQUEUE_INIT_FD (fd, EVFILT_READ, EV_ADD | EV_DISABLE) == -1)
  54                         g_error ("kqueue_register_fd: kevent(read,disable) failed, error (%d) %s", errno, g_strerror (errno));
  55         }
  56         if (events & EVENT_OUT) {
  57                 if (KQUEUE_INIT_FD (fd, EVFILT_WRITE, EV_ADD | EV_ENABLE) == -1)
  58                         g_error ("kqueue_register_fd: kevent(write,enable) failed, error (%d) %s", errno, g_strerror (errno));
  59         } else {
  60                 if (KQUEUE_INIT_FD (fd, EVFILT_WRITE, EV_ADD | EV_DISABLE) == -1)
  61                         g_error ("kqueue_register_fd: kevent(write,disable) failed, error (%d) %s", errno, g_strerror (errno));
  62         }
  63 }
  64
  65 static void
  66 kqueue_remove_fd (gint fd)
  67 {
  68         /* FIXME: a race between closing and adding operation in the Socket managed code trigger a ENOENT error */
  69         if (KQUEUE_INIT_FD (fd, EVFILT_READ, EV_DELETE) == -1)
  70                 g_error ("kqueue_register_fd: kevent(read,delete) failed, error (%d) %s", errno, g_strerror (errno));
  71         if (KQUEUE_INIT_FD (fd, EVFILT_WRITE, EV_DELETE) == -1)
  72                 g_error ("kqueue_register_fd: kevent(write,delete) failed, error (%d) %s", errno, g_strerror (errno));
  73 }
  74
  75 static gint
  76 kqueue_event_wait (void (*callback) (gint fd, gint events, gpointer user_data), gpointer user_data)
  77 {
  78         gint i, ready;
  79
  80         memset (kqueue_events, 0, sizeof (struct kevent) * KQUEUE_NEVENTS);
  81
  82         mono_gc_set_skip_thread (TRUE);
  83
  84         MONO_ENTER_GC_SAFE;
  85         ready = kevent (kqueue_fd, NULL, 0, kqueue_events, KQUEUE_NEVENTS, NULL);
  86         MONO_EXIT_GC_SAFE;
  87
  88         mono_gc_set_skip_thread (FALSE);
  89
  90         if (ready == -1) {
  91                 switch (errno) {
  92                 case EINTR:
  93                         mono_thread_internal_check_for_interruption_critical (mono_thread_internal_current ());
  94                         ready = 0;
  95                         break;
  96                 default:
  97                         g_error ("kqueue_event_wait: kevent () failed, error (%d) %s", errno, g_strerror (errno));
  98                         break;
  99                 }
 100         }
 101
 102         if (ready == -1)
 103                 return -1;
 104
 105         for (i = 0; i < ready; ++i) {
 106                 gint fd, events = 0;
 107
 108                 fd = kqueue_events [i].ident;
 109                 if (kqueue_events [i].filter == EVFILT_READ || (kqueue_events [i].flags & EV_ERROR) != 0)
 110                         events |= EVENT_IN;
 111                 if (kqueue_events [i].filter == EVFILT_WRITE || (kqueue_events [i].flags & EV_ERROR) != 0)
 112                         events |= EVENT_OUT;
 113
 114                 callback (fd, events, user_data);
 115         }
 116
 117         return 0;
 118 }
 119
 120 static ThreadPoolIOBackend backend_kqueue = {
 121         .init = kqueue_init,
 122         .register_fd = kqueue_register_fd,
 123         .remove_fd = kqueue_remove_fd,
 124         .event_wait = kqueue_event_wait,
 125 };
 126
 127 #endif