]>
Commit | Line | Data |
---|---|---|
2025cf9e | 1 | // SPDX-License-Identifier: GPL-2.0-only |
43579b5f PP |
2 | /* |
3 | * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com> | |
43579b5f PP |
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. | |
43579b5f | 16 | */ |
7527a7b1 | 17 | void ib_device_register_rdmacg(struct ib_device *device) |
43579b5f PP |
18 | { |
19 | device->cg_device.name = device->name; | |
7527a7b1 | 20 | rdmacg_register_device(&device->cg_device); |
43579b5f PP |
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 | */ | |
32 | void ib_device_unregister_rdmacg(struct ib_device *device) | |
33 | { | |
34 | rdmacg_unregister_device(&device->cg_device); | |
35 | } | |
36 | ||
37 | int 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 | } | |
44 | EXPORT_SYMBOL(ib_rdmacg_try_charge); | |
45 | ||
46 | void 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 | } | |
53 | EXPORT_SYMBOL(ib_rdmacg_uncharge); |