1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright 2019 Mellanox Technologies, Ltd
6 #include <rte_malloc.h>
9 #include <rte_bus_pci.h>
12 #include <mlx5_glue.h>
13 #include <mlx5_common.h>
14 #include <mlx5_devx_cmds.h>
18 #include "mlx5_vdpa_utils.h"
19 #include "mlx5_vdpa.h"
22 #define MLX5_VDPA_DEFAULT_FEATURES ((1ULL << VHOST_USER_F_PROTOCOL_FEATURES) | \
23 (1ULL << VIRTIO_F_ANY_LAYOUT) | \
24 (1ULL << VIRTIO_NET_F_MQ) | \
25 (1ULL << VIRTIO_NET_F_GUEST_ANNOUNCE) | \
26 (1ULL << VIRTIO_F_ORDER_PLATFORM) | \
27 (1ULL << VHOST_F_LOG_ALL))
29 #define MLX5_VDPA_PROTOCOL_FEATURES \
30 ((1ULL << VHOST_USER_PROTOCOL_F_SLAVE_REQ) | \
31 (1ULL << VHOST_USER_PROTOCOL_F_SLAVE_SEND_FD) | \
32 (1ULL << VHOST_USER_PROTOCOL_F_HOST_NOTIFIER) | \
33 (1ULL << VHOST_USER_PROTOCOL_F_LOG_SHMFD) | \
34 (1ULL << VHOST_USER_PROTOCOL_F_MQ))
36 TAILQ_HEAD(mlx5_vdpa_privs, mlx5_vdpa_priv) priv_list =
37 TAILQ_HEAD_INITIALIZER(priv_list);
38 static pthread_mutex_t priv_list_lock = PTHREAD_MUTEX_INITIALIZER;
39 int mlx5_vdpa_logtype;
41 static struct mlx5_vdpa_priv *
42 mlx5_vdpa_find_priv_resource_by_did(int did)
44 struct mlx5_vdpa_priv *priv;
47 pthread_mutex_lock(&priv_list_lock);
48 TAILQ_FOREACH(priv, &priv_list, next) {
49 if (did == priv->id) {
54 pthread_mutex_unlock(&priv_list_lock);
56 DRV_LOG(ERR, "Invalid device id: %d.", did);
64 mlx5_vdpa_get_queue_num(int did, uint32_t *queue_num)
66 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
69 DRV_LOG(ERR, "Invalid device id: %d.", did);
72 *queue_num = priv->caps.max_num_virtio_queues;
77 mlx5_vdpa_get_vdpa_features(int did, uint64_t *features)
79 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
82 DRV_LOG(ERR, "Invalid device id: %d.", did);
85 *features = MLX5_VDPA_DEFAULT_FEATURES;
86 if (priv->caps.virtio_queue_type & (1 << MLX5_VIRTQ_TYPE_PACKED))
87 *features |= (1ULL << VIRTIO_F_RING_PACKED);
88 if (priv->caps.tso_ipv4)
89 *features |= (1ULL << VIRTIO_NET_F_HOST_TSO4);
90 if (priv->caps.tso_ipv6)
91 *features |= (1ULL << VIRTIO_NET_F_HOST_TSO6);
92 if (priv->caps.tx_csum)
93 *features |= (1ULL << VIRTIO_NET_F_CSUM);
94 if (priv->caps.rx_csum)
95 *features |= (1ULL << VIRTIO_NET_F_GUEST_CSUM);
96 if (priv->caps.virtio_version_1_0)
97 *features |= (1ULL << VIRTIO_F_VERSION_1);
102 mlx5_vdpa_get_protocol_features(int did, uint64_t *features)
104 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
107 DRV_LOG(ERR, "Invalid device id: %d.", did);
110 *features = MLX5_VDPA_PROTOCOL_FEATURES;
115 mlx5_vdpa_set_vring_state(int vid, int vring, int state)
117 int did = rte_vhost_get_vdpa_device_id(vid);
118 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
121 DRV_LOG(ERR, "Invalid device id: %d.", did);
124 if (!priv->configured || vring >= RTE_MIN((int)priv->nr_virtqs,
125 (int)priv->caps.max_num_virtio_queues * 2) ||
126 !priv->virtqs[vring].virtq) {
127 DRV_LOG(ERR, "Invalid or unconfigured vring id: %d.", vring);
130 return mlx5_vdpa_virtq_enable(&priv->virtqs[vring], state);
134 mlx5_vdpa_direct_db_prepare(struct mlx5_vdpa_priv *priv)
138 if (priv->direct_notifier) {
139 ret = rte_vhost_host_notifier_ctrl(priv->vid, false);
141 DRV_LOG(INFO, "Direct HW notifier FD cannot be "
142 "destroyed for device %d: %d.", priv->vid, ret);
145 priv->direct_notifier = 0;
147 ret = rte_vhost_host_notifier_ctrl(priv->vid, true);
149 DRV_LOG(INFO, "Direct HW notifier FD cannot be configured for"
150 " device %d: %d.", priv->vid, ret);
152 priv->direct_notifier = 1;
157 mlx5_vdpa_features_set(int vid)
159 int did = rte_vhost_get_vdpa_device_id(vid);
160 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
161 uint64_t log_base, log_size;
166 DRV_LOG(ERR, "Invalid device id: %d.", did);
169 ret = rte_vhost_get_negotiated_features(vid, &features);
171 DRV_LOG(ERR, "Failed to get negotiated features.");
174 if (RTE_VHOST_NEED_LOG(features)) {
175 ret = rte_vhost_get_log_base(vid, &log_base, &log_size);
177 DRV_LOG(ERR, "Failed to get log base.");
180 ret = mlx5_vdpa_dirty_bitmap_set(priv, log_base, log_size);
182 DRV_LOG(ERR, "Failed to set dirty bitmap.");
185 DRV_LOG(INFO, "mlx5 vdpa: enabling dirty logging...");
186 ret = mlx5_vdpa_logging_enable(priv, 1);
188 DRV_LOG(ERR, "Failed t enable dirty logging.");
196 mlx5_vdpa_dev_close(int vid)
198 int did = rte_vhost_get_vdpa_device_id(vid);
199 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
203 DRV_LOG(ERR, "Invalid device id: %d.", did);
206 if (priv->configured)
207 ret |= mlx5_vdpa_lm_log(priv);
208 mlx5_vdpa_cqe_event_unset(priv);
209 ret |= mlx5_vdpa_steer_unset(priv);
210 mlx5_vdpa_virtqs_release(priv);
211 mlx5_vdpa_event_qp_global_release(priv);
212 mlx5_vdpa_mem_dereg(priv);
213 priv->configured = 0;
215 DRV_LOG(INFO, "vDPA device %d was closed.", vid);
220 mlx5_vdpa_dev_config(int vid)
222 int did = rte_vhost_get_vdpa_device_id(vid);
223 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
226 DRV_LOG(ERR, "Invalid device id: %d.", did);
229 if (priv->configured && mlx5_vdpa_dev_close(vid)) {
230 DRV_LOG(ERR, "Failed to reconfigure vid %d.", vid);
234 if (mlx5_vdpa_mem_register(priv) || mlx5_vdpa_direct_db_prepare(priv) ||
235 mlx5_vdpa_virtqs_prepare(priv) || mlx5_vdpa_steer_setup(priv) ||
236 mlx5_vdpa_cqe_event_setup(priv)) {
237 mlx5_vdpa_dev_close(vid);
240 priv->configured = 1;
241 DRV_LOG(INFO, "vDPA device %d was configured.", vid);
246 mlx5_vdpa_get_device_fd(int vid)
248 int did = rte_vhost_get_vdpa_device_id(vid);
249 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
252 DRV_LOG(ERR, "Invalid device id: %d.", did);
255 return priv->ctx->cmd_fd;
259 mlx5_vdpa_get_notify_area(int vid, int qid, uint64_t *offset, uint64_t *size)
261 int did = rte_vhost_get_vdpa_device_id(vid);
262 struct mlx5_vdpa_priv *priv = mlx5_vdpa_find_priv_resource_by_did(did);
266 DRV_LOG(ERR, "Invalid device id: %d.", did);
270 DRV_LOG(ERR, "VAR was not created for device %d, is the device"
271 " configured?.", did);
274 *offset = priv->var->mmap_off;
275 *size = priv->var->length;
279 static struct rte_vdpa_dev_ops mlx5_vdpa_ops = {
280 .get_queue_num = mlx5_vdpa_get_queue_num,
281 .get_features = mlx5_vdpa_get_vdpa_features,
282 .get_protocol_features = mlx5_vdpa_get_protocol_features,
283 .dev_conf = mlx5_vdpa_dev_config,
284 .dev_close = mlx5_vdpa_dev_close,
285 .set_vring_state = mlx5_vdpa_set_vring_state,
286 .set_features = mlx5_vdpa_features_set,
287 .migration_done = NULL,
288 .get_vfio_group_fd = NULL,
289 .get_vfio_device_fd = mlx5_vdpa_get_device_fd,
290 .get_notify_area = mlx5_vdpa_get_notify_area,
293 static struct ibv_device *
294 mlx5_vdpa_get_ib_device_match(struct rte_pci_addr *addr)
297 struct ibv_device **ibv_list = mlx5_glue->get_device_list(&n);
298 struct ibv_device *ibv_match = NULL;
305 struct rte_pci_addr pci_addr;
307 DRV_LOG(DEBUG, "Checking device \"%s\"..", ibv_list[n]->name);
308 if (mlx5_dev_to_pci_addr(ibv_list[n]->ibdev_path, &pci_addr))
310 if (memcmp(addr, &pci_addr, sizeof(pci_addr)))
312 ibv_match = ibv_list[n];
317 mlx5_glue->free_device_list(ibv_list);
321 /* Try to disable ROCE by Netlink\Devlink. */
323 mlx5_vdpa_nl_roce_disable(const char *addr)
325 int nlsk_fd = mlx5_nl_init(NETLINK_GENERIC);
332 devlink_id = mlx5_nl_devlink_family_id_get(nlsk_fd);
333 if (devlink_id < 0) {
335 DRV_LOG(DEBUG, "Failed to get devlink id for ROCE operations by"
339 ret = mlx5_nl_enable_roce_get(nlsk_fd, devlink_id, addr, &enable);
341 DRV_LOG(DEBUG, "Failed to get ROCE enable by Netlink: %d.",
344 } else if (!enable) {
345 DRV_LOG(INFO, "ROCE has already disabled(Netlink).");
348 ret = mlx5_nl_enable_roce_set(nlsk_fd, devlink_id, addr, 0);
350 DRV_LOG(DEBUG, "Failed to disable ROCE by Netlink: %d.", ret);
352 DRV_LOG(INFO, "ROCE is disabled by Netlink successfully.");
358 /* Try to disable ROCE by sysfs. */
360 mlx5_vdpa_sys_roce_disable(const char *addr)
366 MKSTR(file_p, "/sys/bus/pci/devices/%s/roce_enable", addr);
367 file_o = fopen(file_p, "rb");
372 ret = fscanf(file_o, "%d", &enable);
377 } else if (!enable) {
379 DRV_LOG(INFO, "ROCE has already disabled(sysfs).");
383 file_o = fopen(file_p, "wb");
388 fprintf(file_o, "0\n");
392 DRV_LOG(DEBUG, "Failed to disable ROCE by sysfs: %d.", ret);
394 DRV_LOG(INFO, "ROCE is disabled by sysfs successfully.");
399 #define MLX5_VDPA_MAX_RETRIES 20
400 #define MLX5_VDPA_USEC 1000
402 mlx5_vdpa_roce_disable(struct rte_pci_addr *addr, struct ibv_device **ibv)
404 char addr_name[64] = {0};
406 rte_pci_device_name(addr, addr_name, sizeof(addr_name));
407 /* Firstly try to disable ROCE by Netlink and fallback to sysfs. */
408 if (mlx5_vdpa_nl_roce_disable(addr_name) == 0 ||
409 mlx5_vdpa_sys_roce_disable(addr_name) == 0) {
411 * Succeed to disable ROCE, wait for the IB device to appear
412 * again after reload.
415 struct ibv_device *ibv_new;
417 for (r = MLX5_VDPA_MAX_RETRIES; r; r--) {
418 ibv_new = mlx5_vdpa_get_ib_device_match(addr);
423 usleep(MLX5_VDPA_USEC);
425 DRV_LOG(ERR, "Cannot much device %s after ROCE disable, "
426 "retries exceed %d", addr_name, MLX5_VDPA_MAX_RETRIES);
433 * DPDK callback to register a PCI device.
435 * This function spawns vdpa device out of a given PCI device.
438 * PCI driver structure (mlx5_vpda_driver).
440 * PCI device information.
443 * 0 on success, 1 to skip this driver, a negative errno value otherwise
444 * and rte_errno is set.
447 mlx5_vdpa_pci_probe(struct rte_pci_driver *pci_drv __rte_unused,
448 struct rte_pci_device *pci_dev __rte_unused)
450 struct ibv_device *ibv;
451 struct mlx5_vdpa_priv *priv = NULL;
452 struct ibv_context *ctx = NULL;
453 struct mlx5_hca_attr attr;
456 if (mlx5_class_get(pci_dev->device.devargs) != MLX5_CLASS_VDPA) {
457 DRV_LOG(DEBUG, "Skip probing - should be probed by other mlx5"
461 ibv = mlx5_vdpa_get_ib_device_match(&pci_dev->addr);
463 DRV_LOG(ERR, "No matching IB device for PCI slot "
464 PCI_PRI_FMT ".", pci_dev->addr.domain,
465 pci_dev->addr.bus, pci_dev->addr.devid,
466 pci_dev->addr.function);
469 DRV_LOG(INFO, "PCI information matches for device \"%s\".",
472 if (mlx5_vdpa_roce_disable(&pci_dev->addr, &ibv) != 0) {
473 DRV_LOG(WARNING, "Failed to disable ROCE for \"%s\".",
477 ctx = mlx5_glue->dv_open_device(ibv);
479 DRV_LOG(ERR, "Failed to open IB device \"%s\".", ibv->name);
483 ret = mlx5_devx_cmd_query_hca_attr(ctx, &attr);
485 DRV_LOG(ERR, "Unable to read HCA capabilities.");
488 } else if (!attr.vdpa.valid || !attr.vdpa.max_num_virtio_queues) {
489 DRV_LOG(ERR, "Not enough capabilities to support vdpa, maybe "
490 "old FW/OFED version?");
494 priv = rte_zmalloc("mlx5 vDPA device private", sizeof(*priv) +
495 sizeof(struct mlx5_vdpa_virtq) *
496 attr.vdpa.max_num_virtio_queues * 2,
497 RTE_CACHE_LINE_SIZE);
499 DRV_LOG(ERR, "Failed to allocate private memory.");
503 priv->caps = attr.vdpa;
504 priv->log_max_rqt_size = attr.log_max_rqt_size;
506 priv->dev_addr.pci_addr = pci_dev->addr;
507 priv->dev_addr.type = VDPA_ADDR_PCI;
508 priv->var = mlx5_glue->dv_alloc_var(ctx, 0);
510 DRV_LOG(ERR, "Failed to allocate VAR %u.\n", errno);
513 priv->id = rte_vdpa_register_device(&priv->dev_addr, &mlx5_vdpa_ops);
515 DRV_LOG(ERR, "Failed to register vDPA device.");
516 rte_errno = rte_errno ? rte_errno : EINVAL;
519 SLIST_INIT(&priv->mr_list);
520 pthread_mutex_lock(&priv_list_lock);
521 TAILQ_INSERT_TAIL(&priv_list, priv, next);
522 pthread_mutex_unlock(&priv_list_lock);
528 mlx5_glue->dv_free_var(priv->var);
532 mlx5_glue->close_device(ctx);
537 * DPDK callback to remove a PCI device.
539 * This function removes all vDPA devices belong to a given PCI device.
542 * Pointer to the PCI device.
545 * 0 on success, the function cannot fail.
548 mlx5_vdpa_pci_remove(struct rte_pci_device *pci_dev)
550 struct mlx5_vdpa_priv *priv = NULL;
553 pthread_mutex_lock(&priv_list_lock);
554 TAILQ_FOREACH(priv, &priv_list, next) {
555 if (memcmp(&priv->dev_addr.pci_addr, &pci_dev->addr,
556 sizeof(pci_dev->addr)) == 0) {
562 TAILQ_REMOVE(&priv_list, priv, next);
563 pthread_mutex_unlock(&priv_list_lock);
565 if (priv->configured)
566 mlx5_vdpa_dev_close(priv->vid);
568 mlx5_glue->dv_free_var(priv->var);
571 mlx5_glue->close_device(priv->ctx);
577 static const struct rte_pci_id mlx5_vdpa_pci_id_map[] = {
579 RTE_PCI_DEVICE(PCI_VENDOR_ID_MELLANOX,
580 PCI_DEVICE_ID_MELLANOX_CONNECTX6)
583 RTE_PCI_DEVICE(PCI_VENDOR_ID_MELLANOX,
584 PCI_DEVICE_ID_MELLANOX_CONNECTX6VF)
587 RTE_PCI_DEVICE(PCI_VENDOR_ID_MELLANOX,
588 PCI_DEVICE_ID_MELLANOX_CONNECTX6DX)
591 RTE_PCI_DEVICE(PCI_VENDOR_ID_MELLANOX,
592 PCI_DEVICE_ID_MELLANOX_CONNECTX6DXVF)
595 RTE_PCI_DEVICE(PCI_VENDOR_ID_MELLANOX,
596 PCI_DEVICE_ID_MELLANOX_CONNECTX6DXBF)
603 static struct rte_pci_driver mlx5_vdpa_driver = {
607 .id_table = mlx5_vdpa_pci_id_map,
608 .probe = mlx5_vdpa_pci_probe,
609 .remove = mlx5_vdpa_pci_remove,
614 * Driver initialization routine.
616 RTE_INIT(rte_mlx5_vdpa_init)
618 /* Initialize common log type. */
619 mlx5_vdpa_logtype = rte_log_register("pmd.vdpa.mlx5");
620 if (mlx5_vdpa_logtype >= 0)
621 rte_log_set_level(mlx5_vdpa_logtype, RTE_LOG_NOTICE);
623 rte_pci_register(&mlx5_vdpa_driver);
626 RTE_PMD_EXPORT_NAME(net_mlx5_vdpa, __COUNTER__);
627 RTE_PMD_REGISTER_PCI_TABLE(net_mlx5_vdpa, mlx5_vdpa_pci_id_map);
628 RTE_PMD_REGISTER_KMOD_DEP(net_mlx5_vdpa, "* ib_uverbs & mlx5_core & mlx5_ib");