LXR linux/tools/testing/selftests/bpf/test_libbpf

   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}
 145