Fix: may be used uninitialized warnings

[lttng-tools.git] / tests / utils / testapp / gen-ust-nevents-str / gen-ust-nevents-str.c

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 6d90bfc66524c49cf961e7e624538d0e53905cbc..49ad5180678cac48a0f32feba8bc84cfb020a57b 100644 (file)
--- 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;