chiark - git - mdw - mLib-python/blob - fdutils.pyx

   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(fd), fbix, 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 --------------------------------------------------