Add trace_pipe utility

Author: r41k0u

BCC-Examples/hello_world.py

@@ -1,4 +1,4 @@
-from pythonbpf import bpf, section, bpfglobal, compile
+from pythonbpf import bpf, section, bpfglobal, BPF, trace_pipe
 from ctypes import c_void_p, c_int64
 
 
@@ -15,4 +15,8 @@ def LICENSE() -> str:
     return "GPL"
 
 
-compile()
+# compile()
+b = BPF()
+b.load_and_attach()
+
+trace_pipe()

BCC-Examples/sys_sync.py

@@ -0,0 +1,19 @@
+from pythonbpf import bpf, section, bpfglobal, compile
+from ctypes import c_void_p, c_int64
+
+
+@bpf
+@section("tracepoint/syscalls/sys_enter_sync")
+def hello_world(ctx: c_void_p) -> c_int64:
+    print("sys_sync() called")
+    return c_int64(0)
+
+
+@bpf
+@bpfglobal
+def LICENSE() -> str:
+    return "GPL"
+
+
+compile()
+print("Tracing sys_sync()... Ctrl-C to end.")

pythonbpf/__init__.py

@@ -1,5 +1,6 @@
 from .decorators import bpf, map, section, bpfglobal, struct
 from .codegen import compile_to_ir, compile, BPF
+from .utils import trace_pipe
 
 __all__ = [
     "bpf",
@@ -10,4 +11,5 @@
     "compile_to_ir",
     "compile",
     "BPF",
+    "trace_pipe",
 ]

pythonbpf/utils.py

@@ -0,0 +1,9 @@
+import subprocess
+
+
+def trace_pipe():
+    """Util to read from the trace pipe."""
+    try:
+        subprocess.run(["cat", "/sys/kernel/tracing/trace_pipe"])
+    except KeyboardInterrupt:
+        print("Tracing stopped.")