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         u64                     *dma;
  49 };
  50
  51 enum rds_ib_fr_state {
  52         FRMR_IS_FREE,   /* mr invalidated & ready for use */
  53         FRMR_IS_INUSE,  /* mr is in use or used & can be invalidated */
  54         FRMR_IS_STALE,  /* Stale MR and needs to be dropped  */
  55 };
  56
  57 struct rds_ib_frmr {
  58         struct ib_mr            *mr;
  59         enum rds_ib_fr_state    fr_state;
  60         bool                    fr_inv;
  61         struct ib_send_wr       fr_wr;
  62         unsigned int            dma_npages;
  63         unsigned int            sg_byte_len;
  64 };
  65
  66 /* This is stored as mr->r_trans_private. */
  67 struct rds_ib_mr {
  68         struct rds_ib_device            *device;
  69         struct rds_ib_mr_pool           *pool;
  70         struct rds_ib_connection        *ic;
  71
  72         struct llist_node               llnode;
  73
  74         /* unmap_list is for freeing */
  75         struct list_head                unmap_list;
  76         unsigned int                    remap_count;
  77
  78         struct scatterlist              *sg;
  79         unsigned int                    sg_len;
  80         int                             sg_dma_len;
  81
  82         union {
  83                 struct rds_ib_fmr       fmr;
  84                 struct rds_ib_frmr      frmr;
  85         } u;
  86 };
  87
  88 /* Our own little MR pool */
  89 struct rds_ib_mr_pool {
  90         unsigned int            pool_type;
  91         struct mutex            flush_lock;     /* serialize fmr invalidate */
  92         struct delayed_work     flush_worker;   /* flush worker */
  93
  94         atomic_t                item_count;     /* total # of MRs */
  95         atomic_t                dirty_count;    /* # dirty of MRs */
  96
  97         struct llist_head       drop_list;      /* MRs not reached max_maps */
  98         struct llist_head       free_list;      /* unused MRs */
  99         struct llist_head       clean_list;     /* unused & unmapped MRs */
 100         wait_queue_head_t       flush_wait;
 101
 102         atomic_t                free_pinned;    /* memory pinned by free MRs */
 103         unsigned long           max_items;
 104         unsigned long           max_items_soft;
 105         unsigned long           max_free_pinned;
 106         struct ib_fmr_attr      fmr_attr;
 107         bool                    use_fastreg;
 108 };
 109
 110 extern struct workqueue_struct *rds_ib_mr_wq;
 111 extern unsigned int rds_ib_mr_1m_pool_size;
 112 extern unsigned int rds_ib_mr_8k_pool_size;
 113 extern bool prefer_frmr;
 114
 115 struct rds_ib_mr_pool *rds_ib_create_mr_pool(struct rds_ib_device *rds_dev,
 116                                              int npages);
 117 void rds_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
 118                         struct rds_info_rdma_connection *iinfo);
 119 void rds_ib_destroy_mr_pool(struct rds_ib_mr_pool *);
 120 void *rds_ib_get_mr(struct scatterlist *sg, unsigned long nents,
 121                     struct rds_sock *rs, u32 *key_ret);
 122 void rds_ib_sync_mr(void *trans_private, int dir);
 123 void rds_ib_free_mr(void *trans_private, int invalidate);
 124 void rds_ib_flush_mrs(void);
 125 int rds_ib_mr_init(void);
 126 void rds_ib_mr_exit(void);
 127
 128 void __rds_ib_teardown_mr(struct rds_ib_mr *);
 129 void rds_ib_teardown_mr(struct rds_ib_mr *);
 130 struct rds_ib_mr *rds_ib_alloc_fmr(struct rds_ib_device *, int);
 131 int rds_ib_map_fmr(struct rds_ib_device *, struct rds_ib_mr *,
 132                    struct scatterlist *, unsigned int);
 133 struct rds_ib_mr *rds_ib_reuse_mr(struct rds_ib_mr_pool *);
 134 int rds_ib_flush_mr_pool(struct rds_ib_mr_pool *, int, struct rds_ib_mr **);
 135 struct rds_ib_mr *rds_ib_reg_fmr(struct rds_ib_device *, struct scatterlist *,
 136                                  unsigned long, u32 *);
 137 struct rds_ib_mr *rds_ib_try_reuse_ibmr(struct rds_ib_mr_pool *);
 138 void rds_ib_unreg_fmr(struct list_head *, unsigned int *,
 139                       unsigned long *, unsigned int);
 140 void rds_ib_free_fmr_list(struct rds_ib_mr *);
 141 struct rds_ib_mr *rds_ib_reg_frmr(struct rds_ib_device *rds_ibdev,
 142                                   struct rds_ib_connection *ic,
 143                                   struct scatterlist *sg,
 144                                   unsigned long nents, u32 *key);
 145 void rds_ib_unreg_frmr(struct list_head *list, unsigned int *nfreed,
 146                        unsigned long *unpinned, unsigned int goal);
 147 void rds_ib_free_frmr_list(struct rds_ib_mr *);
 148 #endif