From 2590a7c8ad4d9ca00a8e32a52cd570b1de36771f Mon Sep 17 00:00:00 2001
From: Fabian Keil <fk@fabiankeil.de>
Date: Mon, 31 Aug 2015 10:51:16 +0200
Subject: [PATCH 062/325] release/scripts/image-checksum.sh: Allow to checksum
 multiple images at once

Obtained from: ElectroBSD
---
 release/scripts/image-checksum.sh | 79 +++++++++++++++++--------------
 1 file changed, 44 insertions(+), 35 deletions(-)

diff --git a/release/scripts/image-checksum.sh b/release/scripts/image-checksum.sh
index cc31e9055fe3..30ac62a6c168 100755
--- a/release/scripts/image-checksum.sh
+++ b/release/scripts/image-checksum.sh
@@ -87,42 +87,11 @@ partition_count_acceptable() {
     fi
 }
 
-main() {
+generate_partial_image_checksum() {
     local image_file \
-          args md_unit spec_file gpart_file
-
-    args=$(getopt m:rv $*)
-    if [ $? -ne 0 ]; then
-        echo 'You are doing it wrong: Invalid flag specified'
-        exit 2
-    fi
-    set -- ${args}
-    while true; do
-        case "$1" in
-            -m)
-                shift
-                MTREE_KEYWORDS="${1}"
-                shift
-                ;;
-            -r)
-                REUSE_EXISTING_CACHE_FILES=true
-                shift
-                ;;
-            -v)
-                VERBOSE=1
-                shift
-                ;;
-            --)
-                shift; break
-                ;;
-        esac
-    done
+          md_unit spec_file gpart_file
 
-    image_file=${1}
-    if [ -z "${image_file}" ]; then
-        echo "No image file provided"
-        return 1
-    fi
+    image_file="${1}"
     spec_file="${image_file}.mtree"
     if [ -f "${spec_file}" ]; then
         echo "Spec file ${spec_file} already exists"
@@ -159,8 +128,48 @@ main() {
     verbose_log "mtree checksum: ${mtree_checksum}"
 
     weak_image_checksum=$(echo "${gpart_checksum} ${bootcode_checksum} ${mtree_checksum}" | sha256)
-    echo "Partial image checksum: ${weak_image_checksum}"
+    echo "Partial image checksum for ${image_file}: ${weak_image_checksum}"
+}
 
+main() {
+    local image_file \
+          args
+
+    args=$(getopt m:rv $*)
+    if [ $? -ne 0 ]; then
+        echo 'You are doing it wrong: Invalid flag specified'
+        exit 2
+    fi
+    set -- ${args}
+    while true; do
+        case "$1" in
+            -m)
+                shift
+                MTREE_KEYWORDS="${1}"
+                shift
+                ;;
+            -r)
+                REUSE_EXISTING_CACHE_FILES=true
+                shift
+                ;;
+            -v)
+                VERBOSE=1
+                shift
+                ;;
+            --)
+                shift; break
+                ;;
+        esac
+    done
+
+    if [ -z "${1}" ]; then
+        echo "No image file provided"
+        return 1
+    fi
+
+    for image_file in "${@}"; do
+        generate_partial_image_checksum "${image_file}" || return 1
+    done
 }
 
 main "${@}"
-- 
2.32.0