trace-cmd: Record CPU trace stats before disable all

The disable_all() function is called before we print out the per cpu
buffer stats. But the disable_all() function also resets those same stats
so the output is useless.

Instead of printing out the stats while tracing is still active (although
tracing_on is off), record the stats into individual per cpu trace_seq
buffers and then still print when all has been disabled.

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

1 files changed, 13 insertions, 8 deletions

diff --git a/trace-record.c b/trace-record.c
index 62758e8..45439f4 100644
--- a/trace-record.c
+++ b/trace-record.c
@@ -1426,8 +1426,6 @@ static void start_threads(void)
 {
         int i;
 
-        cpu_count = count_cpus();
-
         if (host)
                 setup_network();
 
@@ -1905,7 +1903,7 @@ void trace_record (int argc, char **argv)
         struct event_list *event;
         struct event_list *last_event;
         struct tracecmd_event_list *list;
-        struct trace_seq s;
+        struct trace_seq *s;
         char *date2ts = NULL;
         int record_all = 0;
         int disable = 0;
@@ -1921,6 +1919,8 @@ void trace_record (int argc, char **argv)
 
         int c;
 
+        cpu_count = count_cpus();
+
         if ((record = (strcmp(argv[1], "record") == 0)))
                 ; /* do nothing */
         else if (strcmp(argv[1], "start") == 0)
@@ -2184,6 +2184,8 @@ void trace_record (int argc, char **argv)
 
         set_options();
 
+        s = malloc_or_die(sizeof(*s) * cpu_count);
+
         if (record || extract) {
                 signal(SIGINT, finish);
                 if (!latency)
@@ -2219,6 +2221,11 @@ void trace_record (int argc, char **argv)
 
         stop_threads();
 
+        for (cpu = 0; cpu < cpu_count; cpu++) {
+                trace_seq_init(&s[cpu]);
+                trace_seq_printf(&s[cpu], "CPU: %d\n", cpu);
+                tracecmd_stat_cpu(&s[cpu], cpu);
+        }
 
         if (!keep)
                 disable_all();
@@ -2227,14 +2234,12 @@ void trace_record (int argc, char **argv)
                "  Note: \"entries\" are the entries left in the kernel ring buffer and are not\n"
                "        recorded in the trace data. They should all be zero.\n\n");
         for (cpu = 0; cpu < cpu_count; cpu++) {
-                trace_seq_init(&s);
-                trace_seq_printf(&s, "CPU: %d\n", cpu);
-                tracecmd_stat_cpu(&s, cpu);
-                trace_seq_do_printf(&s);
-                trace_seq_destroy(&s);
+                trace_seq_do_printf(&s[cpu]);
+                trace_seq_destroy(&s[cpu]);
                 printf("\n");
         }
 
+
         /* extract records the date after extraction */
         if (extract && date)
                 date2ts = get_date_to_ts();