bpf.h 50.3 KB
Newer Older
1
/* SPDX-License-Identifier: GPL-2.0-only */
2 3 4 5 6 7
/* Copyright (c) 2011-2014 PLUMgrid, http://plumgrid.com
 */
#ifndef _LINUX_BPF_H
#define _LINUX_BPF_H 1

#include <uapi/linux/bpf.h>
8

9
#include <linux/workqueue.h>
10
#include <linux/file.h>
11
#include <linux/percpu.h>
Z
Zi Shen Lim 已提交
12
#include <linux/err.h>
13
#include <linux/rbtree_latch.h>
14
#include <linux/numa.h>
15
#include <linux/mm_types.h>
16
#include <linux/wait.h>
A
Alexei Starovoitov 已提交
17
#include <linux/u64_stats_sync.h>
A
Alexei Starovoitov 已提交
18 19
#include <linux/refcount.h>
#include <linux/mutex.h>
20
#include <linux/module.h>
J
Jiri Olsa 已提交
21
#include <linux/kallsyms.h>
22

23
struct bpf_verifier_env;
24
struct bpf_verifier_log;
25
struct perf_event;
26
struct bpf_prog;
27
struct bpf_prog_aux;
28
struct bpf_map;
29
struct sock;
30
struct seq_file;
31
struct btf;
32
struct btf_type;
33
struct exception_table_entry;
34

35 36 37
extern struct idr btf_idr;
extern spinlock_t btf_idr_lock;

38 39 40
/* map is generic key/value storage optionally accesible by eBPF programs */
struct bpf_map_ops {
	/* funcs callable from userspace (via syscall) */
41
	int (*map_alloc_check)(union bpf_attr *attr);
42
	struct bpf_map *(*map_alloc)(union bpf_attr *attr);
43 44
	void (*map_release)(struct bpf_map *map, struct file *map_file);
	void (*map_free)(struct bpf_map *map);
45
	int (*map_get_next_key)(struct bpf_map *map, void *key, void *next_key);
46
	void (*map_release_uref)(struct bpf_map *map);
47
	void *(*map_lookup_elem_sys_only)(struct bpf_map *map, void *key);
48 49
	int (*map_lookup_batch)(struct bpf_map *map, const union bpf_attr *attr,
				union bpf_attr __user *uattr);
50 51 52
	int (*map_lookup_and_delete_batch)(struct bpf_map *map,
					   const union bpf_attr *attr,
					   union bpf_attr __user *uattr);
53 54 55 56
	int (*map_update_batch)(struct bpf_map *map, const union bpf_attr *attr,
				union bpf_attr __user *uattr);
	int (*map_delete_batch)(struct bpf_map *map, const union bpf_attr *attr,
				union bpf_attr __user *uattr);
57 58 59

	/* funcs callable from userspace and from eBPF programs */
	void *(*map_lookup_elem)(struct bpf_map *map, void *key);
60
	int (*map_update_elem)(struct bpf_map *map, void *key, void *value, u64 flags);
61
	int (*map_delete_elem)(struct bpf_map *map, void *key);
M
Mauricio Vasquez B 已提交
62 63 64
	int (*map_push_elem)(struct bpf_map *map, void *value, u64 flags);
	int (*map_pop_elem)(struct bpf_map *map, void *value);
	int (*map_peek_elem)(struct bpf_map *map, void *value);
65 66

	/* funcs called by prog_array and perf_event_array map */
67 68 69
	void *(*map_fd_get_ptr)(struct bpf_map *map, struct file *map_file,
				int fd);
	void (*map_fd_put_ptr)(void *ptr);
70
	u32 (*map_gen_lookup)(struct bpf_map *map, struct bpf_insn *insn_buf);
71
	u32 (*map_fd_sys_lookup_elem)(void *ptr);
72 73
	void (*map_seq_show_elem)(struct bpf_map *map, void *key,
				  struct seq_file *m);
74
	int (*map_check_btf)(const struct bpf_map *map,
75
			     const struct btf *btf,
76 77
			     const struct btf_type *key_type,
			     const struct btf_type *value_type);
78

79 80 81 82 83 84
	/* Prog poke tracking helpers. */
	int (*map_poke_track)(struct bpf_map *map, struct bpf_prog_aux *aux);
	void (*map_poke_untrack)(struct bpf_map *map, struct bpf_prog_aux *aux);
	void (*map_poke_run)(struct bpf_map *map, u32 key, struct bpf_prog *old,
			     struct bpf_prog *new);

85 86 87 88 89
	/* Direct value access helpers. */
	int (*map_direct_value_addr)(const struct bpf_map *map,
				     u64 *imm, u32 off);
	int (*map_direct_value_meta)(const struct bpf_map *map,
				     u64 imm, u32 *off);
90
	int (*map_mmap)(struct bpf_map *map, struct vm_area_struct *vma);
91 92
};

93 94 95 96 97
struct bpf_map_memory {
	u32 pages;
	struct user_struct *user;
};

98
struct bpf_map {
99
	/* The first two cachelines with read-mostly members of which some
100 101 102 103 104 105 106
	 * are also accessed in fast-path (e.g. ops, max_entries).
	 */
	const struct bpf_map_ops *ops ____cacheline_aligned;
	struct bpf_map *inner_map_meta;
#ifdef CONFIG_SECURITY
	void *security;
#endif
107 108 109 110
	enum bpf_map_type map_type;
	u32 key_size;
	u32 value_size;
	u32 max_entries;
111
	u32 map_flags;
112
	int spin_lock_off; /* >=0 valid offset, <0 error */
M
Martin KaFai Lau 已提交
113
	u32 id;
114
	int numa_node;
115 116
	u32 btf_key_type_id;
	u32 btf_value_type_id;
117
	struct btf *btf;
118
	struct bpf_map_memory memory;
119
	char name[BPF_OBJ_NAME_LEN];
120
	u32 btf_vmlinux_value_type_id;
121
	bool unpriv_array;
122 123
	bool frozen; /* write-once; write-protected by freeze_mutex */
	/* 22 bytes hole */
124

125
	/* The 3rd and 4th cacheline with misc members to avoid false sharing
126 127
	 * particularly with refcounting.
	 */
128 129
	atomic64_t refcnt ____cacheline_aligned;
	atomic64_t usercnt;
130
	struct work_struct work;
131 132
	struct mutex freeze_mutex;
	u64 writecnt; /* writable mmap cnt; protected by freeze_mutex */
133 134
};

135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
static inline bool map_value_has_spin_lock(const struct bpf_map *map)
{
	return map->spin_lock_off >= 0;
}

static inline void check_and_init_map_lock(struct bpf_map *map, void *dst)
{
	if (likely(!map_value_has_spin_lock(map)))
		return;
	*(struct bpf_spin_lock *)(dst + map->spin_lock_off) =
		(struct bpf_spin_lock){};
}

/* copy everything but bpf_spin_lock */
static inline void copy_map_value(struct bpf_map *map, void *dst, void *src)
{
	if (unlikely(map_value_has_spin_lock(map))) {
		u32 off = map->spin_lock_off;

		memcpy(dst, src, off);
		memcpy(dst + off + sizeof(struct bpf_spin_lock),
		       src + off + sizeof(struct bpf_spin_lock),
		       map->value_size - off - sizeof(struct bpf_spin_lock));
	} else {
		memcpy(dst, src, map->value_size);
	}
}
162 163
void copy_map_value_locked(struct bpf_map *map, void *dst, void *src,
			   bool lock_src);
164

165
struct bpf_offload_dev;
166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
struct bpf_offloaded_map;

struct bpf_map_dev_ops {
	int (*map_get_next_key)(struct bpf_offloaded_map *map,
				void *key, void *next_key);
	int (*map_lookup_elem)(struct bpf_offloaded_map *map,
			       void *key, void *value);
	int (*map_update_elem)(struct bpf_offloaded_map *map,
			       void *key, void *value, u64 flags);
	int (*map_delete_elem)(struct bpf_offloaded_map *map, void *key);
};

struct bpf_offloaded_map {
	struct bpf_map map;
	struct net_device *netdev;
	const struct bpf_map_dev_ops *dev_ops;
	void *dev_priv;
	struct list_head offloads;
};

static inline struct bpf_offloaded_map *map_to_offmap(struct bpf_map *map)
{
	return container_of(map, struct bpf_offloaded_map, map);
}

191 192 193 194 195
static inline bool bpf_map_offload_neutral(const struct bpf_map *map)
{
	return map->map_type == BPF_MAP_TYPE_PERF_EVENT_ARRAY;
}

196 197
static inline bool bpf_map_support_seq_show(const struct bpf_map *map)
{
198 199
	return (map->btf_value_type_id || map->btf_vmlinux_value_type_id) &&
		map->ops->map_seq_show_elem;
200 201
}

202
int map_check_no_btf(const struct bpf_map *map,
203
		     const struct btf *btf,
204 205 206
		     const struct btf_type *key_type,
		     const struct btf_type *value_type);

207 208
extern const struct bpf_map_ops bpf_map_offload_ops;

209 210
/* function argument constraints */
enum bpf_arg_type {
211
	ARG_DONTCARE = 0,	/* unused argument in helper function */
212 213 214 215 216 217 218

	/* the following constraints used to prototype
	 * bpf_map_lookup/update/delete_elem() functions
	 */
	ARG_CONST_MAP_PTR,	/* const argument used as pointer to bpf_map */
	ARG_PTR_TO_MAP_KEY,	/* pointer to stack used as map key */
	ARG_PTR_TO_MAP_VALUE,	/* pointer to stack used as map value */
219
	ARG_PTR_TO_UNINIT_MAP_VALUE,	/* pointer to valid memory used to store a map value */
220
	ARG_PTR_TO_MAP_VALUE_OR_NULL,	/* pointer to stack used as map value or NULL */
221 222 223 224

	/* the following constraints used to prototype bpf_memcmp() and other
	 * functions that access data on eBPF program stack
	 */
225
	ARG_PTR_TO_MEM,		/* pointer to valid memory (stack, packet, map value) */
226
	ARG_PTR_TO_MEM_OR_NULL, /* pointer to valid memory or NULL */
227 228 229
	ARG_PTR_TO_UNINIT_MEM,	/* pointer to memory does not need to be initialized,
				 * helper function must fill all bytes or clear
				 * them in error case.
230 231
				 */

232 233
	ARG_CONST_SIZE,		/* number of bytes accessed from memory */
	ARG_CONST_SIZE_OR_ZERO,	/* number of bytes accessed from memory or 0 */
234

235
	ARG_PTR_TO_CTX,		/* pointer to context */
236
	ARG_ANYTHING,		/* any (initialized) argument is ok */
237
	ARG_PTR_TO_SPIN_LOCK,	/* pointer to bpf_spin_lock */
238
	ARG_PTR_TO_SOCK_COMMON,	/* pointer to sock_common */
239 240
	ARG_PTR_TO_INT,		/* pointer to int */
	ARG_PTR_TO_LONG,	/* pointer to long */
241
	ARG_PTR_TO_SOCKET,	/* pointer to bpf_sock (fullsock) */
242
	ARG_PTR_TO_BTF_ID,	/* pointer to in-kernel struct */
243 244 245 246 247 248
};

/* type of values returned from helper functions */
enum bpf_return_type {
	RET_INTEGER,			/* function returns integer */
	RET_VOID,			/* function doesn't return anything */
249
	RET_PTR_TO_MAP_VALUE,		/* returns a pointer to map elem value */
250
	RET_PTR_TO_MAP_VALUE_OR_NULL,	/* returns a pointer to map elem value or NULL */
251
	RET_PTR_TO_SOCKET_OR_NULL,	/* returns a pointer to a socket or NULL */
252
	RET_PTR_TO_TCP_SOCK_OR_NULL,	/* returns a pointer to a tcp_sock or NULL */
253
	RET_PTR_TO_SOCK_COMMON_OR_NULL,	/* returns a pointer to a sock_common or NULL */
254 255
};

256 257 258 259 260 261 262
/* eBPF function prototype used by verifier to allow BPF_CALLs from eBPF programs
 * to in-kernel helper functions and for adjusting imm32 field in BPF_CALL
 * instructions after verifying
 */
struct bpf_func_proto {
	u64 (*func)(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5);
	bool gpl_only;
263
	bool pkt_access;
264
	enum bpf_return_type ret_type;
265 266 267 268 269 270 271 272 273 274
	union {
		struct {
			enum bpf_arg_type arg1_type;
			enum bpf_arg_type arg2_type;
			enum bpf_arg_type arg3_type;
			enum bpf_arg_type arg4_type;
			enum bpf_arg_type arg5_type;
		};
		enum bpf_arg_type arg_type[5];
	};
275
	int *btf_id; /* BTF ids of arguments */
276 277 278 279 280 281 282 283 284 285 286
};

/* bpf_context is intentionally undefined structure. Pointer to bpf_context is
 * the first argument to eBPF programs.
 * For socket filters: 'struct bpf_context *' == 'struct sk_buff *'
 */
struct bpf_context;

enum bpf_access_type {
	BPF_READ = 1,
	BPF_WRITE = 2
287 288
};

289
/* types of values stored in eBPF registers */
290 291 292 293 294 295 296 297 298
/* Pointer types represent:
 * pointer
 * pointer + imm
 * pointer + (u16) var
 * pointer + (u16) var + imm
 * if (range > 0) then [ptr, ptr + range - off) is safe to access
 * if (id > 0) means that some 'var' was added
 * if (off > 0) means that 'imm' was added
 */
299 300
enum bpf_reg_type {
	NOT_INIT = 0,		 /* nothing was written into register */
301
	SCALAR_VALUE,		 /* reg doesn't contain a valid pointer */
302 303 304 305
	PTR_TO_CTX,		 /* reg points to bpf_context */
	CONST_PTR_TO_MAP,	 /* reg points to struct bpf_map */
	PTR_TO_MAP_VALUE,	 /* reg points to map element value */
	PTR_TO_MAP_VALUE_OR_NULL,/* points to map elem value or NULL */
306
	PTR_TO_STACK,		 /* reg == frame_pointer + offset */
307
	PTR_TO_PACKET_META,	 /* skb->data - meta_len */
308
	PTR_TO_PACKET,		 /* reg points to skb->data */
309
	PTR_TO_PACKET_END,	 /* skb->data + headlen */
310
	PTR_TO_FLOW_KEYS,	 /* reg points to bpf_flow_keys */
311 312
	PTR_TO_SOCKET,		 /* reg points to struct bpf_sock */
	PTR_TO_SOCKET_OR_NULL,	 /* reg points to struct bpf_sock or NULL */
313 314
	PTR_TO_SOCK_COMMON,	 /* reg points to sock_common */
	PTR_TO_SOCK_COMMON_OR_NULL, /* reg points to sock_common or NULL */
315 316
	PTR_TO_TCP_SOCK,	 /* reg points to struct tcp_sock */
	PTR_TO_TCP_SOCK_OR_NULL, /* reg points to struct tcp_sock or NULL */
317
	PTR_TO_TP_BUFFER,	 /* reg points to a writable raw tp's buffer */
318
	PTR_TO_XDP_SOCK,	 /* reg points to struct xdp_sock */
319
	PTR_TO_BTF_ID,		 /* reg points to kernel struct */
320 321
};

322 323 324 325 326
/* The information passed from prog-specific *_is_valid_access
 * back to the verifier.
 */
struct bpf_insn_access_aux {
	enum bpf_reg_type reg_type;
327 328 329 330 331
	union {
		int ctx_field_size;
		u32 btf_id;
	};
	struct bpf_verifier_log *log; /* for verbose logs */
332 333
};

334 335 336 337 338 339
static inline void
bpf_ctx_record_field_size(struct bpf_insn_access_aux *aux, u32 size)
{
	aux->ctx_field_size = size;
}

340 341 342 343 344
struct bpf_prog_ops {
	int (*test_run)(struct bpf_prog *prog, const union bpf_attr *kattr,
			union bpf_attr __user *uattr);
};

345 346
struct bpf_verifier_ops {
	/* return eBPF function prototype for verification */
347 348 349
	const struct bpf_func_proto *
	(*get_func_proto)(enum bpf_func_id func_id,
			  const struct bpf_prog *prog);
350 351 352 353

	/* return true if 'size' wide access at offset 'off' within bpf_context
	 * with 'type' (read or write) is allowed
	 */
354
	bool (*is_valid_access)(int off, int size, enum bpf_access_type type,
355
				const struct bpf_prog *prog,
356
				struct bpf_insn_access_aux *info);
357 358
	int (*gen_prologue)(struct bpf_insn *insn, bool direct_write,
			    const struct bpf_prog *prog);
359 360
	int (*gen_ld_abs)(const struct bpf_insn *orig,
			  struct bpf_insn *insn_buf);
361 362 363
	u32 (*convert_ctx_access)(enum bpf_access_type type,
				  const struct bpf_insn *src,
				  struct bpf_insn *dst,
364
				  struct bpf_prog *prog, u32 *target_size);
365 366 367 368
	int (*btf_struct_access)(struct bpf_verifier_log *log,
				 const struct btf_type *t, int off, int size,
				 enum bpf_access_type atype,
				 u32 *next_btf_id);
369 370
};

371
struct bpf_prog_offload_ops {
372
	/* verifier basic callbacks */
373 374
	int (*insn_hook)(struct bpf_verifier_env *env,
			 int insn_idx, int prev_insn_idx);
375
	int (*finalize)(struct bpf_verifier_env *env);
376 377 378 379 380
	/* verifier optimization callbacks (called after .finalize) */
	int (*replace_insn)(struct bpf_verifier_env *env, u32 off,
			    struct bpf_insn *insn);
	int (*remove_insns)(struct bpf_verifier_env *env, u32 off, u32 cnt);
	/* program management callbacks */
381 382
	int (*prepare)(struct bpf_prog *prog);
	int (*translate)(struct bpf_prog *prog);
383
	void (*destroy)(struct bpf_prog *prog);
384 385
};

386
struct bpf_prog_offload {
387 388
	struct bpf_prog		*prog;
	struct net_device	*netdev;
389
	struct bpf_offload_dev	*offdev;
390 391 392
	void			*dev_priv;
	struct list_head	offloads;
	bool			dev_state;
393
	bool			opt_failed;
394 395
	void			*jited_image;
	u32			jited_len;
396 397
};

398 399
enum bpf_cgroup_storage_type {
	BPF_CGROUP_STORAGE_SHARED,
400
	BPF_CGROUP_STORAGE_PERCPU,
401 402 403 404 405
	__BPF_CGROUP_STORAGE_MAX
};

#define MAX_BPF_CGROUP_STORAGE_TYPE __BPF_CGROUP_STORAGE_MAX

406 407 408 409 410
/* The longest tracepoint has 12 args.
 * See include/trace/bpf_probe.h
 */
#define MAX_BPF_FUNC_ARGS 12

A
Alexei Starovoitov 已提交
411 412 413 414
struct bpf_prog_stats {
	u64 cnt;
	u64 nsecs;
	struct u64_stats_sync syncp;
E
Eric Dumazet 已提交
415
} __aligned(2 * sizeof(u64));
A
Alexei Starovoitov 已提交
416

A
Alexei Starovoitov 已提交
417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436
struct btf_func_model {
	u8 ret_size;
	u8 nr_args;
	u8 arg_size[MAX_BPF_FUNC_ARGS];
};

/* Restore arguments before returning from trampoline to let original function
 * continue executing. This flag is used for fentry progs when there are no
 * fexit progs.
 */
#define BPF_TRAMP_F_RESTORE_REGS	BIT(0)
/* Call original function after fentry progs, but before fexit progs.
 * Makes sense for fentry/fexit, normal calls and indirect calls.
 */
#define BPF_TRAMP_F_CALL_ORIG		BIT(1)
/* Skip current frame and return to parent.  Makes sense for fentry/fexit
 * programs only. Should not be used with normal calls and indirect calls.
 */
#define BPF_TRAMP_F_SKIP_FRAME		BIT(2)

K
KP Singh 已提交
437 438 439 440 441 442 443 444 445 446
/* Each call __bpf_prog_enter + call bpf_func + call __bpf_prog_exit is ~50
 * bytes on x86.  Pick a number to fit into BPF_IMAGE_SIZE / 2
 */
#define BPF_MAX_TRAMP_PROGS 40

struct bpf_tramp_progs {
	struct bpf_prog *progs[BPF_MAX_TRAMP_PROGS];
	int nr_progs;
};

A
Alexei Starovoitov 已提交
447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466
/* Different use cases for BPF trampoline:
 * 1. replace nop at the function entry (kprobe equivalent)
 *    flags = BPF_TRAMP_F_RESTORE_REGS
 *    fentry = a set of programs to run before returning from trampoline
 *
 * 2. replace nop at the function entry (kprobe + kretprobe equivalent)
 *    flags = BPF_TRAMP_F_CALL_ORIG | BPF_TRAMP_F_SKIP_FRAME
 *    orig_call = fentry_ip + MCOUNT_INSN_SIZE
 *    fentry = a set of program to run before calling original function
 *    fexit = a set of program to run after original function
 *
 * 3. replace direct call instruction anywhere in the function body
 *    or assign a function pointer for indirect call (like tcp_congestion_ops->cong_avoid)
 *    With flags = 0
 *      fentry = a set of programs to run before returning from trampoline
 *    With flags = BPF_TRAMP_F_CALL_ORIG
 *      orig_call = original callback addr or direct function addr
 *      fentry = a set of program to run before calling original function
 *      fexit = a set of program to run after original function
 */
467 468
int arch_prepare_bpf_trampoline(void *image, void *image_end,
				const struct btf_func_model *m, u32 flags,
K
KP Singh 已提交
469
				struct bpf_tramp_progs *tprogs,
A
Alexei Starovoitov 已提交
470 471 472 473 474
				void *orig_call);
/* these two functions are called from generated trampoline */
u64 notrace __bpf_prog_enter(void);
void notrace __bpf_prog_exit(struct bpf_prog *prog, u64 start);

J
Jiri Olsa 已提交
475 476 477
struct bpf_ksym {
	unsigned long		 start;
	unsigned long		 end;
J
Jiri Olsa 已提交
478
	char			 name[KSYM_NAME_LEN];
479
	struct list_head	 lnode;
J
Jiri Olsa 已提交
480 481
};

A
Alexei Starovoitov 已提交
482 483 484
enum bpf_tramp_prog_type {
	BPF_TRAMP_FENTRY,
	BPF_TRAMP_FEXIT,
K
KP Singh 已提交
485
	BPF_TRAMP_MODIFY_RETURN,
486 487
	BPF_TRAMP_MAX,
	BPF_TRAMP_REPLACE, /* more than MAX */
A
Alexei Starovoitov 已提交
488 489 490 491 492 493 494 495 496 497 498 499
};

struct bpf_trampoline {
	/* hlist for trampoline_table */
	struct hlist_node hlist;
	/* serializes access to fields of this trampoline */
	struct mutex mutex;
	refcount_t refcnt;
	u64 key;
	struct {
		struct btf_func_model model;
		void *addr;
500
		bool ftrace_managed;
A
Alexei Starovoitov 已提交
501
	} func;
502 503 504 505 506
	/* if !NULL this is BPF_PROG_TYPE_EXT program that extends another BPF
	 * program by replacing one of its functions. func.addr is the address
	 * of the function it replaced.
	 */
	struct bpf_prog *extension_prog;
A
Alexei Starovoitov 已提交
507 508 509 510 511 512 513 514
	/* list of BPF programs using this trampoline */
	struct hlist_head progs_hlist[BPF_TRAMP_MAX];
	/* Number of attached programs. A counter per kind. */
	int progs_cnt[BPF_TRAMP_MAX];
	/* Executable image of trampoline */
	void *image;
	u64 selector;
};
B
Björn Töpel 已提交
515

516
#define BPF_DISPATCHER_MAX 48 /* Fits in 2048B */
B
Björn Töpel 已提交
517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532

struct bpf_dispatcher_prog {
	struct bpf_prog *prog;
	refcount_t users;
};

struct bpf_dispatcher {
	/* dispatcher mutex */
	struct mutex mutex;
	void *func;
	struct bpf_dispatcher_prog progs[BPF_DISPATCHER_MAX];
	int num_progs;
	void *image;
	u32 image_off;
};

533
static __always_inline unsigned int bpf_dispatcher_nop_func(
534 535 536 537 538 539 540
	const void *ctx,
	const struct bpf_insn *insnsi,
	unsigned int (*bpf_func)(const void *,
				 const struct bpf_insn *))
{
	return bpf_func(ctx, insnsi);
}
A
Alexei Starovoitov 已提交
541 542 543 544 545
#ifdef CONFIG_BPF_JIT
struct bpf_trampoline *bpf_trampoline_lookup(u64 key);
int bpf_trampoline_link_prog(struct bpf_prog *prog);
int bpf_trampoline_unlink_prog(struct bpf_prog *prog);
void bpf_trampoline_put(struct bpf_trampoline *tr);
B
Björn Töpel 已提交
546 547
#define BPF_DISPATCHER_INIT(name) {			\
	.mutex = __MUTEX_INITIALIZER(name.mutex),	\
548
	.func = &name##_func,				\
B
Björn Töpel 已提交
549 550 551 552 553 554 555
	.progs = {},					\
	.num_progs = 0,					\
	.image = NULL,					\
	.image_off = 0					\
}

#define DEFINE_BPF_DISPATCHER(name)					\
556
	noinline unsigned int bpf_dispatcher_##name##_func(		\
B
Björn Töpel 已提交
557 558 559 560 561 562 563
		const void *ctx,					\
		const struct bpf_insn *insnsi,				\
		unsigned int (*bpf_func)(const void *,			\
					 const struct bpf_insn *))	\
	{								\
		return bpf_func(ctx, insnsi);				\
	}								\
564 565 566
	EXPORT_SYMBOL(bpf_dispatcher_##name##_func);			\
	struct bpf_dispatcher bpf_dispatcher_##name =			\
		BPF_DISPATCHER_INIT(bpf_dispatcher_##name);
B
Björn Töpel 已提交
567
#define DECLARE_BPF_DISPATCHER(name)					\
568
	unsigned int bpf_dispatcher_##name##_func(			\
B
Björn Töpel 已提交
569 570 571 572
		const void *ctx,					\
		const struct bpf_insn *insnsi,				\
		unsigned int (*bpf_func)(const void *,			\
					 const struct bpf_insn *));	\
573 574 575
	extern struct bpf_dispatcher bpf_dispatcher_##name;
#define BPF_DISPATCHER_FUNC(name) bpf_dispatcher_##name##_func
#define BPF_DISPATCHER_PTR(name) (&bpf_dispatcher_##name)
B
Björn Töpel 已提交
576 577
void bpf_dispatcher_change_prog(struct bpf_dispatcher *d, struct bpf_prog *from,
				struct bpf_prog *to);
578 579 580 581 582 583 584
struct bpf_image {
	struct latch_tree_node tnode;
	unsigned char data[];
};
#define BPF_IMAGE_SIZE (PAGE_SIZE - sizeof(struct bpf_image))
bool is_bpf_image_address(unsigned long address);
void *bpf_image_alloc(void);
A
Alexei Starovoitov 已提交
585 586 587 588 589 590 591 592 593 594 595 596 597 598
#else
static inline struct bpf_trampoline *bpf_trampoline_lookup(u64 key)
{
	return NULL;
}
static inline int bpf_trampoline_link_prog(struct bpf_prog *prog)
{
	return -ENOTSUPP;
}
static inline int bpf_trampoline_unlink_prog(struct bpf_prog *prog)
{
	return -ENOTSUPP;
}
static inline void bpf_trampoline_put(struct bpf_trampoline *tr) {}
B
Björn Töpel 已提交
599 600
#define DEFINE_BPF_DISPATCHER(name)
#define DECLARE_BPF_DISPATCHER(name)
601
#define BPF_DISPATCHER_FUNC(name) bpf_dispatcher_nop_func
B
Björn Töpel 已提交
602 603 604 605
#define BPF_DISPATCHER_PTR(name) NULL
static inline void bpf_dispatcher_change_prog(struct bpf_dispatcher *d,
					      struct bpf_prog *from,
					      struct bpf_prog *to) {}
606 607 608 609
static inline bool is_bpf_image_address(unsigned long address)
{
	return false;
}
A
Alexei Starovoitov 已提交
610 611
#endif

612
struct bpf_func_info_aux {
613
	u16 linkage;
614 615 616
	bool unreliable;
};

617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634
enum bpf_jit_poke_reason {
	BPF_POKE_REASON_TAIL_CALL,
};

/* Descriptor of pokes pointing /into/ the JITed image. */
struct bpf_jit_poke_descriptor {
	void *ip;
	union {
		struct {
			struct bpf_map *map;
			u32 key;
		} tail_call;
	};
	bool ip_stable;
	u8 adj_off;
	u16 reason;
};

635
struct bpf_prog_aux {
636
	atomic64_t refcnt;
637
	u32 used_map_cnt;
638
	u32 max_ctx_offset;
639
	u32 max_pkt_offset;
640
	u32 max_tp_access;
641
	u32 stack_depth;
M
Martin KaFai Lau 已提交
642
	u32 id;
643 644
	u32 func_cnt; /* used by non-func prog as the number of func progs */
	u32 func_idx; /* 0 for non-func prog, the index in func array for func prog */
645
	u32 attach_btf_id; /* in-kernel BTF type id to attach to */
646
	struct bpf_prog *linked_prog;
647
	bool verifier_zext; /* Zero extensions has been inserted by verifier. */
648
	bool offload_requested;
649
	bool attach_btf_trace; /* true if attaching to BTF-enabled raw tp */
650
	bool func_proto_unreliable;
A
Alexei Starovoitov 已提交
651 652 653
	enum bpf_tramp_prog_type trampoline_prog_type;
	struct bpf_trampoline *trampoline;
	struct hlist_node tramp_hlist;
654 655 656 657
	/* BTF_KIND_FUNC_PROTO for valid attach_btf_id */
	const struct btf_type *attach_func_proto;
	/* function name for valid attach_btf_id */
	const char *attach_func_name;
658 659
	struct bpf_prog **func;
	void *jit_data; /* JIT specific data. arch dependent */
660 661
	struct bpf_jit_poke_descriptor *poke_tab;
	u32 size_poke_tab;
662
	struct latch_tree_node ksym_tnode;
J
Jiri Olsa 已提交
663
	struct bpf_ksym ksym;
664
	const struct bpf_prog_ops *ops;
665 666
	struct bpf_map **used_maps;
	struct bpf_prog *prog;
667
	struct user_struct *user;
668
	u64 load_time; /* ns since boottime */
669
	struct bpf_map *cgroup_storage[MAX_BPF_CGROUP_STORAGE_TYPE];
670
	char name[BPF_OBJ_NAME_LEN];
671 672 673
#ifdef CONFIG_SECURITY
	void *security;
#endif
674
	struct bpf_prog_offload *offload;
Y
Yonghong Song 已提交
675
	struct btf *btf;
676
	struct bpf_func_info *func_info;
677
	struct bpf_func_info_aux *func_info_aux;
M
Martin KaFai Lau 已提交
678 679 680 681 682 683 684 685 686 687 688 689 690 691 692
	/* bpf_line_info loaded from userspace.  linfo->insn_off
	 * has the xlated insn offset.
	 * Both the main and sub prog share the same linfo.
	 * The subprog can access its first linfo by
	 * using the linfo_idx.
	 */
	struct bpf_line_info *linfo;
	/* jited_linfo is the jited addr of the linfo.  It has a
	 * one to one mapping to linfo:
	 * jited_linfo[i] is the jited addr for the linfo[i]->insn_off.
	 * Both the main and sub prog share the same jited_linfo.
	 * The subprog can access its first jited_linfo by
	 * using the linfo_idx.
	 */
	void **jited_linfo;
693
	u32 func_info_cnt;
M
Martin KaFai Lau 已提交
694 695 696 697 698 699
	u32 nr_linfo;
	/* subprog can use linfo_idx to access its first linfo and
	 * jited_linfo.
	 * main prog always has linfo_idx == 0
	 */
	u32 linfo_idx;
700 701
	u32 num_exentries;
	struct exception_table_entry *extable;
A
Alexei Starovoitov 已提交
702
	struct bpf_prog_stats __percpu *stats;
703 704 705 706
	union {
		struct work_struct work;
		struct rcu_head	rcu;
	};
707 708
};

709 710 711 712 713 714 715 716
struct bpf_array_aux {
	/* 'Ownership' of prog array is claimed by the first program that
	 * is going to use this map or by the first program which FD is
	 * stored in the map to make sure that all callers and callees have
	 * the same prog type and JITed flag.
	 */
	enum bpf_prog_type type;
	bool jited;
717 718 719 720 721
	/* Programs with direct jumps into programs part of this array. */
	struct list_head poke_progs;
	struct bpf_map *map;
	struct mutex poke_mutex;
	struct work_struct work;
722 723
};

724
struct bpf_struct_ops_value;
725 726 727 728 729 730 731 732 733
struct btf_type;
struct btf_member;

#define BPF_STRUCT_OPS_MAX_NR_MEMBERS 64
struct bpf_struct_ops {
	const struct bpf_verifier_ops *verifier_ops;
	int (*init)(struct btf *btf);
	int (*check_member)(const struct btf_type *t,
			    const struct btf_member *member);
734 735 736 737 738
	int (*init_member)(const struct btf_type *t,
			   const struct btf_member *member,
			   void *kdata, const void *udata);
	int (*reg)(void *kdata);
	void (*unreg)(void *kdata);
739
	const struct btf_type *type;
740
	const struct btf_type *value_type;
741 742 743
	const char *name;
	struct btf_func_model func_models[BPF_STRUCT_OPS_MAX_NR_MEMBERS];
	u32 type_id;
744
	u32 value_id;
745 746 747
};

#if defined(CONFIG_BPF_JIT) && defined(CONFIG_BPF_SYSCALL)
748
#define BPF_MODULE_OWNER ((void *)((0xeB9FUL << 2) + POISON_POINTER_DELTA))
749
const struct bpf_struct_ops *bpf_struct_ops_find(u32 type_id);
750
void bpf_struct_ops_init(struct btf *btf, struct bpf_verifier_log *log);
751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768
bool bpf_struct_ops_get(const void *kdata);
void bpf_struct_ops_put(const void *kdata);
int bpf_struct_ops_map_sys_lookup_elem(struct bpf_map *map, void *key,
				       void *value);
static inline bool bpf_try_module_get(const void *data, struct module *owner)
{
	if (owner == BPF_MODULE_OWNER)
		return bpf_struct_ops_get(data);
	else
		return try_module_get(owner);
}
static inline void bpf_module_put(const void *data, struct module *owner)
{
	if (owner == BPF_MODULE_OWNER)
		bpf_struct_ops_put(data);
	else
		module_put(owner);
}
769 770 771 772 773
#else
static inline const struct bpf_struct_ops *bpf_struct_ops_find(u32 type_id)
{
	return NULL;
}
774 775 776 777
static inline void bpf_struct_ops_init(struct btf *btf,
				       struct bpf_verifier_log *log)
{
}
778 779 780 781 782 783 784 785 786 787 788 789 790 791
static inline bool bpf_try_module_get(const void *data, struct module *owner)
{
	return try_module_get(owner);
}
static inline void bpf_module_put(const void *data, struct module *owner)
{
	module_put(owner);
}
static inline int bpf_struct_ops_map_sys_lookup_elem(struct bpf_map *map,
						     void *key,
						     void *value)
{
	return -EINVAL;
}
792 793
#endif

794 795 796
struct bpf_array {
	struct bpf_map map;
	u32 elem_size;
797
	u32 index_mask;
798
	struct bpf_array_aux *aux;
799 800
	union {
		char value[0] __aligned(8);
801
		void *ptrs[0] __aligned(8);
802
		void __percpu *pptrs[0] __aligned(8);
803 804
	};
};
805

806
#define BPF_COMPLEXITY_LIMIT_INSNS      1000000 /* yes. 1M insns */
807 808
#define MAX_TAIL_CALL_CNT 32

809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837
#define BPF_F_ACCESS_MASK	(BPF_F_RDONLY |		\
				 BPF_F_RDONLY_PROG |	\
				 BPF_F_WRONLY |		\
				 BPF_F_WRONLY_PROG)

#define BPF_MAP_CAN_READ	BIT(0)
#define BPF_MAP_CAN_WRITE	BIT(1)

static inline u32 bpf_map_flags_to_cap(struct bpf_map *map)
{
	u32 access_flags = map->map_flags & (BPF_F_RDONLY_PROG | BPF_F_WRONLY_PROG);

	/* Combination of BPF_F_RDONLY_PROG | BPF_F_WRONLY_PROG is
	 * not possible.
	 */
	if (access_flags & BPF_F_RDONLY_PROG)
		return BPF_MAP_CAN_READ;
	else if (access_flags & BPF_F_WRONLY_PROG)
		return BPF_MAP_CAN_WRITE;
	else
		return BPF_MAP_CAN_READ | BPF_MAP_CAN_WRITE;
}

static inline bool bpf_map_flags_access_ok(u32 access_flags)
{
	return (access_flags & (BPF_F_RDONLY_PROG | BPF_F_WRONLY_PROG)) !=
	       (BPF_F_RDONLY_PROG | BPF_F_WRONLY_PROG);
}

838 839 840 841 842 843 844
struct bpf_event_entry {
	struct perf_event *event;
	struct file *perf_file;
	struct file *map_file;
	struct rcu_head rcu;
};

845
bool bpf_prog_array_compatible(struct bpf_array *array, const struct bpf_prog *fp);
846
int bpf_prog_calc_tag(struct bpf_prog *fp);
847
const char *kernel_type_name(u32 btf_type_id);
848

849
const struct bpf_func_proto *bpf_get_trace_printk_proto(void);
850 851

typedef unsigned long (*bpf_ctx_copy_t)(void *dst, const void *src,
852
					unsigned long off, unsigned long len);
853 854 855 856 857
typedef u32 (*bpf_convert_ctx_access_t)(enum bpf_access_type type,
					const struct bpf_insn *src,
					struct bpf_insn *dst,
					struct bpf_prog *prog,
					u32 *target_size);
858 859 860

u64 bpf_event_output(struct bpf_map *map, u64 flags, void *meta, u64 meta_size,
		     void *ctx, u64 ctx_size, bpf_ctx_copy_t ctx_copy);
861

862 863 864 865 866 867 868 869 870 871 872 873
/* an array of programs to be executed under rcu_lock.
 *
 * Typical usage:
 * ret = BPF_PROG_RUN_ARRAY(&bpf_prog_array, ctx, BPF_PROG_RUN);
 *
 * the structure returned by bpf_prog_array_alloc() should be populated
 * with program pointers and the last pointer must be NULL.
 * The user has to keep refcnt on the program and make sure the program
 * is removed from the array before bpf_prog_put().
 * The 'struct bpf_prog_array *' should only be replaced with xchg()
 * since other cpus are walking the array of pointers in parallel.
 */
874 875
struct bpf_prog_array_item {
	struct bpf_prog *prog;
876
	struct bpf_cgroup_storage *cgroup_storage[MAX_BPF_CGROUP_STORAGE_TYPE];
877 878
};

879 880
struct bpf_prog_array {
	struct rcu_head rcu;
881
	struct bpf_prog_array_item items[];
882 883
};

884
struct bpf_prog_array *bpf_prog_array_alloc(u32 prog_cnt, gfp_t flags);
885 886
void bpf_prog_array_free(struct bpf_prog_array *progs);
int bpf_prog_array_length(struct bpf_prog_array *progs);
887
bool bpf_prog_array_is_empty(struct bpf_prog_array *array);
888
int bpf_prog_array_copy_to_user(struct bpf_prog_array *progs,
889
				__u32 __user *prog_ids, u32 cnt);
890

891
void bpf_prog_array_delete_safe(struct bpf_prog_array *progs,
892
				struct bpf_prog *old_prog);
893
int bpf_prog_array_copy_info(struct bpf_prog_array *array,
894 895
			     u32 *prog_ids, u32 request_cnt,
			     u32 *prog_cnt);
896
int bpf_prog_array_copy(struct bpf_prog_array *old_array,
897 898 899 900 901
			struct bpf_prog *exclude_prog,
			struct bpf_prog *include_prog,
			struct bpf_prog_array **new_array);

#define __BPF_PROG_RUN_ARRAY(array, ctx, func, check_non_null)	\
902
	({						\
903 904
		struct bpf_prog_array_item *_item;	\
		struct bpf_prog *_prog;			\
905
		struct bpf_prog_array *_array;		\
906
		u32 _ret = 1;				\
907
		migrate_disable();			\
908
		rcu_read_lock();			\
909 910 911
		_array = rcu_dereference(array);	\
		if (unlikely(check_non_null && !_array))\
			goto _out;			\
912 913 914 915 916
		_item = &_array->items[0];		\
		while ((_prog = READ_ONCE(_item->prog))) {		\
			bpf_cgroup_storage_set(_item->cgroup_storage);	\
			_ret &= func(_prog, ctx);	\
			_item++;			\
917 918
		}					\
_out:							\
919
		rcu_read_unlock();			\
920
		migrate_enable();			\
921 922 923
		_ret;					\
	 })

924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953
/* To be used by __cgroup_bpf_run_filter_skb for EGRESS BPF progs
 * so BPF programs can request cwr for TCP packets.
 *
 * Current cgroup skb programs can only return 0 or 1 (0 to drop the
 * packet. This macro changes the behavior so the low order bit
 * indicates whether the packet should be dropped (0) or not (1)
 * and the next bit is a congestion notification bit. This could be
 * used by TCP to call tcp_enter_cwr()
 *
 * Hence, new allowed return values of CGROUP EGRESS BPF programs are:
 *   0: drop packet
 *   1: keep packet
 *   2: drop packet and cn
 *   3: keep packet and cn
 *
 * This macro then converts it to one of the NET_XMIT or an error
 * code that is then interpreted as drop packet (and no cn):
 *   0: NET_XMIT_SUCCESS  skb should be transmitted
 *   1: NET_XMIT_DROP     skb should be dropped and cn
 *   2: NET_XMIT_CN       skb should be transmitted and cn
 *   3: -EPERM            skb should be dropped
 */
#define BPF_PROG_CGROUP_INET_EGRESS_RUN_ARRAY(array, ctx, func)		\
	({						\
		struct bpf_prog_array_item *_item;	\
		struct bpf_prog *_prog;			\
		struct bpf_prog_array *_array;		\
		u32 ret;				\
		u32 _ret = 1;				\
		u32 _cn = 0;				\
954
		migrate_disable();			\
955 956 957 958 959 960 961 962 963 964 965
		rcu_read_lock();			\
		_array = rcu_dereference(array);	\
		_item = &_array->items[0];		\
		while ((_prog = READ_ONCE(_item->prog))) {		\
			bpf_cgroup_storage_set(_item->cgroup_storage);	\
			ret = func(_prog, ctx);		\
			_ret &= (ret & 1);		\
			_cn |= (ret & 2);		\
			_item++;			\
		}					\
		rcu_read_unlock();			\
966
		migrate_enable();			\
967 968 969 970 971 972 973
		if (_ret)				\
			_ret = (_cn ? NET_XMIT_CN : NET_XMIT_SUCCESS);	\
		else					\
			_ret = (_cn ? NET_XMIT_DROP : -EPERM);		\
		_ret;					\
	})

974 975 976 977 978 979
#define BPF_PROG_RUN_ARRAY(array, ctx, func)		\
	__BPF_PROG_RUN_ARRAY(array, ctx, func, false)

#define BPF_PROG_RUN_ARRAY_CHECK(array, ctx, func)	\
	__BPF_PROG_RUN_ARRAY(array, ctx, func, true)

980
#ifdef CONFIG_BPF_SYSCALL
981 982
DECLARE_PER_CPU(int, bpf_prog_active);

983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012
/*
 * Block execution of BPF programs attached to instrumentation (perf,
 * kprobes, tracepoints) to prevent deadlocks on map operations as any of
 * these events can happen inside a region which holds a map bucket lock
 * and can deadlock on it.
 *
 * Use the preemption safe inc/dec variants on RT because migrate disable
 * is preemptible on RT and preemption in the middle of the RMW operation
 * might lead to inconsistent state. Use the raw variants for non RT
 * kernels as migrate_disable() maps to preempt_disable() so the slightly
 * more expensive save operation can be avoided.
 */
static inline void bpf_disable_instrumentation(void)
{
	migrate_disable();
	if (IS_ENABLED(CONFIG_PREEMPT_RT))
		this_cpu_inc(bpf_prog_active);
	else
		__this_cpu_inc(bpf_prog_active);
}

static inline void bpf_enable_instrumentation(void)
{
	if (IS_ENABLED(CONFIG_PREEMPT_RT))
		this_cpu_dec(bpf_prog_active);
	else
		__this_cpu_dec(bpf_prog_active);
	migrate_enable();
}

1013 1014 1015
extern const struct file_operations bpf_map_fops;
extern const struct file_operations bpf_prog_fops;

A
Alexei Starovoitov 已提交
1016
#define BPF_PROG_TYPE(_id, _name, prog_ctx_type, kern_ctx_type) \
1017 1018
	extern const struct bpf_prog_ops _name ## _prog_ops; \
	extern const struct bpf_verifier_ops _name ## _verifier_ops;
1019 1020
#define BPF_MAP_TYPE(_id, _ops) \
	extern const struct bpf_map_ops _ops;
1021 1022
#include <linux/bpf_types.h>
#undef BPF_PROG_TYPE
1023
#undef BPF_MAP_TYPE
1024

1025
extern const struct bpf_prog_ops bpf_offload_prog_ops;
1026 1027 1028
extern const struct bpf_verifier_ops tc_cls_act_analyzer_ops;
extern const struct bpf_verifier_ops xdp_analyzer_ops;

1029
struct bpf_prog *bpf_prog_get(u32 ufd);
1030
struct bpf_prog *bpf_prog_get_type_dev(u32 ufd, enum bpf_prog_type type,
1031
				       bool attach_drv);
1032
void bpf_prog_add(struct bpf_prog *prog, int i);
1033
void bpf_prog_sub(struct bpf_prog *prog, int i);
1034
void bpf_prog_inc(struct bpf_prog *prog);
1035
struct bpf_prog * __must_check bpf_prog_inc_not_zero(struct bpf_prog *prog);
1036
void bpf_prog_put(struct bpf_prog *prog);
1037 1038
int __bpf_prog_charge(struct user_struct *user, u32 pages);
void __bpf_prog_uncharge(struct user_struct *user, u32 pages);
1039 1040
void __bpf_free_used_maps(struct bpf_prog_aux *aux,
			  struct bpf_map **used_maps, u32 len);
1041

1042
void bpf_prog_free_id(struct bpf_prog *prog, bool do_idr_lock);
1043
void bpf_map_free_id(struct bpf_map *map, bool do_idr_lock);
1044

1045
struct bpf_map *bpf_map_get(u32 ufd);
1046
struct bpf_map *bpf_map_get_with_uref(u32 ufd);
1047
struct bpf_map *__bpf_map_get(struct fd f);
1048 1049 1050
void bpf_map_inc(struct bpf_map *map);
void bpf_map_inc_with_uref(struct bpf_map *map);
struct bpf_map * __must_check bpf_map_inc_not_zero(struct bpf_map *map);
1051
void bpf_map_put_with_uref(struct bpf_map *map);
1052
void bpf_map_put(struct bpf_map *map);
1053 1054
int bpf_map_charge_memlock(struct bpf_map *map, u32 pages);
void bpf_map_uncharge_memlock(struct bpf_map *map, u32 pages);
1055
int bpf_map_charge_init(struct bpf_map_memory *mem, u64 size);
1056 1057 1058
void bpf_map_charge_finish(struct bpf_map_memory *mem);
void bpf_map_charge_move(struct bpf_map_memory *dst,
			 struct bpf_map_memory *src);
1059 1060
void *bpf_map_area_alloc(u64 size, int numa_node);
void *bpf_map_area_mmapable_alloc(u64 size, int numa_node);
1061
void bpf_map_area_free(void *base);
1062
void bpf_map_init_from_attr(struct bpf_map *map, union bpf_attr *attr);
1063 1064
int  generic_map_lookup_batch(struct bpf_map *map,
			      const union bpf_attr *attr,
1065 1066 1067 1068 1069 1070
			      union bpf_attr __user *uattr);
int  generic_map_update_batch(struct bpf_map *map,
			      const union bpf_attr *attr,
			      union bpf_attr __user *uattr);
int  generic_map_delete_batch(struct bpf_map *map,
			      const union bpf_attr *attr,
1071
			      union bpf_attr __user *uattr);
1072

1073 1074
extern int sysctl_unprivileged_bpf_disabled;

1075
int bpf_map_new_fd(struct bpf_map *map, int flags);
1076 1077
int bpf_prog_new_fd(struct bpf_prog *prog);

1078 1079 1080 1081
struct bpf_link;

struct bpf_link_ops {
	void (*release)(struct bpf_link *link);
1082
	void (*dealloc)(struct bpf_link *link);
1083 1084 1085 1086 1087 1088 1089
};

void bpf_link_init(struct bpf_link *link, const struct bpf_link_ops *ops,
		   struct bpf_prog *prog);
void bpf_link_inc(struct bpf_link *link);
void bpf_link_put(struct bpf_link *link);
int bpf_link_new_fd(struct bpf_link *link);
1090
struct file *bpf_link_new_file(struct bpf_link *link, int *reserved_fd);
1091 1092
struct bpf_link *bpf_link_get_from_fd(u32 ufd);

1093
int bpf_obj_pin_user(u32 ufd, const char __user *pathname);
1094
int bpf_obj_get_user(const char __user *pathname, int flags);
1095

1096 1097 1098 1099 1100 1101
int bpf_percpu_hash_copy(struct bpf_map *map, void *key, void *value);
int bpf_percpu_array_copy(struct bpf_map *map, void *key, void *value);
int bpf_percpu_hash_update(struct bpf_map *map, void *key, void *value,
			   u64 flags);
int bpf_percpu_array_update(struct bpf_map *map, void *key, void *value,
			    u64 flags);
1102

1103
int bpf_stackmap_copy(struct bpf_map *map, void *key, void *value);
1104

1105 1106
int bpf_fd_array_map_update_elem(struct bpf_map *map, struct file *map_file,
				 void *key, void *value, u64 map_flags);
1107
int bpf_fd_array_map_lookup_elem(struct bpf_map *map, void *key, u32 *value);
M
Martin KaFai Lau 已提交
1108 1109
int bpf_fd_htab_map_update_elem(struct bpf_map *map, struct file *map_file,
				void *key, void *value, u64 map_flags);
1110
int bpf_fd_htab_map_lookup_elem(struct bpf_map *map, void *key, u32 *value);
1111

1112
int bpf_get_file_flag(int flags);
1113 1114
int bpf_check_uarg_tail_zero(void __user *uaddr, size_t expected_size,
			     size_t actual_size);
1115

1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131
/* memcpy that is used with 8-byte aligned pointers, power-of-8 size and
 * forced to use 'long' read/writes to try to atomically copy long counters.
 * Best-effort only.  No barriers here, since it _will_ race with concurrent
 * updates from BPF programs. Called from bpf syscall and mostly used with
 * size 8 or 16 bytes, so ask compiler to inline it.
 */
static inline void bpf_long_memcpy(void *dst, const void *src, u32 size)
{
	const long *lsrc = src;
	long *ldst = dst;

	size /= sizeof(long);
	while (size--)
		*ldst++ = *lsrc++;
}

1132
/* verify correctness of eBPF program */
Y
Yonghong Song 已提交
1133 1134
int bpf_check(struct bpf_prog **fp, union bpf_attr *attr,
	      union bpf_attr __user *uattr);
1135
void bpf_patch_call_args(struct bpf_insn *insn, u32 stack_depth);
1136 1137

/* Map specifics */
1138
struct xdp_buff;
1139
struct sk_buff;
1140 1141

struct bpf_dtab_netdev *__dev_map_lookup_elem(struct bpf_map *map, u32 key);
1142
struct bpf_dtab_netdev *__dev_map_hash_lookup_elem(struct bpf_map *map, u32 key);
1143 1144 1145
void __dev_flush(void);
int dev_xdp_enqueue(struct net_device *dev, struct xdp_buff *xdp,
		    struct net_device *dev_rx);
1146 1147
int dev_map_enqueue(struct bpf_dtab_netdev *dst, struct xdp_buff *xdp,
		    struct net_device *dev_rx);
1148 1149
int dev_map_generic_redirect(struct bpf_dtab_netdev *dst, struct sk_buff *skb,
			     struct bpf_prog *xdp_prog);
1150

1151
struct bpf_cpu_map_entry *__cpu_map_lookup_elem(struct bpf_map *map, u32 key);
1152
void __cpu_map_flush(void);
1153 1154 1155
int cpu_map_enqueue(struct bpf_cpu_map_entry *rcpu, struct xdp_buff *xdp,
		    struct net_device *dev_rx);

1156 1157 1158 1159 1160 1161 1162
/* Return map's numa specified by userspace */
static inline int bpf_map_attr_numa_node(const union bpf_attr *attr)
{
	return (attr->map_flags & BPF_F_NUMA_NODE) ?
		attr->numa_node : NUMA_NO_NODE;
}

1163
struct bpf_prog *bpf_prog_get_type_path(const char *name, enum bpf_prog_type type);
1164
int array_map_alloc_check(union bpf_attr *attr);
1165

1166 1167 1168 1169
int bpf_prog_test_run_xdp(struct bpf_prog *prog, const union bpf_attr *kattr,
			  union bpf_attr __user *uattr);
int bpf_prog_test_run_skb(struct bpf_prog *prog, const union bpf_attr *kattr,
			  union bpf_attr __user *uattr);
1170 1171 1172
int bpf_prog_test_run_tracing(struct bpf_prog *prog,
			      const union bpf_attr *kattr,
			      union bpf_attr __user *uattr);
1173 1174 1175
int bpf_prog_test_run_flow_dissector(struct bpf_prog *prog,
				     const union bpf_attr *kattr,
				     union bpf_attr __user *uattr);
1176 1177 1178 1179 1180 1181 1182
bool btf_ctx_access(int off, int size, enum bpf_access_type type,
		    const struct bpf_prog *prog,
		    struct bpf_insn_access_aux *info);
int btf_struct_access(struct bpf_verifier_log *log,
		      const struct btf_type *t, int off, int size,
		      enum bpf_access_type atype,
		      u32 *next_btf_id);
1183 1184
int btf_resolve_helper_id(struct bpf_verifier_log *log,
			  const struct bpf_func_proto *fn, int);
1185

A
Alexei Starovoitov 已提交
1186 1187 1188 1189 1190 1191
int btf_distill_func_proto(struct bpf_verifier_log *log,
			   struct btf *btf,
			   const struct btf_type *func_proto,
			   const char *func_name,
			   struct btf_func_model *m);

1192 1193 1194 1195 1196
struct bpf_reg_state;
int btf_check_func_arg_match(struct bpf_verifier_env *env, int subprog,
			     struct bpf_reg_state *regs);
int btf_prepare_func_args(struct bpf_verifier_env *env, int subprog,
			  struct bpf_reg_state *reg);
1197 1198
int btf_check_type_match(struct bpf_verifier_env *env, struct bpf_prog *prog,
			 struct btf *btf, const struct btf_type *t);
1199

1200 1201
struct bpf_prog *bpf_prog_by_id(u32 id);

1202
#else /* !CONFIG_BPF_SYSCALL */
1203 1204 1205 1206 1207
static inline struct bpf_prog *bpf_prog_get(u32 ufd)
{
	return ERR_PTR(-EOPNOTSUPP);
}

1208 1209
static inline struct bpf_prog *bpf_prog_get_type_dev(u32 ufd,
						     enum bpf_prog_type type,
1210
						     bool attach_drv)
1211 1212 1213 1214
{
	return ERR_PTR(-EOPNOTSUPP);
}

1215
static inline void bpf_prog_add(struct bpf_prog *prog, int i)
1216 1217
{
}
1218

1219 1220 1221 1222
static inline void bpf_prog_sub(struct bpf_prog *prog, int i)
{
}

1223 1224 1225
static inline void bpf_prog_put(struct bpf_prog *prog)
{
}
1226

1227
static inline void bpf_prog_inc(struct bpf_prog *prog)
1228 1229
{
}
1230

1231 1232 1233 1234 1235 1236
static inline struct bpf_prog *__must_check
bpf_prog_inc_not_zero(struct bpf_prog *prog)
{
	return ERR_PTR(-EOPNOTSUPP);
}

1237 1238 1239 1240 1241 1242 1243 1244
static inline int __bpf_prog_charge(struct user_struct *user, u32 pages)
{
	return 0;
}

static inline void __bpf_prog_uncharge(struct user_struct *user, u32 pages)
{
}
1245

1246
static inline int bpf_obj_get_user(const char __user *pathname, int flags)
1247 1248 1249 1250
{
	return -EOPNOTSUPP;
}

1251 1252 1253 1254 1255 1256
static inline struct net_device  *__dev_map_lookup_elem(struct bpf_map *map,
						       u32 key)
{
	return NULL;
}

1257 1258 1259 1260 1261 1262
static inline struct net_device  *__dev_map_hash_lookup_elem(struct bpf_map *map,
							     u32 key)
{
	return NULL;
}

1263
static inline void __dev_flush(void)
1264 1265
{
}
1266

1267 1268 1269
struct xdp_buff;
struct bpf_dtab_netdev;

1270 1271 1272 1273 1274 1275 1276
static inline
int dev_xdp_enqueue(struct net_device *dev, struct xdp_buff *xdp,
		    struct net_device *dev_rx)
{
	return 0;
}

1277
static inline
1278 1279
int dev_map_enqueue(struct bpf_dtab_netdev *dst, struct xdp_buff *xdp,
		    struct net_device *dev_rx)
1280 1281 1282 1283
{
	return 0;
}

1284 1285 1286 1287 1288 1289 1290 1291 1292
struct sk_buff;

static inline int dev_map_generic_redirect(struct bpf_dtab_netdev *dst,
					   struct sk_buff *skb,
					   struct bpf_prog *xdp_prog)
{
	return 0;
}

1293 1294 1295 1296 1297 1298
static inline
struct bpf_cpu_map_entry *__cpu_map_lookup_elem(struct bpf_map *map, u32 key)
{
	return NULL;
}

1299
static inline void __cpu_map_flush(void)
1300 1301 1302 1303 1304 1305 1306 1307 1308
{
}

static inline int cpu_map_enqueue(struct bpf_cpu_map_entry *rcpu,
				  struct xdp_buff *xdp,
				  struct net_device *dev_rx)
{
	return 0;
}
1309 1310 1311 1312 1313 1314

static inline struct bpf_prog *bpf_prog_get_type_path(const char *name,
				enum bpf_prog_type type)
{
	return ERR_PTR(-EOPNOTSUPP);
}
1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329

static inline int bpf_prog_test_run_xdp(struct bpf_prog *prog,
					const union bpf_attr *kattr,
					union bpf_attr __user *uattr)
{
	return -ENOTSUPP;
}

static inline int bpf_prog_test_run_skb(struct bpf_prog *prog,
					const union bpf_attr *kattr,
					union bpf_attr __user *uattr)
{
	return -ENOTSUPP;
}

1330 1331 1332 1333 1334 1335 1336
static inline int bpf_prog_test_run_tracing(struct bpf_prog *prog,
					    const union bpf_attr *kattr,
					    union bpf_attr __user *uattr)
{
	return -ENOTSUPP;
}

1337 1338 1339 1340 1341 1342
static inline int bpf_prog_test_run_flow_dissector(struct bpf_prog *prog,
						   const union bpf_attr *kattr,
						   union bpf_attr __user *uattr)
{
	return -ENOTSUPP;
}
1343 1344 1345 1346

static inline void bpf_map_put(struct bpf_map *map)
{
}
1347 1348 1349 1350 1351

static inline struct bpf_prog *bpf_prog_by_id(u32 id)
{
	return ERR_PTR(-ENOTSUPP);
}
1352
#endif /* CONFIG_BPF_SYSCALL */
1353

1354 1355 1356 1357 1358 1359
static inline struct bpf_prog *bpf_prog_get_type(u32 ufd,
						 enum bpf_prog_type type)
{
	return bpf_prog_get_type_dev(ufd, type, false);
}

1360 1361
bool bpf_prog_get_ok(struct bpf_prog *, enum bpf_prog_type *, bool);

1362 1363
int bpf_prog_offload_compile(struct bpf_prog *prog);
void bpf_prog_offload_destroy(struct bpf_prog *prog);
1364 1365
int bpf_prog_offload_info_fill(struct bpf_prog_info *info,
			       struct bpf_prog *prog);
1366

1367 1368
int bpf_map_offload_info_fill(struct bpf_map_info *info, struct bpf_map *map);

1369 1370 1371 1372 1373 1374 1375
int bpf_map_offload_lookup_elem(struct bpf_map *map, void *key, void *value);
int bpf_map_offload_update_elem(struct bpf_map *map,
				void *key, void *value, u64 flags);
int bpf_map_offload_delete_elem(struct bpf_map *map, void *key);
int bpf_map_offload_get_next_key(struct bpf_map *map,
				 void *key, void *next_key);

1376
bool bpf_offload_prog_map_match(struct bpf_prog *prog, struct bpf_map *map);
1377

1378
struct bpf_offload_dev *
1379
bpf_offload_dev_create(const struct bpf_prog_offload_ops *ops, void *priv);
1380
void bpf_offload_dev_destroy(struct bpf_offload_dev *offdev);
1381
void *bpf_offload_dev_priv(struct bpf_offload_dev *offdev);
1382 1383 1384 1385
int bpf_offload_dev_netdev_register(struct bpf_offload_dev *offdev,
				    struct net_device *netdev);
void bpf_offload_dev_netdev_unregister(struct bpf_offload_dev *offdev,
				       struct net_device *netdev);
1386
bool bpf_offload_dev_match(struct bpf_prog *prog, struct net_device *netdev);
1387

1388 1389 1390
#if defined(CONFIG_NET) && defined(CONFIG_BPF_SYSCALL)
int bpf_prog_offload_init(struct bpf_prog *prog, union bpf_attr *attr);

1391
static inline bool bpf_prog_is_dev_bound(const struct bpf_prog_aux *aux)
1392
{
1393
	return aux->offload_requested;
1394
}
1395 1396 1397 1398 1399 1400 1401 1402

static inline bool bpf_map_is_dev_bound(struct bpf_map *map)
{
	return unlikely(map->ops == &bpf_map_offload_ops);
}

struct bpf_map *bpf_map_offload_map_alloc(union bpf_attr *attr);
void bpf_map_offload_map_free(struct bpf_map *map);
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413
#else
static inline int bpf_prog_offload_init(struct bpf_prog *prog,
					union bpf_attr *attr)
{
	return -EOPNOTSUPP;
}

static inline bool bpf_prog_is_dev_bound(struct bpf_prog_aux *aux)
{
	return false;
}
1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427

static inline bool bpf_map_is_dev_bound(struct bpf_map *map)
{
	return false;
}

static inline struct bpf_map *bpf_map_offload_map_alloc(union bpf_attr *attr)
{
	return ERR_PTR(-EOPNOTSUPP);
}

static inline void bpf_map_offload_map_free(struct bpf_map *map)
{
}
1428 1429
#endif /* CONFIG_NET && CONFIG_BPF_SYSCALL */

1430 1431 1432
#if defined(CONFIG_BPF_STREAM_PARSER)
int sock_map_prog_update(struct bpf_map *map, struct bpf_prog *prog, u32 which);
int sock_map_get_from_fd(const union bpf_attr *attr, struct bpf_prog *prog);
1433 1434
void sock_map_unhash(struct sock *sk);
void sock_map_close(struct sock *sk, long timeout);
1435
#else
1436 1437
static inline int sock_map_prog_update(struct bpf_map *map,
				       struct bpf_prog *prog, u32 which)
1438 1439 1440
{
	return -EOPNOTSUPP;
}
1441

1442 1443
static inline int sock_map_get_from_fd(const union bpf_attr *attr,
				       struct bpf_prog *prog)
1444 1445 1446
{
	return -EINVAL;
}
1447
#endif /* CONFIG_BPF_STREAM_PARSER */
1448

1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475
#if defined(CONFIG_INET) && defined(CONFIG_BPF_SYSCALL)
void bpf_sk_reuseport_detach(struct sock *sk);
int bpf_fd_reuseport_array_lookup_elem(struct bpf_map *map, void *key,
				       void *value);
int bpf_fd_reuseport_array_update_elem(struct bpf_map *map, void *key,
				       void *value, u64 map_flags);
#else
static inline void bpf_sk_reuseport_detach(struct sock *sk)
{
}

#ifdef CONFIG_BPF_SYSCALL
static inline int bpf_fd_reuseport_array_lookup_elem(struct bpf_map *map,
						     void *key, void *value)
{
	return -EOPNOTSUPP;
}

static inline int bpf_fd_reuseport_array_update_elem(struct bpf_map *map,
						     void *key, void *value,
						     u64 map_flags)
{
	return -EOPNOTSUPP;
}
#endif /* CONFIG_BPF_SYSCALL */
#endif /* defined(CONFIG_INET) && defined(CONFIG_BPF_SYSCALL) */

1476
/* verifier prototypes for helper functions called from eBPF programs */
1477 1478 1479
extern const struct bpf_func_proto bpf_map_lookup_elem_proto;
extern const struct bpf_func_proto bpf_map_update_elem_proto;
extern const struct bpf_func_proto bpf_map_delete_elem_proto;
M
Mauricio Vasquez B 已提交
1480 1481 1482
extern const struct bpf_func_proto bpf_map_push_elem_proto;
extern const struct bpf_func_proto bpf_map_pop_elem_proto;
extern const struct bpf_func_proto bpf_map_peek_elem_proto;
1483

1484
extern const struct bpf_func_proto bpf_get_prandom_u32_proto;
1485
extern const struct bpf_func_proto bpf_get_smp_processor_id_proto;
1486
extern const struct bpf_func_proto bpf_get_numa_node_id_proto;
1487
extern const struct bpf_func_proto bpf_tail_call_proto;
1488
extern const struct bpf_func_proto bpf_ktime_get_ns_proto;
1489 1490 1491
extern const struct bpf_func_proto bpf_get_current_pid_tgid_proto;
extern const struct bpf_func_proto bpf_get_current_uid_gid_proto;
extern const struct bpf_func_proto bpf_get_current_comm_proto;
1492
extern const struct bpf_func_proto bpf_get_stackid_proto;
Y
Yonghong Song 已提交
1493
extern const struct bpf_func_proto bpf_get_stack_proto;
1494
extern const struct bpf_func_proto bpf_sock_map_update_proto;
1495
extern const struct bpf_func_proto bpf_sock_hash_update_proto;
1496
extern const struct bpf_func_proto bpf_get_current_cgroup_id_proto;
1497 1498 1499 1500
extern const struct bpf_func_proto bpf_msg_redirect_hash_proto;
extern const struct bpf_func_proto bpf_msg_redirect_map_proto;
extern const struct bpf_func_proto bpf_sk_redirect_hash_proto;
extern const struct bpf_func_proto bpf_sk_redirect_map_proto;
1501 1502
extern const struct bpf_func_proto bpf_spin_lock_proto;
extern const struct bpf_func_proto bpf_spin_unlock_proto;
1503
extern const struct bpf_func_proto bpf_get_local_storage_proto;
1504 1505
extern const struct bpf_func_proto bpf_strtol_proto;
extern const struct bpf_func_proto bpf_strtoul_proto;
1506
extern const struct bpf_func_proto bpf_tcp_sock_proto;
M
Martin KaFai Lau 已提交
1507
extern const struct bpf_func_proto bpf_jiffies64_proto;
1508
extern const struct bpf_func_proto bpf_get_ns_current_pid_tgid_proto;
1509

1510 1511 1512 1513
/* Shared helpers among cBPF and eBPF. */
void bpf_user_rnd_init_once(void);
u64 bpf_user_rnd_u32(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5);

1514
#if defined(CONFIG_NET)
1515 1516 1517
bool bpf_sock_common_is_valid_access(int off, int size,
				     enum bpf_access_type type,
				     struct bpf_insn_access_aux *info);
1518 1519 1520 1521 1522 1523 1524 1525
bool bpf_sock_is_valid_access(int off, int size, enum bpf_access_type type,
			      struct bpf_insn_access_aux *info);
u32 bpf_sock_convert_ctx_access(enum bpf_access_type type,
				const struct bpf_insn *si,
				struct bpf_insn *insn_buf,
				struct bpf_prog *prog,
				u32 *target_size);
#else
1526 1527 1528 1529 1530 1531
static inline bool bpf_sock_common_is_valid_access(int off, int size,
						   enum bpf_access_type type,
						   struct bpf_insn_access_aux *info)
{
	return false;
}
1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547
static inline bool bpf_sock_is_valid_access(int off, int size,
					    enum bpf_access_type type,
					    struct bpf_insn_access_aux *info)
{
	return false;
}
static inline u32 bpf_sock_convert_ctx_access(enum bpf_access_type type,
					      const struct bpf_insn *si,
					      struct bpf_insn *insn_buf,
					      struct bpf_prog *prog,
					      u32 *target_size)
{
	return 0;
}
#endif

1548
#ifdef CONFIG_INET
A
Alexei Starovoitov 已提交
1549 1550 1551 1552 1553 1554 1555 1556 1557
struct sk_reuseport_kern {
	struct sk_buff *skb;
	struct sock *sk;
	struct sock *selected_sk;
	void *data_end;
	u32 hash;
	u32 reuseport_id;
	bool bind_inany;
};
1558 1559 1560 1561 1562 1563 1564 1565
bool bpf_tcp_sock_is_valid_access(int off, int size, enum bpf_access_type type,
				  struct bpf_insn_access_aux *info);

u32 bpf_tcp_sock_convert_ctx_access(enum bpf_access_type type,
				    const struct bpf_insn *si,
				    struct bpf_insn *insn_buf,
				    struct bpf_prog *prog,
				    u32 *target_size);
1566 1567 1568 1569 1570 1571 1572 1573 1574

bool bpf_xdp_sock_is_valid_access(int off, int size, enum bpf_access_type type,
				  struct bpf_insn_access_aux *info);

u32 bpf_xdp_sock_convert_ctx_access(enum bpf_access_type type,
				    const struct bpf_insn *si,
				    struct bpf_insn *insn_buf,
				    struct bpf_prog *prog,
				    u32 *target_size);
1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590
#else
static inline bool bpf_tcp_sock_is_valid_access(int off, int size,
						enum bpf_access_type type,
						struct bpf_insn_access_aux *info)
{
	return false;
}

static inline u32 bpf_tcp_sock_convert_ctx_access(enum bpf_access_type type,
						  const struct bpf_insn *si,
						  struct bpf_insn *insn_buf,
						  struct bpf_prog *prog,
						  u32 *target_size)
{
	return 0;
}
1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605
static inline bool bpf_xdp_sock_is_valid_access(int off, int size,
						enum bpf_access_type type,
						struct bpf_insn_access_aux *info)
{
	return false;
}

static inline u32 bpf_xdp_sock_convert_ctx_access(enum bpf_access_type type,
						  const struct bpf_insn *si,
						  struct bpf_insn *insn_buf,
						  struct bpf_prog *prog,
						  u32 *target_size)
{
	return 0;
}
1606 1607
#endif /* CONFIG_INET */

1608
enum bpf_text_poke_type {
1609 1610
	BPF_MOD_CALL,
	BPF_MOD_JUMP,
1611
};
1612

1613 1614 1615
int bpf_arch_text_poke(void *ip, enum bpf_text_poke_type t,
		       void *addr1, void *addr2);

1616
#endif /* _LINUX_BPF_H */