Handle better memory management of tracecmd_read_at

The memory management of the cached record let records leak like crazy
when tracecmd_read_at was used.

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

1 files changed, 7 insertions, 2 deletions

diff --git a/trace-input.c b/trace-input.c
index 6fee129..eead50c 100644
--- a/trace-input.c
+++ b/trace-input.c
@@ -593,7 +593,7 @@ static struct record *
 read_event(struct tracecmd_handle *handle, unsigned long long offset,
            int cpu)
 {
-        struct record *record;
+        struct record *record = NULL;
 
         /*
          * Since the timestamp is calculated from the beginnnig
@@ -607,8 +607,13 @@ read_event(struct tracecmd_handle *handle, unsigned long long offset,
                 handle->cpu_data[cpu].index = 0;
 
         do {
+                if (record)
+                        free(record);
                 /* Make sure peek returns new data */
-                handle->cpu_data[cpu].next = NULL;
+                if (handle->cpu_data[cpu].next) {
+                        free(handle->cpu_data[cpu].next);
+                        handle->cpu_data[cpu].next = NULL;
+                }
                 record = tracecmd_read_data(handle, cpu);
         } while (record && (record->offset + record->record_size) <= offset);