From 654685d1e4d5612eeb6b465f146e6987418551c0 Mon Sep 17 00:00:00 2001
From: HikariKnight <2557889+HikariKnight@users.noreply.github.com>
Date: Fri, 7 Apr 2023 14:06:06 +0200
Subject: [PATCH] Add some comments and a better error message if ls-iommu
 returns an error

---
 internal/tuimode/tuimode.go | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/internal/tuimode/tuimode.go b/internal/tuimode/tuimode.go
index 70b04b4..56c49f0 100644
--- a/internal/tuimode/tuimode.go
+++ b/internal/tuimode/tuimode.go
@@ -234,27 +234,30 @@ func App() {
 func GetIOMMU(args ...string) []list.Item {
 	var stdout, stderr bytes.Buffer
 
+	// Configure the ls-iommu command
 	cmd := exec.Command("utils/ls-iommu", args...)
 	cmd.Stderr = &stderr
 	cmd.Stdout = &stdout
 
+	// Execute the command
 	err := cmd.Run()
-	errorcheck.ErrorCheck(err)
 
+	// If ls-iommu returns an error then IOMMU is disabled
+	errorcheck.ErrorCheck(err, "IOMMU disabled in either UEFI/BIOS or in bootloader!")
+
+	// Read the output
 	items := []list.Item{}
 	output, _ := io.ReadAll(&stdout)
 
-	// If we get an error from ls-iommu then print the error and exit
-	// As the system might not have IOMMU enabled
-	if stderr.String() != "" {
-		log.Fatal(stderr.String())
-	}
-
+	// Parse the output line by line
 	scanner := bufio.NewScanner(strings.NewReader(string(output)))
 	for scanner.Scan() {
+		// Get the current line and split by :
 		objects := strings.Split(scanner.Text(), ": ")
+		// Write the objects into the list
 		items = append(items, item{title: objects[1], desc: objects[0]})
 	}
 
+	// Return our list of items
 	return items
 }