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         talloc_set_name_const(ret->talloc_ctx, py_type->tp_name);
  51         ret->ptr = ptr;
  52         return (PyObject *)ret;
  53 }
  54
  55 /**
  56  * Import an existing talloc pointer into a Python object.
  57  */
  58 PyObject *py_talloc_steal(PyTypeObject *py_type, void *ptr)
  59 {
  60         return py_talloc_steal_ex(py_type, ptr, ptr);
  61 }
  62
  63
  64 /**
  65  * Import an existing talloc pointer into a Python object, leaving the
  66  * original parent, and creating a reference to the object in the python
  67  * object
  68  */
  69 PyObject *py_talloc_reference_ex(PyTypeObject *py_type, TALLOC_CTX *mem_ctx, void *ptr)
  70 {
  71         py_talloc_Object *ret;
  72
  73         if (ptr == NULL) {
  74                 Py_RETURN_NONE;
  75         }
  76
  77         ret = (py_talloc_Object *)py_type->tp_alloc(py_type, 0);
  78         ret->talloc_ctx = talloc_new(NULL);
  79         if (ret->talloc_ctx == NULL) {
  80                 return NULL;
  81         }
  82         if (talloc_reference(ret->talloc_ctx, mem_ctx) == NULL) {
  83                 return NULL;
  84         }
  85         talloc_set_name_const(ret->talloc_ctx, py_type->tp_name);
  86         ret->ptr = ptr;
  87         return (PyObject *)ret;
  88 }
  89
  90 /**
  91  * Default (but slightly more useful than the default) implementation of Repr().
  92  */
  93 PyObject *py_talloc_default_repr(PyObject *obj)
  94 {
  95         py_talloc_Object *talloc_obj = (py_talloc_Object *)obj;
  96         PyTypeObject *type = (PyTypeObject*)PyObject_Type(obj);
  97
  98         return PyString_FromFormat("<%s talloc object at 0x%p>",
  99                                    type->tp_name, talloc_obj->ptr);
 100 }
 101
 102 static void py_cobject_talloc_free(void *ptr)
 103 {
 104         talloc_free(ptr);
 105 }
 106
 107 PyObject *PyCObject_FromTallocPtr(void *ptr)
 108 {
 109         if (ptr == NULL) {
 110                 Py_RETURN_NONE;
 111         }
 112         return PyCObject_FromVoidPtr(ptr, py_cobject_talloc_free);
 113 }
 114
 115 PyObject *PyString_FromString_check_null(const char *ptr)
 116 {
 117         if (ptr == NULL) {
 118                 Py_RETURN_NONE;
 119         }
 120         return PyString_FromString(ptr);
 121 }