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 151