X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Futils%2Ftestapp%2Fgen-ust-nevents-str%2Fgen-ust-nevents-str.c;h=49ad5180678cac48a0f32feba8bc84cfb020a57b;hb=20dd2de1fa1efe1519d1b6e88386efa89d60d1b9;hp=6d90bfc66524c49cf961e7e624538d0e53905cbc;hpb=5400d18f909dbcf6cb8c794cdd33d0c2213fda18;p=lttng-tools.git

diff --git a/tests/utils/testapp/gen-ust-nevents-str/gen-ust-nevents-str.c b/tests/utils/testapp/gen-ust-nevents-str/gen-ust-nevents-str.c
index 6d90bfc66..49ad51806 100644
--- a/tests/utils/testapp/gen-ust-nevents-str/gen-ust-nevents-str.c
+++ b/tests/utils/testapp/gen-ust-nevents-str/gen-ust-nevents-str.c
@@ -17,15 +17,18 @@
 
 #define _LGPL_SOURCE
 #include <stdio.h>
+#include "signal-helper.h"
 
 #define TRACEPOINT_DEFINE
 #include "tp.h"
 
 int main(int argc, char **argv)
 {
-	int count;
-	int i;
-	int arg_i;
+	int count = 0, i = 0, arg_i = 0;
+
+	if (set_signal_handler()) {
+		return 1;
+	}
 
 	if (argc <= 3) {
 		fprintf(stderr, "Usage: %s COUNT STRING [STRING]...\n",
@@ -48,6 +51,9 @@ int main(int argc, char **argv)
 		if (arg_i == argc) {
 			arg_i = 2;
 		}
+		if (should_quit) {
+			break;
+		}
 	}
 
 	return 0;