X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Futils%2Ftestapp%2Fgen-ust-nevents-str%2Fgen-ust-nevents-str.c;h=904991a4d40cf055cfa8340c3d718c9854091a2e;hp=6d90bfc66524c49cf961e7e624538d0e53905cbc;hb=95983a02c557a99506716fc2c28f74be37def2b3;hpb=9a318688b5958debecae5560842109d77abd996e

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..904991a4d 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,6 +17,7 @@
 
 #define _LGPL_SOURCE
 #include <stdio.h>
+#include "signal-helper.h"
 
 #define TRACEPOINT_DEFINE
 #include "tp.h"
@@ -27,6 +28,10 @@ int main(int argc, char **argv)
 	int i;
 	int arg_i;
 
+	if (set_signal_handler()) {
+		return 1;
+	}
+
 	if (argc <= 3) {
 		fprintf(stderr, "Usage: %s COUNT STRING [STRING]...\n",
 			argv[0]);
@@ -48,6 +53,9 @@ int main(int argc, char **argv)
 		if (arg_i == argc) {
 			arg_i = 2;
 		}
+		if (should_quit) {
+			break;
+		}
 	}
 
 	return 0;