[linux-block.git] / include / linux / bpf_mem_alloc.h

/* SPDX-License-Identifier: GPL-2.0-only */
/* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
#ifndef _BPF_MEM_ALLOC_H
#define _BPF_MEM_ALLOC_H
#include <linux/compiler_types.h>
#include <linux/workqueue.h>

struct bpf_mem_cache;
struct bpf_mem_caches;

struct bpf_mem_alloc {
	struct bpf_mem_caches __percpu *caches;
	struct bpf_mem_cache __percpu *cache;
	struct work_struct work;
};

int bpf_mem_alloc_init(struct bpf_mem_alloc *ma, int size, bool percpu);
void bpf_mem_alloc_destroy(struct bpf_mem_alloc *ma);

/* kmalloc/kfree equivalent: */
void *bpf_mem_alloc(struct bpf_mem_alloc *ma, size_t size);
void bpf_mem_free(struct bpf_mem_alloc *ma, void *ptr);

/* kmem_cache_alloc/free equivalent: */
void *bpf_mem_cache_alloc(struct bpf_mem_alloc *ma);
void bpf_mem_cache_free(struct bpf_mem_alloc *ma, void *ptr);

#endif /* _BPF_MEM_ALLOC_H */
Commit	Line	Data
7c8199e2 AS	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>
9f2c6e96	6	#include <linux/workqueue.h>
7c8199e2 AS	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;
9f2c6e96	14	struct work_struct work;
7c8199e2 AS	15	};
7c8199e2 AS	16
4ab67149	17	int bpf_mem_alloc_init(struct bpf_mem_alloc *ma, int size, bool percpu);
7c8199e2 AS	18	void bpf_mem_alloc_destroy(struct bpf_mem_alloc *ma);
	19
	20	/* kmalloc/kfree equivalent: */
	21	void bpf_mem_alloc(struct bpf_mem_alloc ma, size_t size);
	22	void bpf_mem_free(struct bpf_mem_alloc ma, void ptr);
	23
	24	/* kmem_cache_alloc/free equivalent: */
	25	void bpf_mem_cache_alloc(struct bpf_mem_alloc ma);
	26	void bpf_mem_cache_free(struct bpf_mem_alloc ma, void ptr);
	27
	28	#endif /* _BPF_MEM_ALLOC_H */