From 4fe4e2196954be5caa13ebb799c7080938f09794 Mon Sep 17 00:00:00 2001
From: The-going <48602507+The-going@users.noreply.github.com>
Date: Fri, 20 Jan 2023 14:15:52 +0300
Subject: [PATCH] fix: pass an argument to the awk script as a variable

MonitorIO: Add an argument check and a hint to the user
---
 packages/bsp/common/usr/bin/armbianmonitor | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/packages/bsp/common/usr/bin/armbianmonitor b/packages/bsp/common/usr/bin/armbianmonitor
index 916b626c37..962f0cb6cc 100755
--- a/packages/bsp/common/usr/bin/armbianmonitor
+++ b/packages/bsp/common/usr/bin/armbianmonitor
@@ -503,13 +503,21 @@ ProcessStats() {
 } # ProcessStats
 
 MonitorIO() {
-	LastPagesOut=$(awk -F" " '/pgpgout/ {print $2}' </proc/vmstat)
-	LastWrite=$(awk -F" " "/ $1 / {print \$8}" </proc/diskstats)
+	LastPagesOut=$(awk '/pgpgout/ {print $2}' </proc/vmstat)
+	if grep -q "$1" /proc/diskstats; then
+		LastWrite=$(awk -v d="$1" '{if($3 == d) print $8}' </proc/diskstats)
+	else
+		echo "Bud argument: [$1]"
+		echo "Disks valid for monitoring: $(
+			awk '{if($8 != 0) printf "%s ", $3}' /proc/diskstats
+			)"
+		exit 1
+	fi
 	LastTimeChecked=$(date "+%s")
 	while true ; do
-		CurrentWrite=$(awk -F" " "/ $1 / {print \$8}" </proc/diskstats)
+		CurrentWrite=$(awk -v d="$1" '{if($3 == d) print $8}' </proc/diskstats)
 		if [ ${CurrentWrite} -gt ${LastWrite} ]; then
-			PagesOut=$(awk -F" " '/pgpgout/ {print $2}' </proc/vmstat)
+			PagesOut=$(awk '/pgpgout/ {print $2}' </proc/vmstat)
 			TimeNow=$(date "+%s")
 			PagesWritten=$((CurrentWrite - LastWrite))
 			PageOuts=$((PagesOut - LastPagesOut))