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