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