/* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
/* Copyright (c) 2024 NVIDIA Corporation & Affiliates */

#ifndef __MLX5_FS_POOL_H__
#define __MLX5_FS_POOL_H__

#include <linux/mlx5/driver.h>

struct mlx5_fs_bulk {
 struct list_head pool_list;
 int bulk_len;
 unsigned long *bitmask;
};

struct mlx5_fs_pool_index {
 struct mlx5_fs_bulk *fs_bulk;
 int index;
};

struct mlx5_fs_pool;

struct mlx5_fs_pool_ops {
 int (*bulk_destroy)(struct mlx5_core_dev *dev, struct mlx5_fs_bulk *bulk);
 struct mlx5_fs_bulk * (*bulk_create)(struct mlx5_core_dev *dev,
          void *pool_ctx);
 void (*update_threshold)(struct mlx5_fs_pool *pool);
};

struct mlx5_fs_pool {
 struct mlx5_core_dev *dev;
 void *pool_ctx;
 const struct mlx5_fs_pool_ops *ops;
 struct mutex pool_lock; /* protects pool lists */
 struct list_head fully_used;
 struct list_head partially_used;
 struct list_head unused;
 int available_units;
 int used_units;
 int threshold;
};

int mlx5_fs_bulk_init(struct mlx5_core_dev *dev, struct mlx5_fs_bulk *fs_bulk,
        int bulk_len);
void mlx5_fs_bulk_cleanup(struct mlx5_fs_bulk *fs_bulk);
int mlx5_fs_bulk_get_free_amount(struct mlx5_fs_bulk *bulk);

void mlx5_fs_pool_init(struct mlx5_fs_pool *pool, struct mlx5_core_dev *dev,
         const struct mlx5_fs_pool_ops *ops, void *pool_ctx);
void mlx5_fs_pool_cleanup(struct mlx5_fs_pool *pool);
int mlx5_fs_pool_acquire_index(struct mlx5_fs_pool *fs_pool,
          struct mlx5_fs_pool_index *pool_index);
int mlx5_fs_pool_release_index(struct mlx5_fs_pool *fs_pool,
          struct mlx5_fs_pool_index *pool_index);

#endif /* __MLX5_FS_POOL_H__ */