pyldb: py_ldb_dn_len checks dn and ldb validity

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 18613f59f2f793358cc18b2ed259f976f43a5720..386c9478ea402d7f4cebb54e87aeed335c271684 100644 (file)
--- a/lib/ldb/pyldb.c
+++ b/lib/ldb/pyldb.c
@@ -938,7 +938,12 @@ static PyMethodDef py_ldb_dn_methods[] = {
 
 static Py_ssize_t py_ldb_dn_len(PyLdbDnObject *self)
 {
-       return ldb_dn_get_comp_num(pyldb_Dn_AS_DN((PyObject *)self));
+       struct ldb_dn *dn = pyldb_Dn_AS_DN(self);
+       if (dn == NULL || self->pyldb->ldb_ctx != ldb_dn_get_ldb_context(dn)) {
+               return -1;
+       }
+
+       return ldb_dn_get_comp_num(dn);
 }
 
 /*