s4-pyglue: added talloc_total_blocks() python call

[metze/samba/wip.git] / source4 / scripting / python / pyglue.c

diff --git a/source4/scripting/python/pyglue.c b/source4/scripting/python/pyglue.c
index ddd44aa7df05292ca7af03c0a9c539ca223dc709..ec15e77a9378a26f2a1cb165628f94e6065ab78b 100644 (file)
--- a/source4/scripting/python/pyglue.c
+++ b/source4/scripting/python/pyglue.c
@@ -165,7 +165,7 @@ static PyObject *py_interface_ips(PyObject *self, PyObject *args)
        return pylist;
 }
 
-/* return a talloc tree string for a talloc python object */
+/* print a talloc tree report for a talloc python object */
 static PyObject *py_talloc_report_full(PyObject *self, PyObject *args)
 {
        PyObject *py_obj;
@@ -183,13 +183,31 @@ static PyObject *py_talloc_report_full(PyObject *self, PyObject *args)
        return Py_None;
 }
 
-/* return a talloc tree string for a talloc python object */
+/* enable null tracking */
 static PyObject *py_talloc_enable_null_tracking(PyObject *self, PyObject *args)
 {
        talloc_enable_null_tracking();
        return Py_None;
 }
 
+/* return the number of talloc blocks */
+static PyObject *py_talloc_total_blocks(PyObject *self, PyObject *args)
+{
+       PyObject *py_obj;
+       PyTypeObject *type;
+
+       if (!PyArg_ParseTuple(args, "O", &py_obj))
+               return NULL;
+
+       if (py_obj == Py_None) {
+               return PyLong_FromLong(talloc_total_blocks(NULL));
+       }
+
+       type = (PyTypeObject*)PyObject_Type(py_obj);
+
+       return PyLong_FromLong(talloc_total_blocks(py_talloc_get_mem_ctx(py_obj)));
+}
+
 
 static PyMethodDef py_misc_methods[] = {
        { "generate_random_str", (PyCFunction)py_generate_random_str, METH_VARARGS,
@@ -212,6 +230,8 @@ static PyMethodDef py_misc_methods[] = {
                "show a talloc tree for an object"},
        { "talloc_enable_null_tracking", (PyCFunction)py_talloc_enable_null_tracking, METH_VARARGS,
                "enable tracking of the NULL object"},
+       { "talloc_total_blocks", (PyCFunction)py_talloc_total_blocks, METH_VARARGS,
+               "return talloc block count"},
        { NULL }
 };