pyldb: py_ldb_dn_get_extended_component() uses PyErr_LDB_DN_OR_RAISE

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/lib/ldb/pyldb.c b/lib/ldb/pyldb.c
index 98e0e4d9c8915cb798d360cbcc1a458607f2ce17..9fb256421fa8a6adf032038211d584a3cb2034a2 100644 (file)
--- a/lib/ldb/pyldb.c
+++ b/lib/ldb/pyldb.c
@@ -516,14 +516,16 @@ static PyObject *py_ldb_dn_extended_str(PyObject *self, PyObject *args, PyObject
        return PyUnicode_FromString(ldb_dn_get_extended_linearized(dn, dn, mode));
 }
 
-static PyObject *py_ldb_dn_get_extended_component(PyLdbDnObject *self, PyObject *args)
+static PyObject *py_ldb_dn_get_extended_component(PyObject *self, PyObject *args)
 {
        char *name;
-       const struct ldb_val *val;
+       const struct ldb_val *val = NULL;
+       struct ldb_dn *dn = NULL;
+       PyErr_LDB_DN_OR_RAISE(self, dn);
 
        if (!PyArg_ParseTuple(args, "s", &name))
                return NULL;
-       val = ldb_dn_get_extended_component(self->dn, name);
+       val = ldb_dn_get_extended_component(dn, name);
        if (val == NULL) {
                Py_RETURN_NONE;
        }