From f14d849862946c8bc241b901706929a2920005e8 Mon Sep 17 00:00:00 2001
From: Hans Petter Selasky <hselasky@FreeBSD.org>
Date: Wed, 12 Feb 2020 09:58:19 +0000
Subject: [PATCH] Add support for disabling and polling MSIX interrupts in
 mlx5core.

MFC after:	1 week
Sponsored by:	Mellanox Technologies
---
 sys/dev/mlx5/driver.h              |  3 +++
 sys/dev/mlx5/mlx5_core/mlx5_eq.c   | 25 +++++++++++++++++++++++++
 sys/dev/mlx5/mlx5_core/mlx5_main.c |  4 ++--
 3 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/sys/dev/mlx5/driver.h b/sys/dev/mlx5/driver.h
index e2a68725751..b9d453e6ebd 100644
--- a/sys/dev/mlx5/driver.h
+++ b/sys/dev/mlx5/driver.h
@@ -1180,4 +1180,7 @@ static inline bool mlx5_rl_is_supported(struct mlx5_core_dev *dev)
 }
 #endif
 
+void mlx5_disable_interrupts(struct mlx5_core_dev *);
+void mlx5_poll_interrupts(struct mlx5_core_dev *);
+
 #endif /* MLX5_DRIVER_H */
diff --git a/sys/dev/mlx5/mlx5_core/mlx5_eq.c b/sys/dev/mlx5/mlx5_core/mlx5_eq.c
index c83973cc14b..0aed35fec71 100644
--- a/sys/dev/mlx5/mlx5_core/mlx5_eq.c
+++ b/sys/dev/mlx5/mlx5_core/mlx5_eq.c
@@ -739,3 +739,28 @@ static void mlx5_port_general_notification_event(struct mlx5_core_dev *dev,
 	}
 }
 
+void
+mlx5_disable_interrupts(struct mlx5_core_dev *dev)
+{
+	int nvec = dev->priv.eq_table.num_comp_vectors + MLX5_EQ_VEC_COMP_BASE;
+	int x;
+
+	for (x = 0; x != nvec; x++)
+		disable_irq(dev->priv.msix_arr[x].vector);
+}
+
+void
+mlx5_poll_interrupts(struct mlx5_core_dev *dev)
+{
+	struct mlx5_eq *eq;
+
+	if (unlikely(dev->priv.disable_irqs != 0))
+		return;
+
+	mlx5_eq_int(dev, &dev->priv.eq_table.cmd_eq);
+	mlx5_eq_int(dev, &dev->priv.eq_table.async_eq);
+	mlx5_eq_int(dev, &dev->priv.eq_table.pages_eq);
+
+	list_for_each_entry(eq, &dev->priv.eq_table.comp_eqs_list, list)
+		mlx5_eq_int(dev, eq);
+}
diff --git a/sys/dev/mlx5/mlx5_core/mlx5_main.c b/sys/dev/mlx5/mlx5_core/mlx5_main.c
index b13fa93ae58..f5809a28540 100644
--- a/sys/dev/mlx5/mlx5_core/mlx5_main.c
+++ b/sys/dev/mlx5/mlx5_core/mlx5_main.c
@@ -1585,7 +1585,7 @@ done:
 	return 0;
 }
 
-static void mlx5_disable_interrupts(struct mlx5_core_dev *mdev)
+static void mlx5_shutdown_disable_interrupts(struct mlx5_core_dev *mdev)
 {
 	int nvec = mdev->priv.eq_table.num_comp_vectors + MLX5_EQ_VEC_COMP_BASE;
 	int x;
@@ -1609,7 +1609,7 @@ static void shutdown_one(struct pci_dev *pdev)
 	set_bit(MLX5_INTERFACE_STATE_TEARDOWN, &dev->intf_state);
 
 	/* disable all interrupts */
-	mlx5_disable_interrupts(dev);
+	mlx5_shutdown_disable_interrupts(dev);
 
 	err = mlx5_try_fast_unload(dev);
 	if (err)