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 *prog_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 = bpf_object__find_program_by_name(*obj, prog_name);
27 	if (!main_prog)
28 		return -1;
29 
30 	*prog_fd = bpf_program__fd(main_prog);
31 	if (*prog_fd < 0)
32 		return -1;
33 
34 	prog_array = bpf_object__find_map_by_name(*obj, map_name);
35 	if (!prog_array)
36 		return -1;
37 
38 	prog_array_fd = bpf_map__fd(prog_array);
39 	if (prog_array_fd < 0)
40 		return -1;
41 
42 	if (keys_map_name && keys_fd) {
43 		keys = bpf_object__find_map_by_name(*obj, keys_map_name);
44 		if (!keys)
45 			return -1;
46 
47 		*keys_fd = bpf_map__fd(keys);
48 		if (*keys_fd < 0)
49 			return -1;
50 	}
51 
52 	i = 0;
53 	bpf_object__for_each_program(prog, *obj) {
54 		fd = bpf_program__fd(prog);
55 		if (fd < 0)
56 			return fd;
57 
58 		if (fd != *prog_fd) {
59 			bpf_map_update_elem(prog_array_fd, &i, &fd, BPF_ANY);
60 			++i;
61 		}
62 	}
63 
64 	return 0;
65 }
66 
67 #endif /* FLOW_DISSECTOR_LOAD */
68