From 229d270deb0ab74101dcc0cbee09edfbe8578dec Mon Sep 17 00:00:00 2001 From: Joseph Sutton Date: Fri, 7 Jul 2023 10:42:38 +1200 Subject: [PATCH] ldb: Check talloc_zero_array() return value Signed-off-by: Joseph Sutton Reviewed-by: Andrew Bartlett --- lib/ldb/pyldb.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/lib/ldb/pyldb.c b/lib/ldb/pyldb.c index 8d38ed83fc9..25c7bb17f41 100644 --- a/lib/ldb/pyldb.c +++ b/lib/ldb/pyldb.c @@ -1392,6 +1392,11 @@ static struct ldb_message *PyDict_AsMessage(TALLOC_CTX *mem_ctx, return NULL; } msg->elements = talloc_zero_array(msg, struct ldb_message_element, PyDict_Size(py_obj)); + if (msg->elements == NULL) { + PyErr_NoMemory(); + TALLOC_FREE(msg); + return NULL; + } if (dn_value) { if (!pyldb_Object_AsDn(msg, dn_value, ldb_ctx, &msg->dn)) { -- 2.47.3