path: root/test_pytalloc.c

/*
   Samba Unix SMB/CIFS implementation.

   C utilities for the pytalloc test suite.
   Provides the "_test_pytalloc" Python module.

   NOTE: Please read talloc_guide.txt for full documentation

   Copyright (C) Petr Viktorin 2015

     ** NOTE! The following LGPL license applies to the talloc
     ** library. This does NOT imply that all of Samba is released
     ** under the LGPL

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 3 of the License, or (at your option) any later version.

   This library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, see <http://www.gnu.org/licenses/>.
*/

#include "lib/replace/system/python.h"
#include <talloc.h>
#include <pytalloc.h>

static PyObject *testpytalloc_new(PyTypeObject *mod,
		PyObject *Py_UNUSED(ignored))
{
	char *obj = talloc_strdup(NULL, "This is a test string");;
	return pytalloc_steal(pytalloc_GetObjectType(), obj);
}

static PyObject *testpytalloc_get_object_type(PyObject *mod,
		PyObject *Py_UNUSED(ignored))
{
	PyObject *type = (PyObject *)pytalloc_GetObjectType();
	Py_INCREF(type);
	return type;
}

static PyObject *testpytalloc_base_new(PyTypeObject *mod,
		PyObject *Py_UNUSED(ignored))
{
	char *obj = talloc_strdup(NULL, "This is a test string for a BaseObject");;
	return pytalloc_steal(pytalloc_GetBaseObjectType(), obj);
}

static PyObject *testpytalloc_base_get_object_type(PyObject *mod,
		PyObject *Py_UNUSED(ignored))
{
	PyObject *type = (PyObject *)pytalloc_GetBaseObjectType();
	Py_INCREF(type);
	return type;
}

static PyObject *testpytalloc_reference(PyObject *mod, PyObject *args) {
	PyObject *source = NULL;
	void *ptr;

	if (!PyArg_ParseTuple(args, "O!", pytalloc_GetObjectType(), &source))
		return NULL;

	ptr = pytalloc_get_ptr(source);
	return pytalloc_reference_ex(pytalloc_GetObjectType(), ptr, ptr);
}

static PyObject *testpytalloc_base_reference(PyObject *mod, PyObject *args) {
	PyObject *source = NULL;
	void *mem_ctx;

	if (!PyArg_ParseTuple(args, "O!", pytalloc_GetBaseObjectType(), &source)) {
		return NULL;
	}
	mem_ctx = pytalloc_get_mem_ctx(source);
	return pytalloc_reference_ex(pytalloc_GetBaseObjectType(), mem_ctx, mem_ctx);
}

static PyMethodDef test_talloc_methods[] = {
	{ "new", (PyCFunction)testpytalloc_new, METH_NOARGS,
		"create a talloc Object with a testing string"},
	{ "get_object_type", (PyCFunction)testpytalloc_get_object_type, METH_NOARGS,
		"call pytalloc_GetObjectType"},
	{ "base_new", (PyCFunction)testpytalloc_base_new, METH_NOARGS,
		"create a talloc BaseObject with a testing string"},
	{ "base_get_object_type", (PyCFunction)testpytalloc_base_get_object_type, METH_NOARGS,
		"call pytalloc_GetBaseObjectType"},
	{ "reference", (PyCFunction)testpytalloc_reference, METH_VARARGS,
		"call pytalloc_reference_ex"},
	{ "base_reference", (PyCFunction)testpytalloc_base_reference, METH_VARARGS,
		"call pytalloc_reference_ex"},
	{0}
};

static PyTypeObject DObject_Type;

static int dobject_destructor(void *ptr)
{
	PyObject *destructor_func = *talloc_get_type(ptr, PyObject*);
	PyObject *ret;
	ret = PyObject_CallObject(destructor_func, NULL);
	Py_DECREF(destructor_func);
	if (ret == NULL) {
		PyErr_Print();
	} else {
		Py_DECREF(ret);
	}
	return 0;
}

static PyObject *dobject_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
{
	PyObject *destructor_func = NULL;
	PyObject **obj;

	if (!PyArg_ParseTuple(args, "O", &destructor_func))
		return NULL;
	Py_INCREF(destructor_func);

	obj = talloc(NULL, PyObject*);
	*obj = destructor_func;

	talloc_set_destructor((void*)obj, dobject_destructor);
	return pytalloc_steal(&DObject_Type, obj);
}

static PyTypeObject DObject_Type = {
	.tp_name = "_test_pytalloc.DObject",
	.tp_basicsize = sizeof(pytalloc_Object),
	.tp_methods = NULL,
	.tp_new = dobject_new,
	.tp_flags = Py_TPFLAGS_DEFAULT,
	.tp_doc = "test talloc object that calls a function when underlying data is freed\n",
};

static PyTypeObject DBaseObject_Type;

static int d_base_object_destructor(void *ptr)
{
	PyObject *destructor_func = *talloc_get_type(ptr, PyObject*);
	PyObject *ret;
	ret = PyObject_CallObject(destructor_func, NULL);
	Py_DECREF(destructor_func);
	if (ret == NULL) {
		PyErr_Print();
	} else {
		Py_DECREF(ret);
	}
	return 0;
}

static PyObject *d_base_object_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
{
	PyObject *destructor_func = NULL;
	PyObject **obj;

	if (!PyArg_ParseTuple(args, "O", &destructor_func))
		return NULL;
	Py_INCREF(destructor_func);

	obj = talloc(NULL, PyObject*);
	*obj = destructor_func;

	talloc_set_destructor((void*)obj, d_base_object_destructor);
	return pytalloc_steal(&DBaseObject_Type, obj);
}

static PyTypeObject DBaseObject_Type = {
	.tp_name = "_test_pytalloc.DBaseObject",
	.tp_methods = NULL,
	.tp_new = d_base_object_new,
	.tp_flags = Py_TPFLAGS_DEFAULT,
	.tp_doc = "test talloc object that calls a function when underlying data is freed\n",
};

#define MODULE_DOC PyDoc_STR("Test utility module for pytalloc")

static struct PyModuleDef moduledef = {
    PyModuleDef_HEAD_INIT,
    .m_name = "_test_pytalloc",
    .m_doc = PyDoc_STR("Test utility module for pytalloc"),
    .m_size = -1,
    .m_methods = test_talloc_methods,
};

static PyObject *module_init(void);
static PyObject *module_init(void)
{
	PyObject *m;

	DObject_Type.tp_base = pytalloc_GetObjectType();
	if (PyType_Ready(&DObject_Type) < 0) {
		return NULL;
	}

	DBaseObject_Type.tp_basicsize = pytalloc_BaseObject_size();
	DBaseObject_Type.tp_base = pytalloc_GetBaseObjectType();
	if (PyType_Ready(&DBaseObject_Type) < 0) {
		return NULL;
	}

	m = PyModule_Create(&moduledef);

	if (m == NULL) {
		return NULL;
	}

	Py_INCREF(&DObject_Type);
	Py_INCREF(DObject_Type.tp_base);
	PyModule_AddObject(m, "DObject", (PyObject *)&DObject_Type);

	Py_INCREF(&DBaseObject_Type);
	Py_INCREF(DBaseObject_Type.tp_base);
	PyModule_AddObject(m, "DBaseObject", (PyObject *)&DBaseObject_Type);

	return m;
}


PyMODINIT_FUNC PyInit__test_pytalloc(void);
PyMODINIT_FUNC PyInit__test_pytalloc(void)
{
	return module_init();
}