/* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
 *
 * Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 */

#ifndef RDMA_CORE_FRMR_POOLS_H
#define RDMA_CORE_FRMR_POOLS_H

#include <rdma/frmr_pools.h>
#include <linux/rbtree_types.h>
#include <linux/spinlock_types.h>
#include <linux/types.h>
#include <asm/page.h>
#include <linux/workqueue.h>

#define NUM_HANDLES_PER_PAGE \
	((PAGE_SIZE - sizeof(struct list_head)) / sizeof(u32))

struct frmr_handles_page {
	struct list_head list;
	u32 handles[NUM_HANDLES_PER_PAGE];
};

/* FRMR queue holds a list of frmr_handles_page.
 * num_pages: number of pages in the queue.
 * ci: current index in the handles array across all pages.
 */
struct frmr_queue {
	struct list_head pages_list;
	u32 num_pages;
	unsigned long ci;
};

struct ib_frmr_pool {
	struct rb_node node;
	struct ib_frmr_key key; /* Pool key */

	/* Protect access to the queue */
	spinlock_t lock;
	struct frmr_queue queue;
	struct frmr_queue inactive_queue;

	struct delayed_work aging_work;
	struct ib_device *device;

	u32 max_in_use;
	u32 in_use;
	u32 pinned_handles;
};

struct ib_frmr_pools {
	struct rb_root rb_root;
	rwlock_t rb_lock;
	const struct ib_frmr_pool_ops *pool_ops;

	struct workqueue_struct *aging_wq;
	u32 aging_period_sec;
};

int ib_frmr_pools_set_pinned(struct ib_device *device, struct ib_frmr_key *key,
			     u32 pinned_handles);
int ib_frmr_pools_set_aging_period(struct ib_device *device, u32 period_sec);
#endif /* RDMA_CORE_FRMR_POOLS_H */