/* 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 */