From a9696ac713e12833181fe0c23d974a6edfe95b61 Mon Sep 17 00:00:00 2001
From: bryan <bryanroessler@gmail.com>
Date: Fri, 13 Jan 2023 21:55:45 -0500
Subject: [PATCH] Refactor verifyImage

---
 openwrtbuilder | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/openwrtbuilder b/openwrtbuilder
index 162e7c0..25728cb 100755
--- a/openwrtbuilder
+++ b/openwrtbuilder
@@ -375,7 +375,7 @@ getImageBuilder() {
 verifyImage() {
     debug "${FUNCNAME[0]}"
 
-    declare checksum
+    declare checksum file
 
     if hash sha256sum &>/dev/null; then
         debug "Verifying Image Builder checksums"
@@ -388,13 +388,12 @@ verifyImage() {
             debug "Failed to download checksum"
         fi
 
-        checksum=$(grep "${IB_ARCHIVE##*/}" "$SHA256_FILE") # | cut -f1 -d' ')
-
-        debug "checksum: $checksum"
-
-        echo "$checksum ${IB_ARCHIVE##*/}" | sha256sum --check --status
-
-        debug "checksum return code: $?"
+        for file in "$FACTORYIMGGZFNAME" "$SYSUPGRADEIMGGZFNAME"; do
+            checksum=$(grep "$file" "$SHA256_FILE" | cut -f1 -d' ')
+            debug "checksum: $checksum"
+            echo "$checksum $file" | sha256sum --check --status
+            debug "checksum return code: $?"
+        done
     fi
 }
 
@@ -719,6 +718,7 @@ main() {
 
         declare -g FACTORYIMG="$img_prefix-factory.img"
         declare -g FACTORYIMGGZ="$img_prefix-factory.img.gz"
+        declare -g FACTORYIMGGZFNAME="${FACTORYIMGGZ##*/}"
         declare -g SYSUPGRADEIMG="$img_prefix-sysupgrade.img"
         declare -g SYSUPGRADEIMGGZ="$img_prefix-sysupgrade.img.gz"
         declare -g SYSUPGRADEIMGGZFNAME="${SYSUPGRADEIMGGZ##*/}"