d636a4f39476378e057154f14d51b8869fb5dff9
[sfrench/cifs-2.6.git] / tools / testing / selftests / bpf / prog_tests / task_fd_query_tp.c
1 // SPDX-License-Identifier: GPL-2.0
2 #include <test_progs.h>
3
4 static void test_task_fd_query_tp_core(const char *probe_name,
5                                        const char *tp_name)
6 {
7         const char *file = "./test_tracepoint.o";
8         int err, bytes, efd, prog_fd, pmu_fd;
9         struct perf_event_attr attr = {};
10         __u64 probe_offset, probe_addr;
11         __u32 len, prog_id, fd_type;
12         struct bpf_object *obj;
13         __u32 duration = 0;
14         char buf[256];
15
16         err = bpf_prog_load(file, BPF_PROG_TYPE_TRACEPOINT, &obj, &prog_fd);
17         if (CHECK(err, "bpf_prog_load", "err %d errno %d\n", err, errno))
18                 goto close_prog;
19
20         snprintf(buf, sizeof(buf),
21                  "/sys/kernel/debug/tracing/events/%s/id", probe_name);
22         efd = open(buf, O_RDONLY, 0);
23         if (CHECK(efd < 0, "open", "err %d errno %d\n", efd, errno))
24                 goto close_prog;
25         bytes = read(efd, buf, sizeof(buf));
26         close(efd);
27         if (CHECK(bytes <= 0 || bytes >= sizeof(buf), "read",
28                   "bytes %d errno %d\n", bytes, errno))
29                 goto close_prog;
30
31         attr.config = strtol(buf, NULL, 0);
32         attr.type = PERF_TYPE_TRACEPOINT;
33         attr.sample_type = PERF_SAMPLE_RAW;
34         attr.sample_period = 1;
35         attr.wakeup_events = 1;
36         pmu_fd = syscall(__NR_perf_event_open, &attr, -1 /* pid */,
37                          0 /* cpu 0 */, -1 /* group id */,
38                          0 /* flags */);
39         if (CHECK(err, "perf_event_open", "err %d errno %d\n", err, errno))
40                 goto close_pmu;
41
42         err = ioctl(pmu_fd, PERF_EVENT_IOC_ENABLE, 0);
43         if (CHECK(err, "perf_event_ioc_enable", "err %d errno %d\n", err,
44                   errno))
45                 goto close_pmu;
46
47         err = ioctl(pmu_fd, PERF_EVENT_IOC_SET_BPF, prog_fd);
48         if (CHECK(err, "perf_event_ioc_set_bpf", "err %d errno %d\n", err,
49                   errno))
50                 goto close_pmu;
51
52         /* query (getpid(), pmu_fd) */
53         len = sizeof(buf);
54         err = bpf_task_fd_query(getpid(), pmu_fd, 0, buf, &len, &prog_id,
55                                 &fd_type, &probe_offset, &probe_addr);
56         if (CHECK(err < 0, "bpf_task_fd_query", "err %d errno %d\n", err,
57                   errno))
58                 goto close_pmu;
59
60         err = (fd_type == BPF_FD_TYPE_TRACEPOINT) && !strcmp(buf, tp_name);
61         if (CHECK(!err, "check_results", "fd_type %d tp_name %s\n",
62                   fd_type, buf))
63                 goto close_pmu;
64
65         close(pmu_fd);
66         goto close_prog_noerr;
67
68 close_pmu:
69         close(pmu_fd);
70 close_prog:
71         error_cnt++;
72 close_prog_noerr:
73         bpf_object__close(obj);
74 }
75
76 void test_task_fd_query_tp(void)
77 {
78         test_task_fd_query_tp_core("sched/sched_switch",
79                                    "sched_switch");
80         test_task_fd_query_tp_core("syscalls/sys_enter_read",
81                                    "sys_enter_read");
82 }