From 78c9eb1a06a696fc6ba85110a4f7d661bbb661d3 Mon Sep 17 00:00:00 2001 From: Matthias Dieter Wallnöfer Date: Sun, 6 Mar 2011 20:16:43 +0100 Subject: ldb:ldb_controls.c - always allocate enough space The size for an additional "struct ldb_control" shouldn't hurt and so the excluded control can also be NULL. Added an ending "talloc_realloc" to resize the chunk to the effective needed size (requested by tridge). --- source4/lib/ldb/common/ldb_controls.c | 40 +++++++++++++++++++++++------------ 1 file changed, 27 insertions(+), 13 deletions(-) diff --git a/source4/lib/ldb/common/ldb_controls.c b/source4/lib/ldb/common/ldb_controls.c index 8c72250abf..8bf80f0dba 100644 --- a/source4/lib/ldb/common/ldb_controls.c +++ b/source4/lib/ldb/common/ldb_controls.c @@ -71,22 +71,26 @@ struct ldb_control *ldb_reply_get_control(struct ldb_reply *rep, const char *oid return NULL; } -/* saves the current controls list into the "saver" and replace the one in req with a new one excluding -the "exclude" control */ -/* returns 0 on error */ +/* + * Saves the current controls list into the "saver" and replace the one in "req" + * with a new one excluding the "exclude" control (if it is NULL then the list + * remains the same) + * + * Returns 0 on error. + */ int ldb_save_controls(struct ldb_control *exclude, struct ldb_request *req, struct ldb_control ***saver) { struct ldb_control **lcs; unsigned int i, j; *saver = req->controls; - for (i = 0; req->controls[i]; i++); - if (i == 1) { + for (i = 0; req->controls && req->controls[i]; i++); + if (i == 0) { req->controls = NULL; return 1; } - lcs = talloc_array(req, struct ldb_control *, i); + lcs = talloc_array(req, struct ldb_control *, i + 1); if (!lcs) { return 0; } @@ -98,25 +102,32 @@ int ldb_save_controls(struct ldb_control *exclude, struct ldb_request *req, stru } lcs[j] = NULL; - req->controls = lcs; + req->controls = talloc_realloc(req, lcs, struct ldb_control *, j + 1); + if (req->controls == NULL) { + return 0; + } return 1; } -/* Returns a list of controls, except the one specified. Included - * controls become a child of returned list if they were children of - * controls_in */ +/* + * Returns a list of controls, except the one specified with "exclude" (can + * also be NULL). Included controls become a child of returned list if they + * were children of "controls_in". + * + * Returns NULL on error (OOM) or an empty control list. + */ struct ldb_control **ldb_controls_except_specified(struct ldb_control **controls_in, TALLOC_CTX *mem_ctx, struct ldb_control *exclude) { struct ldb_control **lcs = NULL; - unsigned int i, j; + unsigned int i, j, n; for (i = 0; controls_in && controls_in[i]; i++); - if (i == 0) { return NULL; } + n = i; for (i = 0, j = 0; controls_in && controls_in[i]; i++) { if (exclude == controls_in[i]) continue; @@ -126,7 +137,8 @@ struct ldb_control **ldb_controls_except_specified(struct ldb_control **controls * control, or there were no controls, we * don't allocate at all, and just return * NULL */ - lcs = talloc_array(mem_ctx, struct ldb_control *, i); + lcs = talloc_array(mem_ctx, struct ldb_control *, + n + 1); if (!lcs) { return NULL; } @@ -138,6 +150,8 @@ struct ldb_control **ldb_controls_except_specified(struct ldb_control **controls } if (lcs) { lcs[j] = NULL; + + lcs = talloc_realloc(mem_ctx, lcs, struct ldb_control *, j + 1); } return lcs; -- cgit