vdpa/mlx5: fix large VM memory region registration
[dpdk.git] / drivers / vdpa / mlx5 / mlx5_vdpa_lm.c
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright 2019 Mellanox Technologies, Ltd
3  */
4 #include <rte_malloc.h>
5 #include <rte_errno.h>
6
7 #include "mlx5_vdpa_utils.h"
8 #include "mlx5_vdpa.h"
9
10
11 int
12 mlx5_vdpa_logging_enable(struct mlx5_vdpa_priv *priv, int enable)
13 {
14         struct mlx5_devx_virtq_attr attr = {
15                 .type = MLX5_VIRTQ_MODIFY_TYPE_DIRTY_BITMAP_DUMP_ENABLE,
16                 .dirty_bitmap_dump_enable = enable,
17         };
18         int i;
19
20         for (i = 0; i < priv->nr_virtqs; ++i) {
21                 attr.queue_index = i;
22                 if (!priv->virtqs[i].virtq) {
23                         DRV_LOG(DEBUG, "virtq %d is invalid for dirty bitmap "
24                                 "enabling.", i);
25                 } else if (mlx5_devx_cmd_modify_virtq(priv->virtqs[i].virtq,
26                            &attr)) {
27                         DRV_LOG(ERR, "Failed to modify virtq %d for dirty "
28                                 "bitmap enabling.", i);
29                         return -1;
30                 }
31         }
32         return 0;
33 }
34
35 int
36 mlx5_vdpa_dirty_bitmap_set(struct mlx5_vdpa_priv *priv, uint64_t log_base,
37                            uint64_t log_size)
38 {
39         struct mlx5_devx_mkey_attr mkey_attr = {
40                         .addr = (uintptr_t)log_base,
41                         .size = log_size,
42                         .pd = priv->pdn,
43                         .pg_access = 1,
44         };
45         struct mlx5_devx_virtq_attr attr = {
46                 .type = MLX5_VIRTQ_MODIFY_TYPE_DIRTY_BITMAP_PARAMS,
47                 .dirty_bitmap_addr = log_base,
48                 .dirty_bitmap_size = log_size,
49         };
50         struct mlx5_vdpa_query_mr *mr = rte_malloc(__func__, sizeof(*mr), 0);
51         int i;
52
53         if (!mr) {
54                 DRV_LOG(ERR, "Failed to allocate mem for lm mr.");
55                 return -1;
56         }
57         mr->umem = mlx5_glue->devx_umem_reg(priv->ctx,
58                                             (void *)(uintptr_t)log_base,
59                                             log_size, IBV_ACCESS_LOCAL_WRITE);
60         if (!mr->umem) {
61                 DRV_LOG(ERR, "Failed to register umem for lm mr.");
62                 goto err;
63         }
64         mkey_attr.umem_id = mr->umem->umem_id;
65         mr->mkey = mlx5_devx_cmd_mkey_create(priv->ctx, &mkey_attr);
66         if (!mr->mkey) {
67                 DRV_LOG(ERR, "Failed to create Mkey for lm.");
68                 goto err;
69         }
70         attr.dirty_bitmap_mkey = mr->mkey->id;
71         for (i = 0; i < priv->nr_virtqs; ++i) {
72                 attr.queue_index = i;
73                 if (!priv->virtqs[i].virtq) {
74                         DRV_LOG(DEBUG, "virtq %d is invalid for LM.", i);
75                 } else if (mlx5_devx_cmd_modify_virtq(priv->virtqs[i].virtq,
76                                                       &attr)) {
77                         DRV_LOG(ERR, "Failed to modify virtq %d for LM.", i);
78                         goto err;
79                 }
80         }
81         mr->is_indirect = 0;
82         SLIST_INSERT_HEAD(&priv->mr_list, mr, next);
83         return 0;
84 err:
85         if (mr->mkey)
86                 mlx5_devx_cmd_destroy(mr->mkey);
87         if (mr->umem)
88                 mlx5_glue->devx_umem_dereg(mr->umem);
89         rte_free(mr);
90         return -1;
91 }
92
93 #define MLX5_VDPA_USED_RING_LEN(size) \
94         ((size) * sizeof(struct vring_used_elem) + sizeof(uint16_t) * 3)
95
96 int
97 mlx5_vdpa_lm_log(struct mlx5_vdpa_priv *priv)
98 {
99         uint64_t features;
100         int ret = rte_vhost_get_negotiated_features(priv->vid, &features);
101         int i;
102
103         if (ret) {
104                 DRV_LOG(ERR, "Failed to get negotiated features.");
105                 return -1;
106         }
107         if (!RTE_VHOST_NEED_LOG(features))
108                 return 0;
109         for (i = 0; i < priv->nr_virtqs; ++i) {
110                 if (!priv->virtqs[i].virtq) {
111                         DRV_LOG(DEBUG, "virtq %d is invalid for LM log.", i);
112                 } else {
113                         ret = mlx5_vdpa_virtq_stop(priv, i);
114                         if (ret) {
115                                 DRV_LOG(ERR, "Failed to stop virtq %d for LM "
116                                         "log.", i);
117                                 return -1;
118                         }
119                 }
120                 rte_vhost_log_used_vring(priv->vid, i, 0,
121                               MLX5_VDPA_USED_RING_LEN(priv->virtqs[i].vq_size));
122         }
123         return 0;
124 }