X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Fplugins%2Fctf%2Flttng-live%2Flttng-live.c;h=774b91c8e39926c9bb7ee6c68b4cc53ca61da9a4;hb=f797424a5aa3b965937203073b27d666c25a6987;hp=af7237bdc2ed721c5d7d3f10e34678384ac735d8;hpb=4c5287b9f6618d173d210d82f365436c266cabd1;p=babeltrace.git

diff --git a/src/plugins/ctf/lttng-live/lttng-live.c b/src/plugins/ctf/lttng-live/lttng-live.c
index af7237bd..774b91c8 100644
--- a/src/plugins/ctf/lttng-live/lttng-live.c
+++ b/src/plugins/ctf/lttng-live/lttng-live.c
@@ -35,6 +35,7 @@
 
 #include <glib.h>
 #include <inttypes.h>
+#include <stdbool.h>
 #include <unistd.h>
 
 #include "common/assert.h"
@@ -455,13 +456,30 @@ enum lttng_live_iterator_status lttng_live_get_session(
 			status != LTTNG_LIVE_ITERATOR_STATUS_END) {
 		goto end;
 	}
-	for (trace_idx = 0; trace_idx < session->traces->len; trace_idx++) {
+	trace_idx = 0;
+	while (trace_idx < session->traces->len) {
 		struct lttng_live_trace *trace =
 			g_ptr_array_index(session->traces, trace_idx);
 
 		status = lttng_live_metadata_update(trace);
-		if (status != LTTNG_LIVE_ITERATOR_STATUS_OK &&
-				status != LTTNG_LIVE_ITERATOR_STATUS_END) {
+		switch (status) {
+		case LTTNG_LIVE_ITERATOR_STATUS_OK:
+			trace_idx++;
+			break;
+		case LTTNG_LIVE_ITERATOR_STATUS_END:
+			/*
+			 * The trace has ended. Remove it of the array an
+			 * continue the iteration.
+			 * We can remove the trace safely when using the
+			 * g_ptr_array_remove_index_fast because it replaces
+			 * the element at trace_idx with the array's last
+			 * element. trace_idx is not incremented because of
+			 * that.
+			 */
+			(void) g_ptr_array_remove_index_fast(session->traces,
+				trace_idx);
+			break;
+		default:
 			goto end;
 		}
 	}