1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun #include <stdio.h>
3*4882a593Smuzhiyun #include <fcntl.h>
4*4882a593Smuzhiyun #include <poll.h>
5*4882a593Smuzhiyun #include <time.h>
6*4882a593Smuzhiyun #include <signal.h>
7*4882a593Smuzhiyun #include <bpf/libbpf.h>
8*4882a593Smuzhiyun
time_get_ns(void)9*4882a593Smuzhiyun static __u64 time_get_ns(void)
10*4882a593Smuzhiyun {
11*4882a593Smuzhiyun struct timespec ts;
12*4882a593Smuzhiyun
13*4882a593Smuzhiyun clock_gettime(CLOCK_MONOTONIC, &ts);
14*4882a593Smuzhiyun return ts.tv_sec * 1000000000ull + ts.tv_nsec;
15*4882a593Smuzhiyun }
16*4882a593Smuzhiyun
17*4882a593Smuzhiyun static __u64 start_time;
18*4882a593Smuzhiyun static __u64 cnt;
19*4882a593Smuzhiyun
20*4882a593Smuzhiyun #define MAX_CNT 100000ll
21*4882a593Smuzhiyun
print_bpf_output(void * ctx,int cpu,void * data,__u32 size)22*4882a593Smuzhiyun static void print_bpf_output(void *ctx, int cpu, void *data, __u32 size)
23*4882a593Smuzhiyun {
24*4882a593Smuzhiyun struct {
25*4882a593Smuzhiyun __u64 pid;
26*4882a593Smuzhiyun __u64 cookie;
27*4882a593Smuzhiyun } *e = data;
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun if (e->cookie != 0x12345678) {
30*4882a593Smuzhiyun printf("BUG pid %llx cookie %llx sized %d\n",
31*4882a593Smuzhiyun e->pid, e->cookie, size);
32*4882a593Smuzhiyun return;
33*4882a593Smuzhiyun }
34*4882a593Smuzhiyun
35*4882a593Smuzhiyun cnt++;
36*4882a593Smuzhiyun
37*4882a593Smuzhiyun if (cnt == MAX_CNT) {
38*4882a593Smuzhiyun printf("recv %lld events per sec\n",
39*4882a593Smuzhiyun MAX_CNT * 1000000000ll / (time_get_ns() - start_time));
40*4882a593Smuzhiyun return;
41*4882a593Smuzhiyun }
42*4882a593Smuzhiyun }
43*4882a593Smuzhiyun
main(int argc,char ** argv)44*4882a593Smuzhiyun int main(int argc, char **argv)
45*4882a593Smuzhiyun {
46*4882a593Smuzhiyun struct perf_buffer_opts pb_opts = {};
47*4882a593Smuzhiyun struct bpf_link *link = NULL;
48*4882a593Smuzhiyun struct bpf_program *prog;
49*4882a593Smuzhiyun struct perf_buffer *pb;
50*4882a593Smuzhiyun struct bpf_object *obj;
51*4882a593Smuzhiyun int map_fd, ret = 0;
52*4882a593Smuzhiyun char filename[256];
53*4882a593Smuzhiyun FILE *f;
54*4882a593Smuzhiyun
55*4882a593Smuzhiyun snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
56*4882a593Smuzhiyun obj = bpf_object__open_file(filename, NULL);
57*4882a593Smuzhiyun if (libbpf_get_error(obj)) {
58*4882a593Smuzhiyun fprintf(stderr, "ERROR: opening BPF object file failed\n");
59*4882a593Smuzhiyun return 0;
60*4882a593Smuzhiyun }
61*4882a593Smuzhiyun
62*4882a593Smuzhiyun /* load BPF program */
63*4882a593Smuzhiyun if (bpf_object__load(obj)) {
64*4882a593Smuzhiyun fprintf(stderr, "ERROR: loading BPF object file failed\n");
65*4882a593Smuzhiyun goto cleanup;
66*4882a593Smuzhiyun }
67*4882a593Smuzhiyun
68*4882a593Smuzhiyun map_fd = bpf_object__find_map_fd_by_name(obj, "my_map");
69*4882a593Smuzhiyun if (map_fd < 0) {
70*4882a593Smuzhiyun fprintf(stderr, "ERROR: finding a map in obj file failed\n");
71*4882a593Smuzhiyun goto cleanup;
72*4882a593Smuzhiyun }
73*4882a593Smuzhiyun
74*4882a593Smuzhiyun prog = bpf_object__find_program_by_name(obj, "bpf_prog1");
75*4882a593Smuzhiyun if (libbpf_get_error(prog)) {
76*4882a593Smuzhiyun fprintf(stderr, "ERROR: finding a prog in obj file failed\n");
77*4882a593Smuzhiyun goto cleanup;
78*4882a593Smuzhiyun }
79*4882a593Smuzhiyun
80*4882a593Smuzhiyun link = bpf_program__attach(prog);
81*4882a593Smuzhiyun if (libbpf_get_error(link)) {
82*4882a593Smuzhiyun fprintf(stderr, "ERROR: bpf_program__attach failed\n");
83*4882a593Smuzhiyun link = NULL;
84*4882a593Smuzhiyun goto cleanup;
85*4882a593Smuzhiyun }
86*4882a593Smuzhiyun
87*4882a593Smuzhiyun pb_opts.sample_cb = print_bpf_output;
88*4882a593Smuzhiyun pb = perf_buffer__new(map_fd, 8, &pb_opts);
89*4882a593Smuzhiyun ret = libbpf_get_error(pb);
90*4882a593Smuzhiyun if (ret) {
91*4882a593Smuzhiyun printf("failed to setup perf_buffer: %d\n", ret);
92*4882a593Smuzhiyun return 1;
93*4882a593Smuzhiyun }
94*4882a593Smuzhiyun
95*4882a593Smuzhiyun f = popen("taskset 1 dd if=/dev/zero of=/dev/null", "r");
96*4882a593Smuzhiyun (void) f;
97*4882a593Smuzhiyun
98*4882a593Smuzhiyun start_time = time_get_ns();
99*4882a593Smuzhiyun while ((ret = perf_buffer__poll(pb, 1000)) >= 0 && cnt < MAX_CNT) {
100*4882a593Smuzhiyun }
101*4882a593Smuzhiyun kill(0, SIGINT);
102*4882a593Smuzhiyun
103*4882a593Smuzhiyun cleanup:
104*4882a593Smuzhiyun bpf_link__destroy(link);
105*4882a593Smuzhiyun bpf_object__close(obj);
106*4882a593Smuzhiyun return ret;
107*4882a593Smuzhiyun }
108