1#!/usr/bin/python 2# 3# This is a Hello World example that uses BPF_PERF_OUTPUT. 4 5from bcc import BPF 6from bcc.utils import printb 7 8# define BPF program 9prog = """ 10#include <linux/sched.h> 11 12// define output data structure in C 13struct data_t { 14 u32 pid; 15 u64 ts; 16 char comm[TASK_COMM_LEN]; 17}; 18BPF_PERF_OUTPUT(events); 19 20int hello(struct pt_regs *ctx) { 21 struct data_t data = {}; 22 23 data.pid = bpf_get_current_pid_tgid(); 24 data.ts = bpf_ktime_get_ns(); 25 bpf_get_current_comm(&data.comm, sizeof(data.comm)); 26 27 events.perf_submit(ctx, &data, sizeof(data)); 28 29 return 0; 30} 31""" 32 33# load BPF program 34b = BPF(text=prog) 35b.attach_kprobe(event=b.get_syscall_fnname("clone"), fn_name="hello") 36 37# header 38print("%-18s %-16s %-6s %s" % ("TIME(s)", "COMM", "PID", "MESSAGE")) 39 40# process event 41start = 0 42def print_event(cpu, data, size): 43 global start 44 event = b["events"].event(data) 45 if start == 0: 46 start = event.ts 47 time_s = (float(event.ts - start)) / 1000000000 48 printb(b"%-18.9f %-16s %-6d %s" % (time_s, event.comm, event.pid, 49 b"Hello, perf_output!")) 50 51# loop with callback to print_event 52b["events"].open_perf_buffer(print_event) 53while 1: 54 try: 55 b.perf_buffer_poll() 56 except KeyboardInterrupt: 57 exit() 58