1 

     2 /* Wrap void* pointers to be passed between C modules */

     3 

     4 #include "Python.h"

     5 

     6 

     7 /* Declarations for objects of type PyCObject */

     8 

     9 typedef void (*destructor1)(void *);

    10 typedef void (*destructor2)(void *, void*);

    11 

    12 PyObject *

    13 PyCObject_FromVoidPtr(void *cobj, void (*destr)(void *))

    14 {

    15     PyCObject *self;

    16 

    17     self = PyObject_NEW(PyCObject, &PyCObject_Type);

    18     if (self == NULL)

    19         return NULL;

    20     self->cobject=cobj;

    21     self->destructor=destr;

    22     self->desc=NULL;

    23 

    24     return (PyObject *)self;

    25 }

    26 

    27 PyObject *

    28 PyCObject_FromVoidPtrAndDesc(void *cobj, void *desc,

    29                              void (*destr)(void *, void *))

    30 {

    31     PyCObject *self;

    32 

    33     if (!desc) {

    34         PyErr_SetString(PyExc_TypeError,

    35                         "PyCObject_FromVoidPtrAndDesc called with null"

    36                         " description");

    37         return NULL;

    38     }

    39     self = PyObject_NEW(PyCObject, &PyCObject_Type);

    40     if (self == NULL)

    41         return NULL;

    42     self->cobject = cobj;

    43     self->destructor = (destructor1)destr;

    44     self->desc = desc;

    45 

    46     return (PyObject *)self;

    47 }

    48 

    49 void *

    50 PyCObject_AsVoidPtr(PyObject *self)

    51 {

    52     if (self) {

    53         if (self->ob_type == &PyCObject_Type)

    54             return ((PyCObject *)self)->cobject;

    55         PyErr_SetString(PyExc_TypeError,

    56                         "PyCObject_AsVoidPtr with non-C-object");

    57     }

    58     if (!PyErr_Occurred())

    59         PyErr_SetString(PyExc_TypeError,

    60                         "PyCObject_AsVoidPtr called with null pointer");

    61     return NULL;

    62 }

    63 

    64 void *

    65 PyCObject_GetDesc(PyObject *self)

    66 {

    67     if (self) {

    68         if (self->ob_type == &PyCObject_Type)

    69             return ((PyCObject *)self)->desc;

    70         PyErr_SetString(PyExc_TypeError,

    71                         "PyCObject_GetDesc with non-C-object");

    72     }

    73     if (!PyErr_Occurred())

    74         PyErr_SetString(PyExc_TypeError,

    75                         "PyCObject_GetDesc called with null pointer");

    76     return NULL;

    77 }

    78 

    79 void *

    80 PyCObject_Import(char *module_name, char *name)

    81 {

    82     PyObject *m, *c;

    83     void *r = NULL;

    84 

    85     if ((m = PyImport_ImportModule(module_name))) {

    86         if ((c = PyObject_GetAttrString(m,name))) {

    87             r = PyCObject_AsVoidPtr(c);

    88             Py_DECREF(c);

    89 	}

    90         Py_DECREF(m);

    91     }

    92     return r;

    93 }

    94 

    95 int

    96 PyCObject_SetVoidPtr(PyObject *self, void *cobj)

    97 {

    98     PyCObject* cself = (PyCObject*)self;

    99     if (cself == NULL || !PyCObject_Check(cself) ||

   100 	cself->destructor != NULL) {

   101 	PyErr_SetString(PyExc_TypeError, 

   102 			"Invalid call to PyCObject_SetVoidPtr");

   103 	return 0;

   104     }

   105     cself->cobject = cobj;

   106     return 1;

   107 }

   108 

   109 static void

   110 PyCObject_dealloc(PyCObject *self)

   111 {

   112     if (self->destructor) {

   113         if(self->desc)

   114             ((destructor2)(self->destructor))(self->cobject, self->desc);

   115         else

   116             (self->destructor)(self->cobject);

   117     }

   118     PyObject_DEL(self);

   119 }

   120 

   121 

   122 PyDoc_STRVAR(PyCObject_Type__doc__,

   123 "C objects to be exported from one extension module to another\n\

   124 \n\

   125 C objects are used for communication between extension modules.  They\n\

   126 provide a way for an extension module to export a C interface to other\n\

   127 extension modules, so that extension modules can use the Python import\n\

   128 mechanism to link to one another.");

   129 

   130 PyTypeObject PyCObject_Type = {

   131     PyVarObject_HEAD_INIT(&PyType_Type, 0)

   132     "PyCObject",		/*tp_name*/

   133     sizeof(PyCObject),		/*tp_basicsize*/

   134     0,				/*tp_itemsize*/

   135     /* methods */

   136     (destructor)PyCObject_dealloc, /*tp_dealloc*/

   137     0,				/*tp_print*/

   138     0,				/*tp_getattr*/

   139     0,				/*tp_setattr*/

   140     0,				/*tp_compare*/

   141     0,				/*tp_repr*/

   142     0,				/*tp_as_number*/

   143     0,				/*tp_as_sequence*/

   144     0,				/*tp_as_mapping*/

   145     0,				/*tp_hash*/

   146     0,				/*tp_call*/

   147     0,				/*tp_str*/

   148     0,				/*tp_getattro*/

   149     0,				/*tp_setattro*/

   150     0,				/*tp_as_buffer*/

   151     0,				/*tp_flags*/

   152     PyCObject_Type__doc__	/*tp_doc*/

   153 };