1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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
46struct rds_ib_fmr {
47 struct ib_fmr *fmr;
48 u64 *dma;
49};
50
51enum rds_ib_fr_state {
52 FRMR_IS_FREE,
53 FRMR_IS_INUSE,
54 FRMR_IS_STALE,
55};
56
57struct 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
67struct 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
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
89struct rds_ib_mr_pool {
90 unsigned int pool_type;
91 struct mutex flush_lock;
92 struct delayed_work flush_worker;
93
94 atomic_t item_count;
95 atomic_t dirty_count;
96
97 struct llist_head drop_list;
98 struct llist_head free_list;
99 struct llist_head clean_list;
100 wait_queue_head_t flush_wait;
101
102 atomic_t free_pinned;
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
110extern struct workqueue_struct *rds_ib_mr_wq;
111extern unsigned int rds_ib_mr_1m_pool_size;
112extern unsigned int rds_ib_mr_8k_pool_size;
113extern bool prefer_frmr;
114
115struct rds_ib_mr_pool *rds_ib_create_mr_pool(struct rds_ib_device *rds_dev,
116 int npages);
117void rds_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
118 struct rds_info_rdma_connection *iinfo);
119void rds_ib_destroy_mr_pool(struct rds_ib_mr_pool *);
120void *rds_ib_get_mr(struct scatterlist *sg, unsigned long nents,
121 struct rds_sock *rs, u32 *key_ret);
122void rds_ib_sync_mr(void *trans_private, int dir);
123void rds_ib_free_mr(void *trans_private, int invalidate);
124void rds_ib_flush_mrs(void);
125int rds_ib_mr_init(void);
126void rds_ib_mr_exit(void);
127
128void __rds_ib_teardown_mr(struct rds_ib_mr *);
129void rds_ib_teardown_mr(struct rds_ib_mr *);
130struct rds_ib_mr *rds_ib_alloc_fmr(struct rds_ib_device *, int);
131int rds_ib_map_fmr(struct rds_ib_device *, struct rds_ib_mr *,
132 struct scatterlist *, unsigned int);
133struct rds_ib_mr *rds_ib_reuse_mr(struct rds_ib_mr_pool *);
134int rds_ib_flush_mr_pool(struct rds_ib_mr_pool *, int, struct rds_ib_mr **);
135struct rds_ib_mr *rds_ib_reg_fmr(struct rds_ib_device *, struct scatterlist *,
136 unsigned long, u32 *);
137struct rds_ib_mr *rds_ib_try_reuse_ibmr(struct rds_ib_mr_pool *);
138void rds_ib_unreg_fmr(struct list_head *, unsigned int *,
139 unsigned long *, unsigned int);
140void rds_ib_free_fmr_list(struct rds_ib_mr *);
141struct 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);
145void rds_ib_unreg_frmr(struct list_head *list, unsigned int *nfreed,
146 unsigned long *unpinned, unsigned int goal);
147void rds_ib_free_frmr_list(struct rds_ib_mr *);
148#endif
149