1// SPDX-License-Identifier: GPL-2.0 2#include <test_progs.h> 3 4void serial_test_tp_attach_query(void) 5{ 6 const int num_progs = 3; 7 int i, j, bytes, efd, err, prog_fd[num_progs], pmu_fd[num_progs]; 8 __u32 duration = 0, info_len, saved_prog_ids[num_progs]; 9 const char *file = "./test_tracepoint.bpf.o"; 10 struct perf_event_query_bpf *query; 11 struct perf_event_attr attr = {}; 12 struct bpf_object *obj[num_progs]; 13 struct bpf_prog_info prog_info; 14 char buf[256]; 15 16 for (i = 0; i < num_progs; i++) 17 obj[i] = NULL; 18 19 if (access("/sys/kernel/tracing/trace", F_OK) == 0) { 20 snprintf(buf, sizeof(buf), 21 "/sys/kernel/tracing/events/sched/sched_switch/id"); 22 } else { 23 snprintf(buf, sizeof(buf), 24 "/sys/kernel/debug/tracing/events/sched/sched_switch/id"); 25 } 26 efd = open(buf, O_RDONLY, 0); 27 if (CHECK(efd < 0, "open", "err %d errno %d\n", efd, errno)) 28 return; 29 bytes = read(efd, buf, sizeof(buf)); 30 close(efd); 31 if (CHECK(bytes <= 0 || bytes >= sizeof(buf), 32 "read", "bytes %d errno %d\n", bytes, errno)) 33 return; 34 35 attr.config = strtol(buf, NULL, 0); 36 attr.type = PERF_TYPE_TRACEPOINT; 37 attr.sample_type = PERF_SAMPLE_RAW | PERF_SAMPLE_CALLCHAIN; 38 attr.sample_period = 1; 39 attr.wakeup_events = 1; 40 41 query = malloc(sizeof(*query) + sizeof(__u32) * num_progs); 42 for (i = 0; i < num_progs; i++) { 43 err = bpf_prog_test_load(file, BPF_PROG_TYPE_TRACEPOINT, &obj[i], 44 &prog_fd[i]); 45 if (CHECK(err, "prog_load", "err %d errno %d\n", err, errno)) 46 goto cleanup1; 47 48 bzero(&prog_info, sizeof(prog_info)); 49 prog_info.jited_prog_len = 0; 50 prog_info.xlated_prog_len = 0; 51 prog_info.nr_map_ids = 0; 52 info_len = sizeof(prog_info); 53 err = bpf_prog_get_info_by_fd(prog_fd[i], &prog_info, 54 &info_len); 55 if (CHECK(err, "bpf_prog_get_info_by_fd", "err %d errno %d\n", 56 err, errno)) 57 goto cleanup1; 58 saved_prog_ids[i] = prog_info.id; 59 60 pmu_fd[i] = syscall(__NR_perf_event_open, &attr, -1 /* pid */, 61 0 /* cpu 0 */, -1 /* group id */, 62 0 /* flags */); 63 if (CHECK(pmu_fd[i] < 0, "perf_event_open", "err %d errno %d\n", 64 pmu_fd[i], errno)) 65 goto cleanup2; 66 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_ENABLE, 0); 67 if (CHECK(err, "perf_event_ioc_enable", "err %d errno %d\n", 68 err, errno)) 69 goto cleanup3; 70 71 if (i == 0) { 72 /* check NULL prog array query */ 73 query->ids_len = num_progs; 74 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_QUERY_BPF, query); 75 if (CHECK(err || query->prog_cnt != 0, 76 "perf_event_ioc_query_bpf", 77 "err %d errno %d query->prog_cnt %u\n", 78 err, errno, query->prog_cnt)) 79 goto cleanup3; 80 } 81 82 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_SET_BPF, prog_fd[i]); 83 if (CHECK(err, "perf_event_ioc_set_bpf", "err %d errno %d\n", 84 err, errno)) 85 goto cleanup3; 86 87 if (i == 1) { 88 /* try to get # of programs only */ 89 query->ids_len = 0; 90 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_QUERY_BPF, query); 91 if (CHECK(err || query->prog_cnt != 2, 92 "perf_event_ioc_query_bpf", 93 "err %d errno %d query->prog_cnt %u\n", 94 err, errno, query->prog_cnt)) 95 goto cleanup3; 96 97 /* try a few negative tests */ 98 /* invalid query pointer */ 99 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_QUERY_BPF, 100 (struct perf_event_query_bpf *)0x1); 101 if (CHECK(!err || errno != EFAULT, 102 "perf_event_ioc_query_bpf", 103 "err %d errno %d\n", err, errno)) 104 goto cleanup3; 105 106 /* no enough space */ 107 query->ids_len = 1; 108 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_QUERY_BPF, query); 109 if (CHECK(!err || errno != ENOSPC || query->prog_cnt != 2, 110 "perf_event_ioc_query_bpf", 111 "err %d errno %d query->prog_cnt %u\n", 112 err, errno, query->prog_cnt)) 113 goto cleanup3; 114 } 115 116 query->ids_len = num_progs; 117 err = ioctl(pmu_fd[i], PERF_EVENT_IOC_QUERY_BPF, query); 118 if (CHECK(err || query->prog_cnt != (i + 1), 119 "perf_event_ioc_query_bpf", 120 "err %d errno %d query->prog_cnt %u\n", 121 err, errno, query->prog_cnt)) 122 goto cleanup3; 123 for (j = 0; j < i + 1; j++) 124 if (CHECK(saved_prog_ids[j] != query->ids[j], 125 "perf_event_ioc_query_bpf", 126 "#%d saved_prog_id %x query prog_id %x\n", 127 j, saved_prog_ids[j], query->ids[j])) 128 goto cleanup3; 129 } 130 131 i = num_progs - 1; 132 for (; i >= 0; i--) { 133 cleanup3: 134 ioctl(pmu_fd[i], PERF_EVENT_IOC_DISABLE); 135 cleanup2: 136 close(pmu_fd[i]); 137 cleanup1: 138 bpf_object__close(obj[i]); 139 } 140 free(query); 141} 142