at v6.5 1.3 kB view raw
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 8struct bpf_mem_cache; 9struct bpf_mem_caches; 10 11struct bpf_mem_alloc { 12 struct bpf_mem_caches __percpu *caches; 13 struct bpf_mem_cache __percpu *cache; 14 struct work_struct work; 15}; 16 17/* 'size != 0' is for bpf_mem_alloc which manages fixed-size objects. 18 * Alloc and free are done with bpf_mem_cache_{alloc,free}(). 19 * 20 * 'size = 0' is for bpf_mem_alloc which manages many fixed-size objects. 21 * Alloc and free are done with bpf_mem_{alloc,free}() and the size of 22 * the returned object is given by the size argument of bpf_mem_alloc(). 23 */ 24int bpf_mem_alloc_init(struct bpf_mem_alloc *ma, int size, bool percpu); 25void bpf_mem_alloc_destroy(struct bpf_mem_alloc *ma); 26 27/* kmalloc/kfree equivalent: */ 28void *bpf_mem_alloc(struct bpf_mem_alloc *ma, size_t size); 29void bpf_mem_free(struct bpf_mem_alloc *ma, void *ptr); 30 31/* kmem_cache_alloc/free equivalent: */ 32void *bpf_mem_cache_alloc(struct bpf_mem_alloc *ma); 33void bpf_mem_cache_free(struct bpf_mem_alloc *ma, void *ptr); 34void bpf_mem_cache_raw_free(void *ptr); 35void *bpf_mem_cache_alloc_flags(struct bpf_mem_alloc *ma, gfp_t flags); 36 37#endif /* _BPF_MEM_ALLOC_H */