The import MR verb take care of importing the generic part of the MR
object and then triggers provider's specific callback to take care of
provider's specific attributes.
Add implementation of mlx5 related MR attributes.
Signed-off-by: Yuval Shaia <yuval.shaia@xxxxxxxxxx>
Signed-off-by: Shamir Rabinovitch <shamir.rabinovitch@xxxxxxxxxx>
Signed-off-by: Shamir Rabinovitch <srabinov7@xxxxxxxxx>
---
providers/mlx5/mlx5.c | 1 +
providers/mlx5/mlx5.h | 2 ++
providers/mlx5/verbs.c | 26 ++++++++++++++++++++++++++
3 files changed, 29 insertions(+)
diff --git a/providers/mlx5/mlx5.c b/providers/mlx5/mlx5.c
index c16b30b3..8d1fa232 100644
--- a/providers/mlx5/mlx5.c
+++ b/providers/mlx5/mlx5.c
@@ -91,6 +91,7 @@ static const struct verbs_context_ops mlx5_ctx_common_ops = {
.alloc_pd = mlx5_alloc_pd,
.async_event = mlx5_async_event,
.dealloc_pd = mlx5_free_pd,
+ .import_mr = mlx5_import_mr,
.import_pd = mlx5_import_pd,
.reg_mr = mlx5_reg_mr,
.rereg_mr = mlx5_rereg_mr,
diff --git a/providers/mlx5/mlx5.h b/providers/mlx5/mlx5.h
index 06e2b471..858ae7c2 100644
--- a/providers/mlx5/mlx5.h
+++ b/providers/mlx5/mlx5.h
@@ -816,6 +816,8 @@ int mlx5_query_port(struct ibv_context *context, uint8_t port,
struct ibv_pd *mlx5_alloc_pd(struct ibv_context *context);
int mlx5_free_pd(struct ibv_pd *pd);
+struct ibv_mr *mlx5_import_mr(struct ibv_context *context, uint32_t fd,
+ uint32_t handle);
struct ibv_pd *mlx5_import_pd(struct ibv_context *context, uint32_t fd,
uint32_t handle);
diff --git a/providers/mlx5/verbs.c b/providers/mlx5/verbs.c
index 3d2510c3..b4964b17 100644
--- a/providers/mlx5/verbs.c
+++ b/providers/mlx5/verbs.c
@@ -178,6 +178,32 @@ struct ibv_pd *mlx5_alloc_pd(struct ibv_context *context)
return &pd->ibv_pd;
}
+struct ibv_mr *mlx5_import_mr(struct ibv_context *context, uint32_t fd,
+ uint32_t handle)
+{
+ struct ibv_import_mr cmd = {
+ .handle = handle,
+ .type = UVERBS_OBJECT_MR,
+ .fd = fd,
+ };
+ struct ib_uverbs_import_fr_fd_resp resp = {0};
+ struct mlx5_mr *mr;
+ int ret;
+
+ mr = calloc(1, sizeof(*mr));
+ if (!mr)
+ return NULL;
+
+ ret = ibv_cmd_import_mr(context, &mr->vmr, &cmd, sizeof(cmd), &resp,
+ sizeof(resp));
+ if (ret) {
+ free(mr);
+ return NULL;
+ }
+
+ return &mr->vmr.ibv_mr;
+}
+
struct ibv_pd *mlx5_import_pd(struct ibv_context *context, uint32_t fd,
uint32_t handle)
{
--
2.20.1
