drivers/infiniband/sw/rxe/rxe_pool.h

   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 /* connect already allocated object to pool */
  58 int __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 */
  65 void *rxe_pool_get_index(struct rxe_pool *pool, u32 index);
  66
  67 int __rxe_get(struct rxe_pool_elem *elem);
  68 #define rxe_get(obj) __rxe_get(&(obj)->elem)
  69
  70 int __rxe_put(struct rxe_pool_elem *elem);
  71 #define rxe_put(obj) __rxe_put(&(obj)->elem)
  72
  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)
  76
  77 #define rxe_read(obj) kref_read(&(obj)->elem.ref_cnt)
  78
  79 void __rxe_finalize(struct rxe_pool_elem *elem);
  80 #define rxe_finalize(obj) __rxe_finalize(&(obj)->elem)
  81
  82 #endif /* RXE_POOL_H */