Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 1 | /*********************************************************** |
Guido van Rossum | 524b588 | 1995-01-04 19:10:35 +0000 | [diff] [blame] | 2 | Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam, |
| 3 | The Netherlands. |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 4 | |
| 5 | All Rights Reserved |
| 6 | |
Guido van Rossum | d266eb4 | 1996-10-25 14:44:06 +0000 | [diff] [blame] | 7 | Permission to use, copy, modify, and distribute this software and its |
| 8 | documentation for any purpose and without fee is hereby granted, |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 9 | provided that the above copyright notice appear in all copies and that |
Guido van Rossum | d266eb4 | 1996-10-25 14:44:06 +0000 | [diff] [blame] | 10 | both that copyright notice and this permission notice appear in |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 11 | supporting documentation, and that the names of Stichting Mathematisch |
Guido van Rossum | d266eb4 | 1996-10-25 14:44:06 +0000 | [diff] [blame] | 12 | Centrum or CWI or Corporation for National Research Initiatives or |
| 13 | CNRI not be used in advertising or publicity pertaining to |
| 14 | distribution of the software without specific, written prior |
| 15 | permission. |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 16 | |
Guido van Rossum | d266eb4 | 1996-10-25 14:44:06 +0000 | [diff] [blame] | 17 | While CWI is the initial source for this software, a modified version |
| 18 | is made available by the Corporation for National Research Initiatives |
| 19 | (CNRI) at the Internet address ftp://ftp.python.org. |
| 20 | |
| 21 | STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH |
| 22 | REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF |
| 23 | MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH |
| 24 | CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL |
| 25 | DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR |
| 26 | PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER |
| 27 | TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR |
| 28 | PERFORMANCE OF THIS SOFTWARE. |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 29 | |
| 30 | ******************************************************************/ |
| 31 | |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 32 | /* select - Module containing unix select(2) call. |
| 33 | Under Unix, the file descriptors are small integers. |
| 34 | Under Win32, select only exists for sockets, and sockets may |
| 35 | have any value except INVALID_SOCKET. |
| 36 | */ |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 37 | |
| 38 | #include "allobjects.h" |
| 39 | #include "modsupport.h" |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 40 | #include "ceval.h" |
| 41 | |
Guido van Rossum | a376cc5 | 1996-12-05 23:43:35 +0000 | [diff] [blame] | 42 | #ifdef HAVE_UNISTD_H |
| 43 | #include <unistd.h> |
| 44 | #endif |
| 45 | |
Guido van Rossum | 3727317 | 1996-12-09 18:47:43 +0000 | [diff] [blame^] | 46 | #ifdef __sgi |
| 47 | /* This is missing from unistd.h */ |
| 48 | extern void bzero(); |
| 49 | #endif |
| 50 | |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 51 | #include <sys/types.h> |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 52 | |
Guido van Rossum | 6f489d9 | 1996-06-28 20:15:15 +0000 | [diff] [blame] | 53 | #ifdef MS_WINDOWS |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 54 | #include <winsock.h> |
| 55 | #else |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 56 | #include "myselect.h" /* Also includes mytime.h */ |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 57 | #define SOCKET int |
| 58 | #endif |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 59 | |
| 60 | static object *SelectError; |
| 61 | |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 62 | typedef struct { /* list of Python objects and their file descriptor */ |
| 63 | object *obj; |
| 64 | SOCKET fd; |
| 65 | } pylist; |
| 66 | |
| 67 | static int |
Guido van Rossum | 4fbf798 | 1992-08-04 09:13:45 +0000 | [diff] [blame] | 68 | list2set(list, set, fd2obj) |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 69 | object *list; |
| 70 | fd_set *set; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 71 | pylist fd2obj[FD_SETSIZE + 3]; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 72 | { |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 73 | int i, len, index, max = -1; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 74 | object *o, *filenomethod, *fno; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 75 | SOCKET v; |
Guido van Rossum | 07432c0 | 1995-03-29 16:47:45 +0000 | [diff] [blame] | 76 | |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 77 | index = 0; |
| 78 | fd2obj[0].obj = (object*)0; /* set list to zero size */ |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 79 | |
| 80 | FD_ZERO(set); |
| 81 | len = getlistsize(list); |
| 82 | for( i=0; i<len; i++ ) { |
| 83 | o = getlistitem(list, i); |
| 84 | if ( is_intobject(o) ) { |
| 85 | v = getintvalue(o); |
| 86 | } else if ( (filenomethod = getattr(o, "fileno")) != NULL ) { |
| 87 | fno = call_object(filenomethod, NULL); |
Guido van Rossum | a849b83 | 1993-05-12 11:35:44 +0000 | [diff] [blame] | 88 | DECREF(filenomethod); |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 89 | if ( fno == NULL ) |
| 90 | return -1; |
| 91 | if ( !is_intobject(fno) ) { |
| 92 | err_badarg(); |
Guido van Rossum | a849b83 | 1993-05-12 11:35:44 +0000 | [diff] [blame] | 93 | DECREF(fno); |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 94 | return -1; |
| 95 | } |
| 96 | v = getintvalue(fno); |
| 97 | DECREF(fno); |
| 98 | } else { |
| 99 | err_badarg(); |
| 100 | return -1; |
| 101 | } |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 102 | #ifdef _MSC_VER |
| 103 | max = 0; /* not used for Win32 */ |
| 104 | #else |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 105 | if ( v < 0 || v >= FD_SETSIZE ) { |
| 106 | err_setstr(ValueError, "filedescriptor out of range in select()"); |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 107 | return -1; |
| 108 | } |
| 109 | if ( v > max ) max = v; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 110 | #endif |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 111 | FD_SET(v, set); |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 112 | /* add object and its file descriptor to the list */ |
| 113 | if ( index >= FD_SETSIZE ) { |
| 114 | err_setstr(ValueError, "too many file descriptors in select()"); |
| 115 | return -1; |
| 116 | } |
| 117 | fd2obj[index].obj = o; |
| 118 | fd2obj[index].fd = v; |
| 119 | fd2obj[++index].obj = (object *)0; /* sentinel */ |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 120 | } |
| 121 | return max+1; |
| 122 | } |
| 123 | |
| 124 | static object * |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 125 | set2list(set, fd2obj) |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 126 | fd_set *set; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 127 | pylist fd2obj[FD_SETSIZE + 3]; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 128 | { |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 129 | int j, num=0; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 130 | object *list, *o; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 131 | SOCKET fd; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 132 | |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 133 | for(j=0; fd2obj[j].obj; j++) |
| 134 | if ( FD_ISSET(fd2obj[j].fd, set) ) |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 135 | num++; |
| 136 | list = newlistobject(num); |
| 137 | num = 0; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 138 | for(j=0; fd2obj[j].obj; j++) { |
| 139 | fd = fd2obj[j].fd; |
| 140 | if ( FD_ISSET(fd, set) ) { |
| 141 | #ifndef _MSC_VER |
| 142 | if ( fd > FD_SETSIZE ) { |
Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 143 | err_setstr(SystemError, |
| 144 | "filedescriptor out of range returned in select()"); |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 145 | return NULL; |
| 146 | } |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 147 | #endif |
| 148 | o = fd2obj[j].obj; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 149 | INCREF(o); |
| 150 | setlistitem(list, num, o); |
| 151 | num++; |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 152 | } |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 153 | } |
| 154 | return list; |
| 155 | } |
| 156 | |
| 157 | static object * |
| 158 | select_select(self, args) |
| 159 | object *self; |
| 160 | object *args; |
| 161 | { |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 162 | pylist rfd2obj[FD_SETSIZE + 3], wfd2obj[FD_SETSIZE + 3], efd2obj[FD_SETSIZE + 3]; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 163 | object *ifdlist, *ofdlist, *efdlist; |
Guido van Rossum | c7a2270 | 1993-11-01 16:27:16 +0000 | [diff] [blame] | 164 | object *ret, *tout; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 165 | fd_set ifdset, ofdset, efdset; |
| 166 | double timeout; |
| 167 | struct timeval tv, *tvp; |
| 168 | int seconds; |
| 169 | int imax, omax, emax, max; |
| 170 | int n; |
| 171 | |
| 172 | |
| 173 | /* Get args. Looks funny because of optional timeout argument */ |
Guido van Rossum | c7a2270 | 1993-11-01 16:27:16 +0000 | [diff] [blame] | 174 | if ( getargs(args, "(OOOO)", &ifdlist, &ofdlist, &efdlist, &tout) ) { |
Guido van Rossum | c7a2270 | 1993-11-01 16:27:16 +0000 | [diff] [blame] | 175 | if (tout == None) |
| 176 | tvp = (struct timeval *)0; |
| 177 | else { |
Sjoerd Mullender | 78ed420 | 1993-11-02 15:34:23 +0000 | [diff] [blame] | 178 | if (!getargs(tout, "d;timeout must be float or None", &timeout)) |
Guido van Rossum | c7a2270 | 1993-11-01 16:27:16 +0000 | [diff] [blame] | 179 | return NULL; |
Sjoerd Mullender | 78ed420 | 1993-11-02 15:34:23 +0000 | [diff] [blame] | 180 | seconds = (int)timeout; |
Guido van Rossum | c7a2270 | 1993-11-01 16:27:16 +0000 | [diff] [blame] | 181 | timeout = timeout - (double)seconds; |
| 182 | tv.tv_sec = seconds; |
| 183 | tv.tv_usec = (int)(timeout*1000000.0); |
| 184 | tvp = &tv; |
| 185 | } |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 186 | } else { |
| 187 | /* Doesn't have 4 args, that means no timeout */ |
| 188 | err_clear(); |
| 189 | if (!getargs(args, "(OOO)", &ifdlist, &ofdlist, &efdlist) ) |
| 190 | return 0; |
| 191 | tvp = (struct timeval *)0; |
| 192 | } |
| 193 | if ( !is_listobject(ifdlist) || !is_listobject(ofdlist) || |
| 194 | !is_listobject(efdlist) ) { |
| 195 | err_badarg(); |
| 196 | return 0; |
| 197 | } |
| 198 | |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 199 | /* Convert lists to fd_sets, and get maximum fd number */ |
Guido van Rossum | 07432c0 | 1995-03-29 16:47:45 +0000 | [diff] [blame] | 200 | if( (imax=list2set(ifdlist, &ifdset, rfd2obj)) < 0 ) |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 201 | return 0; |
Guido van Rossum | 07432c0 | 1995-03-29 16:47:45 +0000 | [diff] [blame] | 202 | if( (omax=list2set(ofdlist, &ofdset, wfd2obj)) < 0 ) |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 203 | return 0; |
Guido van Rossum | 07432c0 | 1995-03-29 16:47:45 +0000 | [diff] [blame] | 204 | if( (emax=list2set(efdlist, &efdset, efd2obj)) < 0 ) |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 205 | return 0; |
| 206 | max = imax; |
| 207 | if ( omax > max ) max = omax; |
| 208 | if ( emax > max ) max = emax; |
| 209 | |
Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 210 | BGN_SAVE |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 211 | n = select(max, &ifdset, &ofdset, &efdset, tvp); |
Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 212 | END_SAVE |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 213 | |
| 214 | if ( n < 0 ) { |
| 215 | err_errno(SelectError); |
| 216 | return 0; |
| 217 | } |
| 218 | |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 219 | if ( n == 0 ) { /* Speedup hack */ |
| 220 | ifdlist = newlistobject(0); |
| 221 | ret = mkvalue("OOO", ifdlist, ifdlist, ifdlist); |
| 222 | XDECREF(ifdlist); |
| 223 | return ret; |
| 224 | } |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 225 | |
Guido van Rossum | 4f0fbf8 | 1996-06-12 04:22:53 +0000 | [diff] [blame] | 226 | ifdlist = set2list(&ifdset, rfd2obj); |
| 227 | ofdlist = set2list(&ofdset, wfd2obj); |
| 228 | efdlist = set2list(&efdset, efd2obj); |
Guido van Rossum | 6f5afc9 | 1993-02-05 09:46:15 +0000 | [diff] [blame] | 229 | ret = mkvalue("OOO", ifdlist, ofdlist, efdlist); |
| 230 | XDECREF(ifdlist); |
| 231 | XDECREF(ofdlist); |
| 232 | XDECREF(efdlist); |
| 233 | return ret; |
Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 234 | } |
| 235 | |
| 236 | |
| 237 | static struct methodlist select_methods[] = { |
| 238 | { "select", select_select }, |
| 239 | { 0, 0 }, |
| 240 | }; |
| 241 | |
| 242 | |
| 243 | void |
| 244 | initselect() |
| 245 | { |
| 246 | object *m, *d; |
| 247 | m = initmodule("select", select_methods); |
| 248 | d = getmoduledict(m); |
| 249 | SelectError = newstringobject("select.error"); |
| 250 | if ( SelectError == NULL || dictinsert(d, "error", SelectError) ) |
| 251 | fatal("Cannot define select.error"); |
| 252 | } |