1/*
 2 * Copyright (c) 2016 HGST, a Western Digital Company.
 3 *
 4 * This program is free software; you can redistribute it and/or modify it
 5 * under the terms and conditions of the GNU General Public License,
 6 * version 2, as published by the Free Software Foundation.
 7 *
 8 * This program is distributed in the hope it will be useful, but WITHOUT
 9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
11 * more details.
12 */
13#ifndef _RDMA_RW_H
14#define _RDMA_RW_H
15
16#include <linux/dma-mapping.h>
17#include <linux/scatterlist.h>
18#include <rdma/ib_verbs.h>
19#include <rdma/rdma_cm.h>
20#include <rdma/mr_pool.h>
21
22struct rdma_rw_ctx {
23	/* number of RDMA READ/WRITE WRs (not counting MR WRs) */
24	u32			nr_ops;
25
26	/* tag for the union below: */
27	u8			type;
28
29	union {
30		/* for mapping a single SGE: */
31		struct {
32			struct ib_sge		sge;
33			struct ib_rdma_wr	wr;
34		} single;
35
36		/* for mapping of multiple SGEs: */
37		struct {
38			struct ib_sge		*sges;
39			struct ib_rdma_wr	*wrs;
40		} map;
41
42		/* for registering multiple WRs: */
43		struct rdma_rw_reg_ctx {
44			struct ib_sge		sge;
45			struct ib_rdma_wr	wr;
46			struct ib_reg_wr	reg_wr;
47			struct ib_send_wr	inv_wr;
48			struct ib_mr		*mr;
49		} *reg;
50
51		struct {
52			struct rdma_rw_reg_ctx	data;
53			struct rdma_rw_reg_ctx	prot;
54			struct ib_send_wr	sig_inv_wr;
55			struct ib_mr		*sig_mr;
56			struct ib_sge		sig_sge;
57			struct ib_sig_handover_wr sig_wr;
58		} *sig;
59	};
60};
61
62int rdma_rw_ctx_init(struct rdma_rw_ctx *ctx, struct ib_qp *qp, u8 port_num,
63		struct scatterlist *sg, u32 sg_cnt, u32 sg_offset,
64		u64 remote_addr, u32 rkey, enum dma_data_direction dir);
65void rdma_rw_ctx_destroy(struct rdma_rw_ctx *ctx, struct ib_qp *qp, u8 port_num,
66		struct scatterlist *sg, u32 sg_cnt,
67		enum dma_data_direction dir);
68
69int rdma_rw_ctx_signature_init(struct rdma_rw_ctx *ctx, struct ib_qp *qp,
70		u8 port_num, struct scatterlist *sg, u32 sg_cnt,
71		struct scatterlist *prot_sg, u32 prot_sg_cnt,
72		struct ib_sig_attrs *sig_attrs, u64 remote_addr, u32 rkey,
73		enum dma_data_direction dir);
74void rdma_rw_ctx_destroy_signature(struct rdma_rw_ctx *ctx, struct ib_qp *qp,
75		u8 port_num, struct scatterlist *sg, u32 sg_cnt,
76		struct scatterlist *prot_sg, u32 prot_sg_cnt,
77		enum dma_data_direction dir);
78
79struct ib_send_wr *rdma_rw_ctx_wrs(struct rdma_rw_ctx *ctx, struct ib_qp *qp,
80		u8 port_num, struct ib_cqe *cqe, struct ib_send_wr *chain_wr);
81int rdma_rw_ctx_post(struct rdma_rw_ctx *ctx, struct ib_qp *qp, u8 port_num,
82		struct ib_cqe *cqe, struct ib_send_wr *chain_wr);
83
84void rdma_rw_init_qp(struct ib_device *dev, struct ib_qp_init_attr *attr);
85int rdma_rw_init_mrs(struct ib_qp *qp, struct ib_qp_init_attr *attr);
86void rdma_rw_cleanup_mrs(struct ib_qp *qp);
87
88#endif /* _RDMA_RW_H */