3 * Reading and writing buffers of stuff
5 * (c) 2005 Straylight/Edgeware
8 /*----- Licensing notice --------------------------------------------------*
10 * This file is part of the Python interface to Catacomb.
12 * Catacomb/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.
17 * Catacomb/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.
22 * You should have received a copy of the GNU General Public License
23 * along with Catacomb/Python; if not, write to the Free Software Foundation,
24 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
27 /*----- Header files ------------------------------------------------------*/
29 #include "catacomb-python.h"
31 /*----- Read buffers ------------------------------------------------------*/
33 PyTypeObject *rbuf_pytype;
35 static PyObject *rbuf_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
40 static const char *const kwlist[] = { "data", 0 };
42 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&:new", KWLIST, convbin, &in))
45 memcpy(q, in.p, in.sz);
46 me = (buf_pyobj *)ty->tp_alloc(ty, 0);
47 me->sub = 0; me->lk = 0;
48 buf_init(&me->b, q, in.sz);
50 return ((PyObject *)me);
53 static void buf_pydealloc(PyObject *me)
56 if (BUF_SUB(me)) Py_DECREF(BUF_SUB(me));
57 else xfree(BBASE(BUF_B(me)));
62 static int rbuf_pygetbuf(PyObject *me, Py_buffer *vw, int f)
65 return (PyBuffer_FillInfo(vw, me, BCUR(b), BLEFT(b), 1, f));
68 static Py_ssize_t rbuf_pysegcount(PyObject *me, Py_ssize_t *nn)
69 { if (nn) *nn = BSZ(BUF_B(me)); return (1); }
70 static Py_ssize_t rbuf_pyreadbuf(PyObject *me, Py_ssize_t seg, void **q)
71 { assert(seg == 0); *q = BCUR(BUF_B(me)); return (BLEFT(BUF_B(me))); }
74 static PyObject *rbmeth_skip(PyObject *me, PyObject *arg)
78 if (!PyArg_ParseTuple(arg, "O&:skip", convszt, &n)) goto end;
79 if (!buf_get(BUF_B(me), n)) BUFERR("buffer exhausted");
85 static PyObject *rbmeth_get(PyObject *me, PyObject *arg)
90 if (!PyArg_ParseTuple(arg, "O&:get", convszt, &n)) goto end;
91 if ((p = buf_get(BUF_B(me), n)) == 0) BUFERR("buffer exhausted");
92 return (bytestring_pywrap(p, n));
97 #define RBMETH_GETU_(n, W, w) \
98 static PyObject *rbmeth_getu##w(PyObject *me) \
101 if (buf_getu##w(BUF_B(me), &x)) BUFERR("buffer exhausted"); \
102 if (MASK##W <= ULONG_MAX) return (getulong(x)); \
103 else { kludge64 y; ASSIGN64(y, x); return (getk64(y)); } \
107 DOUINTCONV(RBMETH_GETU_)
109 #define RBMETH_GETBLK_(n, W, w) \
110 static PyObject *rbmeth_getblk##w(PyObject *me) \
114 if ((q = buf_getmem##w(BUF_B(me), &sz)) == 0) \
115 BUFERR("buffer exhausted"); \
116 return (bytestring_pywrap(q, sz)); \
120 BUF_DOSUFFIXES(RBMETH_GETBLK_)
122 #define RBMETH_GETBUF_(n, W, w) \
123 static PyObject *rbmeth_getbuf##w(PyObject *me) \
127 if (buf_getbuf##w(BUF_B(me), &bb)) BUFERR("buffer exhausted"); \
128 b = PyObject_NEW(buf_pyobj, rbuf_pytype); \
133 return ((PyObject *)b); \
137 BUF_DOSUFFIXES(RBMETH_GETBUF_)
139 static PyObject *rbmeth_getmp(PyObject *me)
142 if ((x = buf_getmp(BUF_B(me))) == 0) BUFERR("buffer exhausted");
143 return (mp_pywrap(x));
148 static PyObject *rbmeth_getgf(PyObject *me)
151 if ((x = buf_getmp(BUF_B(me))) == 0) BUFERR("buffer exhausted");
152 return (gf_pywrap(x));
157 static PyObject *rbmeth_getecpt(PyObject *me, PyObject *arg, PyObject *kw)
159 PyObject *cobj = Py_None;
160 static const char *const kwlist[] = { "curve", 0 };
162 if (!PyArg_ParseTupleAndKeywords(arg, kw, "|O:getecpt", KWLIST, &cobj))
164 if (cobj == Py_None) cobj = (PyObject *)ecpt_pytype;
165 if (!PyType_Check(cobj) ||
166 !PyType_IsSubtype((PyTypeObject *)cobj, ecpt_pytype))
167 TYERR("expected elliptic curve type");
168 if (buf_getec(BUF_B(me), &pt)) BUFERR("buffer exhausted");
169 return (ecpt_pywrapout(cobj, &pt));
174 static PyObject *rbmeth_getecptraw(PyObject *me, PyObject *arg)
179 if (!PyArg_ParseTuple(arg, "O!:getecptraw", eccurve_pytype, &cobj))
181 if (ec_getraw(ECCURVE_C(cobj), BUF_B(me), &pt)) BUFERR("buffer exhausted");
182 rc = ecpt_pywrapout(cobj, &pt);
187 static PyObject *rbmeth_os2ecp(PyObject *me, PyObject *arg, PyObject *kw)
191 unsigned f = EC_XONLY | EC_LSB | EC_SORT | EC_EXPLY;
193 static const char *const kwlist[] = { "curve", "flags", 0 };
194 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O!|O&:os2ecp", KWLIST,
195 eccurve_pytype, &cobj, convuint, &f))
197 if (ec_os2ecp(ECCURVE_C(cobj), f, BUF_B(me), &pt)) VALERR("bad point");
198 rc = ecpt_pywrapout(cobj, &pt);
203 static PyObject *rbmeth_getge(PyObject *me, PyObject *arg)
207 if (!PyArg_ParseTuple(arg, "O!:getge", group_pytype, &gobj)) goto end;
208 x = G_CREATE(GROUP_G(gobj));
209 if (G_FROMBUF(GROUP_G(gobj), BUF_B(me), x)) BUFERR("buffer exhausted");
210 return (ge_pywrap(gobj, x));
212 if (x) G_DESTROY(GROUP_G(gobj), x);
216 static PyObject *rbmeth_getgeraw(PyObject *me, PyObject *arg)
220 if (!PyArg_ParseTuple(arg, "O!:getgeraw", group_pytype, &gobj)) goto end;
221 x = G_CREATE(GROUP_G(gobj));
222 if (G_FROMRAW(GROUP_G(gobj), BUF_B(me), x)) BUFERR("buffer exhausted");
223 return (ge_pywrap(gobj, x));
225 if (x) G_DESTROY(GROUP_G(gobj), x);
229 static PyObject *rbget_size(PyObject *me, void *hunoz)
230 { return (PyInt_FromLong(BSZ(BUF_B(me)))); }
231 static PyObject *rbget_left(PyObject *me, void *hunoz)
232 { return (PyInt_FromLong(BLEFT(BUF_B(me)))); }
233 static PyObject *rbget_endp(PyObject *me, void *hunoz)
234 { return (getbool(!BLEFT(BUF_B(me)))); }
235 static PyObject *rbget_offset(PyObject *me, void *hunoz)
236 { return (PyInt_FromLong(BLEN(BUF_B(me)))); }
237 static int rbset_offset(PyObject *me, PyObject *x, void *hunoz)
240 if (!x) NIERR("__del__");
241 if (!convszt(x, &n)) goto end;
242 if (n > BSZ(BUF_B(me))) VALERR("out of range");
243 BCUR(BUF_B(me)) = BBASE(BUF_B(me)) + n;
249 static const PyGetSetDef rbuf_pygetset[] = {
250 #define GETSETNAME(op, name) rb##op##_##name
251 GET (size, "RBUF.size -> SIZE")
252 GET (left, "RBUF.left -> REMAINDER")
253 GET (endp, "RBUF.endp -> BOOL")
254 GETSET(offset, "RBUF.offset -> OFFSET")
259 static const PyMethodDef rbuf_pymethods[] = {
260 #define METHNAME(func) rbmeth_##func
261 METH (skip, "RBUF.skip(N)")
262 METH (get, "RBUF.get(N) -> BYTES")
263 #define RBMETH_DECL_GETU_(n, W, w) \
264 NAMETH(getu##w, "RBUF.getu" #w "() -> INT")
265 DOUINTCONV(RBMETH_DECL_GETU_)
266 #define RBMETH_DECL_GETBLK_(n, W, w) \
267 NAMETH(getblk##w, "RBUF.getblk" #w "() -> BYTES")
268 BUF_DOSUFFIXES(RBMETH_DECL_GETBLK_)
269 #define RBMETH_DECL_GETBUF_(n, W, w) \
270 NAMETH(getbuf##w, "RBUF.getbuf" #w "() -> RBUF'")
271 BUF_DOSUFFIXES(RBMETH_DECL_GETBUF_)
272 NAMETH(getmp, "RBUF.getmp() -> X")
273 NAMETH(getgf, "RBUF.getgf() -> X")
274 KWMETH(getecpt, "RBUF.getecpt([curve = None]) -> P")
275 METH (getecptraw, "RBUF.getecptraw(CURVE) -> P")
276 KWMETH(os2ecp, "RBUF.os2ecp(CURVE, [flags = ...]) -> P")
277 METH (getge, "RBUF.getge(GROUP) -> X")
278 METH (getgeraw, "RBUF.getgeraw(GROUP) -> X")
283 static const PyBufferProcs rbuf_pybuffer = {
285 rbuf_pygetbuf, /* @bf_getbuffer@ */
286 0, /* @bf_releasebuffer@ */
288 rbuf_pyreadbuf, /* @bf_getreadbuffer@ */
289 0, /* @bf_getwritebuffer@ */
290 rbuf_pysegcount, /* @bf_getsegcount@ */
291 0 /* @bf_getcharbuffer@ */
295 static const PyTypeObject rbuf_pytype_skel = {
296 PyVarObject_HEAD_INIT(0, 0) /* Header */
297 "ReadBuffer", /* @tp_name@ */
298 sizeof(buf_pyobj), /* @tp_basicsize@ */
299 0, /* @tp_itemsize@ */
301 buf_pydealloc, /* @tp_dealloc@ */
303 0, /* @tp_getattr@ */
304 0, /* @tp_setattr@ */
305 0, /* @tp_compare@ */
307 0, /* @tp_as_number@ */
308 0, /* @tp_as_sequence@ */
309 0, /* @tp_as_mapping@ */
313 0, /* @tp_getattro@ */
314 0, /* @tp_setattro@ */
315 PYBUFFER(rbuf), /* @tp_as_buffer@ */
316 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
320 "ReadBuffer(STR): a read buffer.",
322 0, /* @tp_traverse@ */
324 0, /* @tp_richcompare@ */
325 0, /* @tp_weaklistoffset@ */
327 0, /* @tp_iternext@ */
328 PYMETHODS(rbuf), /* @tp_methods@ */
329 0, /* @tp_members@ */
330 PYGETSET(rbuf), /* @tp_getset@ */
333 0, /* @tp_descr_get@ */
334 0, /* @tp_descr_set@ */
335 0, /* @tp_dictoffset@ */
337 PyType_GenericAlloc, /* @tp_alloc@ */
338 rbuf_pynew, /* @tp_new@ */
343 /*----- Write buffers -----------------------------------------------------*/
345 PyTypeObject *wbuf_pytype;
347 int ensurebuf(PyObject *me, size_t n)
352 size_t want = BLEN(b) + n;
357 BUFERR("buffer locked");
359 while (nn < want) nn <<= 1;
360 p = xrealloc(BBASE(b), nn, BSZ(b));
361 BCUR(b) = p + BLEN(b);
370 static PyObject *wbuf_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
375 static const char *const kwlist[] = { "size", 0 };
377 if (!PyArg_ParseTupleAndKeywords(arg, kw, "|O&:new", KWLIST,
380 me = (buf_pyobj *)ty->tp_alloc(ty, 0);
382 me->sub = 0; me->lk = 0;
383 buf_init(&me->b, p, n);
385 return ((PyObject *)me);
389 static int wbuf_pygetbuf(PyObject *me, Py_buffer *vw, int f)
392 if (PyBuffer_FillInfo(vw, me, BBASE(b), BLEN(b), 0, f)) return (-1);
393 BUF_LK(me)++; return (0);
395 static void wbuf_pyrlsbuf(PyObject *me, Py_buffer *vw)
398 static Py_ssize_t wbuf_pysegcount(PyObject *me, Py_ssize_t *nn)
399 { if (nn) *nn = BLEN(BUF_B(me)); return (1); }
400 static Py_ssize_t wbuf_pyreadbuf(PyObject *me, Py_ssize_t seg, void **q)
401 { assert(seg == 0); *q = BBASE(BUF_B(me)); return (BLEN(BUF_B(me))); }
404 static PyObject *wbmeth_zero(PyObject *me, PyObject *arg)
408 if (!PyArg_ParseTuple(arg, "O&:zero", convszt, &n)) return (0);
409 if (ensurebuf(me, n)) return (0);
410 p = buf_get(BUF_B(me), n); assert(p && BOK(BUF_B(me)));
415 static PyObject *wbmeth_put(PyObject *me, PyObject *arg)
418 if (!PyArg_ParseTuple(arg, "O&:put", convbin, &in)) return (0);
419 if (ensurebuf(me, in.sz)) return (0);
420 buf_put(BUF_B(me), in.p, in.sz); assert(BOK(BUF_B(me)));
424 #define WBMETH_PUTU_(n, W, w) \
425 static PyObject *wbmeth_putu##w(PyObject *me, PyObject *arg) \
428 if (!PyArg_ParseTuple(arg, "O&:putu" #w, convu##n, &i)) return (0); \
429 if (ensurebuf(me, SZ_##n)) return (0); \
430 buf_putu##w(BUF_B(me), i); assert(BOK(BUF_B(me))); \
433 DOUINTCONV(WBMETH_PUTU_)
437 #define WBMETH_PUTBLK_(n, W, w) \
438 static PyObject *wbmeth_putblk##w(PyObject *me, PyObject *arg) \
441 if (!PyArg_ParseTuple(arg, "O&:putblk" #w, convbin, &in)) goto end; \
442 if (MASK##W && in.sz > MASK##W) VALERR("too large"); \
443 if (ensurebuf(me, in.sz + SZ_##n)) return (0); \
444 buf_putmem##w(BUF_B(me), in.p, in.sz); assert(BOK(BUF_B(me))); \
449 BUF_DOSUFFIXES(WBMETH_PUTBLK_)
451 static PyObject *wbmeth_putmp(PyObject *me, PyObject *arg)
454 if (!PyArg_ParseTuple(arg, "O&:putmp", convmp, &x)) return (0);
455 if (ensurebuf(me, mp_octets(x) + 2)) return (0);
456 buf_putmp(BUF_B(me), x); assert(BOK(BUF_B(me)));
460 static PyObject *wbmeth_putgf(PyObject *me, PyObject *arg)
463 if (!PyArg_ParseTuple(arg, "O&:putgf", convgf, &x)) return (0);
464 if (ensurebuf(me, mp_octets(x) + 2)) return (0);
465 buf_putmp(BUF_B(me), x); assert(BOK(BUF_B(me)));
470 static PyObject *wbmeth_putecpt(PyObject *me, PyObject *arg)
473 if (!PyArg_ParseTuple(arg, "O&:putecpt", convecpt, &pt)) return (0);
474 if (ensurebuf(me, EC_ATINF(&pt) ? 2 :
475 6 + mp_octets(pt.x) + mp_octets(pt.y)))
477 buf_putec(BUF_B(me), &pt); assert(BOK(BUF_B(me)));
482 static PyObject *wbmeth_putecptraw(PyObject *me, PyObject *arg)
487 if (!PyArg_ParseTuple(arg, "O!:putecptraw", ecptcurve_pytype, &ptobj))
490 EC_OUT(cc, &pt, ECPT_P(ptobj));
491 if (ensurebuf(me, 2*cc->f->noctets + 1)) return (0);
492 ec_putraw(cc, BUF_B(me), &pt); assert(BOK(BUF_B(me)));
497 static PyObject *wbmeth_ec2osp(PyObject *me, PyObject *arg, PyObject *kw)
502 unsigned f = EC_EXPLY;
503 static const char *const kwlist[] = { "point", "flags", 0 };
504 if (!PyArg_ParseTupleAndKeywords(arg, kw, "O!|O&:os2ecp", KWLIST,
505 ecptcurve_pytype, &ptobj, convuint, &f))
508 EC_OUT(cc, &pt, ECPT_P(ptobj));
509 if (ensurebuf(me, 2*cc->f->noctets + 1)) return (0);
510 if (ec_ec2osp(cc, f, BUF_B(me), &pt)) VALERR("invalid flags");
516 static PyObject *wbmeth_putge(PyObject *me, PyObject *arg)
519 if (!PyArg_ParseTuple(arg, "O!:putge", ge_pytype, &geobj)) return (0);
520 if (ensurebuf(me, GE_G(geobj)->noctets)) return (0);
521 G_TOBUF(GE_G(geobj), BUF_B(me), GE_X(geobj)); assert(BOK(BUF_B(me)));
525 static PyObject *wbmeth_putgeraw(PyObject *me, PyObject *arg)
528 if (!PyArg_ParseTuple(arg, "O!:putgeraw", ge_pytype, &geobj)) return (0);
529 if (ensurebuf(me, GE_G(geobj)->noctets)) return (0);
530 G_TORAW(GE_G(geobj), BUF_B(me), GE_X(geobj)); assert(BOK(BUF_B(me)));
534 static PyObject *wbget_size(PyObject *me, void *hunoz)
535 { return (PyInt_FromLong(BLEN(BUF_B(me)))); }
537 static PyObject *wbget_contents(PyObject *me, void *hunoz)
538 { return (bytestring_pywrap(BBASE(BUF_B(me)), BLEN(BUF_B(me)))); }
540 static const PyGetSetDef wbuf_pygetset[] = {
541 #define GETSETNAME(op, name) wb##op##_##name
542 GET (size, "WBUF.size -> SIZE")
543 GET (contents, "WBUF.contents -> STR")
548 static const PyMethodDef wbuf_pymethods[] = {
549 #define METHNAME(func) wbmeth_##func
550 METH (zero, "WBUF.zero(N)")
551 METH (put, "WBUF.put(BYTES)")
552 #define WBMETH_DECL_PUTU_(n, W, w) \
553 METH(putu##w, "WBUF.putu" #w "(INT)")
554 DOUINTCONV(WBMETH_DECL_PUTU_)
555 #define WBMETH_DECL_PUTBLK_(n, W, w) \
556 METH(putblk##w, "WBUF.putblk" #w "(BYTES)")
557 BUF_DOSUFFIXES(WBMETH_DECL_PUTBLK_)
558 METH (putmp, "WBUF.putmp(X)")
559 METH (putgf, "WBUF.putgf(X)")
560 METH (putecpt, "WBUF.putecpt(P)")
561 METH (putecptraw, "WBUF.putecptraw(P)")
562 KWMETH(ec2osp, "WBUF.ec2osp(P, [flags = EC_EXPLY])")
563 METH (putge, "WBUF.putge(X)")
564 METH (putgeraw, "WBUF.putgeraw(X)")
569 static const PyBufferProcs wbuf_pybuffer = {
571 wbuf_pygetbuf, /* @bf_getbuffer@ */
572 wbuf_pyrlsbuf /* @bf_releasebuffer@ */
574 wbuf_pyreadbuf, /* @bf_getreadbuffer@ */
575 0, /* @bf_getwritebuffer@ */
576 wbuf_pysegcount, /* @bf_getsegcount@ */
577 0 /* @bf_getcharbuffer@ */
581 static const PyTypeObject wbuf_pytype_skel = {
582 PyVarObject_HEAD_INIT(0, 0) /* Header */
583 "WriteBuffer", /* @tp_name@ */
584 sizeof(buf_pyobj), /* @tp_basicsize@ */
585 0, /* @tp_itemsize@ */
587 buf_pydealloc, /* @tp_dealloc@ */
589 0, /* @tp_getattr@ */
590 0, /* @tp_setattr@ */
591 0, /* @tp_compare@ */
593 0, /* @tp_as_number@ */
594 0, /* @tp_as_sequence@ */
595 0, /* @tp_as_mapping@ */
599 0, /* @tp_getattro@ */
600 0, /* @tp_setattro@ */
601 PYBUFFER(wbuf), /* @tp_as_buffer@ */
602 Py_TPFLAGS_DEFAULT | /* @tp_flags@ */
606 "WriteBuffer([size = ?]): a write buffer.",
608 0, /* @tp_traverse@ */
610 0, /* @tp_richcompare@ */
611 0, /* @tp_weaklistoffset@ */
613 0, /* @tp_iternext@ */
614 PYMETHODS(wbuf), /* @tp_methods@ */
615 0, /* @tp_members@ */
616 PYGETSET(wbuf), /* @tp_getset@ */
619 0, /* @tp_descr_get@ */
620 0, /* @tp_descr_set@ */
621 0, /* @tp_dictoffset@ */
623 PyType_GenericAlloc, /* @tp_alloc@ */
624 wbuf_pynew, /* @tp_new@ */
629 /*----- Initialization ----------------------------------------------------*/
633 void buffer_pyinit(void)
635 INITTYPE(rbuf, root);
636 INITTYPE(wbuf, root);
639 void buffer_pyinsert(PyObject *mod)
641 INSEXC("BufferError", buferr, PyExc_Exception, 0);
642 INSERT("ReadBuffer", rbuf_pytype);
643 INSERT("WriteBuffer", wbuf_pytype);
646 /*----- That's all, folks -------------------------------------------------*/