trace-cmd: Add signal handler for SIGINT in trace-cmd read

When python plugins are loaded, they override SIGINT, and it becomes
quite annoying that you can't stop a print with Ctrl-C. By adding
a signal handler for SIGINT, we can stop the output with Ctrl-C.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

1 files changed, 9 insertions, 0 deletions

diff --git a/trace-read.c b/trace-read.c
index bbe34ba..a5c27e2 100644
--- a/trace-read.c
+++ b/trace-read.c
@@ -32,6 +32,7 @@
 #include <sys/mman.h>
 #include <pthread.h>
 #include <fcntl.h>
+#include <signal.h>
 #include <unistd.h>
 #include <ctype.h>
 #include <errno.h>
@@ -553,6 +554,12 @@ struct tracecmd_input *read_trace_header(void)
         return tracecmd_alloc_fd(input_fd);
 }
 
+static void sig_end(int sig)
+{
+        fprintf(stderr, "trace-cmd: Received SIGINT\n");
+        exit(0);
+}
+
 void trace_report (int argc, char **argv)
 {
         struct tracecmd_input *handle;
@@ -574,6 +581,8 @@ void trace_report (int argc, char **argv)
         if (strcmp(argv[1], "report") != 0)
                 usage(argv);
 
+        signal(SIGINT, sig_end);
+
         for (;;) {
                 int option_index = 0;
                 static struct option long_options[] = {