common: add lttng_dynamic_array_set_count()

[lttng-tools.git] / src / common / dynamic-array.c

diff --git a/src/common/dynamic-array.c b/src/common/dynamic-array.c
index 198df19bbaa89d5bff75911fdf2d1b915b0cfd3e..81c2c3efa15ba1927a61ecc608ea07a9bd9b6f07 100644 (file)
--- 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)