2021-09-08 06:23:17

by Shunsuke Mie

[permalink] [raw]
Subject: [RFC PATCH] Providers/rxe: Add dma-buf support

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 <[email protected]>
---
providers/rxe/rxe.c | 20 ++++++++++++++++++++
1 file changed, 20 insertions(+)

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,
--
2.17.1