lib/talloc/pytalloc.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Python/Talloc glue
   4    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2008
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include <Python.h>
  21 #include "replace.h"
  22 #include <talloc.h>
  23 #include "pytalloc.h"
  24
  25 /**
  26  * Simple dealloc for talloc-wrapping PyObjects
  27  */
  28 void py_talloc_dealloc(PyObject* self)
  29 {
  30         py_talloc_Object *obj = (py_talloc_Object *)self;
  31         talloc_free(obj->talloc_ctx);
  32         obj->talloc_ctx = NULL;
  33         self->ob_type->tp_free(self);
  34 }
  35
  36 /**
  37  * Import an existing talloc pointer into a Python object.
  38  */
  39 PyObject *py_talloc_steal_ex(PyTypeObject *py_type, TALLOC_CTX *mem_ctx,
  40                                                    void *ptr)
  41 {
  42         py_talloc_Object *ret = (py_talloc_Object *)py_type->tp_alloc(py_type, 0);
  43         ret->talloc_ctx = talloc_new(NULL);
  44         if (ret->talloc_ctx == NULL) {
  45                 return NULL;
  46         }
  47         if (talloc_steal(ret->talloc_ctx, mem_ctx) == NULL) {
  48                 return NULL;
  49         }
  50         ret->ptr = ptr;
  51         return (PyObject *)ret;
  52 }
  53
  54
  55 /**
  56  * Import an existing talloc pointer into a Python object, leaving the
  57  * original parent, and creating a reference to the object in the python
  58  * object
  59  */
  60 PyObject *py_talloc_reference_ex(PyTypeObject *py_type, TALLOC_CTX *mem_ctx, void *ptr)
  61 {
  62         py_talloc_Object *ret = (py_talloc_Object *)py_type->tp_alloc(py_type, 0);
  63         ret->talloc_ctx = talloc_new(NULL);
  64         if (ret->talloc_ctx == NULL) {
  65                 return NULL;
  66         }
  67         if (talloc_reference(ret->talloc_ctx, mem_ctx) == NULL) {
  68                 return NULL;
  69         }
  70         ret->ptr = ptr;
  71         return (PyObject *)ret;
  72 }
  73
  74 /**
  75  * Default (but slightly more useful than the default) implementation of Repr().
  76  */
  77 PyObject *py_talloc_default_repr(PyObject *obj)
  78 {
  79         py_talloc_Object *talloc_obj = (py_talloc_Object *)obj;
  80         PyTypeObject *type = (PyTypeObject*)PyObject_Type(obj);
  81
  82         return PyString_FromFormat("<%s talloc object at 0x%p>",
  83                                    type->tp_name, talloc_obj->ptr);
  84 }
  85
  86 static void py_cobject_talloc_free(void *ptr)
  87 {
  88         talloc_free(ptr);
  89 }
  90
  91 PyObject *PyCObject_FromTallocPtr(void *ptr)
  92 {
  93         return PyCObject_FromVoidPtr(ptr, py_cobject_talloc_free);
  94 }