1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com>
4*4882a593Smuzhiyun */
5*4882a593Smuzhiyun
6*4882a593Smuzhiyun #include "core_priv.h"
7*4882a593Smuzhiyun
8*4882a593Smuzhiyun /**
9*4882a593Smuzhiyun * ib_device_register_rdmacg - register with rdma cgroup.
10*4882a593Smuzhiyun * @device: device to register to participate in resource
11*4882a593Smuzhiyun * accounting by rdma cgroup.
12*4882a593Smuzhiyun *
13*4882a593Smuzhiyun * Register with the rdma cgroup. Should be called before
14*4882a593Smuzhiyun * exposing rdma device to user space applications to avoid
15*4882a593Smuzhiyun * resource accounting leak.
16*4882a593Smuzhiyun */
ib_device_register_rdmacg(struct ib_device * device)17*4882a593Smuzhiyun void ib_device_register_rdmacg(struct ib_device *device)
18*4882a593Smuzhiyun {
19*4882a593Smuzhiyun device->cg_device.name = device->name;
20*4882a593Smuzhiyun rdmacg_register_device(&device->cg_device);
21*4882a593Smuzhiyun }
22*4882a593Smuzhiyun
23*4882a593Smuzhiyun /**
24*4882a593Smuzhiyun * ib_device_unregister_rdmacg - unregister with rdma cgroup.
25*4882a593Smuzhiyun * @device: device to unregister.
26*4882a593Smuzhiyun *
27*4882a593Smuzhiyun * Unregister with the rdma cgroup. Should be called after
28*4882a593Smuzhiyun * all the resources are deallocated, and after a stage when any
29*4882a593Smuzhiyun * other resource allocation by user application cannot be done
30*4882a593Smuzhiyun * for this device to avoid any leak in accounting.
31*4882a593Smuzhiyun */
ib_device_unregister_rdmacg(struct ib_device * device)32*4882a593Smuzhiyun void ib_device_unregister_rdmacg(struct ib_device *device)
33*4882a593Smuzhiyun {
34*4882a593Smuzhiyun rdmacg_unregister_device(&device->cg_device);
35*4882a593Smuzhiyun }
36*4882a593Smuzhiyun
ib_rdmacg_try_charge(struct ib_rdmacg_object * cg_obj,struct ib_device * device,enum rdmacg_resource_type resource_index)37*4882a593Smuzhiyun int ib_rdmacg_try_charge(struct ib_rdmacg_object *cg_obj,
38*4882a593Smuzhiyun struct ib_device *device,
39*4882a593Smuzhiyun enum rdmacg_resource_type resource_index)
40*4882a593Smuzhiyun {
41*4882a593Smuzhiyun return rdmacg_try_charge(&cg_obj->cg, &device->cg_device,
42*4882a593Smuzhiyun resource_index);
43*4882a593Smuzhiyun }
44*4882a593Smuzhiyun EXPORT_SYMBOL(ib_rdmacg_try_charge);
45*4882a593Smuzhiyun
ib_rdmacg_uncharge(struct ib_rdmacg_object * cg_obj,struct ib_device * device,enum rdmacg_resource_type resource_index)46*4882a593Smuzhiyun void ib_rdmacg_uncharge(struct ib_rdmacg_object *cg_obj,
47*4882a593Smuzhiyun struct ib_device *device,
48*4882a593Smuzhiyun enum rdmacg_resource_type resource_index)
49*4882a593Smuzhiyun {
50*4882a593Smuzhiyun rdmacg_uncharge(cg_obj->cg, &device->cg_device,
51*4882a593Smuzhiyun resource_index);
52*4882a593Smuzhiyun }
53*4882a593Smuzhiyun EXPORT_SYMBOL(ib_rdmacg_uncharge);
54