From: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Date: Thu, 14 Mar 2024 22:00:50 +0000 (+1300)
Subject: pyldb: py_ldb_dn_get_extended_component uses PyErr_LDB_DN_OR_RAISE()
X-Git-Tag: tdb-1.4.11~1216
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=85ba5d2c8f2708d284a973bb6b372faea542a393;p=thirdparty%2Fsamba.git

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 2d713ea00f4..0987aaf262f 100644
--- a/lib/ldb/pyldb.c
+++ b/lib/ldb/pyldb.c
@@ -526,23 +526,25 @@ static PyObject *py_ldb_dn_get_extended_component(PyLdbDnObject *self, PyObject
 	return PyBytes_FromStringAndSize((const char *)val->data, val->length);
 }
 
-static PyObject *py_ldb_dn_set_extended_component(PyLdbDnObject *self, PyObject *args)
+static PyObject *py_ldb_dn_set_extended_component(PyObject *self, PyObject *args)
 {
 	char *name;
 	int err;
 	uint8_t *value = NULL;
 	Py_ssize_t size = 0;
+	struct ldb_dn *dn = NULL;
+	PyErr_LDB_DN_OR_RAISE(self, dn);
 
 	if (!PyArg_ParseTuple(args, "sz#", &name, (char **)&value, &size))
 		return NULL;
 
 	if (value == NULL) {
-		err = ldb_dn_set_extended_component(self->dn, name, NULL);
+		err = ldb_dn_set_extended_component(dn, name, NULL);
 	} else {
 		struct ldb_val val;
 		val.data = (uint8_t *)value;
 		val.length = size;
-		err = ldb_dn_set_extended_component(self->dn, name, &val);
+		err = ldb_dn_set_extended_component(dn, name, &val);
 	}
 
 	if (err != LDB_SUCCESS) {