net/rds/ib_mr.h

   1 /*
   2  * Copyright (c) 2016 Oracle.  All rights reserved.
   3  *
   4  * This software is available to you under a choice of one of two
   5  * licenses.  You may choose to be licensed under the terms of the GNU
   6  * General Public License (GPL) Version 2, available from the file
   7  * COPYING in the main directory of this source tree, or the
   8  * OpenIB.org BSD license below:
   9  *
  10  *     Redistribution and use in source and binary forms, with or
  11  *     without modification, are permitted provided that the following
  12  *     conditions are met:
  13  *
  14  *      - Redistributions of source code must retain the above
  15  *        copyright notice, this list of conditions and the following
  16  *        disclaimer.
  17  *
  18  *      - Redistributions in binary form must reproduce the above
  19  *        copyright notice, this list of conditions and the following
  20  *        disclaimer in the documentation and/or other materials
  21  *        provided with the distribution.
  22  *
  23  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  24  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  25  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  26  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  27  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  28  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  29  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  30  * SOFTWARE.
  31  */
  32 #ifndef _RDS_IB_MR_H
  33 #define _RDS_IB_MR_H
  34
  35 #include <linux/kernel.h>
  36
  37 #include "rds.h"
  38 #include "ib.h"
  39
  40 #define RDS_MR_1M_POOL_SIZE             (8192 / 2)
  41 #define RDS_MR_1M_MSG_SIZE              256
  42 #define RDS_MR_8K_MSG_SIZE              2
  43 #define RDS_MR_8K_SCALE                 (256 / (RDS_MR_8K_MSG_SIZE + 1))
  44 #define RDS_MR_8K_POOL_SIZE             (RDS_MR_8K_SCALE * (8192 / 2))
  45
  46 struct rds_ib_fmr {
  47         struct ib_fmr           *fmr;
  48 };
  49
  50 enum rds_ib_fr_state {
  51         FRMR_IS_FREE,   /* mr invalidated & ready for use */
  52         FRMR_IS_INUSE,  /* mr is in use or used & can be invalidated */
  53         FRMR_IS_STALE,  /* Stale MR and needs to be dropped  */
  54 };
  55
  56 struct rds_ib_frmr {
  57         struct ib_mr            *mr;
  58         enum rds_ib_fr_state    fr_state;
  59         bool                    fr_inv;
  60         wait_queue_head_t       fr_inv_done;
  61         bool                    fr_reg;
  62         wait_queue_head_t       fr_reg_done;
  63         struct ib_send_wr       fr_wr;
  64         unsigned int            dma_npages;
  65         unsigned int            sg_byte_len;
  66 };
  67
  68 /* This is stored as mr->r_trans_private. */
  69 struct rds_ib_mr {
  70         struct rds_ib_device            *device;
  71         struct rds_ib_mr_pool           *pool;
  72         struct rds_ib_connection        *ic;
  73
  74         struct llist_node               llnode;
  75
  76         /* unmap_list is for freeing */
  77         struct list_head                unmap_list;
  78         unsigned int                    remap_count;
  79
  80         struct scatterlist              *sg;
  81         unsigned int                    sg_len;
  82         int                             sg_dma_len;
  83
  84         union {
  85                 struct rds_ib_fmr       fmr;
  86                 struct rds_ib_frmr      frmr;
  87         } u;
  88 };
  89
  90 /* Our own little MR pool */
  91 struct rds_ib_mr_pool {
  92         unsigned int            pool_type;
  93         struct mutex            flush_lock;     /* serialize fmr invalidate */
  94         struct delayed_work     flush_worker;   /* flush worker */
  95
  96         atomic_t                item_count;     /* total # of MRs */
  97         atomic_t                dirty_count;    /* # dirty of MRs */
  98
  99         struct llist_head       drop_list;      /* MRs not reached max_maps */
 100         struct llist_head       free_list;      /* unused MRs */
 101         struct llist_head       clean_list;     /* unused & unmapped MRs */
 102         wait_queue_head_t       flush_wait;
 103         spinlock_t              clean_lock;     /* "clean_list" concurrency */
 104
 105         atomic_t                free_pinned;    /* memory pinned by free MRs */
 106         unsigned long           max_items;
 107         unsigned long           max_items_soft;
 108         unsigned long           max_free_pinned;
 109         struct ib_fmr_attr      fmr_attr;
 110         bool                    use_fastreg;
 111 };
 112
 113 extern struct workqueue_struct *rds_ib_mr_wq;
 114 extern bool prefer_frmr;
 115
 116 struct rds_ib_mr_pool *rds_ib_create_mr_pool(struct rds_ib_device *rds_dev,
 117                                              int npages);
 118 void rds_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
 119                         struct rds_info_rdma_connection *iinfo);
 120 void rds6_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
 121                          struct rds6_info_rdma_connection *iinfo6);
 122 void rds_ib_destroy_mr_pool(struct rds_ib_mr_pool *);
 123 void *rds_ib_get_mr(struct scatterlist *sg, unsigned long nents,
 124                     struct rds_sock *rs, u32 *key_ret,
 125                     struct rds_connection *conn);
 126 void rds_ib_sync_mr(void *trans_private, int dir);
 127 void rds_ib_free_mr(void *trans_private, int invalidate);
 128 void rds_ib_flush_mrs(void);
 129 int rds_ib_mr_init(void);
 130 void rds_ib_mr_exit(void);
 131
 132 void __rds_ib_teardown_mr(struct rds_ib_mr *);
 133 void rds_ib_teardown_mr(struct rds_ib_mr *);
 134 struct rds_ib_mr *rds_ib_alloc_fmr(struct rds_ib_device *, int);
 135 struct rds_ib_mr *rds_ib_reuse_mr(struct rds_ib_mr_pool *);
 136 int rds_ib_flush_mr_pool(struct rds_ib_mr_pool *, int, struct rds_ib_mr **);
 137 struct rds_ib_mr *rds_ib_reg_fmr(struct rds_ib_device *, struct scatterlist *,
 138                                  unsigned long, u32 *);
 139 struct rds_ib_mr *rds_ib_try_reuse_ibmr(struct rds_ib_mr_pool *);
 140 void rds_ib_unreg_fmr(struct list_head *, unsigned int *,
 141                       unsigned long *, unsigned int);
 142 void rds_ib_free_fmr_list(struct rds_ib_mr *);
 143 struct rds_ib_mr *rds_ib_reg_frmr(struct rds_ib_device *rds_ibdev,
 144                                   struct rds_ib_connection *ic,
 145                                   struct scatterlist *sg,
 146                                   unsigned long nents, u32 *key);
 147 void rds_ib_unreg_frmr(struct list_head *list, unsigned int *nfreed,
 148                        unsigned long *unpinned, unsigned int goal);
 149 void rds_ib_free_frmr_list(struct rds_ib_mr *);
 150 #endif