tcrt/src/select.cpp

   1 /*=============================================================================
   2         select.cpp :
   3
   4         Copyright © 2008 Bruno Santos <nayart3@gmail.com>
   5 =============================================================================*/
   6
   7 #include <sys/select.h>
   8 #include <string.h>
   9 #include <errno.h>
  10 #include "Ws2_32.h"
  11
  12 ///////////////////////////////////////////////////////////////////////////////
  13 void FD_ZERO(fd_set* fs)
  14 {
  15         memset(fs, 0, sizeof(fd_set));
  16 }
  17
  18 void FD_SET(int fildes, fd_set* fs)
  19 {
  20         HANDLE hd = tcrt::fildes_to_handle(fildes);
  21
  22         if (fs->count >= FD_SETSIZE)
  23                 return;
  24         for (uint i = 0; i < fs->count; ++i) {
  25                 if (fs->fildes[i] == hd)
  26                         return;
  27         }
  28         fs->fildes[fs->count++] = hd;
  29 }
  30
  31 int FD_ISSET(int fildes, fd_set* fs)
  32 {
  33         return Ws2_FDIsSet(tcrt::fildes_to_handle(fildes), fs);
  34 }
  35
  36 int select(int nfds, fd_set* readfds, fd_set* writefds, fd_set* errorfds, struct timeval* timeout)
  37 {
  38         int ret;
  39
  40         if (!nfds) {
  41                 DWORD msec = timeout->tv_sec * 1000 + timeout->tv_usec / 1000;
  42                 Sleep(msec);
  43         }
  44         ret = Ws2_Select(nfds, readfds, writefds, errorfds, timeout);
  45         if (ret < 0) {
  46                 errno = Ws2_GetLastError();
  47                 return -1;
  48         }
  49         return ret;
  50 }
  51
  52 // EOF ////////////////////////////////////////////////////////////////////////