summaryrefslogtreecommitdiff
path: root/lib/sunos57-select.c
blob: 9ee5b6fd89c084687f4a0c5b1d8e91e2e68d3c61 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
/* Work around the bug in Solaris 7 whereby a fd that is opened on
   /dev/null will cause select/poll to hang when given a NULL timeout.

   Copyright (C) 2004 Free Software Foundation, Inc.

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2, or (at your option)
   any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software Foundation,
   Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */

/* written by Mark D. Baushke */

/*
 * Observed on Solaris 7:
 *   If /dev/null is in the readfds set, it will never be marked as
 *   ready by the OS. In the case of a /dev/null fd being the only fd
 *   in the select set and timeout == NULL, the select will hang.
 *   If /dev/null is in the exceptfds set, it will not be set on
 *   return from select().
 */
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif /* HAVE_CONFIG_H */

/* The rpl_select function calls the real select. */
#undef select

#include <stdbool.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>

#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif /* HAVE_UNISTD_H */

#include "minmax.h"
#include "xtime.h"

static struct stat devnull;
static int devnull_set = -1;
int
rpl_select (int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds,
            struct timeval *timeout)
{
  int ret = 0;

  /* Argument checking */
  if (nfds < 1 || nfds > FD_SETSIZE)
    {
      errno = EINVAL;
      return -1;
    }

  /* Perform the initial stat on /dev/null */
  if (devnull_set == -1)
    devnull_set = stat ("/dev/null", &devnull);

  if (devnull_set >= 0)
    {
      int fd;
      int maxfd = -1;
      fd_set null_rfds, null_wfds;
      bool altered = false;	/* Whether we have altered the caller's args.
				 */

      FD_ZERO (&null_rfds);
      FD_ZERO (&null_wfds);

      for (fd = 0; fd < nfds; fd++)
	{
	  /* Check the callers bits for interesting fds */
	  bool isread = (readfds && FD_ISSET (fd, readfds));
	  bool isexcept = (exceptfds && FD_ISSET (fd, exceptfds));
	  bool iswrite = (writefds && FD_ISSET (fd, writefds));

	  /* Check interesting fds against /dev/null */
	  if (isread || iswrite || isexcept)
	    {
	      struct stat sb;

	      /* Equivalent to /dev/null ? */
	      if (fstat (fd, &sb) >= 0
		  && sb.st_dev == devnull.st_dev
		  && sb.st_ino == devnull.st_ino
		  && sb.st_mode == devnull.st_mode
		  && sb.st_uid == devnull.st_uid
		  && sb.st_gid == devnull.st_gid
		  && sb.st_size == devnull.st_size
		  && sb.st_blocks == devnull.st_blocks
		  && sb.st_blksize == devnull.st_blksize)
		{
		  /* Save the interesting bits for later use. */
		  if (isread)
		    {
		      FD_SET (fd, &null_rfds);
		      FD_CLR (fd, readfds);
		      altered = true;
		    }
		  if (isexcept)
		    /* Pass exception bits through.
		     *
		     * At the moment, we only know that this bug
		     * exists in Solaris 7 and so this file should
		     * only be compiled on Solaris 7. Since Solaris 7
		     * never returns ready for exceptions on
		     * /dev/null, we probably could assume this too,
		     * but since Solaris 9 is known to always return
		     * ready for exceptions on /dev/null, pass this
		     * through in case any other systems turn out to
		     * do the same. Besides, this will cause the
		     * timeout to be processed as it would have been
		     * otherwise.
		     */
		    maxfd = MAX (maxfd, fd);
		  if (iswrite)
		    {
		      /* We know of no bugs involving selecting /dev/null
		       * writefds, but we also know that /dev/null is always
		       * ready for write.  Therefore, since we have already
		       * performed all the necessary processing, avoid calling
		       * the system select for this case.
		       */
		      FD_SET (fd, &null_wfds);
		      FD_CLR (fd, writefds);
		      altered = true;
		    }
		}
	      else
		/* A non-/dev/null fd is present. */
		maxfd = MAX (maxfd, fd);
	    }
	}

      if (maxfd >= 0)
	{
	  /* we need to call select, one way or another.  */
	  if (altered)
	    {
	      /* We already have some ready bits set, so timeout immediately
	       * if no bits are set.
	       */
	      struct timeval ztime;
	      ztime.tv_sec = 0;
	      ztime.tv_usec = 0;
	      ret = select (maxfd + 1, readfds, writefds, exceptfds, &ztime);
	      if (ret == 0)
		{
		  /* Timeout.  Zero the sets since the system select might
		   * not have.
		   */
		  if (readfds)
		    FD_ZERO (readfds);
		  if (exceptfds)
		    FD_ZERO (exceptfds);
		  if (writefds)
		    FD_ZERO (writefds);
		}
	    }
	  else
	    /* No /dev/null fds.  Call select just as the user specified.  */
	    ret = select (maxfd + 1, readfds, writefds, exceptfds, timeout);
	}

      /*
       * Borrowed from the Solaris 7 man page for select(3c):
       * 
       *   On successful completion, the objects pointed to by the
       *   readfds, writefds, and exceptfds arguments are modified to
       *   indicate which file descriptors are ready for reading,
       *   ready for writing, or have an error condition pending,
       *   respectively. For each file descriptor less than nfds, the
       *   corresponding bit will be set on successful completion if
       *   it was set on input and the associated condition is true
       *   for that file descriptor.
       *  
       *   On failure, the objects pointed to by the readfds,
       *   writefds, and exceptfds arguments are not modified. If the
       *   timeout interval expires without the specified condition
       *   being true for any of the specified file descriptors, the
       *   objects pointed to by the readfs, writefs, and errorfds
       *   arguments have all bits set to 0.
       *  
       *   On successful completion, select() returns the total number
       *   of bits set in the bit masks. Otherwise, -1 is returned,
       *   and errno is set to indicate the error.
       */

      /* Fix up the fd sets for any changes we may have made. */
      if (altered)
	{
	  /* Tell the caller that nothing is blocking the /dev/null fds */
	  for (fd = 0; fd < nfds; fd++)
	    {
	      /* If ret < 0, then we still need to restore the fd sets.  */
	      if (FD_ISSET (fd, &null_rfds))
		{
		  FD_SET (fd, readfds);
		  if (ret >= 0)
		    ret++;
		}
	      if (FD_ISSET (fd, &null_wfds))
		{
		  FD_SET (fd, writefds);
		  if (ret >= 0)
		    ret++;
		}		  
	    }
	}
    }
  else
    ret = select (nfds, readfds, writefds, exceptfds, timeout);

  return ret;
}