lib/silcutil/win32/silcwin32schedule.c

   1 /*
   2
   3   silcwin32schedule.c
   4
   5   Author: Pekka Riikonen <priikone@silcnet.org>
   6
   7   Copyright (C) 2001 Pekka Riikonen
   8
   9   This program is free software; you can redistribute it and/or modify
  10   it under the terms of the GNU General Public License as published by
  11   the Free Software Foundation; either version 2 of the License, or
  12   (at your option) any later version.
  13
  14   This program is distributed in the hope that it will be useful,
  15   but WITHOUT ANY WARRANTY; without even the implied warranty of
  16   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17   GNU General Public License for more details.
  18
  19 */
  20 /* $Id$ */
  21
  22 #include "silcincludes.h"
  23
  24 /* Our "select()" for WIN32. This mimics the behaviour of select() system
  25    call. It does not call the Winsock's select() though. Its functions
  26    are derived from GLib's g_poll() and from some old Xemacs's sys_select().
  27
  28    This makes following assumptions, which I don't know whether they
  29    are correct or not:
  30
  31    o writefds are ignored, if set this will return immediately.
  32    o exceptfds are ignored totally
  33    o If all arguments except timeout are NULL then this will register
  34      a timeout with SetTimer and will wait just for Windows messages
  35      with WaitMessage.
  36    o MsgWaitForMultipleObjects is used to wait all kind of events, this
  37      includes SOCKETs and Windows messages.
  38    o All Windows messages are dispatched from this function.
  39    o The Operating System has Winsock 2.
  40
  41    References:
  42
  43    o http://msdn.microsoft.com/library/default.asp?
  44      url=/library/en-us/winui/hh/winui/messques_77zk.asp
  45    o http://msdn.microsoft.com/library/default.asp?
  46      url=/library/en-us/winsock/hh/winsock/apistart_9g1e.asp
  47    o http://developer.novell.com/support/winsock/doc/toc.htm
  48
  49 */
  50
  51 int silc_select(int n, fd_set *readfds, fd_set *writefds,
  52                 fd_set *exceptfds, struct timeval *timeout)
  53 {
  54   HANDLE handles[MAXIMUM_WAIT_OBJECTS];
  55   DWORD ready, curtime, timeo;
  56   int nhandles = 0, i;
  57   MSG msg;
  58
  59   /* Check fd sets (ignoring the exceptfds) */
  60   if (readfds) {
  61     for (i = 0; i < n - 1; i++)
  62       if (FD_ISSET(i, readfds))
  63         handles[nhandles++] = (HANDLE)i;
  64
  65     FD_ZERO(readfds);
  66   }
  67
  68   /* If writefds is set then return immediately */
  69   if (writefds) {
  70     for (i = 0; i < n - 1; i++)
  71       if (FD_ISSET(i, writefds))
  72         return 1;
  73   }
  74
  75   timeo = (timeout ? (timeout.tv_sec * 1000) + (timeout.tv_usec / 1000) :
  76            INFINITE);
  77
  78   /* If we have nothing to wait and timeout is set then register a timeout
  79      and wait just for windows messages. */
  80   if (nhandles == 0 && timeout) {
  81     UINT timer = SetTimer(NULL, 0, timeo, NULL);
  82     if (timer) {
  83       WaitMessage();
  84       KillTimer(NULL, timer);
  85
  86       while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
  87         TranslateMessage(&msg);
  88         DispatchMessage(&msg);
  89       }
  90
  91       return 0;
  92     }
  93   }
  94
  95  retry:
  96   curtime = GetTickCount();
  97   ready = MsgWaitForMultipleObjects(nhandles, handles, FALSE, timeo,
  98                                     QS_ALLINPUT);
  99
 100   if (ready == WAIT_FAILED) {
 101     /* Wait failed with error */
 102     SILC_LOG_WARNING(("WaitForMultipleObjects() failed"));
 103     return -1;
 104
 105   } else if (ready >= WAIT_ABANDONED_0 &&
 106              ready < WAIT_ABANDONED_0 + nhandles) {
 107     /* Signal abandoned */
 108     SILC_LOG_WARNING(("WaitForMultipleObjects() failed (ABANDONED)"));
 109     return -1;
 110   } else if (ready == WAIT_TIMEOUT) {
 111     /* Timeout */
 112     return 0;
 113   } else if (ready == WAIT_OBJECT_0 + nhandles) {
 114     /* Windows messages. The MSDN online says that if the application
 115        creates a window then its main loop (and we're assuming that
 116        it is our SILC Scheduler) must handle the Windows messages, so do
 117        it here as the MSDN suggests. */
 118     while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
 119       TranslateMessage(&msg);
 120       DispatchMessage(&msg);
 121     }
 122
 123     /* If timeout is set then we must update the timeout since we won't
 124        return and we will give the wait another try. */
 125     if (timeo != INFINITE) {
 126       timeo -= GetTickCount() - curtime;
 127       if (timeo < 0)
 128         timeo = 0;
 129     }
 130
 131     /* Give the wait another try */
 132    goto retry;
 133   } else if (ready >= WAIT_OBJECT_0 && ready < WAIT_OBJECT_0 + nhandles &&
 134              readfds) {
 135     /* Some other event, like SOCKET or something. */
 136
 137     /* Go through all fds even though only one was set. This is to avoid
 138        starvation of high numbered fds. */
 139     ready -= WAIT_OBJECT_0;
 140     i = 0;
 141     do {
 142       /* Set the handle to fd set */
 143       FD_SET(handle[ready], readfds);
 144       i++;
 145
 146       /* Check the status of the next handle and set it's fd to the fd
 147          set if data is available. */
 148       while (++ready < n)
 149         if (WaitForSingleObject(handle[ready], 0) == WAIT_OBJECT_0)
 150           break;
 151     } while (ready < n);
 152
 153     return i;
 154   }
 155
 156   return -1;
 157 }