pyldb: Fix two memory leaks of attribute lists.
[samba.git] / source4 / lib / ldb / pyldb.c
index ef2af060f2738bbd52ca86dacafe028df586f557..ce38e066384c15fbc63ff13eb2cdd4a9c13abd5d 100644 (file)
@@ -1,16 +1,16 @@
 /*
    Unix SMB/CIFS implementation.
 
-   Swig interface to ldb.
+   Python interface to ldb.
 
    Copyright (C) 2005,2006 Tim Potter <tpot@samba.org>
    Copyright (C) 2006 Simo Sorce <idra@samba.org>
-   Copyright (C) 2007-2008 Jelmer Vernooij <jelmer@samba.org>
+   Copyright (C) 2007-2009 Jelmer Vernooij <jelmer@samba.org>
 
         ** NOTE! The following LGPL license applies to the ldb
         ** 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
@@ -41,47 +41,40 @@ typedef intargfunc ssizeargfunc;
 #define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
 #endif
 
-/* Picked out of thin air. To do this properly, we should probably have some part of the 
- * errors in LDB be allocated to bindings ? */
-#define LDB_ERR_PYTHON_EXCEPTION       142
-
 static PyObject *PyExc_LdbError;
 
-void PyErr_SetLdbError(int ret, struct ldb_context *ldb_ctx)
-{
-       if (ret == LDB_ERR_PYTHON_EXCEPTION)
-               return; /* Python exception should already be set, just keep that */
-       PyErr_SetObject(PyExc_LdbError, Py_BuildValue((char *)"(i,s)", ret, ldb_ctx == NULL?ldb_strerror(ret):ldb_errstring(ldb_ctx)));
-}
+PyAPI_DATA(PyTypeObject) PyLdbMessage;
+PyAPI_DATA(PyTypeObject) PyLdbModule;
+PyAPI_DATA(PyTypeObject) PyLdbDn;
+PyAPI_DATA(PyTypeObject) PyLdb;
+PyAPI_DATA(PyTypeObject) PyLdbMessageElement;
+PyAPI_DATA(PyTypeObject) PyLdbTree;
+
 static PyObject *PyObject_FromLdbValue(struct ldb_context *ldb_ctx, 
                                                           struct ldb_message_element *el, 
                                                           struct ldb_val *val)
 {
-       const struct ldb_schema_attribute *a;
        struct ldb_val new_val;
        TALLOC_CTX *mem_ctx = talloc_new(NULL);
        PyObject *ret;
-       
+
        new_val = *val;
-       
-       if (ldb_ctx != NULL) {          
-               a = ldb_schema_attribute_by_name(ldb_ctx, el->name);
-       
-               if (a != NULL) {
-                       if (a->syntax->ldif_write_fn(ldb_ctx, mem_ctx, val, &new_val) != 0) {
-                               talloc_free(mem_ctx);
-                               return NULL;
-                       }
-               }
-       } 
-       
+
        ret = PyString_FromStringAndSize((const char *)new_val.data, new_val.length);
-       
+
        talloc_free(mem_ctx);
-       
+
        return ret;
 }
 
+/**
+ * Obtain a ldb DN from a Python object.
+ *
+ * @param mem_ctx Memory context
+ * @param object Python object
+ * @param ldb_ctx LDB context
+ * @return Whether or not the conversion succeeded
+ */
 bool PyObject_AsDn(TALLOC_CTX *mem_ctx, PyObject *object, 
                   struct ldb_context *ldb_ctx, struct ldb_dn **dn)
 {
@@ -102,6 +95,12 @@ bool PyObject_AsDn(TALLOC_CTX *mem_ctx, PyObject *object,
        return false;
 }
 
+/**
+ * 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 *PyLdbResult_FromResult(struct ldb_result *result)
 {
        PyObject *ret;
@@ -117,11 +116,20 @@ static PyObject *PyLdbResult_FromResult(struct ldb_result *result)
        return ret;
 }
 
-static struct ldb_result *PyLdbResult_AsResult(TALLOC_CTX *mem_ctx, PyObject *obj)
+/**
+ * Create a LDB Result from a Python object. 
+ * If conversion fails, NULL will be returned and a Python exception set.
+ *
+ * @param mem_ctx Memory context in which to allocate the LDB Result
+ * @param obj Python object to convert
+ * @return a ldb_result, or NULL if the conversion failed
+ */
+static struct ldb_result *PyLdbResult_AsResult(TALLOC_CTX *mem_ctx, 
+                                                                                          PyObject *obj)
 {
        struct ldb_result *res;
        int i;
-       
+
        if (obj == Py_None)
                return NULL;
 
@@ -198,7 +206,15 @@ static int py_ldb_dn_compare(PyLdbDnObject *dn1, PyLdbDnObject *dn2)
 static PyObject *py_ldb_dn_get_parent(PyLdbDnObject *self)
 {
        struct ldb_dn *dn = PyLdbDn_AsDn((PyObject *)self);
-       return PyLdbDn_FromDn(ldb_dn_get_parent(NULL, dn));
+       struct ldb_dn *parent;
+
+       parent = ldb_dn_get_parent(NULL, dn);
+
+       if (parent == NULL) {
+               Py_RETURN_NONE;
+       } else {
+               return PyLdbDn_FromDn(parent);
+       }
 }
 
 #define dn_ldb_ctx(dn) ((struct ldb_context *)dn)
@@ -298,13 +314,15 @@ static PyObject *py_ldb_dn_new(PyTypeObject *type, PyObject *args, PyObject *kwa
        PyObject *py_ldb;
        struct ldb_context *ldb_ctx;
        PyLdbDnObject *py_ret;
-       const char *kwnames[] = { "ldb", "dn", NULL };
+       const char * const kwnames[] = { "ldb", "dn", NULL };
 
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "Os", (char **)kwnames, &py_ldb, &str))
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "Os",
+                                        discard_const_p(char *, kwnames),
+                                        &py_ldb, &str))
                return NULL;
 
        ldb_ctx = PyLdb_AsLdbContext(py_ldb);
-       
+
        ret = ldb_dn_new(ldb_ctx, ldb_ctx, str);
        /* ldb_dn_new() doesn't accept NULL as memory context, so 
           we do it this way... */
@@ -361,21 +379,21 @@ PyTypeObject PyLdbDn = {
 static void py_ldb_debug(void *context, enum ldb_debug_level level, const char *fmt, va_list ap) PRINTF_ATTRIBUTE(3, 0);
 static void py_ldb_debug(void *context, enum ldb_debug_level level, const char *fmt, va_list ap)
 {
-       PyObject *fn = context;
-       PyObject_CallFunction(fn, (char *)"(i,O)", level, PyString_FromFormatV(fmt, ap));
+       PyObject *fn = (PyObject *)context;
+       PyObject_CallFunction(fn, discard_const_p(char, "(i,O)"), level, PyString_FromFormatV(fmt, ap));
 }
 
 static PyObject *py_ldb_set_debug(PyLdbObject *self, PyObject *args)
 {
        PyObject *cb;
-       
+
        if (!PyArg_ParseTuple(args, "O", &cb))
                return NULL;
 
        Py_INCREF(cb);
        /* FIXME: Where do we DECREF cb ? */
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ldb_set_debug(self->ldb_ctx, py_ldb_debug, cb), PyLdb_AsLdbContext(self));
-       
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ldb_set_debug(self->ldb_ctx, py_ldb_debug, cb), PyLdb_AsLdbContext(self));
+
        Py_RETURN_NONE;
 }
 
@@ -403,25 +421,25 @@ static PyObject *py_ldb_set_modules_dir(PyTypeObject *self, PyObject *args)
 
 static PyObject *py_ldb_transaction_start(PyLdbObject *self)
 {
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ldb_transaction_start(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ldb_transaction_start(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
        Py_RETURN_NONE;
 }
 
 static PyObject *py_ldb_transaction_commit(PyLdbObject *self)
 {
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ldb_transaction_commit(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ldb_transaction_commit(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
        Py_RETURN_NONE;
 }
 
 static PyObject *py_ldb_transaction_cancel(PyLdbObject *self)
 {
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ldb_transaction_cancel(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ldb_transaction_cancel(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
        Py_RETURN_NONE;
 }
 
 static PyObject *py_ldb_setup_wellknown_attributes(PyLdbObject *self)
 {
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ldb_setup_wellknown_attributes(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ldb_setup_wellknown_attributes(PyLdb_AsLdbContext(self)), PyLdb_AsLdbContext(self));
        Py_RETURN_NONE;
 }
 
@@ -447,7 +465,6 @@ static PyObject *py_ldb_get_schema_basedn(PyLdbObject *self)
        return PyLdbDn_FromDn(dn);
 }
 
-
 static PyObject *py_ldb_get_config_basedn(PyLdbObject *self)
 {
        struct ldb_dn *dn = ldb_get_config_basedn(PyLdb_AsLdbContext(self));
@@ -456,7 +473,6 @@ static PyObject *py_ldb_get_config_basedn(PyLdbObject *self)
        return PyLdbDn_FromDn(dn);
 }
 
-
 static PyObject *py_ldb_get_default_basedn(PyLdbObject *self)
 {
        struct ldb_dn *dn = ldb_get_default_basedn(PyLdb_AsLdbContext(self));
@@ -465,19 +481,20 @@ static PyObject *py_ldb_get_default_basedn(PyLdbObject *self)
        return PyLdbDn_FromDn(dn);
 }
 
-static const char **PyList_AsStringList(TALLOC_CTX *mem_ctx, PyObject *list)
+static const char **PyList_AsStringList(TALLOC_CTX *mem_ctx, PyObject *list, 
+                                                                               const char *paramname)
 {
        const char **ret;
        int i;
        if (!PyList_Check(list)) {
-               PyErr_SetString(PyExc_TypeError, "options is not a list");
+               PyErr_Format(PyExc_TypeError, "%s is not a list", paramname);
                return NULL;
        }
        ret = talloc_array(NULL, const char *, PyList_Size(list)+1);
        for (i = 0; i < PyList_Size(list); i++) {
                PyObject *item = PyList_GetItem(list, i);
                if (!PyString_Check(item)) {
-                       PyErr_SetString(PyExc_TypeError, "options should be strings");
+                       PyErr_Format(PyExc_TypeError, "%s should be strings", paramname);
                        return NULL;
                }
                ret[i] = PyString_AsString(item);
@@ -488,7 +505,7 @@ static const char **PyList_AsStringList(TALLOC_CTX *mem_ctx, PyObject *list)
 
 static int py_ldb_init(PyLdbObject *self, PyObject *args, PyObject *kwargs)
 {
-       const char *kwnames[] = { "url", "flags", "options", NULL };
+       const char * const kwnames[] = { "url", "flags", "options", NULL };
        char *url = NULL;
        PyObject *py_options = Py_None;
        const char **options;
@@ -496,8 +513,9 @@ static int py_ldb_init(PyLdbObject *self, PyObject *args, PyObject *kwargs)
        int ret;
        struct ldb_context *ldb;
 
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ziO:Ldb.__init__", (char **)kwnames,
-                                                                        &url, &flags, &py_options))
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ziO:Ldb.__init__",
+                                        discard_const_p(char *, kwnames),
+                                        &url, &flags, &py_options))
                return -1;
 
        ldb = PyLdb_AsLdbContext(self);
@@ -505,15 +523,15 @@ static int py_ldb_init(PyLdbObject *self, PyObject *args, PyObject *kwargs)
        if (py_options == Py_None) {
                options = NULL;
        } else {
-               options = PyList_AsStringList(ldb, py_options);
+               options = PyList_AsStringList(ldb, py_options, "options");
                if (options == NULL)
                        return -1;
        }
-       
+
        if (url != NULL) {
                ret = ldb_connect(ldb, url, flags, options);
                if (ret != LDB_SUCCESS) {
-                       PyErr_SetLdbError(ret, ldb);
+                       PyErr_SetLdbError(PyExc_LdbError, ret, ldb);
                        return -1;
                }
        }
@@ -548,23 +566,25 @@ static PyObject *py_ldb_connect(PyLdbObject *self, PyObject *args, PyObject *kwa
        PyObject *py_options = Py_None;
        int ret;
        const char **options;
-       const char *kwnames[] = { "url", "flags", "options", NULL };
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s|iO", (char **)kwnames, &url, &flags,
-                                                                        &py_options))
+       const char * const kwnames[] = { "url", "flags", "options", NULL };
+
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s|iO",
+                                        discard_const_p(char *, kwnames),
+                                        &url, &flags, &py_options))
                return NULL;
 
        if (py_options == Py_None) {
                options = NULL;
        } else {
-               options = PyList_AsStringList(NULL, py_options);
+               options = PyList_AsStringList(NULL, py_options, "options");
                if (options == NULL)
                        return NULL;
        }
-       
+
        ret = ldb_connect(PyLdb_AsLdbContext(self), url, flags, options);
        talloc_free(options);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, PyLdb_AsLdbContext(self));
 
        Py_RETURN_NONE;
 }
@@ -582,7 +602,7 @@ static PyObject *py_ldb_modify(PyLdbObject *self, PyObject *args)
        }
 
        ret = ldb_modify(PyLdb_AsLdbContext(self), PyLdbMessage_AsMessage(py_msg));
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, PyLdb_AsLdbContext(self));
 
        Py_RETURN_NONE;
 }
@@ -637,15 +657,13 @@ static PyObject *py_ldb_add(PyLdbObject *self, PyObject *args)
        } else {
                msg = PyLdbMessage_AsMessage(py_msg);
        }
-       
+
        ret = ldb_add(PyLdb_AsLdbContext(self), msg);
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, PyLdb_AsLdbContext(self));
 
        Py_RETURN_NONE;
 }
 
-
-
 static PyObject *py_ldb_delete(PyLdbObject *self, PyObject *args)
 {
        PyObject *py_dn;
@@ -661,7 +679,7 @@ static PyObject *py_ldb_delete(PyLdbObject *self, PyObject *args)
                return NULL;
 
        ret = ldb_delete(ldb, dn);
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, ldb);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, ldb);
 
        Py_RETURN_NONE;
 }
@@ -683,7 +701,7 @@ static PyObject *py_ldb_rename(PyLdbObject *self, PyObject *args)
                return NULL;
 
        ret = ldb_rename(ldb, dn1, dn2);
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, ldb);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, ldb);
 
        Py_RETURN_NONE;
 }
@@ -709,7 +727,7 @@ static PyObject *py_ldb_schema_attribute_add(PyLdbObject *self, PyObject *args)
 
        ret = ldb_schema_attribute_add(PyLdb_AsLdbContext(self), attribute, flags, syntax);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, PyLdb_AsLdbContext(self));
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, PyLdb_AsLdbContext(self));
 
        Py_RETURN_NONE;
 }
@@ -721,8 +739,9 @@ static PyObject *ldb_ldif_to_pyobject(struct ldb_ldif *ldif)
        } else {
        /* We don't want this attached to the 'ldb' any more */
                talloc_steal(NULL, ldif);
-               return Py_BuildValue((char *)"(iO)", ldif->changetype, 
-                                                        PyLdbMessage_FromMessage(ldif->msg));
+               return Py_BuildValue(discard_const_p(char, "(iO)"),
+                                    ldif->changetype,
+                                    PyLdbMessage_FromMessage(ldif->msg));
        }
 }
 
@@ -755,18 +774,18 @@ static PyObject *py_ldb_schema_format_value(PyLdbObject *self, PyObject *args)
 
        if (!PyArg_ParseTuple(args, "sO", &element_name, &val))
                return NULL;
-       
+
        mem_ctx = talloc_new(NULL);
-       
+
        old_val.data = (uint8_t *)PyString_AsString(val);
        old_val.length = PyString_Size(val);
-               
+
        a = ldb_schema_attribute_by_name(PyLdb_AsLdbContext(self), element_name);
 
        if (a == NULL) {
                Py_RETURN_NONE;
        }
-       
+
        if (a->syntax->ldif_write_fn(PyLdb_AsLdbContext(self), mem_ctx, &old_val, &new_val) != 0) {
                talloc_free(mem_ctx);
                Py_RETURN_NONE;
@@ -786,7 +805,7 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
        char *expr = NULL;
        PyObject *py_attrs = Py_None;
        PyObject *py_controls = Py_None;
-       const char *kwnames[] = { "base", "scope", "expression", "attrs", "controls", NULL };
+       const char * const kwnames[] = { "base", "scope", "expression", "attrs", "controls", NULL };
        int ret;
        struct ldb_result *res;
        struct ldb_request *req;
@@ -794,9 +813,11 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
        struct ldb_context *ldb_ctx;
        struct ldb_control **parsed_controls;
        struct ldb_dn *base;
+       PyObject *py_ret;
 
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OizOO", (char **)kwnames,
-                                                                        &py_base, &scope, &expr, &py_attrs, &py_controls))
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OizOO",
+                                        discard_const_p(char *, kwnames),
+                                        &py_base, &scope, &expr, &py_attrs, &py_controls))
                return NULL;
 
        ldb_ctx = PyLdb_AsLdbContext(self);
@@ -804,7 +825,7 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
        if (py_attrs == Py_None) {
                attrs = NULL;
        } else {
-               attrs = PyList_AsStringList(ldb_ctx, py_attrs);
+               attrs = PyList_AsStringList(NULL, py_attrs, "attrs");
                if (attrs == NULL)
                        return NULL;
        }
@@ -812,14 +833,16 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
        if (py_base == Py_None) {
                base = ldb_get_default_basedn(ldb_ctx);
        } else {
-               if (!PyObject_AsDn(ldb_ctx, py_base, ldb_ctx, &base))
+               if (!PyObject_AsDn(ldb_ctx, py_base, ldb_ctx, &base)) {
+                       talloc_free(attrs);
                        return NULL;
+               }
        }
 
        if (py_controls == Py_None) {
                parsed_controls = NULL;
        } else {
-               const char **controls = PyList_AsStringList(ldb_ctx, py_controls);
+               const char **controls = PyList_AsStringList(ldb_ctx, py_controls, "controls");
                parsed_controls = ldb_parse_control_strings(ldb_ctx, ldb_ctx, controls);
                talloc_free(controls);
        }
@@ -827,6 +850,7 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
        res = talloc_zero(ldb_ctx, struct ldb_result);
        if (res == NULL) {
                PyErr_NoMemory();
+               talloc_free(attrs);
                return NULL;
        }
 
@@ -840,14 +864,16 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
                                   ldb_search_default_callback,
                                   NULL);
 
+       talloc_steal(req, attrs);
+
        if (ret != LDB_SUCCESS) {
                talloc_free(res);
-               PyErr_LDB_ERROR_IS_ERR_RAISE(ret, ldb_ctx);
+               PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, ldb_ctx);
                return NULL;
        }
 
        ret = ldb_request(ldb_ctx, req);
-               
+
        if (ret == LDB_SUCCESS) {
                ret = ldb_wait(req->handle, LDB_WAIT_ALL);
        }
@@ -856,11 +882,15 @@ static PyObject *py_ldb_search(PyLdbObject *self, PyObject *args, PyObject *kwar
 
        if (ret != LDB_SUCCESS) {
                talloc_free(res);
-               PyErr_LDB_ERROR_IS_ERR_RAISE(ret, ldb_ctx);
+               PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, ldb_ctx);
                return NULL;
        }
 
-       return PyLdbResult_FromResult(res);
+       py_ret = PyLdbResult_FromResult(res);
+
+       talloc_free(res);
+
+       return py_ret;
 }
 
 static PyObject *py_ldb_get_opaque(PyLdbObject *self, PyObject *args)
@@ -1009,7 +1039,7 @@ static PyObject *py_ldb_get_firstmodule(PyLdbObject *self, void *closure)
 }
 
 static PyGetSetDef py_ldb_getset[] = {
-       { (char *)"firstmodule", (getter)py_ldb_get_firstmodule, NULL, NULL },
+       { discard_const_p(char, "firstmodule"), (getter)py_ldb_get_firstmodule, NULL, NULL },
        { NULL }
 };
 
@@ -1026,7 +1056,7 @@ static int py_ldb_contains(PyLdbObject *self, PyObject *obj)
 
        ret = ldb_search(ldb_ctx, ldb_ctx, &result, dn, LDB_SCOPE_BASE, NULL, NULL);
        if (ret != LDB_SUCCESS) {
-               PyErr_SetLdbError(ret, ldb_ctx);
+               PyErr_SetLdbError(PyExc_LdbError, ret, ldb_ctx);
                return -1;
        }
 
@@ -1106,28 +1136,47 @@ static PyObject *py_ldb_module_del_transaction(PyLdbModuleObject *self)
 
 static PyObject *py_ldb_module_search(PyLdbModuleObject *self, PyObject *args, PyObject *kwargs)
 {
-       PyObject *py_base, *py_tree, *py_attrs;
+       PyObject *py_base, *py_tree, *py_attrs, *py_ret;
        int ret, scope;
        struct ldb_request *req;
-       const char *kwnames[] = { "base", "scope", "tree", "attrs", NULL };
+       const char * const kwnames[] = { "base", "scope", "tree", "attrs", NULL };
        struct ldb_module *mod;
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OiOO", (char **)kwnames, 
-                                                                        &py_base, &scope, &py_tree, &py_attrs))
+       const char * const*attrs;
+
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OiOO",
+                                        discard_const_p(char *, kwnames),
+                                        &py_base, &scope, &py_tree, &py_attrs))
                return NULL;
 
        mod = self->mod;
 
+       if (py_attrs == Py_None) {
+               attrs = NULL;
+       } else {
+               attrs = PyList_AsStringList(NULL, py_attrs, "attrs");
+               if (attrs == NULL)
+                       return NULL;
+       }
+
        ret = ldb_build_search_req(&req, mod->ldb, NULL, PyLdbDn_AsDn(py_base), 
-                            scope, NULL /* expr */, py_attrs == Py_None?NULL:PyList_AsStringList(req, py_attrs),
+                            scope, NULL /* expr */, attrs,
                             NULL /* controls */, NULL, NULL, NULL);
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, mod->ldb);
+
+       talloc_steal(req, attrs);
+
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, mod->ldb);
+
+       req->op.search.res = NULL;
 
        ret = mod->ops->search(mod, req);
-       talloc_free(req);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, mod->ldb);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, mod->ldb);
+
+       py_ret = PyLdbResult_FromResult(req->op.search.res);
 
-       return PyLdbResult_FromResult(req->op.search.res);
+       talloc_free(req);
+
+       return py_ret;  
 }
 
 
@@ -1148,7 +1197,7 @@ static PyObject *py_ldb_module_add(PyLdbModuleObject *self, PyObject *args)
        mod = PyLdbModule_AsModule(self);
        ret = mod->ops->add(mod, req);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, mod->ldb);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, mod->ldb);
 
        Py_RETURN_NONE;
 }
@@ -1162,15 +1211,15 @@ static PyObject *py_ldb_module_modify(PyLdbModuleObject *self, PyObject *args)
 
        if (!PyArg_ParseTuple(args, "O", &py_message))
                return NULL;
-       
+
        req = talloc_zero(NULL, struct ldb_request);
        req->operation = LDB_MODIFY;
        req->op.mod.message = PyLdbMessage_AsMessage(py_message);
-       
+
        mod = PyLdbModule_AsModule(self);
        ret = mod->ops->modify(mod, req);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, mod->ldb);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, mod->ldb);
 
        Py_RETURN_NONE;
 }
@@ -1183,14 +1232,14 @@ static PyObject *py_ldb_module_delete(PyLdbModuleObject *self, PyObject *args)
 
        if (!PyArg_ParseTuple(args, "O", &py_dn))
                return NULL;
-       
+
        req = talloc_zero(NULL, struct ldb_request);
        req->operation = LDB_DELETE;
        req->op.del.dn = PyLdbDn_AsDn(py_dn);
-       
+
        ret = PyLdbModule_AsModule(self)->ops->del(PyLdbModule_AsModule(self), req);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, NULL);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, NULL);
 
        Py_RETURN_NONE;
 }
@@ -1203,16 +1252,16 @@ static PyObject *py_ldb_module_rename(PyLdbModuleObject *self, PyObject *args)
 
        if (!PyArg_ParseTuple(args, "OO", &py_dn1, &py_dn2))
                return NULL;
-       
+
        req = talloc_zero(NULL, struct ldb_request);
 
        req->operation = LDB_RENAME;
        req->op.rename.olddn = PyLdbDn_AsDn(py_dn1);
        req->op.rename.newdn = PyLdbDn_AsDn(py_dn2);
-       
+
        ret = PyLdbModule_AsModule(self)->ops->rename(PyLdbModule_AsModule(self), req);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, NULL);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, NULL);
 
        Py_RETURN_NONE;
 }
@@ -1245,6 +1294,21 @@ PyTypeObject PyLdbModule = {
        .tp_flags = Py_TPFLAGS_DEFAULT,
 };
 
+
+/**
+ * Create a ldb_message_element from a Python object.
+ *
+ * This will accept any sequence objects that contains strings, or 
+ * a string object.
+ *
+ * A reference to set_obj will be borrowed. 
+ *
+ * @param mem_ctx Memory context
+ * @param set_obj Python object to convert
+ * @param flags ldb_message_element flags to set
+ * @param attr_name Name of the attribute
+ * @return New ldb_message_element, allocated as child of mem_ctx
+ */
 struct ldb_message_element *PyObject_AsMessageElement(TALLOC_CTX *mem_ctx,
                                                                                           PyObject *set_obj, int flags,
                                                                                           const char *attr_name)
@@ -1262,9 +1326,7 @@ struct ldb_message_element *PyObject_AsMessageElement(TALLOC_CTX *mem_ctx,
                me->num_values = 1;
                me->values = talloc_array(me, struct ldb_val, me->num_values);
                me->values[0].length = PyString_Size(set_obj);
-               me->values[0].data = (uint8_t *)talloc_strndup(me->values,
-                                       PyString_AsString(set_obj),
-                                       me->values[0].length);
+               me->values[0].data = (uint8_t *)PyString_AsString(set_obj);
        } else if (PySequence_Check(set_obj)) {
                int i;
                me->num_values = PySequence_Size(set_obj);
@@ -1372,9 +1434,12 @@ static PyObject *py_ldb_msg_element_new(PyTypeObject *type, PyObject *args, PyOb
        struct ldb_message_element *el;
        int flags = 0;
        char *name = NULL;
-       const char *kwnames[] = { "elements", "flags", "name", NULL };
+       const char * const kwnames[] = { "elements", "flags", "name", NULL };
        PyLdbMessageElementObject *ret;
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|Ois", (char **)kwnames, &py_elements, &flags, &name))
+
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|Ois",
+                                        discard_const_p(char *, kwnames),
+                                        &py_elements, &flags, &name))
                return NULL;
 
        el = talloc_zero(NULL, struct ldb_message_element);
@@ -1596,11 +1661,14 @@ static PyMappingMethods py_ldb_msg_mapping = {
 
 static PyObject *py_ldb_msg_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
 {
-       const char *kwnames[] = { "dn", NULL };
+       const char * const kwnames[] = { "dn", NULL };
        struct ldb_message *ret;
        PyObject *pydn = NULL;
        PyLdbMessageObject *py_ret;
-       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|O", (char **)kwnames, &pydn))
+
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|O",
+                                        discard_const_p(char *, kwnames),
+                                        &pydn))
                return NULL;
 
        ret = ldb_msg_new(NULL);
@@ -1650,7 +1718,7 @@ static int py_ldb_msg_set_dn(PyLdbMessageObject *self, PyObject *value, void *cl
 }
 
 static PyGetSetDef py_ldb_msg_getset[] = {
-       { (char *)"dn", (getter)py_ldb_msg_get_dn, (setter)py_ldb_msg_set_dn, NULL },
+       { discard_const_p(char, "dn"), (getter)py_ldb_msg_get_dn, (setter)py_ldb_msg_set_dn, NULL },
        { NULL }
 };
 
@@ -1692,7 +1760,7 @@ PyObject *PyLdbTree_FromTree(struct ldb_parse_tree *tree)
                PyErr_NoMemory();
                return NULL;
        }
-       
+
        ret->mem_ctx = talloc_new(NULL);
        ret->tree = talloc_reference(ret->mem_ctx, tree);
        return (PyObject *)ret;
@@ -1714,7 +1782,7 @@ PyTypeObject PyLdbTree = {
 /* Ldb_module */
 static int py_module_search(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result, *py_base, *py_attrs, *py_tree;
 
        py_base = PyLdbDn_FromDn(req->op.search.base);
@@ -1737,7 +1805,9 @@ static int py_module_search(struct ldb_module *mod, struct ldb_request *req)
                        PyList_SetItem(py_attrs, i, PyString_FromString(req->op.search.attrs[i]));
        }
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"search", (char *)"OiOO", py_base, req->op.search.scope, py_tree, py_attrs);
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "search"),
+                                       discard_const_p(char, "OiOO"),
+                                       py_base, req->op.search.scope, py_tree, py_attrs);
 
        Py_DECREF(py_attrs);
        Py_DECREF(py_tree);
@@ -1759,16 +1829,18 @@ static int py_module_search(struct ldb_module *mod, struct ldb_request *req)
 
 static int py_module_add(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result, *py_msg;
 
-       py_msg = PyLdbMessage_FromMessage((struct ldb_message *)req->op.add.message);
+       py_msg = PyLdbMessage_FromMessage(discard_const_p(struct ldb_message, req->op.add.message));
 
        if (py_msg == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"add", (char *)"O", py_msg);
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "add"),
+                                       discard_const_p(char, "O"),
+                                       py_msg);
 
        Py_DECREF(py_msg);
 
@@ -1783,16 +1855,18 @@ static int py_module_add(struct ldb_module *mod, struct ldb_request *req)
 
 static int py_module_modify(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result, *py_msg;
 
-       py_msg = PyLdbMessage_FromMessage((struct ldb_message *)req->op.mod.message);
+       py_msg = PyLdbMessage_FromMessage(discard_const_p(struct ldb_message, req->op.mod.message));
 
        if (py_msg == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"modify", (char *)"O", py_msg);
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "modify"),
+                                       discard_const_p(char, "O"),
+                                       py_msg);
 
        Py_DECREF(py_msg);
 
@@ -1807,7 +1881,7 @@ static int py_module_modify(struct ldb_module *mod, struct ldb_request *req)
 
 static int py_module_del(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result, *py_dn;
 
        py_dn = PyLdbDn_FromDn(req->op.del.dn);
@@ -1815,7 +1889,9 @@ static int py_module_del(struct ldb_module *mod, struct ldb_request *req)
        if (py_dn == NULL)
                return LDB_ERR_OPERATIONS_ERROR;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"delete", (char *)"O", py_dn);
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "delete"),
+                                       discard_const_p(char, "O"),
+                                       py_dn);
 
        if (py_result == NULL) {
                return LDB_ERR_PYTHON_EXCEPTION;
@@ -1828,7 +1904,7 @@ static int py_module_del(struct ldb_module *mod, struct ldb_request *req)
 
 static int py_module_rename(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result, *py_olddn, *py_newdn;
 
        py_olddn = PyLdbDn_FromDn(req->op.rename.olddn);
@@ -1841,7 +1917,9 @@ static int py_module_rename(struct ldb_module *mod, struct ldb_request *req)
        if (py_newdn == NULL)
                return LDB_ERR_OPERATIONS_ERROR;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"rename", (char *)"OO", py_olddn, py_newdn);
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "rename"),
+                                       discard_const_p(char, "OO"),
+                                       py_olddn, py_newdn);
 
        Py_DECREF(py_olddn);
        Py_DECREF(py_newdn);
@@ -1857,30 +1935,33 @@ static int py_module_rename(struct ldb_module *mod, struct ldb_request *req)
 
 static int py_module_request(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"request", (char *)"");
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "request"),
+                                       discard_const_p(char, ""));
 
        return LDB_ERR_OPERATIONS_ERROR;
 }
 
 static int py_module_extended(struct ldb_module *mod, struct ldb_request *req)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"extended", (char *)"");
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "extended"),
+                                       discard_const_p(char, ""));
 
        return LDB_ERR_OPERATIONS_ERROR;
 }
 
 static int py_module_start_transaction(struct ldb_module *mod)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"start_transaction", (char *)"");
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "start_transaction"),
+                                       discard_const_p(char, ""));
 
        if (py_result == NULL) {
                return LDB_ERR_PYTHON_EXCEPTION;
@@ -1893,10 +1974,11 @@ static int py_module_start_transaction(struct ldb_module *mod)
 
 static int py_module_end_transaction(struct ldb_module *mod)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"end_transaction", (char *)"");
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "end_transaction"),
+                                       discard_const_p(char, ""));
 
        if (py_result == NULL) {
                return LDB_ERR_PYTHON_EXCEPTION;
@@ -1909,10 +1991,11 @@ static int py_module_end_transaction(struct ldb_module *mod)
 
 static int py_module_del_transaction(struct ldb_module *mod)
 {
-       PyObject *py_ldb = mod->private_data;
+       PyObject *py_ldb = (PyObject *)mod->private_data;
        PyObject *py_result;
 
-       py_result = PyObject_CallMethod(py_ldb, (char *)"del_transaction", (char *)"");
+       py_result = PyObject_CallMethod(py_ldb, discard_const_p(char, "del_transaction"),
+                                       discard_const_p(char, ""));
 
        if (py_result == NULL) {
                return LDB_ERR_PYTHON_EXCEPTION;
@@ -1929,9 +2012,9 @@ static int py_module_destructor(struct ldb_module *mod)
        return 0;
 }
 
-static int py_module_init (struct ldb_module *mod)
+static int py_module_init(struct ldb_module *mod)
 {
-       PyObject *py_class = mod->ops->private_data;
+       PyObject *py_class = (PyObject *)mod->ops->private_data;
        PyObject *py_result, *py_next, *py_ldb;
 
        py_ldb = PyLdb_FromLdbContext(mod->ldb);
@@ -1944,7 +2027,8 @@ static int py_module_init (struct ldb_module *mod)
        if (py_next == NULL)
                return LDB_ERR_OPERATIONS_ERROR;
 
-       py_result = PyObject_CallFunction(py_class, (char *)"OO", py_ldb, py_next);
+       py_result = PyObject_CallFunction(py_class, discard_const_p(char, "OO"),
+                                         py_ldb, py_next);
 
        if (py_result == NULL) {
                return LDB_ERR_PYTHON_EXCEPTION;
@@ -1972,7 +2056,7 @@ static PyObject *py_register_module(PyObject *module, PyObject *args)
                return NULL;
        }
 
-       ops->name = talloc_strdup(ops, PyString_AsString(PyObject_GetAttrString(input, (char *)"name")));
+       ops->name = talloc_strdup(ops, PyString_AsString(PyObject_GetAttrString(input, discard_const_p(char, "name"))));
 
        Py_INCREF(input);
        ops->private_data = input;
@@ -1990,7 +2074,7 @@ static PyObject *py_register_module(PyObject *module, PyObject *args)
 
        ret = ldb_register_module(ops);
 
-       PyErr_LDB_ERROR_IS_ERR_RAISE(ret, NULL);
+       PyErr_LDB_ERROR_IS_ERR_RAISE(PyExc_LdbError, ret, NULL);
 
        Py_RETURN_NONE;
 }
@@ -2123,7 +2207,7 @@ void initldb(void)
 
        PyModule_AddObject(m, "__docformat__", PyString_FromString("restructuredText"));
 
-       PyExc_LdbError = PyErr_NewException((char *)"_ldb.LdbError", NULL, NULL);
+       PyExc_LdbError = PyErr_NewException(discard_const_p(char, "_ldb.LdbError"), NULL, NULL);
        PyModule_AddObject(m, "LdbError", PyExc_LdbError);
 
        Py_INCREF(&PyLdb);