drivers/infiniband/core/cgroup.c

   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);