ldb: Check talloc_zero_array() return value

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/lib/ldb/pyldb.c b/lib/ldb/pyldb.c
index 8d38ed83fc9c150636b172311ee986b3a2781769..25c7bb17f41196f196f464450b8d82c33275128e 100644 (file)
--- 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)) {