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
   8 static inline int bpf_flow_load(struct bpf_object **obj,
   9                                 const char *path,
  10                                 const char *section_name,
  11                                 const char *map_name,
  12                                 const char *keys_map_name,
  13                                 int *prog_fd,
  14                                 int *keys_fd)
  15 {
  16         struct bpf_program *prog, *main_prog;
  17         struct bpf_map *prog_array, *keys;
  18         int prog_array_fd;
  19         int ret, fd, i;
  20
  21         ret = bpf_prog_load(path, BPF_PROG_TYPE_FLOW_DISSECTOR, obj,
  22                             prog_fd);
  23         if (ret)
  24                 return ret;
  25
  26         main_prog = NULL;
  27         bpf_object__for_each_program(prog, *obj) {
  28                 if (strcmp(section_name, bpf_program__section_name(prog)) == 0) {
  29                         main_prog = prog;
  30                         break;
  31                 }
  32         }
  33         if (!main_prog)
  34                 return -1;
  35
  36         *prog_fd = bpf_program__fd(main_prog);
  37         if (*prog_fd < 0)
  38                 return -1;
  39
  40         prog_array = bpf_object__find_map_by_name(*obj, map_name);
  41         if (!prog_array)
  42                 return -1;
  43
  44         prog_array_fd = bpf_map__fd(prog_array);
  45         if (prog_array_fd < 0)
  46                 return -1;
  47
  48         if (keys_map_name && keys_fd) {
  49                 keys = bpf_object__find_map_by_name(*obj, keys_map_name);
  50                 if (!keys)
  51                         return -1;
  52
  53                 *keys_fd = bpf_map__fd(keys);
  54                 if (*keys_fd < 0)
  55                         return -1;
  56         }
  57
  58         i = 0;
  59         bpf_object__for_each_program(prog, *obj) {
  60                 fd = bpf_program__fd(prog);
  61                 if (fd < 0)
  62                         return fd;
  63
  64                 if (fd != *prog_fd) {
  65                         bpf_map_update_elem(prog_array_fd, &i, &fd, BPF_ANY);
  66                         ++i;
  67                 }
  68         }
  69
  70         return 0;
  71 }
  72
  73 #endif /* FLOW_DISSECTOR_LOAD */