include/rdma/rw.h

   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
  22 struct 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
  62 int 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);
  65 void 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
  69 int 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);
  74 void 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
  79 struct 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);
  81 int 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
  84 void rdma_rw_init_qp(struct ib_device *dev, struct ib_qp_init_attr *attr);
  85 int rdma_rw_init_mrs(struct ib_qp *qp, struct ib_qp_init_attr *attr);
  86 void rdma_rw_cleanup_mrs(struct ib_qp *qp);
  87
  88 #endif /* _RDMA_RW_H */