X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Fdynamic-array.c;h=81c2c3efa15ba1927a61ecc608ea07a9bd9b6f07;hp=f737c982d6c1b7bfa39212a43ce6a7d404d63449;hb=44635d77b591f83a80d48cd93497bd1cd6aa788d;hpb=ab5be9fa2eb5ba9600a82cd18fd3cfcbac69169a

diff --git a/src/common/dynamic-array.c b/src/common/dynamic-array.c
index f737c982d..81c2c3efa 100644
--- a/src/common/dynamic-array.c
+++ b/src/common/dynamic-array.c
@@ -18,6 +18,35 @@ void lttng_dynamic_array_init(struct lttng_dynamic_array *array,
 	array->destructor = destructor;
 }
 
+LTTNG_HIDDEN
+int lttng_dynamic_array_set_count(struct lttng_dynamic_array *array,
+		size_t new_element_count)
+{
+	int ret;
+
+	if (!array) {
+		ret = -1;
+		goto end;
+	}
+
+	if (array->destructor) {
+		size_t i;
+
+		for (i = new_element_count; i < array->size; i++) {
+			void *element = lttng_dynamic_array_get_element(
+					array, i);
+
+			array->destructor(element);
+		}
+	}
+
+	array->size = new_element_count;
+	ret = lttng_dynamic_buffer_set_size(&array->buffer,
+			new_element_count * array->element_size);
+end:
+	return ret;
+}
+
 LTTNG_HIDDEN
 int lttng_dynamic_array_add_element(struct lttng_dynamic_array *array,
 		const void *element)
@@ -89,7 +118,7 @@ void lttng_dynamic_array_clear(struct lttng_dynamic_array *array)
 		}
 	}
 
-	lttng_dynamic_buffer_set_size(&array->buffer, 0);
+	(void) lttng_dynamic_buffer_set_size(&array->buffer, 0);
 	array->size = 0;
 }