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
 9static 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 */

 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
 8static 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 */