1 | /* SPDX-License-Identifier: GPL-2.0-only */ |
2 | /* |
3 | * Copyright (c) 2016 HGST, a Western Digital Company. |
4 | */ |
5 | #ifndef _RDMA_RW_H |
6 | #define _RDMA_RW_H |
7 | |
8 | #include <linux/dma-mapping.h> |
9 | #include <linux/scatterlist.h> |
10 | #include <rdma/ib_verbs.h> |
11 | #include <rdma/rdma_cm.h> |
12 | #include <rdma/mr_pool.h> |
13 | |
14 | struct rdma_rw_ctx { |
15 | /* number of RDMA READ/WRITE WRs (not counting MR WRs) */ |
16 | u32 nr_ops; |
17 | |
18 | /* tag for the union below: */ |
19 | u8 type; |
20 | |
21 | union { |
22 | /* for mapping a single SGE: */ |
23 | struct { |
24 | struct ib_sge sge; |
25 | struct ib_rdma_wr wr; |
26 | } single; |
27 | |
28 | /* for mapping of multiple SGEs: */ |
29 | struct { |
30 | struct ib_sge *sges; |
31 | struct ib_rdma_wr *wrs; |
32 | } map; |
33 | |
34 | /* for registering multiple WRs: */ |
35 | struct rdma_rw_reg_ctx { |
36 | struct ib_sge sge; |
37 | struct ib_rdma_wr wr; |
38 | struct ib_reg_wr reg_wr; |
39 | struct ib_send_wr inv_wr; |
40 | struct ib_mr *mr; |
41 | } *reg; |
42 | }; |
43 | }; |
44 | |
45 | int rdma_rw_ctx_init(struct rdma_rw_ctx *ctx, struct ib_qp *qp, u32 port_num, |
46 | struct scatterlist *sg, u32 sg_cnt, u32 sg_offset, |
47 | u64 remote_addr, u32 rkey, enum dma_data_direction dir); |
48 | void rdma_rw_ctx_destroy(struct rdma_rw_ctx *ctx, struct ib_qp *qp, |
49 | u32 port_num, struct scatterlist *sg, u32 sg_cnt, |
50 | enum dma_data_direction dir); |
51 | |
52 | int rdma_rw_ctx_signature_init(struct rdma_rw_ctx *ctx, struct ib_qp *qp, |
53 | u32 port_num, struct scatterlist *sg, u32 sg_cnt, |
54 | struct scatterlist *prot_sg, u32 prot_sg_cnt, |
55 | struct ib_sig_attrs *sig_attrs, u64 remote_addr, u32 rkey, |
56 | enum dma_data_direction dir); |
57 | void rdma_rw_ctx_destroy_signature(struct rdma_rw_ctx *ctx, struct ib_qp *qp, |
58 | u32 port_num, struct scatterlist *sg, u32 sg_cnt, |
59 | struct scatterlist *prot_sg, u32 prot_sg_cnt, |
60 | enum dma_data_direction dir); |
61 | |
62 | struct ib_send_wr *rdma_rw_ctx_wrs(struct rdma_rw_ctx *ctx, struct ib_qp *qp, |
63 | u32 port_num, struct ib_cqe *cqe, struct ib_send_wr *chain_wr); |
64 | int rdma_rw_ctx_post(struct rdma_rw_ctx *ctx, struct ib_qp *qp, u32 port_num, |
65 | struct ib_cqe *cqe, struct ib_send_wr *chain_wr); |
66 | |
67 | unsigned int rdma_rw_mr_factor(struct ib_device *device, u32 port_num, |
68 | unsigned int maxpages); |
69 | void rdma_rw_init_qp(struct ib_device *dev, struct ib_qp_init_attr *attr); |
70 | int rdma_rw_init_mrs(struct ib_qp *qp, struct ib_qp_init_attr *attr); |
71 | void rdma_rw_cleanup_mrs(struct ib_qp *qp); |
72 | |
73 | #endif /* _RDMA_RW_H */ |
74 | |