tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
1/* SPDX-License-Identifier: GPL-2.0
2 * Copyright (c) 2018 Jesper Dangaard Brouer, Red Hat Inc.
3 */
4static const char *__doc__ =
5 "Libbpf test program for loading BPF ELF object files";
6
7#include <stdlib.h>
8#include <stdio.h>
9#include <string.h>
10#include <stdarg.h>
11#include <bpf/libbpf.h>
12#include <getopt.h>
13
14#include "bpf_rlimit.h"
15
16static const struct option long_options[] = {
17 {"help", no_argument, NULL, 'h' },
18 {"debug", no_argument, NULL, 'D' },
19 {"quiet", no_argument, NULL, 'q' },
20 {0, 0, NULL, 0 }
21};
22
23static void usage(char *argv[])
24{
25 int i;
26
27 printf("\nDOCUMENTATION:\n%s\n\n", __doc__);
28 printf(" Usage: %s (options-see-below) BPF_FILE\n", argv[0]);
29 printf(" Listing options:\n");
30 for (i = 0; long_options[i].name != 0; i++) {
31 printf(" --%-12s", long_options[i].name);
32 printf(" short-option: -%c",
33 long_options[i].val);
34 printf("\n");
35 }
36 printf("\n");
37}
38
39static bool debug = 0;
40static int libbpf_debug_print(enum libbpf_print_level level,
41 const char *fmt, va_list args)
42{
43 if (level == LIBBPF_DEBUG && !debug)
44 return 0;
45
46 fprintf(stderr, "[%d] ", level);
47 return vfprintf(stderr, fmt, args);
48}
49
50#define EXIT_FAIL_LIBBPF EXIT_FAILURE
51#define EXIT_FAIL_OPTION 2
52
53int test_walk_progs(struct bpf_object *obj, bool verbose)
54{
55 struct bpf_program *prog;
56 int cnt = 0;
57
58 bpf_object__for_each_program(prog, obj) {
59 cnt++;
60 if (verbose)
61 printf("Prog (count:%d) section_name: %s\n", cnt,
62 bpf_program__title(prog, false));
63 }
64 return 0;
65}
66
67int test_walk_maps(struct bpf_object *obj, bool verbose)
68{
69 struct bpf_map *map;
70 int cnt = 0;
71
72 bpf_object__for_each_map(map, obj) {
73 cnt++;
74 if (verbose)
75 printf("Map (count:%d) name: %s\n", cnt,
76 bpf_map__name(map));
77 }
78 return 0;
79}
80
81int test_open_file(char *filename, bool verbose)
82{
83 struct bpf_object *bpfobj = NULL;
84 long err;
85
86 if (verbose)
87 printf("Open BPF ELF-file with libbpf: %s\n", filename);
88
89 /* Load BPF ELF object file and check for errors */
90 bpfobj = bpf_object__open(filename);
91 err = libbpf_get_error(bpfobj);
92 if (err) {
93 char err_buf[128];
94 libbpf_strerror(err, err_buf, sizeof(err_buf));
95 if (verbose)
96 printf("Unable to load eBPF objects in file '%s': %s\n",
97 filename, err_buf);
98 return EXIT_FAIL_LIBBPF;
99 }
100 test_walk_progs(bpfobj, verbose);
101 test_walk_maps(bpfobj, verbose);
102
103 if (verbose)
104 printf("Close BPF ELF-file with libbpf: %s\n",
105 bpf_object__name(bpfobj));
106 bpf_object__close(bpfobj);
107
108 return 0;
109}
110
111int main(int argc, char **argv)
112{
113 char filename[1024] = { 0 };
114 bool verbose = 1;
115 int longindex = 0;
116 int opt;
117
118 libbpf_set_print(libbpf_debug_print);
119
120 /* Parse commands line args */
121 while ((opt = getopt_long(argc, argv, "hDq",
122 long_options, &longindex)) != -1) {
123 switch (opt) {
124 case 'D':
125 debug = 1;
126 break;
127 case 'q': /* Use in scripting mode */
128 verbose = 0;
129 break;
130 case 'h':
131 default:
132 usage(argv);
133 return EXIT_FAIL_OPTION;
134 }
135 }
136 if (optind >= argc) {
137 usage(argv);
138 printf("ERROR: Expected BPF_FILE argument after options\n");
139 return EXIT_FAIL_OPTION;
140 }
141 snprintf(filename, sizeof(filename), "%s", argv[optind]);
142
143 return test_open_file(filename, verbose);
144}