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
10enum 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
22struct 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
31struct 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 */
51void rxe_pool_init(struct rxe_dev *rxe, struct rxe_pool *pool,
52 enum rxe_elem_type type);
53
54/* free resources from object pool */
55void rxe_pool_cleanup(struct rxe_pool *pool);
56
57/* connect already allocated object to pool */
58int __rxe_add_to_pool(struct rxe_pool *pool, struct rxe_pool_elem *elem,
59 bool sleepable);
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)
63
64/* lookup an indexed object from index. takes a reference on object */
65void *rxe_pool_get_index(struct rxe_pool *pool, u32 index);
66
67int __rxe_get(struct rxe_pool_elem *elem);
68#define rxe_get(obj) __rxe_get(&(obj)->elem)
69
70int __rxe_put(struct rxe_pool_elem *elem);
71#define rxe_put(obj) __rxe_put(&(obj)->elem)
72
73int __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)
76
77#define rxe_read(obj) kref_read(&(obj)->elem.ref_cnt)
78
79void __rxe_finalize(struct rxe_pool_elem *elem);
80#define rxe_finalize(obj) __rxe_finalize(&(obj)->elem)
81
82#endif /* RXE_POOL_H */
83

source code of linux/drivers/infiniband/sw/rxe/rxe_pool.h