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

   1 ### -*-pyrex-*-
   2 ###
   3 ### Packet buffering
   4 ###
   5 ### (c) 2005 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 cdef class PacketBuffer:
  27   cdef pkbuf pk
  28   cdef _packet
  29   cdef _eof
  30   def __cinit__(me, packetproc = None, eofproc = None, *hunoz, **hukairz):
  31     pkbuf_init(&me.pk, _pkfunc, <void *>me)
  32     me._packet = _checkcallable(packetproc, 'packet proc')
  33     me._eof = _checkcallable(eofproc, 'eof proc')
  34   def __dealloc__(me):
  35     pkbuf_destroy(&me.pk)
  36   property activep:
  37     def __get__(me):
  38       return _tobool(me.pk.f & PKBUF_ENABLE)
  39   property want:
  40     def __get__(me):
  41       return me.pk.want
  42     def __set__(me, want):
  43       if want <= 0:
  44         raise TypeError, 'want must be positive'
  45       pkbuf_want(&me.pk, pk)
  46   property packetproc:
  47     def __get__(me):
  48       return me._packet
  49     def __set__(me, proc):
  50       me._line = _checkcallable(proc, 'packet proc')
  51     def __del__(me):
  52       me._line = None
  53   property eofproc:
  54     def __get__(me):
  55       return me._eof
  56     def __set__(me, proc):
  57       me._eof = _checkcallable(proc, 'eof proc')
  58     def __del__(me):
  59       me._eof = None
  60   def enable(me):
  61     if me.pk.f & PKBUF_ENABLE:
  62       raise ValueError, 'already enabled'
  63     me.pk.f = me.pk.f | PKBUF_ENABLE
  64     me.enabled()
  65     return me
  66   def disable(me):
  67     if not (me.pk.f & PKBUF_ENABLE):
  68       raise ValueError, 'already disabled'
  69     me.pk.f = me.pk.f & ~PKBUF_ENABLE
  70     me.disabled()
  71     return me
  72   def close(me):
  73     if not (me.pk.f & PKBUF_ENABLE):
  74       raise ValueError, 'buffer disabled'
  75     pkbuf_close(&me.pk)
  76     return me
  77   property free:
  78     def __get__(me):
  79       cdef unsigned char *p
  80       return pkbuf_free(&me.pk, &p)
  81   def flush(me, str):
  82     cdef Py_ssize_t len
  83     cdef unsigned char *p
  84     cdef unsigned char *q
  85     cdef size_t n
  86     PyObject_AsReadBuffer(str, <void **>&p, &len)
  87     while len > 0:
  88       n = pkbuf_free(&me.pk, &q)
  89       if n > len:
  90         n = len
  91       memcpy(q, p, n)
  92       p = p + n
  93       len = len - n
  94       if not (me.pk.f & PKBUF_ENABLE):
  95         break
  96       pkbuf_flush(&me.pk, q, n)
  97     return PyString_FromStringAndSize(<char *>p, len)
  98   def enabled(me):
  99     pass
 100   def disabled(me):
 101     pass
 102   def packet(me, pk):
 103     return _maybecall(me._packet, (pk,))
 104   def eof(me):
 105     return _maybecall(me._eof, ())
 106
 107 cdef void _pkfunc(unsigned char *p, size_t n, pkbuf *pk,
 108                   size_t *keep, void *arg):
 109   cdef PacketBuffer pb
 110   cdef void *rp
 111   cdef Py_ssize_t rn
 112   pb = <PacketBuffer>arg
 113   if p is NULL:
 114     pb.eof()
 115   else:
 116     r = pb.packet(PyString_FromStringAndSize(<char *>p, n))
 117     if r is not None:
 118       PyObject_AsReadBuffer(r, &rp, &rn)
 119       if rn > n:
 120         raise ValueError, 'remaining buffer too large'
 121       if rn:
 122         memcpy(p + n - rn, rp, rn)
 123         keep[0] = rn
 124
 125 ###----- That's all, folks --------------------------------------------------