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

   1 ### -*-pyrex-*-
   2 ###
   3 ### Generic encoder/decoder
   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 ###--------------------------------------------------------------------------
  27 ### Base classes.
  28
  29 cdef extern from 'mLib/codec.h':
  30
  31   ctypedef struct codec
  32
  33   ctypedef struct codec_class:
  34     char *name
  35     codec *(*encoder)(unsigned f, char *ind, unsigned max)
  36     codec *(*decoder)(unsigned f)
  37
  38   ctypedef struct codec_ops:
  39     codec_class *c
  40     int (*code)(codec *c, void *p, size_t, dstr *d)
  41     void (*destroy)(codec *c)
  42   ctypedef struct codec:
  43     codec_ops *ops
  44
  45   enum:
  46     CDCF_LOWERC
  47     CDCF_IGNCASE
  48     CDCF_NOEQPAD
  49     CDCF_IGNEQPAD
  50     CDCF_IGNEQMID
  51     CDCF_IGNZPAD
  52     CDCF_IGNNEWL
  53     CDCF_IGNSPC
  54     CDCF_IGNINVCH
  55     CDCF_IGNJUNK
  56
  57   enum:
  58     CDCERR_OK
  59     CDCERR_INVCH
  60     CDCERR_INVEQPAD
  61     CDCERR_INVZPAD
  62
  63   char *_codec_strerror "codec_strerror"(int err)
  64
  65 class CDCF:
  66   LOWERC = CDCF_LOWERC
  67   IGNCASE = CDCF_IGNCASE
  68   NOEQPAD = CDCF_NOEQPAD
  69   IGNEQPAD = CDCF_IGNEQPAD
  70   IGNEQMID = CDCF_IGNEQMID
  71   IGNZPAD = CDCF_IGNZPAD
  72   IGNNEWL = CDCF_IGNNEWL
  73   IGNSPC = CDCF_IGNSPC
  74   IGNINVCH = CDCF_IGNINVCH
  75   IGNJUNK = CDCF_IGNJUNK
  76
  77 class CDCERR:
  78   OK = CDCERR_OK
  79   INVCH = CDCERR_INVCH
  80   INVEQPAD = CDCERR_INVEQPAD
  81   INVZPAD = CDCERR_INVZPAD
  82
  83 class CodecError (Exception):
  84   def __init__(me, err):
  85     me.err = err
  86     me.msg = _codec_strerror(err)
  87   def __str__(me):
  88     return me.msg
  89
  90 def codec_strerror(err):
  91   return _codec_strerror(err)
  92
  93 cdef int code(codec *c, void *p, size_t len, dstr *d) except -1:
  94   cdef int err
  95   err = c.ops.code(c, p, len, d)
  96   if err:
  97     raise CodecError(err)
  98   return 0
  99
 100 cdef class _BaseCodec:
 101   cdef codec *c
 102   def __cinit__(me, *hunoz, **hukairz):
 103     me.c = NULL
 104   def __init__(me, *hunoz, **hukairz):
 105     raise TypeError, 'abstract class'
 106   def __dealloc__(me):
 107     if me.c is not NULL:
 108       me.c.ops.destroy(me.c)
 109   cdef code(me, text, int finishp):
 110     cdef void *p
 111     cdef Py_ssize_t len
 112     cdef dstr d
 113     cdef int err
 114     if me.c is NULL:
 115       raise ValueError, 'Encoding finished'
 116     DCREATE(&d)
 117     try:
 118       PyObject_AsReadBuffer(text, <cvp *>&p, &len)
 119       code(me.c, p, len, &d)
 120       if finishp:
 121         code(me.c, NULL, 0, &d)
 122         me.c.ops.destroy(me.c)
 123         me.c = NULL
 124       return PyString_FromStringAndSize(d.buf, d.len)
 125     finally:
 126       dstr_destroy(&d)
 127   def done(me):
 128     me.code('', True)
 129
 130 cdef class _BaseEncoder (_BaseCodec):
 131   def encode(me, text, finishp = False):
 132     return me.code(text, finishp)
 133
 134 cdef class _BaseDecoder (_BaseCodec):
 135   def decode(me, text, finishp = False):
 136     return me.code(text, finishp)
 137
 138 ###--------------------------------------------------------------------------
 139 ### Base64.
 140
 141 cdef extern from 'mLib/base64.h':
 142   codec_class base64_class
 143   codec_class file64_class
 144   codec_class base64url_class
 145
 146 cdef class Base64Encoder (_BaseEncoder):
 147   def __init__(me, indent = '\n', maxline = 72, flags = CDCF_IGNJUNK):
 148     me.c = base64_class.encoder(flags, indent, maxline)
 149
 150 cdef class Base64Decoder (_BaseDecoder):
 151   def __init__(me, flags = CDCF_IGNJUNK):
 152     me.c = base64_class.decoder(flags)
 153
 154 cdef class File64Encoder (_BaseEncoder):
 155   def __init__(me, indent = '\n', maxline = 72, flags = CDCF_IGNJUNK):
 156     me.c = file64_class.encoder(flags, indent, maxline)
 157
 158 cdef class File64Decoder (_BaseDecoder):
 159   def __init__(me, flags = CDCF_IGNJUNK):
 160     me.c = file64_class.decoder(flags)
 161
 162 cdef class Base64URLEncoder (_BaseEncoder):
 163   def __init__(me, indent = '\n', maxline = 72, flags = CDCF_IGNJUNK):
 164     me.c = base64url_class.encoder(flags, indent, maxline)
 165
 166 cdef class Base64URLDecoder (_BaseDecoder):
 167   def __init__(me, flags = CDCF_IGNJUNK):
 168     me.c = base64url_class.decoder(flags)
 169
 170 ###--------------------------------------------------------------------------
 171 ### Base32.
 172
 173 cdef extern from 'mLib/base32.h':
 174   codec_class base32_class
 175   codec_class base32hex_class
 176
 177 cdef class Base32Encoder (_BaseEncoder):
 178   def __init__(me, indent = '\n', maxline = 72, flags = CDCF_IGNJUNK):
 179     me.c = base32_class.encoder(flags, indent, maxline)
 180
 181 cdef class Base32Decoder (_BaseDecoder):
 182   def __init__(me, flags = CDCF_IGNJUNK):
 183     me.c = base32_class.decoder(flags)
 184
 185 cdef class Base32HexEncoder (_BaseEncoder):
 186   def __init__(me, indent = '\n', maxline = 72, flags = CDCF_IGNJUNK):
 187     me.c = base32hex_class.encoder(flags, indent, maxline)
 188
 189 cdef class Base32HexDecoder (_BaseDecoder):
 190   def __init__(me, flags = CDCF_IGNJUNK):
 191     me.c = base32hex_class.decoder(flags)
 192
 193 ###--------------------------------------------------------------------------
 194 ### Hex.
 195
 196 cdef extern from 'mLib/hex.h':
 197   codec_class hex_class
 198
 199 cdef class HexEncoder (_BaseEncoder):
 200   def __init__(me, indent = '\n', maxline = 72,
 201                flags = CDCF_IGNJUNK | CDCF_LOWERC):
 202     me.c = hex_class.encoder(flags, indent, maxline)
 203
 204 cdef class HexDecoder (_BaseDecoder):
 205   def __init__(me, flags = CDCF_IGNJUNK | CDCF_LOWERC):
 206     me.c = hex_class.decoder(flags)
 207
 208 ###----- That's all, folks --------------------------------------------------