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 */ 69