125763b3cSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only
21da236b6SYonghong Song /* Copyright (c) 2017 Facebook
31da236b6SYonghong Song */
41da236b6SYonghong Song #include <stdio.h>
51da236b6SYonghong Song #include <unistd.h>
61da236b6SYonghong Song #include <fcntl.h>
71da236b6SYonghong Song #include <stdlib.h>
81da236b6SYonghong Song #include <string.h>
91da236b6SYonghong Song #include <linux/perf_event.h>
101da236b6SYonghong Song #include <errno.h>
11f0c328f8SDaniel T. Lee #include <bpf/libbpf.h>
122bf3e2efSJakub Kicinski #include <bpf/bpf.h>
131da236b6SYonghong Song
141da236b6SYonghong Song /* This program verifies bpf attachment to tracepoint sys_enter_* and sys_exit_*.
151da236b6SYonghong Song * This requires kernel CONFIG_FTRACE_SYSCALLS to be set.
161da236b6SYonghong Song */
171da236b6SYonghong Song
usage(const char * cmd)18a678be5cSYonghong Song static void usage(const char *cmd)
19a678be5cSYonghong Song {
20948189f6SJinghao Jia printf("USAGE: %s [-i nr_tests] [-h]\n", cmd);
21948189f6SJinghao Jia printf(" -i nr_tests # rounds of test to run\n");
22a678be5cSYonghong Song printf(" -h # help\n");
23a678be5cSYonghong Song }
24a678be5cSYonghong Song
verify_map(int map_id)251da236b6SYonghong Song static void verify_map(int map_id)
261da236b6SYonghong Song {
271da236b6SYonghong Song __u32 key = 0;
281da236b6SYonghong Song __u32 val;
291da236b6SYonghong Song
301da236b6SYonghong Song if (bpf_map_lookup_elem(map_id, &key, &val) != 0) {
311da236b6SYonghong Song fprintf(stderr, "map_lookup failed: %s\n", strerror(errno));
321da236b6SYonghong Song return;
331da236b6SYonghong Song }
34a678be5cSYonghong Song if (val == 0) {
351da236b6SYonghong Song fprintf(stderr, "failed: map #%d returns value 0\n", map_id);
36a678be5cSYonghong Song return;
37a678be5cSYonghong Song }
3835f91d1fSSong Chen
3935f91d1fSSong Chen printf("verify map:%d val: %d\n", map_id, val);
4035f91d1fSSong Chen
41a678be5cSYonghong Song val = 0;
42a678be5cSYonghong Song if (bpf_map_update_elem(map_id, &key, &val, BPF_ANY) != 0) {
43a678be5cSYonghong Song fprintf(stderr, "map_update failed: %s\n", strerror(errno));
44a678be5cSYonghong Song return;
45a678be5cSYonghong Song }
461da236b6SYonghong Song }
471da236b6SYonghong Song
test(char * filename,int nr_tests)48948189f6SJinghao Jia static int test(char *filename, int nr_tests)
491da236b6SYonghong Song {
50948189f6SJinghao Jia int map0_fds[nr_tests], map1_fds[nr_tests], fd, i, j = 0;
51*de4825a4SJinghao Jia struct bpf_link **links = NULL;
52948189f6SJinghao Jia struct bpf_object *objs[nr_tests];
53f0c328f8SDaniel T. Lee struct bpf_program *prog;
541da236b6SYonghong Song
55948189f6SJinghao Jia for (i = 0; i < nr_tests; i++) {
56f0c328f8SDaniel T. Lee objs[i] = bpf_object__open_file(filename, NULL);
57f0c328f8SDaniel T. Lee if (libbpf_get_error(objs[i])) {
58f0c328f8SDaniel T. Lee fprintf(stderr, "opening BPF object file failed\n");
59f0c328f8SDaniel T. Lee objs[i] = NULL;
60f0c328f8SDaniel T. Lee goto cleanup;
611da236b6SYonghong Song }
62f0c328f8SDaniel T. Lee
63*de4825a4SJinghao Jia /* One-time initialization */
64*de4825a4SJinghao Jia if (!links) {
65*de4825a4SJinghao Jia int nr_progs = 0;
66*de4825a4SJinghao Jia
67*de4825a4SJinghao Jia bpf_object__for_each_program(prog, objs[i])
68*de4825a4SJinghao Jia nr_progs += 1;
69*de4825a4SJinghao Jia
70*de4825a4SJinghao Jia links = calloc(nr_progs * nr_tests, sizeof(struct bpf_link *));
71*de4825a4SJinghao Jia
72*de4825a4SJinghao Jia if (!links)
73*de4825a4SJinghao Jia goto cleanup;
74*de4825a4SJinghao Jia }
75*de4825a4SJinghao Jia
76f0c328f8SDaniel T. Lee /* load BPF program */
77f0c328f8SDaniel T. Lee if (bpf_object__load(objs[i])) {
78f0c328f8SDaniel T. Lee fprintf(stderr, "loading BPF object file failed\n");
79f0c328f8SDaniel T. Lee goto cleanup;
80f0c328f8SDaniel T. Lee }
81f0c328f8SDaniel T. Lee
82f0c328f8SDaniel T. Lee map0_fds[i] = bpf_object__find_map_fd_by_name(objs[i],
83f0c328f8SDaniel T. Lee "enter_open_map");
84f0c328f8SDaniel T. Lee map1_fds[i] = bpf_object__find_map_fd_by_name(objs[i],
85f0c328f8SDaniel T. Lee "exit_open_map");
86f0c328f8SDaniel T. Lee if (map0_fds[i] < 0 || map1_fds[i] < 0) {
87f0c328f8SDaniel T. Lee fprintf(stderr, "finding a map in obj file failed\n");
88f0c328f8SDaniel T. Lee goto cleanup;
89f0c328f8SDaniel T. Lee }
90f0c328f8SDaniel T. Lee
91f0c328f8SDaniel T. Lee bpf_object__for_each_program(prog, objs[i]) {
92f0c328f8SDaniel T. Lee links[j] = bpf_program__attach(prog);
93f0c328f8SDaniel T. Lee if (libbpf_get_error(links[j])) {
94f0c328f8SDaniel T. Lee fprintf(stderr, "bpf_program__attach failed\n");
95f0c328f8SDaniel T. Lee links[j] = NULL;
96f0c328f8SDaniel T. Lee goto cleanup;
97f0c328f8SDaniel T. Lee }
98f0c328f8SDaniel T. Lee j++;
99f0c328f8SDaniel T. Lee }
100f0c328f8SDaniel T. Lee printf("prog #%d: map ids %d %d\n", i, map0_fds[i], map1_fds[i]);
101a678be5cSYonghong Song }
1021da236b6SYonghong Song
1031da236b6SYonghong Song /* current load_bpf_file has perf_event_open default pid = -1
1041da236b6SYonghong Song * and cpu = 0, which permits attached bpf execution on
1051da236b6SYonghong Song * all cpus for all pid's. bpf program execution ignores
1061da236b6SYonghong Song * cpu affinity.
1071da236b6SYonghong Song */
1081da236b6SYonghong Song /* trigger some "open" operations */
1091da236b6SYonghong Song fd = open(filename, O_RDONLY);
1101da236b6SYonghong Song if (fd < 0) {
1111da236b6SYonghong Song fprintf(stderr, "open failed: %s\n", strerror(errno));
1121da236b6SYonghong Song return 1;
1131da236b6SYonghong Song }
1141da236b6SYonghong Song close(fd);
1151da236b6SYonghong Song
1161da236b6SYonghong Song /* verify the map */
117948189f6SJinghao Jia for (i = 0; i < nr_tests; i++) {
118a678be5cSYonghong Song verify_map(map0_fds[i]);
119a678be5cSYonghong Song verify_map(map1_fds[i]);
120a678be5cSYonghong Song }
1211da236b6SYonghong Song
122f0c328f8SDaniel T. Lee cleanup:
123*de4825a4SJinghao Jia if (links) {
124f0c328f8SDaniel T. Lee for (j--; j >= 0; j--)
125f0c328f8SDaniel T. Lee bpf_link__destroy(links[j]);
126f0c328f8SDaniel T. Lee
127*de4825a4SJinghao Jia free(links);
128*de4825a4SJinghao Jia }
129*de4825a4SJinghao Jia
130f0c328f8SDaniel T. Lee for (i--; i >= 0; i--)
131f0c328f8SDaniel T. Lee bpf_object__close(objs[i]);
1321da236b6SYonghong Song return 0;
1331da236b6SYonghong Song }
134a678be5cSYonghong Song
main(int argc,char ** argv)135a678be5cSYonghong Song int main(int argc, char **argv)
136a678be5cSYonghong Song {
137948189f6SJinghao Jia int opt, nr_tests = 1;
138a678be5cSYonghong Song char filename[256];
139a678be5cSYonghong Song
140a678be5cSYonghong Song while ((opt = getopt(argc, argv, "i:h")) != -1) {
141a678be5cSYonghong Song switch (opt) {
142a678be5cSYonghong Song case 'i':
143948189f6SJinghao Jia nr_tests = atoi(optarg);
144a678be5cSYonghong Song break;
145a678be5cSYonghong Song case 'h':
146a678be5cSYonghong Song default:
147a678be5cSYonghong Song usage(argv[0]);
148a678be5cSYonghong Song return 0;
149a678be5cSYonghong Song }
150a678be5cSYonghong Song }
151a678be5cSYonghong Song
152a678be5cSYonghong Song snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
153a678be5cSYonghong Song
154948189f6SJinghao Jia return test(filename, nr_tests);
155a678be5cSYonghong Song }
156