mirror of
https://github.com/samba-team/samba.git
synced 2024-12-22 13:34:15 +03:00
pyldb: create LdbResult, return value from ldb.search is now a LdbResult
This commit is contained in:
parent
6d48a6ddb3
commit
4c74c5f557
@ -35,6 +35,8 @@ void initldb(void);
|
|||||||
static PyObject *PyLdbMessage_FromMessage(struct ldb_message *msg);
|
static PyObject *PyLdbMessage_FromMessage(struct ldb_message *msg);
|
||||||
static PyObject *PyExc_LdbError;
|
static PyObject *PyExc_LdbError;
|
||||||
|
|
||||||
|
staticforward PyTypeObject PyLdbControl;
|
||||||
|
staticforward PyTypeObject PyLdbResult;
|
||||||
staticforward PyTypeObject PyLdbMessage;
|
staticforward PyTypeObject PyLdbMessage;
|
||||||
staticforward PyTypeObject PyLdbModule;
|
staticforward PyTypeObject PyLdbModule;
|
||||||
staticforward PyTypeObject PyLdbDn;
|
staticforward PyTypeObject PyLdbDn;
|
||||||
@ -62,6 +64,119 @@ typedef intargfunc ssizeargfunc;
|
|||||||
|
|
||||||
#define SIGN(a) (((a) == 0)?0:((a) < 0?-1:1))
|
#define SIGN(a) (((a) == 0)?0:((a) < 0?-1:1))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
static PyObject *py_ldb_control_str(PyLdbControlObject *self)
|
||||||
|
{
|
||||||
|
if (self->data != NULL) {
|
||||||
|
char* control = ldb_control_to_string(self->mem_ctx, self->data);
|
||||||
|
if (control == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return PyString_FromString(control);
|
||||||
|
} else {
|
||||||
|
return PyString_FromFormat("ldb control");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void py_ldb_control_dealloc(PyLdbControlObject *self)
|
||||||
|
{
|
||||||
|
if (self->mem_ctx != NULL) {
|
||||||
|
talloc_free(self->mem_ctx);
|
||||||
|
}
|
||||||
|
self->ob_type->tp_free(self);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_control_get_oid(PyLdbControlObject *self)
|
||||||
|
{
|
||||||
|
return PyString_FromString(self->data->oid);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_control_get_critical(PyLdbControlObject *self)
|
||||||
|
{
|
||||||
|
return PyBool_FromLong(self->data->critical);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_control_set_critical(PyLdbControlObject *self, PyObject *value, void *closure)
|
||||||
|
{
|
||||||
|
if (PyObject_IsTrue(value)) {
|
||||||
|
self->data->critical = true;
|
||||||
|
} else {
|
||||||
|
self->data->critical = false;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_control_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
|
||||||
|
{
|
||||||
|
char *data = NULL;
|
||||||
|
const char *array[2];
|
||||||
|
const char * const kwnames[] = { "ldb", "data", NULL };
|
||||||
|
struct ldb_control *parsed_controls;
|
||||||
|
PyLdbControlObject *ret;
|
||||||
|
PyObject *py_ldb;
|
||||||
|
TALLOC_CTX *mem_ctx;
|
||||||
|
struct ldb_context *ldb_ctx;
|
||||||
|
|
||||||
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "Os",
|
||||||
|
discard_const_p(char *, kwnames),
|
||||||
|
&py_ldb, &data))
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
mem_ctx = talloc_new(NULL);
|
||||||
|
if (mem_ctx == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ldb_ctx = PyLdb_AsLdbContext(py_ldb);
|
||||||
|
parsed_controls = ldb_parse_control_from_string(ldb_ctx, mem_ctx, data);
|
||||||
|
|
||||||
|
if (!parsed_controls) {
|
||||||
|
talloc_free(mem_ctx);
|
||||||
|
PyErr_SetString(PyExc_ValueError, "unable to parse control string");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = PyObject_New(PyLdbControlObject, type);
|
||||||
|
if (ret == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
talloc_free(mem_ctx);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret->mem_ctx = mem_ctx;
|
||||||
|
|
||||||
|
ret->data = talloc_steal(mem_ctx, parsed_controls);
|
||||||
|
if (ret->data == NULL) {
|
||||||
|
Py_DECREF(ret);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
talloc_free(mem_ctx);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return (PyObject *)ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyGetSetDef py_ldb_control_getset[] = {
|
||||||
|
{ discard_const_p(char, "oid"), (getter)py_ldb_control_get_oid, NULL, NULL },
|
||||||
|
{ discard_const_p(char, "critical"), (getter)py_ldb_control_get_critical, (setter)py_ldb_control_set_critical, NULL },
|
||||||
|
{ NULL }
|
||||||
|
};
|
||||||
|
|
||||||
|
static PyTypeObject PyLdbControl = {
|
||||||
|
.tp_name = "ldb.control",
|
||||||
|
.tp_dealloc = (destructor)py_ldb_control_dealloc,
|
||||||
|
.tp_getattro = PyObject_GenericGetAttr,
|
||||||
|
.tp_basicsize = sizeof(PyLdbControlObject),
|
||||||
|
.tp_getset = py_ldb_control_getset,
|
||||||
|
.tp_doc = "LDB control.",
|
||||||
|
.tp_str = (reprfunc)py_ldb_control_str,
|
||||||
|
.tp_new = py_ldb_control_new,
|
||||||
|
.tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
|
||||||
|
};
|
||||||
|
|
||||||
static void PyErr_SetLdbError(PyObject *error, int ret, struct ldb_context *ldb_ctx)
|
static void PyErr_SetLdbError(PyObject *error, int ret, struct ldb_context *ldb_ctx)
|
||||||
{
|
{
|
||||||
if (ret == LDB_ERR_PYTHON_EXCEPTION)
|
if (ret == LDB_ERR_PYTHON_EXCEPTION)
|
||||||
@ -77,6 +192,36 @@ static PyObject *PyObject_FromLdbValue(struct ldb_val *val)
|
|||||||
return PyString_FromStringAndSize((const char *)val->data, val->length);
|
return PyString_FromStringAndSize((const char *)val->data, val->length);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a Python object from a ldb_result.
|
||||||
|
*
|
||||||
|
* @param result LDB result to convert
|
||||||
|
* @return Python object with converted result (a list object)
|
||||||
|
*/
|
||||||
|
static PyObject *PyLdbControl_FromControl(struct ldb_control *control)
|
||||||
|
{
|
||||||
|
TALLOC_CTX *ctl_ctx = talloc_new(NULL);
|
||||||
|
PyLdbControlObject *ctrl;
|
||||||
|
if (ctl_ctx == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ctrl = (PyLdbControlObject *)PyLdbControl.tp_alloc(&PyLdbControl, 0);
|
||||||
|
if (ctrl == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
ctrl->mem_ctx = ctl_ctx;
|
||||||
|
ctrl->data = talloc_steal(ctrl->mem_ctx, control);
|
||||||
|
if (ctrl->data == NULL) {
|
||||||
|
Py_DECREF(ctrl);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return (PyObject*) ctrl;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create a Python object from a ldb_result.
|
* Create a Python object from a ldb_result.
|
||||||
*
|
*
|
||||||
@ -85,16 +230,90 @@ static PyObject *PyObject_FromLdbValue(struct ldb_val *val)
|
|||||||
*/
|
*/
|
||||||
static PyObject *PyLdbResult_FromResult(struct ldb_result *result)
|
static PyObject *PyLdbResult_FromResult(struct ldb_result *result)
|
||||||
{
|
{
|
||||||
PyObject *ret;
|
PyLdbResultObject *ret;
|
||||||
|
PyObject *list, *controls, *referals;
|
||||||
Py_ssize_t i;
|
Py_ssize_t i;
|
||||||
|
|
||||||
if (result == NULL) {
|
if (result == NULL) {
|
||||||
Py_RETURN_NONE;
|
Py_RETURN_NONE;
|
||||||
}
|
|
||||||
ret = PyList_New(result->count);
|
|
||||||
for (i = 0; i < result->count; i++) {
|
|
||||||
PyList_SetItem(ret, i, PyLdbMessage_FromMessage(result->msgs[i]));
|
|
||||||
}
|
}
|
||||||
return ret;
|
|
||||||
|
ret = (PyLdbResultObject *)PyLdbResult.tp_alloc(&PyLdbResult, 0);
|
||||||
|
if (ret == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
list = PyList_New(result->count);
|
||||||
|
if (list == NULL) {
|
||||||
|
PyErr_NoMemory();
|
||||||
|
Py_DECREF(ret);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < result->count; i++) {
|
||||||
|
PyList_SetItem(list, i, PyLdbMessage_FromMessage(result->msgs[i]));
|
||||||
|
}
|
||||||
|
|
||||||
|
ret->mem_ctx = talloc_new(NULL);
|
||||||
|
if (ret->mem_ctx == NULL) {
|
||||||
|
Py_DECREF(list);
|
||||||
|
Py_DECREF(ret);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret->msgs = list;
|
||||||
|
|
||||||
|
if (result->controls) {
|
||||||
|
controls = PyList_New(1);
|
||||||
|
if (controls == NULL) {
|
||||||
|
Py_DECREF(ret);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
for (i=0; result->controls[i]; i++) {
|
||||||
|
PyObject *ctrl = (PyObject*) PyLdbControl_FromControl(result->controls[i]);
|
||||||
|
if (ctrl == NULL) {
|
||||||
|
Py_DECREF(ret);
|
||||||
|
Py_DECREF(controls);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
PyList_SetItem(controls, i, ctrl);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
/*
|
||||||
|
* No controls so we keep an empty list
|
||||||
|
*/
|
||||||
|
controls = PyList_New(0);
|
||||||
|
if (controls == NULL) {
|
||||||
|
Py_DECREF(ret);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ret->controls = controls;
|
||||||
|
|
||||||
|
i = 0;
|
||||||
|
|
||||||
|
while (result->refs && result->refs[i]) {
|
||||||
|
i++;
|
||||||
|
}
|
||||||
|
|
||||||
|
referals = PyList_New(i);
|
||||||
|
if (referals == NULL) {
|
||||||
|
Py_DECREF(ret);
|
||||||
|
PyErr_NoMemory();
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0;result->refs && result->refs[i]; i++) {
|
||||||
|
PyList_SetItem(referals, i, PyString_FromString(result->refs[i]));
|
||||||
|
}
|
||||||
|
ret->referals = referals;
|
||||||
|
return (PyObject *)ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -1520,6 +1739,91 @@ static PyTypeObject PyLdb = {
|
|||||||
.tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
|
.tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
static void py_ldb_result_dealloc(PyLdbResultObject *self)
|
||||||
|
{
|
||||||
|
talloc_free(self->mem_ctx);
|
||||||
|
Py_DECREF(self->msgs);
|
||||||
|
Py_DECREF(self->referals);
|
||||||
|
Py_DECREF(self->controls);
|
||||||
|
self->ob_type->tp_free(self);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_get_msgs(PyLdbResultObject *self, void *closure)
|
||||||
|
{
|
||||||
|
Py_INCREF(self->msgs);
|
||||||
|
return self->msgs;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_get_controls(PyLdbResultObject *self, void *closure)
|
||||||
|
{
|
||||||
|
Py_INCREF(self->controls);
|
||||||
|
return self->controls;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_get_referals(PyLdbResultObject *self, void *closure)
|
||||||
|
{
|
||||||
|
Py_INCREF(self->referals);
|
||||||
|
return self->referals;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_get_count(PyLdbResultObject *self, void *closure)
|
||||||
|
{
|
||||||
|
Py_ssize_t size;
|
||||||
|
if (self->msgs == NULL) {
|
||||||
|
PyErr_SetString(PyExc_AttributeError, "Count attribute is meaningless in this context");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
size = PyList_Size(self->msgs);
|
||||||
|
return PyInt_FromLong(size);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyGetSetDef py_ldb_result_getset[] = {
|
||||||
|
{ discard_const_p(char, "controls"), (getter)py_ldb_result_get_controls, NULL, NULL },
|
||||||
|
{ discard_const_p(char, "msgs"), (getter)py_ldb_result_get_msgs, NULL, NULL },
|
||||||
|
{ discard_const_p(char, "referals"), (getter)py_ldb_result_get_referals, NULL, NULL },
|
||||||
|
{ discard_const_p(char, "count"), (getter)py_ldb_result_get_count, NULL, NULL },
|
||||||
|
{ NULL }
|
||||||
|
};
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_iter(PyLdbResultObject *self)
|
||||||
|
{
|
||||||
|
return PyObject_GetIter(self->msgs);
|
||||||
|
}
|
||||||
|
|
||||||
|
static Py_ssize_t py_ldb_result_len(PyLdbResultObject *self)
|
||||||
|
{
|
||||||
|
return PySequence_Size(self->msgs);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_find(PyLdbResultObject *self, Py_ssize_t idx)
|
||||||
|
{
|
||||||
|
return PySequence_GetItem(self->msgs, idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
static PySequenceMethods py_ldb_result_seq = {
|
||||||
|
.sq_length = (lenfunc)py_ldb_result_len,
|
||||||
|
.sq_item = (ssizeargfunc)py_ldb_result_find,
|
||||||
|
};
|
||||||
|
|
||||||
|
static PyObject *py_ldb_result_repr(PyLdbObject *self)
|
||||||
|
{
|
||||||
|
return PyString_FromFormat("<ldb result>");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static PyTypeObject PyLdbResult = {
|
||||||
|
.tp_name = "ldb.Result",
|
||||||
|
.tp_repr = (reprfunc)py_ldb_result_repr,
|
||||||
|
.tp_dealloc = (destructor)py_ldb_result_dealloc,
|
||||||
|
.tp_iter = (getiterfunc)py_ldb_result_iter,
|
||||||
|
.tp_getset = py_ldb_result_getset,
|
||||||
|
.tp_getattro = PyObject_GenericGetAttr,
|
||||||
|
.tp_basicsize = sizeof(PyLdbResultObject),
|
||||||
|
.tp_as_sequence = &py_ldb_result_seq,
|
||||||
|
.tp_doc = "LDB result.",
|
||||||
|
.tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
|
||||||
|
};
|
||||||
|
|
||||||
static PyObject *py_ldb_module_repr(PyLdbModuleObject *self)
|
static PyObject *py_ldb_module_repr(PyLdbModuleObject *self)
|
||||||
{
|
{
|
||||||
return PyString_FromFormat("<ldb module '%s'>", PyLdbModule_AsModule(self)->ops->name);
|
return PyString_FromFormat("<ldb module '%s'>", PyLdbModule_AsModule(self)->ops->name);
|
||||||
@ -2802,6 +3106,12 @@ void initldb(void)
|
|||||||
if (PyType_Ready(&PyLdbTree) < 0)
|
if (PyType_Ready(&PyLdbTree) < 0)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
|
if (PyType_Ready(&PyLdbResult) < 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (PyType_Ready(&PyLdbControl) < 0)
|
||||||
|
return;
|
||||||
|
|
||||||
m = Py_InitModule3("ldb", py_ldb_global_methods,
|
m = Py_InitModule3("ldb", py_ldb_global_methods,
|
||||||
"An interface to LDB, a LDAP-like API that can either to talk an embedded database (TDB-based) or a standards-compliant LDAP server.");
|
"An interface to LDB, a LDAP-like API that can either to talk an embedded database (TDB-based) or a standards-compliant LDAP server.");
|
||||||
if (m == NULL)
|
if (m == NULL)
|
||||||
@ -2880,6 +3190,8 @@ void initldb(void)
|
|||||||
Py_INCREF(&PyLdbMessage);
|
Py_INCREF(&PyLdbMessage);
|
||||||
Py_INCREF(&PyLdbMessageElement);
|
Py_INCREF(&PyLdbMessageElement);
|
||||||
Py_INCREF(&PyLdbTree);
|
Py_INCREF(&PyLdbTree);
|
||||||
|
Py_INCREF(&PyLdbResult);
|
||||||
|
Py_INCREF(&PyLdbControl);
|
||||||
|
|
||||||
PyModule_AddObject(m, "Ldb", (PyObject *)&PyLdb);
|
PyModule_AddObject(m, "Ldb", (PyObject *)&PyLdb);
|
||||||
PyModule_AddObject(m, "Dn", (PyObject *)&PyLdbDn);
|
PyModule_AddObject(m, "Dn", (PyObject *)&PyLdbDn);
|
||||||
@ -2887,6 +3199,7 @@ void initldb(void)
|
|||||||
PyModule_AddObject(m, "MessageElement", (PyObject *)&PyLdbMessageElement);
|
PyModule_AddObject(m, "MessageElement", (PyObject *)&PyLdbMessageElement);
|
||||||
PyModule_AddObject(m, "Module", (PyObject *)&PyLdbModule);
|
PyModule_AddObject(m, "Module", (PyObject *)&PyLdbModule);
|
||||||
PyModule_AddObject(m, "Tree", (PyObject *)&PyLdbTree);
|
PyModule_AddObject(m, "Tree", (PyObject *)&PyLdbTree);
|
||||||
|
PyModule_AddObject(m, "Control", (PyObject *)&PyLdbControl);
|
||||||
|
|
||||||
PyModule_AddObject(m, "__version__", PyString_FromString(PACKAGE_VERSION));
|
PyModule_AddObject(m, "__version__", PyString_FromString(PACKAGE_VERSION));
|
||||||
}
|
}
|
||||||
|
@ -78,6 +78,20 @@ typedef struct {
|
|||||||
} PyLdbTreeObject;
|
} PyLdbTreeObject;
|
||||||
#define PyLdbTree_AsTree(pyobj) ((PyLdbTreeObject *)pyobj)->tree
|
#define PyLdbTree_AsTree(pyobj) ((PyLdbTreeObject *)pyobj)->tree
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
PyObject_HEAD
|
||||||
|
TALLOC_CTX *mem_ctx;
|
||||||
|
PyObject *msgs;
|
||||||
|
PyObject *referals;
|
||||||
|
PyObject *controls;
|
||||||
|
} PyLdbResultObject;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
PyObject_HEAD
|
||||||
|
TALLOC_CTX *mem_ctx;
|
||||||
|
struct ldb_control *data;
|
||||||
|
} PyLdbControlObject;
|
||||||
|
|
||||||
#define PyErr_LDB_ERROR_IS_ERR_RAISE(err,ret,ldb) \
|
#define PyErr_LDB_ERROR_IS_ERR_RAISE(err,ret,ldb) \
|
||||||
if (ret != LDB_SUCCESS) { \
|
if (ret != LDB_SUCCESS) { \
|
||||||
PyErr_SetLdbError(err, ret, ldb); \
|
PyErr_SetLdbError(err, ret, ldb); \
|
||||||
|
Loading…
Reference in New Issue
Block a user