chiark - git - mdw - pyke/blob - pyke.h

   1 /* -*-c-*-
   2  *
   3  * Pyke: the Python Kit for Extensions
   4  *
   5  * (c) 2019 Straylight/Edgeware
   6  */
   7
   8 /*----- Licensing notice --------------------------------------------------*
   9  *
  10  * This file is part of Pyke: the Python Kit for Extensions.
  11  *
  12  * Pyke is free software: you can redistribute it and/or modify it under
  13  * the terms of the GNU General Public License as published by the Free
  14  * Software Foundation; either version 2 of the License, or (at your
  15  * option) any later version.
  16  *
  17  * Pyke is distributed in the hope that it will be useful, but WITHOUT
  18  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  19  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  20  * for more details.
  21  *
  22  * You should have received a copy of the GNU General Public License
  23  * along with Pyke.  If not, write to the Free Software Foundation, Inc.,
  24  * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  25  */
  26
  27 #ifndef PYKE_H
  28 #define PYKE_H
  29
  30 #ifdef __cplusplus
  31   extern "C" {
  32 #endif
  33
  34 /*----- Header files ------------------------------------------------------*/
  35
  36 #define PY_SSIZE_T_CLEAN
  37
  38 #include <Python.h>
  39 #include <structmember.h>
  40
  41 /*----- Other preliminaries -----------------------------------------------*/
  42
  43 #define NOTHING
  44 #define COMMA ,
  45
  46 /*----- Symbol visibility -------------------------------------------------*
  47  *
  48  * This library is very messy regarding symbol namespace.  Keep this mess
  49  * within our shared-object.
  50  */
  51
  52 #define GOBBLE_SEMI extern int notexist
  53 #if defined(__GNUC__) && defined(__ELF__)
  54 #  define PRIVATE_SYMBOLS _Pragma("GCC visibility push(hidden)") GOBBLE_SEMI
  55 #  define PUBLIC_SYMBOLS _Pragma("GCC visibility pop") GOBBLE_SEMI
  56 #  define EXPORT __attribute__((__visibility__("default")))
  57 #else
  58 #  define PRIVATE_SYMBOLS GOBBLE_SEMI
  59 #  define PUBLIC_SYMBOLS GOBBLE_SEMI
  60 #  define EXPORT
  61 #endif
  62
  63 PRIVATE_SYMBOLS;
  64
  65 /*----- Utilities for returning values and exceptions ---------------------*/
  66
  67 /* Returning values. */
  68 #define RETURN_OBJ(obj) do { Py_INCREF(obj); return (obj); } while (0)
  69 #define RETURN_NONE RETURN_OBJ(Py_None)
  70 #define RETURN_NOTIMPL RETURN_OBJ(Py_NotImplemented)
  71 #define RETURN_TRUE RETURN_OBJ(Py_True)
  72 #define RETURN_FALSE RETURN_OBJ(Py_False)
  73 #define RETURN_ME RETURN_OBJ(me)
  74
  75 /* Returning exceptions.  (Note that `KeyError' is `MAPERR' here, because
  76  * Catacomb has its own kind of `KeyError'.)
  77  */
  78 #define EXCERR(exc, str) do {                                           \
  79   PyErr_SetString(exc, str);                                            \
  80   goto end;                                                             \
  81 } while (0)
  82 #define VALERR(str) EXCERR(PyExc_ValueError, str)
  83 #define OVFERR(str) EXCERR(PyExc_OverflowError, str)
  84 #define TYERR(str) EXCERR(PyExc_TypeError, str)
  85 #define IXERR(str) EXCERR(PyExc_IndexError, str)
  86 #define ZDIVERR(str) EXCERR(PyExc_ZeroDivisionError, str)
  87 #define SYSERR(str) EXCERR(PyExc_SystemError, str)
  88 #define NIERR(str) EXCERR(PyExc_NotImplementedError, str)
  89 #define MAPERR(idx) do {                                                \
  90   PyErr_SetObject(PyExc_KeyError, idx);                                 \
  91   goto end;                                                             \
  92 } while (0)
  93 #define OSERR(name) do {                                                \
  94   PyErr_SetFromErrnoWithFilename(PyExc_OSError, name);                  \
  95   goto end;                                                             \
  96 } while (0)
  97
  98 /* Saving and restoring exceptions. */
  99 struct excinfo { PyObject *ty, *val, *tb; };
 100 #define EXCINFO_INIT { 0, 0, 0 }
 101 #define INIT_EXCINFO(exc) do {                                          \
 102   struct excinfo *_exc = (exc); _exc->ty = _exc->val = _exc->tb = 0;    \
 103 } while (0)
 104 #define RELEASE_EXCINFO(exc) do {                                       \
 105   struct excinfo *_exc = (exc);                                         \
 106   Py_XDECREF(_exc->ty);  _exc->ty  = 0;                                 \
 107   Py_XDECREF(_exc->val); _exc->val = 0;                                 \
 108   Py_XDECREF(_exc->tb);  _exc->tb  = 0;                                 \
 109 } while (0)
 110 #define STASH_EXCINFO(exc) do {                                         \
 111   struct excinfo *_exc = (exc);                                         \
 112   PyErr_Fetch(&_exc->ty, &_exc->val, &_exc->tb);                        \
 113   PyErr_NormalizeException(&_exc->ty, &_exc->val, &_exc->tb);           \
 114 } while (0)
 115 #define RESTORE_EXCINFO(exc) do {                                       \
 116   struct excinfo *_exc = (exc);                                         \
 117   PyErr_Restore(_exc->ty, _exc->val, _exc->tb);                         \
 118   _exc->ty = _exc->val = _exc->tb = 0;                                  \
 119 } while (0)
 120 extern void report_lost_exception(struct excinfo *, const char *, ...);
 121 extern void report_lost_exception_v(struct excinfo *, const char *, va_list);
 122 extern void stash_exception(struct excinfo *, const char *, ...);
 123 extern void restore_exception(struct excinfo *, const char *, ...);
 124
 125 /*----- Conversions -------------------------------------------------------*/
 126
 127 /* Define an input conversion (`O&') function: check that the object has
 128  * Python type TY, and extract a C pointer to CTY by calling EXT on the
 129  * object (which may well be a macro).
 130  */
 131 #define CONVFUNC(ty, cty, ext)                                          \
 132   int conv##ty(PyObject *o, void *p)                                    \
 133   {                                                                     \
 134     if (!PyObject_TypeCheck(o, ty##_pytype))                            \
 135       TYERR("wanted a " #ty);                                           \
 136     *(cty *)p = ext(o);                                                 \
 137     return (1);                                                         \
 138   end:                                                                  \
 139     return (0);                                                         \
 140   }
 141
 142 /* Input conversion functions for standard kinds of objects, with overflow
 143  * checking where applicable.
 144  */
 145 extern int convulong(PyObject *, void *); /* unsigned long */
 146 extern int convuint(PyObject *, void *); /* unsigned int */
 147 extern int convszt(PyObject *, void *); /* size_t */
 148 extern int convbool(PyObject *, void *); /* bool */
 149
 150 /* Output conversions. */
 151 extern PyObject *getbool(int);          /* bool */
 152 extern PyObject *getulong(unsigned long); /* any kind of unsigned integer */
 153
 154 /*----- Miscellaneous utilities -------------------------------------------*/
 155
 156 #define FREEOBJ(obj)                                                    \
 157   (((PyObject *)(obj))->ob_type->tp_free((PyObject *)(obj)))
 158   /* Actually free OBJ, e.g., in a deallocation function. */
 159
 160 extern PyObject *abstract_pynew(PyTypeObject *, PyObject *, PyObject *);
 161   /* A `tp_new' function which refuses to make the object. */
 162
 163 #ifndef CONVERT_CAREFULLY
 164 #  define CONVERT_CAREFULLY(newty, expty, obj)                          \
 165      (!sizeof(*(expty *)0 = (obj)) + (/*unconst*/ newty)(obj))
 166   /* Convert OBJ to the type NEWTY, having previously checked that it is
 167    * convertible to the expected type EXPTY.
 168    *
 169    * Because of the way we set up types, we can make many kinds of tables be
 170    * `const' which can't usually be so (because Python will want to fiddle
 171    * with their reference counts); and, besides, Python's internals are
 172    * generally quite bad at being `const'-correct about tables.  One frequent
 173    * application of this macro, then, is in removing `const' from a type
 174    * without sacrificing all type safety.  The other common use is in
 175    * checking that method function types match up with the signatures
 176    * expected in their method definitions.
 177    */
 178 #endif
 179
 180 #define KWLIST CONVERT_CAREFULLY(char **, const char *const *, kwlist)
 181   /* Strip `const' qualifiers from the keyword list `kwlist'.  Useful when
 182    * calling `PyArg_ParseTupleAndKeywords', which isn't `const'-correct.
 183    */
 184
 185 /*----- Type definitions --------------------------------------------------*
 186  *
 187  * Pyke types are defined in a rather unusual way.
 188  *
 189  * The main code defines a `type skeleton' of type `PyTypeObject',
 190  * conventionally named `TY_pytype_skel'.  Unlike typical Python type
 191  * definitions in extensions, this can (and should) be read-only.  Also,
 192  * there's no point in setting the `tp_base' pointer here, because the actual
 193  * runtime base type object won't, in general, be known at compile time.
 194  * Instead, the type skeletons are converted into Python `heap types' by the
 195  * `INITTYPE' macro.  The main difference is that Python code can add
 196  * attributes to heap types, and we make extensive use of this ability.
 197  */
 198
 199 extern void *newtype(PyTypeObject */*meta*/,
 200                      const PyTypeObject */*skel*/, const char */*name*/);
 201   /* Make and return a new Python type object, of type META (typically
 202    * `PyType_Type', but may be a subclass), filled in from the skeleton SKEL
 203    * (null to inherit everything), and named NAME.  The caller can mess with
 204    * the type object further at this time: call `typeready' when it's set up
 205    * properly.
 206    */
 207
 208 extern void typeready(PyTypeObject *);
 209   /* The type object is now ready to be used. */
 210
 211 extern PyTypeObject *inittype(PyTypeObject */*skel*/,
 212                               PyTypeObject */*meta*/);
 213   /* All-in-one function to construct a working type from a type skeleton
 214    * SKEL, with metaclass META.  The caller is expected to have filled in the
 215    * direct superclass in SKEL->tp_base.
 216    */
 217
 218 /* Alias for built-in types, to fit in with Pyke naming conventions. */
 219 #define root_pytype 0
 220 #define type_pytype &PyType_Type
 221
 222 #define INITTYPE_META(ty, base, meta) do {                              \
 223   ty##_pytype_skel.tp_base = base##_pytype;                             \
 224   ty##_pytype = inittype(&ty##_pytype_skel, meta##_pytype);             \
 225 } while (0)
 226 #define INITTYPE(ty, base) INITTYPE_META(ty, base, type)
 227   /* Macros to initialize a type from its skeleton. */
 228
 229 /* Macros for filling in `PyMethodDef' tables, ensuring that functions have
 230  * the expected signatures.
 231  */
 232 #define STD_METHOD(decor, func, flags, doc)                             \
 233   { #func, decor(func), METH_VARARGS | flags, doc },
 234 #define KEYWORD_METHOD(decor, func, flags, doc)                         \
 235   { #func,                                                              \
 236     CONVERT_CAREFULLY(PyCFunction, PyCFunctionWithKeywords, decor(func)), \
 237     METH_VARARGS | METH_KEYWORDS | flags,                               \
 238     doc },
 239
 240 /* Convenience wrappers for filling in `PyMethodDef' tables, following
 241  * Pyke naming convention.  Define `METHNAME' locally as
 242  *
 243  *      #define METHNAME(name) foometh_##func
 244  *
 245  * around the method table.
 246  */
 247 #define METH(func, doc) STD_METHOD(METHNAME, func, 0, doc)
 248 #define KWMETH(func, doc) KEYWORD_METHOD(METHNAME, func, 0, doc)
 249 #define CMTH(func, doc) STD_METHOD(METHNAME, func, METH_CLASS, doc)
 250 #define KWCMTH(func, doc) KEYWORD_METHOD(METHNAME, func, METH_CLASS, doc)
 251 #define SMTH(func, doc) STD_METHOD(METHNAME, func, METH_STATIC, doc)
 252 #define KWSMTH(func, doc) KEYWORD_METHOD(METHNAME, func, METH_STATIC, doc)
 253
 254 /* Convenience wrappers for filling in `PyGetSetDef' tables, following Pyke
 255  * naming convention.  Define `GETSETNAME' locally as
 256  *
 257  *      #define GETSETNAME(op, name) foo##op##_##func
 258  *
 259  * around the get/set table.
 260  */
 261 #define GET(func, doc)                                                  \
 262   { #func, GETSETNAME(get, func), 0, doc },
 263 #define GETSET(func, doc)                                               \
 264   { #func, GETSETNAME(get, func), GETSETNAME(set, func), doc },
 265
 266 /* Convenience wrapper for filling in `PyMemberDef' tables.  Define
 267  * `MEMBERSTRUCT' locally as
 268  *
 269  *      #define MEMBERSTRUCT foo_pyobj
 270  *
 271  * around the member table.
 272  */
 273 #define MEMBER(name, ty, f, doc)                                        \
 274   { #name, ty, offsetof(MEMBERSTRUCT, name), f, doc },
 275
 276 /* Wrappers for filling in pointers in a `PyTypeObject' structure, (a)
 277  * following Pyke naming convention, and (b) stripping `const' from the types
 278  * without losing type safety.
 279  */
 280 #define UNCONST_TYPE_SLOT(type, suffix, op, ty)                         \
 281   CONVERT_CAREFULLY(type *, const type *, op ty##_py##suffix)
 282 #define PYGETSET(ty) UNCONST_TYPE_SLOT(PyGetSetDef, getset, NOTHING, ty)
 283 #define PYMETHODS(ty) UNCONST_TYPE_SLOT(PyMethodDef, methods, NOTHING, ty)
 284 #define PYMEMBERS(ty) UNCONST_TYPE_SLOT(PyMemberDef, members, NOTHING, ty)
 285 #define PYNUMBER(ty) UNCONST_TYPE_SLOT(PyNumberMethods, number, &, ty)
 286 #define PYSEQUENCE(ty) UNCONST_TYPE_SLOT(PySequenceMethods, sequence, &, ty)
 287 #define PYMAPPING(ty) UNCONST_TYPE_SLOT(PyMappingMethods, mapping, &, ty)
 288 #define PYBUFFER(ty) UNCONST_TYPE_SLOT(PyBufferProcs, buffer, &, ty)
 289
 290 /*----- Populating modules ------------------------------------------------*/
 291
 292 extern PyObject *modname;
 293   /* The overall module name.  Set this with `PyString_FromString'. */
 294
 295 extern PyObject *home_module;
 296   /* The overall module object. */
 297
 298 extern PyObject *mkexc(PyObject */*mod*/, PyObject */*base*/,
 299                        const char */*name*/, const PyMethodDef */*methods*/);
 300   /* Make and return an exception class called NAME, which will end up in
 301    * module MOD (though it is not added at this time).  The new class is a
 302    * subclass of BASE.  Attach the METHODS to it.
 303    */
 304
 305 #define INSERT(name, ob) do {                                           \
 306   PyObject *_o = (PyObject *)(ob);                                      \
 307   Py_INCREF(_o);                                                        \
 308   PyModule_AddObject(mod, name, _o);                                    \
 309 } while (0)
 310   /* Insert a Python object OB into the module `mod' under the given NAME. */
 311
 312 /* Numeric constants. */
 313 struct nameval { const char *name; unsigned f; unsigned long value; };
 314 #define CF_SIGNED 1u
 315 extern void setconstants(PyObject *, const struct nameval *);
 316
 317 #define INSEXC(name, var, base, meth)                                   \
 318   INSERT(name, var = mkexc(mod, base, name, meth))
 319   /* Insert an exception class into the module `mod'; other arguments are as
 320    * for `mkexc'.
 321    */
 322
 323 /*----- Submodules --------------------------------------------------------*
 324  *
 325  * It's useful to split the Python module up into multiple source files, and
 326  * have each one contribute its definitions into the main module.
 327  *
 328  * Define a list-macro `MODULES' in the master header file naming the
 329  * submodules to be processed, and run
 330  *
 331  *      MODULES(DECLARE_MODINIT)
 332  *
 333  * to declare the interface functions.
 334  *
 335  * Each submodule FOO defines two functions: `FOO_pyinit' initializes types
 336  * (see `INITTYPE' above) and accumulates methods (`addmethods' below), while
 337  * `FOO_pyinsert' populates the module with additional definitions
 338  * (especially types, though also constants).
 339  *
 340  * The top-level module initialization should call `INIT_MODULES' before
 341  * creating the Python module, and `INSERT_MODULES' afterwards to make
 342  * everything work.
 343  */
 344
 345 extern void addmethods(const PyMethodDef *);
 346 extern PyMethodDef *donemethods(void);
 347   /* Accumulate method-table fragments, and return the combined table of all
 348    * of the fragments.
 349    */
 350
 351 #define DECLARE_MODINIT(m)                                              \
 352   extern void m##_pyinit(void);                                         \
 353   extern void m##_pyinsert(PyObject *);
 354   /* Declare submodule interface functions. */
 355
 356 #define DOMODINIT(m) m##_pyinit();
 357 #define DOMODINSERT(m) m##_pyinsert(mod);
 358 #define INIT_MODULES do { MODULES(DOMODINIT) } while (0)
 359 #define INSERT_MODULES do { MODULES(DOMODINSERT) } while (0)
 360   /* Top-level dispatch to the various submodules. */
 361
 362 /*----- Generic mapping support -------------------------------------------*/
 363
 364 /* Mapping methods. */
 365 #define GMAP_METMNAME(func) gmapmeth_##func
 366 #define GMAP_METH(func, doc) STD_METHOD(GMAP_METMNAME, func, 0, doc)
 367 #define GMAP_KWMETH(func, doc) KEYWORD_METHOD(GMAP_METMNAME, func, 0, doc)
 368 #define GMAP_METHDECL(func, doc)                                        \
 369   extern PyObject *gmapmeth_##func(PyObject *, PyObject *);
 370 #define GMAP_KWMETHDECL(func, doc)                                      \
 371   extern PyObject *gmapmeth_##func(PyObject *, PyObject *, PyObject *);
 372
 373 #define GMAP_DOROMETHODS(METH, KWMETH)                                  \
 374   METH  (has_key,       "D.has_key(KEY) -> BOOL")                       \
 375   METH  (keys,          "D.keys() -> LIST")                             \
 376   METH  (values,        "D.values() -> LIST")                           \
 377   METH  (items,         "D.items() -> LIST")                            \
 378   METH  (iterkeys,      "D.iterkeys() -> ITER")                         \
 379   METH  (itervalues,    "D.itervalues() -> ITER")                       \
 380   METH  (iteritems,     "D.iteritems() -> ITER")                        \
 381   KWMETH(get,           "D.get(KEY, [default = None]) -> VALUE")
 382
 383 #define GMAP_DOMETHODS(METH, KWMETH)                                    \
 384   GMAP_DOROMETHODS(METH, KWMETH)                                        \
 385   METH  (clear,         "D.clear()")                                    \
 386   KWMETH(setdefault,    "D.setdefault(K, [default = None]) -> VALUE")   \
 387   KWMETH(pop,           "D.pop(KEY, [default = <error>]) -> VALUE")     \
 388   METH  (popitem,       "D.popitem() -> (KEY, VALUE)")                  \
 389   METH  (update,        "D.update(MAP)")
 390
 391 GMAP_DOMETHODS(GMAP_METHDECL, GMAP_KWMETHDECL)
 392 #define GMAP_ROMETHODS GMAP_DOROMETHODS(GMAP_METH, GMAP_KWMETH)
 393 #define GMAP_METHODS GMAP_DOMETHODS(GMAP_METH, GMAP_KWMETH)
 394
 395 /* Mapping protocol implementation. */
 396 extern Py_ssize_t gmap_pysize(PyObject *); /* for `mp_length' */
 397 extern const PySequenceMethods gmap_pysequence; /* for `tp_as_sequence' */
 398 extern const PyMethodDef gmap_pymethods[]; /* all the standard methods */
 399
 400 /*----- That's all, folks -------------------------------------------------*/
 401
 402 #ifdef __cplusplus
 403   }
 404 #endif
 405
 406 #endif