[2] | 1 | /* select - Module containing unix select(2) call.
|
---|
| 2 | Under Unix, the file descriptors are small integers.
|
---|
| 3 | Under Win32, select only exists for sockets, and sockets may
|
---|
| 4 | have any value except INVALID_SOCKET.
|
---|
| 5 | Under BeOS, we suffer the same dichotomy as Win32; sockets can be anything
|
---|
| 6 | >= 0.
|
---|
| 7 | */
|
---|
| 8 |
|
---|
| 9 | #include "Python.h"
|
---|
| 10 | #include <structmember.h>
|
---|
| 11 |
|
---|
| 12 | #ifdef __APPLE__
|
---|
| 13 | /* Perform runtime testing for a broken poll on OSX to make it easier
|
---|
| 14 | * to use the same binary on multiple releases of the OS.
|
---|
| 15 | */
|
---|
| 16 | #undef HAVE_BROKEN_POLL
|
---|
| 17 | #endif
|
---|
| 18 |
|
---|
| 19 | /* Windows #defines FD_SETSIZE to 64 if FD_SETSIZE isn't already defined.
|
---|
| 20 | 64 is too small (too many people have bumped into that limit).
|
---|
| 21 | Here we boost it.
|
---|
| 22 | Users who want even more than the boosted limit should #define
|
---|
| 23 | FD_SETSIZE higher before this; e.g., via compiler /D switch.
|
---|
| 24 | */
|
---|
| 25 | #if defined(MS_WINDOWS) && !defined(FD_SETSIZE)
|
---|
| 26 | #define FD_SETSIZE 512
|
---|
[391] | 27 | #endif
|
---|
[2] | 28 |
|
---|
| 29 | #if defined(HAVE_POLL_H)
|
---|
| 30 | #include <poll.h>
|
---|
| 31 | #elif defined(HAVE_SYS_POLL_H)
|
---|
| 32 | #include <sys/poll.h>
|
---|
| 33 | #endif
|
---|
| 34 |
|
---|
| 35 | #ifdef __sgi
|
---|
| 36 | /* This is missing from unistd.h */
|
---|
| 37 | extern void bzero(void *, int);
|
---|
| 38 | #endif
|
---|
| 39 |
|
---|
| 40 | #ifdef HAVE_SYS_TYPES_H
|
---|
| 41 | #include <sys/types.h>
|
---|
| 42 | #endif
|
---|
| 43 |
|
---|
| 44 | #if defined(PYOS_OS2) && !defined(PYCC_GCC)
|
---|
| 45 | #include <sys/time.h>
|
---|
| 46 | #include <utils.h>
|
---|
| 47 | #endif
|
---|
| 48 |
|
---|
| 49 | #ifdef MS_WINDOWS
|
---|
[391] | 50 | # include <winsock2.h>
|
---|
[2] | 51 | #else
|
---|
| 52 | # define SOCKET int
|
---|
| 53 | # ifdef __BEOS__
|
---|
| 54 | # include <net/socket.h>
|
---|
| 55 | # elif defined(__VMS)
|
---|
| 56 | # include <socket.h>
|
---|
| 57 | # endif
|
---|
| 58 | #endif
|
---|
| 59 |
|
---|
| 60 | static PyObject *SelectError;
|
---|
| 61 |
|
---|
| 62 | /* list of Python objects and their file descriptor */
|
---|
| 63 | typedef struct {
|
---|
[391] | 64 | PyObject *obj; /* owned reference */
|
---|
| 65 | SOCKET fd;
|
---|
| 66 | int sentinel; /* -1 == sentinel */
|
---|
[2] | 67 | } pylist;
|
---|
| 68 |
|
---|
| 69 | static void
|
---|
| 70 | reap_obj(pylist fd2obj[FD_SETSIZE + 1])
|
---|
| 71 | {
|
---|
[391] | 72 | int i;
|
---|
| 73 | for (i = 0; i < FD_SETSIZE + 1 && fd2obj[i].sentinel >= 0; i++) {
|
---|
| 74 | Py_XDECREF(fd2obj[i].obj);
|
---|
| 75 | fd2obj[i].obj = NULL;
|
---|
| 76 | }
|
---|
| 77 | fd2obj[0].sentinel = -1;
|
---|
[2] | 78 | }
|
---|
| 79 |
|
---|
| 80 |
|
---|
| 81 | /* returns -1 and sets the Python exception if an error occurred, otherwise
|
---|
| 82 | returns a number >= 0
|
---|
| 83 | */
|
---|
| 84 | static int
|
---|
| 85 | seq2set(PyObject *seq, fd_set *set, pylist fd2obj[FD_SETSIZE + 1])
|
---|
| 86 | {
|
---|
[391] | 87 | int i;
|
---|
| 88 | int max = -1;
|
---|
| 89 | int index = 0;
|
---|
| 90 | PyObject* fast_seq = NULL;
|
---|
| 91 | PyObject* o = NULL;
|
---|
[2] | 92 |
|
---|
[391] | 93 | fd2obj[0].obj = (PyObject*)0; /* set list to zero size */
|
---|
| 94 | FD_ZERO(set);
|
---|
[2] | 95 |
|
---|
[391] | 96 | fast_seq = PySequence_Fast(seq, "arguments 1-3 must be sequences");
|
---|
| 97 | if (!fast_seq)
|
---|
| 98 | return -1;
|
---|
[2] | 99 |
|
---|
[391] | 100 | for (i = 0; i < PySequence_Fast_GET_SIZE(fast_seq); i++) {
|
---|
| 101 | SOCKET v;
|
---|
[2] | 102 |
|
---|
[391] | 103 | /* any intervening fileno() calls could decr this refcnt */
|
---|
| 104 | if (!(o = PySequence_Fast_GET_ITEM(fast_seq, i)))
|
---|
| 105 | return -1;
|
---|
[2] | 106 |
|
---|
[391] | 107 | Py_INCREF(o);
|
---|
| 108 | v = PyObject_AsFileDescriptor( o );
|
---|
| 109 | if (v == -1) goto finally;
|
---|
[2] | 110 |
|
---|
| 111 | #if defined(_MSC_VER)
|
---|
[391] | 112 | max = 0; /* not used for Win32 */
|
---|
[2] | 113 | #else /* !_MSC_VER */
|
---|
[391] | 114 | if (!_PyIsSelectable_fd(v)) {
|
---|
| 115 | PyErr_SetString(PyExc_ValueError,
|
---|
| 116 | "filedescriptor out of range in select()");
|
---|
| 117 | goto finally;
|
---|
| 118 | }
|
---|
| 119 | if (v > max)
|
---|
| 120 | max = v;
|
---|
[2] | 121 | #endif /* _MSC_VER */
|
---|
[391] | 122 | FD_SET(v, set);
|
---|
[2] | 123 |
|
---|
[391] | 124 | /* add object and its file descriptor to the list */
|
---|
| 125 | if (index >= FD_SETSIZE) {
|
---|
| 126 | PyErr_SetString(PyExc_ValueError,
|
---|
| 127 | "too many file descriptors in select()");
|
---|
| 128 | goto finally;
|
---|
| 129 | }
|
---|
| 130 | fd2obj[index].obj = o;
|
---|
| 131 | fd2obj[index].fd = v;
|
---|
| 132 | fd2obj[index].sentinel = 0;
|
---|
| 133 | fd2obj[++index].sentinel = -1;
|
---|
| 134 | }
|
---|
| 135 | Py_DECREF(fast_seq);
|
---|
| 136 | return max+1;
|
---|
[2] | 137 |
|
---|
| 138 | finally:
|
---|
[391] | 139 | Py_XDECREF(o);
|
---|
| 140 | Py_DECREF(fast_seq);
|
---|
| 141 | return -1;
|
---|
[2] | 142 | }
|
---|
| 143 |
|
---|
| 144 | /* returns NULL and sets the Python exception if an error occurred */
|
---|
| 145 | static PyObject *
|
---|
| 146 | set2list(fd_set *set, pylist fd2obj[FD_SETSIZE + 1])
|
---|
| 147 | {
|
---|
[391] | 148 | int i, j, count=0;
|
---|
| 149 | PyObject *list, *o;
|
---|
| 150 | SOCKET fd;
|
---|
[2] | 151 |
|
---|
[391] | 152 | for (j = 0; fd2obj[j].sentinel >= 0; j++) {
|
---|
| 153 | if (FD_ISSET(fd2obj[j].fd, set))
|
---|
| 154 | count++;
|
---|
| 155 | }
|
---|
| 156 | list = PyList_New(count);
|
---|
| 157 | if (!list)
|
---|
| 158 | return NULL;
|
---|
[2] | 159 |
|
---|
[391] | 160 | i = 0;
|
---|
| 161 | for (j = 0; fd2obj[j].sentinel >= 0; j++) {
|
---|
| 162 | fd = fd2obj[j].fd;
|
---|
| 163 | if (FD_ISSET(fd, set)) {
|
---|
| 164 | o = fd2obj[j].obj;
|
---|
| 165 | fd2obj[j].obj = NULL;
|
---|
| 166 | /* transfer ownership */
|
---|
| 167 | if (PyList_SetItem(list, i, o) < 0)
|
---|
| 168 | goto finally;
|
---|
[2] | 169 |
|
---|
[391] | 170 | i++;
|
---|
| 171 | }
|
---|
| 172 | }
|
---|
| 173 | return list;
|
---|
[2] | 174 | finally:
|
---|
[391] | 175 | Py_DECREF(list);
|
---|
| 176 | return NULL;
|
---|
[2] | 177 | }
|
---|
| 178 |
|
---|
| 179 | #undef SELECT_USES_HEAP
|
---|
| 180 | #if FD_SETSIZE > 1024
|
---|
| 181 | #define SELECT_USES_HEAP
|
---|
| 182 | #endif /* FD_SETSIZE > 1024 */
|
---|
| 183 |
|
---|
| 184 | static PyObject *
|
---|
| 185 | select_select(PyObject *self, PyObject *args)
|
---|
| 186 | {
|
---|
| 187 | #ifdef SELECT_USES_HEAP
|
---|
[391] | 188 | pylist *rfd2obj, *wfd2obj, *efd2obj;
|
---|
[2] | 189 | #else /* !SELECT_USES_HEAP */
|
---|
[391] | 190 | /* XXX: All this should probably be implemented as follows:
|
---|
| 191 | * - find the highest descriptor we're interested in
|
---|
| 192 | * - add one
|
---|
| 193 | * - that's the size
|
---|
| 194 | * See: Stevens, APitUE, $12.5.1
|
---|
| 195 | */
|
---|
| 196 | pylist rfd2obj[FD_SETSIZE + 1];
|
---|
| 197 | pylist wfd2obj[FD_SETSIZE + 1];
|
---|
| 198 | pylist efd2obj[FD_SETSIZE + 1];
|
---|
[2] | 199 | #endif /* SELECT_USES_HEAP */
|
---|
[391] | 200 | PyObject *ifdlist, *ofdlist, *efdlist;
|
---|
| 201 | PyObject *ret = NULL;
|
---|
| 202 | PyObject *tout = Py_None;
|
---|
| 203 | fd_set ifdset, ofdset, efdset;
|
---|
| 204 | double timeout;
|
---|
| 205 | struct timeval tv, *tvp;
|
---|
| 206 | long seconds;
|
---|
| 207 | int imax, omax, emax, max;
|
---|
| 208 | int n;
|
---|
[2] | 209 |
|
---|
[391] | 210 | /* convert arguments */
|
---|
| 211 | if (!PyArg_UnpackTuple(args, "select", 3, 4,
|
---|
| 212 | &ifdlist, &ofdlist, &efdlist, &tout))
|
---|
| 213 | return NULL;
|
---|
[2] | 214 |
|
---|
[391] | 215 | if (tout == Py_None)
|
---|
| 216 | tvp = (struct timeval *)0;
|
---|
| 217 | else if (!PyNumber_Check(tout)) {
|
---|
| 218 | PyErr_SetString(PyExc_TypeError,
|
---|
| 219 | "timeout must be a float or None");
|
---|
| 220 | return NULL;
|
---|
| 221 | }
|
---|
| 222 | else {
|
---|
| 223 | timeout = PyFloat_AsDouble(tout);
|
---|
| 224 | if (timeout == -1 && PyErr_Occurred())
|
---|
| 225 | return NULL;
|
---|
| 226 | if (timeout > (double)LONG_MAX) {
|
---|
| 227 | PyErr_SetString(PyExc_OverflowError,
|
---|
| 228 | "timeout period too long");
|
---|
| 229 | return NULL;
|
---|
| 230 | }
|
---|
| 231 | seconds = (long)timeout;
|
---|
| 232 | timeout = timeout - (double)seconds;
|
---|
| 233 | tv.tv_sec = seconds;
|
---|
| 234 | tv.tv_usec = (long)(timeout * 1E6);
|
---|
| 235 | tvp = &tv;
|
---|
| 236 | }
|
---|
[2] | 237 |
|
---|
| 238 |
|
---|
| 239 | #ifdef SELECT_USES_HEAP
|
---|
[391] | 240 | /* Allocate memory for the lists */
|
---|
| 241 | rfd2obj = PyMem_NEW(pylist, FD_SETSIZE + 1);
|
---|
| 242 | wfd2obj = PyMem_NEW(pylist, FD_SETSIZE + 1);
|
---|
| 243 | efd2obj = PyMem_NEW(pylist, FD_SETSIZE + 1);
|
---|
| 244 | if (rfd2obj == NULL || wfd2obj == NULL || efd2obj == NULL) {
|
---|
| 245 | if (rfd2obj) PyMem_DEL(rfd2obj);
|
---|
| 246 | if (wfd2obj) PyMem_DEL(wfd2obj);
|
---|
| 247 | if (efd2obj) PyMem_DEL(efd2obj);
|
---|
| 248 | return PyErr_NoMemory();
|
---|
| 249 | }
|
---|
[2] | 250 | #endif /* SELECT_USES_HEAP */
|
---|
[391] | 251 | /* Convert sequences to fd_sets, and get maximum fd number
|
---|
| 252 | * propagates the Python exception set in seq2set()
|
---|
| 253 | */
|
---|
| 254 | rfd2obj[0].sentinel = -1;
|
---|
| 255 | wfd2obj[0].sentinel = -1;
|
---|
| 256 | efd2obj[0].sentinel = -1;
|
---|
| 257 | if ((imax=seq2set(ifdlist, &ifdset, rfd2obj)) < 0)
|
---|
| 258 | goto finally;
|
---|
| 259 | if ((omax=seq2set(ofdlist, &ofdset, wfd2obj)) < 0)
|
---|
| 260 | goto finally;
|
---|
| 261 | if ((emax=seq2set(efdlist, &efdset, efd2obj)) < 0)
|
---|
| 262 | goto finally;
|
---|
| 263 | max = imax;
|
---|
| 264 | if (omax > max) max = omax;
|
---|
| 265 | if (emax > max) max = emax;
|
---|
[2] | 266 |
|
---|
[391] | 267 | Py_BEGIN_ALLOW_THREADS
|
---|
| 268 | n = select(max, &ifdset, &ofdset, &efdset, tvp);
|
---|
| 269 | Py_END_ALLOW_THREADS
|
---|
[2] | 270 |
|
---|
| 271 | #ifdef MS_WINDOWS
|
---|
[391] | 272 | if (n == SOCKET_ERROR) {
|
---|
| 273 | PyErr_SetExcFromWindowsErr(SelectError, WSAGetLastError());
|
---|
| 274 | }
|
---|
[2] | 275 | #else
|
---|
[391] | 276 | if (n < 0) {
|
---|
| 277 | PyErr_SetFromErrno(SelectError);
|
---|
| 278 | }
|
---|
[2] | 279 | #endif
|
---|
[391] | 280 | else {
|
---|
| 281 | /* any of these three calls can raise an exception. it's more
|
---|
| 282 | convenient to test for this after all three calls... but
|
---|
| 283 | is that acceptable?
|
---|
| 284 | */
|
---|
| 285 | ifdlist = set2list(&ifdset, rfd2obj);
|
---|
| 286 | ofdlist = set2list(&ofdset, wfd2obj);
|
---|
| 287 | efdlist = set2list(&efdset, efd2obj);
|
---|
| 288 | if (PyErr_Occurred())
|
---|
| 289 | ret = NULL;
|
---|
| 290 | else
|
---|
| 291 | ret = PyTuple_Pack(3, ifdlist, ofdlist, efdlist);
|
---|
[2] | 292 |
|
---|
[391] | 293 | Py_DECREF(ifdlist);
|
---|
| 294 | Py_DECREF(ofdlist);
|
---|
| 295 | Py_DECREF(efdlist);
|
---|
| 296 | }
|
---|
| 297 |
|
---|
[2] | 298 | finally:
|
---|
[391] | 299 | reap_obj(rfd2obj);
|
---|
| 300 | reap_obj(wfd2obj);
|
---|
| 301 | reap_obj(efd2obj);
|
---|
[2] | 302 | #ifdef SELECT_USES_HEAP
|
---|
[391] | 303 | PyMem_DEL(rfd2obj);
|
---|
| 304 | PyMem_DEL(wfd2obj);
|
---|
| 305 | PyMem_DEL(efd2obj);
|
---|
[2] | 306 | #endif /* SELECT_USES_HEAP */
|
---|
[391] | 307 | return ret;
|
---|
[2] | 308 | }
|
---|
| 309 |
|
---|
| 310 | #if defined(HAVE_POLL) && !defined(HAVE_BROKEN_POLL)
|
---|
[391] | 311 | /*
|
---|
[2] | 312 | * poll() support
|
---|
| 313 | */
|
---|
| 314 |
|
---|
| 315 | typedef struct {
|
---|
[391] | 316 | PyObject_HEAD
|
---|
| 317 | PyObject *dict;
|
---|
| 318 | int ufd_uptodate;
|
---|
| 319 | int ufd_len;
|
---|
| 320 | struct pollfd *ufds;
|
---|
| 321 | int poll_running;
|
---|
[2] | 322 | } pollObject;
|
---|
| 323 |
|
---|
| 324 | static PyTypeObject poll_Type;
|
---|
| 325 |
|
---|
[391] | 326 | /* Update the malloc'ed array of pollfds to match the dictionary
|
---|
[2] | 327 | contained within a pollObject. Return 1 on success, 0 on an error.
|
---|
| 328 | */
|
---|
| 329 |
|
---|
| 330 | static int
|
---|
| 331 | update_ufd_array(pollObject *self)
|
---|
| 332 | {
|
---|
[391] | 333 | Py_ssize_t i, pos;
|
---|
| 334 | PyObject *key, *value;
|
---|
| 335 | struct pollfd *old_ufds = self->ufds;
|
---|
[2] | 336 |
|
---|
[391] | 337 | self->ufd_len = PyDict_Size(self->dict);
|
---|
| 338 | PyMem_RESIZE(self->ufds, struct pollfd, self->ufd_len);
|
---|
| 339 | if (self->ufds == NULL) {
|
---|
| 340 | self->ufds = old_ufds;
|
---|
| 341 | PyErr_NoMemory();
|
---|
| 342 | return 0;
|
---|
| 343 | }
|
---|
[2] | 344 |
|
---|
[391] | 345 | i = pos = 0;
|
---|
| 346 | while (PyDict_Next(self->dict, &pos, &key, &value)) {
|
---|
| 347 | assert(i < self->ufd_len);
|
---|
| 348 | /* Never overflow */
|
---|
| 349 | self->ufds[i].fd = (int)PyInt_AsLong(key);
|
---|
| 350 | self->ufds[i].events = (short)PyInt_AsLong(value);
|
---|
| 351 | i++;
|
---|
| 352 | }
|
---|
| 353 | assert(i == self->ufd_len);
|
---|
| 354 | self->ufd_uptodate = 1;
|
---|
| 355 | return 1;
|
---|
[2] | 356 | }
|
---|
| 357 |
|
---|
| 358 | PyDoc_STRVAR(poll_register_doc,
|
---|
| 359 | "register(fd [, eventmask] ) -> None\n\n\
|
---|
| 360 | Register a file descriptor with the polling object.\n\
|
---|
| 361 | fd -- either an integer, or an object with a fileno() method returning an\n\
|
---|
| 362 | int.\n\
|
---|
| 363 | events -- an optional bitmask describing the type of events to check for");
|
---|
| 364 |
|
---|
| 365 | static PyObject *
|
---|
[391] | 366 | poll_register(pollObject *self, PyObject *args)
|
---|
[2] | 367 | {
|
---|
[391] | 368 | PyObject *o, *key, *value;
|
---|
| 369 | int fd;
|
---|
| 370 | short events = POLLIN | POLLPRI | POLLOUT;
|
---|
| 371 | int err;
|
---|
[2] | 372 |
|
---|
[391] | 373 | if (!PyArg_ParseTuple(args, "O|h:register", &o, &events)) {
|
---|
| 374 | return NULL;
|
---|
| 375 | }
|
---|
[2] | 376 |
|
---|
[391] | 377 | fd = PyObject_AsFileDescriptor(o);
|
---|
| 378 | if (fd == -1) return NULL;
|
---|
[2] | 379 |
|
---|
[391] | 380 | /* Add entry to the internal dictionary: the key is the
|
---|
| 381 | file descriptor, and the value is the event mask. */
|
---|
| 382 | key = PyInt_FromLong(fd);
|
---|
| 383 | if (key == NULL)
|
---|
| 384 | return NULL;
|
---|
| 385 | value = PyInt_FromLong(events);
|
---|
| 386 | if (value == NULL) {
|
---|
| 387 | Py_DECREF(key);
|
---|
| 388 | return NULL;
|
---|
| 389 | }
|
---|
| 390 | err = PyDict_SetItem(self->dict, key, value);
|
---|
| 391 | Py_DECREF(key);
|
---|
| 392 | Py_DECREF(value);
|
---|
| 393 | if (err < 0)
|
---|
| 394 | return NULL;
|
---|
[2] | 395 |
|
---|
[391] | 396 | self->ufd_uptodate = 0;
|
---|
| 397 |
|
---|
| 398 | Py_INCREF(Py_None);
|
---|
| 399 | return Py_None;
|
---|
[2] | 400 | }
|
---|
| 401 |
|
---|
| 402 | PyDoc_STRVAR(poll_modify_doc,
|
---|
| 403 | "modify(fd, eventmask) -> None\n\n\
|
---|
| 404 | Modify an already registered file descriptor.\n\
|
---|
| 405 | fd -- either an integer, or an object with a fileno() method returning an\n\
|
---|
| 406 | int.\n\
|
---|
| 407 | events -- an optional bitmask describing the type of events to check for");
|
---|
| 408 |
|
---|
| 409 | static PyObject *
|
---|
| 410 | poll_modify(pollObject *self, PyObject *args)
|
---|
| 411 | {
|
---|
[391] | 412 | PyObject *o, *key, *value;
|
---|
| 413 | int fd, events;
|
---|
| 414 | int err;
|
---|
[2] | 415 |
|
---|
[391] | 416 | if (!PyArg_ParseTuple(args, "Oi:modify", &o, &events)) {
|
---|
| 417 | return NULL;
|
---|
| 418 | }
|
---|
[2] | 419 |
|
---|
[391] | 420 | fd = PyObject_AsFileDescriptor(o);
|
---|
| 421 | if (fd == -1) return NULL;
|
---|
[2] | 422 |
|
---|
[391] | 423 | /* Modify registered fd */
|
---|
| 424 | key = PyInt_FromLong(fd);
|
---|
| 425 | if (key == NULL)
|
---|
| 426 | return NULL;
|
---|
| 427 | if (PyDict_GetItem(self->dict, key) == NULL) {
|
---|
| 428 | errno = ENOENT;
|
---|
| 429 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 430 | return NULL;
|
---|
| 431 | }
|
---|
| 432 | value = PyInt_FromLong(events);
|
---|
| 433 | if (value == NULL) {
|
---|
| 434 | Py_DECREF(key);
|
---|
| 435 | return NULL;
|
---|
| 436 | }
|
---|
| 437 | err = PyDict_SetItem(self->dict, key, value);
|
---|
| 438 | Py_DECREF(key);
|
---|
| 439 | Py_DECREF(value);
|
---|
| 440 | if (err < 0)
|
---|
| 441 | return NULL;
|
---|
[2] | 442 |
|
---|
[391] | 443 | self->ufd_uptodate = 0;
|
---|
| 444 |
|
---|
| 445 | Py_INCREF(Py_None);
|
---|
| 446 | return Py_None;
|
---|
[2] | 447 | }
|
---|
| 448 |
|
---|
| 449 |
|
---|
| 450 | PyDoc_STRVAR(poll_unregister_doc,
|
---|
| 451 | "unregister(fd) -> None\n\n\
|
---|
| 452 | Remove a file descriptor being tracked by the polling object.");
|
---|
| 453 |
|
---|
| 454 | static PyObject *
|
---|
[391] | 455 | poll_unregister(pollObject *self, PyObject *o)
|
---|
[2] | 456 | {
|
---|
[391] | 457 | PyObject *key;
|
---|
| 458 | int fd;
|
---|
[2] | 459 |
|
---|
[391] | 460 | fd = PyObject_AsFileDescriptor( o );
|
---|
| 461 | if (fd == -1)
|
---|
| 462 | return NULL;
|
---|
[2] | 463 |
|
---|
[391] | 464 | /* Check whether the fd is already in the array */
|
---|
| 465 | key = PyInt_FromLong(fd);
|
---|
| 466 | if (key == NULL)
|
---|
| 467 | return NULL;
|
---|
[2] | 468 |
|
---|
[391] | 469 | if (PyDict_DelItem(self->dict, key) == -1) {
|
---|
| 470 | Py_DECREF(key);
|
---|
| 471 | /* This will simply raise the KeyError set by PyDict_DelItem
|
---|
| 472 | if the file descriptor isn't registered. */
|
---|
| 473 | return NULL;
|
---|
| 474 | }
|
---|
[2] | 475 |
|
---|
[391] | 476 | Py_DECREF(key);
|
---|
| 477 | self->ufd_uptodate = 0;
|
---|
[2] | 478 |
|
---|
[391] | 479 | Py_INCREF(Py_None);
|
---|
| 480 | return Py_None;
|
---|
[2] | 481 | }
|
---|
| 482 |
|
---|
| 483 | PyDoc_STRVAR(poll_poll_doc,
|
---|
| 484 | "poll( [timeout] ) -> list of (fd, event) 2-tuples\n\n\
|
---|
| 485 | Polls the set of registered file descriptors, returning a list containing \n\
|
---|
| 486 | any descriptors that have events or errors to report.");
|
---|
| 487 |
|
---|
| 488 | static PyObject *
|
---|
[391] | 489 | poll_poll(pollObject *self, PyObject *args)
|
---|
[2] | 490 | {
|
---|
[391] | 491 | PyObject *result_list = NULL, *tout = NULL;
|
---|
| 492 | int timeout = 0, poll_result, i, j;
|
---|
| 493 | PyObject *value = NULL, *num = NULL;
|
---|
[2] | 494 |
|
---|
[391] | 495 | if (!PyArg_UnpackTuple(args, "poll", 0, 1, &tout)) {
|
---|
| 496 | return NULL;
|
---|
| 497 | }
|
---|
[2] | 498 |
|
---|
[391] | 499 | /* Check values for timeout */
|
---|
| 500 | if (tout == NULL || tout == Py_None)
|
---|
| 501 | timeout = -1;
|
---|
| 502 | else if (!PyNumber_Check(tout)) {
|
---|
| 503 | PyErr_SetString(PyExc_TypeError,
|
---|
| 504 | "timeout must be an integer or None");
|
---|
| 505 | return NULL;
|
---|
| 506 | }
|
---|
| 507 | else {
|
---|
| 508 | tout = PyNumber_Int(tout);
|
---|
| 509 | if (!tout)
|
---|
| 510 | return NULL;
|
---|
| 511 | timeout = _PyInt_AsInt(tout);
|
---|
| 512 | Py_DECREF(tout);
|
---|
| 513 | if (timeout == -1 && PyErr_Occurred())
|
---|
| 514 | return NULL;
|
---|
| 515 | }
|
---|
[2] | 516 |
|
---|
[391] | 517 | /* Avoid concurrent poll() invocation, issue 8865 */
|
---|
| 518 | if (self->poll_running) {
|
---|
| 519 | PyErr_SetString(PyExc_RuntimeError,
|
---|
| 520 | "concurrent poll() invocation");
|
---|
| 521 | return NULL;
|
---|
| 522 | }
|
---|
[2] | 523 |
|
---|
[391] | 524 | /* Ensure the ufd array is up to date */
|
---|
| 525 | if (!self->ufd_uptodate)
|
---|
| 526 | if (update_ufd_array(self) == 0)
|
---|
| 527 | return NULL;
|
---|
[2] | 528 |
|
---|
[391] | 529 | self->poll_running = 1;
|
---|
[2] | 530 |
|
---|
[391] | 531 | /* call poll() */
|
---|
| 532 | Py_BEGIN_ALLOW_THREADS
|
---|
| 533 | poll_result = poll(self->ufds, self->ufd_len, timeout);
|
---|
| 534 | Py_END_ALLOW_THREADS
|
---|
| 535 |
|
---|
| 536 | self->poll_running = 0;
|
---|
| 537 |
|
---|
| 538 | if (poll_result < 0) {
|
---|
| 539 | PyErr_SetFromErrno(SelectError);
|
---|
| 540 | return NULL;
|
---|
| 541 | }
|
---|
| 542 |
|
---|
| 543 | /* build the result list */
|
---|
| 544 |
|
---|
| 545 | result_list = PyList_New(poll_result);
|
---|
| 546 | if (!result_list)
|
---|
| 547 | return NULL;
|
---|
| 548 | else {
|
---|
| 549 | for (i = 0, j = 0; j < poll_result; j++) {
|
---|
| 550 | /* skip to the next fired descriptor */
|
---|
| 551 | while (!self->ufds[i].revents) {
|
---|
| 552 | i++;
|
---|
| 553 | }
|
---|
| 554 | /* if we hit a NULL return, set value to NULL
|
---|
| 555 | and break out of loop; code at end will
|
---|
| 556 | clean up result_list */
|
---|
| 557 | value = PyTuple_New(2);
|
---|
| 558 | if (value == NULL)
|
---|
| 559 | goto error;
|
---|
| 560 | num = PyInt_FromLong(self->ufds[i].fd);
|
---|
| 561 | if (num == NULL) {
|
---|
| 562 | Py_DECREF(value);
|
---|
| 563 | goto error;
|
---|
| 564 | }
|
---|
| 565 | PyTuple_SET_ITEM(value, 0, num);
|
---|
| 566 |
|
---|
| 567 | /* The &0xffff is a workaround for AIX. 'revents'
|
---|
| 568 | is a 16-bit short, and IBM assigned POLLNVAL
|
---|
| 569 | to be 0x8000, so the conversion to int results
|
---|
| 570 | in a negative number. See SF bug #923315. */
|
---|
| 571 | num = PyInt_FromLong(self->ufds[i].revents & 0xffff);
|
---|
| 572 | if (num == NULL) {
|
---|
| 573 | Py_DECREF(value);
|
---|
| 574 | goto error;
|
---|
| 575 | }
|
---|
| 576 | PyTuple_SET_ITEM(value, 1, num);
|
---|
| 577 | if ((PyList_SetItem(result_list, j, value)) == -1) {
|
---|
| 578 | Py_DECREF(value);
|
---|
| 579 | goto error;
|
---|
| 580 | }
|
---|
| 581 | i++;
|
---|
| 582 | }
|
---|
| 583 | }
|
---|
| 584 | return result_list;
|
---|
| 585 |
|
---|
[2] | 586 | error:
|
---|
[391] | 587 | Py_DECREF(result_list);
|
---|
| 588 | return NULL;
|
---|
[2] | 589 | }
|
---|
| 590 |
|
---|
| 591 | static PyMethodDef poll_methods[] = {
|
---|
[391] | 592 | {"register", (PyCFunction)poll_register,
|
---|
| 593 | METH_VARARGS, poll_register_doc},
|
---|
| 594 | {"modify", (PyCFunction)poll_modify,
|
---|
| 595 | METH_VARARGS, poll_modify_doc},
|
---|
| 596 | {"unregister", (PyCFunction)poll_unregister,
|
---|
| 597 | METH_O, poll_unregister_doc},
|
---|
| 598 | {"poll", (PyCFunction)poll_poll,
|
---|
| 599 | METH_VARARGS, poll_poll_doc},
|
---|
| 600 | {NULL, NULL} /* sentinel */
|
---|
[2] | 601 | };
|
---|
| 602 |
|
---|
| 603 | static pollObject *
|
---|
| 604 | newPollObject(void)
|
---|
| 605 | {
|
---|
[391] | 606 | pollObject *self;
|
---|
| 607 | self = PyObject_New(pollObject, &poll_Type);
|
---|
| 608 | if (self == NULL)
|
---|
| 609 | return NULL;
|
---|
| 610 | /* ufd_uptodate is a Boolean, denoting whether the
|
---|
| 611 | array pointed to by ufds matches the contents of the dictionary. */
|
---|
| 612 | self->ufd_uptodate = 0;
|
---|
| 613 | self->ufds = NULL;
|
---|
| 614 | self->poll_running = 0;
|
---|
| 615 | self->dict = PyDict_New();
|
---|
| 616 | if (self->dict == NULL) {
|
---|
| 617 | Py_DECREF(self);
|
---|
| 618 | return NULL;
|
---|
| 619 | }
|
---|
| 620 | return self;
|
---|
[2] | 621 | }
|
---|
| 622 |
|
---|
| 623 | static void
|
---|
| 624 | poll_dealloc(pollObject *self)
|
---|
| 625 | {
|
---|
[391] | 626 | if (self->ufds != NULL)
|
---|
| 627 | PyMem_DEL(self->ufds);
|
---|
| 628 | Py_XDECREF(self->dict);
|
---|
| 629 | PyObject_Del(self);
|
---|
[2] | 630 | }
|
---|
| 631 |
|
---|
| 632 | static PyObject *
|
---|
| 633 | poll_getattr(pollObject *self, char *name)
|
---|
| 634 | {
|
---|
[391] | 635 | return Py_FindMethod(poll_methods, (PyObject *)self, name);
|
---|
[2] | 636 | }
|
---|
| 637 |
|
---|
| 638 | static PyTypeObject poll_Type = {
|
---|
[391] | 639 | /* The ob_type field must be initialized in the module init function
|
---|
| 640 | * to be portable to Windows without using C++. */
|
---|
| 641 | PyVarObject_HEAD_INIT(NULL, 0)
|
---|
| 642 | "select.poll", /*tp_name*/
|
---|
| 643 | sizeof(pollObject), /*tp_basicsize*/
|
---|
| 644 | 0, /*tp_itemsize*/
|
---|
| 645 | /* methods */
|
---|
| 646 | (destructor)poll_dealloc, /*tp_dealloc*/
|
---|
| 647 | 0, /*tp_print*/
|
---|
| 648 | (getattrfunc)poll_getattr, /*tp_getattr*/
|
---|
| 649 | 0, /*tp_setattr*/
|
---|
| 650 | 0, /*tp_compare*/
|
---|
| 651 | 0, /*tp_repr*/
|
---|
| 652 | 0, /*tp_as_number*/
|
---|
| 653 | 0, /*tp_as_sequence*/
|
---|
| 654 | 0, /*tp_as_mapping*/
|
---|
| 655 | 0, /*tp_hash*/
|
---|
[2] | 656 | };
|
---|
| 657 |
|
---|
| 658 | PyDoc_STRVAR(poll_doc,
|
---|
| 659 | "Returns a polling object, which supports registering and\n\
|
---|
| 660 | unregistering file descriptors, and then polling them for I/O events.");
|
---|
| 661 |
|
---|
| 662 | static PyObject *
|
---|
| 663 | select_poll(PyObject *self, PyObject *unused)
|
---|
| 664 | {
|
---|
[391] | 665 | return (PyObject *)newPollObject();
|
---|
[2] | 666 | }
|
---|
| 667 |
|
---|
| 668 | #ifdef __APPLE__
|
---|
[391] | 669 | /*
|
---|
[2] | 670 | * On some systems poll() sets errno on invalid file descriptors. We test
|
---|
| 671 | * for this at runtime because this bug may be fixed or introduced between
|
---|
| 672 | * OS releases.
|
---|
| 673 | */
|
---|
| 674 | static int select_have_broken_poll(void)
|
---|
| 675 | {
|
---|
[391] | 676 | int poll_test;
|
---|
| 677 | int filedes[2];
|
---|
[2] | 678 |
|
---|
[391] | 679 | struct pollfd poll_struct = { 0, POLLIN|POLLPRI|POLLOUT, 0 };
|
---|
[2] | 680 |
|
---|
[391] | 681 | /* Create a file descriptor to make invalid */
|
---|
| 682 | if (pipe(filedes) < 0) {
|
---|
| 683 | return 1;
|
---|
| 684 | }
|
---|
| 685 | poll_struct.fd = filedes[0];
|
---|
| 686 | close(filedes[0]);
|
---|
| 687 | close(filedes[1]);
|
---|
| 688 | poll_test = poll(&poll_struct, 1, 0);
|
---|
| 689 | if (poll_test < 0) {
|
---|
| 690 | return 1;
|
---|
| 691 | } else if (poll_test == 0 && poll_struct.revents != POLLNVAL) {
|
---|
| 692 | return 1;
|
---|
| 693 | }
|
---|
| 694 | return 0;
|
---|
[2] | 695 | }
|
---|
| 696 | #endif /* __APPLE__ */
|
---|
| 697 |
|
---|
| 698 | #endif /* HAVE_POLL */
|
---|
| 699 |
|
---|
| 700 | #ifdef HAVE_EPOLL
|
---|
| 701 | /* **************************************************************************
|
---|
| 702 | * epoll interface for Linux 2.6
|
---|
| 703 | *
|
---|
| 704 | * Written by Christian Heimes
|
---|
| 705 | * Inspired by Twisted's _epoll.pyx and select.poll()
|
---|
| 706 | */
|
---|
| 707 |
|
---|
| 708 | #ifdef HAVE_SYS_EPOLL_H
|
---|
| 709 | #include <sys/epoll.h>
|
---|
| 710 | #endif
|
---|
| 711 |
|
---|
| 712 | typedef struct {
|
---|
[391] | 713 | PyObject_HEAD
|
---|
| 714 | SOCKET epfd; /* epoll control file descriptor */
|
---|
[2] | 715 | } pyEpoll_Object;
|
---|
| 716 |
|
---|
| 717 | static PyTypeObject pyEpoll_Type;
|
---|
| 718 | #define pyepoll_CHECK(op) (PyObject_TypeCheck((op), &pyEpoll_Type))
|
---|
| 719 |
|
---|
| 720 | static PyObject *
|
---|
| 721 | pyepoll_err_closed(void)
|
---|
| 722 | {
|
---|
[391] | 723 | PyErr_SetString(PyExc_ValueError, "I/O operation on closed epoll fd");
|
---|
| 724 | return NULL;
|
---|
[2] | 725 | }
|
---|
| 726 |
|
---|
| 727 | static int
|
---|
| 728 | pyepoll_internal_close(pyEpoll_Object *self)
|
---|
| 729 | {
|
---|
[391] | 730 | int save_errno = 0;
|
---|
| 731 | if (self->epfd >= 0) {
|
---|
| 732 | int epfd = self->epfd;
|
---|
| 733 | self->epfd = -1;
|
---|
| 734 | Py_BEGIN_ALLOW_THREADS
|
---|
| 735 | if (close(epfd) < 0)
|
---|
| 736 | save_errno = errno;
|
---|
| 737 | Py_END_ALLOW_THREADS
|
---|
| 738 | }
|
---|
| 739 | return save_errno;
|
---|
[2] | 740 | }
|
---|
| 741 |
|
---|
| 742 | static PyObject *
|
---|
| 743 | newPyEpoll_Object(PyTypeObject *type, int sizehint, SOCKET fd)
|
---|
| 744 | {
|
---|
[391] | 745 | pyEpoll_Object *self;
|
---|
[2] | 746 |
|
---|
[391] | 747 | if (sizehint == -1) {
|
---|
| 748 | sizehint = FD_SETSIZE-1;
|
---|
| 749 | }
|
---|
| 750 | else if (sizehint < 1) {
|
---|
| 751 | PyErr_Format(PyExc_ValueError,
|
---|
| 752 | "sizehint must be greater zero, got %d",
|
---|
| 753 | sizehint);
|
---|
| 754 | return NULL;
|
---|
| 755 | }
|
---|
[2] | 756 |
|
---|
[391] | 757 | assert(type != NULL && type->tp_alloc != NULL);
|
---|
| 758 | self = (pyEpoll_Object *) type->tp_alloc(type, 0);
|
---|
| 759 | if (self == NULL)
|
---|
| 760 | return NULL;
|
---|
| 761 |
|
---|
| 762 | if (fd == -1) {
|
---|
| 763 | Py_BEGIN_ALLOW_THREADS
|
---|
| 764 | self->epfd = epoll_create(sizehint);
|
---|
| 765 | Py_END_ALLOW_THREADS
|
---|
| 766 | }
|
---|
| 767 | else {
|
---|
| 768 | self->epfd = fd;
|
---|
| 769 | }
|
---|
| 770 | if (self->epfd < 0) {
|
---|
| 771 | Py_DECREF(self);
|
---|
| 772 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 773 | return NULL;
|
---|
| 774 | }
|
---|
| 775 | return (PyObject *)self;
|
---|
[2] | 776 | }
|
---|
| 777 |
|
---|
| 778 |
|
---|
| 779 | static PyObject *
|
---|
| 780 | pyepoll_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
---|
| 781 | {
|
---|
[391] | 782 | int sizehint = -1;
|
---|
| 783 | static char *kwlist[] = {"sizehint", NULL};
|
---|
[2] | 784 |
|
---|
[391] | 785 | if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:epoll", kwlist,
|
---|
| 786 | &sizehint))
|
---|
| 787 | return NULL;
|
---|
[2] | 788 |
|
---|
[391] | 789 | return newPyEpoll_Object(type, sizehint, -1);
|
---|
[2] | 790 | }
|
---|
| 791 |
|
---|
| 792 |
|
---|
| 793 | static void
|
---|
| 794 | pyepoll_dealloc(pyEpoll_Object *self)
|
---|
| 795 | {
|
---|
[391] | 796 | (void)pyepoll_internal_close(self);
|
---|
| 797 | Py_TYPE(self)->tp_free(self);
|
---|
[2] | 798 | }
|
---|
| 799 |
|
---|
| 800 | static PyObject*
|
---|
| 801 | pyepoll_close(pyEpoll_Object *self)
|
---|
| 802 | {
|
---|
[391] | 803 | errno = pyepoll_internal_close(self);
|
---|
| 804 | if (errno < 0) {
|
---|
| 805 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 806 | return NULL;
|
---|
| 807 | }
|
---|
| 808 | Py_RETURN_NONE;
|
---|
[2] | 809 | }
|
---|
| 810 |
|
---|
| 811 | PyDoc_STRVAR(pyepoll_close_doc,
|
---|
| 812 | "close() -> None\n\
|
---|
| 813 | \n\
|
---|
| 814 | Close the epoll control file descriptor. Further operations on the epoll\n\
|
---|
| 815 | object will raise an exception.");
|
---|
| 816 |
|
---|
| 817 | static PyObject*
|
---|
| 818 | pyepoll_get_closed(pyEpoll_Object *self)
|
---|
| 819 | {
|
---|
[391] | 820 | if (self->epfd < 0)
|
---|
| 821 | Py_RETURN_TRUE;
|
---|
| 822 | else
|
---|
| 823 | Py_RETURN_FALSE;
|
---|
[2] | 824 | }
|
---|
| 825 |
|
---|
| 826 | static PyObject*
|
---|
| 827 | pyepoll_fileno(pyEpoll_Object *self)
|
---|
| 828 | {
|
---|
[391] | 829 | if (self->epfd < 0)
|
---|
| 830 | return pyepoll_err_closed();
|
---|
| 831 | return PyInt_FromLong(self->epfd);
|
---|
[2] | 832 | }
|
---|
| 833 |
|
---|
| 834 | PyDoc_STRVAR(pyepoll_fileno_doc,
|
---|
| 835 | "fileno() -> int\n\
|
---|
| 836 | \n\
|
---|
| 837 | Return the epoll control file descriptor.");
|
---|
| 838 |
|
---|
| 839 | static PyObject*
|
---|
| 840 | pyepoll_fromfd(PyObject *cls, PyObject *args)
|
---|
| 841 | {
|
---|
[391] | 842 | SOCKET fd;
|
---|
[2] | 843 |
|
---|
[391] | 844 | if (!PyArg_ParseTuple(args, "i:fromfd", &fd))
|
---|
| 845 | return NULL;
|
---|
[2] | 846 |
|
---|
[391] | 847 | return newPyEpoll_Object((PyTypeObject*)cls, -1, fd);
|
---|
[2] | 848 | }
|
---|
| 849 |
|
---|
| 850 | PyDoc_STRVAR(pyepoll_fromfd_doc,
|
---|
| 851 | "fromfd(fd) -> epoll\n\
|
---|
| 852 | \n\
|
---|
| 853 | Create an epoll object from a given control fd.");
|
---|
| 854 |
|
---|
| 855 | static PyObject *
|
---|
| 856 | pyepoll_internal_ctl(int epfd, int op, PyObject *pfd, unsigned int events)
|
---|
| 857 | {
|
---|
[391] | 858 | struct epoll_event ev;
|
---|
| 859 | int result;
|
---|
| 860 | int fd;
|
---|
[2] | 861 |
|
---|
[391] | 862 | if (epfd < 0)
|
---|
| 863 | return pyepoll_err_closed();
|
---|
[2] | 864 |
|
---|
[391] | 865 | fd = PyObject_AsFileDescriptor(pfd);
|
---|
| 866 | if (fd == -1) {
|
---|
| 867 | return NULL;
|
---|
| 868 | }
|
---|
[2] | 869 |
|
---|
[391] | 870 | switch(op) {
|
---|
| 871 | case EPOLL_CTL_ADD:
|
---|
| 872 | case EPOLL_CTL_MOD:
|
---|
| 873 | ev.events = events;
|
---|
| 874 | ev.data.fd = fd;
|
---|
| 875 | Py_BEGIN_ALLOW_THREADS
|
---|
| 876 | result = epoll_ctl(epfd, op, fd, &ev);
|
---|
| 877 | Py_END_ALLOW_THREADS
|
---|
| 878 | break;
|
---|
| 879 | case EPOLL_CTL_DEL:
|
---|
| 880 | /* In kernel versions before 2.6.9, the EPOLL_CTL_DEL
|
---|
| 881 | * operation required a non-NULL pointer in event, even
|
---|
| 882 | * though this argument is ignored. */
|
---|
| 883 | Py_BEGIN_ALLOW_THREADS
|
---|
| 884 | result = epoll_ctl(epfd, op, fd, &ev);
|
---|
| 885 | if (errno == EBADF) {
|
---|
| 886 | /* fd already closed */
|
---|
| 887 | result = 0;
|
---|
| 888 | errno = 0;
|
---|
| 889 | }
|
---|
| 890 | Py_END_ALLOW_THREADS
|
---|
| 891 | break;
|
---|
| 892 | default:
|
---|
| 893 | result = -1;
|
---|
| 894 | errno = EINVAL;
|
---|
| 895 | }
|
---|
[2] | 896 |
|
---|
[391] | 897 | if (result < 0) {
|
---|
| 898 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 899 | return NULL;
|
---|
| 900 | }
|
---|
| 901 | Py_RETURN_NONE;
|
---|
[2] | 902 | }
|
---|
| 903 |
|
---|
| 904 | static PyObject *
|
---|
| 905 | pyepoll_register(pyEpoll_Object *self, PyObject *args, PyObject *kwds)
|
---|
| 906 | {
|
---|
[391] | 907 | PyObject *pfd;
|
---|
| 908 | unsigned int events = EPOLLIN | EPOLLOUT | EPOLLPRI;
|
---|
| 909 | static char *kwlist[] = {"fd", "eventmask", NULL};
|
---|
[2] | 910 |
|
---|
[391] | 911 | if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|I:register", kwlist,
|
---|
| 912 | &pfd, &events)) {
|
---|
| 913 | return NULL;
|
---|
| 914 | }
|
---|
[2] | 915 |
|
---|
[391] | 916 | return pyepoll_internal_ctl(self->epfd, EPOLL_CTL_ADD, pfd, events);
|
---|
[2] | 917 | }
|
---|
| 918 |
|
---|
| 919 | PyDoc_STRVAR(pyepoll_register_doc,
|
---|
[391] | 920 | "register(fd[, eventmask]) -> None\n\
|
---|
[2] | 921 | \n\
|
---|
[391] | 922 | Registers a new fd or raises an IOError if the fd is already registered.\n\
|
---|
[2] | 923 | fd is the target file descriptor of the operation.\n\
|
---|
| 924 | events is a bit set composed of the various EPOLL constants; the default\n\
|
---|
| 925 | is EPOLL_IN | EPOLL_OUT | EPOLL_PRI.\n\
|
---|
| 926 | \n\
|
---|
| 927 | The epoll interface supports all file descriptors that support poll.");
|
---|
| 928 |
|
---|
| 929 | static PyObject *
|
---|
| 930 | pyepoll_modify(pyEpoll_Object *self, PyObject *args, PyObject *kwds)
|
---|
| 931 | {
|
---|
[391] | 932 | PyObject *pfd;
|
---|
| 933 | unsigned int events;
|
---|
| 934 | static char *kwlist[] = {"fd", "eventmask", NULL};
|
---|
[2] | 935 |
|
---|
[391] | 936 | if (!PyArg_ParseTupleAndKeywords(args, kwds, "OI:modify", kwlist,
|
---|
| 937 | &pfd, &events)) {
|
---|
| 938 | return NULL;
|
---|
| 939 | }
|
---|
[2] | 940 |
|
---|
[391] | 941 | return pyepoll_internal_ctl(self->epfd, EPOLL_CTL_MOD, pfd, events);
|
---|
[2] | 942 | }
|
---|
| 943 |
|
---|
| 944 | PyDoc_STRVAR(pyepoll_modify_doc,
|
---|
| 945 | "modify(fd, eventmask) -> None\n\
|
---|
| 946 | \n\
|
---|
| 947 | fd is the target file descriptor of the operation\n\
|
---|
| 948 | events is a bit set composed of the various EPOLL constants");
|
---|
| 949 |
|
---|
| 950 | static PyObject *
|
---|
| 951 | pyepoll_unregister(pyEpoll_Object *self, PyObject *args, PyObject *kwds)
|
---|
| 952 | {
|
---|
[391] | 953 | PyObject *pfd;
|
---|
| 954 | static char *kwlist[] = {"fd", NULL};
|
---|
[2] | 955 |
|
---|
[391] | 956 | if (!PyArg_ParseTupleAndKeywords(args, kwds, "O:unregister", kwlist,
|
---|
| 957 | &pfd)) {
|
---|
| 958 | return NULL;
|
---|
| 959 | }
|
---|
[2] | 960 |
|
---|
[391] | 961 | return pyepoll_internal_ctl(self->epfd, EPOLL_CTL_DEL, pfd, 0);
|
---|
[2] | 962 | }
|
---|
| 963 |
|
---|
| 964 | PyDoc_STRVAR(pyepoll_unregister_doc,
|
---|
| 965 | "unregister(fd) -> None\n\
|
---|
| 966 | \n\
|
---|
| 967 | fd is the target file descriptor of the operation.");
|
---|
| 968 |
|
---|
| 969 | static PyObject *
|
---|
| 970 | pyepoll_poll(pyEpoll_Object *self, PyObject *args, PyObject *kwds)
|
---|
| 971 | {
|
---|
[391] | 972 | double dtimeout = -1.;
|
---|
| 973 | int timeout;
|
---|
| 974 | int maxevents = -1;
|
---|
| 975 | int nfds, i;
|
---|
| 976 | PyObject *elist = NULL, *etuple = NULL;
|
---|
| 977 | struct epoll_event *evs = NULL;
|
---|
| 978 | static char *kwlist[] = {"timeout", "maxevents", NULL};
|
---|
[2] | 979 |
|
---|
[391] | 980 | if (self->epfd < 0)
|
---|
| 981 | return pyepoll_err_closed();
|
---|
[2] | 982 |
|
---|
[391] | 983 | if (!PyArg_ParseTupleAndKeywords(args, kwds, "|di:poll", kwlist,
|
---|
| 984 | &dtimeout, &maxevents)) {
|
---|
| 985 | return NULL;
|
---|
| 986 | }
|
---|
[2] | 987 |
|
---|
[391] | 988 | if (dtimeout < 0) {
|
---|
| 989 | timeout = -1;
|
---|
| 990 | }
|
---|
| 991 | else if (dtimeout * 1000.0 > INT_MAX) {
|
---|
| 992 | PyErr_SetString(PyExc_OverflowError,
|
---|
| 993 | "timeout is too large");
|
---|
| 994 | return NULL;
|
---|
| 995 | }
|
---|
| 996 | else {
|
---|
| 997 | timeout = (int)(dtimeout * 1000.0);
|
---|
| 998 | }
|
---|
[2] | 999 |
|
---|
[391] | 1000 | if (maxevents == -1) {
|
---|
| 1001 | maxevents = FD_SETSIZE-1;
|
---|
| 1002 | }
|
---|
| 1003 | else if (maxevents < 1) {
|
---|
| 1004 | PyErr_Format(PyExc_ValueError,
|
---|
| 1005 | "maxevents must be greater than 0, got %d",
|
---|
| 1006 | maxevents);
|
---|
| 1007 | return NULL;
|
---|
| 1008 | }
|
---|
[2] | 1009 |
|
---|
[391] | 1010 | evs = PyMem_New(struct epoll_event, maxevents);
|
---|
| 1011 | if (evs == NULL) {
|
---|
| 1012 | Py_DECREF(self);
|
---|
| 1013 | PyErr_NoMemory();
|
---|
| 1014 | return NULL;
|
---|
| 1015 | }
|
---|
[2] | 1016 |
|
---|
[391] | 1017 | Py_BEGIN_ALLOW_THREADS
|
---|
| 1018 | nfds = epoll_wait(self->epfd, evs, maxevents, timeout);
|
---|
| 1019 | Py_END_ALLOW_THREADS
|
---|
| 1020 | if (nfds < 0) {
|
---|
| 1021 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 1022 | goto error;
|
---|
| 1023 | }
|
---|
[2] | 1024 |
|
---|
[391] | 1025 | elist = PyList_New(nfds);
|
---|
| 1026 | if (elist == NULL) {
|
---|
| 1027 | goto error;
|
---|
| 1028 | }
|
---|
[2] | 1029 |
|
---|
[391] | 1030 | for (i = 0; i < nfds; i++) {
|
---|
| 1031 | etuple = Py_BuildValue("iI", evs[i].data.fd, evs[i].events);
|
---|
| 1032 | if (etuple == NULL) {
|
---|
| 1033 | Py_CLEAR(elist);
|
---|
| 1034 | goto error;
|
---|
| 1035 | }
|
---|
| 1036 | PyList_SET_ITEM(elist, i, etuple);
|
---|
| 1037 | }
|
---|
[2] | 1038 |
|
---|
| 1039 | error:
|
---|
[391] | 1040 | PyMem_Free(evs);
|
---|
| 1041 | return elist;
|
---|
[2] | 1042 | }
|
---|
| 1043 |
|
---|
| 1044 | PyDoc_STRVAR(pyepoll_poll_doc,
|
---|
| 1045 | "poll([timeout=-1[, maxevents=-1]]) -> [(fd, events), (...)]\n\
|
---|
| 1046 | \n\
|
---|
| 1047 | Wait for events on the epoll file descriptor for a maximum time of timeout\n\
|
---|
| 1048 | in seconds (as float). -1 makes poll wait indefinitely.\n\
|
---|
| 1049 | Up to maxevents are returned to the caller.");
|
---|
| 1050 |
|
---|
| 1051 | static PyMethodDef pyepoll_methods[] = {
|
---|
[391] | 1052 | {"fromfd", (PyCFunction)pyepoll_fromfd,
|
---|
| 1053 | METH_VARARGS | METH_CLASS, pyepoll_fromfd_doc},
|
---|
| 1054 | {"close", (PyCFunction)pyepoll_close, METH_NOARGS,
|
---|
| 1055 | pyepoll_close_doc},
|
---|
| 1056 | {"fileno", (PyCFunction)pyepoll_fileno, METH_NOARGS,
|
---|
| 1057 | pyepoll_fileno_doc},
|
---|
| 1058 | {"modify", (PyCFunction)pyepoll_modify,
|
---|
| 1059 | METH_VARARGS | METH_KEYWORDS, pyepoll_modify_doc},
|
---|
| 1060 | {"register", (PyCFunction)pyepoll_register,
|
---|
| 1061 | METH_VARARGS | METH_KEYWORDS, pyepoll_register_doc},
|
---|
| 1062 | {"unregister", (PyCFunction)pyepoll_unregister,
|
---|
| 1063 | METH_VARARGS | METH_KEYWORDS, pyepoll_unregister_doc},
|
---|
| 1064 | {"poll", (PyCFunction)pyepoll_poll,
|
---|
| 1065 | METH_VARARGS | METH_KEYWORDS, pyepoll_poll_doc},
|
---|
| 1066 | {NULL, NULL},
|
---|
[2] | 1067 | };
|
---|
| 1068 |
|
---|
| 1069 | static PyGetSetDef pyepoll_getsetlist[] = {
|
---|
[391] | 1070 | {"closed", (getter)pyepoll_get_closed, NULL,
|
---|
| 1071 | "True if the epoll handler is closed"},
|
---|
| 1072 | {0},
|
---|
[2] | 1073 | };
|
---|
| 1074 |
|
---|
| 1075 | PyDoc_STRVAR(pyepoll_doc,
|
---|
| 1076 | "select.epoll([sizehint=-1])\n\
|
---|
| 1077 | \n\
|
---|
| 1078 | Returns an epolling object\n\
|
---|
| 1079 | \n\
|
---|
| 1080 | sizehint must be a positive integer or -1 for the default size. The\n\
|
---|
| 1081 | sizehint is used to optimize internal data structures. It doesn't limit\n\
|
---|
| 1082 | the maximum number of monitored events.");
|
---|
| 1083 |
|
---|
| 1084 | static PyTypeObject pyEpoll_Type = {
|
---|
[391] | 1085 | PyVarObject_HEAD_INIT(NULL, 0)
|
---|
| 1086 | "select.epoll", /* tp_name */
|
---|
| 1087 | sizeof(pyEpoll_Object), /* tp_basicsize */
|
---|
| 1088 | 0, /* tp_itemsize */
|
---|
| 1089 | (destructor)pyepoll_dealloc, /* tp_dealloc */
|
---|
| 1090 | 0, /* tp_print */
|
---|
| 1091 | 0, /* tp_getattr */
|
---|
| 1092 | 0, /* tp_setattr */
|
---|
| 1093 | 0, /* tp_compare */
|
---|
| 1094 | 0, /* tp_repr */
|
---|
| 1095 | 0, /* tp_as_number */
|
---|
| 1096 | 0, /* tp_as_sequence */
|
---|
| 1097 | 0, /* tp_as_mapping */
|
---|
| 1098 | 0, /* tp_hash */
|
---|
| 1099 | 0, /* tp_call */
|
---|
| 1100 | 0, /* tp_str */
|
---|
| 1101 | PyObject_GenericGetAttr, /* tp_getattro */
|
---|
| 1102 | 0, /* tp_setattro */
|
---|
| 1103 | 0, /* tp_as_buffer */
|
---|
| 1104 | Py_TPFLAGS_DEFAULT, /* tp_flags */
|
---|
| 1105 | pyepoll_doc, /* tp_doc */
|
---|
| 1106 | 0, /* tp_traverse */
|
---|
| 1107 | 0, /* tp_clear */
|
---|
| 1108 | 0, /* tp_richcompare */
|
---|
| 1109 | 0, /* tp_weaklistoffset */
|
---|
| 1110 | 0, /* tp_iter */
|
---|
| 1111 | 0, /* tp_iternext */
|
---|
| 1112 | pyepoll_methods, /* tp_methods */
|
---|
| 1113 | 0, /* tp_members */
|
---|
| 1114 | pyepoll_getsetlist, /* tp_getset */
|
---|
| 1115 | 0, /* tp_base */
|
---|
| 1116 | 0, /* tp_dict */
|
---|
| 1117 | 0, /* tp_descr_get */
|
---|
| 1118 | 0, /* tp_descr_set */
|
---|
| 1119 | 0, /* tp_dictoffset */
|
---|
| 1120 | 0, /* tp_init */
|
---|
| 1121 | 0, /* tp_alloc */
|
---|
| 1122 | pyepoll_new, /* tp_new */
|
---|
| 1123 | 0, /* tp_free */
|
---|
[2] | 1124 | };
|
---|
| 1125 |
|
---|
| 1126 | #endif /* HAVE_EPOLL */
|
---|
| 1127 |
|
---|
| 1128 | #ifdef HAVE_KQUEUE
|
---|
| 1129 | /* **************************************************************************
|
---|
| 1130 | * kqueue interface for BSD
|
---|
| 1131 | *
|
---|
| 1132 | * Copyright (c) 2000 Doug White, 2006 James Knight, 2007 Christian Heimes
|
---|
| 1133 | * All rights reserved.
|
---|
| 1134 | *
|
---|
| 1135 | * Redistribution and use in source and binary forms, with or without
|
---|
| 1136 | * modification, are permitted provided that the following conditions
|
---|
| 1137 | * are met:
|
---|
| 1138 | * 1. Redistributions of source code must retain the above copyright
|
---|
| 1139 | * notice, this list of conditions and the following disclaimer.
|
---|
| 1140 | * 2. Redistributions in binary form must reproduce the above copyright
|
---|
| 1141 | * notice, this list of conditions and the following disclaimer in the
|
---|
| 1142 | * documentation and/or other materials provided with the distribution.
|
---|
| 1143 | *
|
---|
| 1144 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
---|
| 1145 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
---|
| 1146 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
---|
| 1147 | * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
---|
| 1148 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
---|
| 1149 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
---|
| 1150 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
---|
| 1151 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
---|
| 1152 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
---|
| 1153 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
---|
| 1154 | * SUCH DAMAGE.
|
---|
| 1155 | */
|
---|
| 1156 |
|
---|
| 1157 | #ifdef HAVE_SYS_EVENT_H
|
---|
| 1158 | #include <sys/event.h>
|
---|
| 1159 | #endif
|
---|
| 1160 |
|
---|
| 1161 | PyDoc_STRVAR(kqueue_event_doc,
|
---|
[391] | 1162 | "kevent(ident, filter=KQ_FILTER_READ, flags=KQ_EV_ADD, fflags=0, data=0, udata=0)\n\
|
---|
[2] | 1163 | \n\
|
---|
| 1164 | This object is the equivalent of the struct kevent for the C API.\n\
|
---|
| 1165 | \n\
|
---|
| 1166 | See the kqueue manpage for more detailed information about the meaning\n\
|
---|
| 1167 | of the arguments.\n\
|
---|
| 1168 | \n\
|
---|
| 1169 | One minor note: while you might hope that udata could store a\n\
|
---|
| 1170 | reference to a python object, it cannot, because it is impossible to\n\
|
---|
| 1171 | keep a proper reference count of the object once it's passed into the\n\
|
---|
| 1172 | kernel. Therefore, I have restricted it to only storing an integer. I\n\
|
---|
| 1173 | recommend ignoring it and simply using the 'ident' field to key off\n\
|
---|
| 1174 | of. You could also set up a dictionary on the python side to store a\n\
|
---|
| 1175 | udata->object mapping.");
|
---|
| 1176 |
|
---|
| 1177 | typedef struct {
|
---|
[391] | 1178 | PyObject_HEAD
|
---|
| 1179 | struct kevent e;
|
---|
[2] | 1180 | } kqueue_event_Object;
|
---|
| 1181 |
|
---|
| 1182 | static PyTypeObject kqueue_event_Type;
|
---|
| 1183 |
|
---|
| 1184 | #define kqueue_event_Check(op) (PyObject_TypeCheck((op), &kqueue_event_Type))
|
---|
| 1185 |
|
---|
| 1186 | typedef struct {
|
---|
[391] | 1187 | PyObject_HEAD
|
---|
| 1188 | SOCKET kqfd; /* kqueue control fd */
|
---|
[2] | 1189 | } kqueue_queue_Object;
|
---|
| 1190 |
|
---|
| 1191 | static PyTypeObject kqueue_queue_Type;
|
---|
| 1192 |
|
---|
| 1193 | #define kqueue_queue_Check(op) (PyObject_TypeCheck((op), &kqueue_queue_Type))
|
---|
| 1194 |
|
---|
[391] | 1195 | #if (SIZEOF_UINTPTR_T != SIZEOF_VOID_P)
|
---|
| 1196 | # error uintptr_t does not match void *!
|
---|
| 1197 | #elif (SIZEOF_UINTPTR_T == SIZEOF_LONG_LONG)
|
---|
| 1198 | # define T_UINTPTRT T_ULONGLONG
|
---|
| 1199 | # define T_INTPTRT T_LONGLONG
|
---|
| 1200 | # define PyLong_AsUintptr_t PyLong_AsUnsignedLongLong
|
---|
| 1201 | # define UINTPTRT_FMT_UNIT "K"
|
---|
| 1202 | # define INTPTRT_FMT_UNIT "L"
|
---|
| 1203 | #elif (SIZEOF_UINTPTR_T == SIZEOF_LONG)
|
---|
| 1204 | # define T_UINTPTRT T_ULONG
|
---|
| 1205 | # define T_INTPTRT T_LONG
|
---|
| 1206 | # define PyLong_AsUintptr_t PyLong_AsUnsignedLong
|
---|
| 1207 | # define UINTPTRT_FMT_UNIT "k"
|
---|
| 1208 | # define INTPTRT_FMT_UNIT "l"
|
---|
| 1209 | #elif (SIZEOF_UINTPTR_T == SIZEOF_INT)
|
---|
| 1210 | # define T_UINTPTRT T_UINT
|
---|
| 1211 | # define T_INTPTRT T_INT
|
---|
| 1212 | # define PyLong_AsUintptr_t PyLong_AsUnsignedLong
|
---|
| 1213 | # define UINTPTRT_FMT_UNIT "I"
|
---|
| 1214 | # define INTPTRT_FMT_UNIT "i"
|
---|
| 1215 | #else
|
---|
| 1216 | # error uintptr_t does not match int, long, or long long!
|
---|
| 1217 | #endif
|
---|
| 1218 |
|
---|
| 1219 | /*
|
---|
| 1220 | * kevent is not standard and its members vary across BSDs.
|
---|
| 1221 | */
|
---|
| 1222 | #if !defined(__OpenBSD__)
|
---|
| 1223 | # define IDENT_TYPE T_UINTPTRT
|
---|
| 1224 | # define IDENT_CAST Py_intptr_t
|
---|
| 1225 | # define DATA_TYPE T_INTPTRT
|
---|
| 1226 | # define DATA_FMT_UNIT INTPTRT_FMT_UNIT
|
---|
| 1227 | # define IDENT_AsType PyLong_AsUintptr_t
|
---|
| 1228 | #else
|
---|
| 1229 | # define IDENT_TYPE T_UINT
|
---|
| 1230 | # define IDENT_CAST int
|
---|
| 1231 | # define DATA_TYPE T_INT
|
---|
| 1232 | # define DATA_FMT_UNIT "i"
|
---|
| 1233 | # define IDENT_AsType PyLong_AsUnsignedLong
|
---|
| 1234 | #endif
|
---|
| 1235 |
|
---|
[2] | 1236 | /* Unfortunately, we can't store python objects in udata, because
|
---|
| 1237 | * kevents in the kernel can be removed without warning, which would
|
---|
| 1238 | * forever lose the refcount on the object stored with it.
|
---|
| 1239 | */
|
---|
| 1240 |
|
---|
| 1241 | #define KQ_OFF(x) offsetof(kqueue_event_Object, x)
|
---|
| 1242 | static struct PyMemberDef kqueue_event_members[] = {
|
---|
[391] | 1243 | {"ident", IDENT_TYPE, KQ_OFF(e.ident)},
|
---|
| 1244 | {"filter", T_SHORT, KQ_OFF(e.filter)},
|
---|
| 1245 | {"flags", T_USHORT, KQ_OFF(e.flags)},
|
---|
| 1246 | {"fflags", T_UINT, KQ_OFF(e.fflags)},
|
---|
| 1247 | {"data", DATA_TYPE, KQ_OFF(e.data)},
|
---|
| 1248 | {"udata", T_UINTPTRT, KQ_OFF(e.udata)},
|
---|
| 1249 | {NULL} /* Sentinel */
|
---|
[2] | 1250 | };
|
---|
| 1251 | #undef KQ_OFF
|
---|
| 1252 |
|
---|
| 1253 | static PyObject *
|
---|
[391] | 1254 |
|
---|
[2] | 1255 | kqueue_event_repr(kqueue_event_Object *s)
|
---|
| 1256 | {
|
---|
[391] | 1257 | char buf[1024];
|
---|
| 1258 | PyOS_snprintf(
|
---|
| 1259 | buf, sizeof(buf),
|
---|
| 1260 | "<select.kevent ident=%zu filter=%d flags=0x%x fflags=0x%x "
|
---|
| 1261 | "data=0x%zd udata=%p>",
|
---|
| 1262 | (size_t)(s->e.ident), s->e.filter, s->e.flags,
|
---|
| 1263 | s->e.fflags, (Py_ssize_t)(s->e.data), s->e.udata);
|
---|
| 1264 | return PyString_FromString(buf);
|
---|
[2] | 1265 | }
|
---|
| 1266 |
|
---|
| 1267 | static int
|
---|
| 1268 | kqueue_event_init(kqueue_event_Object *self, PyObject *args, PyObject *kwds)
|
---|
| 1269 | {
|
---|
[391] | 1270 | PyObject *pfd;
|
---|
| 1271 | static char *kwlist[] = {"ident", "filter", "flags", "fflags",
|
---|
| 1272 | "data", "udata", NULL};
|
---|
| 1273 | static char *fmt = "O|hHI" DATA_FMT_UNIT UINTPTRT_FMT_UNIT ":kevent";
|
---|
[2] | 1274 |
|
---|
[391] | 1275 | EV_SET(&(self->e), 0, EVFILT_READ, EV_ADD, 0, 0, 0); /* defaults */
|
---|
[2] | 1276 |
|
---|
[391] | 1277 | if (!PyArg_ParseTupleAndKeywords(args, kwds, fmt, kwlist,
|
---|
| 1278 | &pfd, &(self->e.filter), &(self->e.flags),
|
---|
| 1279 | &(self->e.fflags), &(self->e.data), &(self->e.udata))) {
|
---|
| 1280 | return -1;
|
---|
| 1281 | }
|
---|
| 1282 |
|
---|
| 1283 | if (PyLong_Check(pfd)
|
---|
| 1284 | #if IDENT_TYPE == T_UINT
|
---|
| 1285 | && PyLong_AsUnsignedLong(pfd) <= UINT_MAX
|
---|
| 1286 | #endif
|
---|
| 1287 | ) {
|
---|
| 1288 | self->e.ident = IDENT_AsType(pfd);
|
---|
| 1289 | }
|
---|
| 1290 | else {
|
---|
| 1291 | self->e.ident = PyObject_AsFileDescriptor(pfd);
|
---|
| 1292 | }
|
---|
| 1293 | if (PyErr_Occurred()) {
|
---|
| 1294 | return -1;
|
---|
| 1295 | }
|
---|
| 1296 | return 0;
|
---|
[2] | 1297 | }
|
---|
| 1298 |
|
---|
| 1299 | static PyObject *
|
---|
| 1300 | kqueue_event_richcompare(kqueue_event_Object *s, kqueue_event_Object *o,
|
---|
[391] | 1301 | int op)
|
---|
[2] | 1302 | {
|
---|
[391] | 1303 | Py_intptr_t result = 0;
|
---|
[2] | 1304 |
|
---|
[391] | 1305 | if (!kqueue_event_Check(o)) {
|
---|
| 1306 | if (op == Py_EQ || op == Py_NE) {
|
---|
| 1307 | PyObject *res = op == Py_EQ ? Py_False : Py_True;
|
---|
| 1308 | Py_INCREF(res);
|
---|
| 1309 | return res;
|
---|
| 1310 | }
|
---|
| 1311 | PyErr_Format(PyExc_TypeError,
|
---|
| 1312 | "can't compare %.200s to %.200s",
|
---|
| 1313 | Py_TYPE(s)->tp_name, Py_TYPE(o)->tp_name);
|
---|
| 1314 | return NULL;
|
---|
| 1315 | }
|
---|
| 1316 | if (((result = (IDENT_CAST)(s->e.ident - o->e.ident)) == 0) &&
|
---|
| 1317 | ((result = s->e.filter - o->e.filter) == 0) &&
|
---|
| 1318 | ((result = s->e.flags - o->e.flags) == 0) &&
|
---|
| 1319 | ((result = (int)(s->e.fflags - o->e.fflags)) == 0) &&
|
---|
| 1320 | ((result = s->e.data - o->e.data) == 0) &&
|
---|
| 1321 | ((result = s->e.udata - o->e.udata) == 0)
|
---|
| 1322 | ) {
|
---|
| 1323 | result = 0;
|
---|
| 1324 | }
|
---|
[2] | 1325 |
|
---|
[391] | 1326 | switch (op) {
|
---|
| 1327 | case Py_EQ:
|
---|
| 1328 | result = (result == 0);
|
---|
| 1329 | break;
|
---|
| 1330 | case Py_NE:
|
---|
| 1331 | result = (result != 0);
|
---|
| 1332 | break;
|
---|
| 1333 | case Py_LE:
|
---|
| 1334 | result = (result <= 0);
|
---|
| 1335 | break;
|
---|
| 1336 | case Py_GE:
|
---|
| 1337 | result = (result >= 0);
|
---|
| 1338 | break;
|
---|
| 1339 | case Py_LT:
|
---|
| 1340 | result = (result < 0);
|
---|
| 1341 | break;
|
---|
| 1342 | case Py_GT:
|
---|
| 1343 | result = (result > 0);
|
---|
| 1344 | break;
|
---|
| 1345 | }
|
---|
| 1346 | return PyBool_FromLong((long)result);
|
---|
[2] | 1347 | }
|
---|
| 1348 |
|
---|
| 1349 | static PyTypeObject kqueue_event_Type = {
|
---|
[391] | 1350 | PyVarObject_HEAD_INIT(NULL, 0)
|
---|
| 1351 | "select.kevent", /* tp_name */
|
---|
| 1352 | sizeof(kqueue_event_Object), /* tp_basicsize */
|
---|
| 1353 | 0, /* tp_itemsize */
|
---|
| 1354 | 0, /* tp_dealloc */
|
---|
| 1355 | 0, /* tp_print */
|
---|
| 1356 | 0, /* tp_getattr */
|
---|
| 1357 | 0, /* tp_setattr */
|
---|
| 1358 | 0, /* tp_compare */
|
---|
| 1359 | (reprfunc)kqueue_event_repr, /* tp_repr */
|
---|
| 1360 | 0, /* tp_as_number */
|
---|
| 1361 | 0, /* tp_as_sequence */
|
---|
| 1362 | 0, /* tp_as_mapping */
|
---|
| 1363 | 0, /* tp_hash */
|
---|
| 1364 | 0, /* tp_call */
|
---|
| 1365 | 0, /* tp_str */
|
---|
| 1366 | 0, /* tp_getattro */
|
---|
| 1367 | 0, /* tp_setattro */
|
---|
| 1368 | 0, /* tp_as_buffer */
|
---|
| 1369 | Py_TPFLAGS_DEFAULT, /* tp_flags */
|
---|
| 1370 | kqueue_event_doc, /* tp_doc */
|
---|
| 1371 | 0, /* tp_traverse */
|
---|
| 1372 | 0, /* tp_clear */
|
---|
| 1373 | (richcmpfunc)kqueue_event_richcompare, /* tp_richcompare */
|
---|
| 1374 | 0, /* tp_weaklistoffset */
|
---|
| 1375 | 0, /* tp_iter */
|
---|
| 1376 | 0, /* tp_iternext */
|
---|
| 1377 | 0, /* tp_methods */
|
---|
| 1378 | kqueue_event_members, /* tp_members */
|
---|
| 1379 | 0, /* tp_getset */
|
---|
| 1380 | 0, /* tp_base */
|
---|
| 1381 | 0, /* tp_dict */
|
---|
| 1382 | 0, /* tp_descr_get */
|
---|
| 1383 | 0, /* tp_descr_set */
|
---|
| 1384 | 0, /* tp_dictoffset */
|
---|
| 1385 | (initproc)kqueue_event_init, /* tp_init */
|
---|
| 1386 | 0, /* tp_alloc */
|
---|
| 1387 | 0, /* tp_new */
|
---|
| 1388 | 0, /* tp_free */
|
---|
[2] | 1389 | };
|
---|
| 1390 |
|
---|
| 1391 | static PyObject *
|
---|
| 1392 | kqueue_queue_err_closed(void)
|
---|
| 1393 | {
|
---|
[391] | 1394 | PyErr_SetString(PyExc_ValueError, "I/O operation on closed kqueue fd");
|
---|
| 1395 | return NULL;
|
---|
[2] | 1396 | }
|
---|
| 1397 |
|
---|
| 1398 | static int
|
---|
| 1399 | kqueue_queue_internal_close(kqueue_queue_Object *self)
|
---|
| 1400 | {
|
---|
[391] | 1401 | int save_errno = 0;
|
---|
| 1402 | if (self->kqfd >= 0) {
|
---|
| 1403 | int kqfd = self->kqfd;
|
---|
| 1404 | self->kqfd = -1;
|
---|
| 1405 | Py_BEGIN_ALLOW_THREADS
|
---|
| 1406 | if (close(kqfd) < 0)
|
---|
| 1407 | save_errno = errno;
|
---|
| 1408 | Py_END_ALLOW_THREADS
|
---|
| 1409 | }
|
---|
| 1410 | return save_errno;
|
---|
[2] | 1411 | }
|
---|
| 1412 |
|
---|
| 1413 | static PyObject *
|
---|
| 1414 | newKqueue_Object(PyTypeObject *type, SOCKET fd)
|
---|
| 1415 | {
|
---|
[391] | 1416 | kqueue_queue_Object *self;
|
---|
| 1417 | assert(type != NULL && type->tp_alloc != NULL);
|
---|
| 1418 | self = (kqueue_queue_Object *) type->tp_alloc(type, 0);
|
---|
| 1419 | if (self == NULL) {
|
---|
| 1420 | return NULL;
|
---|
| 1421 | }
|
---|
| 1422 |
|
---|
| 1423 | if (fd == -1) {
|
---|
| 1424 | Py_BEGIN_ALLOW_THREADS
|
---|
| 1425 | self->kqfd = kqueue();
|
---|
| 1426 | Py_END_ALLOW_THREADS
|
---|
| 1427 | }
|
---|
| 1428 | else {
|
---|
| 1429 | self->kqfd = fd;
|
---|
| 1430 | }
|
---|
| 1431 | if (self->kqfd < 0) {
|
---|
| 1432 | Py_DECREF(self);
|
---|
| 1433 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 1434 | return NULL;
|
---|
| 1435 | }
|
---|
| 1436 | return (PyObject *)self;
|
---|
[2] | 1437 | }
|
---|
| 1438 |
|
---|
| 1439 | static PyObject *
|
---|
| 1440 | kqueue_queue_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
---|
| 1441 | {
|
---|
| 1442 |
|
---|
[391] | 1443 | if ((args != NULL && PyObject_Size(args)) ||
|
---|
| 1444 | (kwds != NULL && PyObject_Size(kwds))) {
|
---|
| 1445 | PyErr_SetString(PyExc_ValueError,
|
---|
| 1446 | "select.kqueue doesn't accept arguments");
|
---|
| 1447 | return NULL;
|
---|
| 1448 | }
|
---|
[2] | 1449 |
|
---|
[391] | 1450 | return newKqueue_Object(type, -1);
|
---|
[2] | 1451 | }
|
---|
| 1452 |
|
---|
| 1453 | static void
|
---|
| 1454 | kqueue_queue_dealloc(kqueue_queue_Object *self)
|
---|
| 1455 | {
|
---|
[391] | 1456 | kqueue_queue_internal_close(self);
|
---|
| 1457 | Py_TYPE(self)->tp_free(self);
|
---|
[2] | 1458 | }
|
---|
| 1459 |
|
---|
| 1460 | static PyObject*
|
---|
| 1461 | kqueue_queue_close(kqueue_queue_Object *self)
|
---|
| 1462 | {
|
---|
[391] | 1463 | errno = kqueue_queue_internal_close(self);
|
---|
| 1464 | if (errno < 0) {
|
---|
| 1465 | PyErr_SetFromErrno(PyExc_IOError);
|
---|
| 1466 | return NULL;
|
---|
| 1467 | }
|
---|
| 1468 | Py_RETURN_NONE;
|
---|
[2] | 1469 | }
|
---|
| 1470 |
|
---|
| 1471 | PyDoc_STRVAR(kqueue_queue_close_doc,
|
---|
| 1472 | "close() -> None\n\
|
---|
| 1473 | \n\
|
---|
| 1474 | Close the kqueue control file descriptor. Further operations on the kqueue\n\
|
---|
| 1475 | object will raise an exception.");
|
---|
| 1476 |
|
---|
| 1477 | static PyObject*
|
---|
| 1478 | kqueue_queue_get_closed(kqueue_queue_Object *self)
|
---|
| 1479 | {
|
---|
[391] | 1480 | if (self->kqfd < 0)
|
---|
| 1481 | Py_RETURN_TRUE;
|
---|
| 1482 | else
|
---|
| 1483 | Py_RETURN_FALSE;
|
---|
[2] | 1484 | }
|
---|
| 1485 |
|
---|
| 1486 | static PyObject*
|
---|
| 1487 | kqueue_queue_fileno(kqueue_queue_Object *self)
|
---|
| 1488 | {
|
---|
[391] | 1489 | if (self->kqfd < 0)
|
---|
| 1490 | return kqueue_queue_err_closed();
|
---|
| 1491 | return PyInt_FromLong(self->kqfd);
|
---|
[2] | 1492 | }
|
---|
| 1493 |
|
---|
| 1494 | PyDoc_STRVAR(kqueue_queue_fileno_doc,
|
---|
| 1495 | "fileno() -> int\n\
|
---|
| 1496 | \n\
|
---|
| 1497 | Return the kqueue control file descriptor.");
|
---|
| 1498 |
|
---|
| 1499 | static PyObject*
|
---|
| 1500 | kqueue_queue_fromfd(PyObject *cls, PyObject *args)
|
---|
| 1501 | {
|
---|
[391] | 1502 | SOCKET fd;
|
---|
[2] | 1503 |
|
---|
[391] | 1504 | if (!PyArg_ParseTuple(args, "i:fromfd", &fd))
|
---|
| 1505 | return NULL;
|
---|
[2] | 1506 |
|
---|
[391] | 1507 | return newKqueue_Object((PyTypeObject*)cls, fd);
|
---|
[2] | 1508 | }
|
---|
| 1509 |
|
---|
| 1510 | PyDoc_STRVAR(kqueue_queue_fromfd_doc,
|
---|
| 1511 | "fromfd(fd) -> kqueue\n\
|
---|
| 1512 | \n\
|
---|
| 1513 | Create a kqueue object from a given control fd.");
|
---|
| 1514 |
|
---|
| 1515 | static PyObject *
|
---|
| 1516 | kqueue_queue_control(kqueue_queue_Object *self, PyObject *args)
|
---|
| 1517 | {
|
---|
[391] | 1518 | int nevents = 0;
|
---|
| 1519 | int gotevents = 0;
|
---|
| 1520 | int nchanges = 0;
|
---|
| 1521 | int i = 0;
|
---|
| 1522 | PyObject *otimeout = NULL;
|
---|
| 1523 | PyObject *ch = NULL;
|
---|
| 1524 | PyObject *it = NULL, *ei = NULL;
|
---|
| 1525 | PyObject *result = NULL;
|
---|
| 1526 | struct kevent *evl = NULL;
|
---|
| 1527 | struct kevent *chl = NULL;
|
---|
| 1528 | struct timespec timeoutspec;
|
---|
| 1529 | struct timespec *ptimeoutspec;
|
---|
[2] | 1530 |
|
---|
[391] | 1531 | if (self->kqfd < 0)
|
---|
| 1532 | return kqueue_queue_err_closed();
|
---|
[2] | 1533 |
|
---|
[391] | 1534 | if (!PyArg_ParseTuple(args, "Oi|O:control", &ch, &nevents, &otimeout))
|
---|
| 1535 | return NULL;
|
---|
[2] | 1536 |
|
---|
[391] | 1537 | if (nevents < 0) {
|
---|
| 1538 | PyErr_Format(PyExc_ValueError,
|
---|
| 1539 | "Length of eventlist must be 0 or positive, got %d",
|
---|
| 1540 | nevents);
|
---|
| 1541 | return NULL;
|
---|
| 1542 | }
|
---|
[2] | 1543 |
|
---|
[391] | 1544 | if (otimeout == Py_None || otimeout == NULL) {
|
---|
| 1545 | ptimeoutspec = NULL;
|
---|
| 1546 | }
|
---|
| 1547 | else if (PyNumber_Check(otimeout)) {
|
---|
| 1548 | double timeout;
|
---|
| 1549 | long seconds;
|
---|
[2] | 1550 |
|
---|
[391] | 1551 | timeout = PyFloat_AsDouble(otimeout);
|
---|
| 1552 | if (timeout == -1 && PyErr_Occurred())
|
---|
| 1553 | return NULL;
|
---|
| 1554 | if (timeout > (double)LONG_MAX) {
|
---|
| 1555 | PyErr_SetString(PyExc_OverflowError,
|
---|
| 1556 | "timeout period too long");
|
---|
| 1557 | return NULL;
|
---|
| 1558 | }
|
---|
| 1559 | if (timeout < 0) {
|
---|
| 1560 | PyErr_SetString(PyExc_ValueError,
|
---|
| 1561 | "timeout must be positive or None");
|
---|
| 1562 | return NULL;
|
---|
| 1563 | }
|
---|
[2] | 1564 |
|
---|
[391] | 1565 | seconds = (long)timeout;
|
---|
| 1566 | timeout = timeout - (double)seconds;
|
---|
| 1567 | timeoutspec.tv_sec = seconds;
|
---|
| 1568 | timeoutspec.tv_nsec = (long)(timeout * 1E9);
|
---|
| 1569 | ptimeoutspec = &timeoutspec;
|
---|
| 1570 | }
|
---|
| 1571 | else {
|
---|
| 1572 | PyErr_Format(PyExc_TypeError,
|
---|
| 1573 | "timeout argument must be an number "
|
---|
| 1574 | "or None, got %.200s",
|
---|
| 1575 | Py_TYPE(otimeout)->tp_name);
|
---|
| 1576 | return NULL;
|
---|
| 1577 | }
|
---|
[2] | 1578 |
|
---|
[391] | 1579 | if (ch != NULL && ch != Py_None) {
|
---|
| 1580 | it = PyObject_GetIter(ch);
|
---|
| 1581 | if (it == NULL) {
|
---|
| 1582 | PyErr_SetString(PyExc_TypeError,
|
---|
| 1583 | "changelist is not iterable");
|
---|
| 1584 | return NULL;
|
---|
| 1585 | }
|
---|
| 1586 | nchanges = PyObject_Size(ch);
|
---|
| 1587 | if (nchanges < 0) {
|
---|
| 1588 | goto error;
|
---|
| 1589 | }
|
---|
[2] | 1590 |
|
---|
[391] | 1591 | chl = PyMem_New(struct kevent, nchanges);
|
---|
| 1592 | if (chl == NULL) {
|
---|
| 1593 | PyErr_NoMemory();
|
---|
| 1594 | goto error;
|
---|
| 1595 | }
|
---|
| 1596 | i = 0;
|
---|
| 1597 | while ((ei = PyIter_Next(it)) != NULL) {
|
---|
| 1598 | if (!kqueue_event_Check(ei)) {
|
---|
| 1599 | Py_DECREF(ei);
|
---|
| 1600 | PyErr_SetString(PyExc_TypeError,
|
---|
| 1601 | "changelist must be an iterable of "
|
---|
| 1602 | "select.kevent objects");
|
---|
| 1603 | goto error;
|
---|
| 1604 | } else {
|
---|
| 1605 | chl[i++] = ((kqueue_event_Object *)ei)->e;
|
---|
| 1606 | }
|
---|
| 1607 | Py_DECREF(ei);
|
---|
| 1608 | }
|
---|
| 1609 | }
|
---|
| 1610 | Py_CLEAR(it);
|
---|
[2] | 1611 |
|
---|
[391] | 1612 | /* event list */
|
---|
| 1613 | if (nevents) {
|
---|
| 1614 | evl = PyMem_New(struct kevent, nevents);
|
---|
| 1615 | if (evl == NULL) {
|
---|
| 1616 | PyErr_NoMemory();
|
---|
| 1617 | goto error;
|
---|
| 1618 | }
|
---|
| 1619 | }
|
---|
[2] | 1620 |
|
---|
[391] | 1621 | Py_BEGIN_ALLOW_THREADS
|
---|
| 1622 | gotevents = kevent(self->kqfd, chl, nchanges,
|
---|
| 1623 | evl, nevents, ptimeoutspec);
|
---|
| 1624 | Py_END_ALLOW_THREADS
|
---|
[2] | 1625 |
|
---|
[391] | 1626 | if (gotevents == -1) {
|
---|
| 1627 | PyErr_SetFromErrno(PyExc_OSError);
|
---|
| 1628 | goto error;
|
---|
| 1629 | }
|
---|
[2] | 1630 |
|
---|
[391] | 1631 | result = PyList_New(gotevents);
|
---|
| 1632 | if (result == NULL) {
|
---|
| 1633 | goto error;
|
---|
| 1634 | }
|
---|
[2] | 1635 |
|
---|
[391] | 1636 | for (i = 0; i < gotevents; i++) {
|
---|
| 1637 | kqueue_event_Object *ch;
|
---|
[2] | 1638 |
|
---|
[391] | 1639 | ch = PyObject_New(kqueue_event_Object, &kqueue_event_Type);
|
---|
| 1640 | if (ch == NULL) {
|
---|
| 1641 | goto error;
|
---|
| 1642 | }
|
---|
| 1643 | ch->e = evl[i];
|
---|
| 1644 | PyList_SET_ITEM(result, i, (PyObject *)ch);
|
---|
| 1645 | }
|
---|
| 1646 | PyMem_Free(chl);
|
---|
| 1647 | PyMem_Free(evl);
|
---|
| 1648 | return result;
|
---|
[2] | 1649 |
|
---|
| 1650 | error:
|
---|
[391] | 1651 | PyMem_Free(chl);
|
---|
| 1652 | PyMem_Free(evl);
|
---|
| 1653 | Py_XDECREF(result);
|
---|
| 1654 | Py_XDECREF(it);
|
---|
| 1655 | return NULL;
|
---|
[2] | 1656 | }
|
---|
| 1657 |
|
---|
| 1658 | PyDoc_STRVAR(kqueue_queue_control_doc,
|
---|
| 1659 | "control(changelist, max_events[, timeout=None]) -> eventlist\n\
|
---|
| 1660 | \n\
|
---|
| 1661 | Calls the kernel kevent function.\n\
|
---|
| 1662 | - changelist must be a list of kevent objects describing the changes\n\
|
---|
| 1663 | to be made to the kernel's watch list or None.\n\
|
---|
| 1664 | - max_events lets you specify the maximum number of events that the\n\
|
---|
| 1665 | kernel will return.\n\
|
---|
| 1666 | - timeout is the maximum time to wait in seconds, or else None,\n\
|
---|
| 1667 | to wait forever. timeout accepts floats for smaller timeouts, too.");
|
---|
| 1668 |
|
---|
| 1669 |
|
---|
| 1670 | static PyMethodDef kqueue_queue_methods[] = {
|
---|
[391] | 1671 | {"fromfd", (PyCFunction)kqueue_queue_fromfd,
|
---|
| 1672 | METH_VARARGS | METH_CLASS, kqueue_queue_fromfd_doc},
|
---|
| 1673 | {"close", (PyCFunction)kqueue_queue_close, METH_NOARGS,
|
---|
| 1674 | kqueue_queue_close_doc},
|
---|
| 1675 | {"fileno", (PyCFunction)kqueue_queue_fileno, METH_NOARGS,
|
---|
| 1676 | kqueue_queue_fileno_doc},
|
---|
| 1677 | {"control", (PyCFunction)kqueue_queue_control,
|
---|
| 1678 | METH_VARARGS , kqueue_queue_control_doc},
|
---|
| 1679 | {NULL, NULL},
|
---|
[2] | 1680 | };
|
---|
| 1681 |
|
---|
| 1682 | static PyGetSetDef kqueue_queue_getsetlist[] = {
|
---|
[391] | 1683 | {"closed", (getter)kqueue_queue_get_closed, NULL,
|
---|
| 1684 | "True if the kqueue handler is closed"},
|
---|
| 1685 | {0},
|
---|
[2] | 1686 | };
|
---|
| 1687 |
|
---|
| 1688 | PyDoc_STRVAR(kqueue_queue_doc,
|
---|
| 1689 | "Kqueue syscall wrapper.\n\
|
---|
| 1690 | \n\
|
---|
| 1691 | For example, to start watching a socket for input:\n\
|
---|
| 1692 | >>> kq = kqueue()\n\
|
---|
| 1693 | >>> sock = socket()\n\
|
---|
| 1694 | >>> sock.connect((host, port))\n\
|
---|
| 1695 | >>> kq.control([kevent(sock, KQ_FILTER_WRITE, KQ_EV_ADD)], 0)\n\
|
---|
| 1696 | \n\
|
---|
| 1697 | To wait one second for it to become writeable:\n\
|
---|
| 1698 | >>> kq.control(None, 1, 1000)\n\
|
---|
| 1699 | \n\
|
---|
| 1700 | To stop listening:\n\
|
---|
| 1701 | >>> kq.control([kevent(sock, KQ_FILTER_WRITE, KQ_EV_DELETE)], 0)");
|
---|
| 1702 |
|
---|
| 1703 | static PyTypeObject kqueue_queue_Type = {
|
---|
[391] | 1704 | PyVarObject_HEAD_INIT(NULL, 0)
|
---|
| 1705 | "select.kqueue", /* tp_name */
|
---|
| 1706 | sizeof(kqueue_queue_Object), /* tp_basicsize */
|
---|
| 1707 | 0, /* tp_itemsize */
|
---|
| 1708 | (destructor)kqueue_queue_dealloc, /* tp_dealloc */
|
---|
| 1709 | 0, /* tp_print */
|
---|
| 1710 | 0, /* tp_getattr */
|
---|
| 1711 | 0, /* tp_setattr */
|
---|
| 1712 | 0, /* tp_compare */
|
---|
| 1713 | 0, /* tp_repr */
|
---|
| 1714 | 0, /* tp_as_number */
|
---|
| 1715 | 0, /* tp_as_sequence */
|
---|
| 1716 | 0, /* tp_as_mapping */
|
---|
| 1717 | 0, /* tp_hash */
|
---|
| 1718 | 0, /* tp_call */
|
---|
| 1719 | 0, /* tp_str */
|
---|
| 1720 | 0, /* tp_getattro */
|
---|
| 1721 | 0, /* tp_setattro */
|
---|
| 1722 | 0, /* tp_as_buffer */
|
---|
| 1723 | Py_TPFLAGS_DEFAULT, /* tp_flags */
|
---|
| 1724 | kqueue_queue_doc, /* tp_doc */
|
---|
| 1725 | 0, /* tp_traverse */
|
---|
| 1726 | 0, /* tp_clear */
|
---|
| 1727 | 0, /* tp_richcompare */
|
---|
| 1728 | 0, /* tp_weaklistoffset */
|
---|
| 1729 | 0, /* tp_iter */
|
---|
| 1730 | 0, /* tp_iternext */
|
---|
| 1731 | kqueue_queue_methods, /* tp_methods */
|
---|
| 1732 | 0, /* tp_members */
|
---|
| 1733 | kqueue_queue_getsetlist, /* tp_getset */
|
---|
| 1734 | 0, /* tp_base */
|
---|
| 1735 | 0, /* tp_dict */
|
---|
| 1736 | 0, /* tp_descr_get */
|
---|
| 1737 | 0, /* tp_descr_set */
|
---|
| 1738 | 0, /* tp_dictoffset */
|
---|
| 1739 | 0, /* tp_init */
|
---|
| 1740 | 0, /* tp_alloc */
|
---|
| 1741 | kqueue_queue_new, /* tp_new */
|
---|
| 1742 | 0, /* tp_free */
|
---|
[2] | 1743 | };
|
---|
| 1744 |
|
---|
| 1745 | #endif /* HAVE_KQUEUE */
|
---|
| 1746 | /* ************************************************************************ */
|
---|
| 1747 |
|
---|
| 1748 | PyDoc_STRVAR(select_doc,
|
---|
| 1749 | "select(rlist, wlist, xlist[, timeout]) -> (rlist, wlist, xlist)\n\
|
---|
| 1750 | \n\
|
---|
| 1751 | Wait until one or more file descriptors are ready for some kind of I/O.\n\
|
---|
| 1752 | The first three arguments are sequences of file descriptors to be waited for:\n\
|
---|
| 1753 | rlist -- wait until ready for reading\n\
|
---|
| 1754 | wlist -- wait until ready for writing\n\
|
---|
| 1755 | xlist -- wait for an ``exceptional condition''\n\
|
---|
| 1756 | If only one kind of condition is required, pass [] for the other lists.\n\
|
---|
| 1757 | A file descriptor is either a socket or file object, or a small integer\n\
|
---|
| 1758 | gotten from a fileno() method call on one of those.\n\
|
---|
| 1759 | \n\
|
---|
| 1760 | The optional 4th argument specifies a timeout in seconds; it may be\n\
|
---|
| 1761 | a floating point number to specify fractions of seconds. If it is absent\n\
|
---|
| 1762 | or None, the call will never time out.\n\
|
---|
| 1763 | \n\
|
---|
| 1764 | The return value is a tuple of three lists corresponding to the first three\n\
|
---|
| 1765 | arguments; each contains the subset of the corresponding file descriptors\n\
|
---|
| 1766 | that are ready.\n\
|
---|
| 1767 | \n\
|
---|
| 1768 | *** IMPORTANT NOTICE ***\n\
|
---|
| 1769 | On Windows and OpenVMS, only sockets are supported; on Unix, all file\n\
|
---|
| 1770 | descriptors can be used.");
|
---|
| 1771 |
|
---|
| 1772 | static PyMethodDef select_methods[] = {
|
---|
[391] | 1773 | {"select", select_select, METH_VARARGS, select_doc},
|
---|
| 1774 | #if defined(HAVE_POLL) && !defined(HAVE_BROKEN_POLL)
|
---|
| 1775 | {"poll", select_poll, METH_NOARGS, poll_doc},
|
---|
[2] | 1776 | #endif /* HAVE_POLL */
|
---|
[391] | 1777 | {0, 0}, /* sentinel */
|
---|
[2] | 1778 | };
|
---|
| 1779 |
|
---|
| 1780 | PyDoc_STRVAR(module_doc,
|
---|
| 1781 | "This module supports asynchronous I/O on multiple file descriptors.\n\
|
---|
| 1782 | \n\
|
---|
| 1783 | *** IMPORTANT NOTICE ***\n\
|
---|
| 1784 | On Windows and OpenVMS, only sockets are supported; on Unix, all file descriptors.");
|
---|
| 1785 |
|
---|
| 1786 | PyMODINIT_FUNC
|
---|
| 1787 | initselect(void)
|
---|
| 1788 | {
|
---|
[391] | 1789 | PyObject *m;
|
---|
| 1790 | m = Py_InitModule3("select", select_methods, module_doc);
|
---|
| 1791 | if (m == NULL)
|
---|
| 1792 | return;
|
---|
[2] | 1793 |
|
---|
[391] | 1794 | SelectError = PyErr_NewException("select.error", NULL, NULL);
|
---|
| 1795 | Py_INCREF(SelectError);
|
---|
| 1796 | PyModule_AddObject(m, "error", SelectError);
|
---|
[2] | 1797 |
|
---|
[391] | 1798 | #ifdef PIPE_BUF
|
---|
| 1799 | #ifdef HAVE_BROKEN_PIPE_BUF
|
---|
| 1800 | #undef PIPE_BUF
|
---|
| 1801 | #define PIPE_BUF 512
|
---|
| 1802 | #endif
|
---|
| 1803 | PyModule_AddIntConstant(m, "PIPE_BUF", PIPE_BUF);
|
---|
| 1804 | #endif
|
---|
| 1805 |
|
---|
| 1806 | #if defined(HAVE_POLL) && !defined(HAVE_BROKEN_POLL)
|
---|
[2] | 1807 | #ifdef __APPLE__
|
---|
[391] | 1808 | if (select_have_broken_poll()) {
|
---|
| 1809 | if (PyObject_DelAttrString(m, "poll") == -1) {
|
---|
| 1810 | PyErr_Clear();
|
---|
| 1811 | }
|
---|
| 1812 | } else {
|
---|
[2] | 1813 | #else
|
---|
[391] | 1814 | {
|
---|
[2] | 1815 | #endif
|
---|
[391] | 1816 | Py_TYPE(&poll_Type) = &PyType_Type;
|
---|
| 1817 | PyModule_AddIntConstant(m, "POLLIN", POLLIN);
|
---|
| 1818 | PyModule_AddIntConstant(m, "POLLPRI", POLLPRI);
|
---|
| 1819 | PyModule_AddIntConstant(m, "POLLOUT", POLLOUT);
|
---|
| 1820 | PyModule_AddIntConstant(m, "POLLERR", POLLERR);
|
---|
| 1821 | PyModule_AddIntConstant(m, "POLLHUP", POLLHUP);
|
---|
| 1822 | PyModule_AddIntConstant(m, "POLLNVAL", POLLNVAL);
|
---|
[2] | 1823 |
|
---|
| 1824 | #ifdef POLLRDNORM
|
---|
[391] | 1825 | PyModule_AddIntConstant(m, "POLLRDNORM", POLLRDNORM);
|
---|
[2] | 1826 | #endif
|
---|
| 1827 | #ifdef POLLRDBAND
|
---|
[391] | 1828 | PyModule_AddIntConstant(m, "POLLRDBAND", POLLRDBAND);
|
---|
[2] | 1829 | #endif
|
---|
| 1830 | #ifdef POLLWRNORM
|
---|
[391] | 1831 | PyModule_AddIntConstant(m, "POLLWRNORM", POLLWRNORM);
|
---|
[2] | 1832 | #endif
|
---|
| 1833 | #ifdef POLLWRBAND
|
---|
[391] | 1834 | PyModule_AddIntConstant(m, "POLLWRBAND", POLLWRBAND);
|
---|
[2] | 1835 | #endif
|
---|
| 1836 | #ifdef POLLMSG
|
---|
[391] | 1837 | PyModule_AddIntConstant(m, "POLLMSG", POLLMSG);
|
---|
[2] | 1838 | #endif
|
---|
[391] | 1839 | }
|
---|
[2] | 1840 | #endif /* HAVE_POLL */
|
---|
| 1841 |
|
---|
| 1842 | #ifdef HAVE_EPOLL
|
---|
[391] | 1843 | Py_TYPE(&pyEpoll_Type) = &PyType_Type;
|
---|
| 1844 | if (PyType_Ready(&pyEpoll_Type) < 0)
|
---|
| 1845 | return;
|
---|
[2] | 1846 |
|
---|
[391] | 1847 | Py_INCREF(&pyEpoll_Type);
|
---|
| 1848 | PyModule_AddObject(m, "epoll", (PyObject *) &pyEpoll_Type);
|
---|
[2] | 1849 |
|
---|
[391] | 1850 | PyModule_AddIntConstant(m, "EPOLLIN", EPOLLIN);
|
---|
| 1851 | PyModule_AddIntConstant(m, "EPOLLOUT", EPOLLOUT);
|
---|
| 1852 | PyModule_AddIntConstant(m, "EPOLLPRI", EPOLLPRI);
|
---|
| 1853 | PyModule_AddIntConstant(m, "EPOLLERR", EPOLLERR);
|
---|
| 1854 | PyModule_AddIntConstant(m, "EPOLLHUP", EPOLLHUP);
|
---|
| 1855 | PyModule_AddIntConstant(m, "EPOLLET", EPOLLET);
|
---|
[2] | 1856 | #ifdef EPOLLONESHOT
|
---|
[391] | 1857 | /* Kernel 2.6.2+ */
|
---|
| 1858 | PyModule_AddIntConstant(m, "EPOLLONESHOT", EPOLLONESHOT);
|
---|
[2] | 1859 | #endif
|
---|
[391] | 1860 | /* PyModule_AddIntConstant(m, "EPOLL_RDHUP", EPOLLRDHUP); */
|
---|
| 1861 | PyModule_AddIntConstant(m, "EPOLLRDNORM", EPOLLRDNORM);
|
---|
| 1862 | PyModule_AddIntConstant(m, "EPOLLRDBAND", EPOLLRDBAND);
|
---|
| 1863 | PyModule_AddIntConstant(m, "EPOLLWRNORM", EPOLLWRNORM);
|
---|
| 1864 | PyModule_AddIntConstant(m, "EPOLLWRBAND", EPOLLWRBAND);
|
---|
| 1865 | PyModule_AddIntConstant(m, "EPOLLMSG", EPOLLMSG);
|
---|
[2] | 1866 | #endif /* HAVE_EPOLL */
|
---|
| 1867 |
|
---|
| 1868 | #ifdef HAVE_KQUEUE
|
---|
[391] | 1869 | kqueue_event_Type.tp_new = PyType_GenericNew;
|
---|
| 1870 | Py_TYPE(&kqueue_event_Type) = &PyType_Type;
|
---|
| 1871 | if(PyType_Ready(&kqueue_event_Type) < 0)
|
---|
| 1872 | return;
|
---|
[2] | 1873 |
|
---|
[391] | 1874 | Py_INCREF(&kqueue_event_Type);
|
---|
| 1875 | PyModule_AddObject(m, "kevent", (PyObject *)&kqueue_event_Type);
|
---|
[2] | 1876 |
|
---|
[391] | 1877 | Py_TYPE(&kqueue_queue_Type) = &PyType_Type;
|
---|
| 1878 | if(PyType_Ready(&kqueue_queue_Type) < 0)
|
---|
| 1879 | return;
|
---|
| 1880 | Py_INCREF(&kqueue_queue_Type);
|
---|
| 1881 | PyModule_AddObject(m, "kqueue", (PyObject *)&kqueue_queue_Type);
|
---|
| 1882 |
|
---|
| 1883 | /* event filters */
|
---|
| 1884 | PyModule_AddIntConstant(m, "KQ_FILTER_READ", EVFILT_READ);
|
---|
| 1885 | PyModule_AddIntConstant(m, "KQ_FILTER_WRITE", EVFILT_WRITE);
|
---|
| 1886 | PyModule_AddIntConstant(m, "KQ_FILTER_AIO", EVFILT_AIO);
|
---|
| 1887 | PyModule_AddIntConstant(m, "KQ_FILTER_VNODE", EVFILT_VNODE);
|
---|
| 1888 | PyModule_AddIntConstant(m, "KQ_FILTER_PROC", EVFILT_PROC);
|
---|
[2] | 1889 | #ifdef EVFILT_NETDEV
|
---|
[391] | 1890 | PyModule_AddIntConstant(m, "KQ_FILTER_NETDEV", EVFILT_NETDEV);
|
---|
[2] | 1891 | #endif
|
---|
[391] | 1892 | PyModule_AddIntConstant(m, "KQ_FILTER_SIGNAL", EVFILT_SIGNAL);
|
---|
| 1893 | PyModule_AddIntConstant(m, "KQ_FILTER_TIMER", EVFILT_TIMER);
|
---|
[2] | 1894 |
|
---|
[391] | 1895 | /* event flags */
|
---|
| 1896 | PyModule_AddIntConstant(m, "KQ_EV_ADD", EV_ADD);
|
---|
| 1897 | PyModule_AddIntConstant(m, "KQ_EV_DELETE", EV_DELETE);
|
---|
| 1898 | PyModule_AddIntConstant(m, "KQ_EV_ENABLE", EV_ENABLE);
|
---|
| 1899 | PyModule_AddIntConstant(m, "KQ_EV_DISABLE", EV_DISABLE);
|
---|
| 1900 | PyModule_AddIntConstant(m, "KQ_EV_ONESHOT", EV_ONESHOT);
|
---|
| 1901 | PyModule_AddIntConstant(m, "KQ_EV_CLEAR", EV_CLEAR);
|
---|
[2] | 1902 |
|
---|
[391] | 1903 | PyModule_AddIntConstant(m, "KQ_EV_SYSFLAGS", EV_SYSFLAGS);
|
---|
| 1904 | PyModule_AddIntConstant(m, "KQ_EV_FLAG1", EV_FLAG1);
|
---|
[2] | 1905 |
|
---|
[391] | 1906 | PyModule_AddIntConstant(m, "KQ_EV_EOF", EV_EOF);
|
---|
| 1907 | PyModule_AddIntConstant(m, "KQ_EV_ERROR", EV_ERROR);
|
---|
[2] | 1908 |
|
---|
[391] | 1909 | /* READ WRITE filter flag */
|
---|
| 1910 | PyModule_AddIntConstant(m, "KQ_NOTE_LOWAT", NOTE_LOWAT);
|
---|
[2] | 1911 |
|
---|
[391] | 1912 | /* VNODE filter flags */
|
---|
| 1913 | PyModule_AddIntConstant(m, "KQ_NOTE_DELETE", NOTE_DELETE);
|
---|
| 1914 | PyModule_AddIntConstant(m, "KQ_NOTE_WRITE", NOTE_WRITE);
|
---|
| 1915 | PyModule_AddIntConstant(m, "KQ_NOTE_EXTEND", NOTE_EXTEND);
|
---|
| 1916 | PyModule_AddIntConstant(m, "KQ_NOTE_ATTRIB", NOTE_ATTRIB);
|
---|
| 1917 | PyModule_AddIntConstant(m, "KQ_NOTE_LINK", NOTE_LINK);
|
---|
| 1918 | PyModule_AddIntConstant(m, "KQ_NOTE_RENAME", NOTE_RENAME);
|
---|
| 1919 | PyModule_AddIntConstant(m, "KQ_NOTE_REVOKE", NOTE_REVOKE);
|
---|
[2] | 1920 |
|
---|
[391] | 1921 | /* PROC filter flags */
|
---|
| 1922 | PyModule_AddIntConstant(m, "KQ_NOTE_EXIT", NOTE_EXIT);
|
---|
| 1923 | PyModule_AddIntConstant(m, "KQ_NOTE_FORK", NOTE_FORK);
|
---|
| 1924 | PyModule_AddIntConstant(m, "KQ_NOTE_EXEC", NOTE_EXEC);
|
---|
| 1925 | PyModule_AddIntConstant(m, "KQ_NOTE_PCTRLMASK", NOTE_PCTRLMASK);
|
---|
| 1926 | PyModule_AddIntConstant(m, "KQ_NOTE_PDATAMASK", NOTE_PDATAMASK);
|
---|
[2] | 1927 |
|
---|
[391] | 1928 | PyModule_AddIntConstant(m, "KQ_NOTE_TRACK", NOTE_TRACK);
|
---|
| 1929 | PyModule_AddIntConstant(m, "KQ_NOTE_CHILD", NOTE_CHILD);
|
---|
| 1930 | PyModule_AddIntConstant(m, "KQ_NOTE_TRACKERR", NOTE_TRACKERR);
|
---|
| 1931 |
|
---|
| 1932 | /* NETDEV filter flags */
|
---|
[2] | 1933 | #ifdef EVFILT_NETDEV
|
---|
[391] | 1934 | PyModule_AddIntConstant(m, "KQ_NOTE_LINKUP", NOTE_LINKUP);
|
---|
| 1935 | PyModule_AddIntConstant(m, "KQ_NOTE_LINKDOWN", NOTE_LINKDOWN);
|
---|
| 1936 | PyModule_AddIntConstant(m, "KQ_NOTE_LINKINV", NOTE_LINKINV);
|
---|
[2] | 1937 | #endif
|
---|
| 1938 |
|
---|
| 1939 | #endif /* HAVE_KQUEUE */
|
---|
| 1940 | }
|
---|