2 files changed, 16 insertions, 1 deletions

diff --git a/trace-cmd.h b/trace-cmd.h
index 83e7a34..22b45cc 100644
--- a/trace-cmd.h
+++ b/trace-cmd.h
@@ -87,6 +87,15 @@ int tracecmd_init_data(struct tracecmd_input *handle);
 struct record *
 tracecmd_peek_data(struct tracecmd_input *handle, int cpu);
 
+static inline struct record *
+tracecmd_peek_data_ref(struct tracecmd_input *handle, int cpu)
+{
+        struct record *rec = tracecmd_peek_data(handle, cpu);
+        if (rec)
+                rec->ref_count++;
+        return rec;
+}
+
 struct record *
 tracecmd_read_data(struct tracecmd_input *handle, int cpu);
 
diff --git a/tracecmd.py b/tracecmd.py
index 75a639d..39e7043 100644
--- a/tracecmd.py
+++ b/tracecmd.py
@@ -213,7 +213,13 @@ class Trace(object):
         return Event(self._pevent, rec, format)
 
     def peek_event(self, cpu):
-        pass
+        rec = tracecmd_peek_data_ref(self._handle, cpu)
+        if rec is None:
+            return None
+        type = pevent_data_type(self._pevent, rec)
+        format = pevent_data_event_from_type(self._pevent, type)
+        # rec ownership goes over to Event instance
+        return Event(self._pevent, rec, format)
 
 
 # Basic builtin test, execute module directly