1 // SPDX-License-Identifier: GPL-2.0 2 // Copyright (c) 2020 Facebook 3 #include <linux/bpf.h> 4 #include <asm/unistd.h> 5 #include <bpf/bpf_helpers.h> 6 #include <bpf/bpf_tracing.h> 7 #include "bpf_misc.h" 8 9 char _license[] SEC("license") = "GPL"; 10 11 #define CPU_MASK 255 12 #define MAX_CPUS (CPU_MASK + 1) /* should match MAX_BUCKETS in benchs/bench_trigger.c */ 13 14 /* matches struct counter in bench.h */ 15 struct counter { 16 long value; 17 } __attribute__((aligned(128))); 18 19 struct counter hits[MAX_CPUS]; 20 21 static __always_inline void inc_counter(void) 22 { 23 int cpu = bpf_get_smp_processor_id(); 24 25 __sync_add_and_fetch(&hits[cpu & CPU_MASK].value, 1); 26 } 27 28 SEC("?uprobe") 29 int bench_trigger_uprobe(void *ctx) 30 { 31 inc_counter(); 32 return 0; 33 } 34 35 const volatile int batch_iters = 0; 36 37 SEC("?raw_tp") 38 int trigger_count(void *ctx) 39 { 40 int i; 41 42 for (i = 0; i < batch_iters; i++) 43 inc_counter(); 44 45 return 0; 46 } 47 48 SEC("?raw_tp") 49 int trigger_driver(void *ctx) 50 { 51 int i; 52 53 for (i = 0; i < batch_iters; i++) 54 (void)bpf_get_numa_node_id(); /* attach point for benchmarking */ 55 56 return 0; 57 } 58 59 SEC("?kprobe/bpf_get_numa_node_id") 60 int bench_trigger_kprobe(void *ctx) 61 { 62 inc_counter(); 63 return 0; 64 } 65 66 SEC("?kretprobe/bpf_get_numa_node_id") 67 int bench_trigger_kretprobe(void *ctx) 68 { 69 inc_counter(); 70 return 0; 71 } 72 73 SEC("?kprobe.multi/bpf_get_numa_node_id") 74 int bench_trigger_kprobe_multi(void *ctx) 75 { 76 inc_counter(); 77 return 0; 78 } 79 80 SEC("?kretprobe.multi/bpf_get_numa_node_id") 81 int bench_trigger_kretprobe_multi(void *ctx) 82 { 83 inc_counter(); 84 return 0; 85 } 86 87 SEC("?fentry/bpf_get_numa_node_id") 88 int bench_trigger_fentry(void *ctx) 89 { 90 inc_counter(); 91 return 0; 92 } 93 94 SEC("?fexit/bpf_get_numa_node_id") 95 int bench_trigger_fexit(void *ctx) 96 { 97 inc_counter(); 98 return 0; 99 } 100