1
2
3
4
5
6
7
8
9
10
11
12
13#ifndef __BPF_UTIL__
14#define __BPF_UTIL__
15
16#include <linux/bpf.h>
17#include <linux/btf.h>
18#include <linux/filter.h>
19#include <linux/magic.h>
20#include <linux/elf-em.h>
21#include <linux/if_alg.h>
22
23#include "utils.h"
24#include "bpf_scm.h"
25
26#define BPF_ENV_UDS "TC_BPF_UDS"
27#define BPF_ENV_MNT "TC_BPF_MNT"
28
29#ifndef BPF_MAX_LOG
30# define BPF_MAX_LOG 4096
31#endif
32
33#define BPF_DIR_GLOBALS "globals"
34
35#ifndef BPF_FS_MAGIC
36# define BPF_FS_MAGIC 0xcafe4a11
37#endif
38
39#define BPF_DIR_MNT "/sys/fs/bpf"
40
41#ifndef TRACEFS_MAGIC
42# define TRACEFS_MAGIC 0x74726163
43#endif
44
45#define TRACE_DIR_MNT "/sys/kernel/tracing"
46
47#ifndef AF_ALG
48# define AF_ALG 38
49#endif
50
51#ifndef EM_BPF
52# define EM_BPF 247
53#endif
54
55struct bpf_cfg_ops {
56 void (*cbpf_cb)(void *nl, const struct sock_filter *ops, int ops_len);
57 void (*ebpf_cb)(void *nl, int fd, const char *annotation);
58};
59
60enum bpf_mode {
61 CBPF_BYTECODE,
62 CBPF_FILE,
63 EBPF_OBJECT,
64 EBPF_PINNED,
65 BPF_MODE_MAX,
66};
67
68struct bpf_cfg_in {
69 const char *object;
70 const char *section;
71 const char *uds;
72 enum bpf_prog_type type;
73 enum bpf_mode mode;
74 __u32 ifindex;
75 bool verbose;
76 int argc;
77 char **argv;
78 struct sock_filter opcodes[BPF_MAXINSNS];
79 union {
80 int n_opcodes;
81 int prog_fd;
82 };
83};
84
85
86
87#define BPF_ALU64_REG(OP, DST, SRC) \
88 ((struct bpf_insn) { \
89 .code = BPF_ALU64 | BPF_OP(OP) | BPF_X, \
90 .dst_reg = DST, \
91 .src_reg = SRC, \
92 .off = 0, \
93 .imm = 0 })
94
95#define BPF_ALU32_REG(OP, DST, SRC) \
96 ((struct bpf_insn) { \
97 .code = BPF_ALU | BPF_OP(OP) | BPF_X, \
98 .dst_reg = DST, \
99 .src_reg = SRC, \
100 .off = 0, \
101 .imm = 0 })
102
103
104
105#define BPF_ALU64_IMM(OP, DST, IMM) \
106 ((struct bpf_insn) { \
107 .code = BPF_ALU64 | BPF_OP(OP) | BPF_K, \
108 .dst_reg = DST, \
109 .src_reg = 0, \
110 .off = 0, \
111 .imm = IMM })
112
113#define BPF_ALU32_IMM(OP, DST, IMM) \
114 ((struct bpf_insn) { \
115 .code = BPF_ALU | BPF_OP(OP) | BPF_K, \
116 .dst_reg = DST, \
117 .src_reg = 0, \
118 .off = 0, \
119 .imm = IMM })
120
121
122
123#define BPF_MOV64_REG(DST, SRC) \
124 ((struct bpf_insn) { \
125 .code = BPF_ALU64 | BPF_MOV | BPF_X, \
126 .dst_reg = DST, \
127 .src_reg = SRC, \
128 .off = 0, \
129 .imm = 0 })
130
131#define BPF_MOV32_REG(DST, SRC) \
132 ((struct bpf_insn) { \
133 .code = BPF_ALU | BPF_MOV | BPF_X, \
134 .dst_reg = DST, \
135 .src_reg = SRC, \
136 .off = 0, \
137 .imm = 0 })
138
139
140
141#define BPF_MOV64_IMM(DST, IMM) \
142 ((struct bpf_insn) { \
143 .code = BPF_ALU64 | BPF_MOV | BPF_K, \
144 .dst_reg = DST, \
145 .src_reg = 0, \
146 .off = 0, \
147 .imm = IMM })
148
149#define BPF_MOV32_IMM(DST, IMM) \
150 ((struct bpf_insn) { \
151 .code = BPF_ALU | BPF_MOV | BPF_K, \
152 .dst_reg = DST, \
153 .src_reg = 0, \
154 .off = 0, \
155 .imm = IMM })
156
157
158#define BPF_LD_IMM64(DST, IMM) \
159 BPF_LD_IMM64_RAW(DST, 0, IMM)
160
161#define BPF_LD_IMM64_RAW(DST, SRC, IMM) \
162 ((struct bpf_insn) { \
163 .code = BPF_LD | BPF_DW | BPF_IMM, \
164 .dst_reg = DST, \
165 .src_reg = SRC, \
166 .off = 0, \
167 .imm = (__u32) (IMM) }), \
168 ((struct bpf_insn) { \
169 .code = 0, \
170 .dst_reg = 0, \
171 .src_reg = 0, \
172 .off = 0, \
173 .imm = ((__u64) (IMM)) >> 32 })
174
175#ifndef BPF_PSEUDO_MAP_FD
176# define BPF_PSEUDO_MAP_FD 1
177#endif
178
179
180#define BPF_LD_MAP_FD(DST, MAP_FD) \
181 BPF_LD_IMM64_RAW(DST, BPF_PSEUDO_MAP_FD, MAP_FD)
182
183
184
185
186#define BPF_LD_ABS(SIZE, IMM) \
187 ((struct bpf_insn) { \
188 .code = BPF_LD | BPF_SIZE(SIZE) | BPF_ABS, \
189 .dst_reg = 0, \
190 .src_reg = 0, \
191 .off = 0, \
192 .imm = IMM })
193
194
195
196#define BPF_LDX_MEM(SIZE, DST, SRC, OFF) \
197 ((struct bpf_insn) { \
198 .code = BPF_LDX | BPF_SIZE(SIZE) | BPF_MEM, \
199 .dst_reg = DST, \
200 .src_reg = SRC, \
201 .off = OFF, \
202 .imm = 0 })
203
204
205
206#define BPF_STX_MEM(SIZE, DST, SRC, OFF) \
207 ((struct bpf_insn) { \
208 .code = BPF_STX | BPF_SIZE(SIZE) | BPF_MEM, \
209 .dst_reg = DST, \
210 .src_reg = SRC, \
211 .off = OFF, \
212 .imm = 0 })
213
214
215
216#define BPF_ST_MEM(SIZE, DST, OFF, IMM) \
217 ((struct bpf_insn) { \
218 .code = BPF_ST | BPF_SIZE(SIZE) | BPF_MEM, \
219 .dst_reg = DST, \
220 .src_reg = 0, \
221 .off = OFF, \
222 .imm = IMM })
223
224
225
226#define BPF_JMP_REG(OP, DST, SRC, OFF) \
227 ((struct bpf_insn) { \
228 .code = BPF_JMP | BPF_OP(OP) | BPF_X, \
229 .dst_reg = DST, \
230 .src_reg = SRC, \
231 .off = OFF, \
232 .imm = 0 })
233
234
235
236#define BPF_JMP_IMM(OP, DST, IMM, OFF) \
237 ((struct bpf_insn) { \
238 .code = BPF_JMP | BPF_OP(OP) | BPF_K, \
239 .dst_reg = DST, \
240 .src_reg = 0, \
241 .off = OFF, \
242 .imm = IMM })
243
244
245
246#define BPF_RAW_INSN(CODE, DST, SRC, OFF, IMM) \
247 ((struct bpf_insn) { \
248 .code = CODE, \
249 .dst_reg = DST, \
250 .src_reg = SRC, \
251 .off = OFF, \
252 .imm = IMM })
253
254
255
256#define BPF_EXIT_INSN() \
257 ((struct bpf_insn) { \
258 .code = BPF_JMP | BPF_EXIT, \
259 .dst_reg = 0, \
260 .src_reg = 0, \
261 .off = 0, \
262 .imm = 0 })
263
264int bpf_parse_common(struct bpf_cfg_in *cfg, const struct bpf_cfg_ops *ops);
265int bpf_load_common(struct bpf_cfg_in *cfg, const struct bpf_cfg_ops *ops,
266 void *nl);
267int bpf_parse_and_load_common(struct bpf_cfg_in *cfg,
268 const struct bpf_cfg_ops *ops, void *nl);
269
270const char *bpf_prog_to_default_section(enum bpf_prog_type type);
271
272int bpf_graft_map(const char *map_path, uint32_t *key, int argc, char **argv);
273int bpf_trace_pipe(void);
274
275void bpf_print_ops(struct rtattr *bpf_ops, __u16 len);
276
277int bpf_prog_load_dev(enum bpf_prog_type type, const struct bpf_insn *insns,
278 size_t size_insns, const char *license, __u32 ifindex,
279 char *log, size_t size_log);
280int bpf_program_load(enum bpf_prog_type type, const struct bpf_insn *insns,
281 size_t size_insns, const char *license, char *log,
282 size_t size_log);
283
284int bpf_prog_attach_fd(int prog_fd, int target_fd, enum bpf_attach_type type);
285int bpf_prog_detach_fd(int target_fd, enum bpf_attach_type type);
286int bpf_program_attach(int prog_fd, int target_fd, enum bpf_attach_type type);
287
288int bpf_dump_prog_info(FILE *f, uint32_t id);
289
290#ifdef HAVE_ELF
291int bpf_send_map_fds(const char *path, const char *obj);
292int bpf_recv_map_fds(const char *path, int *fds, struct bpf_map_aux *aux,
293 unsigned int entries);
294#ifdef HAVE_LIBBPF
295int iproute2_bpf_elf_ctx_init(struct bpf_cfg_in *cfg);
296int iproute2_bpf_fetch_ancillary(void);
297int iproute2_get_root_path(char *root_path, size_t len);
298bool iproute2_is_pin_map(const char *libbpf_map_name, char *pathname);
299bool iproute2_is_map_in_map(const char *libbpf_map_name, struct bpf_elf_map *imap,
300 struct bpf_elf_map *omap, char *omap_name);
301int iproute2_find_map_name_by_id(unsigned int map_id, char *name);
302int iproute2_load_libbpf(struct bpf_cfg_in *cfg);
303#endif
304#else
305static inline int bpf_send_map_fds(const char *path, const char *obj)
306{
307 return 0;
308}
309
310static inline int bpf_recv_map_fds(const char *path, int *fds,
311 struct bpf_map_aux *aux,
312 unsigned int entries)
313{
314 return -1;
315}
316#ifdef HAVE_LIBBPF
317static inline int iproute2_load_libbpf(struct bpf_cfg_in *cfg)
318{
319 fprintf(stderr, "No ELF library support compiled in.\n");
320 return -1;
321}
322#endif
323#endif
324
325const char *get_libbpf_version(void);
326
327#endif
328