chiark / gitweb /
Release 1.0.5.
[mLib-python] / pkbuf.pyx
CommitLineData
5b1830f3
MW
1### -*-pyrex-*-
2###
3### Packet buffering
4###
5### (c) 2005 Straylight/Edgeware
6###
579d0169 7
5b1830f3
MW
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.
579d0169 25
26cdef class PacketBuffer:
27 cdef pkbuf pk
28 cdef _packet
29 cdef _eof
376ad06d 30 def __cinit__(me, packetproc = None, eofproc = None, *hunoz, **hukairz):
579d0169 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 int 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
107cdef 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 int 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
5b1830f3 125###----- That's all, folks --------------------------------------------------