LXR linux/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 */
  26int 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 */
  41void ib_device_unregister_rdmacg(struct ib_device *device)
  42{
  43        rdmacg_unregister_device(&device->cg_device);
  44}
  45
  46int 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}
  53EXPORT_SYMBOL(ib_rdmacg_try_charge);
  54
  55void 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}
  62EXPORT_SYMBOL(ib_rdmacg_uncharge);
  63