1 /* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */ 2 /* 3 * Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved. 4 * Copyright (c) 2015 System Fabric Works, Inc. All rights reserved. 5 */ 6 7 #ifndef RXE_POOL_H 8 #define RXE_POOL_H 9 10 enum rxe_elem_type { 11 RXE_TYPE_UC, 12 RXE_TYPE_PD, 13 RXE_TYPE_AH, 14 RXE_TYPE_SRQ, 15 RXE_TYPE_QP, 16 RXE_TYPE_CQ, 17 RXE_TYPE_MR, 18 RXE_TYPE_MW, 19 RXE_NUM_TYPES, /* keep me last */ 20 }; 21 22 struct rxe_pool_elem { 23 struct rxe_pool *pool; 24 void *obj; 25 struct kref ref_cnt; 26 struct list_head list; 27 struct completion complete; 28 u32 index; 29 }; 30 31 struct rxe_pool { 32 struct rxe_dev *rxe; 33 const char *name; 34 void (*cleanup)(struct rxe_pool_elem *elem); 35 enum rxe_elem_type type; 36 37 unsigned int max_elem; 38 atomic_t num_elem; 39 size_t elem_size; 40 size_t elem_offset; 41 42 struct xarray xa; 43 struct xa_limit limit; 44 u32 next; 45 }; 46 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 50 */ 51 void rxe_pool_init(struct rxe_dev *rxe, struct rxe_pool *pool, 52 enum rxe_elem_type type); 53 54 /* free resources from object pool */ 55 void rxe_pool_cleanup(struct rxe_pool *pool); 56 57 /* allocate an object from pool */ 58 void *rxe_alloc(struct rxe_pool *pool); 59 60 /* connect already allocated object to pool */ 61 int __rxe_add_to_pool(struct rxe_pool *pool, struct rxe_pool_elem *elem, 62 bool sleepable); 63 #define rxe_add_to_pool(pool, obj) __rxe_add_to_pool(pool, &(obj)->elem, true) 64 #define rxe_add_to_pool_ah(pool, obj, sleepable) __rxe_add_to_pool(pool, \ 65 &(obj)->elem, sleepable) 66 67 /* lookup an indexed object from index. takes a reference on object */ 68 void *rxe_pool_get_index(struct rxe_pool *pool, u32 index); 69 70 int __rxe_get(struct rxe_pool_elem *elem); 71 #define rxe_get(obj) __rxe_get(&(obj)->elem) 72 73 int __rxe_put(struct rxe_pool_elem *elem); 74 #define rxe_put(obj) __rxe_put(&(obj)->elem) 75 76 int __rxe_cleanup(struct rxe_pool_elem *elem, bool sleepable); 77 #define rxe_cleanup(obj) __rxe_cleanup(&(obj)->elem, true) 78 #define rxe_cleanup_ah(obj, sleepable) __rxe_cleanup(&(obj)->elem, sleepable) 79 80 #define rxe_read(obj) kref_read(&(obj)->elem.ref_cnt) 81 82 void __rxe_finalize(struct rxe_pool_elem *elem); 83 #define rxe_finalize(obj) __rxe_finalize(&(obj)->elem) 84 85 #endif /* RXE_POOL_H */ 86