trace-cmd: Always append '.1' to output file in trace-cmd split

If a output file is specified, always append a '.1' to it.
This removes the confusion of where the output file is written to
and it could cause data to be overwritten that was not expected.

Reported-by: Li Zefan <lizf@cn.fujitsu.com>
Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

1 files changed, 6 insertions, 7 deletions

diff --git a/trace-split.c b/trace-split.c
index 489e6cf..d3594ab 100644
--- a/trace-split.c
+++ b/trace-split.c
@@ -489,13 +489,12 @@ void trace_split (int argc, char **argv)
 
         pevent = tracecmd_get_pevent(handle);
 
-        if (!output) {
-                if (repeat)
-                        output = strdup(input_file);
-                else {
-                        output = malloc_or_die(strlen(input_file) + 3);
-                        sprintf(output, "%s.1", input_file);
-                }
+        if (!output)
+                output = strdup(input_file);
+
+        if (!repeat) {
+                output = realloc(output, strlen(output) + 3);
+                strcat(output, ".1");
         }
 
         current = start_ns;