rex/tracepoint: implement PerfEventStreamer trait

Signed-off-by: MinhPhan8803 <[email protected]>

Author: MinhPhan8803

rex/src/tracepoint/tp_impl.rs

@@ -1,8 +1,12 @@
+use crate::base_helper::termination_check;
 use crate::bindings::uapi::linux::bpf::{
     bpf_map_type, BPF_PROG_TYPE_TRACEPOINT,
 };
+use crate::ffi;
+use crate::map::RexPerfEventArray;
 use crate::prog_type::rex_prog;
 use crate::task_struct::TaskStruct;
+use crate::utils::{to_result, NoRef, PerfEventMaskedCPU, PerfEventStreamer};
 use crate::Result;
 
 use super::{
@@ -65,3 +69,26 @@ impl<C: TracepointContext + 'static> rex_prog for tracepoint<C> {
         ((self.prog)(self, newctx)).unwrap_or_else(|e| e) as u32
     }
 }
+
+impl<C: TracepointContext + 'static> PerfEventStreamer for tracepoint<C> {
+    type Context = C;
+    fn output_event<T: Copy + NoRef>(
+        &self,
+        ctx: &Self::Context,
+        map: &'static RexPerfEventArray<T>,
+        data: &T,
+        cpu: PerfEventMaskedCPU,
+    ) -> Result {
+        let map_kptr = unsafe { core::ptr::read_volatile(&map.kptr) };
+        let ctx_ptr = ctx as *const C as *const ();
+        termination_check!(unsafe {
+            to_result!(ffi::bpf_perf_event_output_tp(
+                ctx_ptr,
+                map_kptr,
+                cpu.masked_cpu,
+                data as *const T as *const (),
+                core::mem::size_of::<T>() as u64
+            ))
+        })
+    }
+}