From 3a8d00e1944d5262c428d0e1900748b01211f4cf Mon Sep 17 00:00:00 2001
From: HikariKnight <2557889+HikariKnight@users.noreply.github.com>
Date: Tue, 1 Mar 2022 13:07:56 +0100
Subject: [PATCH] detect if IOMMU is properly enabled

---
 vfio-setup | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/vfio-setup b/vfio-setup
index 36decf5..956b73c 100755
--- a/vfio-setup
+++ b/vfio-setup
@@ -111,22 +111,36 @@ else
 fi
     
 # Run ls-iommu so we can verify that IOMMU properly working
-"$SCRIPTDIR/utils/ls-iommu"
+LS_IOMMU=$($SCRIPTDIR/utils/ls-iommu)
 
-printf "
-If the above list has everything inside just 1 IOMMU group the please enable
-IOMMU, VT-d or AMD-v inside your UEFI/BIOS and add \"$CMDLINE\"
+if [[ $LS_IOMMU =~ ^IOMMU[[:space:]]Group[[:space:]]+\*: ]];
+then
+    printf "IOMMU IS NOT ENABLED!
+Please enable IOMMU, VT-d or AMD-v inside your UEFI/BIOS and add \"$CMDLINE\"
 to your kernel boot arguments and reboot your system, then re-run this script!
 "
+    exit 1
+
+else
+    echo "$LS_IOMMU"
+fi
+
+echo ""
 read -p "Is there more than 1 group in the output above? [y/N]: " YESNO
 case "${YESNO}" in
     [Yy]*)
         clear
     ;;
     [Nn]*)
+        printf "Please enable IOMMU, VT-d or AMD-v inside your UEFI/BIOS and add \"$CMDLINE\"
+to your kernel boot arguments and reboot your system, then re-run this script!
+"
         exit 1
     ;;
     *)
+        printf "Please enable IOMMU, VT-d or AMD-v inside your UEFI/BIOS and add \"$CMDLINE\"
+to your kernel boot arguments and reboot your system, then re-run this script!
+"
         exit 1
     ;;
 esac