tools/testing/selftests/bpf/flow_dissector_load.h

   1 /* SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause) */
   2 #ifndef FLOW_DISSECTOR_LOAD
   3 #define FLOW_DISSECTOR_LOAD
   4
   5 #include <bpf/bpf.h>
   6 #include <bpf/libbpf.h>
   7 #include "testing_helpers.h"
   8
   9 static inline int bpf_flow_load(struct bpf_object **obj,
  10                                 const char *path,
  11                                 const char *prog_name,
  12                                 const char *map_name,
  13                                 const char *keys_map_name,
  14                                 int *prog_fd,
  15                                 int *keys_fd)
  16 {
  17         struct bpf_program *prog, *main_prog;
  18         struct bpf_map *prog_array, *keys;
  19         int prog_array_fd;
  20         int ret, fd, i;
  21
  22         ret = bpf_prog_test_load(path, BPF_PROG_TYPE_FLOW_DISSECTOR, obj,
  23                             prog_fd);
  24         if (ret)
  25                 return ret;
  26
  27         main_prog = bpf_object__find_program_by_name(*obj, prog_name);
  28         if (!main_prog)
  29                 return -1;
  30
  31         *prog_fd = bpf_program__fd(main_prog);
  32         if (*prog_fd < 0)
  33                 return -1;
  34
  35         prog_array = bpf_object__find_map_by_name(*obj, map_name);
  36         if (!prog_array)
  37                 return -1;
  38
  39         prog_array_fd = bpf_map__fd(prog_array);
  40         if (prog_array_fd < 0)
  41                 return -1;
  42
  43         if (keys_map_name && keys_fd) {
  44                 keys = bpf_object__find_map_by_name(*obj, keys_map_name);
  45                 if (!keys)
  46                         return -1;
  47
  48                 *keys_fd = bpf_map__fd(keys);
  49                 if (*keys_fd < 0)
  50                         return -1;
  51         }
  52
  53         i = 0;
  54         bpf_object__for_each_program(prog, *obj) {
  55                 fd = bpf_program__fd(prog);
  56                 if (fd < 0)
  57                         return fd;
  58
  59                 if (fd != *prog_fd) {
  60                         bpf_map_update_elem(prog_array_fd, &i, &fd, BPF_ANY);
  61                         ++i;
  62                 }
  63         }
  64
  65         return 0;
  66 }
  67
  68 #endif /* FLOW_DISSECTOR_LOAD */