mono/utils/mono-poll.c

   1 /**
   2  * \file
   3  */
   4
   5 #include <config.h>
   6
   7 #ifdef HOST_WIN32
   8 /* For select */
   9 #include <winsock2.h>
  10 #endif
  11
  12 #include "mono-poll.h"
  13 #include <errno.h>
  14 #include <mono/utils/mono-errno.h>
  15
  16 #ifdef DISABLE_SOCKETS
  17 #include <glib.h>
  18
  19 int
  20 mono_poll_can_add (mono_pollfd *ufds, unsigned int nfds, int fd)
  21 {
  22         return 1;
  23 }
  24
  25 int
  26 mono_poll (mono_pollfd *ufds, unsigned int nfds, int timeout)
  27 {
  28         g_assert_not_reached ();
  29         return -1;
  30 }
  31
  32 #else
  33
  34 #if defined(HAVE_POLL) && !defined(__APPLE__)
  35
  36 int
  37 mono_poll_can_add (mono_pollfd *ufds, unsigned int nfds, int fd)
  38 {
  39         return 1;
  40 }
  41
  42 int
  43 mono_poll (mono_pollfd *ufds, unsigned int nfds, int timeout)
  44 {
  45         return poll (ufds, nfds, timeout);
  46 }
  47 #else
  48
  49 int
  50 mono_poll_can_add (mono_pollfd *ufds, unsigned int nfds, int fd)
  51 {
  52         if (fd < 0)
  53                 return 1;
  54 #ifdef HOST_WIN32
  55         return (nfds < FD_SETSIZE);
  56 #else
  57         return (fd < FD_SETSIZE);
  58 #endif
  59 }
  60
  61 int
  62 mono_poll (mono_pollfd *ufds, unsigned int nfds, int timeout)
  63 {
  64         struct timeval tv, *tvptr;
  65         int i, fd, events, affected, count;
  66         fd_set rfds, wfds, efds;
  67         int nexc = 0;
  68         int maxfd = 0;
  69
  70         if (timeout < 0) {
  71                 tvptr = NULL;
  72         } else {
  73                 tv.tv_sec = timeout / 1000;
  74                 tv.tv_usec = (timeout % 1000) * 1000;
  75                 tvptr = &tv;
  76         }
  77
  78         FD_ZERO (&rfds);
  79         FD_ZERO (&wfds);
  80         FD_ZERO (&efds);
  81
  82         for (i = 0; i < nfds; i++) {
  83                 ufds [i].revents = 0;
  84                 fd = ufds [i].fd;
  85                 if (fd < 0)
  86                         continue;
  87
  88 #ifdef HOST_WIN32
  89                 if (nexc >= FD_SETSIZE) {
  90                         ufds [i].revents = MONO_POLLNVAL;
  91                         return 1;
  92                 }
  93 #else
  94                 if (fd >= FD_SETSIZE) {
  95                         ufds [i].revents = MONO_POLLNVAL;
  96                         return 1;
  97                 }
  98 #endif
  99
 100                 events = ufds [i].events;
 101                 if ((events & MONO_POLLIN) != 0)
 102                         FD_SET (fd, &rfds);
 103
 104                 if ((events & MONO_POLLOUT) != 0)
 105                         FD_SET (fd, &wfds);
 106
 107                 FD_SET (fd, &efds);
 108                 nexc++;
 109                 if (fd > maxfd)
 110                         maxfd = fd;
 111
 112         }
 113
 114         affected = select (maxfd + 1, &rfds, &wfds, &efds, tvptr);
 115         if (affected == -1) {
 116 #ifdef HOST_WIN32
 117                 int error = WSAGetLastError ();
 118                 switch (error) {
 119                 case WSAEFAULT: mono_set_errno (EFAULT); break;
 120                 case WSAEINVAL: mono_set_errno (EINVAL); break;
 121                 case WSAEINTR: mono_set_errno (EINTR); break;
 122                 /* case WSAEINPROGRESS: mono_set_errno (EINPROGRESS); break; */
 123                 case WSAEINPROGRESS: mono_set_errno (EINTR); break;
 124                 case WSAENOTSOCK: mono_set_errno (EBADF); break;
 125 #ifdef ENOSR
 126                 case WSAENETDOWN: mono_set_errno (ENOSR); break;
 127 #endif
 128                 default: mono_set_errno (0);
 129                 }
 130 #endif
 131
 132                 return -1;
 133         }
 134
 135         count = 0;
 136         for (i = 0; i < nfds && affected > 0; i++) {
 137                 fd = ufds [i].fd;
 138                 if (fd < 0)
 139                         continue;
 140
 141                 events = ufds [i].events;
 142                 if ((events & MONO_POLLIN) != 0 && FD_ISSET (fd, &rfds)) {
 143                         ufds [i].revents |= MONO_POLLIN;
 144                         affected--;
 145                 }
 146
 147                 if ((events & MONO_POLLOUT) != 0 && FD_ISSET (fd, &wfds)) {
 148                         ufds [i].revents |= MONO_POLLOUT;
 149                         affected--;
 150                 }
 151
 152                 if (FD_ISSET (fd, &efds)) {
 153                         ufds [i].revents |= MONO_POLLERR;
 154                         affected--;
 155                 }
 156
 157                 if (ufds [i].revents != 0)
 158                         count++;
 159         }
 160
 161         return count;
 162 }
 163
 164 #endif
 165
 166 #endif /* #ifndef DISABLE_SOCKETS */