54 lines
1.5 KiB
C
54 lines
1.5 KiB
C
|
// SPDX-License-Identifier: GPL-2.0-only
|
||
|
|
/*
|
||
|
|
* Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com>
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "core_priv.h"
|
||
|
|
|
||
|
|
/**
|
||
|
|
* ib_device_register_rdmacg - register with rdma cgroup.
|
||
|
|
* @device: device to register to participate in resource
|
||
|
|
* accounting by rdma cgroup.
|
||
|
|
*
|
||
|
|
* Register with the rdma cgroup. Should be called before
|
||
|
|
* exposing rdma device to user space applications to avoid
|
||
|
|
* resource accounting leak.
|
||
|
|
*/
|
||
|
|
void ib_device_register_rdmacg(struct ib_device *device)
|
||
|
|
{
|
||
|
|
device->cg_device.name = device->name;
|
||
|
|
rdmacg_register_device(&device->cg_device);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* ib_device_unregister_rdmacg - unregister with rdma cgroup.
|
||
|
|
* @device: device to unregister.
|
||
|
|
*
|
||
|
|
* Unregister with the rdma cgroup. Should be called after
|
||
|
|
* all the resources are deallocated, and after a stage when any
|
||
|
|
* other resource allocation by user application cannot be done
|
||
|
|
* for this device to avoid any leak in accounting.
|
||
|
|
*/
|
||
|
|
void ib_device_unregister_rdmacg(struct ib_device *device)
|
||
|
|
{
|
||
|
|
rdmacg_unregister_device(&device->cg_device);
|
||
|
|
}
|
||
|
|
|
||
|
|
int ib_rdmacg_try_charge(struct ib_rdmacg_object *cg_obj,
|
||
|
|
struct ib_device *device,
|
||
|
|
enum rdmacg_resource_type resource_index)
|
||
|
|
{
|
||
|
|
return rdmacg_try_charge(&cg_obj->cg, &device->cg_device,
|
||
|
|
resource_index);
|
||
|
|
}
|
||
|
|
EXPORT_SYMBOL(ib_rdmacg_try_charge);
|
||
|
|
|
||
|
|
void ib_rdmacg_uncharge(struct ib_rdmacg_object *cg_obj,
|
||
|
|
struct ib_device *device,
|
||
|
|
enum rdmacg_resource_type resource_index)
|
||
|
|
{
|
||
|
|
rdmacg_uncharge(cg_obj->cg, &device->cg_device,
|
||
|
|
resource_index);
|
||
|
|
}
|
||
|
|
EXPORT_SYMBOL(ib_rdmacg_uncharge);
|