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 enum rds_ib_fr_state { 47 FRMR_IS_FREE, /* mr invalidated & ready for use */ 48 FRMR_IS_INUSE, /* mr is in use or used & can be invalidated */ 49 FRMR_IS_STALE, /* Stale MR and needs to be dropped */ 50 }; 51 52 struct rds_ib_frmr { 53 struct ib_mr *mr; 54 enum rds_ib_fr_state fr_state; 55 bool fr_inv; 56 wait_queue_head_t fr_inv_done; 57 bool fr_reg; 58 wait_queue_head_t fr_reg_done; 59 struct ib_send_wr fr_wr; 60 unsigned int dma_npages; 61 unsigned int sg_byte_len; 62 }; 63 64 /* This is stored as mr->r_trans_private. */ 65 struct rds_ib_mr { 66 struct delayed_work work; 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 u8 odp:1; 82 union { 83 struct rds_ib_frmr frmr; 84 struct ib_mr *mr; 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 spinlock_t clean_lock; /* "clean_list" concurrency */ 102 103 atomic_t free_pinned; /* memory pinned by free MRs */ 104 unsigned long max_items; 105 unsigned long max_items_soft; 106 unsigned long max_free_pinned; 107 unsigned int max_pages; 108 }; 109 110 extern struct workqueue_struct *rds_ib_mr_wq; 111 extern bool prefer_frmr; 112 113 struct rds_ib_mr_pool *rds_ib_create_mr_pool(struct rds_ib_device *rds_dev, 114 int npages); 115 void rds_ib_get_mr_info(struct rds_ib_device *rds_ibdev, 116 struct rds_info_rdma_connection *iinfo); 117 void rds6_ib_get_mr_info(struct rds_ib_device *rds_ibdev, 118 struct rds6_info_rdma_connection *iinfo6); 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 struct rds_connection *conn, u64 start, u64 length, 123 int need_odp); 124 void rds_ib_sync_mr(void *trans_private, int dir); 125 void rds_ib_free_mr(void *trans_private, int invalidate); 126 void rds_ib_flush_mrs(void); 127 int rds_ib_mr_init(void); 128 void rds_ib_mr_exit(void); 129 u32 rds_ib_get_lkey(void *trans_private); 130 131 void __rds_ib_teardown_mr(struct rds_ib_mr *); 132 void rds_ib_teardown_mr(struct rds_ib_mr *); 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_try_reuse_ibmr(struct rds_ib_mr_pool *); 136 struct rds_ib_mr *rds_ib_reg_frmr(struct rds_ib_device *rds_ibdev, 137 struct rds_ib_connection *ic, 138 struct scatterlist *sg, 139 unsigned long nents, u32 *key); 140 void rds_ib_unreg_frmr(struct list_head *list, unsigned int *nfreed, 141 unsigned long *unpinned, unsigned int goal); 142 void rds_ib_free_frmr_list(struct rds_ib_mr *); 143 #endif 144