1 files changed, 34 insertions, 5 deletions

diff --git a/trace-output.c b/trace-output.c
index a59099d..a45265d 100644
--- a/trace-output.c
+++ b/trace-output.c
@@ -520,8 +520,8 @@ static int read_ftrace_printk(struct tracecmd_output *handle)
         return 0;
 }
 
-static struct tracecmd_output *create_file(const char *output_file, int cpus,
-                                           struct tracecmd_input *ihandle)
+static struct tracecmd_output *
+create_file_fd(int fd, int cpus, struct tracecmd_input *ihandle)
 {
         struct tracecmd_output *handle;
         unsigned long long endian8;
@@ -539,9 +539,7 @@ static struct tracecmd_output *create_file(const char *output_file, int cpus,
                 return NULL;
         memset(handle, 0, sizeof(*handle));
 
-        handle->fd = open(output_file, O_RDWR | O_CREAT | O_TRUNC | O_LARGEFILE, 0644);
-        if (handle->fd < 0)
-                goto out_free;
+        handle->fd = fd;
 
         buf[0] = 23;
         buf[1] = 8;
@@ -631,6 +629,25 @@ static struct tracecmd_output *create_file(const char *output_file, int cpus,
         return NULL;
 }
 
+static struct tracecmd_output *create_file(const char *output_file, int cpus,
+                                           struct tracecmd_input *ihandle)
+{
+        struct tracecmd_output *handle;
+        int fd;
+
+        fd = open(output_file, O_RDWR | O_CREAT | O_TRUNC | O_LARGEFILE, 0644);
+        if (fd < 0)
+                return NULL;
+
+        handle = create_file_fd(fd, cpus, ihandle);
+        if (!handle) {
+                close(fd);
+                unlink(output_file);
+        }
+
+        return handle;
+}
+
 struct tracecmd_output *tracecmd_create_file_latency(const char *output_file, int cpus)
 {
         struct tracecmd_output *handle;
@@ -760,6 +777,18 @@ struct tracecmd_output *tracecmd_create_file(const char *output_file,
         return handle;
 }
 
+struct tracecmd_output *
+tracecmd_create_init_fd(int fd, int cpus)
+{
+        struct tracecmd_output *handle;
+
+        handle = create_file_fd(fd, cpus, NULL);
+        if (!handle)
+                return NULL;
+
+        return handle;
+}
+
 /**
  * tracecmd_copy - copy the headers of one trace.dat file for another
  * @ihandle: input handle of the trace.dat file to copy