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 7 struct bpf_mem_cache; 8 struct bpf_mem_caches; 9 10 struct bpf_mem_alloc { 11 struct bpf_mem_caches __percpu *caches; 12 struct bpf_mem_cache __percpu *cache; 13 }; 14 15 int bpf_mem_alloc_init(struct bpf_mem_alloc *ma, int size); 16 void bpf_mem_alloc_destroy(struct bpf_mem_alloc *ma); 17 18 /* kmalloc/kfree equivalent: */ 19 void *bpf_mem_alloc(struct bpf_mem_alloc *ma, size_t size); 20 void bpf_mem_free(struct bpf_mem_alloc *ma, void *ptr); 21 22 /* kmem_cache_alloc/free equivalent: */ 23 void *bpf_mem_cache_alloc(struct bpf_mem_alloc *ma); 24 void bpf_mem_cache_free(struct bpf_mem_alloc *ma, void *ptr); 25 26 #endif /* _BPF_MEM_ALLOC_H */ 27