[RFC PATCH v2 08/32] iommu: Register IOMMU FLB with iommufd file handler

[Samiullah Khawaja]

Use the IOMMU FLB register/unregister API to associate it with iommufd
LU file handler.

Signed-off-by: Samiullah Khawaja <skhawaja@xxxxxxxxxx>
---
 drivers/iommu/iommufd/liveupdate.c | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/drivers/iommu/iommufd/liveupdate.c b/drivers/iommu/iommufd/liveupdate.c
index 42b380229c57..782585aff44a 100644
--- a/drivers/iommu/iommufd/liveupdate.c
+++ b/drivers/iommu/iommufd/liveupdate.c
@@ -8,6 +8,7 @@
 #include <linux/kexec_handover.h>
 #include <linux/kho/abi/iommufd.h>
 #include <linux/liveupdate.h>
+#include <linux/iommu-lu.h>
 #include <linux/mm.h>
 #include <linux/pci.h>
 
@@ -362,10 +363,22 @@ static struct liveupdate_file_handler iommufd_lu_handler = {
 
 int iommufd_liveupdate_register_lufs(void)
 {
-	return liveupdate_register_file_handler(&iommufd_lu_handler);
+	int ret;
+
+	ret = liveupdate_register_file_handler(&iommufd_lu_handler);
+	if (ret)
+		return ret;
+
+	ret = iommu_liveupdate_register_flb(&iommufd_lu_handler);
+	if (ret)
+		liveupdate_unregister_file_handler(&iommufd_lu_handler);
+
+	return ret;
 }
 
 int iommufd_liveupdate_unregister_lufs(void)
 {
+	WARN_ON(iommu_liveupdate_unregister_flb(&iommufd_lu_handler));
+
 	return liveupdate_unregister_file_handler(&iommufd_lu_handler);
 }
-- 
2.52.0.158.g65b55ccf14-goog