/* SPDX-License-Identifier: GPL-2.0-only */
  /*
   * Copyright (C) 2016 Parav Pandit <pandit.parav@gmail.com>
   */
  
  #ifndef _CGROUP_RDMA_H
  #define _CGROUP_RDMA_H
  
  #include <linux/cgroup.h>
  
  enum rdmacg_resource_type {
  	RDMACG_RESOURCE_HCA_HANDLE,
  	RDMACG_RESOURCE_HCA_OBJECT,
  	RDMACG_RESOURCE_MAX,
  };
  
  #ifdef CONFIG_CGROUP_RDMA
  
  struct rdma_cgroup {
  	struct cgroup_subsys_state	css;
  
  	/*
  	 * head to keep track of all resource pools
  	 * that belongs to this cgroup.
  	 */
  	struct list_head		rpools;
  };
  
  struct rdmacg_device {
  	struct list_head	dev_node;
  	struct list_head	rpools;
  	char			*name;
  };
  
  /*
   * APIs for RDMA/IB stack to publish when a device wants to
   * participate in resource accounting
   */
  void rdmacg_register_device(struct rdmacg_device *device);
  void rdmacg_unregister_device(struct rdmacg_device *device);
  
  /* APIs for RDMA/IB stack to charge/uncharge pool specific resources */
  int rdmacg_try_charge(struct rdma_cgroup **rdmacg,
  		      struct rdmacg_device *device,
  		      enum rdmacg_resource_type index);
  void rdmacg_uncharge(struct rdma_cgroup *cg,
  		     struct rdmacg_device *device,
  		     enum rdmacg_resource_type index);
  #endif	/* CONFIG_CGROUP_RDMA */
  #endif	/* _CGROUP_RDMA_H */