libs/port/poll.c

   1 /*
   2  * poll function
   3  *
   4  * Copyright 2008 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #ifndef HAVE_POLL
  25
  26 #ifdef HAVE_SYS_TIME_H
  27 #include <sys/time.h>
  28 #endif
  29 #ifdef HAVE_UNISTD_H
  30 #include <unistd.h>
  31 #endif
  32
  33 /* we can't include winsock2.h here so we have to duplicate the definitions for Windows */
  34 #if defined(__MINGW32__) || defined(_MSC_VER)
  35
  36 #define FD_SETSIZE 64
  37
  38 struct __ms_timeval
  39 {
  40     long tv_sec;
  41     long tv_usec;
  42 };
  43 #define timeval __ms_timeval
  44
  45 typedef struct
  46 {
  47     unsigned int fd_count;
  48     int fd_array[FD_SETSIZE];   /* an array of SOCKETs */
  49 } fd_set;
  50
  51 #define FD_ZERO(set)      ((set)->fd_count = 0)
  52 #define FD_ISSET(fd, set) __WSAFDIsSet((fd), (set))
  53 #define FD_SET(fd, set)   do { if ((set)->fd_count < FD_SETSIZE) (set)->fd_array[(set)->fd_count++]=(fd); } while(0)
  54
  55 int __stdcall select(int,fd_set*,fd_set*,fd_set*,const struct timeval*);
  56 int __stdcall __WSAFDIsSet(int,fd_set*);
  57
  58 #endif
  59
  60 int poll( struct pollfd *fds, unsigned int count, int timeout )
  61 {
  62     fd_set read_set, write_set, except_set;
  63     unsigned int i;
  64     int maxfd = -1, ret;
  65
  66     FD_ZERO( &read_set );
  67     FD_ZERO( &write_set );
  68     FD_ZERO( &except_set );
  69
  70     for (i = 0; i < count; i++)
  71     {
  72         if (fds[i].fd == -1) continue;
  73         if (fds[i].events & (POLLIN|POLLPRI)) FD_SET( fds[i].fd, &read_set  );
  74         if (fds[i].events & POLLOUT) FD_SET( fds[i].fd, &write_set );
  75         FD_SET( fds[i].fd, &except_set );  /* POLLERR etc. are always selected */
  76         if (fds[i].fd > maxfd) maxfd = fds[i].fd;
  77     }
  78     if (timeout != -1)
  79     {
  80         struct timeval tv;
  81         tv.tv_sec = timeout / 1000;
  82         tv.tv_usec = timeout % 1000;
  83         ret = select( maxfd + 1, &read_set, &write_set, &except_set, &tv );
  84     }
  85     else ret = select( maxfd + 1, &read_set, &write_set, &except_set, NULL );
  86
  87     if (ret >= 0)
  88     {
  89         for (i = 0; i < count; i++)
  90         {
  91             fds[i].revents = 0;
  92             if (fds[i].fd == -1) continue;
  93             if (FD_ISSET( fds[i].fd, &read_set )) fds[i].revents |= POLLIN;
  94             if (FD_ISSET( fds[i].fd, &write_set )) fds[i].revents |= POLLOUT;
  95             if (FD_ISSET( fds[i].fd, &except_set )) fds[i].revents |= POLLERR;
  96         }
  97     }
  98     return ret;
  99 }
 100
 101 #endif /* HAVE_POLL */