| 1 | ### -*-pyrex-*- |
| 2 | ### |
| 3 | ### Messing with file descriptors |
| 4 | ### |
| 5 | ### (c) 2007 Straylight/Edgeware |
| 6 | ### |
| 7 | |
| 8 | ###----- Licensing notice --------------------------------------------------- |
| 9 | ### |
| 10 | ### This file is part of the Python interface to mLib. |
| 11 | ### |
| 12 | ### mLib/Python is free software; you can redistribute it and/or modify |
| 13 | ### it under the terms of the GNU General Public License as published by |
| 14 | ### the Free Software Foundation; either version 2 of the License, or |
| 15 | ### (at your option) any later version. |
| 16 | ### |
| 17 | ### mLib/Python is distributed in the hope that it will be useful, |
| 18 | ### but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 19 | ### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 20 | ### GNU General Public License for more details. |
| 21 | ### |
| 22 | ### You should have received a copy of the GNU General Public License |
| 23 | ### along with mLib/Python; if not, write to the Free Software Foundation, |
| 24 | ### Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
| 25 | |
| 26 | def fdflags(file, |
| 27 | unsigned fbic = 0, unsigned fxor = 0, |
| 28 | unsigned fdbic = 0, unsigned fdxor = 0): |
| 29 | """ |
| 30 | fdflags(FILE, [fbic = 0], [fxor = 0], [fdbic = 0], [fdxor = 0]) |
| 31 | |
| 32 | Set fcntl(2) file and descriptor flags. If these are FL and FD, then |
| 33 | update: |
| 34 | |
| 35 | * FL = (FL & ~FBIC) ^ FXOR |
| 36 | * FD = (FD & ~FDBIC) ^ FDXOR |
| 37 | |
| 38 | FILE may be integer file descriptor or an object with `fileno' method. |
| 39 | """ |
| 40 | cdef int rc |
| 41 | rc = _fdflags(_getfd(file), fbic, fxor, fdbic, fdxor) |
| 42 | if rc < 0: |
| 43 | _oserror() |
| 44 | return rc |
| 45 | |
| 46 | def fdsend(sock, file, buffer): |
| 47 | """ |
| 48 | fdsend(SOCK, FILE, BUFFER) -> RC: |
| 49 | send FILE over Unix-domain socket SOCK, along with BUFFER |
| 50 | """ |
| 51 | cdef void *p |
| 52 | cdef Py_ssize_t len |
| 53 | cdef int rc |
| 54 | PyObject_AsReadBuffer(buffer, <cvp *>&p, &len) |
| 55 | rc = fdpass_send(_getfd(sock), _getfd(file), p, len) |
| 56 | if rc < 0: |
| 57 | _oserror() |
| 58 | return rc |
| 59 | |
| 60 | def fdrecv(sock, unsigned size): |
| 61 | """ |
| 62 | fdrecv(SOCK, SIZE) -> FD, BUFFER |
| 63 | receive file FD and BUFFER of length up to SIZE from Unix-domain SOCK |
| 64 | """ |
| 65 | cdef void *p |
| 66 | cdef buf |
| 67 | cdef Py_ssize_t len |
| 68 | cdef PyObject *obj |
| 69 | cdef int fd |
| 70 | buf = PyString_FromStringAndSize(NULL, size) |
| 71 | p = PyString_AS_STRING(buf) |
| 72 | len = fdpass_recv(_getfd(sock), &fd, p, size) |
| 73 | if len < 0: |
| 74 | _oserror() |
| 75 | obj = <PyObject *>buf |
| 76 | _PyString_Resize(&obj, len) |
| 77 | return fd, <object>obj |
| 78 | |
| 79 | ###----- That's all, folks -------------------------------------------------- |