include/rdma/rw.h

   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, u8 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, u8 port_num,
  49                 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                 u8 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                 u8 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                 u8 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, u8 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, u8 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 */