[RFC] Providers/rxe: Add dma-buf support

Commit Message

Shunsuke Mie Sept. 8, 2021, 6:20 a.m. UTC

This is the user space counter-part of the kernel patch set to add
dma-buf support to the RXE driver.

Implement a new provider method for dma-buf base memory registration.

Pull request at GitHub: https://github.com/linux-rdma/rdma-core/pull/1055

Signed-off-by: Shunsuke Mie <[e-mail]>
---
 providers/rxe/rxe.c | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

Patch

diff --git a/providers/rxe/rxe.c b/providers/rxe/rxe.c
index 3c3ea8bb..0e6b1ecd 100644
--- a/providers/rxe/rxe.c
+++ b/providers/rxe/rxe.c
@@ -239,6 +239,25 @@  static struct ibv_mr *rxe_reg_mr(struct ibv_pd *pd, void *addr, size_t length,
 	return &vmr->ibv_mr;
 }
 
+static struct ibv_mr* rxe_reg_dmabuf_mr(struct ibv_pd *pd, uint64_t offset, size_t length,
+        uint64_t iova, int fd, int access)
+{
+	struct verbs_mr *vmr;
+    int ret;
+
+    vmr = malloc(sizeof (*vmr));
+    if (!vmr)
+        return NULL;
+
+    ret = ibv_cmd_reg_dmabuf_mr(pd, offset, length, iova, fd, access, vmr);
+    if (ret) {
+        free(vmr);
+        return NULL;
+    }
+
+    return &vmr->ibv_mr;
+}
+
 static int rxe_dereg_mr(struct verbs_mr *vmr)
 {
 	int ret;
@@ -1706,6 +1725,7 @@  static const struct verbs_context_ops rxe_ctx_ops = {
 	.alloc_pd = rxe_alloc_pd,
 	.dealloc_pd = rxe_dealloc_pd,
 	.reg_mr = rxe_reg_mr,
+	.reg_dmabuf_mr = rxe_reg_dmabuf_mr,
 	.dereg_mr = rxe_dereg_mr,
 	.alloc_mw = rxe_alloc_mw,
 	.dealloc_mw = rxe_dealloc_mw,