vdpa/mlx5: fix live migration termination
[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                         .klm_array = NULL,
45                         .klm_num = 0,
46                         .relaxed_ordering = 0,
47         };
48         struct mlx5_devx_virtq_attr attr = {
49                 .type = MLX5_VIRTQ_MODIFY_TYPE_DIRTY_BITMAP_PARAMS,
50                 .dirty_bitmap_addr = log_base,
51                 .dirty_bitmap_size = log_size,
52         };
53         struct mlx5_vdpa_query_mr *mr = rte_malloc(__func__, sizeof(*mr), 0);
54         int i;
55
56         if (!mr) {
57                 DRV_LOG(ERR, "Failed to allocate mem for lm mr.");
58                 return -1;
59         }
60         mr->umem = mlx5_glue->devx_umem_reg(priv->ctx,
61                                             (void *)(uintptr_t)log_base,
62                                             log_size, IBV_ACCESS_LOCAL_WRITE);
63         if (!mr->umem) {
64                 DRV_LOG(ERR, "Failed to register umem for lm mr.");
65                 goto err;
66         }
67         mkey_attr.umem_id = mr->umem->umem_id;
68         mr->mkey = mlx5_devx_cmd_mkey_create(priv->ctx, &mkey_attr);
69         if (!mr->mkey) {
70                 DRV_LOG(ERR, "Failed to create Mkey for lm.");
71                 goto err;
72         }
73         attr.dirty_bitmap_mkey = mr->mkey->id;
74         for (i = 0; i < priv->nr_virtqs; ++i) {
75                 attr.queue_index = i;
76                 if (!priv->virtqs[i].virtq) {
77                         DRV_LOG(DEBUG, "virtq %d is invalid for LM.", i);
78                 } else if (mlx5_devx_cmd_modify_virtq(priv->virtqs[i].virtq,
79                                                       &attr)) {
80                         DRV_LOG(ERR, "Failed to modify virtq %d for LM.", i);
81                         goto err;
82                 }
83         }
84         mr->is_indirect = 0;
85         SLIST_INSERT_HEAD(&priv->mr_list, mr, next);
86         return 0;
87 err:
88         if (mr->mkey)
89                 mlx5_devx_cmd_destroy(mr->mkey);
90         if (mr->umem)
91                 mlx5_glue->devx_umem_dereg(mr->umem);
92         rte_free(mr);
93         return -1;
94 }
95
96 #define MLX5_VDPA_USED_RING_LEN(size) \
97         ((size) * sizeof(struct vring_used_elem) + sizeof(uint16_t) * 3)
98
99 int
100 mlx5_vdpa_lm_log(struct mlx5_vdpa_priv *priv)
101 {
102         uint64_t features;
103         int ret = rte_vhost_get_negotiated_features(priv->vid, &features);
104         int i;
105
106         if (ret) {
107                 DRV_LOG(ERR, "Failed to get negotiated features.");
108                 return -1;
109         }
110         if (!RTE_VHOST_NEED_LOG(features))
111                 return 0;
112         for (i = 0; i < priv->nr_virtqs; ++i) {
113                 if (!priv->virtqs[i].virtq) {
114                         DRV_LOG(DEBUG, "virtq %d is invalid for LM log.", i);
115                 } else {
116                         ret = mlx5_vdpa_virtq_stop(priv, i);
117                         if (ret) {
118                                 DRV_LOG(ERR, "Failed to stop virtq %d for LM "
119                                         "log.", i);
120                                 return -1;
121                         }
122                 }
123                 rte_vhost_log_used_vring(priv->vid, i, 0,
124                               MLX5_VDPA_USED_RING_LEN(priv->virtqs[i].vq_size));
125         }
126         return 0;
127 }