X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Futils%2Ftestapp%2Fgen-ust-events%2Fgen-ust-events.cpp;h=3b62ed0dfee9a80baa82b6cb50c0c05dcd0232ec;hb=eb2dca3bdbb5f9d1fe8dc4c5d83951fe99ea0b29;hp=d64d55421a1e1caa30496c42cfc373e8f626f740;hpb=cd9adb8b829564212158943a0d279bb35322ab30;p=lttng-tools.git

diff --git a/tests/utils/testapp/gen-ust-events/gen-ust-events.cpp b/tests/utils/testapp/gen-ust-events/gen-ust-events.cpp
index d64d55421..3b62ed0df 100644
--- a/tests/utils/testapp/gen-ust-events/gen-ust-events.cpp
+++ b/tests/utils/testapp/gen-ust-events/gen-ust-events.cpp
@@ -32,12 +32,14 @@ static struct option long_options[] = {
 	/* These options set a flag. */
 	{ "iter", required_argument, nullptr, 'i' },
 	{ "wait", required_argument, nullptr, 'w' },
-	{ "sync-after-first-event", required_argument, nullptr, 'a' },
-	{ "sync-before-last-event", required_argument, nullptr, 'b' },
-	{ "sync-before-last-event-touch", required_argument, nullptr, 'c' },
-	{ "sync-before-exit", required_argument, nullptr, 'd' },
-	{ "sync-before-exit-touch", required_argument, nullptr, 'e' },
-	{ "emit-end-event", no_argument, nullptr, 'f' },
+	{ "sync-application-in-main-touch", required_argument, nullptr, 'a' },
+	{ "sync-before-first-event", required_argument, nullptr, 'b' },
+	{ "sync-after-first-event", required_argument, nullptr, 'c' },
+	{ "sync-before-last-event", required_argument, nullptr, 'd' },
+	{ "sync-before-last-event-touch", required_argument, nullptr, 'e' },
+	{ "sync-before-exit", required_argument, nullptr, 'f' },
+	{ "sync-before-exit-touch", required_argument, nullptr, 'g' },
+	{ "emit-end-event", no_argument, nullptr, 'h' },
 	{ nullptr, 0, nullptr, 0 }
 };
 
@@ -54,6 +56,8 @@ int main(int argc, char **argv)
 	uint32_t net_values[] = { 1, 2, 3 };
 	int nr_iter = 100, ret = 0, first_event_file_created = 0;
 	useconds_t nr_usec = 0;
+	char *application_in_main_file_path = nullptr;
+	char *before_first_event_file_path = nullptr;
 	char *after_first_event_file_path = nullptr;
 	char *before_last_event_file_path = nullptr;
 	/*
@@ -72,25 +76,31 @@ int main(int argc, char **argv)
 		net_values[i] = htonl(net_values[i]);
 	}
 
-	while ((option = getopt_long(argc, argv, "i:w:a:b:c:d:e:f", long_options, &option_index)) !=
-	       -1) {
+	while ((option = getopt_long(
+			argc, argv, "i:w:a:b:c:d:e:f:g:h", long_options, &option_index)) != -1) {
 		switch (option) {
 		case 'a':
-			after_first_event_file_path = strdup(optarg);
+			application_in_main_file_path = strdup(optarg);
 			break;
 		case 'b':
-			before_last_event_file_path = strdup(optarg);
+			before_first_event_file_path = strdup(optarg);
 			break;
 		case 'c':
-			before_last_event_file_path_touch = strdup(optarg);
+			after_first_event_file_path = strdup(optarg);
 			break;
 		case 'd':
-			before_exit_file_path = strdup(optarg);
+			before_last_event_file_path = strdup(optarg);
 			break;
 		case 'e':
-			before_exit_file_path_touch = strdup(optarg);
+			before_last_event_file_path_touch = strdup(optarg);
 			break;
 		case 'f':
+			before_exit_file_path = strdup(optarg);
+			break;
+		case 'g':
+			before_exit_file_path_touch = strdup(optarg);
+			break;
+		case 'h':
 			emit_end_event = true;
 			break;
 		case 'i':
@@ -129,6 +139,14 @@ int main(int argc, char **argv)
 		goto end;
 	}
 
+	if (application_in_main_file_path) {
+		create_file(application_in_main_file_path);
+	}
+
+	if (before_first_event_file_path) {
+		wait_on_file(before_first_event_file_path);
+	}
+
 	for (i = 0; nr_iter < 0 || i < nr_iter; i++) {
 		if (nr_iter >= 0 && i == nr_iter - 1) {
 			if (before_last_event_file_path_touch) {
@@ -204,6 +222,8 @@ int main(int argc, char **argv)
 		}
 	}
 end:
+	free(application_in_main_file_path);
+	free(before_first_event_file_path);
 	free(after_first_event_file_path);
 	free(before_last_event_file_path);
 	free(before_last_event_file_path_touch);