pyldb: py_ldb_msg_elements uses PyErr_LDB_MESSAGE_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 b0a0e94ef7cf2bc0b1b72abffbc24837b03db672..6691b8178461295bc06914ea5b6656cde5dbd34e 100644 (file)
--- a/lib/ldb/pyldb.c
+++ b/lib/ldb/pyldb.c
@@ -3905,12 +3905,15 @@ static PyObject *py_ldb_msg_items(PyObject *self,
        return l;
 }
 
-static PyObject *py_ldb_msg_elements(PyLdbMessageObject *self,
+static PyObject *py_ldb_msg_elements(PyObject *self,
                PyObject *Py_UNUSED(ignored))
 {
-       struct ldb_message *msg = pyldb_Message_AsMessage(self);
        Py_ssize_t i = 0;
-       PyObject *l = PyList_New(msg->num_elements);
+       PyObject *l = NULL;
+       struct ldb_message *msg = NULL;
+       PyErr_LDB_MESSAGE_OR_RAISE(self, msg);
+
+       l = PyList_New(msg->num_elements);
        if (l == NULL) {
                return NULL;
        }