include/linux/cgroup_rdma.h

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2 /*
   3  * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com>
   4  */
   5
   6 #ifndef _CGROUP_RDMA_H
   7 #define _CGROUP_RDMA_H
   8
   9 #include <linux/cgroup.h>
  10
  11 enum rdmacg_resource_type {
  12         RDMACG_RESOURCE_HCA_HANDLE,
  13         RDMACG_RESOURCE_HCA_OBJECT,
  14         RDMACG_RESOURCE_MAX,
  15 };
  16
  17 #ifdef CONFIG_CGROUP_RDMA
  18
  19 struct rdma_cgroup {
  20         struct cgroup_subsys_state      css;
  21
  22         /*
  23          * head to keep track of all resource pools
  24          * that belongs to this cgroup.
  25          */
  26         struct list_head                rpools;
  27 };
  28
  29 struct rdmacg_device {
  30         struct list_head        dev_node;
  31         struct list_head        rpools;
  32         char                    *name;
  33 };
  34
  35 /*
  36  * APIs for RDMA/IB stack to publish when a device wants to
  37  * participate in resource accounting
  38  */
  39 void rdmacg_register_device(struct rdmacg_device *device);
  40 void rdmacg_unregister_device(struct rdmacg_device *device);
  41
  42 /* APIs for RDMA/IB stack to charge/uncharge pool specific resources */
  43 int rdmacg_try_charge(struct rdma_cgroup **rdmacg,
  44                       struct rdmacg_device *device,
  45                       enum rdmacg_resource_type index);
  46 void rdmacg_uncharge(struct rdma_cgroup *cg,
  47                      struct rdmacg_device *device,
  48                      enum rdmacg_resource_type index);
  49 #endif  /* CONFIG_CGROUP_RDMA */
  50 #endif  /* _CGROUP_RDMA_H */