Tools/WinSN5.0Ver.iso/NETSCAP.50/WIN1998.ZIP/ns/nsprpub/pr/src/md/os2/os2poll.c

C/C++ Source or Header | 1998-04-08 | 4.7 KB | 166 lines

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
 * The contents of this file are subject to the Netscape Public License
 * Version 1.0 (the "NPL"); you may not use this file except in
 * compliance with the NPL.  You may obtain a copy of the NPL at
 * http://www.mozilla.org/NPL/
 *
 * Software distributed under the NPL is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
 * for the specific language governing rights and limitations under the
 * NPL.
 *
 * The Initial Developer of this code under the NPL is Netscape
 * Communications Corporation.  Portions created by Netscape are
 * Copyright (C) 1998 Netscape Communications Corporation.  All Rights
 * Reserved.
 */

/*
 * This file implements _PR_MD_PR_POLL for OS/2.
 */

#include "primpl.h"

PRInt32
_PR_MD_PR_POLL(PRPollDesc *pds, PRIntn npds,
                        PRIntervalTime timeout)
{
    PRPollDesc *pd, *epd;
    PRInt32 n, err, pdcnt;
    PRThread *me = _PR_MD_CURRENT_THREAD();

       fd_set rd, wt, ex;
       struct timeval tv, *tvp = NULL;
       int maxfd = -1;

    /*
        * For restarting _MD_SELECT() if it is interrupted by a signal.
        * We use these variables to figure out how much time has elapsed
        * and how much of the timeout still remains.
        */
       PRIntervalTime start, elapsed, remaining;

       FD_ZERO(&rd);
       FD_ZERO(&wt);
       FD_ZERO(&ex);

    for (pd = pds, epd = pd + npds; pd < epd; pd++) {
         PRInt32 osfd;
         PRInt16 in_flags = pd->in_flags;
         PRFileDesc *bottom = pd->fd;

         if ((NULL == bottom) || (in_flags == 0)) {
             continue;
         }
         while (bottom->lower != NULL) {
             bottom = bottom->lower;
         }
         osfd = bottom->secret->md.osfd;

       if (osfd > maxfd) {
           maxfd = osfd;
       }
       if (in_flags & PR_POLL_READ)  {
           FD_SET(osfd, &rd);
       }
       if (in_flags & PR_POLL_WRITE) {
           FD_SET(osfd, &wt);
       }
       if (in_flags & PR_POLL_EXCEPT) {
           FD_SET(osfd, &ex);
       }
   }
   if (timeout != PR_INTERVAL_NO_TIMEOUT) {
       tv.tv_sec = PR_IntervalToSeconds(timeout);
       tv.tv_usec = PR_IntervalToMicroseconds(timeout) % PR_USEC_PER_SEC;
       tvp = &tv;
       start = PR_IntervalNow();
   }

retry:
   n = _MD_SELECT(maxfd + 1, &rd, &wt, &ex, tvp);
   if (n == -1 && errno == EINTR) {
         if (timeout == PR_INTERVAL_NO_TIMEOUT) {
              goto retry;
         } else {
             elapsed = (PRIntervalTime) (PR_IntervalNow() - start);
             if (elapsed > timeout) {
                 n = 0;  /* timed out */
            } else {
                remaining = timeout - elapsed;
                tv.tv_sec = PR_IntervalToSeconds(remaining);
                tv.tv_usec = PR_IntervalToMicroseconds(
                    remaining - PR_SecondsToInterval(tv.tv_sec));
                goto retry;
         }
         }
   }

   if (n > 0) {
       n = 0;
       for (pd = pds, epd = pd + npds; pd < epd; pd++) {
           PRInt32 osfd;
           PRInt16 in_flags = pd->in_flags;
           PRInt16 out_flags = 0;
           PRFileDesc *bottom = pd->fd;

             if ((NULL == bottom) || (in_flags == 0)) {
               pd->out_flags = 0;
               continue;
           }
           while (bottom->lower != NULL) {
               bottom = bottom->lower;
           }
           osfd = bottom->secret->md.osfd;

           if ((in_flags & PR_POLL_READ) && FD_ISSET(osfd, &rd))  {
               out_flags |= PR_POLL_READ;
           }
           if ((in_flags & PR_POLL_WRITE) && FD_ISSET(osfd, &wt)) {
               out_flags |= PR_POLL_WRITE;
           }
           if ((in_flags & PR_POLL_EXCEPT) && FD_ISSET(osfd, &ex)) {
               out_flags |= PR_POLL_EXCEPT;
           }
           pd->out_flags = out_flags;
           if (out_flags) {
               n++;
           }
       }
       PR_ASSERT(n > 0);
   } else if (n < 0) {
       err = _MD_ERRNO();
       if (err == EBADF) {
           /* Find the bad fds */
           n = 0;
           for (pd = pds, epd = pd + npds; pd < epd; pd++) {
            int optval;
            int optlen = sizeof(optval);
               PRFileDesc *bottom = pd->fd;
            pd->out_flags = 0;
               if ((NULL == bottom) || (pd->in_flags == 0)) {
                   continue;
               }
               while (bottom->lower != NULL) {
                   bottom = bottom->lower;
               }
            if (getsockopt(bottom->secret->md.osfd, SOL_SOCKET,
                    SO_TYPE, (char *) &optval, &optlen) == -1) {
                PR_ASSERT(_MD_ERRNO() == ENOTSOCK);
                if (_MD_ERRNO() == ENOTSOCK) {
                    pd->out_flags = PR_POLL_NVAL;
                    n++;
                }
            }
           }
           PR_ASSERT(n > 0);
       } else {
           PR_ASSERT(err != EINTR);  /* should have been handled above */
           _PR_MD_MAP_SELECT_ERROR(err);
      }
   }
   return n;
 }