parse-events: Add ref count to pevent

Add a ref counter to pevent so more than on object can share it and
not worry about which one will free it.

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

2 files changed, 19 insertions, 0 deletions

diff --git a/parse-events.c b/parse-events.c
index 672fd84..627f4a1 100644
--- a/parse-events.c
+++ b/parse-events.c
@@ -3830,10 +3830,16 @@ struct pevent *pevent_alloc(void)
         if (!pevent)
                 return NULL;
         memset(pevent, 0, sizeof(*pevent));
+        pevent->ref_count = 1;
 
         return pevent;
 }
 
+void pevent_ref(struct pevent *pevent)
+{
+        pevent->ref_count++;
+}
+
 static void free_format_fields(struct format_field *field)
 {
         struct format_field *next;
@@ -3878,6 +3884,10 @@ void pevent_free(struct pevent *pevent)
         struct printk_list *printklist = pevent->printklist, *printknext;
         int i;
 
+        pevent->ref_count--;
+        if (pevent->ref_count)
+                return;
+
         if (pevent->cmdlines) {
                 for (i = 0; i < pevent->cmdline_count; i++)
                         free(pevent->cmdlines[i].comm);
@@ -3930,3 +3940,8 @@ void pevent_free(struct pevent *pevent)
 
         free(pevent);
 }
+
+void pevent_unref(struct pevent *pevent)
+{
+        pevent_free(pevent);
+}
diff --git a/parse-events.h b/parse-events.h
index 1866299..e6ea2ab 100644
--- a/parse-events.h
+++ b/parse-events.h
@@ -226,6 +226,8 @@ struct func_map;
 struct func_list;
 
 struct pevent {
+        int ref_count;
+
         int header_page_ts_offset;
         int header_page_ts_size;
         int header_page_size_offset;
@@ -444,6 +446,8 @@ static inline void pevent_set_latency_format(struct pevent *pevent, int lat)
 
 struct pevent *pevent_alloc(void);
 void pevent_free(struct pevent *pevent);
+void pevent_ref(struct pevent *pevent);
+void pevent_unref(struct pevent *pevent);
 
 /* for debugging */
 void pevent_print_funcs(struct pevent *pevent);