1 /* SPDX-License-Identifier: GPL-2.0 */
2 /* Copyright (c) 2018 Facebook */
7 #include <linux/types.h>
8 #include <linux/bpfptr.h>
9 #include <linux/bsearch.h>
10 #include <linux/btf_ids.h>
11 #include <uapi/linux/btf.h>
12 #include <uapi/linux/bpf.h>
14 #define BTF_TYPE_EMIT(type) ((void)(type *)0)
15 #define BTF_TYPE_EMIT_ENUM(enum_val) ((void)enum_val)
17 /* These need to be macros, as the expressions are used in assembler input */
18 #define KF_ACQUIRE (1 << 0) /* kfunc is an acquire function */
19 #define KF_RELEASE (1 << 1) /* kfunc is a release function */
20 #define KF_RET_NULL (1 << 2) /* kfunc returns a pointer that may be NULL */
21 /* Trusted arguments are those which are guaranteed to be valid when passed to
22 * the kfunc. It is used to enforce that pointers obtained from either acquire
23 * kfuncs, or from the main kernel on a tracepoint or struct_ops callback
24 * invocation, remain unmodified when being passed to helpers taking trusted
27 * Consider, for example, the following new task tracepoint:
29 * SEC("tp_btf/task_newtask")
30 * int BPF_PROG(new_task_tp, struct task_struct *task, u64 clone_flags)
35 * And the following kfunc:
37 * BTF_ID_FLAGS(func, bpf_task_acquire, KF_ACQUIRE | KF_TRUSTED_ARGS)
39 * All invocations to the kfunc must pass the unmodified, unwalked task:
41 * bpf_task_acquire(task); // Allowed
42 * bpf_task_acquire(task->last_wakee); // Rejected, walked task
44 * Programs may also pass referenced tasks directly to the kfunc:
46 * struct task_struct *acquired;
48 * acquired = bpf_task_acquire(task); // Allowed, same as above
49 * bpf_task_acquire(acquired); // Allowed
50 * bpf_task_acquire(task); // Allowed
51 * bpf_task_acquire(acquired->last_wakee); // Rejected, walked task
53 * Programs may _not_, however, pass a task from an arbitrary fentry/fexit, or
54 * kprobe/kretprobe to the kfunc, as BPF cannot guarantee that all of these
55 * pointers are guaranteed to be safe. For example, the following BPF program
58 * SEC("kretprobe/free_task")
59 * int BPF_PROG(free_task_probe, struct task_struct *tsk)
61 * struct task_struct *acquired;
63 * acquired = bpf_task_acquire(acquired); // Rejected, not a trusted pointer
64 * bpf_task_release(acquired);
69 #define KF_TRUSTED_ARGS (1 << 4) /* kfunc only takes trusted pointer arguments */
70 #define KF_SLEEPABLE (1 << 5) /* kfunc may sleep */
71 #define KF_DESTRUCTIVE (1 << 6) /* kfunc performs destructive actions */
72 #define KF_RCU (1 << 7) /* kfunc takes either rcu or trusted pointer arguments */
73 /* only one of KF_ITER_{NEW,NEXT,DESTROY} could be specified per kfunc */
74 #define KF_ITER_NEW (1 << 8) /* kfunc implements BPF iter constructor */
75 #define KF_ITER_NEXT (1 << 9) /* kfunc implements BPF iter next method */
76 #define KF_ITER_DESTROY (1 << 10) /* kfunc implements BPF iter destructor */
77 #define KF_RCU_PROTECTED (1 << 11) /* kfunc should be protected by rcu cs when they are invoked */
80 * Tag marking a kernel function as a kfunc. This is meant to minimize the
81 * amount of copy-paste that kfunc authors have to include for correctness so
82 * as to avoid issues such as the compiler inlining or eliding either a static
83 * kfunc, or a global kfunc in an LTO build.
85 #define __bpf_kfunc __used __retain noinline
87 #define __bpf_kfunc_start_defs() \
89 __diag_ignore_all("-Wmissing-declarations", \
90 "Global kfuncs as their definitions will be in BTF");\
91 __diag_ignore_all("-Wmissing-prototypes", \
92 "Global kfuncs as their definitions will be in BTF")
94 #define __bpf_kfunc_end_defs() __diag_pop()
95 #define __bpf_hook_start() __bpf_kfunc_start_defs()
96 #define __bpf_hook_end() __bpf_kfunc_end_defs()
99 * Return the name of the passed struct, if exists, or halt the build if for
100 * example the structure gets renamed. In this way, developers have to revisit
101 * the code using that structure name, and update it accordingly.
103 #define stringify_struct(x) \
104 ({ BUILD_BUG_ON(sizeof(struct x) < 0); \
115 typedef int (*btf_kfunc_filter_t)(const struct bpf_prog *prog, u32 kfunc_id);
117 struct btf_kfunc_id_set {
118 struct module *owner;
119 struct btf_id_set8 *set;
120 btf_kfunc_filter_t filter;
123 struct btf_id_dtor_kfunc {
128 struct btf_struct_meta {
130 struct btf_record *record;
133 struct btf_struct_metas {
135 struct btf_struct_meta types[];
138 extern const struct file_operations btf_fops;
140 const char *btf_get_name(const struct btf *btf);
141 void btf_get(struct btf *btf);
142 void btf_put(struct btf *btf);
143 const struct btf_header *btf_header(const struct btf *btf);
144 int btf_new_fd(const union bpf_attr *attr, bpfptr_t uattr, u32 uattr_sz);
145 struct btf *btf_get_by_fd(int fd);
146 int btf_get_info_by_fd(const struct btf *btf,
147 const union bpf_attr *attr,
148 union bpf_attr __user *uattr);
149 /* Figure out the size of a type_id. If type_id is a modifier
150 * (e.g. const), it will be resolved to find out the type with size.
153 * In describing "const void *", type_id is "const" and "const"
154 * refers to "void *". The return type will be "void *".
156 * If type_id is a simple "int", then return type will be "int".
158 * @btf: struct btf object
159 * @type_id: Find out the size of type_id. The type_id of the return
160 * type is set to *type_id.
161 * @ret_size: It can be NULL. If not NULL, the size of the return
162 * type is set to *ret_size.
163 * Return: The btf_type (resolved to another type with size info if needed).
164 * NULL is returned if type_id itself does not have size info
165 * (e.g. void) or it cannot be resolved to another type that
167 * *type_id and *ret_size will not be changed in the
170 const struct btf_type *btf_type_id_size(const struct btf *btf,
175 * Options to control show behaviour.
176 * - BTF_SHOW_COMPACT: no formatting around type information
177 * - BTF_SHOW_NONAME: no struct/union member names/types
178 * - BTF_SHOW_PTR_RAW: show raw (unobfuscated) pointer values;
180 * - BTF_SHOW_ZERO: show zero-valued struct/union members; they
181 * are not displayed by default
182 * - BTF_SHOW_UNSAFE: skip use of bpf_probe_read() to safely read
183 * data before displaying it.
185 #define BTF_SHOW_COMPACT BTF_F_COMPACT
186 #define BTF_SHOW_NONAME BTF_F_NONAME
187 #define BTF_SHOW_PTR_RAW BTF_F_PTR_RAW
188 #define BTF_SHOW_ZERO BTF_F_ZERO
189 #define BTF_SHOW_UNSAFE (1ULL << 4)
191 void btf_type_seq_show(const struct btf *btf, u32 type_id, void *obj,
193 int btf_type_seq_show_flags(const struct btf *btf, u32 type_id, void *obj,
194 struct seq_file *m, u64 flags);
197 * Copy len bytes of string representation of obj of BTF type_id into buf.
199 * @btf: struct btf object
200 * @type_id: type id of type obj points to
201 * @obj: pointer to typed data
202 * @buf: buffer to write to
203 * @len: maximum length to write to buf
204 * @flags: show options (see above)
206 * Return: length that would have been/was copied as per snprintf, or
209 int btf_type_snprintf_show(const struct btf *btf, u32 type_id, void *obj,
210 char *buf, int len, u64 flags);
212 int btf_get_fd_by_id(u32 id);
213 u32 btf_obj_id(const struct btf *btf);
214 bool btf_is_kernel(const struct btf *btf);
215 bool btf_is_module(const struct btf *btf);
216 bool btf_is_vmlinux(const struct btf *btf);
217 struct module *btf_try_get_module(const struct btf *btf);
218 u32 btf_nr_types(const struct btf *btf);
219 struct btf *btf_base_btf(const struct btf *btf);
220 bool btf_member_is_reg_int(const struct btf *btf, const struct btf_type *s,
221 const struct btf_member *m,
222 u32 expected_offset, u32 expected_size);
223 struct btf_record *btf_parse_fields(const struct btf *btf, const struct btf_type *t,
224 u32 field_mask, u32 value_size);
225 int btf_check_and_fixup_fields(const struct btf *btf, struct btf_record *rec);
226 bool btf_type_is_void(const struct btf_type *t);
227 s32 btf_find_by_name_kind(const struct btf *btf, const char *name, u8 kind);
228 s32 bpf_find_btf_id(const char *name, u32 kind, struct btf **btf_p);
229 const struct btf_type *btf_type_skip_modifiers(const struct btf *btf,
230 u32 id, u32 *res_id);
231 const struct btf_type *btf_type_resolve_ptr(const struct btf *btf,
232 u32 id, u32 *res_id);
233 const struct btf_type *btf_type_resolve_func_ptr(const struct btf *btf,
234 u32 id, u32 *res_id);
235 const struct btf_type *
236 btf_resolve_size(const struct btf *btf, const struct btf_type *type,
238 const char *btf_type_str(const struct btf_type *t);
240 #define for_each_member(i, struct_type, member) \
241 for (i = 0, member = btf_type_member(struct_type); \
242 i < btf_type_vlen(struct_type); \
245 #define for_each_vsi(i, datasec_type, member) \
246 for (i = 0, member = btf_type_var_secinfo(datasec_type); \
247 i < btf_type_vlen(datasec_type); \
250 static inline bool btf_type_is_ptr(const struct btf_type *t)
252 return BTF_INFO_KIND(t->info) == BTF_KIND_PTR;
255 static inline bool btf_type_is_int(const struct btf_type *t)
257 return BTF_INFO_KIND(t->info) == BTF_KIND_INT;
260 static inline bool btf_type_is_small_int(const struct btf_type *t)
262 return btf_type_is_int(t) && t->size <= sizeof(u64);
265 static inline u8 btf_int_encoding(const struct btf_type *t)
267 return BTF_INT_ENCODING(*(u32 *)(t + 1));
270 static inline bool btf_type_is_signed_int(const struct btf_type *t)
272 return btf_type_is_int(t) && (btf_int_encoding(t) & BTF_INT_SIGNED);
275 static inline bool btf_type_is_enum(const struct btf_type *t)
277 return BTF_INFO_KIND(t->info) == BTF_KIND_ENUM;
280 static inline bool btf_is_any_enum(const struct btf_type *t)
282 return BTF_INFO_KIND(t->info) == BTF_KIND_ENUM ||
283 BTF_INFO_KIND(t->info) == BTF_KIND_ENUM64;
286 static inline bool btf_kind_core_compat(const struct btf_type *t1,
287 const struct btf_type *t2)
289 return BTF_INFO_KIND(t1->info) == BTF_INFO_KIND(t2->info) ||
290 (btf_is_any_enum(t1) && btf_is_any_enum(t2));
293 static inline bool str_is_empty(const char *s)
298 static inline u16 btf_kind(const struct btf_type *t)
300 return BTF_INFO_KIND(t->info);
303 static inline bool btf_is_enum(const struct btf_type *t)
305 return btf_kind(t) == BTF_KIND_ENUM;
308 static inline bool btf_is_enum64(const struct btf_type *t)
310 return btf_kind(t) == BTF_KIND_ENUM64;
313 static inline u64 btf_enum64_value(const struct btf_enum64 *e)
315 return ((u64)e->val_hi32 << 32) | e->val_lo32;
318 static inline bool btf_is_composite(const struct btf_type *t)
320 u16 kind = btf_kind(t);
322 return kind == BTF_KIND_STRUCT || kind == BTF_KIND_UNION;
325 static inline bool btf_is_array(const struct btf_type *t)
327 return btf_kind(t) == BTF_KIND_ARRAY;
330 static inline bool btf_is_int(const struct btf_type *t)
332 return btf_kind(t) == BTF_KIND_INT;
335 static inline bool btf_is_ptr(const struct btf_type *t)
337 return btf_kind(t) == BTF_KIND_PTR;
340 static inline u8 btf_int_offset(const struct btf_type *t)
342 return BTF_INT_OFFSET(*(u32 *)(t + 1));
345 static inline __u8 btf_int_bits(const struct btf_type *t)
347 return BTF_INT_BITS(*(__u32 *)(t + 1));
350 static inline bool btf_type_is_scalar(const struct btf_type *t)
352 return btf_type_is_int(t) || btf_type_is_enum(t);
355 static inline bool btf_type_is_typedef(const struct btf_type *t)
357 return BTF_INFO_KIND(t->info) == BTF_KIND_TYPEDEF;
360 static inline bool btf_type_is_volatile(const struct btf_type *t)
362 return BTF_INFO_KIND(t->info) == BTF_KIND_VOLATILE;
365 static inline bool btf_type_is_func(const struct btf_type *t)
367 return BTF_INFO_KIND(t->info) == BTF_KIND_FUNC;
370 static inline bool btf_type_is_func_proto(const struct btf_type *t)
372 return BTF_INFO_KIND(t->info) == BTF_KIND_FUNC_PROTO;
375 static inline bool btf_type_is_var(const struct btf_type *t)
377 return BTF_INFO_KIND(t->info) == BTF_KIND_VAR;
380 static inline bool btf_type_is_type_tag(const struct btf_type *t)
382 return BTF_INFO_KIND(t->info) == BTF_KIND_TYPE_TAG;
385 /* union is only a special case of struct:
386 * all its offsetof(member) == 0
388 static inline bool btf_type_is_struct(const struct btf_type *t)
390 u8 kind = BTF_INFO_KIND(t->info);
392 return kind == BTF_KIND_STRUCT || kind == BTF_KIND_UNION;
395 static inline bool __btf_type_is_struct(const struct btf_type *t)
397 return BTF_INFO_KIND(t->info) == BTF_KIND_STRUCT;
400 static inline bool btf_type_is_array(const struct btf_type *t)
402 return BTF_INFO_KIND(t->info) == BTF_KIND_ARRAY;
405 static inline u16 btf_type_vlen(const struct btf_type *t)
407 return BTF_INFO_VLEN(t->info);
410 static inline u16 btf_vlen(const struct btf_type *t)
412 return btf_type_vlen(t);
415 static inline u16 btf_func_linkage(const struct btf_type *t)
417 return BTF_INFO_VLEN(t->info);
420 static inline bool btf_type_kflag(const struct btf_type *t)
422 return BTF_INFO_KFLAG(t->info);
425 static inline u32 __btf_member_bit_offset(const struct btf_type *struct_type,
426 const struct btf_member *member)
428 return btf_type_kflag(struct_type) ? BTF_MEMBER_BIT_OFFSET(member->offset)
432 static inline u32 __btf_member_bitfield_size(const struct btf_type *struct_type,
433 const struct btf_member *member)
435 return btf_type_kflag(struct_type) ? BTF_MEMBER_BITFIELD_SIZE(member->offset)
439 static inline struct btf_member *btf_members(const struct btf_type *t)
441 return (struct btf_member *)(t + 1);
444 static inline u32 btf_member_bit_offset(const struct btf_type *t, u32 member_idx)
446 const struct btf_member *m = btf_members(t) + member_idx;
448 return __btf_member_bit_offset(t, m);
451 static inline u32 btf_member_bitfield_size(const struct btf_type *t, u32 member_idx)
453 const struct btf_member *m = btf_members(t) + member_idx;
455 return __btf_member_bitfield_size(t, m);
458 static inline const struct btf_member *btf_type_member(const struct btf_type *t)
460 return (const struct btf_member *)(t + 1);
463 static inline struct btf_array *btf_array(const struct btf_type *t)
465 return (struct btf_array *)(t + 1);
468 static inline struct btf_enum *btf_enum(const struct btf_type *t)
470 return (struct btf_enum *)(t + 1);
473 static inline struct btf_enum64 *btf_enum64(const struct btf_type *t)
475 return (struct btf_enum64 *)(t + 1);
478 static inline const struct btf_var_secinfo *btf_type_var_secinfo(
479 const struct btf_type *t)
481 return (const struct btf_var_secinfo *)(t + 1);
484 static inline struct btf_param *btf_params(const struct btf_type *t)
486 return (struct btf_param *)(t + 1);
489 static inline struct btf_decl_tag *btf_decl_tag(const struct btf_type *t)
491 return (struct btf_decl_tag *)(t + 1);
494 static inline int btf_id_cmp_func(const void *a, const void *b)
496 const int *pa = a, *pb = b;
501 static inline bool btf_id_set_contains(const struct btf_id_set *set, u32 id)
503 return bsearch(&id, set->ids, set->cnt, sizeof(u32), btf_id_cmp_func) != NULL;
506 static inline void *btf_id_set8_contains(const struct btf_id_set8 *set, u32 id)
508 return bsearch(&id, set->pairs, set->cnt, sizeof(set->pairs[0]), btf_id_cmp_func);
511 bool btf_param_match_suffix(const struct btf *btf,
512 const struct btf_param *arg,
514 int btf_ctx_arg_offset(const struct btf *btf, const struct btf_type *func_proto,
517 struct bpf_verifier_log;
519 #if defined(CONFIG_BPF_JIT) && defined(CONFIG_BPF_SYSCALL)
520 struct bpf_struct_ops;
521 int __register_bpf_struct_ops(struct bpf_struct_ops *st_ops);
522 const struct bpf_struct_ops_desc *bpf_struct_ops_find_value(struct btf *btf, u32 value_id);
523 const struct bpf_struct_ops_desc *bpf_struct_ops_find(struct btf *btf, u32 type_id);
525 static inline const struct bpf_struct_ops_desc *bpf_struct_ops_find(struct btf *btf, u32 type_id)
531 enum btf_field_iter_kind {
536 struct btf_field_desc {
537 /* once-per-type offsets */
538 int t_off_cnt, t_offs[2];
539 /* member struct size, or zero, if no members */
541 /* repeated per-member offsets */
542 int m_off_cnt, m_offs[1];
545 struct btf_field_iter {
546 struct btf_field_desc desc;
553 #ifdef CONFIG_BPF_SYSCALL
554 const struct btf_type *btf_type_by_id(const struct btf *btf, u32 type_id);
555 void btf_set_base_btf(struct btf *btf, const struct btf *base_btf);
556 int btf_relocate(struct btf *btf, const struct btf *base_btf, __u32 **map_ids);
557 int btf_field_iter_init(struct btf_field_iter *it, struct btf_type *t,
558 enum btf_field_iter_kind iter_kind);
559 __u32 *btf_field_iter_next(struct btf_field_iter *it);
561 const char *btf_name_by_offset(const struct btf *btf, u32 offset);
562 const char *btf_str_by_offset(const struct btf *btf, u32 offset);
563 struct btf *btf_parse_vmlinux(void);
564 struct btf *bpf_prog_get_target_btf(const struct bpf_prog *prog);
565 u32 *btf_kfunc_id_set_contains(const struct btf *btf, u32 kfunc_btf_id,
566 const struct bpf_prog *prog);
567 u32 *btf_kfunc_is_modify_return(const struct btf *btf, u32 kfunc_btf_id,
568 const struct bpf_prog *prog);
569 int register_btf_kfunc_id_set(enum bpf_prog_type prog_type,
570 const struct btf_kfunc_id_set *s);
571 int register_btf_fmodret_id_set(const struct btf_kfunc_id_set *kset);
572 s32 btf_find_dtor_kfunc(struct btf *btf, u32 btf_id);
573 int register_btf_id_dtor_kfuncs(const struct btf_id_dtor_kfunc *dtors, u32 add_cnt,
574 struct module *owner);
575 struct btf_struct_meta *btf_find_struct_meta(const struct btf *btf, u32 btf_id);
576 bool btf_is_projection_of(const char *pname, const char *tname);
577 bool btf_is_prog_ctx_type(struct bpf_verifier_log *log, const struct btf *btf,
578 const struct btf_type *t, enum bpf_prog_type prog_type,
580 int get_kern_ctx_btf_id(struct bpf_verifier_log *log, enum bpf_prog_type prog_type);
581 bool btf_types_are_same(const struct btf *btf1, u32 id1,
582 const struct btf *btf2, u32 id2);
584 static inline const struct btf_type *btf_type_by_id(const struct btf *btf,
590 static inline void btf_set_base_btf(struct btf *btf, const struct btf *base_btf)
594 static inline int btf_relocate(void *log, struct btf *btf, const struct btf *base_btf,
600 static inline int btf_field_iter_init(struct btf_field_iter *it, struct btf_type *t,
601 enum btf_field_iter_kind iter_kind)
606 static inline __u32 *btf_field_iter_next(struct btf_field_iter *it)
611 static inline const char *btf_name_by_offset(const struct btf *btf,
616 static inline u32 *btf_kfunc_id_set_contains(const struct btf *btf,
618 struct bpf_prog *prog)
623 static inline int register_btf_kfunc_id_set(enum bpf_prog_type prog_type,
624 const struct btf_kfunc_id_set *s)
628 static inline s32 btf_find_dtor_kfunc(struct btf *btf, u32 btf_id)
632 static inline int register_btf_id_dtor_kfuncs(const struct btf_id_dtor_kfunc *dtors,
633 u32 add_cnt, struct module *owner)
637 static inline struct btf_struct_meta *btf_find_struct_meta(const struct btf *btf, u32 btf_id)
642 btf_is_prog_ctx_type(struct bpf_verifier_log *log, const struct btf *btf,
643 const struct btf_type *t, enum bpf_prog_type prog_type,
648 static inline int get_kern_ctx_btf_id(struct bpf_verifier_log *log,
649 enum bpf_prog_type prog_type) {
652 static inline bool btf_types_are_same(const struct btf *btf1, u32 id1,
653 const struct btf *btf2, u32 id2)
659 static inline bool btf_type_is_struct_ptr(struct btf *btf, const struct btf_type *t)
661 if (!btf_type_is_ptr(t))
664 t = btf_type_skip_modifiers(btf, t->type, NULL);
666 return btf_type_is_struct(t);