1 /* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
3 * Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved.
4 * Copyright (c) 2015 System Fabric Works, Inc. All rights reserved.
19 RXE_NUM_TYPES, /* keep me last */
22 struct rxe_pool_elem {
23 struct rxe_pool *pool;
26 struct list_head list;
27 struct completion complete;
34 void (*cleanup)(struct rxe_pool_elem *elem);
35 enum rxe_elem_type type;
37 unsigned int max_elem;
43 struct xa_limit limit;
47 /* initialize a pool of objects with given limit on
48 * number of elements. gets parameters from rxe_type_info
49 * pool elements will be allocated out of a slab cache
51 void rxe_pool_init(struct rxe_dev *rxe, struct rxe_pool *pool,
52 enum rxe_elem_type type);
54 /* free resources from object pool */
55 void rxe_pool_cleanup(struct rxe_pool *pool);
57 /* connect already allocated object to pool */
58 int __rxe_add_to_pool(struct rxe_pool *pool, struct rxe_pool_elem *elem,
60 #define rxe_add_to_pool(pool, obj) __rxe_add_to_pool(pool, &(obj)->elem, true)
61 #define rxe_add_to_pool_ah(pool, obj, sleepable) __rxe_add_to_pool(pool, \
62 &(obj)->elem, sleepable)
64 /* lookup an indexed object from index. takes a reference on object */
65 void *rxe_pool_get_index(struct rxe_pool *pool, u32 index);
67 int __rxe_get(struct rxe_pool_elem *elem);
68 #define rxe_get(obj) __rxe_get(&(obj)->elem)
70 int __rxe_put(struct rxe_pool_elem *elem);
71 #define rxe_put(obj) __rxe_put(&(obj)->elem)
73 int __rxe_cleanup(struct rxe_pool_elem *elem, bool sleepable);
74 #define rxe_cleanup(obj) __rxe_cleanup(&(obj)->elem, true)
75 #define rxe_cleanup_ah(obj, sleepable) __rxe_cleanup(&(obj)->elem, sleepable)
77 #define rxe_read(obj) kref_read(&(obj)->elem.ref_cnt)
79 void __rxe_finalize(struct rxe_pool_elem *elem);
80 #define rxe_finalize(obj) __rxe_finalize(&(obj)->elem)
82 #endif /* RXE_POOL_H */