Parav Pandit | 43579b5 | 2017-01-10 00:02:14 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com> |
| 3 | * |
| 4 | * This program is free software; you can redistribute it and/or modify it |
| 5 | * under the terms and conditions of the GNU General Public License, |
| 6 | * version 2, as published by the Free Software Foundation. |
| 7 | * |
| 8 | * This program is distributed in the hope it will be useful, but WITHOUT |
| 9 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| 10 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for |
| 11 | * more details. |
| 12 | */ |
| 13 | |
| 14 | #include "core_priv.h" |
| 15 | |
| 16 | /** |
| 17 | * ib_device_register_rdmacg - register with rdma cgroup. |
| 18 | * @device: device to register to participate in resource |
| 19 | * accounting by rdma cgroup. |
| 20 | * |
| 21 | * Register with the rdma cgroup. Should be called before |
| 22 | * exposing rdma device to user space applications to avoid |
| 23 | * resource accounting leak. |
| 24 | * Returns 0 on success or otherwise failure code. |
| 25 | */ |
| 26 | int ib_device_register_rdmacg(struct ib_device *device) |
| 27 | { |
| 28 | device->cg_device.name = device->name; |
| 29 | return rdmacg_register_device(&device->cg_device); |
| 30 | } |
| 31 | |
| 32 | /** |
| 33 | * ib_device_unregister_rdmacg - unregister with rdma cgroup. |
| 34 | * @device: device to unregister. |
| 35 | * |
| 36 | * Unregister with the rdma cgroup. Should be called after |
| 37 | * all the resources are deallocated, and after a stage when any |
| 38 | * other resource allocation by user application cannot be done |
| 39 | * for this device to avoid any leak in accounting. |
| 40 | */ |
| 41 | void ib_device_unregister_rdmacg(struct ib_device *device) |
| 42 | { |
| 43 | rdmacg_unregister_device(&device->cg_device); |
| 44 | } |
| 45 | |
| 46 | int ib_rdmacg_try_charge(struct ib_rdmacg_object *cg_obj, |
| 47 | struct ib_device *device, |
| 48 | enum rdmacg_resource_type resource_index) |
| 49 | { |
| 50 | return rdmacg_try_charge(&cg_obj->cg, &device->cg_device, |
| 51 | resource_index); |
| 52 | } |
| 53 | EXPORT_SYMBOL(ib_rdmacg_try_charge); |
| 54 | |
| 55 | void ib_rdmacg_uncharge(struct ib_rdmacg_object *cg_obj, |
| 56 | struct ib_device *device, |
| 57 | enum rdmacg_resource_type resource_index) |
| 58 | { |
| 59 | rdmacg_uncharge(cg_obj->cg, &device->cg_device, |
| 60 | resource_index); |
| 61 | } |
| 62 | EXPORT_SYMBOL(ib_rdmacg_uncharge); |