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