s4-pyglue: added talloc_total_blocks() python call

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 }
 };
 

diff --git a/source4/scripting/python/samba/__init__.py b/source4/scripting/python/samba/__init__.py
index fcd224202b5fabcf07cdb81b45e5adbb6da7e09f..72bbb4a7e35f5d14c6eac2b67d0399bdb972f5cf 100644 (file)
--- a/source4/scripting/python/samba/__init__.py
+++ b/source4/scripting/python/samba/__init__.py
@@ -329,3 +329,4 @@ unix2nttime = _glue.unix2nttime
 generate_random_password = _glue.generate_random_password
 talloc_report_full = _glue.talloc_report_full
 talloc_enable_null_tracking = _glue.talloc_enable_null_tracking
+talloc_total_blocks = _glue.talloc_total_blocks