| 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 | |
|---|
| 11 | enum 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 | |
|---|
| 19 | struct 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 | |
|---|
| 29 | struct 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 | */ |
|---|
| 39 | void rdmacg_register_device(struct rdmacg_device *device); |
|---|
| 40 | void rdmacg_unregister_device(struct rdmacg_device *device); |
|---|
| 41 | |
|---|
| 42 | /* APIs for RDMA/IB stack to charge/uncharge pool specific resources */ |
|---|
| 43 | int rdmacg_try_charge(struct rdma_cgroup **rdmacg, |
|---|
| 44 | struct rdmacg_device *device, |
|---|
| 45 | enum rdmacg_resource_type index); |
|---|
| 46 | void 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 | |
|---|
