1/* SPDX-License-Identifier: GPL-2.0-only */
2/*
3 * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com>
4 */
5
6#ifndef _CGROUP_RDMA_H
7#define _CGROUP_RDMA_H
8
9#include <linux/cgroup.h>
10
11enum rdmacg_resource_type {
12	RDMACG_RESOURCE_HCA_HANDLE,
13	RDMACG_RESOURCE_HCA_OBJECT,
14	RDMACG_RESOURCE_MAX,
15};
16
17#ifdef CONFIG_CGROUP_RDMA
18
19struct rdma_cgroup {
20	struct cgroup_subsys_state	css;
21
22	/*
23	 * head to keep track of all resource pools
24	 * that belongs to this cgroup.
25	 */
26	struct list_head		rpools;
27};
28
29struct rdmacg_device {
30	struct list_head	dev_node;
31	struct list_head	rpools;
32	char			*name;
33};
34
35/*
36 * APIs for RDMA/IB stack to publish when a device wants to
37 * participate in resource accounting
38 */
39void rdmacg_register_device(struct rdmacg_device *device);
40void rdmacg_unregister_device(struct rdmacg_device *device);
41
42/* APIs for RDMA/IB stack to charge/uncharge pool specific resources */
43int rdmacg_try_charge(struct rdma_cgroup **rdmacg,
44		      struct rdmacg_device *device,
45		      enum rdmacg_resource_type index);
46void rdmacg_uncharge(struct rdma_cgroup *cg,
47		     struct rdmacg_device *device,
48		     enum rdmacg_resource_type index);
49#endif	/* CONFIG_CGROUP_RDMA */
50#endif	/* _CGROUP_RDMA_H */
51