1 /* SPDX-License-Identifier: GPL-2.0-only */
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3 #ifndef _BPF_MEM_ALLOC_H
4 #define _BPF_MEM_ALLOC_H
5 #include <linux/compiler_types.h>
6 #include <linux/workqueue.h>
7 
8 struct bpf_mem_cache;
9 struct bpf_mem_caches;
10 
11 struct bpf_mem_alloc {
12 	struct bpf_mem_caches __percpu *caches;
13 	struct bpf_mem_cache __percpu *cache;
14 	bool percpu;
15 	struct work_struct work;
16 };
17 
18 /* 'size != 0' is for bpf_mem_alloc which manages fixed-size objects.
19  * Alloc and free are done with bpf_mem_cache_{alloc,free}().
20  *
21  * 'size = 0' is for bpf_mem_alloc which manages many fixed-size objects.
22  * Alloc and free are done with bpf_mem_{alloc,free}() and the size of
23  * the returned object is given by the size argument of bpf_mem_alloc().
24  */
25 int bpf_mem_alloc_init(struct bpf_mem_alloc *ma, int size, bool percpu);
26 void bpf_mem_alloc_destroy(struct bpf_mem_alloc *ma);
27 
28 /* kmalloc/kfree equivalent: */
29 void *bpf_mem_alloc(struct bpf_mem_alloc *ma, size_t size);
30 void bpf_mem_free(struct bpf_mem_alloc *ma, void *ptr);
31 void bpf_mem_free_rcu(struct bpf_mem_alloc *ma, void *ptr);
32 
33 /* kmem_cache_alloc/free equivalent: */
34 void *bpf_mem_cache_alloc(struct bpf_mem_alloc *ma);
35 void bpf_mem_cache_free(struct bpf_mem_alloc *ma, void *ptr);
36 void bpf_mem_cache_free_rcu(struct bpf_mem_alloc *ma, void *ptr);
37 void bpf_mem_cache_raw_free(void *ptr);
38 void *bpf_mem_cache_alloc_flags(struct bpf_mem_alloc *ma, gfp_t flags);
39 
40 #endif /* _BPF_MEM_ALLOC_H */
41