X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Futils%2Ftestapp%2Fgen-ust-events%2Fgen-ust-events.c;h=57c33f59728125fef76b1e79a58d978ba479d5e5;hp=df1e58e417fa5f464b50a01f52fc2e3a97755881;hb=6ffce1f5bbc5c3e956bc91ce801c779588bf37af;hpb=e45dd625d3e802d8e6e2ec3de180c73546e8f9fe

diff --git a/tests/utils/testapp/gen-ust-events/gen-ust-events.c b/tests/utils/testapp/gen-ust-events/gen-ust-events.c
index df1e58e41..57c33f597 100644
--- a/tests/utils/testapp/gen-ust-events/gen-ust-events.c
+++ b/tests/utils/testapp/gen-ust-events/gen-ust-events.c
@@ -38,6 +38,7 @@ static struct option long_options[] =
 	{"sync-before-last-event-touch", required_argument, 0, 'c'},
 	{"sync-before-exit", required_argument, 0, 'd'},
 	{"sync-before-exit-touch", required_argument, 0, 'e'},
+	{"emit-end-event", no_argument, 0, 'f'},
 	{0, 0, 0, 0}
 };
 
@@ -65,13 +66,15 @@ int main(int argc, char **argv)
 	char *before_exit_file_path_touch = NULL;
 	/* Wait on file before exiting */
 	char *before_exit_file_path = NULL;
+	/* Emit an end event */
+	bool emit_end_event = false;
 
 	for (i = 0; i < 3; i++) {
 		net_values[i] = htonl(net_values[i]);
 	}
 
-	while ((option = getopt_long(argc, argv, "i:w:a:b:c:d:",
-			long_options, &option_index)) != -1) {
+	while ((option = getopt_long(argc, argv, "i:w:a:b:c:d:e:f",
+				long_options, &option_index)) != -1) {
 		switch (option) {
 		case 'a':
 			after_first_event_file_path = strdup(optarg);
@@ -88,6 +91,9 @@ int main(int argc, char **argv)
 		case 'e':
 			before_exit_file_path_touch = strdup(optarg);
 			break;
+		case 'f':
+			emit_end_event = true;
+			break;
 		case 'i':
 			nr_iter = atoi(optarg);
 			break;
@@ -174,6 +180,10 @@ int main(int argc, char **argv)
 		}
 	}
 
+	if (emit_end_event) {
+		tracepoint(tp, end);
+	}
+
 	if (before_exit_file_path_touch) {
 		ret = create_file(before_exit_file_path_touch);
 		if (ret != 0) {