From: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Date: Fri, 15 Mar 2024 04:16:34 +0000 (+1300)
Subject: pyldb: py_ldb_msg_elements uses PyErr_LDB_MESSAGE_OR_RAISE
X-Git-Tag: tdb-1.4.11~1196
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=61ba0cc17df6a0ecf59b61c7d6da3bb7c0bdeea3;p=thirdparty%2Fsamba.git

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 b0a0e94ef7c..6691b817846 100644
--- 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;
 	}